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" );
  17   this->m_Parameters.ConfigureAsBool( "UseImageSpacing" );
  18   this->m_Parameters.ConfigureAsReal( "ConductanceParameter" );
  19   this->m_Parameters.ConfigureAsReal( "TimeStep" );
  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   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_Image_ScalarPixels_AllDims_1( o, _GD0 )
  47     this->_Error( "No valid input image." );
  48 }
  49
  50 // -------------------------------------------------------------------------
  51 template< class _TImage >
  52 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  53 _GD0( _TImage* image )
  54 {
  55   std::string out_res =
  56     this->m_Parameters.GetSelectedChoice( "OutputResolution" );
  57 #ifdef cpPlugins_CONFIG_REAL_TYPES_float
  58   if( out_res == "float" ) this->_GD1< _TImage, float >( image );
  59 #endif // cpPlugins_CONFIG_REAL_TYPES_float
  60 #ifdef cpPlugins_CONFIG_REAL_TYPES_double
  61   if( out_res == "double" ) this->_GD1< _TImage, double >( image );
  62 #endif // cpPlugins_CONFIG_REAL_TYPES_double
  63 }
  64
  65 // -------------------------------------------------------------------------
  66 template< class _TImage, class _TScalar >
  67 void cpPluginsAnisotropicDiffusion::CurvatureAnisotropicDiffusionImageFilter::
  68 _GD1( _TImage* image )
  69 {
  70   typedef itk::Image< _TScalar, _TImage::ImageDimension > _TOutImage;
  71   typedef
  72     itk::CurvatureAnisotropicDiffusionImageFilter< _TImage, _TOutImage >
  73     _TFilter;
  74
  75   // Configure filter
  76   _TFilter* filter = this->_CreateITK< _TFilter >( );
  77   filter->SetInput( image );
  78   filter->SetNumberOfIterations( this->m_Parameters.GetUint( "NumberOfIterations" ) );
  79   filter->SetUseImageSpacing( this->m_Parameters.GetBool( "UseImageSpacing" ) );
  80   filter->SetConductanceParameter( this->m_Parameters.GetReal( "ConductanceParameter" ) );
  81   filter->SetTimeStep( this->m_Parameters.GetReal( "TimeStep" ) );
  82   filter->Update( );
  83
  84   // Connect output
  85   this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
  86 }
  87
  88 // eof - $RCSfile$