lib/cpExtensions/Algorithms/RGBExtractFunction.h

   1 // -------------------------------------------------------------------------
   2 // @author Leonardo Florez-Valencia (florez-l@javeriana.edu.co)
   3 // -------------------------------------------------------------------------
   4
   5 #ifndef __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__
   6 #define __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__
   7
   8 #include <cmath>
   9 #include <limits>
  10 #include <vnl/vnl_math.h>
  11
  12 #include <itkRGBPixel.h>
  13 #include <itkVector.h>
  14
  15 namespace cpExtensions
  16 {
  17   namespace Algorithms
  18   {
  19     /**
  20      */
  21     template< class O >
  22     struct RGBExtractFunction
  23     {
  24       typedef RGBExtractFunction  Self;
  25       typedef itk::Vector< O, 3 > TOutPixel;
  26
  27       template< class Tr, class Tg, class Tb >
  28       TOutPixel operator()( const Tr& r, const Tg& g, const Tb& b ) const
  29         {
  30           TOutPixel rgb;
  31           rgb[ 0 ] = O( r );
  32           rgb[ 1 ] = O( g );
  33           rgb[ 2 ] = O( b );
  34           return( rgb );
  35         }
  36
  37       template< class C >
  38       TOutPixel operator()( const itk::RGBPixel< C >& rgb ) const
  39         {
  40           return(
  41             this->operator()(
  42               rgb.GetRed( ), rgb.GetGreen( ), rgb.GetBlue( )
  43               )
  44             );
  45         }
  46     };
  47
  48   } // ecapseman
  49
  50 } // ecapseman
  51
  52 #endif // __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__
  53
  54 // eof - $RCSfile$