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