X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpPlugins%2FInterface%2FDataObject.cxx;h=c937e4d32d868d8359018e8183136168f7ea4283;hb=f654620df52b811be7bd263a1775c93d29c69a65;hp=4ec6223b6096d995f662b45e276311ae598035f7;hpb=1adce86c283e253ec41f762652bc477d56d617a5;p=cpPlugins.git

diff --git a/lib/cpPlugins/Interface/DataObject.cxx b/lib/cpPlugins/Interface/DataObject.cxx
index 4ec6223..c937e4d 100644
--- a/lib/cpPlugins/Interface/DataObject.cxx
+++ b/lib/cpPlugins/Interface/DataObject.cxx
@@ -1,52 +1,82 @@
 #include <cpPlugins/Interface/DataObject.h>
 
-// -------------------------------------------------------------------------
-cpPlugins::Interface::DataObject::
-DataObject( )
-  : Superclass( ),
-    m_Source( NULL )
-{
-}
+#include <cpPlugins/Interface/ProcessObject.h>
+#include <itkDataObject.h>
+#include <vtkImageData.h>
+#include <vtkPolyData.h>
 
 // -------------------------------------------------------------------------
-cpPlugins::Interface::DataObject::
-~DataObject( )
+cpPlugins::Interface::ProcessObject* cpPlugins::Interface::DataObject::
+GetSource( )
 {
+  return( this->m_Source );
 }
 
 // -------------------------------------------------------------------------
-std::string cpPlugins::Interface::DataObject::
-GetClassName( ) const
+const cpPlugins::Interface::ProcessObject* cpPlugins::Interface::DataObject::
+GetSource( ) const
 {
-  return( "cpPlugins::Interface::DataObject" );
+  return( this->m_Source );
 }
 
 // -------------------------------------------------------------------------
-itk::DataObject* cpPlugins::Interface::DataObject::
-GetDataObject( ) const
+void cpPlugins::Interface::DataObject::
+SetSource( cpPlugins::Interface::ProcessObject* src )
 {
-  return( this->m_DataObject );
+  if( this->m_Source != src )
+  {
+    this->m_Source = src;
+    this->Modified( );
+
+  } // fi
 }
 
 // -------------------------------------------------------------------------
 void cpPlugins::Interface::DataObject::
-SetDataObject( itk::DataObject* dobj )
+DisconnectPipeline( )
 {
-  this->m_DataObject = dobj;
+  // Disconnect input pipelines (ITK)
+  itk::DataObject* itk_obj = this->GetITK< itk::DataObject >( );
+  if( itk_obj != NULL )
+    itk_obj->DisconnectPipeline( );
+
+  // 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
+
+  // Unbind source
+  this->m_Source = NULL;
 }
 
 // -------------------------------------------------------------------------
-cpPlugins::Interface::ProcessObject* cpPlugins::Interface::DataObject::
-GetSource( ) const
+cpPlugins::Interface::DataObject::
+DataObject( )
+  : Superclass( ),
+    m_ITKObject( NULL ),
+    m_VTKObject( NULL ),
+    m_Source( NULL )
 {
-  return( this->m_Source );
 }
 
 // -------------------------------------------------------------------------
-void cpPlugins::Interface::DataObject::
-SetSource( cpPlugins::Interface::ProcessObject* src )
+cpPlugins::Interface::DataObject::
+~DataObject( )
 {
-  this->m_Source = src;
 }
 
 // eof - $RCSfile$