Integration of new functionalities into ProcessObject and DataObject

[cpPlugins.git] / lib / cpPlugins / Interface / Image.cxx

diff --git a/lib/cpPlugins/Interface/Image.cxx b/lib/cpPlugins/Interface/Image.cxx
index 9b55ed27d17cea446ab6e26a8a325ea3ca9905f7..d8a47c81cb27eb46688d74918d3c04f90fc5a441 100644 (file)
--- a/lib/cpPlugins/Interface/Image.cxx
+++ b/lib/cpPlugins/Interface/Image.cxx
@@ -1,9 +1,29 @@
 #include <cpPlugins/Interface/Image.h>
 
+#include <itkImage.h>
+#include <itkRGBPixel.h>
+#include <itkImageToVTKImageFilter.h>
+
+// -------------------------------------------------------------------------
+#define cpPlugins_Interface_Image_Dimension( d, dobj, func )            \
+  if( dynamic_cast< itk::ImageBase< d >* >( dobj ) != NULL )            \
+    this->func< d >( )
+
+// -------------------------------------------------------------------------
+#define cpPlugins_Interface_Image_Pixel( p, d, dobj, func )             \
+  if( dynamic_cast< itk::Image< p, d >* >( dobj ) != NULL )             \
+    this->func< p, d >( )
+
+// -------------------------------------------------------------------------
+#define cpPlugins_Interface_Image_RGB( p, d, dobj, func )               \
+  if( dynamic_cast< itk::Image< itk::RGBPixel< p >, d >* >( dobj ) != NULL ) \
+    this->func< itk::RGBPixel< p >, d >( )
+
 // -------------------------------------------------------------------------
 cpPlugins::Interface::Image::
 Image( )
-  : Superclass( )
+  : Superclass( ),
+    m_VTKImageData( NULL )
 {
 }
 
@@ -20,4 +40,68 @@ GetClassName( ) const
   return( "cpPlugins::Interface::Image" );
 }
 
+// -------------------------------------------------------------------------
+void cpPlugins::Interface::Image::
+SetDataObject( itk::DataObject* dobj )
+{
+  this->Superclass::SetDataObject( dobj );
+  
+  // WARNING: Only 2 and 3 dimensions at this moment
+  cpPlugins_Interface_Image_Dimension( 2, dobj, _ConnectToVTK_0 );
+  else cpPlugins_Interface_Image_Dimension( 3, dobj, _ConnectToVTK_0 );
+}
+
+// -------------------------------------------------------------------------
+vtkImageData* cpPlugins::Interface::Image::
+GetVTKImageData( ) const
+{
+  return( this->m_VTKImageData );
+}
+
+// -------------------------------------------------------------------------
+template< unsigned int D >
+void cpPlugins::Interface::Image::
+_ConnectToVTK_0( )
+{
+  itk::DataObject* dobj = this->Superclass::GetDataObject( );
+
+  cpPlugins_Interface_Image_Pixel( char, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( short, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( int, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( long, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( unsigned char, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( unsigned short, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( unsigned int, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( unsigned long, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( float, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_Pixel( double, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( char, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( short, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( int, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( long, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( unsigned char, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( unsigned short, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( unsigned int, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( unsigned long, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( float, D, dobj, _ConnectToVTK_1 );
+  else cpPlugins_Interface_Image_RGB( double, D, dobj, _ConnectToVTK_1 );
+}
+
+// -------------------------------------------------------------------------
+template< class P, unsigned int D >
+void cpPlugins::Interface::Image::
+_ConnectToVTK_1( )
+{
+  typedef itk::Image< P, D > _TImage;
+  typedef itk::ImageToVTKImageFilter< _TImage > _TFilter;
+
+  _TImage* img =
+    dynamic_cast< _TImage* >( this->Superclass::GetDataObject( ) );
+  typename _TFilter::Pointer filter = _TFilter::New( );
+  filter->SetInput( img );
+  filter->Update( );
+  this->m_VTKImageData = filter->GetOutput( );
+  this->m_Image2VTKImageData = filter;
+}
+
 // eof - $RCSfile$