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
|
// -*- c++ -*-
#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; }
};
#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
|
