lib/cpPlugins/Interface/Image.cxx

   1 #include <cpPlugins/Interface/Image.h>
   2
   3 #define ITK_MANUAL_INSTANTIATION
   4 #include <itkImage.h>
   5 #include <itkRGBPixel.h>
   6 #include <itkImageToVTKImageFilter.h>
   7
   8 // -------------------------------------------------------------------------
   9 #define cpPlugins_Image_Dimension( d, o, f )                         \
  10   if( dynamic_cast< itk::ImageBase< d >* >( o ) != NULL )            \
  11     this->f< d >( )
  12
  13 // -------------------------------------------------------------------------
  14 #define cpPlugins_Image_Pixel( p, d, o, f )                             \
  15   if( dynamic_cast< itk::Image< p, d >* >( o ) != NULL )                \
  16     this->f< p, d >( )
  17
  18 // -------------------------------------------------------------------------
  19 #define cpPlugins_Image_RGB( p, d, o, f )                               \
  20   if(                                                                   \
  21     dynamic_cast< itk::Image< itk::RGBPixel< p >, d >* >( o ) != NULL   \
  22     )                                                                   \
  23     this->f< itk::RGBPixel< p >, d >( )
  24
  25 // -------------------------------------------------------------------------
  26 cpPlugins::Interface::Image::
  27 Image( )
  28   : Superclass( ),
  29     m_VTKImageData( NULL )
  30 {
  31 }
  32
  33 // -------------------------------------------------------------------------
  34 cpPlugins::Interface::Image::
  35 ~Image( )
  36 {
  37 }
  38
  39 // -------------------------------------------------------------------------
  40 std::string cpPlugins::Interface::Image::
  41 GetClassName( ) const
  42 {
  43   return( "cpPlugins::Interface::Image" );
  44 }
  45
  46 // -------------------------------------------------------------------------
  47 void cpPlugins::Interface::Image::
  48 SetDataObject( itk::DataObject* dobj )
  49 {
  50   this->Superclass::SetDataObject( dobj );
  51
  52   // WARNING: Only 2 and 3 dimensions at this moment
  53   cpPlugins_Image_Dimension( 2, dobj, _ConnectToVTK_0 );
  54   else cpPlugins_Image_Dimension( 3, dobj, _ConnectToVTK_0 );
  55 }
  56
  57 // -------------------------------------------------------------------------
  58 vtkImageData* cpPlugins::Interface::Image::
  59 GetVTKImageData( ) const
  60 {
  61   return( this->m_VTKImageData );
  62 }
  63
  64 // -------------------------------------------------------------------------
  65 void cpPlugins::Interface::Image::
  66 UpdateVTKImageData( )
  67 {
  68   this->m_Image2VTKImageData->Modified( );
  69   this->m_Image2VTKImageData->Update( );
  70   this->m_VTKImageData->Modified( );
  71 }
  72
  73 // -------------------------------------------------------------------------
  74 template< unsigned int D >
  75 void cpPlugins::Interface::Image::
  76 _ConnectToVTK_0( )
  77 {
  78   itk::DataObject* dobj = this->Superclass::GetDataObject( );
  79
  80   cpPlugins_Image_Pixel( char, D, dobj, _ConnectToVTK_1 );
  81   else cpPlugins_Image_Pixel( short, D, dobj, _ConnectToVTK_1 );
  82   else cpPlugins_Image_Pixel( int, D, dobj, _ConnectToVTK_1 );
  83   else cpPlugins_Image_Pixel( long, D, dobj, _ConnectToVTK_1 );
  84   else cpPlugins_Image_Pixel( unsigned char, D, dobj, _ConnectToVTK_1 );
  85   else cpPlugins_Image_Pixel( unsigned short, D, dobj, _ConnectToVTK_1 );
  86   else cpPlugins_Image_Pixel( unsigned int, D, dobj, _ConnectToVTK_1 );
  87   else cpPlugins_Image_Pixel( unsigned long, D, dobj, _ConnectToVTK_1 );
  88   else cpPlugins_Image_Pixel( float, D, dobj, _ConnectToVTK_1 );
  89   else cpPlugins_Image_Pixel( double, D, dobj, _ConnectToVTK_1 );
  90   else cpPlugins_Image_RGB( char, D, dobj, _ConnectToVTK_1 );
  91   else cpPlugins_Image_RGB( short, D, dobj, _ConnectToVTK_1 );
  92   else cpPlugins_Image_RGB( int, D, dobj, _ConnectToVTK_1 );
  93   else cpPlugins_Image_RGB( long, D, dobj, _ConnectToVTK_1 );
  94   else cpPlugins_Image_RGB( unsigned char, D, dobj, _ConnectToVTK_1 );
  95   else cpPlugins_Image_RGB( unsigned short, D, dobj, _ConnectToVTK_1 );
  96   else cpPlugins_Image_RGB( unsigned int, D, dobj, _ConnectToVTK_1 );
  97   else cpPlugins_Image_RGB( unsigned long, D, dobj, _ConnectToVTK_1 );
  98   else cpPlugins_Image_RGB( float, D, dobj, _ConnectToVTK_1 );
  99   else cpPlugins_Image_RGB( double, D, dobj, _ConnectToVTK_1 );
 100 }
 101
 102 // -------------------------------------------------------------------------
 103 template< class P, unsigned int D >
 104 void cpPlugins::Interface::Image::
 105 _ConnectToVTK_1( )
 106 {
 107   typedef itk::Image< P, D > _TImage;
 108   typedef itk::ImageToVTKImageFilter< _TImage > _TFilter;
 109
 110   _TImage* img =
 111     dynamic_cast< _TImage* >( this->Superclass::GetDataObject( ) );
 112   typename _TFilter::Pointer filter = _TFilter::New( );
 113   filter->SetInput( img );
 114   filter->Update( );
 115   this->m_VTKImageData = filter->GetOutput( );
 116   this->m_Image2VTKImageData = filter;
 117 }
 118
 119 // eof - $RCSfile$