#include <cpPlugins/Plugins/ImageSeriesReader.h>
#include <cpPlugins/Interface/Image.h>

#include <set>
#include <sstream>

#include <itkImageSeriesReader.h>

#define ITK_MANUAL_INSTANTIATION
#include <itkImage.h>
#include <itkRGBPixel.h>

// -------------------------------------------------------------------------
std::string cpPlugins::Plugins::ImageSeriesReader::
GetClassName( ) const
{
  return( "cpPlugins::Plugins::ImageSeriesReader" );
}

// -------------------------------------------------------------------------
cpPlugins::Plugins::ImageSeriesReader::
ImageSeriesReader( )
  : Superclass( )
{
  this->SetNumberOfOutputs( 1 );
  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;
}

// -------------------------------------------------------------------------
cpPlugins::Plugins::ImageSeriesReader::
~ImageSeriesReader( )
{
}

// -------------------------------------------------------------------------
std::string cpPlugins::Plugins::ImageSeriesReader::
_GenerateData( )
{
  using namespace cpPlugins::Interface;
  Parameters::TUint dim = this->m_Parameters.GetValueAsUint( "Dimension" );

  std::string r = "cpPlugins::Plugins::ImageSeriesReader: itk::Image dimension not supported.";
  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::ImageSeriesReader::
_GD0( )
{
  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::ImageSeriesReader: itk::Image pixel type not supported";
  if( ci == 0 )
  {
    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( ci == 1 )
  {
    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( r );
}

// -------------------------------------------------------------------------
template< class P, unsigned int D >
std::string cpPlugins::Plugins::ImageSeriesReader::
_GD1( )
{
  using namespace cpPlugins::Interface;
  std::vector< Parameters::TString > unordered_names;
  this->m_Parameters.GetValueAsStringList( unordered_names, "FileNames" );
  std::set< std::string > ordered_names;
  for( unsigned int i = 0; i < unordered_names.size( ); ++i )
    ordered_names.insert( unordered_names[ i ] );

  // Reader
  typedef itk::Image< P, D > _TImage;
  typedef itk::ImageSeriesReader< _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( ) );

  } // fi
  std::set< std::string >::const_iterator fnIt = ordered_names.begin( );
  for( ; fnIt != ordered_names.end( ); ++fnIt )
    reader->AddFileName( *fnIt );
  try
  {
    reader->Update( );
  }
  catch( itk::ExceptionObject& err )
  {
    return( err.GetDescription( ) );

  } // yrt
  this->_SetOutput( 0, reader->GetOutput( ) );
  return( "" );
}

// eof - $RCSfile$