colour_properties_tool.h

Go to the documentation of this file.

#include "image.h"
#include "rgba.h"
#include <string>

#ifndef COLOUR_PROPERTIES_TOOL_H
#define COLOUR_PROPERTIES_TOOL_H

namespace mimas {
template <typename T>
class colour_properties_tool : public object
{
  private:
    void interpolate(image< rgba< T > > &orig, image< rgba< T > > &ref, double weight);
      
  public:
    colour_properties_tool(void) {}
    void brightness(image< rgba< T > > &im, double brightval); 
    void saturation(image< rgba< T > > &im, double satval); 
    void contrast(image< rgba< T > > &im, double contval); 
    void setChannel(image< rgba< T > > &im, T val, const char *channel); 

};

template <typename T>
void colour_properties_tool<T>::interpolate(image< rgba< T > > &orig, image< rgba< T > > &ref, double weight)
{
  double outr, outg, outb;

  for (int j=0; j<orig.getHeight(); j++)
    for (int i=0; i<orig.getWidth(); i++)
    {
         rgba< T > *origPixel = &orig.pixel( i, j );
         const rgba< T > *refPixel = &ref.pixel( i, j );
      outr=(1.0-weight)*(double)refPixel->getRed() + weight*(double)origPixel->getRed();
      outg=(1.0-weight)*(double)refPixel->getGreen() + weight*(double)origPixel->getGreen();
      outb=(1.0-weight)*(double)refPixel->getBlue() + weight*(double)origPixel->getBlue();

      outr=(outr>(T)255)?(T)255:outr;
      outg=(outg>(T)255)?(T)255:outg;
      outb=(outb>(T)255)?(T)255:outb;
      outr=(outr<(T)0)?(T)0:outr;
      outg=(outg<(T)0)?(T)0:outg;
      outb=(outb<(T)0)?(T)0:outb;
      
      origPixel->setRed((T)outr);
      origPixel->setGreen((T)outg);
      origPixel->setBlue((T)outb);
    }
}


template <typename T>
void colour_properties_tool<T>::brightness(image< rgba< T > > &im, double weight)
{
  image< rgba< T > > ref;
  double w;

  //weight=(weight>1.0)?1.0:weight;
  //weight=(weight<-1.0)?-1.0:weight;

   rgba< T > initVal;

  if (weight>=0.0)
  {
    w=1.0-weight;
      initVal = rgba< T >( 255, 255, 255 );
  }
  else
  {
    w=1.0+weight;
      initVal = rgba< T >(   0,   0,   0 );
  }

  ref.init(im.getWidth(),im.getHeight());
   ref.fill( initVal );
  interpolate(im,ref,w);
}

template <typename T>
void colour_properties_tool<T>::saturation(image< rgba< T > > &im, double weight)
{
  image< rgba< T > > ref;
  double w;

  w=weight+1.0;
  
  ref=im;

  double gray;
  for (int j=0; j<ref.getHeight(); j++)
    for (int i=0; i<ref.getWidth(); i++)
    {
        const rgba< T > *pixel  = &ref.pixel( i, j );
        gray = (double)( pixel->getRed() +
                         pixel->getGreen() +
                         pixel->getBlue() )/3.0;
        ref.pixel( i, j ) = rgba< T >( gray, gray, gray );
    }
  
  interpolate(im,ref,w);

}

template <typename T>
void colour_properties_tool<T>::contrast(image< rgba< T > > &im, double weight)
{
  image< rgba< T > > ref;
  double w;

  w=weight+1.0;

  ref.init(im.getWidth(),im.getHeight());
  ref.fill( rgba< T >( 127, 127, 127 ) );

  interpolate(im,ref,w);

}

template <typename T>
void colour_properties_tool<T>::setChannel(image< rgba< T > > &im, T val, const char *channel)
{
  bool R,G,B,A;
   typedef std::string string;
   string c(channel);
  
  R=c.find("R")!=string::npos?true:false;
  G=c.find("G")!=string::npos?true:false;
  B=c.find("B")!=string::npos?true:false;
  A=c.find("A")!=string::npos?true:false;

   
   MMERROR( (int)(R+G+B+A) == (signed)c.length(), mimasexception, ,
            "colour_properties_tool<T>::zeroChannel() - illegal channel "
            "string" );

  for (int j=0; j<im.getHeight(); j++)
    for (int i=0; i<im.getWidth(); i++)
    {
        rgba< T > *pixel = &im.pixel( i, j );
        if (R) pixel->setRed( (T)val );
        if (G) pixel->setGreen( (T)val );
        if (B) pixel->setBlue( (T)val );
        if (A) pixel->setAlpha( (T)val );
    }
}

}

#endif

---