plugins/AnisotropicDiffusion/CurvatureAnisotropicDiffusionImageFilter.cxx

   1 #include <plugins/AnisotropicDiffusion/CurvatureAnisotropicDiffusionImageFilter.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3
   4 #include <itkCurvatureAnisotropicDiffusionImageFilter.h>
   5 #include <itkCurvatureNDAnisotropicDiffusionFunction.hxx>
   6 #include <itkFiniteDifferenceFunction.hxx>
   7 #include <itkNeighborhoodInnerProduct.hxx>
   8 #include <itkNeighborhoodOperator.hxx>
   9 #include <itkFiniteDifferenceImageFilter.hxx>
  10 #include <itkAnisotropicDiffusionImageFilter.hxx>
  11 #include <itkDenseFiniteDifferenceImageFilter.hxx>
  12 #include <itkNeighborhoodAlgorithm.hxx>
  13 #include <itkDerivativeOperator.hxx>
  14 #include <itkScalarAnisotropicDiffusionFunction.hxx>
  15
  16 // -------------------------------------------------------------------------
  17 cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  18 CurvatureAnisotropicDiffusionImageFilter( )
  19   : Superclass( )
  20 {
  21   this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input", true, false );
  22   this->_ConfigureOutput< cpPlugins::DataObjects::Image >( "Output" );
  23
  24   this->m_Parameters.ConfigureAsUint( "NumberOfIterations" );
  25   this->m_Parameters.ConfigureAsBool( "UseImageSpacing" );
  26   this->m_Parameters.ConfigureAsReal( "ConductanceParameter" );
  27   this->m_Parameters.ConfigureAsReal( "TimeStep" );
  28
  29   this->m_Parameters.SetUint( "NumberOfIterations", 5 );
  30   this->m_Parameters.SetBool( "UseImageSpacing", false );
  31   this->m_Parameters.SetReal( "ConductanceParameter", 3 );
  32   this->m_Parameters.SetReal( "TimeStep", 0.0625 );
  33 }
  34
  35 // -------------------------------------------------------------------------
  36 cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  37 ~CurvatureAnisotropicDiffusionImageFilter( )
  38 {
  39 }
  40
  41 // -------------------------------------------------------------------------
  42 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  43 _GenerateData( )
  44 {
  45   auto o = this->GetInputData( "Input" );
  46   cpPlugins_Demangle_ImageScalars( o, _GD0, 2 );
  47   else cpPlugins_Demangle_ImageScalars( o, _GD0, 3 );
  48   else this->_Error( "No valid input image." );
  49 }
  50
  51 // -------------------------------------------------------------------------
  52 template< class _TImage >
  53 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  54 _GD0( _TImage* image )
  55 {
  56   typedef itk::CurvatureAnisotropicDiffusionImageFilter< _TImage, _TImage > _TFilter;
  57
  58   // Configure filter
  59   _TFilter* filter = this->_CreateITK< _TFilter >( );
  60   filter->SetInput( image );
  61   filter->SetNumberOfIterations( this->m_Parameters.GetUint( "NumberOfIterations" ) );
  62   filter->SetUseImageSpacing( this->m_Parameters.GetBool( "UseImageSpacing" ) );
  63   filter->SetConductanceParameter( this->m_Parameters.GetReal( "ConductanceParameter" ) );
  64   filter->SetTimeStep( this->m_Parameters.GetReal( "TimeStep" ) );
  65   filter->Update( );
  66
  67   // Connect output
  68   this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
  69 }
  70
  71 // eof - $RCSfile$