#include <plugins/AnisotropicDiffusion/CurvatureAnisotropicDiffusionImageFilter.h>
#include <cpPlugins/DataObjects/Image.h>

#include <itkCurvatureAnisotropicDiffusionImageFilter.h>
#include <itkCurvatureNDAnisotropicDiffusionFunction.hxx>
#include <itkFiniteDifferenceFunction.hxx>
#include <itkNeighborhoodInnerProduct.hxx>
#include <itkNeighborhoodOperator.hxx>
#include <itkFiniteDifferenceImageFilter.hxx>
#include <itkAnisotropicDiffusionImageFilter.hxx>
#include <itkDenseFiniteDifferenceImageFilter.hxx>
#include <itkNeighborhoodAlgorithm.hxx>
#include <itkDerivativeOperator.hxx>
#include <itkScalarAnisotropicDiffusionFunction.hxx>

// -------------------------------------------------------------------------
cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
CurvatureAnisotropicDiffusionImageFilter( )
  : Superclass( )
{
  this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input", true, false );
  this->_ConfigureOutput< cpPlugins::DataObjects::Image >( "Output" );

  this->m_Parameters.ConfigureAsUint( "NumberOfIterations" );
  this->m_Parameters.ConfigureAsBool( "UseImageSpacing" );
  this->m_Parameters.ConfigureAsReal( "ConductanceParameter" );
  this->m_Parameters.ConfigureAsReal( "TimeStep" );

  this->m_Parameters.SetUint( "NumberOfIterations", 5 );
  this->m_Parameters.SetBool( "UseImageSpacing", false );
  this->m_Parameters.SetReal( "ConductanceParameter", 3 );
  this->m_Parameters.SetReal( "TimeStep", 0.0625 );
}

// -------------------------------------------------------------------------
cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
~CurvatureAnisotropicDiffusionImageFilter( )
{
}

// -------------------------------------------------------------------------
void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
_GenerateData( )
{
  auto o = this->GetInputData( "Input" );
  cpPlugins_Demangle_ImageScalars( o, _GD0, 2 );
  else cpPlugins_Demangle_ImageScalars( o, _GD0, 3 );
  else this->_Error( "No valid input image." );
}

// -------------------------------------------------------------------------
template< class _TImage >
void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
_GD0( _TImage* image )
{
  typedef itk::CurvatureAnisotropicDiffusionImageFilter< _TImage, _TImage > _TFilter;

  // Configure filter
  _TFilter* filter = this->_CreateITK< _TFilter >( );
  filter->SetInput( image );
  filter->SetNumberOfIterations( this->m_Parameters.GetUint( "NumberOfIterations" ) );
  filter->SetUseImageSpacing( this->m_Parameters.GetBool( "UseImageSpacing" ) );
  filter->SetConductanceParameter( this->m_Parameters.GetReal( "ConductanceParameter" ) );
  filter->SetTimeStep( this->m_Parameters.GetReal( "TimeStep" ) );
  filter->Update( );

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

// eof - $RCSfile$