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   std::vector< std::string > choices;
  21   choices.push_back( "HSV" );
  22   choices.push_back( "YPbPr" );
  23   this->m_Parameters->ConfigureAsChoices( "OutChannel", choices );
  24 }
  25
  26 // -------------------------------------------------------------------------
  27 cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  28 ~RGBImageToOtherChannelsFilter( )
  29 {
  30 }
  31
  32 // -------------------------------------------------------------------------
  33 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  34 _GenerateData( )
  35 {
  36   cpPlugins::Interface::Image* image =
  37     this->GetInput< cpPlugins::Interface::Image >( 0 );
  38   if( image == NULL )
  39     return( "RGBImageToOtherChannelsFilter: No input image." );
  40
  41   itk::DataObject* itk_image = NULL;
  42   std::string r = "";
  43   cpPlugins_Image_Demangle_AllRGBTypes( 2, image, itk_image, r, _GD0 );
  44   else cpPlugins_Image_Demangle_AllRGBTypes( 3, image, itk_image, r, _GD0 );
  45   else cpPlugins_Image_Demangle_AllRGBTypes( 4, image, itk_image, r, _GD0 );
  46   else r = "RGBImageToOtherChannelsFilter: Input image type not supported.";
  47   return( r );
  48 }
  49
  50 // -------------------------------------------------------------------------
  51 template< class I >
  52 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  53 _GD0( itk::DataObject* image )
  54 {
  55   typedef typename I::PixelType _P;
  56
  57   std::string outc = this->m_Parameters->GetSelectedChoice( "OutChannel" );
  58   std::string r = "";
  59
  60   using namespace cpExtensions::Algorithms;
  61   if( outc == "HSV" )
  62     r = this->_RealGD< I, RGBToHSVFunction< _P > >( image );
  63   else if( outc == "YPbPr" )
  64     r = this->_RealGD< I, RGBToYPbPrFunction< _P > >( image );
  65   else if( outc == "Copy" )
  66     r = this->_RealGD< I, RGBExtractFunction< _P > >( image );
  67   else
  68     r =
  69       std::string( "RGBImageToOtherChannelsFilter: Conversor not available: " ) +
  70       outc;
  71   return( r );
  72 }
  73
  74 // -------------------------------------------------------------------------
  75 template< class I, class C >
  76 std::string cpPlugins::BasicFilters::RGBImageToOtherChannelsFilter::
  77 _RealGD( itk::DataObject* image )
  78 {
  79   typedef itk::Image< itk::RGBPixel< double >, I::ImageDimension > _O;
  80   typedef
  81     cpExtensions::Algorithms::RGBImageToOtherChannelsFilter< I, _O, C >
  82     _F;
  83
  84   // Configure filter
  85   _F* filter = this->_CreateITK< _F >( );
  86   filter->SetInput( dynamic_cast< I* >( image ) );
  87   filter->Update( );
  88
  89   // Connect output
  90   cpPlugins::Interface::Image* out =
  91     this->GetOutput< cpPlugins::Interface::Image >( 0 );
  92   if( out != NULL )
  93   {
  94     out->SetITK< _O >( filter->GetOutput( ) );
  95     return( "" );
  96   }
  97   else
  98     return( "RGBImageToOtherChannelsFilter: output not correctly created." );
  99
 100   return( "" );
 101 }
 102
 103 // eof - $RCSfile$