src/common/color.c

#ifndef lint
static const char       RCSid[] = "$Id: color.c,v 2.15 2004/09/14 02:53:50 greg Exp $";
#endif
/*
 *  color.c - routines for color calculations.
 *
 *  Externals declared in color.h
 */

#include "copyright.h"

#include  <stdio.h>

#include  <stdlib.h>

#include  <math.h>

#include  "color.h"

#ifdef getc_unlocked            /* avoid horrendous overhead of flockfile */
#undef getc
#undef putc
#define getc    getc_unlocked
#define putc    putc_unlocked
#endif

#define  MINELEN        8       /* minimum scanline length for encoding */
#define  MAXELEN        0x7fff  /* maximum scanline length for encoding */
#define  MINRUN         4       /* minimum run length */


char *
tempbuffer(len)                 /* get a temporary buffer */
unsigned int  len;
{
        static char  *tempbuf = NULL;
        static unsigned  tempbuflen = 0;

        if (len > tempbuflen) {
                if (tempbuflen > 0)
                        tempbuf = (char *)realloc((void *)tempbuf, len);
                else
                        tempbuf = (char *)malloc(len);
                tempbuflen = tempbuf==NULL ? 0 : len;
        }
        return(tempbuf);
}


int
fwritecolrs(scanline, len, fp)          /* write out a colr scanline */
register COLR  *scanline;
int  len;
register FILE  *fp;
{
        register int  i, j, beg, cnt = 1;
        int  c2;
        
        if ((len < MINELEN) | (len > MAXELEN))  /* OOBs, write out flat */
                return(fwrite((char *)scanline,sizeof(COLR),len,fp) - len);
                                        /* put magic header */
        putc(2, fp);
        putc(2, fp);
        putc(len>>8, fp);
        putc(len&255, fp);
                                        /* put components seperately */
        for (i = 0; i < 4; i++) {
            for (j = 0; j < len; j += cnt) {    /* find next run */
                for (beg = j; beg < len; beg += cnt) {
                    for (cnt = 1; cnt < 127 && beg+cnt < len &&
                            scanline[beg+cnt][i] == scanline[beg][i]; cnt++)
                        ;
                    if (cnt >= MINRUN)
                        break;                  /* long enough */
                }
                if (beg-j > 1 && beg-j < MINRUN) {
                    c2 = j+1;
                    while (scanline[c2++][i] == scanline[j][i])
                        if (c2 == beg) {        /* short run */
                            putc(128+beg-j, fp);
                            putc(scanline[j][i], fp);
                            j = beg;
                            break;
                        }
                }
                while (j < beg) {               /* write out non-run */
                    if ((c2 = beg-j) > 128) c2 = 128;
                    putc(c2, fp);
                    while (c2--)
                        putc(scanline[j++][i], fp);
                }
                if (cnt >= MINRUN) {            /* write out run */
                    putc(128+cnt, fp);
                    putc(scanline[beg][i], fp);
                } else
                    cnt = 0;
            }
        }
        return(ferror(fp) ? -1 : 0);
}


static int
oldreadcolrs(scanline, len, fp)         /* read in an old colr scanline */
register COLR  *scanline;
int  len;
register FILE  *fp;
{
        int  rshift;
        register int  i;
        
        rshift = 0;
        
        while (len > 0) {
                scanline[0][RED] = getc(fp);
                scanline[0][GRN] = getc(fp);
                scanline[0][BLU] = getc(fp);
                scanline[0][EXP] = getc(fp);
                if (feof(fp) || ferror(fp))
                        return(-1);
                if (scanline[0][RED] == 1 &&
                                scanline[0][GRN] == 1 &&
                                scanline[0][BLU] == 1) {
                        for (i = scanline[0][EXP] << rshift; i > 0; i--) {
                                copycolr(scanline[0], scanline[-1]);
                                scanline++;
                                len--;
                        }
                        rshift += 8;
                } else {
                        scanline++;
                        len--;
                        rshift = 0;
                }
        }
        return(0);
}


int
freadcolrs(scanline, len, fp)           /* read in an encoded colr scanline */
register COLR  *scanline;
int  len;
register FILE  *fp;
{
        register int  i, j;
        int  code, val;
                                        /* determine scanline type */
        if ((len < MINELEN) | (len > MAXELEN))
                return(oldreadcolrs(scanline, len, fp));
        if ((i = getc(fp)) == EOF)
                return(-1);
        if (i != 2) {
                ungetc(i, fp);
                return(oldreadcolrs(scanline, len, fp));
        }
        scanline[0][GRN] = getc(fp);
        scanline[0][BLU] = getc(fp);
        if ((i = getc(fp)) == EOF)
                return(-1);
        if (scanline[0][GRN] != 2 || scanline[0][BLU] & 128) {
                scanline[0][RED] = 2;
                scanline[0][EXP] = i;
                return(oldreadcolrs(scanline+1, len-1, fp));
        }
        if ((scanline[0][BLU]<<8 | i) != len)
                return(-1);             /* length mismatch! */
                                        /* read each component */
        for (i = 0; i < 4; i++)
            for (j = 0; j < len; ) {
                if ((code = getc(fp)) == EOF)
                    return(-1);
                if (code > 128) {       /* run */
                    code &= 127;
                    if ((val = getc(fp)) == EOF)
                        return -1;
                    if (j + code > len)
                        return -1;      /* overrun */
                    while (code--)
                        scanline[j++][i] = val;
                } else {                /* non-run */
                    if (j + code > len)
                        return -1;      /* overrun */
                    while (code--) {
                        if ((val = getc(fp)) == EOF)
                            return -1;
                        scanline[j++][i] = val;
                    }
                }
            }
        return(0);
}


int
fwritescan(scanline, len, fp)           /* write out a scanline */
register COLOR  *scanline;
int  len;
FILE  *fp;
{
        COLR  *clrscan;
        int  n;
        register COLR  *sp;
                                        /* get scanline buffer */
        if ((sp = (COLR *)tempbuffer(len*sizeof(COLR))) == NULL)
                return(-1);
        clrscan = sp;
                                        /* convert scanline */
        n = len;
        while (n-- > 0) {
                setcolr(sp[0], scanline[0][RED],
                                  scanline[0][GRN],
                                  scanline[0][BLU]);
                scanline++;
                sp++;
        }
        return(fwritecolrs(clrscan, len, fp));
}


int
freadscan(scanline, len, fp)            /* read in a scanline */
register COLOR  *scanline;
int  len;
FILE  *fp;
{
        register COLR  *clrscan;

        if ((clrscan = (COLR *)tempbuffer(len*sizeof(COLR))) == NULL)
                return(-1);
        if (freadcolrs(clrscan, len, fp) < 0)
                return(-1);
                                        /* convert scanline */
        colr_color(scanline[0], clrscan[0]);
        while (--len > 0) {
                scanline++; clrscan++;
                if (clrscan[0][RED] == clrscan[-1][RED] &&
                            clrscan[0][GRN] == clrscan[-1][GRN] &&
                            clrscan[0][BLU] == clrscan[-1][BLU] &&
                            clrscan[0][EXP] == clrscan[-1][EXP])
                        copycolor(scanline[0], scanline[-1]);
                else
                        colr_color(scanline[0], clrscan[0]);
        }
        return(0);
}


void
setcolr(clr, r, g, b)           /* assign a short color value */
register COLR  clr;
double  r, g, b;
{
        double  d;
        int  e;
        
        d = r > g ? r : g;
        if (b > d) d = b;

        if (d <= 1e-32) {
                clr[RED] = clr[GRN] = clr[BLU] = 0;
                clr[EXP] = 0;
                return;
        }

        d = frexp(d, &e) * 255.9999 / d;

        if (r > 0.0)
                clr[RED] = r * d;
        else
                clr[RED] = 0;
        if (g > 0.0)
                clr[GRN] = g * d;
        else
                clr[GRN] = 0;
        if (b > 0.0)
                clr[BLU] = b * d;
        else
                clr[BLU] = 0;

        clr[EXP] = e + COLXS;
}


void
colr_color(col, clr)            /* convert short to float color */
register COLOR  col;
register COLR  clr;
{
        double  f;
        
        if (clr[EXP] == 0)
                col[RED] = col[GRN] = col[BLU] = 0.0;
        else {
                f = ldexp(1.0, (int)clr[EXP]-(COLXS+8));
                col[RED] = (clr[RED] + 0.5)*f;
                col[GRN] = (clr[GRN] + 0.5)*f;
                col[BLU] = (clr[BLU] + 0.5)*f;
        }
}


int
bigdiff(c1, c2, md)                     /* c1 delta c2 > md? */
register COLOR  c1, c2;
double  md;
{
        register int  i;

        for (i = 0; i < 3; i++)
                if (colval(c1,i)-colval(c2,i) > md*colval(c2,i) ||
                        colval(c2,i)-colval(c1,i) > md*colval(c1,i))
                        return(1);
        return(0);
}
Revision:	2.16
Committed:	Wed Feb 9 00:00:17 2005 UTC (19 years, 2 months ago) by greg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad3R7P2, rad3R7P1, rad4R1, rad4R0, rad3R8, rad3R9
Changes since 2.15:	+7 -2 lines
Log Message:	Added checks for overrun scanlines
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.16	static const char RCSid[] = "$Id: color.c,v 2.15 2004/09/14 02:53:50 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* color.c - routines for color calculations.
6			*
7	greg	2.9	* Externals declared in color.h
8			*/
9
10	greg	2.10	#include "copyright.h"
11	greg	1.1
12			#include <stdio.h>
13
14	greg	2.9	#include <stdlib.h>
15
16	greg	2.7	#include <math.h>
17
18	greg	1.1	#include "color.h"
19	greg	2.14
20			#ifdef getc_unlocked /* avoid horrendous overhead of flockfile */
21	greg	2.15	#undef getc
22			#undef putc
23	greg	2.14	#define getc getc_unlocked
24			#define putc putc_unlocked
25			#endif
26	greg	1.1
27	greg	1.14	#define MINELEN 8 /* minimum scanline length for encoding */
28	greg	2.6	#define MAXELEN 0x7fff /* maximum scanline length for encoding */
29	greg	1.14	#define MINRUN 4 /* minimum run length */
30
31	greg	2.4
32	greg	1.14	char *
33			tempbuffer(len) /* get a temporary buffer */
34	greg	2.9	unsigned int len;
35	greg	1.14	{
36			static char *tempbuf = NULL;
37	greg	2.5	static unsigned tempbuflen = 0;
38	greg	1.14
39			if (len > tempbuflen) {
40			if (tempbuflen > 0)
41	greg	2.12	tempbuf = (char )realloc((void )tempbuf, len);
42	greg	1.14	else
43	greg	2.9	tempbuf = (char *)malloc(len);
44	greg	1.14	tempbuflen = tempbuf==NULL ? 0 : len;
45			}
46			return(tempbuf);
47			}
48
49
50	greg	2.9	int
51	greg	1.1	fwritecolrs(scanline, len, fp) /* write out a colr scanline */
52			register COLR *scanline;
53	greg	2.9	int len;
54	greg	1.1	register FILE *fp;
55			{
56	greg	2.9	register int i, j, beg, cnt = 1;
57	greg	1.14	int c2;
58	greg	1.1
59	schorsch	2.13	if ((len < MINELEN) \| (len > MAXELEN)) /* OOBs, write out flat */
60	greg	1.14	return(fwrite((char *)scanline,sizeof(COLR),len,fp) - len);
61	greg	2.2	/* put magic header */
62			putc(2, fp);
63	greg	1.14	putc(2, fp);
64			putc(len>>8, fp);
65			putc(len&255, fp);
66			/* put components seperately */
67			for (i = 0; i < 4; i++) {
68			for (j = 0; j < len; j += cnt) { /* find next run */
69			for (beg = j; beg < len; beg += cnt) {
70			for (cnt = 1; cnt < 127 && beg+cnt < len &&
71			scanline[beg+cnt][i] == scanline[beg][i]; cnt++)
72			;
73			if (cnt >= MINRUN)
74			break; /* long enough */
75	greg	1.1	}
76	greg	1.15	if (beg-j > 1 && beg-j < MINRUN) {
77			c2 = j+1;
78			while (scanline[c2++][i] == scanline[j][i])
79			if (c2 == beg) { /* short run */
80			putc(128+beg-j, fp);
81			putc(scanline[j][i], fp);
82			j = beg;
83			break;
84			}
85			}
86			while (j < beg) { /* write out non-run */
87	greg	1.14	if ((c2 = beg-j) > 128) c2 = 128;
88			putc(c2, fp);
89			while (c2--)
90			putc(scanline[j++][i], fp);
91			}
92			if (cnt >= MINRUN) { /* write out run */
93			putc(128+cnt, fp);
94			putc(scanline[beg][i], fp);
95			} else
96			cnt = 0;
97			}
98	greg	1.1	}
99			return(ferror(fp) ? -1 : 0);
100			}
101
102
103	greg	2.9	static int
104			oldreadcolrs(scanline, len, fp) /* read in an old colr scanline */
105			register COLR *scanline;
106			int len;
107			register FILE *fp;
108			{
109			int rshift;
110			register int i;
111
112			rshift = 0;
113
114			while (len > 0) {
115			scanline[0][RED] = getc(fp);
116			scanline[0][GRN] = getc(fp);
117			scanline[0][BLU] = getc(fp);
118			scanline[0][EXP] = getc(fp);
119			if (feof(fp) \|\| ferror(fp))
120			return(-1);
121			if (scanline[0][RED] == 1 &&
122			scanline[0][GRN] == 1 &&
123			scanline[0][BLU] == 1) {
124			for (i = scanline[0][EXP] << rshift; i > 0; i--) {
125			copycolr(scanline[0], scanline[-1]);
126			scanline++;
127			len--;
128			}
129			rshift += 8;
130			} else {
131			scanline++;
132			len--;
133			rshift = 0;
134			}
135			}
136			return(0);
137			}
138
139
140			int
141	greg	1.14	freadcolrs(scanline, len, fp) /* read in an encoded colr scanline */
142	greg	1.1	register COLR *scanline;
143			int len;
144			register FILE *fp;
145			{
146	greg	1.14	register int i, j;
147	greg	2.6	int code, val;
148	greg	1.14	/* determine scanline type */
149	schorsch	2.13	if ((len < MINELEN) \| (len > MAXELEN))
150	greg	1.14	return(oldreadcolrs(scanline, len, fp));
151			if ((i = getc(fp)) == EOF)
152			return(-1);
153			if (i != 2) {
154			ungetc(i, fp);
155			return(oldreadcolrs(scanline, len, fp));
156			}
157			scanline[0][GRN] = getc(fp);
158			scanline[0][BLU] = getc(fp);
159			if ((i = getc(fp)) == EOF)
160			return(-1);
161			if (scanline[0][GRN] != 2 \|\| scanline[0][BLU] & 128) {
162			scanline[0][RED] = 2;
163			scanline[0][EXP] = i;
164			return(oldreadcolrs(scanline+1, len-1, fp));
165			}
166			if ((scanline[0][BLU]<<8 \| i) != len)
167			return(-1); /* length mismatch! */
168			/* read each component */
169			for (i = 0; i < 4; i++)
170			for (j = 0; j < len; ) {
171			if ((code = getc(fp)) == EOF)
172			return(-1);
173			if (code > 128) { /* run */
174	greg	2.6	code &= 127;
175	greg	2.9	if ((val = getc(fp)) == EOF)
176			return -1;
177	greg	2.16	if (j + code > len)
178			return -1; /* overrun */
179	greg	2.6	while (code--)
180			scanline[j++][i] = val;
181	greg	2.16	} else { /* non-run */
182			if (j + code > len)
183			return -1; /* overrun */
184	greg	2.9	while (code--) {
185			if ((val = getc(fp)) == EOF)
186			return -1;
187			scanline[j++][i] = val;
188			}
189	greg	2.16	}
190	greg	1.14	}
191	greg	1.1	return(0);
192			}
193
194
195	greg	2.9	int
196	greg	1.1	fwritescan(scanline, len, fp) /* write out a scanline */
197			register COLOR *scanline;
198			int len;
199	greg	1.14	FILE *fp;
200	greg	1.1	{
201	greg	1.14	COLR *clrscan;
202			int n;
203			register COLR *sp;
204			/* get scanline buffer */
205			if ((sp = (COLR )tempbuffer(lensizeof(COLR))) == NULL)
206			return(-1);
207			clrscan = sp;
208			/* convert scanline */
209			n = len;
210			while (n-- > 0) {
211			setcolr(sp[0], scanline[0][RED],
212	greg	1.1	scanline[0][GRN],
213			scanline[0][BLU]);
214			scanline++;
215	greg	1.14	sp++;
216	greg	1.1	}
217	greg	1.14	return(fwritecolrs(clrscan, len, fp));
218	greg	1.1	}
219
220
221	greg	2.9	int
222	greg	1.1	freadscan(scanline, len, fp) /* read in a scanline */
223			register COLOR *scanline;
224			int len;
225	greg	1.14	FILE *fp;
226	greg	1.1	{
227	greg	1.14	register COLR *clrscan;
228
229			if ((clrscan = (COLR )tempbuffer(lensizeof(COLR))) == NULL)
230			return(-1);
231			if (freadcolrs(clrscan, len, fp) < 0)
232			return(-1);
233			/* convert scanline */
234			colr_color(scanline[0], clrscan[0]);
235			while (--len > 0) {
236			scanline++; clrscan++;
237			if (clrscan[0][RED] == clrscan[-1][RED] &&
238			clrscan[0][GRN] == clrscan[-1][GRN] &&
239			clrscan[0][BLU] == clrscan[-1][BLU] &&
240			clrscan[0][EXP] == clrscan[-1][EXP])
241			copycolor(scanline[0], scanline[-1]);
242			else
243			colr_color(scanline[0], clrscan[0]);
244	greg	1.1	}
245			return(0);
246			}
247
248
249	greg	2.9	void
250	greg	1.1	setcolr(clr, r, g, b) /* assign a short color value */
251			register COLR clr;
252			double r, g, b;
253			{
254			double d;
255			int e;
256
257			d = r > g ? r : g;
258			if (b > d) d = b;
259
260	greg	1.4	if (d <= 1e-32) {
261	greg	1.1	clr[RED] = clr[GRN] = clr[BLU] = 0;
262			clr[EXP] = 0;
263			return;
264			}
265
266	greg	2.8	d = frexp(d, &e) * 255.9999 / d;
267	greg	1.1
268	greg	2.11	if (r > 0.0)
269			clr[RED] = r * d;
270			else
271			clr[RED] = 0;
272			if (g > 0.0)
273			clr[GRN] = g * d;
274			else
275			clr[GRN] = 0;
276			if (b > 0.0)
277			clr[BLU] = b * d;
278			else
279			clr[BLU] = 0;
280
281	greg	1.1	clr[EXP] = e + COLXS;
282			}
283
284
285	greg	2.9	void
286	greg	1.1	colr_color(col, clr) /* convert short to float color */
287			register COLOR col;
288			register COLR clr;
289			{
290	greg	1.6	double f;
291	greg	1.1
292			if (clr[EXP] == 0)
293			col[RED] = col[GRN] = col[BLU] = 0.0;
294			else {
295	greg	1.2	f = ldexp(1.0, (int)clr[EXP]-(COLXS+8));
296	greg	1.1	col[RED] = (clr[RED] + 0.5)*f;
297			col[GRN] = (clr[GRN] + 0.5)*f;
298			col[BLU] = (clr[BLU] + 0.5)*f;
299			}
300	greg	1.6	}
301
302
303	greg	2.9	int
304	greg	1.7	bigdiff(c1, c2, md) /* c1 delta c2 > md? */
305			register COLOR c1, c2;
306			double md;
307			{
308			register int i;
309
310			for (i = 0; i < 3; i++)
311			if (colval(c1,i)-colval(c2,i) > md*colval(c2,i) \|\|
312			colval(c2,i)-colval(c1,i) > md*colval(c1,i))
313			return(1);
314			return(0);
315			}