lib/cpPlugins/Plugins/ImageReader.cxx

   1 #include <cpPlugins/Plugins/ImageReader.h>
   2
   3 // TODO: interesting... #define ITK_MANUAL_INSTANTIATION
   4 #include <itkImage.h>
   5 #include <itkImageFileReader.h>
   6 #include <itkRGBPixel.h>
   7
   8 // -------------------------------------------------------------------------
   9 cpPlugins::Plugins::ImageReader::
  10 ImageReader( )
  11   : Superclass( )
  12 {
  13   this->m_DefaultParameters[ "FileName" ] =
  14     TParameter( "string", "no_file_name" );
  15   this->m_DefaultParameters[ "PixelType" ] = TParameter( "type", "uchar" );
  16   this->m_DefaultParameters[ "ImageDimension" ] = TParameter( "int", "2" );
  17   this->m_DefaultParameters[ "IsColorImage" ] = TParameter( "bool", "0" );
  18 }
  19
  20 // -------------------------------------------------------------------------
  21 cpPlugins::Plugins::ImageReader::
  22 ~ImageReader( )
  23 {
  24 }
  25
  26 // -------------------------------------------------------------------------
  27 std::string cpPlugins::Plugins::ImageReader::
  28 GetClassName( ) const
  29 {
  30   return( "cpPlugins::Plugins::ImageReader" );
  31 }
  32
  33 // -------------------------------------------------------------------------
  34 bool cpPlugins::Plugins::ImageReader::
  35 Update( )
  36 {
  37   TParameters::const_iterator dIt;
  38
  39   // Get image dimension
  40   dIt = this->m_Parameters.find( "ImageDimension" );
  41   if( dIt == this->m_Parameters.end( ) )
  42     dIt = this->m_DefaultParameters.find( "ImageDimension" );
  43
  44   bool ret = false;
  45   if     ( dIt->second.second == "1" ) ret = this->_Update0< 1 >( );
  46   else if( dIt->second.second == "2" ) ret = this->_Update0< 2 >( );
  47   else if( dIt->second.second == "3" ) ret = this->_Update0< 3 >( );
  48   else if( dIt->second.second == "4" ) ret = this->_Update0< 4 >( );
  49
  50   return( ret );
  51 }
  52
  53 // -------------------------------------------------------------------------
  54 template< unsigned int D >
  55 bool cpPlugins::Plugins::ImageReader::
  56 _Update0( )
  57 {
  58   TParameters::const_iterator tIt, cIt;
  59
  60   // Get image pixelType
  61   tIt = this->m_Parameters.find( "PixelType" );
  62   if( tIt == this->m_Parameters.end( ) )
  63     tIt = this->m_DefaultParameters.find( "PixelType" );
  64   cIt = this->m_Parameters.find( "IsColorImage" );
  65   if( cIt == this->m_Parameters.end( ) )
  66     cIt = this->m_DefaultParameters.find( "IsColorImage" );
  67
  68   bool ret = false;
  69   if( cIt->second.second == "0" )
  70   {
  71     if( tIt->second.second == "char" )
  72       ret = this->_Update1< char, D >( );
  73     else if( tIt->second.second == "short" )
  74       ret = this->_Update1< short, D >( );
  75     else if( tIt->second.second == "int" )
  76       ret = this->_Update1< int, D >( );
  77     else if( tIt->second.second == "long" )
  78       ret = this->_Update1< long, D >( );
  79     else if( tIt->second.second == "uchar" )
  80       ret = this->_Update1< unsigned char, D >( );
  81     else if( tIt->second.second == "ushort" )
  82       ret = this->_Update1< unsigned short, D >( );
  83     else if( tIt->second.second == "uint" )
  84       ret = this->_Update1< unsigned int, D >( );
  85     else if( tIt->second.second == "ulong" )
  86       ret = this->_Update1< unsigned long, D >( );
  87     else if( tIt->second.second == "float" )
  88       ret = this->_Update1< float, D >( );
  89     else if( tIt->second.second == "double" )
  90       ret = this->_Update1< double, D >( );
  91   }
  92   else if( cIt->second.second == "1" )
  93   {
  94     if( tIt->second.second == "char" )
  95       ret = this->_Update1< itk::RGBPixel< char >, D >( );
  96     else if( tIt->second.second == "short" )
  97       ret = this->_Update1< itk::RGBPixel< short >, D >( );
  98     else if( tIt->second.second == "int" )
  99       ret = this->_Update1< itk::RGBPixel< int >, D >( );
 100     else if( tIt->second.second == "long" )
 101       ret = this->_Update1< itk::RGBPixel< long >, D >( );
 102     else if( tIt->second.second == "uchar" )
 103       ret = this->_Update1< itk::RGBPixel< unsigned char >, D >( );
 104     else if( tIt->second.second == "ushort" )
 105       ret = this->_Update1< itk::RGBPixel< unsigned short >, D >( );
 106     else if( tIt->second.second == "uint" )
 107       ret = this->_Update1< itk::RGBPixel< unsigned int >, D >( );
 108     else if( tIt->second.second == "ulong" )
 109       ret = this->_Update1< itk::RGBPixel< unsigned long >, D >( );
 110     else if( tIt->second.second == "float" )
 111       ret = this->_Update1< itk::RGBPixel< float >, D >( );
 112     else if( tIt->second.second == "double" )
 113       ret = this->_Update1< itk::RGBPixel< double >, D >( );
 114   } // fi
 115   return( ret );
 116 }
 117
 118 // -------------------------------------------------------------------------
 119 template< class P, unsigned int D >
 120 bool cpPlugins::Plugins::ImageReader::
 121 _Update1( )
 122 {
 123   TParameters::const_iterator fIt;
 124
 125   // Get image pixelType
 126   fIt = this->m_Parameters.find( "FileName" );
 127   if( fIt == this->m_Parameters.end( ) )
 128     fIt = this->m_DefaultParameters.find( "FileName" );
 129
 130   typedef itk::Image< P, D > _TImage;
 131   typedef itk::ImageFileReader< _TImage > _TImageReader;
 132
 133   _TImageReader* reader =
 134     dynamic_cast< _TImageReader* >( this->m_Reader.GetPointer( ) );
 135   if( reader == NULL )
 136   {
 137     this->m_Reader = _TImageReader::New( );
 138     reader = dynamic_cast< _TImageReader* >( this->m_Reader.GetPointer( ) );
 139
 140   } // fi
 141   reader->SetFileName( fIt->second.second );
 142   try
 143   {
 144     reader->Update( );
 145   }
 146   catch( itk::ExceptionObject& err )
 147   {
 148     return( false );
 149
 150   } // yrt
 151   return( true );
 152 }
 153
 154 // eof - $RCSfile$