plugins/ImageGradientFilters/GulsunTekImageFilter.cxx

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