/* Copyright (c) 1995 Regents of the University of California */

#ifndef lint
static char SCCSid[] = "$SunId$ LBL";
#endif

/*
 * Calibrate a scanned MacBeth Color Checker Chart
 *
 * Produce a .cal file suitable for use with pcomb.
 */

#include <stdio.h>
#ifdef MSDOS
#include <fcntl.h>
#endif
#include "color.h"
#include "resolu.h"
#include "pmap.h"

				/* MacBeth colors */
#define	DarkSkin	0
#define	LightSkin	1
#define	BlueSky		2
#define	Foliage		3
#define	BlueFlower	4
#define	BluishGreen	5
#define	Orange		6
#define	PurplishBlue	7
#define	ModerateRed	8
#define	Purple		9
#define	YellowGreen	10
#define	OrangeYellow	11
#define	Blue		12
#define	Green		13
#define	Red		14
#define	Yellow		15
#define	Magenta		16
#define	Cyan		17
#define	White		18
#define	Neutral8	19
#define	Neutral65	20
#define	Neutral5	21
#define	Neutral35	22
#define	Black		23
				/* computed from 5nm spectral measurements */
				/* CIE 1931 2 degree obs, equal-energy white */
float	mbxyY[24][3] = {
		{0.462, 0.3769, 0.0932961},	/* DarkSkin */
		{0.4108, 0.3542, 0.410348},	/* LightSkin */
		{0.2626, 0.267, 0.181554},	/* BlueSky */
		{0.36, 0.4689, 0.108447},	/* Foliage */
		{0.2977, 0.2602, 0.248407},	/* BlueFlower */
		{0.2719, 0.3485, 0.401156},	/* BluishGreen */
		{0.52, 0.4197, 0.357899},	/* Orange */
		{0.229, 0.1866, 0.103911},	/* PurplishBlue */
		{0.4909, 0.3262, 0.242615},	/* ModerateRed */
		{0.3361, 0.2249, 0.0600102},	/* Purple */
		{0.3855, 0.4874, 0.42963},	/* YellowGreen */
		{0.4853, 0.4457, 0.476343},	/* OrangeYellow */
		{0.2026, 0.1369, 0.0529249},	/* Blue */
		{0.3007, 0.4822, 0.221226},	/* Green */
		{0.5805, 0.3238, 0.162167},	/* Red */
		{0.4617, 0.472, 0.64909},	/* Yellow */
		{0.4178, 0.2625, 0.233662},	/* Magenta */
		{0.2038, 0.2508, 0.167275},	/* Cyan */
		{0.3358, 0.337, 0.916877},	/* White */
		{0.3338, 0.3348, 0.604678},	/* Neutral.8 */
		{0.3333, 0.3349, 0.364566},	/* Neutral.65 */
		{0.3353, 0.3359, 0.200238},	/* Neutral.5 */
		{0.3363, 0.336, 0.0878721},	/* Neutral.35 */
		{0.3346, 0.3349, 0.0308383}	/* Black */
	};

COLOR	mbRGB[24];		/* MacBeth RGB values */

#define	NMBNEU		6	/* Number of MacBeth neutral colors */
short	mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutral65,Neutral8,White};

#define NMBMOD		16	/* Number of MacBeth unsaturated colors */
short	mbmod[NMBMOD] = {
		DarkSkin,LightSkin,BlueSky,Foliage,BlueFlower,BluishGreen,
		PurplishBlue,ModerateRed,YellowGreen,OrangeYellow,
		Black,Neutral35,Neutral5,Neutral65,Neutral8,White
	};

#define NMBSAT		8	/* Number of MacBeth saturated colors */
short	mbsat[NMBSAT] = {
		Red,Green,Blue,Magenta,Yellow,Cyan,
		Orange,Purple
	};

#define  REQFLGS	(1L<<White|1L<<Neutral8|1L<<Neutral65| \
				1L<<Neutral5|1L<<Neutral35|1L<<Black)

#define  CENTCVG	0.3	/* measured coverage of square sample */
#define  FULLCVG	0.9	/* coverage of entire square */

int	xmax, ymax;		/* input image dimensions */
int	bounds[4][2];		/* image coordinates of chart corners */
double	imgxfm[3][3];		/* coordinate transformation matrix */

COLOR	inpRGB[24];		/* measured or scanned input colors */
long	inpflags = 0;		/* flags of which colors were input */

COLOR	bramp[NMBNEU][2];	/* brightness ramp (per primary) */
double	solmat[3][3];		/* color mapping matrix */

FILE	*debugfp = NULL;	/* debug output picture */
char	*progname;

extern char	*malloc();


main(argc, argv)
int	argc;
char	**argv;
{
	int	inpispic = 1;
	int	i;

	progname = argv[0];
	for (i = 1; i < argc && argv[i][0] == '-'; i++)
		switch (argv[i][1]) {
		case 'd':				/* debug output */
			i++;
			if (badarg(argc-i, argv+i, "s"))
				goto userr;
			if ((debugfp = fopen(argv[i], "w")) == NULL) {
				perror(argv[i]);
				exit(1);
			}
#ifdef MSDOS
			setmode(fileno(debugfp), O_BINARY);
#endif
			newheader("RADIANCE", debugfp);		/* start */
			printargs(argc, argv, debugfp);		/* header */
			break;
		case 'p':				/* picture position */
			if (badarg(argc-i-1, argv+i+1, "iiiiiiii"))
				goto userr;
			bounds[0][0] = atoi(argv[++i]);
			bounds[0][1] = atoi(argv[++i]);
			bounds[1][0] = atoi(argv[++i]);
			bounds[1][1] = atoi(argv[++i]);
			bounds[2][0] = atoi(argv[++i]);
			bounds[2][1] = atoi(argv[++i]);
			bounds[3][0] = atoi(argv[++i]);
			bounds[3][1] = atoi(argv[++i]);
			inpispic = 2;
			break;
		case 'c':				/* color input */
			inpispic = 0;
			break;
		default:
			goto userr;
		}
							/* open files */
	if (i < argc && freopen(argv[i], "r", stdin) == NULL) {
		perror(argv[1]);
		exit(1);
	}
	if (i+1 < argc && freopen(argv[i+1], "w", stdout) == NULL) {
		perror(argv[2]);
		exit(1);
	}
	if (inpispic) {			/* load input picture header */
#ifdef MSDOS
		setmode(fileno(stdin), O_BINARY);
#endif
		if (checkheader(stdin, COLRFMT, NULL) < 0 ||
				fgetresolu(&xmax, &ymax, stdin) < 0) {
			fprintf(stderr, "%s: bad input picture\n", progname);
			exit(1);
		}
	} else {			/* else set default xmax and ymax */
		xmax = 512;
		ymax = 2*512/3;
	}
	if (inpispic != 2) {		/* use default boundaries */
		bounds[0][0] = bounds[2][0] = .029*xmax + .5;
		bounds[0][1] = bounds[1][1] = .956*ymax + .5;
		bounds[1][0] = bounds[3][0] = .971*xmax + .5;
		bounds[2][1] = bounds[3][1] = .056*ymax + .5;
	}
	init();				/* initialize */
	if (inpispic)			/* get picture colors */
		getpicture();
	else
		getcolors();
	compute();			/* compute color mapping */
					/* print comment */
	printf("{ Color correction file computed by:\n\t");
	printargs(argc, argv, stdout);
	printf("}\n");
	putmapping();			/* put out color mapping */
	if (debugfp != NULL)		/* put out debug picture */
		if (inpispic)
			picdebug();
		else
			clrdebug();
	exit(0);
userr:
	fprintf(stderr,
"Usage: %s [-d dbg.pic][-p xul yul xur yur xll yll xlr ylr] input.pic [output.cal]\n",
			progname);
	fprintf(stderr, "   or: %s [-d dbg.pic] -c [xyY.dat [output.cal]]\n",
			progname);
	exit(1);
}


init()				/* initialize */
{
	double	quad[4][2];
	register int	i;
					/* make coordinate transformation */
	quad[0][0] = bounds[0][0];
	quad[0][1] = bounds[0][1];
	quad[1][0] = bounds[1][0];
	quad[1][1] = bounds[1][1];
	quad[2][0] = bounds[3][0];
	quad[2][1] = bounds[3][1];
	quad[3][0] = bounds[2][0];
	quad[3][1] = bounds[2][1];

	if (pmap_quad_rect(0., 0., 6., 4., quad, imgxfm) == PMAP_BAD) {
		fprintf(stderr, "%s: bad chart boundaries\n", progname);
		exit(1);
	}
					/* map MacBeth colors to RGB space */
	for (i = 0; i < 24; i++)
		xyY2RGB(mbRGB[i], mbxyY[i]);
}


int
chartndx(x, y, cvg)			/* find color number for position */
int	x, y;
double	cvg;
{
	double	ipos[3], cpos[3];
	int	ix, iy;
	double	fx, fy;
	double  cmin, cmax;

	ipos[0] = x;
	ipos[1] = y;
	ipos[2] = 1;
	mx3d_transform(ipos, imgxfm, cpos);
	cpos[0] /= cpos[2];
	cpos[1] /= cpos[2];
	if (cpos[0] < 0. || cpos[0] >= 6. || cpos[1] < 0. || cpos[1] >= 4.)
		return(-1);
	ix = cpos[0];
	iy = cpos[1];
	fx = cpos[0] - ix;
	fy = cpos[1] - iy;
	cmin = .5*(1.-cvg);
	cmax = 1. - cmin;
	if (fx < cmin || fx >= cmax || fy < cmin || fy >= cmax)
		return(-1);
	return(iy*6 + ix);
}


getpicture()				/* load in picture colors */
{
	COLR	*scanln;
	COLOR	pval;
	int	ccount[24];
	double	d;
	int	y;
	register int	x, i;

	scanln = (COLR *)malloc(xmax*sizeof(COLR));
	if (scanln == NULL) {
		perror(progname);
		exit(1);
	}
	for (i = 0; i < 24; i++) {
		setcolor(inpRGB[i], 0., 0., 0.);
		ccount[i] = 0;
	}
	for (y = ymax-1; y >= 0; y--) {
		if (freadcolrs(scanln, xmax, stdin) < 0) {
			fprintf(stderr, "%s: error reading input picture\n",
					progname);
			exit(1);
		}
		for (x = 0; x < xmax; x++) {
			i = chartndx(x, y, CENTCVG);
			if (i >= 0) {
				colr_color(pval, scanln[x]);
				addcolor(inpRGB[i], pval);
				ccount[i]++;
			}
		}
	}
	for (i = 0; i < 24; i++) {		/* compute averages */
		if (ccount[i] == 0)
			continue;
		d = 1./ccount[i];
		scalecolor(inpRGB[i], d);
		inpflags |= 1L<<i;
	}
	free((char *)scanln);
}


getcolors()			/* get xyY colors from standard input */
{
	int	gotwhite = 0;
	COLOR	whiteclr;
	int	n;
	float	xyYin[3];

	while (fgetval(stdin, 'i', &n) == 1) {		/* read colors */
		if (n < 0 | n > 24 ||
				fgetval(stdin, 'f', &xyYin[0]) != 1 ||
				fgetval(stdin, 'f', &xyYin[1]) != 1 ||
				fgetval(stdin, 'f', &xyYin[2]) != 1 ||
				xyYin[0] < 0. | xyYin[0] > 1. |
				xyYin[1] < 0. | xyYin[1] > 1.) {
			fprintf(stderr, "%s: bad color input data\n",
					progname);
			exit(1);
		}
		if (n == 0) {				/* calibration white */
			xyY2RGB(whiteclr, xyYin);
			gotwhite++;
		} else {				/* standard color */
			n--;
			xyY2RGB(inpRGB[n], xyYin);
			inpflags |= 1L<<n;
		}
	}
					/* normalize colors */
	if (!gotwhite) {
		if (!(inpflags & 1L<<White)) {
			fprintf(stderr, "%s: missing input for White\n",
					progname);
			exit(1);
		}
		setcolor(whiteclr,
			colval(inpRGB[White],RED)/colval(mbRGB[White],RED),
			colval(inpRGB[White],GRN)/colval(mbRGB[White],GRN),
			colval(inpRGB[White],BLU)/colval(mbRGB[White],BLU));
	}
	for (n = 0; n < 24; n++)
		if (inpflags & 1L<<n)
			setcolor(inpRGB[n],
				colval(inpRGB[n],RED)/colval(whiteclr,RED),
				colval(inpRGB[n],GRN)/colval(whiteclr,GRN),
				colval(inpRGB[n],BLU)/colval(whiteclr,BLU));
}


bresp(y, x)		/* piecewise linear interpolation of primaries */
COLOR	y, x;
{
	double	cv[3];
	register int	i, n;

	for (i = 0; i < 3; i++) {
		for (n = 0; n < NMBNEU-2; n++)
			if (colval(x,i) < colval(bramp[n+1][0],i))
				break;
		cv[i] = ((colval(bramp[n+1][0],i) - colval(x,i)) *
						colval(bramp[n][1],i) +
				(colval(x,i) - colval(bramp[n][0],i)) *
						colval(bramp[n+1][1],i)) /
			(colval(bramp[n+1][0],i) - colval(bramp[n][0],i));
		if (cv[i] < 0.) cv[i] = 0.;
	}
	setcolor(y, cv[0], cv[1], cv[2]);
}


compute()			/* compute color mapping */
{
	COLOR	clrin[NMBMOD], clrout[NMBMOD];
	register int	i, n;
					/* did we get what we need? */
	if ((inpflags & REQFLGS) != REQFLGS) {
		fprintf(stderr, "%s: missing required input colors\n",
				progname);
		exit(1);
	}
					/* compute piecewise luminance curve */
	for (i = 0; i < NMBNEU; i++) {
		copycolor(bramp[i][0], inpRGB[mbneu[i]]);
		copycolor(bramp[i][1], mbRGB[mbneu[i]]);
	}
					/* compute color matrix */
	for (n = 0, i = 0; i < NMBMOD; i++)
		if (inpflags & 1L<<mbmod[i]) {
			bresp(clrin[n], inpRGB[mbmod[i]]);
			copycolor(clrout[n], mbRGB[mbmod[i]]);
			n++;
		}
	compsoln(clrin, clrout, n);
}


putmapping()			/* put out color mapping for pcomb -f */
{
	static char	cchar[3] = {'r', 'g', 'b'};
	register int	i, j;
					/* print brightness mapping */
	for (j = 0; j < 3; j++) {
		printf("%cxa(i) : select(i", cchar[j]);
		for (i = 0; i < NMBNEU; i++)
			printf(",%g", colval(bramp[i][0],j));
		printf(");\n");
		printf("%cya(i) : select(i", cchar[j]);
		for (i = 0; i < NMBNEU; i++)
			printf(",%g", colval(bramp[i][1],j));
		printf(");\n");
		printf("%c = %ci(1);\n", cchar[j], cchar[j]);
		printf("%cfi(n) = if(n-%g, %d, if(%cxa(n+1)-%c, n, %cfi(n+1)));\n",
				cchar[j], NMBNEU-1.5, NMBNEU-1, cchar[j],
				cchar[j], cchar[j]);
		printf("%cndx = %cfi(1);\n", cchar[j], cchar[j]);
		printf("%cn = ((%cxa(%cndx+1)-%c)*%cya(%cndx) + ",
				cchar[j], cchar[j], cchar[j],
				cchar[j], cchar[j], cchar[j]);
		printf("(%c-%cxa(%cndx))*%cya(%cndx+1)) /\n",
				cchar[j], cchar[j], cchar[j],
				cchar[j], cchar[j]);
		printf("\t\t(%cxa(%cndx+1) - %cxa(%cndx)) ;\n",
				cchar[j], cchar[j], cchar[j], cchar[j]);
	}
					/* print color mapping */
	printf("ro = %g*rn + %g*gn + %g*bn ;\n",
			solmat[0][0], solmat[0][1], solmat[0][2]);
	printf("go = %g*rn + %g*gn + %g*bn ;\n",
			solmat[1][0], solmat[1][1], solmat[1][2]);
	printf("bo = %g*rn + %g*gn + %g*bn ;\n",
			solmat[2][0], solmat[2][1], solmat[2][2]);
}


compsoln(cin, cout, n)		/* solve 3xN system using least-squares */
COLOR	cin[], cout[];
int	n;
{
	extern double	mx3d_adjoint(), fabs();
	double	mat[3][3], invmat[3][3];
	double	det;
	double	colv[3], rowv[3];
	register int	i, j, k;

	if (n < 3 | n > NMBMOD) {
		fprintf(stderr, "%s: bad number of colors to match\n", progname);
		exit(1);
	}
	if (n == 3)
		for (i = 0; i < 3; i++)
			for (j = 0; j < 3; j++)
				mat[i][j] = colval(cin[j],i);
	else {				/* compute A^t A */
		for (i = 0; i < 3; i++)
			for (j = i; j < 3; j++) {
				mat[i][j] = 0.;
				for (k = 0; k < n; k++)
					mat[i][j] += colval(cin[k],i) *
							colval(cin[k],j);
			}
		for (i = 1; i < 3; i++)		/* using symmetry */
			for (j = 0; j < i; j++)
				mat[i][j] = mat[j][i];
	}
	det = mx3d_adjoint(mat, invmat);
	if (fabs(det) < 1e-4) {
		fprintf(stderr, "%s: cannot compute color mapping\n",
				progname);
		solmat[0][0] = solmat[1][1] = solmat[2][2] = 1.;
		solmat[0][1] = solmat[0][2] = solmat[1][0] =
		solmat[1][2] = solmat[2][0] = solmat[2][1] = 0.;
		return;
	}
	for (i = 0; i < 3; i++)
		for (j = 0; j < 3; j++)
			invmat[i][j] /= det;
	for (i = 0; i < 3; i++) {
		if (n == 3)
			for (j = 0; j < 3; j++)
				colv[j] = colval(cout[j],i);
		else
			for (j = 0; j < 3; j++) {
				colv[j] = 0.;
				for (k = 0; k < n; k++)
					colv[j] += colval(cout[k],i) *
							colval(cin[k],j);
			}
		mx3d_transform(colv, invmat, rowv);
		for (j = 0; j < 3; j++)
			solmat[i][j] = rowv[j];
	}
}


cvtcolor(cout, cin)		/* convert color according to our mapping */
COLOR	cout, cin;
{
	double	r, g, b;

	bresp(cout, cin);
	r = colval(cout,0)*solmat[0][0] + colval(cout,1)*solmat[0][1]
			+ colval(cout,2)*solmat[0][2];
	if (r < 0) r = 0;
	g = colval(cout,0)*solmat[1][0] + colval(cout,1)*solmat[1][1]
			+ colval(cout,2)*solmat[1][2];
	if (g < 0) g = 0;
	b = colval(cout,0)*solmat[2][0] + colval(cout,1)*solmat[2][1]
			+ colval(cout,2)*solmat[2][2];
	if (b < 0) b = 0;
	setcolor(cout, r, g, b);
}


xyY2RGB(rgbout, xyYin)		/* convert xyY to RGB */
COLOR	rgbout;
register float	xyYin[3];
{
	COLOR	ctmp;
	double	d;

	d = xyYin[2] / xyYin[1];
	ctmp[0] = xyYin[0] * d;
	ctmp[1] = xyYin[2];
	ctmp[2] = (1. - xyYin[0] - xyYin[1]) * d;
	cie_rgb(rgbout, ctmp);
}


picdebug()			/* put out debugging picture */
{
	COLOR	*scan;
	int	y;
	register int	x, i;

	if (fseek(stdin, 0L, 0) == EOF) {
		fprintf(stderr, "%s: cannot seek on input picture\n", progname);
		exit(1);
	}
	getheader(stdin, NULL, NULL);		/* skip input header */
	fgetresolu(&xmax, &ymax, stdin);
						/* allocate scanline */
	scan = (COLOR *)malloc(xmax*sizeof(COLOR));
	if (scan == NULL) {
		perror(progname);
		exit(1);
	}
						/* finish debug header */
	fputformat(COLRFMT, debugfp);
	putc('\n', debugfp);
	fprtresolu(xmax, ymax, debugfp);
						/* write debug picture */
	for (y = ymax-1; y >= 0; y--) {
		if (freadscan(scan, xmax, stdin) < 0) {
			fprintf(stderr, "%s: error rereading input picture\n",
					progname);
			exit(1);
		}
		for (x = 0; x < xmax; x++) {
			i = chartndx(x, y, CENTCVG);
			if (i < 0)
				cvtcolor(scan[x], scan[x]);
			else
				copycolor(scan[x], mbRGB[i]);
		}
		if (fwritescan(scan, xmax, debugfp) < 0) {
			fprintf(stderr, "%s: error writing debugging picture\n",
					progname);
			exit(1);
		}
	}
						/* clean up */
	fclose(debugfp);
	free((char *)scan);
}


clrdebug()			/* put out debug picture from color input */
{
	static COLR	blkclr = BLKCOLR;
	COLR	mbclr[24], cvclr[24];
	COLR	*scan;
	COLOR	ctmp;
	int	y;
	register int	i, x;
						/* convert colors */
	for (i = 0; i < 24; i++) {
		setcolr(mbclr[i], colval(mbRGB[i],RED),
				colval(mbRGB[i],GRN), colval(mbRGB[i],BLU));
		if (inpflags & 1L<<i) {
			cvtcolor(ctmp, inpRGB[i]);
			setcolr(cvclr[i], colval(ctmp,RED),
					colval(ctmp,GRN), colval(ctmp,BLU));
		}
	}
						/* allocate scanline */
	scan = (COLR *)malloc(xmax*sizeof(COLR));
	if (scan == NULL) {
		perror(progname);
		exit(1);
	}
						/* finish debug header */
	fputformat(COLRFMT, debugfp);
	putc('\n', debugfp);
	fprtresolu(xmax, ymax, debugfp);
						/* write debug picture */
	for (y = ymax-1; y >= 0; y--) {
		for (x = 0; x < xmax; x++)
			if ((i = chartndx(x, y, CENTCVG)) >= 0)
				copycolr(scan[x], mbclr[i]);
			else if ((i = chartndx(x, y, FULLCVG)) >= 0 &&
					inpflags & 1L<<i)
				copycolr(scan[x], cvclr[i]);
			else
				copycolr(scan[x], blkclr);
		if (fwritecolrs(scan, xmax, debugfp) < 0) {
			fprintf(stderr, "%s: error writing debugging picture\n",
					progname);
			exit(1);
		}
	}
						/* clean up */
	fclose(debugfp);
	free((char *)scan);
}