Modules/ImageProcessor/VLCImageProcessor/VLCColorSpace.h

Go to the documentation of this file.

/**
* @file VLCColorSpace.h
* 
* This file contains a class for Image Processing.
*
* @author Tobias Wegner
*/

#ifndef __VLCColorSpace_h_
#define __VLCColorSpace_h_

#include "Tools/Math/Common.h"
#include "Tools/ColorClasses.h"

#define BRIGHTNESS_LEVELS 16
#define DIRECTION_LEVELS  8

class ColorSpace
{
public:
  ColorSpace ();

  void Normalize ();
  double Sum ();

  void AddSample (colorClass color, double brightness, double dY);

  colorClass getMajorColor ();

  ColorSpace& operator+=(const ColorSpace& other)
  {
    int i;
    for (i = 0; i < numOfColors; i++)
      colorCount[i] += other.colorCount[i];

    for (i = 0; i < BRIGHTNESS_LEVELS; i++)
      yCount[i] += other.yCount[i];

    y += other.y;
    dY += other.dY;

    return *this;
  }


  double  colorCount[numOfColors];
  double  yCount[BRIGHTNESS_LEVELS];
  double  y;
  double  dY;
};

class RobotDirection
{
public:
  RobotDirection ();

  void AddSample (double direction) { AddSample (direction, 1); };
  void AddSample (double direction, int weight);

  double getDirection ();
  double getReliability ();

  double directionCount[DIRECTION_LEVELS];
};
#endif

---