--- ray/src/px/macbethcal.c 1995/10/20 15:26:11 2.7 +++ ray/src/px/macbethcal.c 2004/03/28 20:33:13 2.23 @@ -1,22 +1,27 @@ -/* Copyright (c) 1995 Regents of the University of California */ - #ifndef lint -static char SCCSid[] = "$SunId$ LBL"; +static const char RCSid[] = "$Id: macbethcal.c,v 2.23 2004/03/28 20:33:13 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 -#ifdef MSDOS -#include -#endif +#include +#include + +#include "platform.h" +#include "rtprocess.h" +#include "rtio.h" #include "color.h" #include "resolu.h" #include "pmap.h" +#include "warp3d.h" +#include "mx3.h" /* MacBeth colors */ #define DarkSkin 0 @@ -77,46 +82,73 @@ 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 NEUFLGS (1L<= argc) + goto userr; + pickchartpos(argv[i]); + scanning = 2; + } } else { /* else set default xmax and ymax */ xmax = 512; ymax = 2*512/3; } - if (inpispic != 2) { /* use default boundaries */ + if (scanning != 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 */ + if (scanning) /* 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) + 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("#\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(); 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", +"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); + return 1; /* pro forma return */ } -init() /* initialize */ +static void +init(void) /* initialize */ { double quad[4][2]; register int i; @@ -229,20 +301,23 @@ 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); + } } -int -chartndx(x, y, cvg) /* find color number for position */ -int x, y; -double cvg; +static int +chartndx( /* find color number for position */ + int x, + int y, + int *np +) { double ipos[3], cpos[3]; int ix, iy; double fx, fy; - double cmin, cmax; ipos[0] = x; ipos[1] = y; @@ -251,27 +326,31 @@ double cvg; cpos[0] /= cpos[2]; cpos[1] /= cpos[2]; if (cpos[0] < 0. || cpos[0] >= 6. || cpos[1] < 0. || cpos[1] >= 4.) - return(-1); + return(RG_BORD); 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); + *np = iy*6 + ix; + if (fx >= 0.35 && fx < 0.65 && fy >= 0.35 && fy < 0.65) + return(RG_CENT); + if (fx < 0.05 || fx >= 0.95 || fy < 0.05 || fy >= 0.95) + return(RG_BORD); + if (fx >= 0.5) /* right side is corrected */ + return(RG_CORR); + return(RG_ORIG); /* left side is original */ } -getpicture() /* load in picture colors */ +static void +getpicture(void) /* load in picture colors */ { COLR *scanln; COLOR pval; int ccount[24]; double d; - int y; - register int x, i; + int y, i; + register int x; scanln = (COLR *)malloc(xmax*sizeof(COLR)); if (scanln == NULL) { @@ -288,14 +367,12 @@ getpicture() /* load in picture colors */ progname); exit(1); } - for (x = 0; x < xmax; x++) { - i = chartndx(x, y, CENTCVG); - if (i >= 0) { + for (x = 0; x < xmax; x++) + if (chartndx(x, y, &i) == RG_CENT) { colr_color(pval, scanln[x]); addcolor(inpRGB[i], pval); ccount[i]++; } - } } for (i = 0; i < 24; i++) { /* compute averages */ if (ccount[i] == 0) @@ -304,11 +381,12 @@ getpicture() /* load in picture colors */ scalecolor(inpRGB[i], d); inpflags |= 1L< 24 || + 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.) { + (xyYin[0] < 0.) | (xyYin[1] < 0.) || + xyYin[0] + xyYin[1] > 1.) { fprintf(stderr, "%s: bad color input data\n", progname); exit(1); @@ -356,30 +434,33 @@ getcolors() /* get xyY colors from standard input */ } -bresp(y, x) /* piecewise linear interpolation of primaries */ -COLOR y, x; +static void +bresp( /* piecewise linear interpolation of primaries */ + COLOR y, + COLOR 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(y,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 */ +static void +compute(void) /* compute color mapping */ { - COLOR clrin[NMBMOD], clrout[NMBMOD]; + COLOR clrin[24], clrout[24]; + long cflags; + COLOR ctmp; register int i, n; /* did we get what we need? */ if ((inpflags & REQFLGS) != REQFLGS) { @@ -392,18 +473,49 @@ compute() /* compute color mapping */ 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< 0.99 && irrad < 1.01) /* check gamut */ + for (i = 0; i < 24; i++) + if (cflags & 1L< NMBMOD) { - fprintf(stderr, "%s: bad number of colors to match\n", progname); + if (n < 3) { + fprintf(stderr, "%s: too few colors to match!\n", progname); exit(1); } if (n == 3) @@ -501,45 +626,82 @@ int n; } -cvtcolor(cout, cin) /* convert color according to our mapping */ -COLOR cout, cin; +static void +cwarp(void) /* compute color warp map */ { - double r, g, b; + register int i; - 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); + 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); } -xyY2RGB(rgbout, xyYin) /* convert xyY to RGB */ -COLOR rgbout; -register float xyYin[3]; +static int +cvtcolor( /* convert color according to our mapping */ + COLOR cout, + COLOR cin +) { COLOR ctmp; + int clipped; + + if (wcor != NULL) { + clipped = warp3d(cout, cin, wcor); + clipped |= clipgamut(cout,bright(cout),CGAMUT,colmin,colmax); + } else if (scanning) { + bresp(ctmp, cin); + clipped = cresp(cout, ctmp); + } else { + clipped = cresp(ctmp, cin); + bresp(cout, ctmp); + } + return(clipped); +} + + +static int +cresp( /* transform color according to matrix */ + COLOR cout, + COLOR cin +) +{ + colortrans(cout, solmat, cin); + return(clipgamut(cout, bright(cout), CGAMUT, colmin, colmax)); +} + + +static void +xyY2RGB( /* 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); + /* allow negative values */ + colortrans(rgbout, xyz2rgbmat, ctmp); } -picdebug() /* put out debugging picture */ +static void +picdebug(void) /* put out debugging picture */ { + static COLOR blkcol = BLKCOLOR; COLOR *scan; - int y; - register int x, i; + int y, i; + register int x, rg; if (fseek(stdin, 0L, 0) == EOF) { fprintf(stderr, "%s: cannot seek on input picture\n", progname); @@ -565,11 +727,18 @@ picdebug() /* put out debugging picture */ exit(1); } for (x = 0; x < xmax; x++) { - i = chartndx(x, y, CENTCVG); - if (i < 0) + rg = chartndx(x, y, &i); + if (rg == RG_CENT) { + if (!(1L<= 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<