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->SetNumberOfInputs( 1 );
  12   this->SetNumberOfOutputs( 1 );
  13   this->_MakeOutput< cpPlugins::Interface::Image >( 0 );
  14
  15   using namespace cpPlugins::Interface;
  16   this->m_DefaultParameters.Configure(
  17     Parameters::Uint, "NumberOfHistogramBins"
  18     );
  19   this->m_DefaultParameters.Configure( Parameters::Real, "InsideValue" );
  20   this->m_DefaultParameters.Configure( Parameters::Real, "OutsideValue" );
  21   this->m_DefaultParameters.SetValueAsUint( "NumberOfHistogramBins", 100 );
  22   this->m_DefaultParameters.SetValueAsReal( "InsideValue", 255 );
  23   this->m_DefaultParameters.SetValueAsReal( "OutsideValue", 0 );
  24   this->m_Parameters = this->m_DefaultParameters;
  25 }
  26
  27 // -------------------------------------------------------------------------
  28 cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  29 ~OtsuThresholdImageFilter( )
  30 {
  31 }
  32
  33 // -------------------------------------------------------------------------
  34 std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  35 _GenerateData( )
  36 {
  37   cpPlugins::Interface::Image* image =
  38     this->GetInput< cpPlugins::Interface::Image >( 0 );
  39   if( image == NULL )
  40     return( "OtsuThresholdImageFilter: No input image." );
  41
  42   itk::DataObject* itk_image = NULL;
  43   std::string r = "";
  44   cpPlugins_Image_Demangle_AllScalarTypes( 2, image, itk_image, r, _GD0 );
  45   else cpPlugins_Image_Demangle_AllScalarTypes( 3, image, itk_image, r, _GD0 );
  46   else cpPlugins_Image_Demangle_AllScalarTypes( 4, image, itk_image, r, _GD0 );
  47   else r = "OtsuThresholdImageFilter: Input image type not supported.";
  48   return( r );
  49 }
  50
  51 // -------------------------------------------------------------------------
  52 template< class I >
  53 std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  54 _GD0( itk::DataObject* image )
  55 {
  56   return(
  57     this->_RealGD< I, itk::Image< unsigned char, I::ImageDimension > >(
  58       image
  59       )
  60     );
  61 }
  62
  63 // -------------------------------------------------------------------------
  64 template< class I, class O >
  65 inline std::string cpPlugins::BasicFilters::OtsuThresholdImageFilter::
  66 _RealGD( itk::DataObject* image )
  67 {
  68   typedef itk::OtsuThresholdImageFilter< I, O > _F;
  69   typedef typename O::PixelType _OP;
  70
  71   // Get parameters
  72   unsigned int bins =
  73     this->m_Parameters.GetValueAsUint( "NumberOfHistogramBins" );
  74   _OP in_val = _OP( this->m_Parameters.GetValueAsReal( "InsideValue" ) );
  75   _OP out_val = _OP( this->m_Parameters.GetValueAsReal( "OutsideValue" ) );
  76
  77   // Configure filter
  78   _F* filter = this->_CreateITK< _F >( );
  79   filter->SetInput( dynamic_cast< I* >( image ) );
  80   filter->SetNumberOfHistogramBins( bins );
  81   filter->SetInsideValue( in_val );
  82   filter->SetOutsideValue( out_val );
  83   filter->Update( );
  84
  85   // Connect output
  86   cpPlugins::Interface::Image* out =
  87     this->GetOutput< cpPlugins::Interface::Image >( 0 );
  88   if( out != NULL )
  89   {
  90     out->SetITK< O >( filter->GetOutput( ) );
  91     return( "" );
  92   }
  93   else
  94     return( "OtsuThresholdImageFilter: output not correctly created." );
  95 }
  96
  97 // eof - $RCSfile$