...

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

diff --git a/lib/cpPlugins/Interface/Image.h b/lib/cpPlugins/Interface/Image.h
index 729acdf682aaaa382d4e50db41ab40dadda85666..45b47faf37175195f59d7249eafa4b2721c70464 100644 (file)
--- a/lib/cpPlugins/Interface/Image.h
+++ b/lib/cpPlugins/Interface/Image.h
@@ -5,9 +5,6 @@
 
 #include <itkProcessObject.h>
 
-// Some forward declarations
-class vtkObject;
-
 namespace cpPlugins
 {
   namespace Interface
@@ -29,17 +26,21 @@ namespace cpPlugins
       cpPlugins_Id_Macro( Image, DataObject );
 
     public:
-      template< class I >
-        inline void SetITK( itk::Object* object );
-
-      virtual void SetVTK( vtkObject* image );
+      virtual void SetITK( itk::LightObject* o );
+      virtual void SetVTK( vtkObjectBase* o );
 
     protected:
       Image( );
       virtual ~Image( );
 
+      template< unsigned int D >
+        inline void _ITK_2_VTK_0( itk::LightObject* o );
+
       template< class P, unsigned int D >
-        inline void _ITK_2_VTK( itk::Object* object );
+        inline void _ITK_2_VTK_1( itk::LightObject* o );
+
+      template< class I >
+        inline void _ITK_2_VTK_2( itk::LightObject* o );
 
     private:
       // Purposely not implemented
@@ -47,7 +48,7 @@ namespace cpPlugins
       Self& operator=( const Self& );
 
     protected:
-      itk::ProcessObject::Pointer m_ITKvVTKConnection;
+      itk::ProcessObject::Pointer m_ITKvVTK;
     };
 
   } // ecapseman