From be8a1c8994497facc13d5fbff8e41fc9f0a4b277 Mon Sep 17 00:00:00 2001
From: Krad <kradchen123@163.com>
Date: Thu, 5 May 2022 15:54:10 +0800
Subject: [PATCH] Add transform matrix to Property.

---
 src/src/base/ExtendMedicalImageProperties.h | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/src/src/base/ExtendMedicalImageProperties.h b/src/src/base/ExtendMedicalImageProperties.h
index 15b1296..8e7771b 100644
--- a/src/src/base/ExtendMedicalImageProperties.h
+++ b/src/src/base/ExtendMedicalImageProperties.h
@@ -7,6 +7,8 @@
 
 #include <vtkMedicalImageProperties.h>
 #include <vector>
+#include <vtkNew.h>
+#include <vtkMatrix4x4.h>
 
 class ExtendMedicalImageProperties: public vtkMedicalImageProperties{
 public:
@@ -72,6 +74,12 @@ public:
     void SetUniqueID(const std::string& id){
         uniqueID = id;
     }
+    vtkMatrix4x4* GetWorldToModelMatrix(){
+        return worldToModelMatrix.Get();
+    }
+    vtkMatrix4x4* GetModelToWorldMatrix(){
+        return modelToWorldMatrix.Get();
+    }
 
 protected:
     ExtendMedicalImageProperties();
@@ -84,6 +92,8 @@ protected:
     unsigned short SamplePerPixel;
     std::vector<std::string> FileNames;
     std::string uniqueID;
+    vtkNew<vtkMatrix4x4> worldToModelMatrix;
+    vtkNew<vtkMatrix4x4> modelToWorldMatrix;
 private:
     ExtendMedicalImageProperties(const ExtendMedicalImageProperties&) = delete;
     void operator=(const ExtendMedicalImageProperties&) = delete;