Integration of new functionalities into ProcessObject and DataObject

[cpPlugins.git] / lib / cpPlugins / Interface / Image.h

diff --git a/lib/cpPlugins/Interface/Image.h b/lib/cpPlugins/Interface/Image.h
index cbdc219f7590e296f375204abbaffc360e35b323..e6b2b7762708360c5781af9651fe1f6ffc91212d 100644 (file)
--- a/lib/cpPlugins/Interface/Image.h
+++ b/lib/cpPlugins/Interface/Image.h
@@ -7,6 +7,8 @@
 #include <cpPlugins/Interface/cpPlugins_Interface_Export.h>
 #include <cpPlugins/Interface/DataObject.h>
 
+class vtkImageData;
+
 namespace cpPlugins
 {
   namespace Interface
@@ -25,6 +27,20 @@ namespace cpPlugins
       virtual ~Image( );
 
       virtual std::string GetClassName( ) const;
+      virtual void SetDataObject( itk::DataObject* dobj );
+
+      vtkImageData* GetVTKImageData( ) const;
+
+    protected:
+      template< unsigned int D >
+      void _ConnectToVTK_0( );
+
+      template< class P, unsigned int D >
+      void _ConnectToVTK_1( );
+
+    protected:
+      itk::ProcessObject::Pointer m_Image2VTKImageData;
+      vtkImageData* m_VTKImageData;
     };
 
   } // ecapseman