--- ray/src/px/macbethcal.c	1997/01/30 19:14:11	2.12
+++ ray/src/px/macbethcal.c	2003/06/05 19:29:34	2.18
@@ -1,22 +1,24 @@
-/* Copyright (c) 1997 Regents of the University of California */
-
 #ifndef lint
-static char SCCSid[] = "$SunId$ LBL";
+static const char	RCSid[] = "$Id: macbethcal.c,v 2.18 2003/06/05 19:29:34 schorsch Exp $";
 #endif
-
 /*
  * Calibrate a scanned MacBeth Color Checker Chart
  *
- * Produce a .cal file suitable for use with pcomb.
+ * Produce a .cal file suitable for use with pcomb,
+ * or .cwp file suitable for use with pcwarp.
+ *
+ * Warping code depends on conformance of COLOR and W3VEC types.
  */
 
 #include <stdio.h>
-#ifdef MSDOS
-#include <fcntl.h>
-#endif
+#include <math.h>
+#include <time.h>
+
+#include "platform.h"
 #include "color.h"
 #include "resolu.h"
 #include "pmap.h"
+#include "warp3d.h"
 
 				/* MacBeth colors */
 #define	DarkSkin	0
@@ -95,7 +97,13 @@ short	mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutra
 #define  RG_ORIG	02	/* original color region */
 #define  RG_CORR	04	/* corrected color region */
 
+#ifndef  DISPCOM
+#define  DISPCOM	"ximage -op %s"
+#endif
+
 int	scanning = 1;		/* scanned input (or recorded output)? */
+double	irrad = 1.0;		/* irradiance multiplication factor */
+int	rawmap = 0;		/* put out raw color mapping? */
 
 int	xmax, ymax;		/* input image dimensions */
 int	bounds[4][2];		/* image coordinates of chart corners */
@@ -106,14 +114,15 @@ long	inpflags = 0;		/* flags of which colors were inpu
 long	gmtflags = 0;		/* flags of out-of-gamut colors */
 
 COLOR	bramp[NMBNEU][2];	/* brightness ramp (per primary) */
-double	solmat[3][3];		/* color mapping matrix */
+COLORMAT	solmat;		/* color mapping matrix */
+COLOR	colmin, colmax;		/* gamut limits */
 
+WARP3D	*wcor = NULL;		/* color space warp */
+
 FILE	*debugfp = NULL;	/* debug output picture */
 char	*progname;
 
-extern char	*malloc();
 
-
 main(argc, argv)
 int	argc;
 char	**argv;
@@ -131,9 +140,7 @@ char	**argv;
 				perror(argv[i]);
 				exit(1);
 			}
-#ifdef MSDOS
-			setmode(fileno(debugfp), O_BINARY);
-#endif
+			SET_FILE_BINARY(debugfp);
 			newheader("RADIANCE", debugfp);		/* start */
 			printargs(argc, argv, debugfp);		/* header */
 			break;
@@ -150,6 +157,18 @@ char	**argv;
 			bounds[3][1] = atoi(argv[++i]);
 			scanning = 2;
 			break;
+		case 'P':				/* pick position */
+			scanning = 3;
+			break;
+		case 'i':				/* irradiance factor */
+			i++;
+			if (badarg(argc-i, argv+i, "f"))
+				goto userr;
+			irrad = atof(argv[i]);
+			break;
+		case 'm':				/* raw map output */
+			rawmap = 1;
+			break;
 		case 'c':				/* color input */
 			scanning = 0;
 			break;
@@ -158,22 +177,26 @@ char	**argv;
 		}
 							/* open files */
 	if (i < argc && freopen(argv[i], "r", stdin) == NULL) {
-		perror(argv[1]);
+		perror(argv[i]);
 		exit(1);
 	}
 	if (i+1 < argc && freopen(argv[i+1], "w", stdout) == NULL) {
-		perror(argv[2]);
+		perror(argv[i+1]);
 		exit(1);
 	}
 	if (scanning) {			/* load input picture header */
-#ifdef MSDOS
-		setmode(fileno(stdin), O_BINARY);
-#endif
+		SET_FILE_BINARY(stdin);
 		if (checkheader(stdin, COLRFMT, NULL) < 0 ||
 				fgetresolu(&xmax, &ymax, stdin) < 0) {
 			fprintf(stderr, "%s: bad input picture\n", progname);
 			exit(1);
 		}
+		if (scanning == 3) {
+			if (i >= argc)
+				goto userr;
+			pickchartpos(argv[i]);
+			scanning = 2;
+		}
 	} else {			/* else set default xmax and ymax */
 		xmax = 512;
 		ymax = 2*512/3;
@@ -190,16 +213,34 @@ char	**argv;
 	else
 		getcolors();
 	compute();			/* compute color mapping */
-					/* print comment */
-	printf("{\n\tColor correction file computed by:\n\t\t");
-	printargs(argc, argv, stdout);
-	printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
-			i+1 < argc ? argv[i+1] : "{this_file}");
-	if (!scanning)
-		printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
+	if (rawmap) {			/* print out raw correspondence */
+		register int	j;
+
+		printf("# Color correspondence produced by:\n#\t\t");
+		printargs(argc, argv, stdout);
+		printf("#\tUsage: pcwarp %s uncorrected.pic > corrected.pic\n",
 				i+1 < argc ? argv[i+1] : "{this_file}");
-	printf("}\n");
-	putmapping();			/* put out color mapping */
+		printf("#\t   Or: pcond [options] -m %s orig.pic > output.pic\n",
+				i+1 < argc ? argv[i+1] : "{this_file}");
+		for (j = 0; j < 24; j++)
+			printf("%f %f %f    %f %f %f\n",
+				colval(inpRGB[j],RED), colval(inpRGB[j],GRN),
+				colval(inpRGB[j],BLU), colval(mbRGB[j],RED),
+				colval(mbRGB[j],GRN), colval(mbRGB[j],BLU));
+		if (scanning && debugfp != NULL)
+			cwarp();		/* color warp for debugging */
+	} else {			/* print color mapping */
+						/* print header */
+		printf("{\n\tColor correction file computed by:\n\t\t");
+		printargs(argc, argv, stdout);
+		printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
+				i+1 < argc ? argv[i+1] : "{this_file}");
+		if (!scanning)
+			printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
+					i+1 < argc ? argv[i+1] : "{this_file}");
+		printf("}\n");
+		putmapping();			/* put out color mapping */
+	}
 	if (debugfp != NULL)		/* put out debug picture */
 		if (scanning)
 			picdebug();
@@ -208,9 +249,9 @@ char	**argv;
 	exit(0);
 userr:
 	fprintf(stderr,
-"Usage: %s [-d dbg.pic][-p xul yul xur yur xll yll xlr ylr] input.pic [output.cal]\n",
+"Usage: %s [-d dbg.pic][-P | -p xul yul xur yur xll yll xlr ylr][-i irrad][-m] input.pic [output.{cal|cwp}]\n",
 			progname);
-	fprintf(stderr, "   or: %s [-d dbg.pic] -c [xyY.dat [output.cal]]\n",
+	fprintf(stderr, "   or: %s [-d dbg.pic][-i irrad][-m] -c [xyY.dat [output.{cal|cwp}]]\n",
 			progname);
 	exit(1);
 }
@@ -235,8 +276,10 @@ init()				/* initialize */
 		exit(1);
 	}
 					/* map MacBeth colors to RGB space */
-	for (i = 0; i < 24; i++)
+	for (i = 0; i < 24; i++) {
 		xyY2RGB(mbRGB[i], mbxyY[i]);
+		scalecolor(mbRGB[i], irrad);
+	}
 }
 
 
@@ -310,7 +353,7 @@ getpicture()				/* load in picture colors */
 		scalecolor(inpRGB[i], d);
 		inpflags |= 1L<<i;
 	}
-	free((char *)scanln);
+	free((void *)scanln);
 }
 
 
@@ -385,7 +428,7 @@ compute()			/* compute color mapping */
 	COLOR	clrin[24], clrout[24];
 	long	cflags;
 	COLOR	ctmp;
-	register int	i, j, n;
+	register int	i, n;
 					/* did we get what we need? */
 	if ((inpflags & REQFLGS) != REQFLGS) {
 		fprintf(stderr, "%s: missing required input colors\n",
@@ -397,6 +440,24 @@ compute()			/* compute color mapping */
 		copycolor(bramp[i][0], inpRGB[mbneu[i]]);
 		copycolor(bramp[i][1], mbRGB[mbneu[i]]);
 	}
+					/* compute color space gamut */
+	if (scanning) {
+		copycolor(colmin, cblack);
+		copycolor(colmax, cwhite);
+		scalecolor(colmax, irrad);
+	} else
+		for (i = 0; i < 3; i++) {
+			colval(colmin,i) = colval(bramp[0][0],i) -
+				colval(bramp[0][1],i) *
+				(colval(bramp[1][0],i)-colval(bramp[0][0],i)) /
+				(colval(bramp[1][1],i)-colval(bramp[1][0],i));
+			colval(colmax,i) = colval(bramp[NMBNEU-2][0],i) +
+				(1.-colval(bramp[NMBNEU-2][1],i)) *
+				(colval(bramp[NMBNEU-1][0],i) -
+					colval(bramp[NMBNEU-2][0],i)) /
+				(colval(bramp[NMBNEU-1][1],i) -
+					colval(bramp[NMBNEU-2][1],i));
+		}
 					/* compute color mapping */
 	do {
 		cflags = inpflags & ~gmtflags;
@@ -408,17 +469,10 @@ compute()			/* compute color mapping */
 				n++;
 			}
 		compsoln(clrin, clrout, n);
-						/* check out-of-gamut colors */
-		for (i = 0; i < 24; i++)
-			if (cflags & 1L<<i) {
-				cvtcolor(ctmp, mbRGB[i]);
-				for (j = 0; j < 3; j++)
-					if (colval(ctmp,j) <= 1e-6 ||
-						colval(ctmp,j) >= 1.-1e-6) {
-						gmtflags |= 1L<<i;
-						break;
-					}
-			}
+		if (irrad > 0.99 && irrad < 1.01)	/* check gamut */
+			for (i = 0; i < 24; i++)
+				if (cflags & 1L<<i && cvtcolor(ctmp, mbRGB[i]))
+					gmtflags |= 1L<<i;
 	} while (cflags & gmtflags);
 	if (gmtflags & MODFLGS)
 		fprintf(stderr,
@@ -427,7 +481,7 @@ compute()			/* compute color mapping */
 }
 
 
-putmapping()			/* put out color mapping for pcomb -f */
+putmapping()			/* put out color mapping */
 {
 	static char	cchar[3] = {'r', 'g', 'b'};
 	register int	i, j;
@@ -535,39 +589,49 @@ int	n;
 }
 
 
+cwarp()				/* compute color warp map */
+{
+	register int	i;
+
+	if ((wcor = new3dw(W3EXACT)) == NULL)
+		goto memerr;
+	for (i = 0; i < 24; i++)
+		if (!add3dpt(wcor, inpRGB[i], mbRGB[i]))
+			goto memerr;
+	return;
+memerr:
+	perror(progname);
+	exit(1);
+}
+
+
+int
 cvtcolor(cout, cin)		/* convert color according to our mapping */
 COLOR	cout, cin;
 {
 	COLOR	ctmp;
+	int	clipped;
 
-	if (scanning) {
+	if (wcor != NULL) {
+		clipped = warp3d(cout, cin, wcor);
+		clipped |= clipgamut(cout,bright(cout),CGAMUT,colmin,colmax);
+	} else if (scanning) {
 		bresp(ctmp, cin);
-		cresp(cout, ctmp);
+		clipped = cresp(cout, ctmp);
 	} else {
-		cresp(ctmp, cin);
+		clipped = cresp(ctmp, cin);
 		bresp(cout, ctmp);
 	}
-	if (colval(cout,RED) < 0.)
-		colval(cout,RED) = 0.;
-	if (colval(cout,GRN) < 0.)
-		colval(cout,GRN) = 0.;
-	if (colval(cout,BLU) < 0.)
-		colval(cout,BLU) = 0.;
+	return(clipped);
 }
 
 
+int
 cresp(cout, cin)		/* transform color according to matrix */
 COLOR	cout, cin;
 {
-	double	r, g, b;
-
-	r = colval(cin,0)*solmat[0][0] + colval(cin,1)*solmat[0][1]
-			+ colval(cin,2)*solmat[0][2];
-	g = colval(cin,0)*solmat[1][0] + colval(cin,1)*solmat[1][1]
-			+ colval(cin,2)*solmat[1][2];
-	b = colval(cin,0)*solmat[2][0] + colval(cin,1)*solmat[2][1]
-			+ colval(cin,2)*solmat[2][2];
-	setcolor(cout, r, g, b);
+	colortrans(cout, solmat, cin);
+	return(clipgamut(cout, bright(cout), CGAMUT, colmin, colmax));
 }
 
 
@@ -582,9 +646,8 @@ register float	xyYin[3];
 	ctmp[0] = xyYin[0] * d;
 	ctmp[1] = xyYin[2];
 	ctmp[2] = (1. - xyYin[0] - xyYin[1]) * d;
-	cie_rgb(rgbout, ctmp);
 				/* allow negative values */
-	colortrans(rgbout, xyz2rgbmat, ctmp, 0);
+	colortrans(rgbout, xyz2rgbmat, ctmp);
 }
 
 
@@ -621,9 +684,11 @@ picdebug()			/* put out debugging picture */
 		for (x = 0; x < xmax; x++) {
 			rg = chartndx(x, y, &i);
 			if (rg == RG_CENT) {
-				if (!(1L<<i & gmtflags) || (x+y)&07)
+				if (!(1L<<i & gmtflags) || (x+y)&07) {
 					copycolor(scan[x], mbRGB[i]);
-				else
+					clipgamut(scan[x], bright(scan[x]),
+						CGAMUT, colmin, colmax);
+				} else
 					copycolor(scan[x], blkcol);
 			} else if (rg == RG_CORR)
 				cvtcolor(scan[x], scan[x]);
@@ -638,7 +703,7 @@ picdebug()			/* put out debugging picture */
 	}
 						/* clean up */
 	fclose(debugfp);
-	free((char *)scan);
+	free((void *)scan);
 }
 
 
@@ -647,20 +712,31 @@ clrdebug()			/* put out debug picture from color input
 	static COLR	blkclr = BLKCOLR;
 	COLR	mbclr[24], cvclr[24], orclr[24];
 	COLR	*scan;
-	COLOR	ctmp;
+	COLOR	ctmp, ct2;
 	int	y, i;
 	register int	x, rg;
 						/* convert colors */
 	for (i = 0; i < 24; i++) {
-		setcolr(mbclr[i], colval(mbRGB[i],RED),
-				colval(mbRGB[i],GRN), colval(mbRGB[i],BLU));
+		copycolor(ctmp, mbRGB[i]);
+		clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
+		setcolr(mbclr[i], colval(ctmp,RED),
+				colval(ctmp,GRN), colval(ctmp,BLU));
 		if (inpflags & 1L<<i) {
-			setcolr(orclr[i], colval(inpRGB[i],RED),
-					colval(inpRGB[i],GRN),
-					colval(inpRGB[i],BLU));
-			cvtcolor(ctmp, inpRGB[i]);
-			setcolr(cvclr[i], colval(ctmp,RED),
+			copycolor(ctmp, inpRGB[i]);
+			clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
+			setcolr(orclr[i], colval(ctmp,RED),
 					colval(ctmp,GRN), colval(ctmp,BLU));
+			if (rawmap)
+				copycolr(cvclr[i], mbclr[i]);
+			else {
+				bresp(ctmp, inpRGB[i]);
+				colortrans(ct2, solmat, ctmp);
+				clipgamut(ct2, bright(ct2), CGAMUT,
+						cblack, cwhite);
+				setcolr(cvclr[i], colval(ct2,RED),
+						colval(ct2,GRN),
+						colval(ct2,BLU));
+			}
 		}
 	}
 						/* allocate scanline */
@@ -697,5 +773,43 @@ clrdebug()			/* put out debug picture from color input
 	}
 						/* clean up */
 	fclose(debugfp);
-	free((char *)scan);
+	free((void *)scan);
+}
+
+
+getpos(name, bnds, fp)		/* get boundary position */
+char	*name;
+int	bnds[2];
+FILE	*fp;
+{
+	char	buf[64];
+
+	fprintf(stderr, "\tSelect corner: %s\n", name);
+	if (fgets(buf, sizeof(buf), fp) == NULL ||
+			sscanf(buf, "%d %d", &bnds[0], &bnds[1]) != 2) {
+		fprintf(stderr, "%s: read error from display process\n",
+				progname);
+		exit(1);
+	}
+}
+
+
+pickchartpos(pfn)		/* display picture and pick chart location */
+char	*pfn;
+{
+	char	combuf[512];
+	FILE	*pfp;
+
+	sprintf(combuf, DISPCOM, pfn);
+	if ((pfp = popen(combuf, "r")) == NULL) {
+		perror(combuf);
+		exit(1);
+	}
+	fputs("Use middle mouse button to select chart corners:\n", stderr);
+	getpos("upper left (dark skin)", bounds[0], pfp);
+	getpos("upper right (bluish green)", bounds[1], pfp);
+	getpos("lower left (white)", bounds[2], pfp);
+	getpos("lower right (black)", bounds[3], pfp);
+	fputs("Got it -- quit display program.\n", stderr);
+	pclose(pfp);
 }