X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpPlugins%2FInterface%2FDataObject.cxx;h=580434548740a34c758e10bb702be2e0705363ba;hb=4f9005fe71854d7cff6ab68e48627989a6d1fbc8;hp=da2cdf0747a4fbac100c38ff6513246183aa1165;hpb=7c7bc497af96e7b5845be9a2fc277036ec752be9;p=cpPlugins.git

diff --git a/lib/cpPlugins/Interface/DataObject.cxx b/lib/cpPlugins/Interface/DataObject.cxx
index da2cdf0..5804345 100644
--- a/lib/cpPlugins/Interface/DataObject.cxx
+++ b/lib/cpPlugins/Interface/DataObject.cxx
@@ -1,37 +1,9 @@
 #include <cpPlugins/Interface/DataObject.h>
+
+#include <itkDataObject.h>
 #include <vtkImageData.h>
 #include <vtkPolyData.h>
 
-// -------------------------------------------------------------------------
-/* TODO
-   itk::DataObject* cpPlugins::Interface::DataObject::
-   GetITKDataObject( )
-   {
-   return( this->m_ITKObject.GetPointer( ) );
-   }
-
-   // -------------------------------------------------------------------------
-   const itk::DataObject* cpPlugins::Interface::DataObject::
-   GetITKDataObject( ) const
-   {
-   return( this->m_ITKObject.GetPointer( ) );
-   }
-
-   // -------------------------------------------------------------------------
-   vtkDataObject* cpPlugins::Interface::DataObject::
-   GetVTKDataObject( )
-   {
-   return( this->m_VTKObject.GetPointer( ) );
-   }
-
-   // -------------------------------------------------------------------------
-   const vtkDataObject* cpPlugins::Interface::DataObject::
-   GetVTKDataObject( ) const
-   {
-   return( this->m_VTKObject.GetPointer( ) );
-   }
-*/
-
 // -------------------------------------------------------------------------
 cpPlugins::Interface::Object* cpPlugins::Interface::DataObject::
 GetSource( )
@@ -62,29 +34,27 @@ SetSource( cpPlugins::Interface::Object* src )
 void cpPlugins::Interface::DataObject::
 DisconnectPipeline( )
 {
-  if( this->m_ITKObject.IsNotNull( ) )
-  {
-    // Disconnect input pipelines
-    this->m_ITKObject->DisconnectPipeline( );
-    vtkImageData* vtk_image =
-      dynamic_cast< vtkImageData* >( this->m_VTKObject.GetPointer( ) );
-    vtkPolyData* vtk_pd =
-      dynamic_cast< vtkPolyData* >( this->m_VTKObject.GetPointer( ) );
-    if( vtk_image != NULL )
-    {
-      vtkSmartPointer< vtkImageData > d =
-        vtkSmartPointer< vtkImageData >::New( );
-      d->ShallowCopy( vtk_image );
-      this->m_VTKObject = d;
-    }
-    else if( vtk_pd != NULL )
-    {
-      vtkSmartPointer< vtkPolyData > d =
-        vtkSmartPointer< vtkPolyData >::New( );
-      d->ShallowCopy( vtk_pd );
-      this->m_VTKObject = d;
+  // Disconnect input pipelines (ITK)
+  itk::DataObject* itk_obj = this->GetITK< itk::DataObject >( );
+  if( itk_obj != NULL )
+    itk_obj->DisconnectPipeline( );
 
-    } // fi
+  // Disconnect input pipelines (VTK)
+  vtkImageData* vtk_image = this->GetVTK< vtkImageData >( );
+  vtkPolyData* vtk_pd = this->GetVTK< vtkPolyData >( );
+  if( vtk_image != NULL )
+  {
+    vtkSmartPointer< vtkImageData > d =
+      vtkSmartPointer< vtkImageData >::New( );
+    d->ShallowCopy( vtk_image );
+    this->m_VTKObject = d;
+  }
+  else if( vtk_pd != NULL )
+  {
+    vtkSmartPointer< vtkPolyData > d =
+      vtkSmartPointer< vtkPolyData >::New( );
+    d->ShallowCopy( vtk_pd );
+    this->m_VTKObject = d;
 
   } // fi
 
@@ -95,10 +65,11 @@ DisconnectPipeline( )
 // -------------------------------------------------------------------------
 cpPlugins::Interface::DataObject::
 DataObject( )
-  : Superclass( )
+  : Superclass( ),
+    m_ITKObject( NULL ),
+    m_VTKObject( NULL ),
+    m_Source( NULL )
 {
-  this->m_ClassName = "cpPlugins::Interface::DataObject";
-  this->m_ClassCategory = "BasicObject";
 }
 
 // -------------------------------------------------------------------------