plugins/ITKAnisotropicSmoothing/CurvatureAnisotropicDiffusionImageFilter.cxx

   1 #include <ITKAnisotropicSmoothing/CurvatureAnisotropicDiffusionImageFilter.h>
   2 #include <cpInstances/DataObjects/Image.h>
   3
   4 #include <itkCurvatureAnisotropicDiffusionImageFilter.h>
   5
   6 // -------------------------------------------------------------------------
   7 cpPluginsITKAnisotropicSmoothing::CurvatureAnisotropicDiffusionImageFilter::
   8 CurvatureAnisotropicDiffusionImageFilter( )
   9   : Superclass( )
  10 {
  11   typedef cpInstances::DataObjects::Image _TImage;
  12   this->_ConfigureInput< _TImage >( "Input", true, false );
  13   this->_ConfigureOutput< _TImage >( "Output" );
  14
  15   this->m_Parameters.ConfigureAsUint( "NumberOfIterations", 5 );
  16   this->m_Parameters.ConfigureAsBool( "UseImageSpacing", false );
  17   this->m_Parameters.ConfigureAsReal( "ConductanceParameter", 3 );
  18   this->m_Parameters.ConfigureAsReal( "TimeStep", 0.0625 );
  19   this->m_Parameters.ConfigureAsRealTypesChoices( "OutputResolution" );
  20 }
  21
  22 // -------------------------------------------------------------------------
  23 cpPluginsITKAnisotropicSmoothing::CurvatureAnisotropicDiffusionImageFilter::
  24 ~CurvatureAnisotropicDiffusionImageFilter( )
  25 {
  26 }
  27
  28 // -------------------------------------------------------------------------
  29 void cpPluginsITKAnisotropicSmoothing::CurvatureAnisotropicDiffusionImageFilter::
  30 _GenerateData( )
  31 {
  32   auto o = this->GetInputData( "Input" );
  33   cpPlugins_Demangle_Image_ScalarPixels_AllDims_1( o, _GD0 )
  34     this->_Error( "No valid input image." );
  35 }
  36
  37 // -------------------------------------------------------------------------
  38 template< class _TImage >
  39 void cpPluginsITKAnisotropicSmoothing::CurvatureAnisotropicDiffusionImageFilter::
  40 _GD0( _TImage* image )
  41 {
  42   std::string out_res =
  43     this->m_Parameters.GetSelectedChoice( "OutputResolution" );
  44   if( out_res == "float" ) this->_GD1< _TImage, float >( image );
  45   if( out_res == "double" ) this->_GD1< _TImage, double >( image );
  46 }
  47
  48 // -------------------------------------------------------------------------
  49 template< class _TImage, class _TScalar >
  50 void cpPluginsITKAnisotropicSmoothing::CurvatureAnisotropicDiffusionImageFilter::
  51 _GD1( _TImage* image )
  52 {
  53   typedef itk::Image< _TScalar, _TImage::ImageDimension > _TOutImage;
  54   typedef
  55     itk::CurvatureAnisotropicDiffusionImageFilter< _TImage, _TOutImage >
  56     _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$