ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/macbethcal.c
Revision: 2.32
Committed: Sat Jun 7 05:09:46 2025 UTC (12 days, 7 hours ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: HEAD
Changes since 2.31: +1 -2 lines
Log Message:
refactor: Put some declarations into "paths.h" and included in "platform.h"

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.32 static const char RCSid[] = "$Id: macbethcal.c,v 2.31 2025/06/03 21:31:51 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * Calibrate a scanned MacBeth Color Checker Chart
6     *
7 greg 2.14 * Produce a .cal file suitable for use with pcomb,
8     * or .cwp file suitable for use with pcwarp.
9     *
10     * Warping code depends on conformance of COLOR and W3VEC types.
11 greg 2.1 */
12    
13 greg 2.14 #include <math.h>
14 schorsch 2.18
15     #include "platform.h"
16 schorsch 2.23 #include "rtio.h"
17 greg 2.1 #include "color.h"
18     #include "resolu.h"
19     #include "pmap.h"
20 greg 2.14 #include "warp3d.h"
21 schorsch 2.23 #include "mx3.h"
22 greg 2.1
23 greg 2.4 /* MacBeth colors */
24     #define DarkSkin 0
25     #define LightSkin 1
26     #define BlueSky 2
27     #define Foliage 3
28     #define BlueFlower 4
29     #define BluishGreen 5
30     #define Orange 6
31     #define PurplishBlue 7
32     #define ModerateRed 8
33     #define Purple 9
34     #define YellowGreen 10
35     #define OrangeYellow 11
36     #define Blue 12
37     #define Green 13
38     #define Red 14
39     #define Yellow 15
40     #define Magenta 16
41     #define Cyan 17
42     #define White 18
43     #define Neutral8 19
44     #define Neutral65 20
45     #define Neutral5 21
46     #define Neutral35 22
47     #define Black 23
48 greg 2.26 /* computed from 10nm spectral measurements */
49 greg 2.4 /* CIE 1931 2 degree obs, equal-energy white */
50 greg 2.1 float mbxyY[24][3] = {
51 greg 2.26 {0.421236, 0.361196, 0.103392}, /* DarkSkin */
52     {0.40868, 0.358157, 0.352867}, /* LightSkin */
53     {0.265063, 0.271424, 0.185124}, /* BlueSky */
54     {0.362851, 0.43055, 0.132625}, /* Foliage */
55     {0.28888, 0.260851, 0.233138}, /* BlueFlower */
56     {0.277642, 0.365326, 0.416443}, /* BluishGreen */
57     {0.524965, 0.40068, 0.312039}, /* Orange */
58     {0.225018, 0.190392, 0.114999}, /* PurplishBlue */
59     {0.487199, 0.315372, 0.198616}, /* ModerateRed */
60     {0.314245, 0.227231, 0.0646047}, /* Purple */
61     {0.396202, 0.489732, 0.440724}, /* YellowGreen */
62     {0.493297, 0.435299, 0.43444}, /* OrangeYellow */
63     {0.198191, 0.149265, 0.0588122}, /* Blue */
64     {0.322838, 0.487601, 0.229258}, /* Green */
65     {0.561833, 0.321165, 0.126978}, /* Red */
66     {0.468113, 0.467021, 0.605289}, /* Yellow */
67     {0.397128, 0.248535, 0.201761}, /* Magenta */
68     {0.209552, 0.276256, 0.190917}, /* Cyan */
69     {0.337219, 0.339042, 0.912482}, /* White */
70     {0.333283, 0.335077, 0.588297}, /* Neutral.8 */
71     {0.332747, 0.334371, 0.3594}, /* Neutral.65 */
72     {0.331925, 0.334202, 0.19114}, /* Neutral.5 */
73     {0.330408, 0.332615, 0.0892964}, /* Neutral.35 */
74     {0.331841, 0.331405, 0.0319541}, /* Black */
75 greg 2.1 };
76    
77     COLOR mbRGB[24]; /* MacBeth RGB values */
78    
79     #define NMBNEU 6 /* Number of MacBeth neutral colors */
80 greg 2.4 short mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutral65,Neutral8,White};
81 greg 2.1
82 greg 2.8 #define NEUFLGS (1L<<White|1L<<Neutral8|1L<<Neutral65| \
83     1L<<Neutral5|1L<<Neutral35|1L<<Black)
84 greg 2.4
85 greg 2.8 #define SATFLGS (1L<<Red|1L<<Green|1L<<Blue|1L<<Magenta|1L<<Yellow| \
86     1L<<Cyan|1L<<Orange|1L<<Purple|1L<<PurplishBlue| \
87     1L<<YellowGreen|1<<OrangeYellow|1L<<BlueFlower)
88 greg 2.4
89 greg 2.8 #define UNSFLGS (1L<<DarkSkin|1L<<LightSkin|1L<<BlueSky|1L<<Foliage| \
90     1L<<BluishGreen|1L<<ModerateRed)
91 greg 2.7
92 greg 2.8 #define REQFLGS NEUFLGS /* need these colors */
93     #define MODFLGS (NEUFLGS|UNSFLGS) /* should be in gamut */
94    
95 greg 2.9 #define RG_BORD 0 /* patch border */
96     #define RG_CENT 01 /* central region of patch */
97     #define RG_ORIG 02 /* original color region */
98     #define RG_CORR 04 /* corrected color region */
99 greg 2.7
100 gwlarson 2.16 #ifndef DISPCOM
101 greg 2.29 #define DISPCOM "ximage -e auto -op \"%s\""
102 gwlarson 2.16 #endif
103    
104 greg 2.11 int scanning = 1; /* scanned input (or recorded output)? */
105 greg 2.14 double irrad = 1.0; /* irradiance multiplication factor */
106     int rawmap = 0; /* put out raw color mapping? */
107 greg 2.11
108 greg 2.1 int xmax, ymax; /* input image dimensions */
109     int bounds[4][2]; /* image coordinates of chart corners */
110     double imgxfm[3][3]; /* coordinate transformation matrix */
111    
112 greg 2.7 COLOR inpRGB[24]; /* measured or scanned input colors */
113     long inpflags = 0; /* flags of which colors were input */
114 greg 2.8 long gmtflags = 0; /* flags of out-of-gamut colors */
115 greg 2.1
116 greg 2.2 COLOR bramp[NMBNEU][2]; /* brightness ramp (per primary) */
117 greg 2.13 COLORMAT solmat; /* color mapping matrix */
118     COLOR colmin, colmax; /* gamut limits */
119 greg 2.1
120 greg 2.14 WARP3D *wcor = NULL; /* color space warp */
121    
122 greg 2.4 FILE *debugfp = NULL; /* debug output picture */
123 greg 2.1
124 schorsch 2.23 static void init(void);
125     static int chartndx(int x, int y, int *np);
126     static void getpicture(void);
127     static void getcolors(void);
128     static void bresp(COLOR y, COLOR x);
129 greg 2.30 static void ibresp(COLOR y, COLOR x);
130 schorsch 2.23 static void compute(void);
131     static void putmapping(void);
132     static void compsoln(COLOR cin[], COLOR cout[], int n);
133     static void cwarp(void);
134     static int cvtcolor(COLOR cout, COLOR cin);
135     static int cresp(COLOR cout, COLOR cin);
136     static void xyY2RGB(COLOR rgbout, float xyYin[3]);
137     static void picdebug(void);
138     static void clrdebug(void);
139     static void getpos(char *name, int bnds[2], FILE *fp);
140     static void pickchartpos(char *pfn);
141 greg 2.1
142 schorsch 2.23
143     int
144     main(
145     int argc,
146     char **argv
147     )
148 greg 2.1 {
149     int i;
150    
151 greg 2.31 fixargv0(argv[0]); /* sets global progname */
152 greg 2.7 for (i = 1; i < argc && argv[i][0] == '-'; i++)
153     switch (argv[i][1]) {
154     case 'd': /* debug output */
155     i++;
156     if (badarg(argc-i, argv+i, "s"))
157     goto userr;
158     if ((debugfp = fopen(argv[i], "w")) == NULL) {
159     perror(argv[i]);
160     exit(1);
161     }
162 schorsch 2.18 SET_FILE_BINARY(debugfp);
163 greg 2.7 newheader("RADIANCE", debugfp); /* start */
164     printargs(argc, argv, debugfp); /* header */
165     break;
166     case 'p': /* picture position */
167     if (badarg(argc-i-1, argv+i+1, "iiiiiiii"))
168     goto userr;
169     bounds[0][0] = atoi(argv[++i]);
170     bounds[0][1] = atoi(argv[++i]);
171     bounds[1][0] = atoi(argv[++i]);
172     bounds[1][1] = atoi(argv[++i]);
173     bounds[2][0] = atoi(argv[++i]);
174     bounds[2][1] = atoi(argv[++i]);
175     bounds[3][0] = atoi(argv[++i]);
176     bounds[3][1] = atoi(argv[++i]);
177 greg 2.11 scanning = 2;
178 greg 2.7 break;
179 gwlarson 2.16 case 'P': /* pick position */
180     scanning = 3;
181     break;
182 greg 2.14 case 'i': /* irradiance factor */
183     i++;
184     if (badarg(argc-i, argv+i, "f"))
185     goto userr;
186     irrad = atof(argv[i]);
187     break;
188     case 'm': /* raw map output */
189     rawmap = 1;
190     break;
191 greg 2.7 case 'c': /* color input */
192 greg 2.11 scanning = 0;
193 greg 2.7 break;
194     default:
195     goto userr;
196     }
197     /* open files */
198     if (i < argc && freopen(argv[i], "r", stdin) == NULL) {
199 greg 2.13 perror(argv[i]);
200 greg 2.1 exit(1);
201     }
202 greg 2.7 if (i+1 < argc && freopen(argv[i+1], "w", stdout) == NULL) {
203 greg 2.13 perror(argv[i+1]);
204 greg 2.1 exit(1);
205     }
206 greg 2.11 if (scanning) { /* load input picture header */
207 schorsch 2.18 SET_FILE_BINARY(stdin);
208 greg 2.7 if (checkheader(stdin, COLRFMT, NULL) < 0 ||
209     fgetresolu(&xmax, &ymax, stdin) < 0) {
210     fprintf(stderr, "%s: bad input picture\n", progname);
211     exit(1);
212     }
213 gwlarson 2.16 if (scanning == 3) {
214     if (i >= argc)
215     goto userr;
216     pickchartpos(argv[i]);
217     scanning = 2;
218     }
219 greg 2.7 } else { /* else set default xmax and ymax */
220     xmax = 512;
221     ymax = 2*512/3;
222 greg 2.1 }
223 greg 2.11 if (scanning != 2) { /* use default boundaries */
224 greg 2.1 bounds[0][0] = bounds[2][0] = .029*xmax + .5;
225     bounds[0][1] = bounds[1][1] = .956*ymax + .5;
226     bounds[1][0] = bounds[3][0] = .971*xmax + .5;
227     bounds[2][1] = bounds[3][1] = .056*ymax + .5;
228     }
229     init(); /* initialize */
230 greg 2.11 if (scanning) /* get picture colors */
231 greg 2.7 getpicture();
232     else
233     getcolors();
234 greg 2.2 compute(); /* compute color mapping */
235 greg 2.14 if (rawmap) { /* print out raw correspondence */
236 greg 2.29 int j;
237 greg 2.14
238     printf("# Color correspondence produced by:\n#\t\t");
239     printargs(argc, argv, stdout);
240 greg 2.25 printf("#\tUsage: pcwarp %s uncorrected.hdr > corrected.hdr\n",
241 greg 2.11 i+1 < argc ? argv[i+1] : "{this_file}");
242 greg 2.25 printf("#\t Or: pcond [options] -m %s orig.hdr > output.hdr\n",
243 greg 2.14 i+1 < argc ? argv[i+1] : "{this_file}");
244     for (j = 0; j < 24; j++)
245     printf("%f %f %f %f %f %f\n",
246     colval(inpRGB[j],RED), colval(inpRGB[j],GRN),
247     colval(inpRGB[j],BLU), colval(mbRGB[j],RED),
248     colval(mbRGB[j],GRN), colval(mbRGB[j],BLU));
249     if (scanning && debugfp != NULL)
250     cwarp(); /* color warp for debugging */
251     } else { /* print color mapping */
252     /* print header */
253     printf("{\n\tColor correction file computed by:\n\t\t");
254     printargs(argc, argv, stdout);
255 greg 2.25 printf("\n\tUsage: pcomb -f %s uncorrected.hdr > corrected.hdr\n",
256 greg 2.14 i+1 < argc ? argv[i+1] : "{this_file}");
257     if (!scanning)
258 greg 2.25 printf("\t Or: pcond [options] -f %s orig.hdr > output.hdr\n",
259 greg 2.14 i+1 < argc ? argv[i+1] : "{this_file}");
260     printf("}\n");
261     putmapping(); /* put out color mapping */
262     }
263 schorsch 2.19 if (debugfp != NULL) { /* put out debug picture */
264 greg 2.11 if (scanning)
265 greg 2.7 picdebug();
266     else
267     clrdebug();
268 schorsch 2.19 }
269 greg 2.1 exit(0);
270     userr:
271 greg 2.7 fprintf(stderr,
272 greg 2.25 "Usage: %s [-d dbg.hdr][-P | -p xul yul xur yur xll yll xlr ylr][-i irrad][-m] input.hdr [output.{cal|cwp}]\n",
273 greg 2.1 progname);
274 greg 2.25 fprintf(stderr, " or: %s [-d dbg.hdr][-i irrad][-m] -c [xyY.dat [output.{cal|cwp}]]\n",
275 greg 2.7 progname);
276 greg 2.1 exit(1);
277 schorsch 2.23 return 1; /* pro forma return */
278 greg 2.1 }
279    
280    
281 schorsch 2.23 static void
282     init(void) /* initialize */
283 greg 2.1 {
284     double quad[4][2];
285 greg 2.29 int i;
286 greg 2.1 /* make coordinate transformation */
287     quad[0][0] = bounds[0][0];
288     quad[0][1] = bounds[0][1];
289     quad[1][0] = bounds[1][0];
290     quad[1][1] = bounds[1][1];
291     quad[2][0] = bounds[3][0];
292     quad[2][1] = bounds[3][1];
293     quad[3][0] = bounds[2][0];
294     quad[3][1] = bounds[2][1];
295    
296     if (pmap_quad_rect(0., 0., 6., 4., quad, imgxfm) == PMAP_BAD) {
297     fprintf(stderr, "%s: bad chart boundaries\n", progname);
298     exit(1);
299     }
300 greg 2.7 /* map MacBeth colors to RGB space */
301 greg 2.14 for (i = 0; i < 24; i++) {
302 greg 2.7 xyY2RGB(mbRGB[i], mbxyY[i]);
303 greg 2.14 scalecolor(mbRGB[i], irrad);
304     }
305 greg 2.1 }
306    
307    
308 schorsch 2.23 static int
309     chartndx( /* find color number for position */
310     int x,
311     int y,
312     int *np
313     )
314 greg 2.1 {
315     double ipos[3], cpos[3];
316     int ix, iy;
317     double fx, fy;
318    
319     ipos[0] = x;
320     ipos[1] = y;
321     ipos[2] = 1;
322     mx3d_transform(ipos, imgxfm, cpos);
323     cpos[0] /= cpos[2];
324     cpos[1] /= cpos[2];
325 greg 2.29 if ((cpos[0] < 0.) | (cpos[0] >= 6.) | (cpos[1] < 0.) | (cpos[1] >= 4.))
326 greg 2.9 return(RG_BORD);
327 greg 2.1 ix = cpos[0];
328     iy = cpos[1];
329     fx = cpos[0] - ix;
330     fy = cpos[1] - iy;
331 greg 2.9 *np = iy*6 + ix;
332 greg 2.29 if ((fx >= 0.35) & (fx < 0.65) & (fy >= 0.35) & (fy < 0.65))
333 greg 2.9 return(RG_CENT);
334 greg 2.29 if ((fx < 0.05) | (fx >= 0.95) | (fy < 0.05) | (fy >= 0.95))
335 greg 2.9 return(RG_BORD);
336     if (fx >= 0.5) /* right side is corrected */
337     return(RG_CORR);
338     return(RG_ORIG); /* left side is original */
339 greg 2.1 }
340    
341    
342 schorsch 2.23 static void
343     getpicture(void) /* load in picture colors */
344 greg 2.1 {
345     COLR *scanln;
346     COLOR pval;
347     int ccount[24];
348     double d;
349 greg 2.9 int y, i;
350 greg 2.29 int x;
351 greg 2.1
352     scanln = (COLR *)malloc(xmax*sizeof(COLR));
353     if (scanln == NULL) {
354     perror(progname);
355     exit(1);
356     }
357     for (i = 0; i < 24; i++) {
358 greg 2.7 setcolor(inpRGB[i], 0., 0., 0.);
359 greg 2.1 ccount[i] = 0;
360     }
361     for (y = ymax-1; y >= 0; y--) {
362     if (freadcolrs(scanln, xmax, stdin) < 0) {
363     fprintf(stderr, "%s: error reading input picture\n",
364     progname);
365     exit(1);
366     }
367 greg 2.9 for (x = 0; x < xmax; x++)
368     if (chartndx(x, y, &i) == RG_CENT) {
369 greg 2.1 colr_color(pval, scanln[x]);
370 greg 2.7 addcolor(inpRGB[i], pval);
371 greg 2.1 ccount[i]++;
372     }
373     }
374 greg 2.7 for (i = 0; i < 24; i++) { /* compute averages */
375     if (ccount[i] == 0)
376     continue;
377     d = 1./ccount[i];
378     scalecolor(inpRGB[i], d);
379     inpflags |= 1L<<i;
380     }
381 greg 2.17 free((void *)scanln);
382 greg 2.7 }
383    
384    
385 schorsch 2.23 static void
386     getcolors(void) /* get xyY colors from standard input */
387 greg 2.7 {
388     int gotwhite = 0;
389     COLOR whiteclr;
390     int n;
391     float xyYin[3];
392    
393     while (fgetval(stdin, 'i', &n) == 1) { /* read colors */
394 schorsch 2.20 if ((n < 0) | (n > 24) ||
395 greg 2.7 fgetval(stdin, 'f', &xyYin[0]) != 1 ||
396     fgetval(stdin, 'f', &xyYin[1]) != 1 ||
397     fgetval(stdin, 'f', &xyYin[2]) != 1 ||
398 schorsch 2.20 (xyYin[0] < 0.) | (xyYin[1] < 0.) ||
399 greg 2.12 xyYin[0] + xyYin[1] > 1.) {
400 greg 2.7 fprintf(stderr, "%s: bad color input data\n",
401 greg 2.1 progname);
402     exit(1);
403     }
404 greg 2.7 if (n == 0) { /* calibration white */
405     xyY2RGB(whiteclr, xyYin);
406     gotwhite++;
407     } else { /* standard color */
408     n--;
409     xyY2RGB(inpRGB[n], xyYin);
410     inpflags |= 1L<<n;
411     }
412 greg 2.1 }
413 greg 2.7 /* normalize colors */
414     if (!gotwhite) {
415     if (!(inpflags & 1L<<White)) {
416     fprintf(stderr, "%s: missing input for White\n",
417     progname);
418     exit(1);
419     }
420     setcolor(whiteclr,
421     colval(inpRGB[White],RED)/colval(mbRGB[White],RED),
422     colval(inpRGB[White],GRN)/colval(mbRGB[White],GRN),
423     colval(inpRGB[White],BLU)/colval(mbRGB[White],BLU));
424     }
425     for (n = 0; n < 24; n++)
426     if (inpflags & 1L<<n)
427     setcolor(inpRGB[n],
428     colval(inpRGB[n],RED)/colval(whiteclr,RED),
429     colval(inpRGB[n],GRN)/colval(whiteclr,GRN),
430     colval(inpRGB[n],BLU)/colval(whiteclr,BLU));
431 greg 2.1 }
432    
433    
434 schorsch 2.23 static void
435     bresp( /* piecewise linear interpolation of primaries */
436 greg 2.30 COLOR y, /* y is linear output */
437     COLOR x /* x is non-linear input */
438 schorsch 2.23 )
439 greg 2.1 {
440 greg 2.29 int i, n;
441 greg 2.1
442 greg 2.2 for (i = 0; i < 3; i++) {
443 greg 2.5 for (n = 0; n < NMBNEU-2; n++)
444     if (colval(x,i) < colval(bramp[n+1][0],i))
445     break;
446 greg 2.8 colval(y,i) = ((colval(bramp[n+1][0],i) - colval(x,i)) *
447 greg 2.2 colval(bramp[n][1],i) +
448     (colval(x,i) - colval(bramp[n][0],i)) *
449     colval(bramp[n+1][1],i)) /
450     (colval(bramp[n+1][0],i) - colval(bramp[n][0],i));
451     }
452 greg 2.1 }
453    
454    
455 schorsch 2.23 static void
456 greg 2.30 ibresp( /* inverse mapping (delinearization) */
457     COLOR x, /* x is non-linear output */
458     COLOR y /* y is linear input */
459     )
460     {
461     int i, n;
462    
463     for (i = 0; i < 3; i++) {
464     for (n = 0; n < NMBNEU-2; n++)
465     if (colval(y,i) < colval(bramp[n+1][1],i))
466     break;
467     colval(x,i) = ((colval(bramp[n+1][1],i) - colval(y,i)) *
468     colval(bramp[n][0],i) +
469     (colval(y,i) - colval(bramp[n][1],i)) *
470     colval(bramp[n+1][0],i)) /
471     (colval(bramp[n+1][1],i) - colval(bramp[n][1],i));
472     }
473     }
474    
475    
476     static void
477 schorsch 2.23 compute(void) /* compute color mapping */
478 greg 2.1 {
479 greg 2.8 COLOR clrin[24], clrout[24];
480     long cflags;
481     COLOR ctmp;
482 greg 2.29 int i, n;
483 greg 2.7 /* did we get what we need? */
484     if ((inpflags & REQFLGS) != REQFLGS) {
485     fprintf(stderr, "%s: missing required input colors\n",
486     progname);
487     exit(1);
488 greg 2.2 }
489 greg 2.1 /* compute piecewise luminance curve */
490     for (i = 0; i < NMBNEU; i++) {
491 greg 2.7 copycolor(bramp[i][0], inpRGB[mbneu[i]]);
492 greg 2.29 for (n = 3*(i>0); n--; )
493 greg 2.24 if (colval(bramp[i][0],n) <=
494     colval(bramp[i-1][0],n)+1e-7) {
495     fprintf(stderr,
496     "%s: non-increasing neutral patch\n", progname);
497     exit(1);
498     }
499 greg 2.2 copycolor(bramp[i][1], mbRGB[mbneu[i]]);
500 greg 2.1 }
501 greg 2.13 /* compute color space gamut */
502 greg 2.30 copycolor(colmin, cblack);
503     copycolor(colmax, cwhite);
504     scalecolor(colmax, irrad);
505     if (!scanning) {
506     ibresp(colmin, colmin);
507     ibresp(colmax, colmax);
508     }
509 greg 2.8 /* compute color mapping */
510     do {
511     cflags = inpflags & ~gmtflags;
512     n = 0; /* compute transform matrix */
513 greg 2.30 for (i = 0; i < 24; i++) {
514     if (!(cflags & 1L<<i))
515     continue;
516     if (scanning) {
517 greg 2.8 bresp(clrin[n], inpRGB[i]);
518     copycolor(clrout[n], mbRGB[i]);
519 greg 2.30 } else {
520     copycolor(clrin[n], inpRGB[i]);
521     ibresp(clrout[n], mbRGB[i]);
522 greg 2.8 }
523 greg 2.30 n++;
524     }
525 greg 2.8 compsoln(clrin, clrout, n);
526 greg 2.29 for (i = 0; i < 24; i++) /* check gamut */
527     if (cflags & 1L<<i && cvtcolor(ctmp, inpRGB[i]))
528     gmtflags |= 1L<<i;
529 greg 2.8 } while (cflags & gmtflags);
530 greg 2.30
531 greg 2.8 if (gmtflags & MODFLGS)
532     fprintf(stderr,
533     "%s: warning - some moderate colors are out of gamut\n",
534     progname);
535 greg 2.2 }
536    
537    
538 schorsch 2.23 static void
539     putmapping(void) /* put out color mapping */
540 greg 2.2 {
541     static char cchar[3] = {'r', 'g', 'b'};
542 greg 2.29 int i, j;
543 greg 2.1 /* print brightness mapping */
544 greg 2.2 for (j = 0; j < 3; j++) {
545     printf("%cxa(i) : select(i", cchar[j]);
546     for (i = 0; i < NMBNEU; i++)
547     printf(",%g", colval(bramp[i][0],j));
548     printf(");\n");
549     printf("%cya(i) : select(i", cchar[j]);
550     for (i = 0; i < NMBNEU; i++)
551     printf(",%g", colval(bramp[i][1],j));
552     printf(");\n");
553     printf("%cfi(n) = if(n-%g, %d, if(%cxa(n+1)-%c, n, %cfi(n+1)));\n",
554     cchar[j], NMBNEU-1.5, NMBNEU-1, cchar[j],
555     cchar[j], cchar[j]);
556     printf("%cndx = %cfi(1);\n", cchar[j], cchar[j]);
557 greg 2.11 printf("%c%c = ((%cxa(%cndx+1)-%c)*%cya(%cndx) + ",
558     cchar[j], scanning?'n':'o', cchar[j],
559     cchar[j], cchar[j], cchar[j], cchar[j]);
560 greg 2.2 printf("(%c-%cxa(%cndx))*%cya(%cndx+1)) /\n",
561     cchar[j], cchar[j], cchar[j],
562     cchar[j], cchar[j]);
563     printf("\t\t(%cxa(%cndx+1) - %cxa(%cndx)) ;\n",
564     cchar[j], cchar[j], cchar[j], cchar[j]);
565     }
566 greg 2.1 /* print color mapping */
567 greg 2.11 if (scanning) {
568     printf("r = ri(1); g = gi(1); b = bi(1);\n");
569     printf("ro = %g*rn + %g*gn + %g*bn ;\n",
570     solmat[0][0], solmat[0][1], solmat[0][2]);
571     printf("go = %g*rn + %g*gn + %g*bn ;\n",
572     solmat[1][0], solmat[1][1], solmat[1][2]);
573     printf("bo = %g*rn + %g*gn + %g*bn ;\n",
574     solmat[2][0], solmat[2][1], solmat[2][2]);
575     } else {
576     printf("r1 = ri(1); g1 = gi(1); b1 = bi(1);\n");
577     printf("r = %g*r1 + %g*g1 + %g*b1 ;\n",
578     solmat[0][0], solmat[0][1], solmat[0][2]);
579     printf("g = %g*r1 + %g*g1 + %g*b1 ;\n",
580     solmat[1][0], solmat[1][1], solmat[1][2]);
581     printf("b = %g*r1 + %g*g1 + %g*b1 ;\n",
582     solmat[2][0], solmat[2][1], solmat[2][2]);
583     }
584 greg 2.1 }
585    
586    
587 schorsch 2.23 static void
588     compsoln( /* solve 3xN system using least-squares */
589     COLOR cin[],
590     COLOR cout[],
591     int n
592     )
593 greg 2.1 {
594     extern double mx3d_adjoint(), fabs();
595     double mat[3][3], invmat[3][3];
596     double det;
597     double colv[3], rowv[3];
598 greg 2.29 int i, j, k;
599 greg 2.1
600 greg 2.8 if (n < 3) {
601     fprintf(stderr, "%s: too few colors to match!\n", progname);
602 greg 2.1 exit(1);
603     }
604 greg 2.4 if (n == 3)
605     for (i = 0; i < 3; i++)
606     for (j = 0; j < 3; j++)
607     mat[i][j] = colval(cin[j],i);
608     else { /* compute A^t A */
609     for (i = 0; i < 3; i++)
610     for (j = i; j < 3; j++) {
611     mat[i][j] = 0.;
612     for (k = 0; k < n; k++)
613     mat[i][j] += colval(cin[k],i) *
614     colval(cin[k],j);
615     }
616     for (i = 1; i < 3; i++) /* using symmetry */
617     for (j = 0; j < i; j++)
618     mat[i][j] = mat[j][i];
619     }
620 greg 2.1 det = mx3d_adjoint(mat, invmat);
621     if (fabs(det) < 1e-4) {
622     fprintf(stderr, "%s: cannot compute color mapping\n",
623     progname);
624     solmat[0][0] = solmat[1][1] = solmat[2][2] = 1.;
625     solmat[0][1] = solmat[0][2] = solmat[1][0] =
626     solmat[1][2] = solmat[2][0] = solmat[2][1] = 0.;
627     return;
628     }
629     for (i = 0; i < 3; i++)
630     for (j = 0; j < 3; j++)
631     invmat[i][j] /= det;
632     for (i = 0; i < 3; i++) {
633 greg 2.4 if (n == 3)
634     for (j = 0; j < 3; j++)
635     colv[j] = colval(cout[j],i);
636     else
637     for (j = 0; j < 3; j++) {
638     colv[j] = 0.;
639     for (k = 0; k < n; k++)
640     colv[j] += colval(cout[k],i) *
641     colval(cin[k],j);
642     }
643 greg 2.3 mx3d_transform(colv, invmat, rowv);
644 greg 2.1 for (j = 0; j < 3; j++)
645 greg 2.3 solmat[i][j] = rowv[j];
646 greg 2.1 }
647     }
648    
649 greg 2.3
650 schorsch 2.23 static void
651     cwarp(void) /* compute color warp map */
652 greg 2.14 {
653 greg 2.29 int i;
654 greg 2.14
655     if ((wcor = new3dw(W3EXACT)) == NULL)
656     goto memerr;
657     for (i = 0; i < 24; i++)
658     if (!add3dpt(wcor, inpRGB[i], mbRGB[i]))
659     goto memerr;
660     return;
661     memerr:
662     perror(progname);
663     exit(1);
664     }
665    
666    
667 schorsch 2.23 static int
668     cvtcolor( /* convert color according to our mapping */
669     COLOR cout,
670     COLOR cin
671     )
672 greg 2.1 {
673 greg 2.8 COLOR ctmp;
674 greg 2.13 int clipped;
675 greg 2.8
676 greg 2.14 if (wcor != NULL) {
677     clipped = warp3d(cout, cin, wcor);
678     clipped |= clipgamut(cout,bright(cout),CGAMUT,colmin,colmax);
679     } else if (scanning) {
680 greg 2.11 bresp(ctmp, cin);
681 greg 2.13 clipped = cresp(cout, ctmp);
682 greg 2.11 } else {
683 greg 2.13 clipped = cresp(ctmp, cin);
684 greg 2.11 bresp(cout, ctmp);
685     }
686 greg 2.13 return(clipped);
687 greg 2.8 }
688    
689    
690 schorsch 2.23 static int
691     cresp( /* transform color according to matrix */
692     COLOR cout,
693     COLOR cin
694     )
695 greg 2.8 {
696 greg 2.13 colortrans(cout, solmat, cin);
697     return(clipgamut(cout, bright(cout), CGAMUT, colmin, colmax));
698 greg 2.1 }
699    
700    
701 schorsch 2.23 static void
702     xyY2RGB( /* convert xyY to RGB */
703     COLOR rgbout,
704 greg 2.29 float xyYin[3]
705 schorsch 2.23 )
706 greg 2.1 {
707 greg 2.7 COLOR ctmp;
708     double d;
709    
710     d = xyYin[2] / xyYin[1];
711     ctmp[0] = xyYin[0] * d;
712     ctmp[1] = xyYin[2];
713     ctmp[2] = (1. - xyYin[0] - xyYin[1]) * d;
714 greg 2.12 /* allow negative values */
715 greg 2.13 colortrans(rgbout, xyz2rgbmat, ctmp);
716 greg 2.7 }
717    
718    
719 schorsch 2.23 static void
720     picdebug(void) /* put out debugging picture */
721 greg 2.7 {
722 greg 2.8 static COLOR blkcol = BLKCOLOR;
723 greg 2.1 COLOR *scan;
724 greg 2.9 int y, i;
725 greg 2.29 int x, rg;
726 greg 2.1
727     if (fseek(stdin, 0L, 0) == EOF) {
728     fprintf(stderr, "%s: cannot seek on input picture\n", progname);
729     exit(1);
730     }
731     getheader(stdin, NULL, NULL); /* skip input header */
732     fgetresolu(&xmax, &ymax, stdin);
733     /* allocate scanline */
734     scan = (COLOR *)malloc(xmax*sizeof(COLOR));
735     if (scan == NULL) {
736     perror(progname);
737     exit(1);
738     }
739     /* finish debug header */
740 greg 2.5 fputformat(COLRFMT, debugfp);
741 greg 2.1 putc('\n', debugfp);
742     fprtresolu(xmax, ymax, debugfp);
743 greg 2.7 /* write debug picture */
744 greg 2.1 for (y = ymax-1; y >= 0; y--) {
745     if (freadscan(scan, xmax, stdin) < 0) {
746     fprintf(stderr, "%s: error rereading input picture\n",
747     progname);
748     exit(1);
749     }
750     for (x = 0; x < xmax; x++) {
751 greg 2.9 rg = chartndx(x, y, &i);
752     if (rg == RG_CENT) {
753 greg 2.13 if (!(1L<<i & gmtflags) || (x+y)&07) {
754 greg 2.9 copycolor(scan[x], mbRGB[i]);
755 greg 2.13 clipgamut(scan[x], bright(scan[x]),
756 greg 2.14 CGAMUT, colmin, colmax);
757 greg 2.13 } else
758 greg 2.9 copycolor(scan[x], blkcol);
759     } else if (rg == RG_CORR)
760 greg 2.1 cvtcolor(scan[x], scan[x]);
761 greg 2.9 else if (rg != RG_ORIG)
762 greg 2.8 copycolor(scan[x], blkcol);
763 greg 2.1 }
764     if (fwritescan(scan, xmax, debugfp) < 0) {
765     fprintf(stderr, "%s: error writing debugging picture\n",
766     progname);
767     exit(1);
768     }
769     }
770 greg 2.7 /* clean up */
771     fclose(debugfp);
772 greg 2.17 free((void *)scan);
773 greg 2.7 }
774    
775    
776 schorsch 2.23 static void
777     clrdebug(void) /* put out debug picture from color input */
778 greg 2.7 {
779     static COLR blkclr = BLKCOLR;
780 greg 2.9 COLR mbclr[24], cvclr[24], orclr[24];
781 greg 2.7 COLR *scan;
782 greg 2.13 COLOR ctmp, ct2;
783 greg 2.9 int y, i;
784 greg 2.29 int x, rg;
785 greg 2.7 /* convert colors */
786     for (i = 0; i < 24; i++) {
787 greg 2.13 copycolor(ctmp, mbRGB[i]);
788     clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
789     setcolr(mbclr[i], colval(ctmp,RED),
790     colval(ctmp,GRN), colval(ctmp,BLU));
791 greg 2.7 if (inpflags & 1L<<i) {
792 greg 2.13 copycolor(ctmp, inpRGB[i]);
793     clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
794     setcolr(orclr[i], colval(ctmp,RED),
795 greg 2.7 colval(ctmp,GRN), colval(ctmp,BLU));
796 greg 2.14 if (rawmap)
797     copycolr(cvclr[i], mbclr[i]);
798     else {
799     bresp(ctmp, inpRGB[i]);
800     colortrans(ct2, solmat, ctmp);
801     clipgamut(ct2, bright(ct2), CGAMUT,
802     cblack, cwhite);
803     setcolr(cvclr[i], colval(ct2,RED),
804     colval(ct2,GRN),
805     colval(ct2,BLU));
806     }
807 greg 2.7 }
808     }
809     /* allocate scanline */
810     scan = (COLR *)malloc(xmax*sizeof(COLR));
811     if (scan == NULL) {
812     perror(progname);
813     exit(1);
814     }
815     /* finish debug header */
816     fputformat(COLRFMT, debugfp);
817     putc('\n', debugfp);
818     fprtresolu(xmax, ymax, debugfp);
819     /* write debug picture */
820     for (y = ymax-1; y >= 0; y--) {
821 greg 2.9 for (x = 0; x < xmax; x++) {
822     rg = chartndx(x, y, &i);
823     if (rg == RG_CENT) {
824 greg 2.8 if (!(1L<<i & gmtflags) || (x+y)&07)
825     copycolr(scan[x], mbclr[i]);
826     else
827     copycolr(scan[x], blkclr);
828 greg 2.9 } else if (rg == RG_BORD || !(1L<<i & inpflags))
829     copycolr(scan[x], blkclr);
830     else if (rg == RG_ORIG)
831     copycolr(scan[x], orclr[i]);
832     else /* rg == RG_CORR */
833 greg 2.7 copycolr(scan[x], cvclr[i]);
834 greg 2.9 }
835 greg 2.7 if (fwritecolrs(scan, xmax, debugfp) < 0) {
836     fprintf(stderr, "%s: error writing debugging picture\n",
837     progname);
838     exit(1);
839     }
840     }
841     /* clean up */
842     fclose(debugfp);
843 greg 2.17 free((void *)scan);
844 gwlarson 2.16 }
845    
846    
847 schorsch 2.23 static void
848     getpos( /* get boundary position */
849     char *name,
850     int bnds[2],
851     FILE *fp
852     )
853 gwlarson 2.16 {
854     char buf[64];
855    
856     fprintf(stderr, "\tSelect corner: %s\n", name);
857     if (fgets(buf, sizeof(buf), fp) == NULL ||
858     sscanf(buf, "%d %d", &bnds[0], &bnds[1]) != 2) {
859     fprintf(stderr, "%s: read error from display process\n",
860     progname);
861     exit(1);
862     }
863     }
864    
865    
866 schorsch 2.23 static void
867     pickchartpos( /* display picture and pick chart location */
868     char *pfn
869     )
870 gwlarson 2.16 {
871 schorsch 2.22 char combuf[PATH_MAX];
872 gwlarson 2.16 FILE *pfp;
873    
874     sprintf(combuf, DISPCOM, pfn);
875     if ((pfp = popen(combuf, "r")) == NULL) {
876     perror(combuf);
877     exit(1);
878     }
879     fputs("Use middle mouse button to select chart corners:\n", stderr);
880     getpos("upper left (dark skin)", bounds[0], pfp);
881     getpos("upper right (bluish green)", bounds[1], pfp);
882     getpos("lower left (white)", bounds[2], pfp);
883     getpos("lower right (black)", bounds[3], pfp);
884     fputs("Got it -- quit display program.\n", stderr);
885     pclose(pfp);
886 greg 2.1 }