--- ray/src/px/pcond2.c	1997/01/31 15:56:20	3.7
+++ ray/src/px/pcond2.c	2005/09/08 22:29:47	3.17
@@ -1,14 +1,13 @@
-/* Copyright (c) 1997 Regents of the University of California */
-
 #ifndef lint
-static char SCCSid[] = "$SunId$ LBL";
+static const char	RCSid[] = "$Id: pcond2.c,v 3.17 2005/09/08 22:29:47 greg Exp $";
 #endif
-
 /*
  * Input and output conditioning routines for pcond.
+ *  Added white-balance adjustment 10/01 (GW).
  */
 
 #include "pcond.h"
+#include "warp3d.h"
 
 
 RGBPRIMP	inprims = stdprims;	/* input primaries */
@@ -19,6 +18,7 @@ double	(*lumf)() = rgblum;		/* input luminance functio
 double	inpexp = 1.0;			/* input exposure value */
 
 char	*mbcalfile = NULL;		/* macbethcal mapping file */
+char	*cwarpfile = NULL;		/* color space warping file */
 
 static struct mbc {
 	COLORMAT	cmat;
@@ -26,14 +26,23 @@ static struct mbc {
 	COLOR	cmin, cmax;
 }	mbcond;				/* macbethcal conditioning struct */
 
+static WARP3D	*cwarp;			/* color warping structure */
+
 static COLOR	*scanbuf;		/* scanline processing buffer */
 static int	nread;			/* number of scanlines processed */
 
+static void sfscan(COLOR *sl, int len, double sf);
+static void matscan(COLOR *sl, int len, COLORMAT mat);
+static void mbscan(COLOR *sl, int len, struct mbc *mb);
+static void cwscan(COLOR *sl, int len, WARP3D *wp);
+static void getmbcalfile(char *fn, struct mbc *mb);
 
-double
-rgblum(clr, scotopic)		/* compute (scotopic) luminance of RGB color */
-COLOR	clr;
-int	scotopic;
+
+extern double
+rgblum(		/* compute (scotopic) luminance of RGB color */
+	COLOR	clr,
+	int	scotopic
+)
 {
 	if (scotopic)		/* approximate */
 		return( WHTSEFFICACY * (colval(clr,RED)*.062 +
@@ -44,10 +53,11 @@ int	scotopic;
 }
 
 
-double
-cielum(xyz, scotopic)		/* compute (scotopic) luminance of CIE color */
-COLOR	xyz;
-int	scotopic;
+extern double
+cielum(		/* compute (scotopic) luminance of CIE color */
+	COLOR	xyz,
+	int	scotopic
+)
 {
 	if (scotopic)		/* approximate */
 		return(colval(xyz,CIEY) *
@@ -57,11 +67,13 @@ int	scotopic;
 }
 
 
-COLOR *
-nextscan()				/* read and condition next scanline */
+extern COLOR *
+nextscan(void)				/* read and condition next scanline */
 {
 	if (nread >= numscans(&inpres)) {
-		free((char *)scanbuf);
+		if (cwarpfile != NULL)
+			free3dw(cwarp);
+		free((void *)scanbuf);
 		return(scanbuf = NULL);
 	}
 	if (what2do&DO_ACUITY)
@@ -81,23 +93,28 @@ nextscan()				/* read and condition next scanline */
 		mapscan(scanbuf, scanlen(&inpres));
 	if (mbcalfile != NULL)			/* device color correction */
 		mbscan(scanbuf, scanlen(&inpres), &mbcond);
-	else if (lumf == cielum | inprims != outprims)
+	else if (cwarpfile != NULL)		/* device color space warp */
+		cwscan(scanbuf, scanlen(&inpres), cwarp);
+	else if ((lumf == cielum) | (inprims != outprims))
 		matscan(scanbuf, scanlen(&inpres), mbcond.cmat);
 	nread++;
 	return(scanbuf);
 }
 
 
-COLOR *
-firstscan()				/* return first processed scanline */
+extern COLOR *
+firstscan(void)				/* return first processed scanline */
 {
 	if (mbcalfile != NULL)		/* load macbethcal file */
 		getmbcalfile(mbcalfile, &mbcond);
-	else
+	else if (cwarpfile != NULL) {
+		if ((cwarp = load3dw(cwarpfile, NULL)) == NULL)
+			syserror(cwarpfile);
+	} else
 		if (lumf == rgblum)
-			comprgb2rgbmat(mbcond.cmat, inprims, outprims);
+			comprgb2rgbWBmat(mbcond.cmat, inprims, outprims);
 		else
-			compxyz2rgbmat(mbcond.cmat, outprims);
+			compxyz2rgbWBmat(mbcond.cmat, outprims);
 	if (what2do&DO_ACUITY)
 		initacuity();
 	scanbuf = (COLOR *)malloc(scanlen(&inpres)*sizeof(COLOR));
@@ -108,10 +125,12 @@ firstscan()				/* return first processed scanline */
 }
 
 
-sfscan(sl, len, sf)			/* apply scalefactor to scanline */
-register COLOR	*sl;
-int	len;
-double	sf;
+static void
+sfscan(			/* apply scalefactor to scanline */
+	register COLOR	*sl,
+	int	len,
+	double	sf
+)
 {
 	while (len--) {
 		scalecolor(sl[0], sf);
@@ -120,30 +139,52 @@ double	sf;
 }
 
 
-matscan(sl, len, mat)			/* apply color matrix to scaline */
-register COLOR	*sl;
-int	len;
-COLORMAT	mat;
+static double
+greypoint(			/* compute gamut mapping grey target */
+	COLOR col
+)
 {
+	COLOR	gryc;
+	int	i;
+				/* improves saturated color rendering */
+	copycolor(gryc, col);
+	for (i = 3; i--; )
+		if (gryc[i] > 1)
+			gryc[i] = 1;
+		else if (gryc[i] < 0)
+			gryc[i] = 0;
+	return(bright(gryc));
+}
+
+
+static void
+matscan(			/* apply color matrix to scaline */
+	register COLOR	*sl,
+	int	len,
+	COLORMAT	mat
+)
+{
 	while (len--) {
 		colortrans(sl[0], mat, sl[0]);
-		clipgamut(sl[0], bright(sl[0]), CGAMUT, cblack, cwhite);
+		clipgamut(sl[0], greypoint(sl[0]), CGAMUT, cblack, cwhite);
 		sl++;
 	}
 }
 
 
-mbscan(sl, len, mb)			/* apply macbethcal adj. to scaline */
-COLOR	*sl;
-int	len;
-register struct mbc	*mb;
+static void
+mbscan(			/* apply macbethcal adj. to scaline */
+	COLOR	*sl,
+	int	len,
+	register struct mbc	*mb
+)
 {
 	double	d;
 	register int	i, j;
 
 	while (len--) {
 		colortrans(sl[0], mb->cmat, sl[0]);
-		clipgamut(sl[0], bright(sl[0]), CGAMUT, mb->cmin, mb->cmax);
+		clipgamut(sl[0], greypoint(sl[0]), CGAMUT, mb->cmin, mb->cmax);
 		for (i = 0; i < 3; i++) {
 			d = colval(sl[0],i);
 			for (j = 0; j < 4 && mb->xa[i][j+1] <= d; j++)
@@ -157,11 +198,36 @@ register struct mbc	*mb;
 }
 
 
-getmbcalfile(fn, mb)			/* load macbethcal file */
-char	*fn;
-register struct mbc	*mb;
+static void
+cwscan(			/* apply color space warp to scaline */
+	COLOR	*sl,
+	int	len,
+	WARP3D	*wp
+)
 {
-	extern char	*fgets();
+	int	rval;
+
+	while (len--) {
+		rval = warp3d(sl[0], sl[0], wp);
+		if (rval & W3ERROR)
+			syserror("warp3d");
+		if (rval & W3BADMAP) {
+			fprintf(stderr, "%s: %s: bad color space map\n",
+					progname, cwarpfile);
+			exit(1);
+		}
+		clipgamut(sl[0], greypoint(sl[0]), CGAMUT, cblack, cwhite);
+		sl++;
+	}
+}
+
+
+static void
+getmbcalfile(			/* load macbethcal file */
+	char	*fn,
+	register struct mbc	*mb
+)
+{
 	char	buf[128];
 	FILE	*fp;
 	int	inpflags = 0;
@@ -220,19 +286,19 @@ register struct mbc	*mb;
 			inpflags |= 040;
 		else if (!(inpflags & 0100) &&
 				sscanf(buf,
-				"r = %f*r1 + %f*g1 + %f*b1",
+				"ro = %f*rn + %f*gn + %f*bn",
 				&mb->cmat[0][0], &mb->cmat[0][1],
 				&mb->cmat[0][2]) == 3)
 			inpflags |= 0100;
 		else if (!(inpflags & 0200) &&
 				sscanf(buf,
-				"g = %f*r1 + %f*g1 + %f*b1",
+				"go = %f*rn + %f*gn + %f*bn",
 				&mb->cmat[1][0], &mb->cmat[1][1],
 				&mb->cmat[1][2]) == 3)
 			inpflags |= 0200;
 		else if (!(inpflags & 0400) &&
 				sscanf(buf,
-				"b = %f*r1 + %f*g1 + %f*b1",
+				"bo = %f*rn + %f*gn + %f*bn",
 				&mb->cmat[2][0], &mb->cmat[2][1],
 				&mb->cmat[2][2]) == 3)
 			inpflags |= 0400;