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