X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;ds=sidebyside;f=common%2FvvImage.h;h=cc0ce76028dd3734eeeb342aa80c9de996846d05;hb=1520bb5daceae6c3a6ec221c60a462a02d5c08ab;hp=6a30a454569d67021084ba8757c294c78f1b3088;hpb=e11a1cffe5f20b584f80d534d400b00afc594146;p=clitk.git

diff --git a/common/vvImage.h b/common/vvImage.h
index 6a30a45..cc0ce76 100644
--- a/common/vvImage.h
+++ b/common/vvImage.h
@@ -21,6 +21,7 @@
 #include <iostream>
 #include <vector>
 #include <itkObjectFactory.h>
+#include <itkProcessObject.h>
 #include <vtkSmartPointer.h>
 
 class vtkImageData;
@@ -33,12 +34,12 @@ class vvImage : public itk::LightObject
 public :
   typedef vvImage Self;
   typedef itk::SmartPointer<Self> Pointer;
+  typedef itk::ProcessObject::Pointer ConverterPointer;
   itkNewMacro(Self);
 
   void Init();
   void Reset();
-  void SetImage(std::vector<vtkImageData*> images);
-  void AddImage(vtkImageData* image);
+  template<class TItkImageType> void AddItkImage(TItkImageType *input);
   const std::vector<vtkImageData*>& GetVTKImages();
   vtkImageData* GetFirstVTKImageData();
   int GetNumberOfDimensions() const;
@@ -63,6 +64,7 @@ private:
   vvImage();
   ~vvImage();
 
+  std::vector< ConverterPointer > mItkToVtkConverters;
   std::vector<vtkImageData*> mVtkImages;
   std::vector< vtkSmartPointer<vtkImageReslice> > mVtkImageReslice;
   vtkSmartPointer<vtkTransform> mTransform;
@@ -71,8 +73,6 @@ private:
 };
 //------------------------------------------------------------------------------
 
-//------------------------------------------------------------------------------
-vtkImageData * CopyAndCastToFloatFrom(vtkImageData * p);
-//------------------------------------------------------------------------------
+#include "vvImage.txx"
 
 #endif