#include <ITKSeparableFilters/SmoothingRecursiveGaussianImageFilter.h>
#include <cpInstances/DataObjects/Image.h>

#include <itkSmoothingRecursiveGaussianImageFilter.h>

// -------------------------------------------------------------------------
cpPluginsITKSeparableFilters::SmoothingRecursiveGaussianImageFilter::
SmoothingRecursiveGaussianImageFilter( )
  : Superclass( )
{
  this->_ConfigureInput< cpInstances::DataObjects::Image >( "Input", true, false );
  this->_ConfigureOutput< cpInstances::DataObjects::Image >( "Output" );
  this->m_Parameters.ConfigureAsRealList( "SigmaArray" );
  this->m_Parameters.ConfigureAsRealTypesChoices( "OutputResolution" );
}

// -------------------------------------------------------------------------
cpPluginsITKSeparableFilters::SmoothingRecursiveGaussianImageFilter::
~SmoothingRecursiveGaussianImageFilter( )
{
}

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

// -------------------------------------------------------------------------
template< class _TImage >
void cpPluginsITKSeparableFilters::SmoothingRecursiveGaussianImageFilter::
_GD0( _TImage* image )
{
  std::string out_res =
    this->m_Parameters.GetSelectedChoice( "OutputResolution" );
  if( out_res == "float" ) this->_GD1< _TImage, float >( image );
  if( out_res == "double" ) this->_GD1< _TImage, double >( image );
}

// -------------------------------------------------------------------------
template< class _TImage, class _TScalar >
void cpPluginsITKSeparableFilters::SmoothingRecursiveGaussianImageFilter::
_GD1( _TImage* image )
{
  typedef itk::Image< _TScalar, _TImage::ImageDimension > _TOutput;
  typedef
    itk::SmoothingRecursiveGaussianImageFilter< _TImage, _TOutput >
    _TFilter;
  typedef typename _TFilter::SigmaArrayType _TSigmas;

  auto param_sigmas = this->m_Parameters.GetRealList( "SigmaArray" );
  unsigned int dim =
    ( param_sigmas.size( ) < _TImage::ImageDimension )? 
    param_sigmas.size( ):
    _TImage::ImageDimension;
  _TSigmas sigmas;
  sigmas.Fill( param_sigmas[ dim - 1 ] );
  for( unsigned d = 0; d < dim; ++d )
    sigmas[ d ] = param_sigmas[ d ];

  // Configure filter
  _TFilter* filter = this->_CreateITK< _TFilter >( );
  filter->SetInput( image );
  filter->SetSigmaArray( sigmas );
  filter->Update( );

  // Connect output
  this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
}

// eof - $RCSfile$