plugins/ImageGradientFilters/MultiScaleGaussianImageFilter.cxx

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