Enable resliced images, put in mTransformedVtkImages

[clitk.git] / common / vvImage.h

diff --git a/common/vvImage.h b/common/vvImage.h
index 2d4776d728cb9707a70ae2dc22c7a34ba27e810e..5c38f31089e554844536385b45c112854899b5a5 100644 (file)
--- a/common/vvImage.h
+++ b/common/vvImage.h
@@ -22,11 +22,12 @@
 #include <vector>
 #include <itkObjectFactory.h>
 #include <itkProcessObject.h>
+
+#include <vtkImageReslice.h>
 #include <vtkSmartPointer.h>
+#include <vtkTransform.h>
 
 class vtkImageData;
-class vtkImageReslice;
-class vtkTransform;
 
 //------------------------------------------------------------------------------
 class vvImage : public itk::LightObject
@@ -41,6 +42,7 @@ public :
   void Reset();
   template<class TItkImageType> void AddItkImage(TItkImageType *input);
   const std::vector<vtkImageData*>& GetVTKImages();
+  const std::vector<vtkImageData*>& GetTransformedVTKImages();
   vtkImageData* GetFirstVTKImageData();
   int GetNumberOfDimensions() const;
   int GetNumberOfSpatialDimensions();
@@ -66,8 +68,11 @@ private:
 
   std::vector< ConverterPointer > mItkToVtkConverters;
   std::vector<vtkImageData*> mVtkImages;
+
   std::vector< vtkSmartPointer<vtkImageReslice> > mVtkImageReslice;
   vtkSmartPointer<vtkTransform> mTransform;
+  std::vector<vtkImageData*> mTransformedVtkImages;
+
   double mTimeOrigin;
   double mTimeSpacing;
   unsigned int mImageDimension;