plugins/ImageGradientFilters/MFluxImageFilter.cxx

   1 #include <plugins/ImageGradientFilters/MFluxImageFilter.h>
   2 #include <cpPlugins/DataObjects/Image.h>
   3
   4 #include <cpExtensions/Algorithms/ImageFunctionFilter.h>
   5 #include <cpExtensions/Algorithms/MFluxMedialness.h>
   6
   7 #include <cpExtensions/Algorithms/ImageFunctionFilter.hxx>
   8 #include <cpExtensions/Algorithms/MFluxMedialness.hxx>
   9 #include <cpExtensions/Algorithms/GradientImageFunctionBase.hxx>
  10 #include <itkImageFunction.hxx>
  11
  12 // -------------------------------------------------------------------------
  13 cpPluginsImageGradientFilters::MFluxImageFilter::
  14 MFluxImageFilter( )
  15   : Superclass( )
  16 {
  17   this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Input", true, false );
  18   this->_ConfigureInput< cpPlugins::DataObjects::Image >( "Mask", false, false );
  19   this->_ConfigureOutput< cpPlugins::DataObjects::Image >( "Output" );
  20
  21   this->m_Parameters.ConfigureAsReal( "MinRadius" );
  22   this->m_Parameters.ConfigureAsReal( "MaxRadius" );
  23   this->m_Parameters.ConfigureAsReal( "RadiusStep" );
  24   this->m_Parameters.ConfigureAsUint( "RadialSampling" );
  25
  26   this->m_Parameters.SetReal( "MinRadius", 0 );
  27   this->m_Parameters.SetReal( "MaxRadius", 1 );
  28   this->m_Parameters.SetReal( "RadiusStep", 1 );
  29   this->m_Parameters.SetUint( "RadialSampling", 4 );
  30 }
  31
  32 // -------------------------------------------------------------------------
  33 cpPluginsImageGradientFilters::MFluxImageFilter::
  34 ~MFluxImageFilter( )
  35 {
  36 }
  37
  38 // -------------------------------------------------------------------------
  39 void cpPluginsImageGradientFilters::MFluxImageFilter::
  40 _GenerateData( )
  41 {
  42   auto o = this->GetInputData( "Input" );
  43   cpPlugins_Demangle_ImageCovariantVectors_Dims( o, _GD0 );
  44   else this->_Error( "Invalid input image." );
  45 }
  46
  47 // -------------------------------------------------------------------------
  48 template< class _TImage >
  49 void cpPluginsImageGradientFilters::MFluxImageFilter::
  50 _GD0( _TImage* image )
  51 {
  52   typedef typename _TImage::PixelType _TGradient;
  53   typedef cpExtensions::Algorithms::MFluxMedialness< _TImage > _TFunction;
  54   typedef typename _TFunction::TOutput _TScalar;
  55   typedef itk::Image< _TScalar, _TImage::ImageDimension > _TOutputImage;
  56   typedef cpExtensions::Algorithms::ImageFunctionFilter< _TImage, _TOutputImage, _TFunction > _TFilter;
  57
  58   auto filter = this->_CreateITK< _TFilter >( );
  59   auto function = filter->GetFunction( );
  60   if( function == NULL )
  61   {
  62     filter->SetFunction( _TFunction::New( ) );
  63     function = filter->GetFunction( );
  64
  65   } // fi
  66   function->SetMinRadius( this->m_Parameters.GetReal( "MinRadius" ) );
  67   function->SetMaxRadius( this->m_Parameters.GetReal( "MaxRadius" ) );
  68   function->SetRadiusStep( this->m_Parameters.GetReal( "RadiusStep" ) );
  69   function->SetRadialSampling( this->m_Parameters.GetUint( "RadialSampling" ) );
  70   filter->SetInput( image );
  71   filter->Update( );
  72   this->GetOutput( "Output" )->SetITK( filter->GetOutput( ) );
  73 }
  74
  75 // eof - $RCSfile$