src/rt/colortab.c

#ifndef lint
static const char       RCSid[] = "$Id: colortab.c,v 2.8 2003/06/30 14:59:12 schorsch Exp $";
#endif
/*
 * colortab.c - allocate and control dynamic color table.
 *
 *      We start off with a uniform partition of color space.
 *      As pixels are sent to the frame buffer, a histogram is built.
 *      When a new color table is requested, the histogram is used
 *      to make a pseudo-optimal partition, after which the
 *      histogram is cleared.  This algorithm
 *      performs only as well as the next drawing's color
 *      distribution is correlated to the last.
 *
 *  External symbols declared in drvier.h
 */

#include "copyright.h"

#include <string.h>

#include "standard.h"
#include "color.h"
#include "driver.h"

                                /* histogram resolution */
#define NRED            24
#define NGRN            32
#define NBLU            16
#define HMAX            NGRN
                                /* minimum box count for adaptive partition */
#define MINSAMP         7
                                /* maximum distance^2 before color reassign */
#define MAXDST2         12
                                /* map a color */
#define map_col(c,p)    clrmap[p][ colval(c,p)<1. ? \
                                (int)(colval(c,p)*256.) : 255 ]
                                /* color partition tree */
#define CNODE           short
#define set_branch(p,c) ((c)<<2|(p))
#define set_pval(pv)    ((pv)<<2|3)
#define is_branch(cn)   (((cn)&3)!=3)
#define is_pval(cn)     (((cn)&3)==3)
#define part(cn)        ((cn)>>2)
#define prim(cn)        ((cn)&3)
#define pval(cn)        ((cn)>>2)
                                /* our color table */
static struct tabent {
        long    sum[3];         /* sum of colors using this entry */
        int     n;              /* number of colors */
        BYTE    ent[3];         /* current table value */
}       *clrtab = NULL;
                                /* color cube partition */
static CNODE    *ctree = NULL;
                                /* our color correction map */
static BYTE     clrmap[3][256];
                                /* histogram of colors used */
static unsigned short   histo[NRED][NGRN][NBLU];
                                /* initial color cube boundary */
static int      CLRCUBE[3][2] = {{0,NRED},{0,NGRN},{0,NBLU}};

static int      split();
static void     cut();


int
new_ctab(ncolors)               /* start new color table with max ncolors */
int     ncolors;
{
        int     treesize;

        if (ncolors < 1)
                return(0);
                                /* free old tables */
        if (clrtab != NULL)
                free((void *)clrtab);
        if (ctree != NULL)
                free((void *)ctree);
                                /* get new tables */
        for (treesize = 1; treesize < ncolors; treesize <<= 1)
                ;
        treesize <<= 1;
        clrtab = (struct tabent *)calloc(ncolors, sizeof(struct tabent));
        ctree = (CNODE *)malloc(treesize*sizeof(CNODE));
        if (clrtab == NULL || ctree == NULL)
                return(0);
                                /* partition color space */
        cut(ctree, 0, CLRCUBE, 0, ncolors);
                                /* clear histogram */
        memset((void *)histo, '\0', sizeof(histo));
                                /* return number of colors used */
        return(ncolors);
}


extern int
get_pixel(      /* get pixel for color */
        COLOR   col,
        dr_newcolrf_t *newcolr
)
{
        int     r, g, b;
        int     cv[3];
        register CNODE  *tp;
        register int    h;
                                                /* map color */
        r = map_col(col,RED);
        g = map_col(col,GRN);
        b = map_col(col,BLU);
                                                /* reduce resolution */
        cv[RED] = (r*NRED)>>8;
        cv[GRN] = (g*NGRN)>>8;
        cv[BLU] = (b*NBLU)>>8;
                                                /* add to histogram */
        histo[cv[RED]][cv[GRN]][cv[BLU]]++;
                                                /* find pixel in tree */
        for (tp = ctree, h = 0; is_branch(*tp); h++)
                if (cv[prim(*tp)] < part(*tp))
                        tp += 1<<h;             /* left branch */
                else
                        tp += 1<<(h+1);         /* right branch */
        h = pval(*tp);
                                                /* add to color table */
        clrtab[h].sum[RED] += r;
        clrtab[h].sum[GRN] += g;
        clrtab[h].sum[BLU] += b;
        clrtab[h].n++;
                                        /* recompute average */
        r = clrtab[h].sum[RED] / clrtab[h].n;
        g = clrtab[h].sum[GRN] / clrtab[h].n;
        b = clrtab[h].sum[BLU] / clrtab[h].n;
                                        /* check for movement */
        if (clrtab[h].n == 1 ||
                        (r-clrtab[h].ent[RED])*(r-clrtab[h].ent[RED]) +
                        (g-clrtab[h].ent[GRN])*(g-clrtab[h].ent[GRN]) +
                        (b-clrtab[h].ent[BLU])*(b-clrtab[h].ent[BLU]) > MAXDST2) {
                clrtab[h].ent[RED] = r;
                clrtab[h].ent[GRN] = g; /* reassign pixel */
                clrtab[h].ent[BLU] = b;
#ifdef DEBUG
                sprintf(errmsg, "pixel %d = (%d,%d,%d) (%d refs)\n",
                                h, r, g, b, clrtab[h].n);
                eputs(errmsg);
#endif
                (*newcolr)(h, r, g, b);
        }
        return(h);                              /* return pixel value */
}


void
make_gmap(gam)                  /* make gamma correction map */
double  gam;
{
        register int    i;
        
        for (i = 0; i < 256; i++)
                clrmap[RED][i] =
                clrmap[GRN][i] =
                clrmap[BLU][i] = 256.0 * pow((i+0.5)/256.0, 1.0/gam);
}


void
set_cmap(rmap, gmap, bmap)      /* set custom color correction map */
BYTE    *rmap, *gmap, *bmap;
{
        memcpy((void *)clrmap[RED], (void *)rmap, 256);
        memcpy((void *)clrmap[GRN], (void *)gmap, 256);
        memcpy((void *)clrmap[BLU], (void *)bmap, 256);
}


void
map_color(rgb, col)             /* map a color to a byte triplet */
BYTE    rgb[3];
COLOR   col;
{
        rgb[RED] = map_col(col,RED);
        rgb[GRN] = map_col(col,GRN);
        rgb[BLU] = map_col(col,BLU);
}


static void
cut(tree, level, box, c0, c1)           /* partition color space */
register CNODE  *tree;
int     level;
register int    box[3][2];
int     c0, c1;
{
        int     kb[3][2];
        
        if (c1-c0 <= 1) {               /* assign pixel */
                *tree = set_pval(c0);
                return;
        }
                                        /* split box */
        *tree = split(box);
        memcpy((void *)kb, (void *)box, sizeof(kb));
                                                /* do left (lesser) branch */
        kb[prim(*tree)][1] = part(*tree);
        cut(tree+(1<<level), level+1, kb, c0, (c0+c1)>>1);
                                                /* do right branch */
        kb[prim(*tree)][0] = part(*tree);
        kb[prim(*tree)][1] = box[prim(*tree)][1];
        cut(tree+(1<<(level+1)), level+1, kb, (c0+c1)>>1, c1);
}


static int
split(box)                              /* find median cut for box */
register int    box[3][2];
{
#define c0      r
        register int    r, g, b;
        int     pri;
        long    t[HMAX], med;
                                        /* find dominant axis */
        pri = RED;
        if (box[GRN][1]-box[GRN][0] > box[pri][1]-box[pri][0])
                pri = GRN;
        if (box[BLU][1]-box[BLU][0] > box[pri][1]-box[pri][0])
                pri = BLU;
                                        /* sum histogram over box */
        med = 0;
        switch (pri) {
        case RED:
                for (r = box[RED][0]; r < box[RED][1]; r++) {
                        t[r] = 0;
                        for (g = box[GRN][0]; g < box[GRN][1]; g++)
                                for (b = box[BLU][0]; b < box[BLU][1]; b++)
                                        t[r] += histo[r][g][b];
                        med += t[r];
                }
                break;
        case GRN:
                for (g = box[GRN][0]; g < box[GRN][1]; g++) {
                        t[g] = 0;
                        for (b = box[BLU][0]; b < box[BLU][1]; b++)
                                for (r = box[RED][0]; r < box[RED][1]; r++)
                                        t[g] += histo[r][g][b];
                        med += t[g];
                }
                break;
        case BLU:
                for (b = box[BLU][0]; b < box[BLU][1]; b++) {
                        t[b] = 0;
                        for (r = box[RED][0]; r < box[RED][1]; r++)
                                for (g = box[GRN][0]; g < box[GRN][1]; g++)
                                        t[b] += histo[r][g][b];
                        med += t[b];
                }
                break;
        }
        if (med < MINSAMP)              /* if too sparse, split at midpoint */
                return(set_branch(pri,(box[pri][0]+box[pri][1])>>1));
                                        /* find median position */
        med >>= 1;
        for (c0 = box[pri][0]; med > 0; c0++)
                med -= t[c0];
        if (c0 > (box[pri][0]+box[pri][1])>>1)  /* if past the midpoint */
                c0--;                           /* part left of median */
        return(set_branch(pri,c0));
#undef c0
}
Revision:	2.9
Committed:	Tue Mar 30 16:13:01 2004 UTC (20 years, 1 month ago) by schorsch
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad3R7P2, rad3R7P1, rad4R0, rad3R6, rad3R6P1, rad3R8, rad3R9
Changes since 2.8:	+8 -6 lines
Log Message:	Continued ANSIfication. There are only bits and pieces left now.
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	schorsch	2.9	static const char RCSid[] = "$Id: colortab.c,v 2.8 2003/06/30 14:59:12 schorsch Exp $";
3	greg	1.1	#endif
4			/*
5			* colortab.c - allocate and control dynamic color table.
6			*
7			* We start off with a uniform partition of color space.
8			* As pixels are sent to the frame buffer, a histogram is built.
9			* When a new color table is requested, the histogram is used
10			* to make a pseudo-optimal partition, after which the
11			* histogram is cleared. This algorithm
12			* performs only as well as the next drawing's color
13			* distribution is correlated to the last.
14	greg	2.5	*
15			* External symbols declared in drvier.h
16			*/
17
18	greg	2.6	#include "copyright.h"
19	greg	1.1
20	schorsch	2.8	#include <string.h>
21
22	greg	1.15	#include "standard.h"
23	schorsch	2.8	#include "color.h"
24	schorsch	2.9	#include "driver.h"
25	greg	1.15
26	greg	1.1	/* histogram resolution */
27	greg	1.3	#define NRED 24
28	greg	1.1	#define NGRN 32
29	greg	1.3	#define NBLU 16
30	greg	1.1	#define HMAX NGRN
31			/* minimum box count for adaptive partition */
32			#define MINSAMP 7
33	greg	1.3	/* maximum distance^2 before color reassign */
34	greg	1.5	#define MAXDST2 12
35	greg	1.1	/* map a color */
36	greg	1.4	#define map_col(c,p) clrmap[p][ colval(c,p)<1. ? \
37			(int)(colval(c,p)*256.) : 255 ]
38	greg	1.1	/* color partition tree */
39			#define CNODE short
40	greg	2.2	#define set_branch(p,c) ((c)<<2\|(p))
41	greg	1.1	#define set_pval(pv) ((pv)<<2\|3)
42	greg	1.7	#define is_branch(cn) (((cn)&3)!=3)
43	greg	1.1	#define is_pval(cn) (((cn)&3)==3)
44			#define part(cn) ((cn)>>2)
45			#define prim(cn) ((cn)&3)
46			#define pval(cn) ((cn)>>2)
47			/* our color table */
48	greg	1.6	static struct tabent {
49	greg	1.3	long sum[3]; /* sum of colors using this entry */
50	greg	1.15	int n; /* number of colors */
51			BYTE ent[3]; /* current table value */
52	greg	1.8	} *clrtab = NULL;
53			/* color cube partition */
54			static CNODE *ctree = NULL;
55	greg	1.1	/* our color correction map */
56	greg	1.4	static BYTE clrmap[3][256];
57	greg	1.1	/* histogram of colors used */
58	greg	1.9	static unsigned short histo[NRED][NGRN][NBLU];
59	greg	1.8	/* initial color cube boundary */
60	greg	2.3	static int CLRCUBE[3][2] = {{0,NRED},{0,NGRN},{0,NBLU}};
61
62	greg	2.5	static int split();
63			static void cut();
64	greg	1.1
65
66	greg	1.3	int
67	greg	1.4	new_ctab(ncolors) /* start new color table with max ncolors */
68	greg	1.1	int ncolors;
69			{
70	greg	1.8	int treesize;
71
72	greg	1.6	if (ncolors < 1)
73	greg	1.3	return(0);
74	greg	1.8	/* free old tables */
75			if (clrtab != NULL)
76	greg	2.5	free((void *)clrtab);
77	greg	1.8	if (ctree != NULL)
78	greg	2.5	free((void *)ctree);
79	greg	1.8	/* get new tables */
80			for (treesize = 1; treesize < ncolors; treesize <<= 1)
81			;
82			treesize <<= 1;
83			clrtab = (struct tabent *)calloc(ncolors, sizeof(struct tabent));
84			ctree = (CNODE )malloc(treesizesizeof(CNODE));
85			if (clrtab == NULL \|\| ctree == NULL)
86			return(0);
87	greg	1.3	/* partition color space */
88	greg	1.7	cut(ctree, 0, CLRCUBE, 0, ncolors);
89	greg	1.1	/* clear histogram */
90	schorsch	2.8	memset((void *)histo, '\0', sizeof(histo));
91	greg	1.3	/* return number of colors used */
92			return(ncolors);
93	greg	1.1	}
94
95
96	schorsch	2.9	extern int
97			get_pixel( /* get pixel for color */
98			COLOR col,
99			dr_newcolrf_t *newcolr
100			)
101	greg	1.1	{
102	greg	1.3	int r, g, b;
103	greg	1.1	int cv[3];
104	greg	1.10	register CNODE *tp;
105	greg	1.1	register int h;
106	greg	1.3	/* map color */
107			r = map_col(col,RED);
108			g = map_col(col,GRN);
109			b = map_col(col,BLU);
110			/* reduce resolution */
111			cv[RED] = (r*NRED)>>8;
112			cv[GRN] = (g*NGRN)>>8;
113			cv[BLU] = (b*NBLU)>>8;
114			/* add to histogram */
115	greg	1.1	histo[cv[RED]][cv[GRN]][cv[BLU]]++;
116	greg	1.3	/* find pixel in tree */
117	greg	1.10	for (tp = ctree, h = 0; is_branch(*tp); h++)
118			if (cv[prim(tp)] < part(tp))
119			tp += 1<<h; /* left branch */
120	greg	1.1	else
121	greg	1.15	tp += 1<<(h+1); /* right branch */
122	greg	1.10	h = pval(*tp);
123	greg	1.3	/* add to color table */
124	greg	1.10	clrtab[h].sum[RED] += r;
125			clrtab[h].sum[GRN] += g;
126			clrtab[h].sum[BLU] += b;
127			clrtab[h].n++;
128	greg	1.3	/* recompute average */
129	greg	1.10	r = clrtab[h].sum[RED] / clrtab[h].n;
130			g = clrtab[h].sum[GRN] / clrtab[h].n;
131			b = clrtab[h].sum[BLU] / clrtab[h].n;
132	greg	1.3	/* check for movement */
133	greg	1.10	if (clrtab[h].n == 1 \|\|
134			(r-clrtab[h].ent[RED])*(r-clrtab[h].ent[RED]) +
135			(g-clrtab[h].ent[GRN])*(g-clrtab[h].ent[GRN]) +
136			(b-clrtab[h].ent[BLU])*(b-clrtab[h].ent[BLU]) > MAXDST2) {
137			clrtab[h].ent[RED] = r;
138	greg	2.2	clrtab[h].ent[GRN] = g; /* reassign pixel */
139	greg	1.10	clrtab[h].ent[BLU] = b;
140	greg	1.12	#ifdef DEBUG
141			sprintf(errmsg, "pixel %d = (%d,%d,%d) (%d refs)\n",
142	greg	1.10	h, r, g, b, clrtab[h].n);
143	greg	1.12	eputs(errmsg);
144	greg	1.1	#endif
145	schorsch	2.9	(*newcolr)(h, r, g, b);
146	greg	1.3	}
147			return(h); /* return pixel value */
148	greg	1.1	}
149
150
151	greg	2.5	void
152	greg	1.4	make_gmap(gam) /* make gamma correction map */
153	greg	2.2	double gam;
154	greg	1.1	{
155			register int i;
156
157	greg	1.4	for (i = 0; i < 256; i++)
158			clrmap[RED][i] =
159			clrmap[GRN][i] =
160	greg	1.15	clrmap[BLU][i] = 256.0 * pow((i+0.5)/256.0, 1.0/gam);
161	greg	1.4	}
162
163
164	greg	2.5	void
165	greg	1.4	set_cmap(rmap, gmap, bmap) /* set custom color correction map */
166			BYTE rmap, gmap, *bmap;
167			{
168	schorsch	2.8	memcpy((void )clrmap[RED], (void )rmap, 256);
169			memcpy((void )clrmap[GRN], (void )gmap, 256);
170			memcpy((void )clrmap[BLU], (void )bmap, 256);
171	greg	1.1	}
172
173
174	greg	2.5	void
175	greg	1.13	map_color(rgb, col) /* map a color to a byte triplet */
176			BYTE rgb[3];
177			COLOR col;
178			{
179			rgb[RED] = map_col(col,RED);
180			rgb[GRN] = map_col(col,GRN);
181			rgb[BLU] = map_col(col,BLU);
182			}
183
184
185	greg	2.5	static void
186	greg	1.7	cut(tree, level, box, c0, c1) /* partition color space */
187	greg	1.1	register CNODE *tree;
188	greg	1.7	int level;
189	greg	1.1	register int box[3][2];
190			int c0, c1;
191			{
192			int kb[3][2];
193
194	greg	1.3	if (c1-c0 <= 1) { /* assign pixel */
195	greg	1.1	*tree = set_pval(c0);
196			return;
197			}
198			/* split box */
199			*tree = split(box);
200	schorsch	2.8	memcpy((void )kb, (void )box, sizeof(kb));
201	greg	1.3	/* do left (lesser) branch */
202	greg	1.1	kb[prim(tree)][1] = part(tree);
203	greg	1.7	cut(tree+(1<<level), level+1, kb, c0, (c0+c1)>>1);
204	greg	1.3	/* do right branch */
205	greg	1.1	kb[prim(tree)][0] = part(tree);
206			kb[prim(tree)][1] = box[prim(tree)][1];
207	greg	1.7	cut(tree+(1<<(level+1)), level+1, kb, (c0+c1)>>1, c1);
208	greg	1.1	}
209
210
211			static int
212			split(box) /* find median cut for box */
213			register int box[3][2];
214			{
215			#define c0 r
216			register int r, g, b;
217			int pri;
218	greg	2.4	long t[HMAX], med;
219	greg	1.1	/* find dominant axis */
220			pri = RED;
221			if (box[GRN][1]-box[GRN][0] > box[pri][1]-box[pri][0])
222			pri = GRN;
223			if (box[BLU][1]-box[BLU][0] > box[pri][1]-box[pri][0])
224			pri = BLU;
225			/* sum histogram over box */
226			med = 0;
227			switch (pri) {
228			case RED:
229			for (r = box[RED][0]; r < box[RED][1]; r++) {
230			t[r] = 0;
231			for (g = box[GRN][0]; g < box[GRN][1]; g++)
232			for (b = box[BLU][0]; b < box[BLU][1]; b++)
233			t[r] += histo[r][g][b];
234			med += t[r];
235			}
236			break;
237			case GRN:
238			for (g = box[GRN][0]; g < box[GRN][1]; g++) {
239			t[g] = 0;
240			for (b = box[BLU][0]; b < box[BLU][1]; b++)
241			for (r = box[RED][0]; r < box[RED][1]; r++)
242			t[g] += histo[r][g][b];
243			med += t[g];
244			}
245			break;
246			case BLU:
247			for (b = box[BLU][0]; b < box[BLU][1]; b++) {
248			t[b] = 0;
249			for (r = box[RED][0]; r < box[RED][1]; r++)
250			for (g = box[GRN][0]; g < box[GRN][1]; g++)
251			t[b] += histo[r][g][b];
252			med += t[b];
253			}
254			break;
255			}
256			if (med < MINSAMP) /* if too sparse, split at midpoint */
257			return(set_branch(pri,(box[pri][0]+box[pri][1])>>1));
258			/* find median position */
259			med >>= 1;
260			for (c0 = box[pri][0]; med > 0; c0++)
261			med -= t[c0];
262			if (c0 > (box[pri][0]+box[pri][1])>>1) /* if past the midpoint */
263			c0--; /* part left of median */
264			return(set_branch(pri,c0));
265			#undef c0
266			}