#include "MedianImageFilter.h"
#include <cpPlugins/Interface/Image.h>

#include <itkMedianImageFilter.h>

// -------------------------------------------------------------------------
cpPlugins::BasicFilters::MedianImageFilter::
MedianImageFilter( )
  : Superclass( )
{
  this->_AddInput( "Input" );
  this->_AddOutput< cpPlugins::Interface::Image >( "Output" );

  this->m_Parameters->ConfigureAsUint( "Radius" );
  this->m_Parameters->SetUint( "Radius", 3 );
}

// -------------------------------------------------------------------------
cpPlugins::BasicFilters::MedianImageFilter::
~MedianImageFilter( )
{
}

// -------------------------------------------------------------------------
std::string cpPlugins::BasicFilters::MedianImageFilter::
_GenerateData( )
{
  auto image = this->GetInputData< cpPlugins::Interface::Image >( "Input" );
  if( image == NULL )
    return( "MedianImageFilter: No input image." );

  itk::DataObject* itk_image = NULL;
  std::string r = "";
  cpPlugins_Image_Demangle_AllScalarTypes( 2, image, itk_image, r, _GD0 );
  else cpPlugins_Image_Demangle_AllScalarTypes( 3, image, itk_image, r, _GD0 );
  else cpPlugins_Image_Demangle_AllScalarTypes( 4, image, itk_image, r, _GD0 );
  else r = "MedianImageFilter: Input image type not supported.";
  return( r );
}

// -------------------------------------------------------------------------
template< class I >
std::string cpPlugins::BasicFilters::MedianImageFilter::
_GD0( itk::DataObject* image )
{
  return(
    this->_RealGD< I, itk::Image< unsigned char, I::ImageDimension > >(
      image
      )
    );
}

// -------------------------------------------------------------------------
template< class I, class O >
inline std::string cpPlugins::BasicFilters::MedianImageFilter::
_RealGD( itk::DataObject* image )
{
  typedef itk::MedianImageFilter< I, O > _F;
  typedef typename _F::RadiusType _RT;

  // Get parameters
  _RT rad_val;
  rad_val.Fill( this->m_Parameters->GetUint( "Radius" ) );

  // Configure filter
  _F* filter = this->_CreateITK< _F >( );
  filter->SetInput( dynamic_cast< I* >( image ) );
  filter->SetRadius( rad_val );
  filter->Update( );

  // Connect output
  auto out = this->GetOutputData< cpPlugins::Interface::Image >( "Output" );
  if( out != NULL )
  {
    out->SetITK< O >( filter->GetOutput( ) );
    return( "" );
  }
  else
    return( "MedianImageFilter: output not correctly created." );
}

// eof - $RCSfile$