#include <cpPlugins/Interface/Image.h>

#define ITK_MANUAL_INSTANTIATION
#include <itkImage.h>
#include <itkRGBPixel.h>
#include <itkImageToVTKImageFilter.h>

// -------------------------------------------------------------------------
#define cpPlugins_Image_Dimension( d, o, f )                         \
  if( dynamic_cast< itk::ImageBase< d >* >( o ) != NULL )            \
    this->f< d >( )

// -------------------------------------------------------------------------
#define cpPlugins_Image_Pixel( p, d, o, f )                             \
  if( dynamic_cast< itk::Image< p, d >* >( o ) != NULL )                \
    this->f< p, d >( )

// -------------------------------------------------------------------------
#define cpPlugins_Image_RGB( p, d, o, f )                               \
  if(                                                                   \
    dynamic_cast< itk::Image< itk::RGBPixel< p >, d >* >( o ) != NULL   \
    )                                                                   \
    this->f< itk::RGBPixel< p >, d >( )

// -------------------------------------------------------------------------
std::string cpPlugins::Interface::Image::
GetClassName( ) const
{
  return( "cpPlugins::Interface::Image" );
}

// -------------------------------------------------------------------------
void cpPlugins::Interface::Image::
SetRealDataObject( itk::DataObject* dobj )
{
  this->Superclass::SetRealDataObject( dobj );

  // NOTE: Only 2 and 3 dimensions at this moment to connect with VTK
  cpPlugins_Image_Dimension( 2, dobj, _VTK_0 );
  else cpPlugins_Image_Dimension( 3, dobj, _VTK_0 );
}

// -------------------------------------------------------------------------
vtkImageData* cpPlugins::Interface::Image::
GetVTKImageData( ) const
{
  return( this->m_VTKImageData );
}

// -------------------------------------------------------------------------
void cpPlugins::Interface::Image::
UpdateVTKImageData( )
{
  this->m_Image2VTKImageData->Modified( );
  this->m_Image2VTKImageData->Update( );
  this->m_VTKImageData->Modified( );
}

// -------------------------------------------------------------------------
cpPlugins::Interface::Image::
Image( )
  : Superclass( ),
    m_VTKImageData( NULL )
{
}

// -------------------------------------------------------------------------
cpPlugins::Interface::Image::
~Image( )
{
}

// -------------------------------------------------------------------------
template< unsigned int D >
void cpPlugins::Interface::Image::
_VTK_0( )
{
  itk::DataObject* dobj = this->Superclass::GetRealDataObject( );

  cpPlugins_Image_Pixel( char, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( short, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( int, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( long, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( unsigned char, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( unsigned short, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( unsigned int, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( unsigned long, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( float, D, dobj, _VTK_1 );
  else cpPlugins_Image_Pixel( double, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( char, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( short, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( int, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( long, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( unsigned char, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( unsigned short, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( unsigned int, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( unsigned long, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( float, D, dobj, _VTK_1 );
  else cpPlugins_Image_RGB( double, D, dobj, _VTK_1 );
}

// -------------------------------------------------------------------------
template< class P, unsigned int D >
void cpPlugins::Interface::Image::
_VTK_1( )
{
  typedef itk::Image< P, D > _TImage;
  typedef itk::ImageToVTKImageFilter< _TImage > _TFilter;

  _TImage* img =
    dynamic_cast< _TImage* >( this->Superclass::GetRealDataObject( ) );
  typename _TFilter::Pointer filter = _TFilter::New( );
  filter->SetInput( img );
  filter->Update( );
  this->m_VTKImageData = filter->GetOutput( );
  this->m_Image2VTKImageData = filter;
}

// eof - $RCSfile$