---

/home/slang/XVision2/src/Segmentation/XVColorSeg.h

// *** BEGIN_XVISION2_COPYRIGHT_NOTICE ***
// *** END_XVISION2_COPYRIGHT_NOTICE ***

#ifndef _XVCOLORSEG_H_
#define _XVCOLORSEG_H_

#include <math.h>
#include <XVSegmentation.h>
#include <XVTools.h>
#include <XVImageRGB.h>
#include <XVImageScalar.h>

#define NULL_HUE 5555
#define DEFAULT_DISTANCE  20
#define DEFAULT_SECTOR_NUM 360
#define DEFAULT_HUERANGE_WIDTH 4

typedef struct{

  float min;
  float max;

} HUERANGE;

static const HUERANGE DEFAULT_HUERANGE = {-1, -1};

/************************************************
 * Table and Segmentation classes for Hue based
 * Segmentation
 ************************************************/

/************************************************
 * Table and Segmentation classes for Sector
 * based Segmentation.  The hue values of the
 * pixels are divided into sectors
 * and an image is segmented based on those 
 * sectors
 ************************************************/

template <class T, class Y>
class HueSectorTable : public virtual RGBTable<T, Y>{

 protected:

  u_char darkPix, brightPix;

  int numOfSectors;

  Y computePixelValue(T);
  
 public:
  
  HueSectorTable(int num, u_char dp, u_char bp) : RGBTable<T, Y>(), numOfSectors(num), darkPix(dp), brightPix(bp) { 
    this->buildTable(); 
  };

  int getSectors(){ return numOfSectors; };
};

template <class T, class Y>
class XVHueSeg : public virtual XVSegmentation<T, Y>{

 protected:

  int numOfSectors;

 public:

  XVHueSeg(int sec = DEFAULT_SECTOR_NUM, 
        u_char dp = DEFAULT_DARK_PIXEL, 
        u_char bp = DEFAULT_BRIGHT_PIXEL);
  XVHueSeg(HueSectorTable<T, Y> *);

  float findBestHue(const XVImageRGB<T> &);

  int getNumOfSectors(){ return this->numOfSectors; }

  void update(const XVImageBase<T> & roiIM){};
};

/************************************************
 * Table and Segmentation classes for Color
 * Segmentation based on the distance from some
 * specified color.  
 ************************************************/

template <class T>
class RGBDistanceTable : public virtual RGBTable<T, bool>{

 protected:

  T baseColor;
  float bcLength;
  u_short distance;

  bool computePixelValue(T);
  
 public:

  RGBDistanceTable(T, u_short);
};

inline bool RGB_DISTANCE_CHECK(const bool val){ return val; };

template <class T>
class XVRGBDistanceSeg : public virtual XVSegmentation<T, bool>{

 public:

  XVRGBDistanceSeg(T, u_short dist = DEFAULT_DISTANCE,
             RGBDistanceTable<T> * table = NULL);

  void update(const XVImageBase<T> & roiIM){};

};


/************************************************
 * Table and Segmentation classes for Color
 * Segmentation based on a range of Hue values
 ************************************************/

template <class T, class Y>
class HueRangeTable : public virtual RGBTable<T, Y>{

 protected:

  Y computePixelValue(T);
  u_char darkPix, brightPix;

 public:

  HUERANGE hueRange;

  HueSectorTable<T, Y> * hueTable;
  
  HueRangeTable(HUERANGE range, u_char dp, u_char bp) : RGBTable<T, Y>() { 
    hueTable = new HueSectorTable<T, Y>(DEFAULT_SECTOR_NUM, dp, bp);
    hueRange = range; 
    this->buildTable(); 
  }

  ~HueRangeTable(){ delete hueTable; };

  void updateTable(HUERANGE);
  HUERANGE getRange() { return hueRange; }
};

template <class Y>
inline bool RANGE_SEGFUNC(const Y val){ return (bool)val; }

template <class T, class Y>
class XVHueRangeSeg : public virtual XVSegmentation<T, Y>{

  int rangeWidth;

 public:

  XVHueRangeSeg(HUERANGE range = DEFAULT_HUERANGE, int rW = DEFAULT_HUERANGE_WIDTH,
          int sec = DEFAULT_SECTOR_NUM, 
          u_char dp = DEFAULT_DARK_PIXEL, u_char bp = DEFAULT_BRIGHT_PIXEL);
  XVHueRangeSeg(HueRangeTable<T, Y> *);
  XVHueRangeSeg(XVImageRGB<T> &, int rW = DEFAULT_HUERANGE_WIDTH,
          int sec = DEFAULT_SECTOR_NUM,
          u_char dp = DEFAULT_DARK_PIXEL, u_char bp = DEFAULT_BRIGHT_PIXEL);
 
  void updateTable(HUERANGE);

  float findBestHue(const XVImageRGB<T> &);

  void update(const XVImageBase<T> &);
  HUERANGE getRange(){ return dynamic_cast<HueRangeTable<T, Y> *>(lookup)->getRange(); }
};

#endif

---