ColorImage.h

/*----------------------------------------------------------------------------*/
/* Copyright (c) FIRST 2008. All Rights Reserved.                                                         */
/* Open Source Software - may be modified and shared by FRC teams. The code   */
/* must be accompanied by the FIRST BSD license file in $(WIND_BASE)/WPILib.  */
/*----------------------------------------------------------------------------*/

#ifndef __COLOR_IMAGE_H__
#define __COLOR_IMAGE_H__

#include "ImageBase.h"
#include "BinaryImage.h"
#include "Threshold.h"

class ColorImage : public ImageBase
{
public:
        ColorImage(ImageType type);
        virtual ~ColorImage();
        BinaryImage *ThresholdRGB(int redLow, int redHigh, int greenLow, int greenHigh, int blueLow, int blueHigh);
        BinaryImage *ThresholdHSL(int hueLow, int hueHigh, int saturationLow, int saturationHigh, int luminenceLow, int luminenceHigh);
        BinaryImage *ThresholdHSV(int hueLow, int hueHigh, int saturationLow, int saturationHigh, int valueHigh, int valueLow);
        BinaryImage *ThresholdHSI(int hueLow, int hueHigh, int saturationLow, int saturationHigh, int intensityLow, int intensityHigh);
        BinaryImage *ThresholdRGB(Threshold &threshold);
        BinaryImage *ThresholdHSL(Threshold &threshold);
        BinaryImage *ThresholdHSV(Threshold &threshold);
        BinaryImage *ThresholdHSI(Threshold &threshold);
        MonoImage *GetRedPlane();
        MonoImage *GetGreenPlane();
        MonoImage *GetBluePlane();
        MonoImage *GetHSLHuePlane();
        MonoImage *GetHSVHuePlane();
        MonoImage *GetHSIHuePlane();
        MonoImage *GetHSLSaturationPlane();
        MonoImage *GetHSVSaturationPlane();
        MonoImage *GetHSISaturationPlane();
        MonoImage *GetLuminancePlane();
        MonoImage *GetValuePlane();
        MonoImage *GetIntensityPlane();
        void ReplaceRedPlane(MonoImage *plane);
        void ReplaceGreenPlane(MonoImage *plane);
        void ReplaceBluePlane(MonoImage *plane);
        void ReplaceHSLHuePlane(MonoImage *plane);
        void ReplaceHSVHuePlane(MonoImage *plane);
        void ReplaceHSIHuePlane(MonoImage *plane);
        void ReplaceHSLSaturationPlane(MonoImage *plane);
        void ReplaceHSVSaturationPlane(MonoImage *plane);
        void ReplaceHSISaturationPlane(MonoImage *plane);
        void ReplaceLuminancePlane(MonoImage *plane);
        void ReplaceValuePlane(MonoImage *plane);
        void ReplaceIntensityPlane(MonoImage *plane);
        void ColorEqualize();
        void LuminanceEqualize();
        
private:
        BinaryImage *ComputeThreshold(ColorMode colorMode, int low1, int high1, int low2, int high2, int low3, int high3);
        void Equalize(bool allPlanes);
        MonoImage * ExtractColorPlane(ColorMode mode, int planeNumber);
        MonoImage * ExtractFirstColorPlane(ColorMode mode);
        MonoImage * ExtractSecondColorPlane(ColorMode mode);
        MonoImage * ExtractThirdColorPlane(ColorMode mode);
        void ReplacePlane(ColorMode mode, MonoImage *plane, int planeNumber);
        void ReplaceFirstColorPlane(ColorMode mode, MonoImage *plane);
        void ReplaceSecondColorPlane(ColorMode mode, MonoImage *plane);
        void ReplaceThirdColorPlane(ColorMode mode, MonoImage *plane);
};

#endif