#include <ImageAlgorithms/RegionGrow.h>
#include <cpInstances/DataObjects/Image.h>

#include <fpa/Image/RegionGrow.h>

// -------------------------------------------------------------------------
fpaPluginsImageAlgorithms::RegionGrow::
RegionGrow( )
  : Superclass( )
{
  typedef cpPlugins::Pipeline::DataObject _TData;

  this->_ConfigureInput< _TData >( "GrowFunction", true, false );
  this->m_Parameters.ConfigureAsInt( "InsideValue", 1 );
  this->m_Parameters.ConfigureAsInt( "OutsideValue", 0 );
  this->m_Parameters.ConfigureAsIntTypesChoices( "ResultType" );
}

// -------------------------------------------------------------------------
fpaPluginsImageAlgorithms::RegionGrow::
~RegionGrow( )
{
}

// -------------------------------------------------------------------------
void fpaPluginsImageAlgorithms::RegionGrow::
_GenerateData( )
{
  auto o = this->GetInputData( "Input" );
  cpPlugins_Demangle_Image_ScalarPixels_AllDims_1( o, _GD0 )
    this->_Error( "Invalid input image." );
}

// -------------------------------------------------------------------------
template< class _TImage >
void fpaPluginsImageAlgorithms::RegionGrow::
_GD0( _TImage* image )
{
  auto rtype = this->m_Parameters.GetSelectedChoice( "ResultType" );
  if( rtype == "char" )        this->_GD1< _TImage, char >( image );
  else if( rtype == "uchar" )  this->_GD1< _TImage, unsigned char >( image );
  else if( rtype == "short" )  this->_GD1< _TImage, short >( image );
  else if( rtype == "ushort" ) this->_GD1< _TImage, unsigned short >( image );
  else if( rtype == "int" )    this->_GD1< _TImage, int >( image );
  else if( rtype == "uint" )   this->_GD1< _TImage, unsigned int >( image );
  else if( rtype == "long" )   this->_GD1< _TImage, long >( image );
  else if( rtype == "ulong" )  this->_GD1< _TImage, unsigned long >( image );
  else                         this->_GD1< _TImage, char >( image );
}

// -------------------------------------------------------------------------
template< class _TInputImage, class _TOutputPixel >
void fpaPluginsImageAlgorithms::RegionGrow::
_GD1( _TInputImage* image )
{
  typedef
    itk::Image< _TOutputPixel, _TInputImage::ImageDimension >
    _TOutputImage;
  typedef fpa::Image::RegionGrow< _TInputImage, _TOutputImage > _TFilter;
  typedef typename _TFilter::TGrowFunction                      _TGrow;

  _TOutputPixel i_val = _TOutputPixel( this->m_Parameters.GetInt( "InsideValue" ) );
  _TOutputPixel o_val = _TOutputPixel( this->m_Parameters.GetInt( "OutsideValue" ) );

  auto filter = this->_CreateITK< _TFilter >( );
  std::vector< typename _TInputImage::IndexType > seeds;
  this->_ConfigureFilter( filter, image, seeds );
  filter->ClearSeeds( );
  for( auto seed : seeds )
    filter->AddSeed( seed, i_val );
  filter->SetGrowFunction( this->GetInputData< _TGrow >( "GrowFunction" ) );
  filter->SetInsideValue( i_val );
  filter->SetOutsideValue( o_val );
  filter->Update( );
  this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
}

// eof - $RCSfile$