#include <cpPlugins/DataObject.h>
#include <itkDataObject.h>
#include <vtkImageData.h>
#include <vtkPolyData.h>

// -------------------------------------------------------------------------
cpPlugins::ProcessObject* cpPlugins::DataObject::
GetSource( )
{
  return( this->m_Source );
}

// -------------------------------------------------------------------------
const cpPlugins::ProcessObject* cpPlugins::DataObject::
GetSource( ) const
{
  return( this->m_Source );
}

// -------------------------------------------------------------------------
void cpPlugins::DataObject::
SetSource( cpPlugins::ProcessObject* src )
{
  if( this->m_Source != src )
  {
    this->m_Source = src;
    this->Modified( );

  } // fi
}

// -------------------------------------------------------------------------
void cpPlugins::DataObject::
DisconnectFromPipeline( )
{
  // 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;
  this->Modified( );
}

// -------------------------------------------------------------------------
vtkProp* cpPlugins::DataObject::
GetVTKActor( )
{
  if( this->m_Actor.GetPointer( ) == NULL )
    this->_CreateVTKActor( );
  return( this->m_Actor.GetPointer( ) );
}

// -------------------------------------------------------------------------
const vtkProp* cpPlugins::DataObject::
GetVTKActor( ) const
{
  if( this->m_Actor.GetPointer( ) == NULL )
    this->_CreateVTKActor( );
  return( this->m_Actor.GetPointer( ) );
}

// -------------------------------------------------------------------------
cpPlugins::DataObject::
DataObject( )
  : Superclass( ),
    m_Source( NULL )
{
}

// -------------------------------------------------------------------------
cpPlugins::DataObject::
~DataObject( )
{
}

// -------------------------------------------------------------------------
void cpPlugins::DataObject::
_CreateVTKActor( ) const
{
}

// eof - $RCSfile$