summaryrefslogtreecommitdiff
path: root/quantize.h
blob: 389090b6e344bac85362951b72ae7a8316c594fc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
/*
 * OSD Picture in Picture plugin for the Video Disk Recorder
 *
 * See the README file for copyright information and how to reach the author.
 */

#ifndef VDR_OSDPIP_QUANTIZE_H
#define VDR_OSDPIP_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_OSDPIP_QUANTIZE_H