src/common/color.c

#ifndef lint
static const char       RCSid[] = "$Id: color.c,v 2.30 2023/11/21 01:30:20 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
#undef ferror
#define getc    getc_unlocked
#define putc    putc_unlocked
#define ferror  ferror_unlocked
#endif

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


int  CNDX[4] = {0,1,2,3};       /* RGBE indices for SCOLOR, SCOLR */
float  WLPART[4] = {780,588,480,380};   /* RGB wavelength limits+partitions (nm) */


int
setspectrsamp(                  /* assign spectral sampling, 1 if good, -1 if bad */
        int cn[4],              /* input cn[3]=nsamps */
        float wlpt[4]           /* input wlpt[0],wlpt[3]=extrema */
)
{
        static const float      PKWL[3] = {607, 553, 469};
        int                     i, j;

        if (cn[3] < 3)
                return(-1);             /* reject this */

        if (wlpt[0] < wlpt[3]) {
                float   tf = wlpt[0];
                wlpt[0] = wlpt[3]; wlpt[3] = tf;
        }
        if (wlpt[0] - wlpt[3] < 50.f)
                return(-1);             /* also reject */

        if (cn[3] > MAXCSAMP)
                cn[3] = MAXCSAMP;

        if ((wlpt[3] >= PKWL[2]) | (wlpt[0] <= PKWL[0])) {
                wlpt[1] = wlpt[0] + 0.333333f*(wlpt[3]-wlpt[0]);
                wlpt[2] = wlpt[0] + 0.666667f*(wlpt[3]-wlpt[0]);
                cn[0] = 0; cn[1] = cn[3]/3; cn[2] = cn[3]*2/3;
                return(0);              /* unhappy but non-fatal return value */
        }
        wlpt[1] = 588.f;                /* tuned for standard green channel */
        wlpt[2] = 480.f;
        if (cn[3] == 3) {               /* nothing to tune? */
                cn[0] = 0; cn[1] = 1; cn[2] = 2;
        } else {                        /* else find nearest color indices */
                double  curwl[3];
                memset(curwl, 0, sizeof(curwl));
                for (i = cn[3]; i--; ) {
                        const float     cwl = (i+.5f)/cn[3]*(wlpt[3]-wlpt[0]) + wlpt[0];
                        for (j = 3; j--; )
                                if (fabs(cwl - PKWL[j]) < fabs(curwl[j] - PKWL[j])) {
                                        curwl[j] = cwl;
                                        cn[j] = i;
                                }
                }
        }
        return(1);                      /* happy return value */
}


void
setscolor(                      /* assign spectral color from RGB */
        SCOLOR scol,
        double r,
        double g,
        double b
)
{
        const double    step = (WLPART[3] - WLPART[0])/(double)NCSAMP;
        double          cwl = WLPART[0] + .5*step;
        int             i;

        for (i = 0; i < NCSAMP; i++) {
                if (cwl >= WLPART[1])
                        scol[i] = r;
                else if (cwl >= WLPART[2])
                        scol[i] = g;
                else
                        scol[i] = b;
                cwl += step;
        }
}


void
scolor2color(                   /* assign RGB color from spectrum */
        COLOR col,
        SCOLOR scol,            /* uses average over bands */
        int ncs,
        const float wlpt[4]
)
{
        const double    step = (wlpt[3] - wlpt[0])/(double)ncs;
        double          cwl = wlpt[0] + .5*step;
        int             i, j=0, n=0;

        setcolor(col, 0, 0, 0);
        for (i = 0; i < ncs; i++) {
                if (cwl < wlpt[j+1]) {
                        if (n > 1) col[j] /= (COLORV)n;
                        j++;
                        n = 0;
                }
                col[j] += scol[i];
                n++;
                cwl += step;
        }
        if (n > 1) col[j] /= (COLORV)n;
}


void
scolor2colr(                    /* assign RGBE from spectral color */
        COLR clr,
        SCOLOR scol,            /* uses average over bands */
        int ncs,
        const float wlpt[4]
)
{
        COLOR   col;

        scolor2color(col, scol, ncs, wlpt);
        setcolr(clr, col[RED], col[GRN], col[BLU]);
}


void
scolor2scolr(                   /* float spectrum to common exponent */
        SCOLR sclr,
        SCOLOR scol,
        int ncs
)
{
        int     i = ncs;
        COLORV  p = scol[--i];

        while (i)
                if (scol[--i] > p)
                        p = scol[i];
        if (p <= 1e-32) {
                memset(sclr, 0, ncs+1);
                return;
        }
        p = frexp(p, &i) * 256.0 / p;
        sclr[ncs] = i + COLXS;
        for (i = ncs; i--; )
                sclr[i] = (scol[i] > 0) * (int)(scol[i]*p);
}


void
scolr2scolor(                   /* common exponent to float spectrum */
        SCOLOR scol,
        SCOLR sclr,
        int ncs
)
{
        double  f;
        int     i;

        if (sclr[ncs] == 0) {
                memset(scol, 0, sizeof(COLORV)*ncs);
                return;
        }
        f = ldexp(1.0, (int)sclr[ncs]-(COLXS+8));

        for (i = ncs; i--; )
                scol[i] = (sclr[i] + 0.5)*f;
}


double
scolor_mean(                    /* compute average for spectral color */
        SCOLOR  scol
)
{
        int     i = NCSAMP;
        double  sum = 0;

        while (i--)
                sum += scol[i];

        return sum/(double)NCSAMP;
}


double
sintens(                        /* find maximum value from spectrum */
        SCOLOR  scol
)
{
        int     i = NCSAMP;
        COLORV  peak = scol[--i];

        while (i)
                if (scol[--i] > peak)
                        peak = scol[i];

        return peak;
}


void
convertscolor(                  /* spectrum conversion, zero-fill ends */
        SCOLOR dst,             /* destination spectrum */
        int dnc,                /* destination # of spectral samples/intervals */
        double dwl0,            /* starting destination wavelength (longer) */
        double dwl1,            /* ending destination wavelength (shorter) */
        const COLORV src[],     /* source spectrum array */
        int snc,
        double swl0,            /* long/short wavelengths may be reversed */
        double swl1
)
{
        const int       sdir = 1 - 2*(swl0 < swl1);
        const double    sstp = (swl1 - swl0)/(double)snc;
        const double    dstp = (dwl1 - dwl0)/(double)dnc;
        const double    rdstp = 1./dstp;
        int             si, ssi, di;
        double          wl;

        if ((dnc < 3) | (dwl0 <= dwl1) | (dst == src))
                return;         /* invalid destination */

        if (dnc == snc && (dwl0-swl0)*(dwl0-swl0) + (dwl1-swl1)*(dwl1-swl1) <= .5) {
                memcpy(dst, src, sizeof(COLORV)*dnc);
                return;         /* same spectral sampling */
        }
        memset(dst, 0, sizeof(COLORV)*dnc);
                                /* set starting positions */
        if ((sdir>0 ? swl0 : swl1) <= dwl0) {
                if (sdir > 0) {
                        wl = swl0;
                        ssi = 0;
                } else {
                        wl = swl1;
                        ssi = snc-1;
                }
                si = 0;
                di = (wl - dwl0)*rdstp;
        } else {
                wl = dwl0;
                if (sdir > 0) {
                        ssi = si = (wl - swl0)/sstp;
                } else {
                        si = (wl - swl1)/sstp;
                        ssi = snc-1 - si;
                }
                di = 0;
        }
        swl0 += (sdir < 0)*sstp;
                                /* step through intervals */
        while ((si < snc) & (di < dnc)) {
                double  intvl;
                if (swl0 + (ssi+sdir)*sstp < dwl0 + (di+1)*dstp) {
                        intvl = dwl0 + (di+1)*dstp - wl;
                        dst[di++] += src[ssi]*intvl*rdstp;
                } else {
                        intvl = swl0 + (ssi+sdir)*sstp - wl;
                        dst[di] += src[ssi]*intvl*rdstp;
                        ssi += sdir;
                        si++;
                }
                wl += intvl;
        }
}


void *
tempbuffer(                     /* get a temporary buffer */
        size_t  len
)
{
        static void     *tempbuf = NULL;
        static size_t   tempbuflen = 0;

        if (!len) {             /* call to free */
                if (tempbuflen) {
                        free(tempbuf);
                        tempbuf = NULL;
                        tempbuflen = 0;
                }
                return(NULL);
        }
        if (len <= tempbuflen)  /* big enough already? */
                return(tempbuf);
                                /* else free & reallocate */
        if (tempbuflen)
                free(tempbuf);
        tempbuf = malloc(len);
        tempbuflen = len*(tempbuf != NULL);
        return(tempbuf);
}


int
fwritecolrs(                    /* write out a colr scanline */
        COLR  *scanline,
        int  len,
        FILE  *fp
)
{
        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&0xff, 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);
}

/*
 * An old-format scanline is either a stream of valid RGBE or XYZE real
 * pixels or at least one real pixel followed by some number of
 * invalid real pixels of the form (1,1,1,n), where n is a count.
 * These can themselves be repeated to create a multibyte repeat
 * count, with the least significant byte first (little-endian order.)
 * Repeat counts are limited by the size of an int; if a repetition
 * leads to an overrun, the rest of the the repetition will be
 * silently ignored.
 */
static int
oldreadcolrs(                   /* read in an old-style colr scanline */
        COLR  *scanline,
        int  len,
        FILE  *fp
)
{
        int  rshift = 0;
        int  i;
        
        while (len > 0) {
                scanline[0][RED] = getc(fp);
                scanline[0][GRN] = getc(fp);
                scanline[0][BLU] = getc(fp);
                scanline[0][EXP] = i = getc(fp);
                if (i == EOF)
                        return(-1);
                if (scanline[0][GRN] == 1 &&
                                (scanline[0][RED] == 1) &
                                (scanline[0][BLU] == 1)) {
                        i = scanline[0][EXP] << rshift;
                        while (i--) {
                                copycolr(scanline[0], scanline[-1]);
                                if (--len <= 0)
                                        return(0);
                                scanline++;
                        }
                        rshift += 8;
                } else {
                        scanline++;
                        len--;
                        rshift = 0;
                }
        }
        return(0);
}

/* 
 * There are two scanline formats: old and new.  The old format
 * compresses runs of RGBE or XYZE four-byte real pixels; the new
 * format breaks the pixels into R, G, B, and E lines (or XYZE lines)
 * which are individually run-length encoded.
 *
 * An old-format scanline always begins with a valid real pixel; at
 * least one of the RGB (or XYZ) values will have its high-order bit
 * set.  A new-format scanline begins with four bytes which are not a
 * valid real pixel: (2, 2, lenhigh, lenlow) where lenhigh is always
 * less than 128 and hence never has a high-order bit set.
 *
 * A new-format scanline is broken into its RGBE or XYZE components.
 * Each is output and run-length encoded separately so that a scanline
 * is broken into four records.  In turn, each record is organized
 * into chunks of up to 128 characters, which begin with a count byte.
 * If the count byte is greater than 128, the following data byte is
 * repeated (count-128) times.  If not, the count byte is followed by
 * that many data bytes.
 */
int
freadcolrs(                     /* read in an encoded colr scanline */
        COLR  *scanline,
        int  len,
        FILE  *fp
)
{
        int  i, j;
        int  code, val;
                                        /* determine scanline type */
        if (len <= 0)
                return(0);
        if ((i = getc(fp)) == EOF)
                return(-1);
        scanline[0][RED] = i;
        scanline[0][GRN] = getc(fp);
        scanline[0][BLU] = getc(fp);
        if ((i = getc(fp)) == EOF)
                return(-1);
        if ((scanline[0][RED] != 2) | (scanline[0][GRN] != 2) |
                        (scanline[0][BLU] & 0x80)) {
                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);
}


/* read an nc-component common-exponent color scanline */
int
freadscolrs(uby8 *scanline, int nc, int len, FILE *fp)
{
        if (fread(scanline, nc+1, len, fp) != len)
                return(-1);
        return(0);
}


/* write an common-exponent spectral color scanline */
int
fwritescolrs(uby8 *sscanline, int nc, int len, FILE *fp)
{
        if (fwrite(sscanline, nc+1, len, fp) != len)
                return(-1);
        return(0);
}


int
fwritescan(                     /* write out a scanline */
        COLOR  *scanline,
        int  len,
        FILE  *fp
)
{
        COLR  *clrscan;
        int  n;
        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(                      /* read in a scanline */
        COLOR  *scanline,
        int  len,
        FILE  *fp
)
{
        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][GRN] == clrscan[-1][GRN] &&
                            (clrscan[0][RED] == clrscan[-1][RED]) &
                            (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);
}


/* read an nc-component color scanline */
int
freadsscan(COLORV *sscanline, int nc, int len, FILE *fp)
{
        uby8    *tscn = (uby8 *)tempbuffer((nc+1)*len);
        int     i;

        if (tscn == NULL || freadscolrs(tscn, nc, len, fp) < 0)
                return(-1);
        for (i = len; i-- > 0; ) {
                scolr2scolor(sscanline, tscn, nc);
                sscanline += nc;
                tscn += nc+1;
        }
        return(0);
}


/* write an spectral color scanline (NCSAMP) */
int
fwritesscan(COLORV *sscanline, int nc, int len, FILE *fp)
{
        uby8    *tscn = (uby8 *)tempbuffer((nc+1)*len);
        int     i;

        if (tscn == NULL)
                return(-1);
        for (i = 0; i < len; i++) {
                scolor2scolr(tscn+i*(nc+1), sscanline, nc);
                sscanline += nc;
        }
        return(fwritescolrs(tscn, nc, len, fp));
}


void
setcolr(                        /* assign a short color value */
        COLR  clr,
        double  r,
        double  g,
        double  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) * 256.0 / d;

        clr[RED] = (r > 0) * (int)(r*d);
        clr[GRN] = (g > 0) * (int)(g*d);
        clr[BLU] = (b > 0) * (int)(b*d);
        clr[EXP] = e + COLXS;
}


void
colr_color(                     /* convert short to float color */
        COLOR  col,
        COLR  clr
)
{
        double  f;
        
        if (clr[EXP] == 0) {
                col[RED] = col[GRN] = col[BLU] = 0.0;
                return;
        }
        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 delta c2 > md? */
        COLOR  c1,
        COLOR  c2,
        double  md
)
{
        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);
}


int
sbigsdiff(                              /* sc1 delta sc2 > md? */
        SCOLOR  c1,
        SCOLOR  c2,
        double  md
)
{
        int  i = NCSAMP;

        while (i--)
                if ((c1[i]-c2[i] > md*c2[i]) | (c2[i]-c1[i] > md*c1[i]))
                        return(1);
        return(0);
}
Revision:	2.31
Committed:	Mon Nov 27 21:00:14 2023 UTC (5 months, 2 weeks ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.30:	+9 -9 lines
Log Message:	refactor: Added number of spectral samples and wavelength splits to arguments
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.31	static const char RCSid[] = "$Id: color.c,v 2.30 2023/11/21 01:30:20 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	greg	2.9	#include <stdlib.h>
14	greg	2.7	#include <math.h>
15	greg	1.1	#include "color.h"
16	greg	2.14
17			#ifdef getc_unlocked /* avoid horrendous overhead of flockfile */
18	greg	2.15	#undef getc
19			#undef putc
20	greg	2.19	#undef ferror
21	greg	2.14	#define getc getc_unlocked
22			#define putc putc_unlocked
23	greg	2.18	#define ferror ferror_unlocked
24	greg	2.14	#endif
25	greg	1.1
26	greg	1.14	#define MINELEN 8 /* minimum scanline length for encoding */
27	greg	2.6	#define MAXELEN 0x7fff /* maximum scanline length for encoding */
28	greg	1.14	#define MINRUN 4 /* minimum run length */
29
30	greg	2.4
31	greg	2.27	int CNDX[4] = {0,1,2,3}; /* RGBE indices for SCOLOR, SCOLR */
32			float WLPART[4] = {780,588,480,380}; /* RGB wavelength limits+partitions (nm) */
33
34
35			int
36			setspectrsamp( /* assign spectral sampling, 1 if good, -1 if bad */
37			int cn[4], /* input cn[3]=nsamps */
38			float wlpt[4] /* input wlpt[0],wlpt[3]=extrema */
39			)
40			{
41			static const float PKWL[3] = {607, 553, 469};
42			int i, j;
43
44			if (cn[3] < 3)
45			return(-1); /* reject this */
46
47			if (wlpt[0] < wlpt[3]) {
48			float tf = wlpt[0];
49			wlpt[0] = wlpt[3]; wlpt[3] = tf;
50			}
51			if (wlpt[0] - wlpt[3] < 50.f)
52			return(-1); /* also reject */
53
54			if (cn[3] > MAXCSAMP)
55			cn[3] = MAXCSAMP;
56
57			if ((wlpt[3] >= PKWL[2]) \| (wlpt[0] <= PKWL[0])) {
58			wlpt[1] = wlpt[0] + 0.333333f*(wlpt[3]-wlpt[0]);
59			wlpt[2] = wlpt[0] + 0.666667f*(wlpt[3]-wlpt[0]);
60			cn[0] = 0; cn[1] = cn[3]/3; cn[2] = cn[3]*2/3;
61			return(0); /* unhappy but non-fatal return value */
62			}
63			wlpt[1] = 588.f; /* tuned for standard green channel */
64			wlpt[2] = 480.f;
65			if (cn[3] == 3) { /* nothing to tune? */
66			cn[0] = 0; cn[1] = 1; cn[2] = 2;
67			} else { /* else find nearest color indices */
68			double curwl[3];
69			memset(curwl, 0, sizeof(curwl));
70			for (i = cn[3]; i--; ) {
71			const float cwl = (i+.5f)/cn[3]*(wlpt[3]-wlpt[0]) + wlpt[0];
72			for (j = 3; j--; )
73			if (fabs(cwl - PKWL[j]) < fabs(curwl[j] - PKWL[j])) {
74			curwl[j] = cwl;
75			cn[j] = i;
76			}
77			}
78			}
79			return(1); /* happy return value */
80			}
81
82
83			void
84			setscolor( /* assign spectral color from RGB */
85			SCOLOR scol,
86			double r,
87			double g,
88			double b
89			)
90			{
91			const double step = (WLPART[3] - WLPART[0])/(double)NCSAMP;
92			double cwl = WLPART[0] + .5*step;
93			int i;
94
95			for (i = 0; i < NCSAMP; i++) {
96			if (cwl >= WLPART[1])
97			scol[i] = r;
98			else if (cwl >= WLPART[2])
99			scol[i] = g;
100			else
101			scol[i] = b;
102			cwl += step;
103			}
104			}
105
106
107			void
108			scolor2color( /* assign RGB color from spectrum */
109			COLOR col,
110			SCOLOR scol, /* uses average over bands */
111			int ncs,
112	greg	2.30	const float wlpt[4]
113	greg	2.27	)
114			{
115			const double step = (wlpt[3] - wlpt[0])/(double)ncs;
116			double cwl = wlpt[0] + .5*step;
117			int i, j=0, n=0;
118
119			setcolor(col, 0, 0, 0);
120			for (i = 0; i < ncs; i++) {
121			if (cwl < wlpt[j+1]) {
122			if (n > 1) col[j] /= (COLORV)n;
123			j++;
124			n = 0;
125			}
126			col[j] += scol[i];
127			n++;
128			cwl += step;
129			}
130			if (n > 1) col[j] /= (COLORV)n;
131			}
132
133
134			void
135			scolor2colr( /* assign RGBE from spectral color */
136			COLR clr,
137			SCOLOR scol, /* uses average over bands */
138			int ncs,
139	greg	2.30	const float wlpt[4]
140	greg	2.27	)
141			{
142			COLOR col;
143
144			scolor2color(col, scol, ncs, wlpt);
145			setcolr(clr, col[RED], col[GRN], col[BLU]);
146			}
147
148
149			void
150			scolor2scolr( /* float spectrum to common exponent */
151			SCOLR sclr,
152			SCOLOR scol,
153			int ncs
154			)
155			{
156			int i = ncs;
157			COLORV p = scol[--i];
158
159			while (i)
160			if (scol[--i] > p)
161			p = scol[i];
162			if (p <= 1e-32) {
163			memset(sclr, 0, ncs+1);
164			return;
165			}
166			p = frexp(p, &i) * 256.0 / p;
167			sclr[ncs] = i + COLXS;
168			for (i = ncs; i--; )
169			sclr[i] = (scol[i] > 0) * (int)(scol[i]*p);
170			}
171
172
173			void
174			scolr2scolor( /* common exponent to float spectrum */
175			SCOLOR scol,
176			SCOLR sclr,
177			int ncs
178			)
179			{
180			double f;
181			int i;
182
183			if (sclr[ncs] == 0) {
184			memset(scol, 0, sizeof(COLORV)*ncs);
185			return;
186			}
187			f = ldexp(1.0, (int)sclr[ncs]-(COLXS+8));
188
189			for (i = ncs; i--; )
190			scol[i] = (sclr[i] + 0.5)*f;
191			}
192
193
194			double
195			scolor_mean( /* compute average for spectral color */
196			SCOLOR scol
197			)
198			{
199			int i = NCSAMP;
200			double sum = 0;
201
202			while (i--)
203			sum += scol[i];
204
205			return sum/(double)NCSAMP;
206			}
207
208
209			double
210			sintens( /* find maximum value from spectrum */
211			SCOLOR scol
212			)
213			{
214			int i = NCSAMP;
215			COLORV peak = scol[--i];
216
217			while (i)
218			if (scol[--i] > peak)
219			peak = scol[i];
220
221			return peak;
222			}
223
224
225			void
226			convertscolor( /* spectrum conversion, zero-fill ends */
227			SCOLOR dst, /* destination spectrum */
228			int dnc, /* destination # of spectral samples/intervals */
229			double dwl0, /* starting destination wavelength (longer) */
230			double dwl1, /* ending destination wavelength (shorter) */
231			const COLORV src[], /* source spectrum array */
232			int snc,
233			double swl0, /* long/short wavelengths may be reversed */
234			double swl1
235			)
236			{
237			const int sdir = 1 - 2*(swl0 < swl1);
238			const double sstp = (swl1 - swl0)/(double)snc;
239			const double dstp = (dwl1 - dwl0)/(double)dnc;
240			const double rdstp = 1./dstp;
241			int si, ssi, di;
242			double wl;
243
244			if ((dnc < 3) \| (dwl0 <= dwl1) \| (dst == src))
245			return; /* invalid destination */
246
247			if (dnc == snc && (dwl0-swl0)(dwl0-swl0) + (dwl1-swl1)(dwl1-swl1) <= .5) {
248			memcpy(dst, src, sizeof(COLORV)*dnc);
249			return; /* same spectral sampling */
250			}
251			memset(dst, 0, sizeof(COLORV)*dnc);
252			/* set starting positions */
253			if ((sdir>0 ? swl0 : swl1) <= dwl0) {
254			if (sdir > 0) {
255			wl = swl0;
256			ssi = 0;
257			} else {
258			wl = swl1;
259			ssi = snc-1;
260			}
261			si = 0;
262			di = (wl - dwl0)*rdstp;
263			} else {
264			wl = dwl0;
265	greg	2.29	if (sdir > 0) {
266			ssi = si = (wl - swl0)/sstp;
267			} else {
268			si = (wl - swl1)/sstp;
269			ssi = snc-1 - si;
270			}
271	greg	2.27	di = 0;
272			}
273			swl0 += (sdir < 0)*sstp;
274			/* step through intervals */
275			while ((si < snc) & (di < dnc)) {
276			double intvl;
277			if (swl0 + (ssi+sdir)sstp < dwl0 + (di+1)dstp) {
278			intvl = dwl0 + (di+1)*dstp - wl;
279			dst[di++] += src[ssi]intvlrdstp;
280			} else {
281			intvl = swl0 + (ssi+sdir)*sstp - wl;
282			dst[di] += src[ssi]intvlrdstp;
283			ssi += sdir;
284			si++;
285			}
286			wl += intvl;
287			}
288			}
289
290
291	greg	2.22	void *
292	greg	2.17	tempbuffer( /* get a temporary buffer */
293	greg	2.27	size_t len
294	greg	2.17	)
295	greg	1.14	{
296	greg	2.27	static void *tempbuf = NULL;
297			static size_t tempbuflen = 0;
298	greg	1.14
299	greg	2.22	if (!len) { /* call to free */
300			if (tempbuflen) {
301	greg	2.20	free(tempbuf);
302	greg	2.22	tempbuf = NULL;
303			tempbuflen = 0;
304			}
305			return(NULL);
306	greg	1.14	}
307	greg	2.22	if (len <= tempbuflen) /* big enough already? */
308			return(tempbuf);
309			/* else free & reallocate */
310			if (tempbuflen)
311			free(tempbuf);
312			tempbuf = malloc(len);
313			tempbuflen = len*(tempbuf != NULL);
314	greg	1.14	return(tempbuf);
315			}
316
317
318	greg	2.9	int
319	greg	2.17	fwritecolrs( /* write out a colr scanline */
320			COLR *scanline,
321			int len,
322			FILE *fp
323			)
324	greg	1.1	{
325	greg	2.17	int i, j, beg, cnt = 1;
326	greg	1.14	int c2;
327	greg	1.1
328	schorsch	2.13	if ((len < MINELEN) \| (len > MAXELEN)) /* OOBs, write out flat */
329	greg	1.14	return(fwrite((char *)scanline,sizeof(COLR),len,fp) - len);
330	greg	2.2	/* put magic header */
331			putc(2, fp);
332	greg	1.14	putc(2, fp);
333			putc(len>>8, fp);
334	greg	2.21	putc(len&0xff, fp);
335	greg	1.14	/* put components seperately */
336			for (i = 0; i < 4; i++) {
337			for (j = 0; j < len; j += cnt) { /* find next run */
338			for (beg = j; beg < len; beg += cnt) {
339	greg	2.20	for (cnt = 1; (cnt < 127) & (beg+cnt < len) &&
340	greg	1.14	scanline[beg+cnt][i] == scanline[beg][i]; cnt++)
341			;
342			if (cnt >= MINRUN)
343			break; /* long enough */
344	greg	1.1	}
345	greg	2.20	if ((beg-j > 1) & (beg-j < MINRUN)) {
346	greg	1.15	c2 = j+1;
347			while (scanline[c2++][i] == scanline[j][i])
348			if (c2 == beg) { /* short run */
349			putc(128+beg-j, fp);
350			putc(scanline[j][i], fp);
351			j = beg;
352			break;
353			}
354			}
355			while (j < beg) { /* write out non-run */
356	greg	1.14	if ((c2 = beg-j) > 128) c2 = 128;
357			putc(c2, fp);
358			while (c2--)
359			putc(scanline[j++][i], fp);
360			}
361			if (cnt >= MINRUN) { /* write out run */
362			putc(128+cnt, fp);
363			putc(scanline[beg][i], fp);
364			} else
365			cnt = 0;
366			}
367	greg	1.1	}
368			return(ferror(fp) ? -1 : 0);
369			}
370
371	greg	2.23	/*
372			* An old-format scanline is either a stream of valid RGBE or XYZE real
373			* pixels or at least one real pixel followed by some number of
374			* invalid real pixels of the form (1,1,1,n), where n is a count.
375			* These can themselves be repeated to create a multibyte repeat
376			* count, with the least significant byte first (little-endian order.)
377			* Repeat counts are limited by the size of an int; if a repetition
378			* leads to an overrun, the rest of the the repetition will be
379			* silently ignored.
380			*/
381	greg	2.9	static int
382	greg	2.19	oldreadcolrs( /* read in an old-style colr scanline */
383	greg	2.17	COLR *scanline,
384			int len,
385			FILE *fp
386			)
387	greg	2.9	{
388	greg	2.25	int rshift = 0;
389	greg	2.17	int i;
390	greg	2.9
391			while (len > 0) {
392			scanline[0][RED] = getc(fp);
393			scanline[0][GRN] = getc(fp);
394			scanline[0][BLU] = getc(fp);
395	greg	2.18	scanline[0][EXP] = i = getc(fp);
396			if (i == EOF)
397	greg	2.9	return(-1);
398	greg	2.20	if (scanline[0][GRN] == 1 &&
399			(scanline[0][RED] == 1) &
400	greg	2.25	(scanline[0][BLU] == 1)) {
401	greg	2.20	i = scanline[0][EXP] << rshift;
402			while (i--) {
403	greg	2.9	copycolr(scanline[0], scanline[-1]);
404	greg	2.20	if (--len <= 0)
405			return(0);
406	greg	2.9	scanline++;
407			}
408			rshift += 8;
409			} else {
410			scanline++;
411			len--;
412			rshift = 0;
413			}
414			}
415			return(0);
416			}
417
418	greg	2.23	/*
419			* There are two scanline formats: old and new. The old format
420			* compresses runs of RGBE or XYZE four-byte real pixels; the new
421			* format breaks the pixels into R, G, B, and E lines (or XYZE lines)
422			* which are individually run-length encoded.
423			*
424			* An old-format scanline always begins with a valid real pixel; at
425			* least one of the RGB (or XYZ) values will have its high-order bit
426			* set. A new-format scanline begins with four bytes which are not a
427			* valid real pixel: (2, 2, lenhigh, lenlow) where lenhigh is always
428			* less than 128 and hence never has a high-order bit set.
429			*
430			* A new-format scanline is broken into its RGBE or XYZE components.
431			* Each is output and run-length encoded separately so that a scanline
432			* is broken into four records. In turn, each record is organized
433			* into chunks of up to 128 characters, which begin with a count byte.
434			* If the count byte is greater than 128, the following data byte is
435			* repeated (count-128) times. If not, the count byte is followed by
436			* that many data bytes.
437			*/
438	greg	2.9	int
439	greg	2.17	freadcolrs( /* read in an encoded colr scanline */
440			COLR *scanline,
441			int len,
442			FILE *fp
443			)
444	greg	1.1	{
445	greg	2.17	int i, j;
446	greg	2.6	int code, val;
447	greg	1.14	/* determine scanline type */
448	greg	2.26	if (len <= 0)
449			return(0);
450	greg	1.14	if ((i = getc(fp)) == EOF)
451			return(-1);
452	greg	2.26	scanline[0][RED] = i;
453	greg	1.14	scanline[0][GRN] = getc(fp);
454			scanline[0][BLU] = getc(fp);
455			if ((i = getc(fp)) == EOF)
456			return(-1);
457	greg	2.26	if ((scanline[0][RED] != 2) \| (scanline[0][GRN] != 2) \|
458			(scanline[0][BLU] & 0x80)) {
459	greg	1.14	scanline[0][EXP] = i;
460			return(oldreadcolrs(scanline+1, len-1, fp));
461			}
462			if ((scanline[0][BLU]<<8 \| i) != len)
463			return(-1); /* length mismatch! */
464			/* read each component */
465			for (i = 0; i < 4; i++)
466			for (j = 0; j < len; ) {
467			if ((code = getc(fp)) == EOF)
468			return(-1);
469			if (code > 128) { /* run */
470	greg	2.6	code &= 127;
471	greg	2.9	if ((val = getc(fp)) == EOF)
472			return -1;
473	greg	2.16	if (j + code > len)
474			return -1; /* overrun */
475	greg	2.6	while (code--)
476			scanline[j++][i] = val;
477	greg	2.16	} else { /* non-run */
478			if (j + code > len)
479			return -1; /* overrun */
480	greg	2.9	while (code--) {
481			if ((val = getc(fp)) == EOF)
482			return -1;
483			scanline[j++][i] = val;
484			}
485	greg	2.16	}
486	greg	1.14	}
487	greg	1.1	return(0);
488			}
489
490
491	greg	2.30	/* read an nc-component common-exponent color scanline */
492			int
493			freadscolrs(uby8 scanline, int nc, int len, FILE fp)
494			{
495			if (fread(scanline, nc+1, len, fp) != len)
496			return(-1);
497			return(0);
498			}
499
500
501	greg	2.31	/* write an common-exponent spectral color scanline */
502	greg	2.30	int
503	greg	2.31	fwritescolrs(uby8 sscanline, int nc, int len, FILE fp)
504	greg	2.30	{
505	greg	2.31	if (fwrite(sscanline, nc+1, len, fp) != len)
506	greg	2.30	return(-1);
507			return(0);
508			}
509
510
511	greg	2.9	int
512	greg	2.17	fwritescan( /* write out a scanline */
513			COLOR *scanline,
514			int len,
515			FILE *fp
516			)
517	greg	1.1	{
518	greg	1.14	COLR *clrscan;
519			int n;
520	greg	2.17	COLR *sp;
521	greg	1.14	/* get scanline buffer */
522			if ((sp = (COLR )tempbuffer(lensizeof(COLR))) == NULL)
523			return(-1);
524			clrscan = sp;
525			/* convert scanline */
526			n = len;
527			while (n-- > 0) {
528			setcolr(sp[0], scanline[0][RED],
529	greg	1.1	scanline[0][GRN],
530			scanline[0][BLU]);
531			scanline++;
532	greg	1.14	sp++;
533	greg	1.1	}
534	greg	1.14	return(fwritecolrs(clrscan, len, fp));
535	greg	1.1	}
536
537
538	greg	2.9	int
539	greg	2.17	freadscan( /* read in a scanline */
540			COLOR *scanline,
541			int len,
542			FILE *fp
543			)
544	greg	1.1	{
545	greg	2.17	COLR *clrscan;
546	greg	1.14
547			if ((clrscan = (COLR )tempbuffer(lensizeof(COLR))) == NULL)
548			return(-1);
549			if (freadcolrs(clrscan, len, fp) < 0)
550			return(-1);
551			/* convert scanline */
552			colr_color(scanline[0], clrscan[0]);
553			while (--len > 0) {
554			scanline++; clrscan++;
555	greg	2.20	if (clrscan[0][GRN] == clrscan[-1][GRN] &&
556			(clrscan[0][RED] == clrscan[-1][RED]) &
557			(clrscan[0][BLU] == clrscan[-1][BLU]) &
558			(clrscan[0][EXP] == clrscan[-1][EXP]))
559	greg	1.14	copycolor(scanline[0], scanline[-1]);
560			else
561			colr_color(scanline[0], clrscan[0]);
562	greg	1.1	}
563			return(0);
564			}
565
566
567	greg	2.30	/* read an nc-component color scanline */
568			int
569			freadsscan(COLORV sscanline, int nc, int len, FILE fp)
570			{
571			uby8 tscn = (uby8 )tempbuffer((nc+1)*len);
572			int i;
573
574			if (tscn == NULL \|\| freadscolrs(tscn, nc, len, fp) < 0)
575			return(-1);
576			for (i = len; i-- > 0; ) {
577			scolr2scolor(sscanline, tscn, nc);
578			sscanline += nc;
579			tscn += nc+1;
580			}
581			return(0);
582			}
583
584
585			/* write an spectral color scanline (NCSAMP) */
586			int
587	greg	2.31	fwritesscan(COLORV sscanline, int nc, int len, FILE fp)
588	greg	2.30	{
589	greg	2.31	uby8 tscn = (uby8 )tempbuffer((nc+1)*len);
590	greg	2.30	int i;
591
592			if (tscn == NULL)
593			return(-1);
594			for (i = 0; i < len; i++) {
595	greg	2.31	scolor2scolr(tscn+i*(nc+1), sscanline, nc);
596			sscanline += nc;
597	greg	2.30	}
598	greg	2.31	return(fwritescolrs(tscn, nc, len, fp));
599	greg	2.30	}
600
601
602	greg	2.9	void
603	greg	2.17	setcolr( /* assign a short color value */
604			COLR clr,
605			double r,
606			double g,
607			double b
608			)
609	greg	1.1	{
610			double d;
611			int e;
612
613			d = r > g ? r : g;
614			if (b > d) d = b;
615
616	greg	1.4	if (d <= 1e-32) {
617	greg	1.1	clr[RED] = clr[GRN] = clr[BLU] = 0;
618			clr[EXP] = 0;
619			return;
620			}
621
622	greg	2.21	d = frexp(d, &e) * 256.0 / d;
623	greg	1.1
624	greg	2.27	clr[RED] = (r > 0) * (int)(r*d);
625			clr[GRN] = (g > 0) * (int)(g*d);
626			clr[BLU] = (b > 0) * (int)(b*d);
627	greg	1.1	clr[EXP] = e + COLXS;
628			}
629
630
631	greg	2.9	void
632	greg	2.17	colr_color( /* convert short to float color */
633			COLOR col,
634			COLR clr
635			)
636	greg	1.1	{
637	greg	1.6	double f;
638	greg	1.1
639	greg	2.27	if (clr[EXP] == 0) {
640	greg	1.1	col[RED] = col[GRN] = col[BLU] = 0.0;
641	greg	2.27	return;
642	greg	1.1	}
643	greg	2.27	f = ldexp(1.0, (int)clr[EXP]-(COLXS+8));
644			col[RED] = (clr[RED] + 0.5)*f;
645			col[GRN] = (clr[GRN] + 0.5)*f;
646			col[BLU] = (clr[BLU] + 0.5)*f;
647	greg	1.6	}
648
649
650	greg	2.9	int
651	greg	2.17	bigdiff( /* c1 delta c2 > md? */
652			COLOR c1,
653			COLOR c2,
654			double md
655			)
656	greg	1.7	{
657	greg	2.17	int i;
658	greg	1.7
659			for (i = 0; i < 3; i++)
660	greg	2.27	if ((colval(c1,i)-colval(c2,i) > md*colval(c2,i)) \|
661			(colval(c2,i)-colval(c1,i) > md*colval(c1,i)))
662			return(1);
663			return(0);
664			}
665
666
667			int
668			sbigsdiff( /* sc1 delta sc2 > md? */
669			SCOLOR c1,
670			SCOLOR c2,
671			double md
672			)
673			{
674			int i = NCSAMP;
675
676			while (i--)
677			if ((c1[i]-c2[i] > mdc2[i]) \| (c2[i]-c1[i] > mdc1[i]))
678	greg	1.7	return(1);
679			return(0);
680			}