lib/cpPlugins/Plugins/BasicFilters/MedianImageFilter.cxx

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