X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpPlugins%2FInterface%2FDataObject.h;h=05b70239dc06bda7ff7ac5e7451bbca89f050b95;hb=de874ea850042e77a99a456188f423c8df2e374f;hp=c08365463e5e9a8abf736a85a4df4f90bf1da078;hpb=c29448936abc15d110205b2da435035131398a87;p=cpPlugins.git

diff --git a/lib/cpPlugins/Interface/DataObject.h b/lib/cpPlugins/Interface/DataObject.h
index c083654..05b7023 100644
--- a/lib/cpPlugins/Interface/DataObject.h
+++ b/lib/cpPlugins/Interface/DataObject.h
@@ -1,16 +1,17 @@
 #ifndef __CPPLUGINS__INTERFACE__DATAOBJECT__H__
 #define __CPPLUGINS__INTERFACE__DATAOBJECT__H__
 
-#include <map>
-#include <string>
-#include <itkDataObject.h>
-#include <cpPlugins/Interface/cpPlugins_Interface_Export.h>
 #include <cpPlugins/Interface/Object.h>
 
+#include <itkObject.h>
+#include <vtkSmartPointer.h>
+#include <vtkObject.h>
+
 namespace cpPlugins
 {
   namespace Interface
   {
+    // Some forward declarations
     class ProcessObject;
 
     /**
@@ -19,30 +20,56 @@ namespace cpPlugins
       : public Object
     {
     public:
-      typedef DataObject Self;
-      typedef Object     Superclass;
+      typedef DataObject                      Self;
+      typedef Object                          Superclass;
+      typedef itk::SmartPointer< Self >       Pointer;
+      typedef itk::SmartPointer< const Self > ConstPointer;
 
     public:
-      DataObject( );
-      virtual ~DataObject( );
+      itkNewMacro( Self );
+      itkTypeMacro( DataObject, Object );
+      cpPlugins_Id_Macro( DataObject, BasicObject );
+
+    public:
+      ProcessObject* GetSource( );
+      const ProcessObject* GetSource( ) const;
+      void SetSource( ProcessObject* src );
 
-      virtual std::string GetClassName( ) const;
+      void DisconnectPipeline( );
 
-      itk::DataObject* GetDataObject( ) const;
-      virtual void SetDataObject( itk::DataObject* src );
+      template< class T >
+        inline T* GetITK( );
 
-      ProcessObject* GetSource( ) const;
-      void SetSource( ProcessObject* src );
+      template< class T >
+        inline const T* GetITK( ) const;
+
+      template< class T >
+        inline T* GetVTK( );
+
+      template< class T >
+        inline const T* GetVTK( ) const;
 
     protected:
-      itk::DataObject::Pointer m_DataObject;
-      ProcessObject*           m_Source;
+      DataObject( );
+      virtual ~DataObject( );
+
+    private:
+      // Purposely not implemented
+      DataObject( const Self& );
+      Self& operator=( const Self& );
+
+    protected:
+      itk::Object::Pointer         m_ITKObject;
+      vtkSmartPointer< vtkObject > m_VTKObject;
+      ProcessObject*               m_Source;
     };
 
   } // ecapseman
 
 } // ecapseman
 
+#include <cpPlugins/Interface/DataObject.hxx>
+
 #endif // __CPPLUGINS__INTERFACE__DATAOBJECT__H__
 
 // eof - $RCSfile$