...

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

diff --git a/lib/cpPlugins/Interface/Mesh.h b/lib/cpPlugins/Interface/Mesh.h
index ffaec80ce9809f4589893c4b35dabcc320957bfc..75c5eb3683e766a6a546e3091330116d170a8b2e 100644 (file)
--- a/lib/cpPlugins/Interface/Mesh.h
+++ b/lib/cpPlugins/Interface/Mesh.h
@@ -4,7 +4,10 @@
 #include <cpPlugins/Interface/cpPlugins_Interface_Export.h>
 #include <cpPlugins/Interface/DataObject.h>
 
-class vtkPolyData;
+#include <vtkSmartPointer.h>
+#include <vtkActor.h>
+#include <vtkPolyData.h>
+#include <vtkPolyDataMapper.h>
 
 namespace cpPlugins
 {
@@ -26,38 +29,41 @@ namespace cpPlugins
       itkTypeMacro( Mesh, DataObject );
 
     public:
-      virtual void SetITKDataObject( itk::DataObject* o );
-      virtual void SetVTKDataObject( vtkDataObject* o );
+      template< class M >
+        inline void SetITKMesh( itk::DataObject* object );
 
-      vtkPolyData* GetVTKPolyData( );
-      const vtkPolyData* GetVTKPolyData( ) const;
+      template< class M >
+        inline M* GetITKMesh( );
+
+      template< class M >
+        inline const M* GetITKMesh( ) const;
+
+      virtual void SetVTKMesh( vtkPolyData* mesh );
+      virtual vtkPolyData* GetVTKMesh( );
+      virtual const vtkPolyData* GetVTKMesh( ) const;
+      virtual vtkActor* GetVTKActor( );
+      virtual const vtkActor* GetVTKActor( ) const;
 
     protected:
       Mesh( );
       virtual ~Mesh( );
 
-      template< unsigned int D >
-        void _ITK_2_VTK_0( itk::DataObject* o );
-
-      template< class P, unsigned int D >
-        void _ITK_2_VTK_1( itk::DataObject* o );
-
-      template< unsigned int D >
-        void _VTK_2_ITK_0( itk::DataObject* o );
-
-      template< class P, unsigned int D >
-        void _VTK_2_ITK_1( itk::DataObject* o );
-
     private:
       // Purposely not implemented
       Mesh( const Self& );
       Self& operator=( const Self& );
+
+    protected:
+      vtkSmartPointer< vtkActor >          m_Actor;
+      vtkSmartPointer< vtkPolyDataMapper > m_Mapper;
     };
 
   } // ecapseman
 
 } // ecapseman
 
+#include <cpPlugins/Interface/Mesh.hxx>
+
 #endif // __CPPLUGINS__INTERFACE__MESH__H__
 
 // eof - $RCSfile$