New pixel converter

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

diff --git a/lib/cpPlugins/Interface/Image.cxx b/lib/cpPlugins/Interface/Image.cxx
index 9b55ed27d17cea446ab6e26a8a325ea3ca9905f7..c72663d5dbda9f65cb3cc2eddf021e64b4796fa3 100644 (file)
--- a/lib/cpPlugins/Interface/Image.cxx
+++ b/lib/cpPlugins/Interface/Image.cxx
@@ -1,9 +1,66 @@
 #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( )
+  : Superclass( ),
+    m_VTKImageData( NULL )
 {
 }
 
@@ -14,10 +71,49 @@ cpPlugins::Interface::Image::
 }
 
 // -------------------------------------------------------------------------
-std::string cpPlugins::Interface::Image::
-GetClassName( ) const
+template< unsigned int D >
+void cpPlugins::Interface::Image::
+_VTK_0( )
 {
-  return( "cpPlugins::Interface::Image" );
+  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$