plugins/AnisotropicDiffusion/CurvatureAnisotropicDiffusionImageFilter.cxx

   1 #include <AnisotropicDiffusion/CurvatureAnisotropicDiffusionImageFilter.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3 #include <cpPlugins/DataObjects/Image_Demanglers.h>
   4
   5 #include <itkCurvatureAnisotropicDiffusionImageFilter.h>
   6
   7 // -------------------------------------------------------------------------
   8 cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
   9 CurvatureAnisotropicDiffusionImageFilter( )
  10   : Superclass( )
  11 {
  12   typedef cpPlugins::DataObjects::Image _TImage;
  13   this->_ConfigureInput< _TImage >( "Input", true, false );
  14   this->_ConfigureOutput< _TImage >( "Output" );
  15
  16   this->m_Parameters.ConfigureAsUint( "NumberOfIterations", 5 );
  17   this->m_Parameters.ConfigureAsBool( "UseImageSpacing", false );
  18   this->m_Parameters.ConfigureAsReal( "ConductanceParameter", 3 );
  19   this->m_Parameters.ConfigureAsReal( "TimeStep", 0.0625 );
  20   std::vector< std::string > choices;
  21 #ifdef cpPlugins_CONFIG_REAL_TYPES_float
  22   choices.push_back( "float" );
  23 #endif // cpPlugins_CONFIG_REAL_TYPES_float
  24 #ifdef cpPlugins_CONFIG_REAL_TYPES_double
  25   choices.push_back( "double" );
  26 #endif // cpPlugins_CONFIG_REAL_TYPES_double
  27   this->m_Parameters.ConfigureAsChoices( "OutputResolution", choices );
  28 }
  29
  30 // -------------------------------------------------------------------------
  31 cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  32 ~CurvatureAnisotropicDiffusionImageFilter( )
  33 {
  34 }
  35
  36 // -------------------------------------------------------------------------
  37 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  38 _GenerateData( )
  39 {
  40   auto o = this->GetInputData( "Input" );
  41   cpPlugins_Demangle_Image_ScalarPixels_AllDims_1( o, _GD0 )
  42     this->_Error( "No valid input image." );
  43 }
  44
  45 // -------------------------------------------------------------------------
  46 template< class _TImage >
  47 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  48 _GD0( _TImage* image )
  49 {
  50   std::string out_res =
  51     this->m_Parameters.GetSelectedChoice( "OutputResolution" );
  52 #ifdef cpPlugins_CONFIG_REAL_TYPES_float
  53   if( out_res == "float" ) this->_GD1< _TImage, float >( image );
  54 #endif // cpPlugins_CONFIG_REAL_TYPES_float
  55 #ifdef cpPlugins_CONFIG_REAL_TYPES_double
  56   if( out_res == "double" ) this->_GD1< _TImage, double >( image );
  57 #endif // cpPlugins_CONFIG_REAL_TYPES_double
  58 }
  59
  60 // -------------------------------------------------------------------------
  61 template< class _TImage, class _TScalar >
  62 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  63 _GD1( _TImage* image )
  64 {
  65   typedef itk::Image< _TScalar, _TImage::ImageDimension > _TOutImage;
  66   typedef
  67     itk::CurvatureAnisotropicDiffusionImageFilter< _TImage, _TOutImage >
  68     _TFilter;
  69
  70   // Configure filter
  71   _TFilter* filter = this->_CreateITK< _TFilter >( );
  72   filter->SetInput( image );
  73   filter->SetNumberOfIterations( this->m_Parameters.GetUint( "NumberOfIterations" ) );
  74   filter->SetUseImageSpacing( this->m_Parameters.GetBool( "UseImageSpacing" ) );
  75   filter->SetConductanceParameter( this->m_Parameters.GetReal( "ConductanceParameter" ) );
  76   filter->SetTimeStep( this->m_Parameters.GetReal( "TimeStep" ) );
  77   filter->Update( );
  78
  79   // Connect output
  80   this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
  81 }
  82
  83 // eof - $RCSfile$