#include <plugins/ImageArithmeticFilters/DivideOrZeroOutImageFilter.h>
#include <cpPlugins/DataObjects/Image.h>

#include <itkDivideOrZeroOutImageFilter.h>
#include <itkBinaryFunctorImageFilter.hxx>

// -------------------------------------------------------------------------
cpPluginsImageArithmeticFilters::DivideOrZeroOutImageFilter::
DivideOrZeroOutImageFilter( )
  : Superclass( )
{
  this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input0", true, false );
  this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input1", true, false );
  this->_ConfigureOutput< cpPlugins::DataObjects::Image >( "Output" );

  this->m_Parameters.ConfigureAsReal( "Threshold" );
  this->m_Parameters.SetReal( "Threshold", 1e-10 );
}

// -------------------------------------------------------------------------
cpPluginsImageArithmeticFilters::DivideOrZeroOutImageFilter::
~DivideOrZeroOutImageFilter( )
{
}

// -------------------------------------------------------------------------
void cpPluginsImageArithmeticFilters::DivideOrZeroOutImageFilter::
_GenerateData( )
{
  auto o = this->GetInputData( "Input0" );
  cpPlugins_Demangle_ImageScalars_Dims( o, _GD0 );
  else this->_Error( "Invalid input image." );
}

// -------------------------------------------------------------------------
template< class _TImage >
void cpPluginsImageArithmeticFilters::DivideOrZeroOutImageFilter::
_GD0( _TImage* image0 )
{
  typedef itk::DivideOrZeroOutImageFilter< _TImage, _TImage, _TImage > _TFilter;

  auto image1 = this->GetInputData< _TImage >( "Input1" );
  if( image1 == NULL )
    this->_Error( "Incompatible second input image." );

  // Configure filter
  auto filter = this->_CreateITK< _TFilter >( );
  filter->SetInput( 0, image0 );
  filter->SetInput( 1, image1 );
  filter->SetThreshold( this->m_Parameters.GetReal( "Threshold" ) );
  filter->Update( );

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

// eof - $RCSfile$