]> Creatis software - cpPlugins.git/blobdiff - lib/cpPlugins/Plugins/ImageReader.cxx
Light compensation algorithm added
[cpPlugins.git] / lib / cpPlugins / Plugins / ImageReader.cxx
index 0aa4bfc233f7ae56f0f57c259ec86ed8cd9cd711..da7b949485bcb72e7bb9947b3a63df4dd88f1ea7 100644 (file)
@@ -2,6 +2,7 @@
 #include <cpPlugins/Interface/Image.h>
 
 #include <itkImageFileReader.h>
+#include <itkImageSeriesReader.h>
 
 #define ITK_MANUAL_INSTANTIATION
 #include <itkImage.h>
@@ -22,11 +23,15 @@ ImageReader( )
   this->SetNumberOfOutputs( 1 );
   this->_MakeOutput< cpPlugins::Interface::Image >( 0 );
 
-  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" );
+  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;
 }
 
 // -------------------------------------------------------------------------
@@ -39,18 +44,13 @@ 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" );
-
+  using namespace cpPlugins::Interface;
+  Parameters::TUint dim = this->m_Parameters.GetValueAsUint( "Dimension" );
   std::string r = "cpPlugins::Plugins::ImageReader: itk::Image dimension not supported.";
-  if     ( dIt->second.second == "1" ) r = this->_GD0< 1 >( );
-  else if( dIt->second.second == "2" ) r = this->_GD0< 2 >( );
-  else if( dIt->second.second == "3" ) r = this->_GD0< 3 >( );
-  else if( dIt->second.second == "4" ) r = this->_GD0< 4 >( );
+  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 );
 }
@@ -60,61 +60,55 @@ template< unsigned int D >
 std::string cpPlugins::Plugins::ImageReader::
 _GD0( )
 {
-  TParameters::const_iterator tIt, cIt;
-
-  // Get image pixel type
-  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" );
+  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( cIt->second.second == "0" )
+  if( ci == 0 )
   {
-    if( tIt->second.second == "char" )
+    if( pt == "char" )
       r = this->_GD1< char, D >( );
-    else if( tIt->second.second == "short" )
+    else if( pt == "short" )
       r = this->_GD1< short, D >( );
-    else if( tIt->second.second == "int" )
+    else if( pt == "int" )
       r = this->_GD1< int, D >( );
-    else if( tIt->second.second == "long" )
+    else if( pt == "long" )
       r = this->_GD1< long, D >( );
-    else if( tIt->second.second == "uchar" )
+    else if( pt == "uchar" )
       r = this->_GD1< unsigned char, D >( );
-    else if( tIt->second.second == "ushort" )
+    else if( pt == "ushort" )
       r = this->_GD1< unsigned short, D >( );
-    else if( tIt->second.second == "uint" )
+    else if( pt == "uint" )
       r = this->_GD1< unsigned int, D >( );
-    else if( tIt->second.second == "ulong" )
+    else if( pt == "ulong" )
       r = this->_GD1< unsigned long, D >( );
-    else if( tIt->second.second == "float" )
+    else if( pt == "float" )
       r = this->_GD1< float, D >( );
-    else if( tIt->second.second == "double" )
+    else if( pt == "double" )
       r = this->_GD1< double, D >( );
   }
-  else if( cIt->second.second == "1" )
+  else
   {
-    if( tIt->second.second == "char" )
+    if( pt == "char" )
       r = this->_GD1< itk::RGBPixel< char >, D >( );
-    else if( tIt->second.second == "short" )
+    else if( pt == "short" )
       r = this->_GD1< itk::RGBPixel< short >, D >( );
-    else if( tIt->second.second == "int" )
+    else if( pt == "int" )
       r = this->_GD1< itk::RGBPixel< int >, D >( );
-    else if( tIt->second.second == "long" )
+    else if( pt == "long" )
       r = this->_GD1< itk::RGBPixel< long >, D >( );
-    else if( tIt->second.second == "uchar" )
+    else if( pt == "uchar" )
       r = this->_GD1< itk::RGBPixel< unsigned char >, D >( );
-    else if( tIt->second.second == "ushort" )
+    else if( pt == "ushort" )
       r = this->_GD1< itk::RGBPixel< unsigned short >, D >( );
-    else if( tIt->second.second == "uint" )
+    else if( pt == "uint" )
       r = this->_GD1< itk::RGBPixel< unsigned int >, D >( );
-    else if( tIt->second.second == "ulong" )
+    else if( pt == "ulong" )
       r = this->_GD1< itk::RGBPixel< unsigned long >, D >( );
-    else if( tIt->second.second == "float" )
+    else if( pt == "float" )
       r = this->_GD1< itk::RGBPixel< float >, D >( );
-    else if( tIt->second.second == "double" )
+    else if( pt == "double" )
       r = this->_GD1< itk::RGBPixel< double >, D >( );
   } // fi
   return( r );
@@ -125,37 +119,82 @@ template< class P, unsigned int D >
 std::string cpPlugins::Plugins::ImageReader::
 _GD1( )
 {
-  TParameters::const_iterator fIt;
-
-  // Get filename
-  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 > _TReader;
 
-  _TReader* reader =
-    dynamic_cast< _TReader* >( this->m_RealProcessObject.GetPointer( ) );
-  if( reader == NULL )
-  {
-    this->m_RealProcessObject = _TReader::New( );
-    reader =
-      dynamic_cast< _TReader* >( this->m_RealProcessObject.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$