/*
 * Graphtft plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 */

#ifndef VDR_GRAPHTFT_QUANTIZE_H
#define VDR_GRAPHTFT_QUANTIZE_H

#define MAXCOLOR 256

class cQuantize
{
protected:
	unsigned int paletteOutput[MAXCOLOR];
	unsigned char imageOutput[1024*1024];
public:
	cQuantize();
	virtual ~cQuantize();
	virtual int Quantize(unsigned char * input, int size, int colors) = 0;
	unsigned int * OutputPalette() { return paletteOutput; }
	unsigned char * OutputImage() { return imageOutput; }
};

class cQuantizeFixed : public cQuantize
{
private:
	unsigned char * imageInput;
	int imageSize;

	unsigned char redLevels[32];
	unsigned char greenLevels[32];
	unsigned char blueLevels[32];
	int redLevelCount;
	int greenLevelCount;
	int blueLevelCount;
public:
	cQuantizeFixed();
	virtual ~cQuantizeFixed();
	virtual int Quantize(unsigned char * input, int size, int colors);
};

#define RED   2
#define	GREEN 1
#define BLUE  0

//#define NOINVERT        // RGB or BGR

struct box
{
	int r0;
	int r1;
	int g0;
	int g1;
	int b0;
	int b1;
	int vol;
};

// Histogram is in elements 1..HISTSIZE along each axis,
// element 0 is for base or marginal value.
// NB: these must start out 0!
#define BOX 33

class cQuantizeWu : public cQuantize
{
private:
	long wt[BOX][BOX][BOX];
	long mr[BOX][BOX][BOX];
	long mg[BOX][BOX][BOX];
	long mb[BOX][BOX][BOX];
	float m2[BOX][BOX][BOX];

	unsigned short * Qadd;

	unsigned char * imageInput;
	int imageSize;
	int palSize;

	void Hist3d(long *vwt, long *vmr, long *vmg, long *vmb, float *m_2);
	void Momt3d(long *vwt, long *vmr, long *vmg, long *vmb, float *m_2);
	int Cut(struct box * set1, struct box * set2);
	long Vol(struct box * cube, long mmt[BOX][BOX][BOX]);
	float Maximize(struct box * cube, unsigned char dir, int first, int last, int * cut, long whole_r, long whole_g, long whole_b, long whole_w);
	float Var(struct box * cube);
	long Top(struct box * cube, unsigned char dir, int pos, long mmt[BOX][BOX][BOX]);
	long Bottom(struct box * cube, unsigned char dir, long mmt[BOX][BOX][BOX]);
	void Mark(struct box * cube, int label, unsigned char * tag);
public:
	cQuantizeWu();
	virtual ~cQuantizeWu();
	virtual int Quantize(unsigned char * input, int size, int colors);
};

#endif // VDR_GRAPHTFT_QUANTIZE_H
