...

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

diff --git a/lib/cpPlugins/Interface/DataObject.h b/lib/cpPlugins/Interface/DataObject.h
index be40af5b2989d4c5978760fab68308c172c11d58..e2d79ffcc16ba3313d4b50b2b1e4e7f26759a60e 100644 (file)
--- a/lib/cpPlugins/Interface/DataObject.h
+++ b/lib/cpPlugins/Interface/DataObject.h
@@ -1,20 +1,15 @@
 #ifndef __CPPLUGINS__INTERFACE__DATAOBJECT__H__
 #define __CPPLUGINS__INTERFACE__DATAOBJECT__H__
 
-#include <map>
-#include <string>
-#include <vtkDataObject.h>
-#include <vtkSmartPointer.h>
-#include <cpPlugins/Interface/cpPlugins_Interface_Export.h>
 #include <cpPlugins/Interface/Object.h>
 
-#define ITK_MANUAL_INSTANTIATION
-#include <itkDataObject.h>
-
 namespace cpPlugins
 {
   namespace Interface
   {
+    // Some forward declarations
+    class ProcessObject;
+
     /**
      */
     class cpPlugins_Interface_EXPORT DataObject
@@ -27,20 +22,14 @@ namespace cpPlugins
       typedef itk::SmartPointer< const Self > ConstPointer;
 
     public:
+      itkNewMacro( Self );
       itkTypeMacro( DataObject, Object );
+      cpPlugins_Id_Macro( DataObject, BasicObject );
 
     public:
-      /* TODO
-         virtual itk::DataObject* GetITKDataObject( );
-         virtual const itk::DataObject* GetITKDataObject( ) const;
-
-         virtual vtkDataObject* GetVTKDataObject( );
-         virtual const vtkDataObject* GetVTKDataObject( ) const;
-      */
-
-      Object* GetSource( );
-      const Object* GetSource( ) const;
-      void SetSource( Object* src );
+      ProcessObject* GetSource( );
+      const ProcessObject* GetSource( ) const;
+      void SetSource( ProcessObject* src );
 
       void DisconnectPipeline( );
 
@@ -54,9 +43,7 @@ namespace cpPlugins
       Self& operator=( const Self& );
 
     protected:
-      itk::DataObject::Pointer         m_ITKObject;
-      vtkSmartPointer< vtkDataObject > m_VTKObject;
-      Object::Pointer                  m_Source;
+      ProcessObject* m_Source;
     };
 
   } // ecapseman