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 std::string cpPlugins::Interface::Image::
  27 GetClassName( ) const
  28 {
  29   return( "cpPlugins::Interface::Image" );
  30 }
  31
  32 // -------------------------------------------------------------------------
  33 void cpPlugins::Interface::Image::
  34 SetRealDataObject( itk::DataObject* dobj )
  35 {
  36   this->Superclass::SetRealDataObject( dobj );
  37
  38   // NOTE: Only 2 and 3 dimensions at this moment to connect with VTK
  39   cpPlugins_Image_Dimension( 2, dobj, _VTK_0 );
  40   else cpPlugins_Image_Dimension( 3, dobj, _VTK_0 );
  41 }
  42
  43 // -------------------------------------------------------------------------
  44 vtkImageData* cpPlugins::Interface::Image::
  45 GetVTKImageData( ) const
  46 {
  47   return( this->m_VTKImageData );
  48 }
  49
  50 // -------------------------------------------------------------------------
  51 void cpPlugins::Interface::Image::
  52 UpdateVTKImageData( )
  53 {
  54   this->m_Image2VTKImageData->Modified( );
  55   this->m_Image2VTKImageData->Update( );
  56   this->m_VTKImageData->Modified( );
  57 }
  58
  59 // -------------------------------------------------------------------------
  60 cpPlugins::Interface::Image::
  61 Image( )
  62   : Superclass( ),
  63     m_VTKImageData( NULL )
  64 {
  65 }
  66
  67 // -------------------------------------------------------------------------
  68 cpPlugins::Interface::Image::
  69 ~Image( )
  70 {
  71 }
  72
  73 // -------------------------------------------------------------------------
  74 template< unsigned int D >
  75 void cpPlugins::Interface::Image::
  76 _VTK_0( )
  77 {
  78   itk::DataObject* dobj = this->Superclass::GetRealDataObject( );
  79
  80   cpPlugins_Image_Pixel( char, D, dobj, _VTK_1 );
  81   else cpPlugins_Image_Pixel( short, D, dobj, _VTK_1 );
  82   else cpPlugins_Image_Pixel( int, D, dobj, _VTK_1 );
  83   else cpPlugins_Image_Pixel( long, D, dobj, _VTK_1 );
  84   else cpPlugins_Image_Pixel( unsigned char, D, dobj, _VTK_1 );
  85   else cpPlugins_Image_Pixel( unsigned short, D, dobj, _VTK_1 );
  86   else cpPlugins_Image_Pixel( unsigned int, D, dobj, _VTK_1 );
  87   else cpPlugins_Image_Pixel( unsigned long, D, dobj, _VTK_1 );
  88   else cpPlugins_Image_Pixel( float, D, dobj, _VTK_1 );
  89   else cpPlugins_Image_Pixel( double, D, dobj, _VTK_1 );
  90   else cpPlugins_Image_RGB( char, D, dobj, _VTK_1 );
  91   else cpPlugins_Image_RGB( short, D, dobj, _VTK_1 );
  92   else cpPlugins_Image_RGB( int, D, dobj, _VTK_1 );
  93   else cpPlugins_Image_RGB( long, D, dobj, _VTK_1 );
  94   else cpPlugins_Image_RGB( unsigned char, D, dobj, _VTK_1 );
  95   else cpPlugins_Image_RGB( unsigned short, D, dobj, _VTK_1 );
  96   else cpPlugins_Image_RGB( unsigned int, D, dobj, _VTK_1 );
  97   else cpPlugins_Image_RGB( unsigned long, D, dobj, _VTK_1 );
  98   else cpPlugins_Image_RGB( float, D, dobj, _VTK_1 );
  99   else cpPlugins_Image_RGB( double, D, dobj, _VTK_1 );
 100 }
 101
 102 // -------------------------------------------------------------------------
 103 template< class P, unsigned int D >
 104 void cpPlugins::Interface::Image::
 105 _VTK_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::GetRealDataObject( ) );
 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$