lib/cpPlugins/Plugins/BasicFilters/OtsuThresholdImageFilter.cxx

   1 #include "OtsuThresholdImageFilter.h"
   2 #include <cpPlugins/Interface/Image.h>
   3
   4 #include <itkOtsuThresholdImageFilter.h>
   5
   6 // -------------------------------------------------------------------------
   7 cpPlugins::BasicFilters::OtsuThresholdImageFilter::
   8 OtsuThresholdImageFilter( )
   9   : Superclass( )
  10 {
  11   this->_AddInput( "Input" );
  12   this->_AddOutput< cpPlugins::Interface::Image >( "Output" );
  13
  14   this->m_Parameters->ConfigureAsUint( "NumberOfHistogramBins" );
  15   this->m_Parameters->ConfigureAsUint( "InsideValue" );
  16   this->m_Parameters->ConfigureAsUint( "OutsideValue" );
  17
  18   this->m_Parameters->SetUint( "NumberOfHistogramBins", 100 );
  19   this->m_Parameters->SetUint( "InsideValue", 255 );
  20   this->m_Parameters->SetUint( "OutsideValue", 0 );
  21 }
  22
  23 // -------------------------------------------------------------------------
  24 cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  25 ~OtsuThresholdImageFilter( )
  26 {
  27 }
  28
  29 // -------------------------------------------------------------------------
  30 std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  31 _GenerateData( )
  32 {
  33   auto image = this->GetInputData( "Input" );
  34   itk::DataObject* itk_image = NULL;
  35   std::string r = "";
  36   cpPlugins_Image_Demangle_AllScalarTypes( 2, image, itk_image, r, _GD0 );
  37   else cpPlugins_Image_Demangle_AllScalarTypes( 3, image, itk_image, r, _GD0 );
  38   else cpPlugins_Image_Demangle_AllScalarTypes( 4, image, itk_image, r, _GD0 );
  39   else r = "OtsuThresholdImageFilter: Input image type not supported.";
  40   return( r );
  41 }
  42
  43 // -------------------------------------------------------------------------
  44 template< class I >
  45 std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  46 _GD0( itk::DataObject* image )
  47 {
  48   return(
  49     this->_RealGD< I, itk::Image< unsigned char, I::ImageDimension > >(
  50       image
  51       )
  52     );
  53 }
  54
  55 // -------------------------------------------------------------------------
  56 template< class I, class O >
  57 inline std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  58 _RealGD( itk::DataObject* image )
  59 {
  60   typedef itk::OtsuThresholdImageFilter< I, O > _F;
  61   typedef typename O::PixelType _OP;
  62
  63   // Get parameters
  64   unsigned int bins = this->m_Parameters->GetUint( "NumberOfHistogramBins" );
  65   _OP in_val = _OP( this->m_Parameters->GetUint( "InsideValue" ) );
  66   _OP out_val = _OP( this->m_Parameters->GetUint( "OutsideValue" ) );
  67
  68   // Configure filter
  69   _F* filter = this->_CreateITK< _F >( );
  70   filter->SetInput( dynamic_cast< I* >( image ) );
  71   filter->SetNumberOfHistogramBins( bins );
  72   filter->SetInsideValue( out_val ); // WARNING: these are inverted
  73   filter->SetOutsideValue( in_val );
  74   filter->Update( );
  75
  76   // Connect output
  77   auto out = this->GetOutputData( "Output" );
  78   out->SetITK( filter->GetOutput( ) );
  79   return( "" );
  80 }
  81
  82 // eof - $RCSfile$