plugins/ImageGradientFilters/MultiScaleGaussianImageFilter.cxx

   1 #include <ImageGradientFilters/MultiScaleGaussianImageFilter.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3 #include <cpPlugins/DataObjects/Image_Demanglers.h>
   4
   5 #include <cpExtensions/Algorithms/MultiScaleGaussianImageFilter.h>
   6
   7 // -------------------------------------------------------------------------
   8 cpPluginsImageGradientFilters::MultiScaleGaussianImageFilter::
   9 MultiScaleGaussianImageFilter( )
  10   : Superclass( )
  11 {
  12   this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input", true, false );
  13   this->_ConfigureOutput< cpPlugins::DataObjects::Image >( "Output" );
  14
  15   this->m_Parameters.ConfigureAsRealList( "Sigmas" );
  16
  17   std::vector< std::string > choices;
  18   choices.push_back( "float" );
  19   choices.push_back( "double" );
  20   this->m_Parameters.ConfigureAsChoices( "ScalarType", choices );
  21   this->m_Parameters.SetSelectedChoice( "ScalarType", "float" );
  22 }
  23
  24 // -------------------------------------------------------------------------
  25 cpPluginsImageGradientFilters::MultiScaleGaussianImageFilter::
  26 ~MultiScaleGaussianImageFilter( )
  27 {
  28 }
  29
  30 // -------------------------------------------------------------------------
  31 void cpPluginsImageGradientFilters::MultiScaleGaussianImageFilter::
  32 _GenerateData( )
  33 {
  34   auto o = this->GetInputData( "Input" );
  35   cpPlugins_Demangle_Image_ScalarPixels_AllDims_1( o, _GD0 )
  36     this->_Error( "Invalid input image." );
  37 }
  38
  39 // -------------------------------------------------------------------------
  40 template< class _TImage >
  41 void cpPluginsImageGradientFilters::MultiScaleGaussianImageFilter::
  42 _GD0( _TImage* image )
  43 {
  44   auto choice = this->m_Parameters.GetSelectedChoice( "ScalarType" );
  45   if( choice == "float" )       this->_GD1< _TImage, float >( image );
  46   else if( choice == "double" ) this->_GD1< _TImage, double >( image );
  47   else this->_Error( "Invalid scalar type." );
  48 }
  49
  50 // -------------------------------------------------------------------------
  51 template< class _TInputImage, class _TScalar >
  52 void cpPluginsImageGradientFilters::MultiScaleGaussianImageFilter::
  53 _GD1( _TInputImage* image )
  54 {
  55   typedef itk::CovariantVector< _TScalar, _TInputImage::ImageDimension > _TGradient;
  56   typedef itk::Image< _TGradient, _TInputImage::ImageDimension > _TOutputImage;
  57   typedef
  58     cpExtensions::Algorithms::MultiScaleGaussianImageFilter< _TInputImage, _TOutputImage >
  59     _TFilter;
  60
  61   auto sigmas = this->m_Parameters.GetRealList( "Sigmas" );
  62   if( sigmas.size( ) == 0 )
  63     this->_Error( "No given sigmas." );
  64
  65   // Configure filter
  66   _TFilter* filter = this->_CreateITK< _TFilter >( );
  67   filter->SetInput( image );
  68   for( auto sIt = sigmas.begin( ); sIt != sigmas.end( ); ++sIt )
  69     filter->AddScale( *sIt );
  70   filter->Update( );
  71
  72   // Connect output
  73   this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
  74 }
  75
  76 // eof - $RCSfile$