// -------------------------------------------------------------------------
// @author Leonardo Florez-Valencia (florez-l@javeriana.edu.co)
// -------------------------------------------------------------------------

#ifndef __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__
#define __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__

#include <cmath>
#include <limits>
#include <vnl/vnl_math.h>

#include <itkRGBPixel.h>
#include <itkVector.h>

namespace cpExtensions
  {
    namespace Algorithms
    {
      /**
       */
      template< class O >
      struct RGBExtractFunction
      {
        typedef RGBExtractFunction  Self;
        typedef itk::Vector< O, 3 > TOutPixel;

        template< class Tr, class Tg, class Tb >
        TOutPixel operator()( const Tr& r, const Tg& g, const Tb& b ) const
          {
            TOutPixel rgb;
            rgb[ 0 ] = O( r );
            rgb[ 1 ] = O( g );
            rgb[ 2 ] = O( b );
            return( rgb );
          }

        template< class C >
        TOutPixel operator()( const itk::RGBPixel< C >& rgb ) const
          {
            return(
              this->operator()(
                rgb.GetRed( ), rgb.GetGreen( ), rgb.GetBlue( )
                )
              );
          }
      };

    } // ecapseman

} // ecapseman

#endif // __CPEXTENSIONS__ALGORITHMS__RGBEXTRACTFUNCTION__H__

// eof - $RCSfile$