X-Git-Url: https://git.creatis.insa-lyon.fr/pubgit/?a=blobdiff_plain;f=lib%2FcpPlugins%2FPlugins%2FImageReader.cxx;h=da7b949485bcb72e7bb9947b3a63df4dd88f1ea7;hb=aa6a578004bddb5b0bb07b780483fda0ecc6cb5e;hp=feca19b52bb063cb79c3f5a23164b41d40e39385;hpb=cf8ebcdb1e31f332e74569b2be5dfb5f5100df07;p=cpPlugins.git diff --git a/lib/cpPlugins/Plugins/ImageReader.cxx b/lib/cpPlugins/Plugins/ImageReader.cxx index feca19b..da7b949 100644 --- a/lib/cpPlugins/Plugins/ImageReader.cxx +++ b/lib/cpPlugins/Plugins/ImageReader.cxx @@ -1,22 +1,37 @@ #include +#include -// TODO: interesting... #define ITK_MANUAL_INSTANTIATION -#include #include +#include + +#define ITK_MANUAL_INSTANTIATION +#include #include +// ------------------------------------------------------------------------- +std::string cpPlugins::Plugins::ImageReader:: +GetClassName( ) const +{ + return( "cpPlugins::Plugins::ImageReader" ); +} + // ------------------------------------------------------------------------- cpPlugins::Plugins::ImageReader:: ImageReader( ) : Superclass( ) { this->SetNumberOfOutputs( 1 ); - - this->m_DefaultParameters[ "FileName" ] = - TParameter( "string", "no_file_name" ); - this->m_DefaultParameters[ "PixelType" ] = TParameter( "type", "uchar" ); - this->m_DefaultParameters[ "ImageDimension" ] = TParameter( "int", "2" ); - this->m_DefaultParameters[ "IsColorImage" ] = TParameter( "bool", "0" ); + this->_MakeOutput< cpPlugins::Interface::Image >( 0 ); + + using namespace cpPlugins::Interface; + this->m_DefaultParameters.Configure( Parameters::StringList, "FileNames" ); + this->m_DefaultParameters.Configure( Parameters::String, "PixelType" ); + this->m_DefaultParameters.Configure( Parameters::Uint, "Dimension" ); + this->m_DefaultParameters.Configure( Parameters::Uint, "IsColorImage" ); + this->m_DefaultParameters.SetValueAsString( "PixelType", "uchar" ); + this->m_DefaultParameters.SetValueAsUint( "Dimension", 3 ); + this->m_DefaultParameters.SetValueAsUint( "IsColorImage", 0 ); + this->m_Parameters = this->m_DefaultParameters; } // ------------------------------------------------------------------------- @@ -25,133 +40,161 @@ cpPlugins::Plugins::ImageReader:: { } -// ------------------------------------------------------------------------- -std::string cpPlugins::Plugins::ImageReader:: -GetClassName( ) const -{ - return( "cpPlugins::Plugins::ImageReader" ); -} - // ------------------------------------------------------------------------- std::string cpPlugins::Plugins::ImageReader:: _GenerateData( ) { - TParameters::const_iterator dIt; - - // Get image dimension - dIt = this->m_Parameters.find( "ImageDimension" ); - if( dIt == this->m_Parameters.end( ) ) - dIt = this->m_DefaultParameters.find( "ImageDimension" ); - - std::string ret = "itk::Image dimension not supported."; - if ( dIt->second.second == "1" ) ret = this->_GenerateData0< 1 >( ); - else if( dIt->second.second == "2" ) ret = this->_GenerateData0< 2 >( ); - else if( dIt->second.second == "3" ) ret = this->_GenerateData0< 3 >( ); - else if( dIt->second.second == "4" ) ret = this->_GenerateData0< 4 >( ); - - return( ret ); + using namespace cpPlugins::Interface; + Parameters::TUint dim = this->m_Parameters.GetValueAsUint( "Dimension" ); + std::string r = "cpPlugins::Plugins::ImageReader: itk::Image dimension not supported."; + if ( dim == 1 ) r = this->_GD0< 1 >( ); + else if( dim == 2 ) r = this->_GD0< 2 >( ); + else if( dim == 3 ) r = this->_GD0< 3 >( ); + else if( dim == 4 ) r = this->_GD0< 4 >( ); + + return( r ); } // ------------------------------------------------------------------------- template< unsigned int D > std::string cpPlugins::Plugins::ImageReader:: -_GenerateData0( ) +_GD0( ) { - TParameters::const_iterator tIt, cIt; - - // Get image pixelType - tIt = this->m_Parameters.find( "PixelType" ); - if( tIt == this->m_Parameters.end( ) ) - tIt = this->m_DefaultParameters.find( "PixelType" ); - cIt = this->m_Parameters.find( "IsColorImage" ); - if( cIt == this->m_Parameters.end( ) ) - cIt = this->m_DefaultParameters.find( "IsColorImage" ); - - std::string ret = "itk::Image pixel type not supported"; - if( cIt->second.second == "0" ) + using namespace cpPlugins::Interface; + Parameters::TString pt = this->m_Parameters.GetValueAsString( "PixelType" ); + Parameters::TUint ci = this->m_Parameters.GetValueAsUint( "IsColorImage" ); + + std::string r = "cpPlugins::Plugins::ImageReader: itk::Image pixel type not supported"; + if( ci == 0 ) { - if( tIt->second.second == "char" ) - ret = this->_GenerateData1< char, D >( ); - else if( tIt->second.second == "short" ) - ret = this->_GenerateData1< short, D >( ); - else if( tIt->second.second == "int" ) - ret = this->_GenerateData1< int, D >( ); - else if( tIt->second.second == "long" ) - ret = this->_GenerateData1< long, D >( ); - else if( tIt->second.second == "uchar" ) - ret = this->_GenerateData1< unsigned char, D >( ); - else if( tIt->second.second == "ushort" ) - ret = this->_GenerateData1< unsigned short, D >( ); - else if( tIt->second.second == "uint" ) - ret = this->_GenerateData1< unsigned int, D >( ); - else if( tIt->second.second == "ulong" ) - ret = this->_GenerateData1< unsigned long, D >( ); - else if( tIt->second.second == "float" ) - ret = this->_GenerateData1< float, D >( ); - else if( tIt->second.second == "double" ) - ret = this->_GenerateData1< double, D >( ); + if( pt == "char" ) + r = this->_GD1< char, D >( ); + else if( pt == "short" ) + r = this->_GD1< short, D >( ); + else if( pt == "int" ) + r = this->_GD1< int, D >( ); + else if( pt == "long" ) + r = this->_GD1< long, D >( ); + else if( pt == "uchar" ) + r = this->_GD1< unsigned char, D >( ); + else if( pt == "ushort" ) + r = this->_GD1< unsigned short, D >( ); + else if( pt == "uint" ) + r = this->_GD1< unsigned int, D >( ); + else if( pt == "ulong" ) + r = this->_GD1< unsigned long, D >( ); + else if( pt == "float" ) + r = this->_GD1< float, D >( ); + else if( pt == "double" ) + r = this->_GD1< double, D >( ); } - else if( cIt->second.second == "1" ) + else { - if( tIt->second.second == "char" ) - ret = this->_GenerateData1< itk::RGBPixel< char >, D >( ); - else if( tIt->second.second == "short" ) - ret = this->_GenerateData1< itk::RGBPixel< short >, D >( ); - else if( tIt->second.second == "int" ) - ret = this->_GenerateData1< itk::RGBPixel< int >, D >( ); - else if( tIt->second.second == "long" ) - ret = this->_GenerateData1< itk::RGBPixel< long >, D >( ); - else if( tIt->second.second == "uchar" ) - ret = this->_GenerateData1< itk::RGBPixel< unsigned char >, D >( ); - else if( tIt->second.second == "ushort" ) - ret = this->_GenerateData1< itk::RGBPixel< unsigned short >, D >( ); - else if( tIt->second.second == "uint" ) - ret = this->_GenerateData1< itk::RGBPixel< unsigned int >, D >( ); - else if( tIt->second.second == "ulong" ) - ret = this->_GenerateData1< itk::RGBPixel< unsigned long >, D >( ); - else if( tIt->second.second == "float" ) - ret = this->_GenerateData1< itk::RGBPixel< float >, D >( ); - else if( tIt->second.second == "double" ) - ret = this->_GenerateData1< itk::RGBPixel< double >, D >( ); + if( pt == "char" ) + r = this->_GD1< itk::RGBPixel< char >, D >( ); + else if( pt == "short" ) + r = this->_GD1< itk::RGBPixel< short >, D >( ); + else if( pt == "int" ) + r = this->_GD1< itk::RGBPixel< int >, D >( ); + else if( pt == "long" ) + r = this->_GD1< itk::RGBPixel< long >, D >( ); + else if( pt == "uchar" ) + r = this->_GD1< itk::RGBPixel< unsigned char >, D >( ); + else if( pt == "ushort" ) + r = this->_GD1< itk::RGBPixel< unsigned short >, D >( ); + else if( pt == "uint" ) + r = this->_GD1< itk::RGBPixel< unsigned int >, D >( ); + else if( pt == "ulong" ) + r = this->_GD1< itk::RGBPixel< unsigned long >, D >( ); + else if( pt == "float" ) + r = this->_GD1< itk::RGBPixel< float >, D >( ); + else if( pt == "double" ) + r = this->_GD1< itk::RGBPixel< double >, D >( ); } // fi - return( ret ); + return( r ); } // ------------------------------------------------------------------------- template< class P, unsigned int D > std::string cpPlugins::Plugins::ImageReader:: -_GenerateData1( ) +_GD1( ) { - TParameters::const_iterator fIt; - - // Get image pixelType - fIt = this->m_Parameters.find( "FileName" ); - if( fIt == this->m_Parameters.end( ) ) - fIt = this->m_DefaultParameters.find( "FileName" ); - typedef itk::Image< P, D > _TImage; - typedef itk::ImageFileReader< _TImage > _TImageReader; - _TImageReader* reader = - dynamic_cast< _TImageReader* >( this->m_Reader.GetPointer( ) ); - if( reader == NULL ) - { - this->m_Reader = _TImageReader::New( ); - reader = dynamic_cast< _TImageReader* >( this->m_Reader.GetPointer( ) ); + // Get filenames + using namespace cpPlugins::Interface; + std::vector< Parameters::TString > unordered_names; + this->m_Parameters.GetValueAsStringList( unordered_names, "FileNames" ); - } // fi - reader->SetFileName( fIt->second.second ); - try + if( unordered_names.size( ) == 1 ) { - reader->Update( ); + // Read single image + typedef itk::ImageFileReader< _TImage > _TSingleReader; + + _TSingleReader* singleReader = + dynamic_cast< _TSingleReader* >( + this->m_RealProcessObject.GetPointer( ) + ); + if( singleReader == NULL ) + { + this->m_RealProcessObject = _TSingleReader::New( ); + singleReader = + dynamic_cast< _TSingleReader* >( + this->m_RealProcessObject.GetPointer( ) + ); + + } // fi + singleReader->SetFileName( unordered_names.front( ) ); + try + { + singleReader->Update( ); + } + catch( itk::ExceptionObject& err ) + { + return( err.GetDescription( ) ); + + } // yrt + this->_SetOutput( 0, singleReader->GetOutput( ) ); + return( "" ); } - catch( itk::ExceptionObject& err ) + else if( unordered_names.size( ) > 1 ) { - return( err.GetDescription( ) ); - - } // yrt - this->_SetOutput( 0, reader->GetOutput( ) ); - return( "" ); + // Read image series + std::set< std::string > ordered_names; + for( unsigned int i = 0; i < unordered_names.size( ); ++i ) + ordered_names.insert( unordered_names[ i ] ); + + typedef itk::ImageSeriesReader< _TImage > _TMultiReader; + _TMultiReader* multiReader = + dynamic_cast< _TMultiReader* >( + this->m_RealProcessObject.GetPointer( ) + ); + if( multiReader == NULL ) + { + this->m_RealProcessObject = _TMultiReader::New( ); + multiReader = + dynamic_cast< _TMultiReader* >( + this->m_RealProcessObject.GetPointer( ) + ); + + } // fi + std::set< std::string >::const_iterator fnIt = ordered_names.begin( ); + for( ; fnIt != ordered_names.end( ); ++fnIt ) + multiReader->AddFileName( *fnIt ); + try + { + multiReader->Update( ); + } + catch( itk::ExceptionObject& err ) + { + return( err.GetDescription( ) ); + + } // yrt + this->_SetOutput( 0, multiReader->GetOutput( ) ); + return( "" ); + } + else + return( "No image filename(s) given." ); } // eof - $RCSfile$