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( dynamic_cast< itk::Image< itk::RGBPixel< p >, d >* >( o ) != NULL ) \
  21     this->f< itk::RGBPixel< p >, d >( )
  22
  23 // -------------------------------------------------------------------------
  24 std::string cpPlugins::Interface::Image::
  25 GetClassName( ) const
  26 {
  27   return( "cpPlugins::Interface::Image" );
  28 }
  29
  30 // -------------------------------------------------------------------------
  31 void cpPlugins::Interface::Image::
  32 SetRealDataObject( itk::DataObject* dobj )
  33 {
  34   this->Superclass::SetRealDataObject( dobj );
  35
  36   // NOTE: Only 2 and 3 dimensions at this moment to connect with VTK
  37   cpPlugins_Image_Dimension( 2, dobj, _VTK_0 );
  38   else cpPlugins_Image_Dimension( 3, dobj, _VTK_0 );
  39 }
  40
  41 // -------------------------------------------------------------------------
  42 vtkImageData* cpPlugins::Interface::Image::
  43 GetVTKImageData( ) const
  44 {
  45   return( this->m_VTKImageData );
  46 }
  47
  48 // -------------------------------------------------------------------------
  49 void cpPlugins::Interface::Image::
  50 UpdateVTKImageData( )
  51 {
  52   this->m_Image2VTKImageData->Modified( );
  53   this->m_Image2VTKImageData->Update( );
  54   this->m_VTKImageData->Modified( );
  55 }
  56
  57 // -------------------------------------------------------------------------
  58 cpPlugins::Interface::Image::
  59 Image( )
  60   : Superclass( ),
  61     m_VTKImageData( NULL )
  62 {
  63 }
  64
  65 // -------------------------------------------------------------------------
  66 cpPlugins::Interface::Image::
  67 ~Image( )
  68 {
  69 }
  70
  71 // -------------------------------------------------------------------------
  72 template< unsigned int D >
  73 void cpPlugins::Interface::Image::
  74 _VTK_0( )
  75 {
  76   itk::DataObject* dobj = this->Superclass::GetRealDataObject( );
  77
  78   cpPlugins_Image_Pixel( char, D, dobj, _VTK_1 );
  79   else cpPlugins_Image_Pixel( short, D, dobj, _VTK_1 );
  80   else cpPlugins_Image_Pixel( int, D, dobj, _VTK_1 );
  81   else cpPlugins_Image_Pixel( long, D, dobj, _VTK_1 );
  82   else cpPlugins_Image_Pixel( unsigned char, D, dobj, _VTK_1 );
  83   else cpPlugins_Image_Pixel( unsigned short, D, dobj, _VTK_1 );
  84   else cpPlugins_Image_Pixel( unsigned int, D, dobj, _VTK_1 );
  85   else cpPlugins_Image_Pixel( unsigned long, D, dobj, _VTK_1 );
  86   else cpPlugins_Image_Pixel( float, D, dobj, _VTK_1 );
  87   else cpPlugins_Image_Pixel( double, D, dobj, _VTK_1 );
  88   else cpPlugins_Image_RGB( char, D, dobj, _VTK_1 );
  89   else cpPlugins_Image_RGB( short, D, dobj, _VTK_1 );
  90   else cpPlugins_Image_RGB( int, D, dobj, _VTK_1 );
  91   else cpPlugins_Image_RGB( long, D, dobj, _VTK_1 );
  92   else cpPlugins_Image_RGB( unsigned char, D, dobj, _VTK_1 );
  93   else cpPlugins_Image_RGB( unsigned short, D, dobj, _VTK_1 );
  94   else cpPlugins_Image_RGB( unsigned int, D, dobj, _VTK_1 );
  95   else cpPlugins_Image_RGB( unsigned long, D, dobj, _VTK_1 );
  96   else cpPlugins_Image_RGB( float, D, dobj, _VTK_1 );
  97   else cpPlugins_Image_RGB( double, D, dobj, _VTK_1 );
  98 }
  99
 100 // -------------------------------------------------------------------------
 101 template< class P, unsigned int D >
 102 void cpPlugins::Interface::Image::
 103 _VTK_1( )
 104 {
 105   typedef itk::Image< P, D > _TImage;
 106   typedef itk::ImageToVTKImageFilter< _TImage > _TFilter;
 107
 108   _TImage* img =
 109     dynamic_cast< _TImage* >( this->Superclass::GetRealDataObject( ) );
 110   typename _TFilter::Pointer filter = _TFilter::New( );
 111   filter->SetInput( img );
 112   filter->Update( );
 113   this->m_VTKImageData = filter->GetOutput( );
 114   this->m_Image2VTKImageData = filter;
 115 }
 116
 117 // eof - $RCSfile$