lib/cpPlugins/Plugins/BasicFilters/RGBImageToOtherChannelsFilter.cxx

   1 #include "RGBImageToOtherChannelsFilter.h"
   2 #include <cpPlugins/Interface/Image.h>
   3
   4 #include <cpExtensions/Algorithms/RGBImageToOtherChannelsFilter.h>
   5 #include <cpExtensions/Algorithms/RGBExtractFunction.h>
   6 #include <cpExtensions/Algorithms/RGBToHSVFunction.h>
   7 #include <cpExtensions/Algorithms/RGBToYPbPrFunction.h>
   8
   9 // -------------------------------------------------------------------------
  10 cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  11 RGBImageToOtherChannelsFilter( )
  12   : Superclass( )
  13 {
  14   typedef cpPlugins::Interface::Parameters TParameters;
  15
  16   this->SetNumberOfInputs( 1 );
  17   this->SetNumberOfOutputs( 1 );
  18   this->_MakeOutput< cpPlugins::Interface::Image >( 0 );
  19
  20   this->m_DefaultParameters.Configure( TParameters::String, "OutChannel" );
  21   this->m_Parameters = this->m_DefaultParameters;
  22 }
  23
  24 // -------------------------------------------------------------------------
  25 cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  26 ~RGBImageToOtherChannelsFilter( )
  27 {
  28 }
  29
  30 // -------------------------------------------------------------------------
  31 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  32 _GenerateData( )
  33 {
  34   cpPlugins::Interface::Image* image =
  35     this->GetInput< cpPlugins::Interface::Image >( 0 );
  36   if( image == NULL )
  37     return( "RGBImageToOtherChannelsFilter: No input image." );
  38
  39   itk::DataObject* itk_image = NULL;
  40   std::string r = "";
  41   cpPlugins_Image_Demangle_AllRGBTypes( 2, image, itk_image, r, _GD0 );
  42   else cpPlugins_Image_Demangle_AllRGBTypes( 3, image, itk_image, r, _GD0 );
  43   else cpPlugins_Image_Demangle_AllRGBTypes( 4, image, itk_image, r, _GD0 );
  44   else r = "RGBImageToOtherChannelsFilter: Input image type not supported.";
  45   return( r );
  46 }
  47
  48 // -------------------------------------------------------------------------
  49 template< class I >
  50 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  51 _GD0( itk::DataObject* image )
  52 {
  53   typedef typename I::PixelType _P;
  54
  55   std::string outc = this->m_Parameters.GetValueAsString( "OutChannel" );
  56   std::string r = "";
  57
  58   using namespace cpExtensions::Algorithms;
  59   if( outc == "HSV" )
  60     r = this->_RealGD< I, RGBToHSVFunction< _P > >( image );
  61   else if( outc == "YPbPr" )
  62     r = this->_RealGD< I, RGBToYPbPrFunction< _P > >( image );
  63   else if( outc == "Copy" )
  64     r = this->_RealGD< I, RGBExtractFunction< _P > >( image );
  65   else
  66     r =
  67       std::string( "RGBImageToOtherChannelsFilter: Conversor not available: " ) +
  68       outc;
  69   return( r );
  70 }
  71
  72 // -------------------------------------------------------------------------
  73 template< class I, class C >
  74 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  75 _RealGD( itk::DataObject* image )
  76 {
  77   typedef itk::Image< itk::RGBPixel< double >, I::ImageDimension > _O;
  78   typedef
  79     cpExtensions::Algorithms::RGBImageToOtherChannelsFilter< I, _O, C >
  80     _F;
  81
  82   // Configure filter
  83   _F* filter = this->_CreateITK< _F >( );
  84   filter->SetInput( dynamic_cast< I* >( image ) );
  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( "RGBImageToOtherChannelsFilter: output not correctly created." );
  97
  98   return( "" );
  99 }
 100
 101 // eof - $RCSfile$