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