ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/macbethcal.c
(Generate patch)

Comparing ray/src/px/macbethcal.c (file contents):
Revision 2.1 by greg, Wed Oct 11 10:40:13 1995 UTC vs.
Revision 2.12 by greg, Thu Jan 30 19:14:11 1997 UTC

# Line 1 | Line 1
1 < /* Copyright (c) 1995 Regents of the University of California */
1 > /* Copyright (c) 1997 Regents of the University of California */
2  
3   #ifndef lint
4   static char SCCSid[] = "$SunId$ LBL";
# Line 18 | Line 18 | static char SCCSid[] = "$SunId$ LBL";
18   #include "resolu.h"
19   #include "pmap.h"
20  
21 <                                /* MacBeth colors (CIE 1931, absolute white) */
22 <                                /* computed from spectral measurements */
21 >                                /* MacBeth colors */
22 > #define DarkSkin        0
23 > #define LightSkin       1
24 > #define BlueSky         2
25 > #define Foliage         3
26 > #define BlueFlower      4
27 > #define BluishGreen     5
28 > #define Orange          6
29 > #define PurplishBlue    7
30 > #define ModerateRed     8
31 > #define Purple          9
32 > #define YellowGreen     10
33 > #define OrangeYellow    11
34 > #define Blue            12
35 > #define Green           13
36 > #define Red             14
37 > #define Yellow          15
38 > #define Magenta         16
39 > #define Cyan            17
40 > #define White           18
41 > #define Neutral8        19
42 > #define Neutral65       20
43 > #define Neutral5        21
44 > #define Neutral35       22
45 > #define Black           23
46 >                                /* computed from 5nm spectral measurements */
47 >                                /* CIE 1931 2 degree obs, equal-energy white */
48   float   mbxyY[24][3] = {
49                  {0.462, 0.3769, 0.0932961},     /* DarkSkin */
50                  {0.4108, 0.3542, 0.410348},     /* LightSkin */
# Line 50 | Line 75 | float  mbxyY[24][3] = {
75   COLOR   mbRGB[24];              /* MacBeth RGB values */
76  
77   #define NMBNEU          6       /* Number of MacBeth neutral colors */
78 < short   mbneu[NMBNEU] = {23,22,21,20,19,18};
54 < #define NMBMOD          3       /* Number of MacBeth moderate colors */
55 < short   mbmod[NMBMOD] = {1,3,2};
56 < #define NMBSAT          6       /* Number of MacBeth saturated colors */
57 < short   mbsat[NMBSAT] = {14,12,13,15,16,17};
78 > short   mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutral65,Neutral8,White};
79  
80 + #define  NEUFLGS        (1L<<White|1L<<Neutral8|1L<<Neutral65| \
81 +                                1L<<Neutral5|1L<<Neutral35|1L<<Black)
82 +
83 + #define  SATFLGS        (1L<<Red|1L<<Green|1L<<Blue|1L<<Magenta|1L<<Yellow| \
84 +                        1L<<Cyan|1L<<Orange|1L<<Purple|1L<<PurplishBlue| \
85 +                        1L<<YellowGreen|1<<OrangeYellow|1L<<BlueFlower)
86 +
87 + #define  UNSFLGS        (1L<<DarkSkin|1L<<LightSkin|1L<<BlueSky|1L<<Foliage| \
88 +                        1L<<BluishGreen|1L<<ModerateRed)
89 +
90 + #define  REQFLGS        NEUFLGS                 /* need these colors */
91 + #define  MODFLGS        (NEUFLGS|UNSFLGS)       /* should be in gamut */
92 +
93 + #define  RG_BORD        0       /* patch border */
94 + #define  RG_CENT        01      /* central region of patch */
95 + #define  RG_ORIG        02      /* original color region */
96 + #define  RG_CORR        04      /* corrected color region */
97 +
98 + int     scanning = 1;           /* scanned input (or recorded output)? */
99 +
100   int     xmax, ymax;             /* input image dimensions */
101   int     bounds[4][2];           /* image coordinates of chart corners */
102   double  imgxfm[3][3];           /* coordinate transformation matrix */
103  
104 < COLOR   picRGB[24];             /* picture colors */
104 > COLOR   inpRGB[24];             /* measured or scanned input colors */
105 > long    inpflags = 0;           /* flags of which colors were input */
106 > long    gmtflags = 0;           /* flags of out-of-gamut colors */
107  
108 < double  bramp[NMBNEU][2];       /* brightness ramp */
108 > COLOR   bramp[NMBNEU][2];       /* brightness ramp (per primary) */
109   double  solmat[3][3];           /* color mapping matrix */
110  
111 < FILE    *debugfp = NULL;
111 > FILE    *debugfp = NULL;        /* debug output picture */
112   char    *progname;
113  
114   extern char     *malloc();
# Line 78 | Line 121 | char   **argv;
121          int     i;
122  
123          progname = argv[0];
124 <        if (argc > 2 && !strcmp(argv[1], "-d")) {       /* debug output */
125 <                if ((debugfp = fopen(argv[2], "w")) == NULL) {
126 <                        perror(argv[2]);
127 <                        exit(1);
128 <                }
124 >        for (i = 1; i < argc && argv[i][0] == '-'; i++)
125 >                switch (argv[i][1]) {
126 >                case 'd':                               /* debug output */
127 >                        i++;
128 >                        if (badarg(argc-i, argv+i, "s"))
129 >                                goto userr;
130 >                        if ((debugfp = fopen(argv[i], "w")) == NULL) {
131 >                                perror(argv[i]);
132 >                                exit(1);
133 >                        }
134   #ifdef MSDOS
135 <                setmode(fileno(debugfp), O_BINARY);
135 >                        setmode(fileno(debugfp), O_BINARY);
136   #endif
137 <                newheader("RADIANCE", debugfp);
138 <                printargs(argc, argv, debugfp);
139 <                argv += 2;
140 <                argc -= 2;
141 <        }
142 <        if (argc != 3 && argc != 11)
143 <                goto userr;
144 <        if (strcmp(argv[1], "-") && freopen(argv[1], "r", stdin) == NULL) {
137 >                        newheader("RADIANCE", debugfp);         /* start */
138 >                        printargs(argc, argv, debugfp);         /* header */
139 >                        break;
140 >                case 'p':                               /* picture position */
141 >                        if (badarg(argc-i-1, argv+i+1, "iiiiiiii"))
142 >                                goto userr;
143 >                        bounds[0][0] = atoi(argv[++i]);
144 >                        bounds[0][1] = atoi(argv[++i]);
145 >                        bounds[1][0] = atoi(argv[++i]);
146 >                        bounds[1][1] = atoi(argv[++i]);
147 >                        bounds[2][0] = atoi(argv[++i]);
148 >                        bounds[2][1] = atoi(argv[++i]);
149 >                        bounds[3][0] = atoi(argv[++i]);
150 >                        bounds[3][1] = atoi(argv[++i]);
151 >                        scanning = 2;
152 >                        break;
153 >                case 'c':                               /* color input */
154 >                        scanning = 0;
155 >                        break;
156 >                default:
157 >                        goto userr;
158 >                }
159 >                                                        /* open files */
160 >        if (i < argc && freopen(argv[i], "r", stdin) == NULL) {
161                  perror(argv[1]);
162                  exit(1);
163          }
164 <        if (strcmp(argv[2], "-") && freopen(argv[2], "w", stdout) == NULL) {
164 >        if (i+1 < argc && freopen(argv[i+1], "w", stdout) == NULL) {
165                  perror(argv[2]);
166                  exit(1);
167          }
168 +        if (scanning) {                 /* load input picture header */
169   #ifdef MSDOS
170 <        setmode(fileno(stdin), O_BINARY);
170 >                setmode(fileno(stdin), O_BINARY);
171   #endif
172 <        if (checkheader(stdin, COLRFMT, NULL) < 0 ||
173 <                        fgetresolu(&xmax, &ymax, stdin) < 0) {
174 <                fprintf(stderr, "%s: bad input picture\n", progname);
175 <                exit(1);
111 <        }
112 <                                        /* get chart boundaries */
113 <        if (argc == 11) {
114 <                for (i = 0; i < 4; i++) {
115 <                        if (!isint(argv[2*i+3]) | !isint(argv[2*i+4]))
116 <                                goto userr;
117 <                        bounds[i][0] = atoi(argv[2*i+3]);
118 <                        bounds[i][1] = atoi(argv[2*i+4]);
172 >                if (checkheader(stdin, COLRFMT, NULL) < 0 ||
173 >                                fgetresolu(&xmax, &ymax, stdin) < 0) {
174 >                        fprintf(stderr, "%s: bad input picture\n", progname);
175 >                        exit(1);
176                  }
177 <        } else {
177 >        } else {                        /* else set default xmax and ymax */
178 >                xmax = 512;
179 >                ymax = 2*512/3;
180 >        }
181 >        if (scanning != 2) {            /* use default boundaries */
182                  bounds[0][0] = bounds[2][0] = .029*xmax + .5;
183                  bounds[0][1] = bounds[1][1] = .956*ymax + .5;
184                  bounds[1][0] = bounds[3][0] = .971*xmax + .5;
185                  bounds[2][1] = bounds[3][1] = .056*ymax + .5;
186          }
187          init();                         /* initialize */
188 <        getcolors();                    /* get picture colors */
188 >        if (scanning)                   /* get picture colors */
189 >                getpicture();
190 >        else
191 >                getcolors();
192 >        compute();                      /* compute color mapping */
193 >                                        /* print comment */
194 >        printf("{\n\tColor correction file computed by:\n\t\t");
195 >        printargs(argc, argv, stdout);
196 >        printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
197 >                        i+1 < argc ? argv[i+1] : "{this_file}");
198 >        if (!scanning)
199 >                printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
200 >                                i+1 < argc ? argv[i+1] : "{this_file}");
201 >        printf("}\n");
202          putmapping();                   /* put out color mapping */
203 <        putdebug();                     /* put out debug picture */
203 >        if (debugfp != NULL)            /* put out debug picture */
204 >                if (scanning)
205 >                        picdebug();
206 >                else
207 >                        clrdebug();
208          exit(0);
209   userr:
210 <        fprintf(stderr, "Usage: %s [-d dbg.pic] input.pic output.cal [xul yul xur yur xll yll xlr ylr]\n",
210 >        fprintf(stderr,
211 > "Usage: %s [-d dbg.pic][-p xul yul xur yur xll yll xlr ylr] input.pic [output.cal]\n",
212                          progname);
213 +        fprintf(stderr, "   or: %s [-d dbg.pic] -c [xyY.dat [output.cal]]\n",
214 +                        progname);
215          exit(1);
216   }
217  
# Line 138 | Line 219 | userr:
219   init()                          /* initialize */
220   {
221          double  quad[4][2];
222 <        COLOR   ctmp;
142 <        double  d;
143 <        int     i;
222 >        register int    i;
223                                          /* make coordinate transformation */
224          quad[0][0] = bounds[0][0];
225          quad[0][1] = bounds[0][1];
# Line 156 | Line 235 | init()                         /* initialize */
235                  exit(1);
236          }
237                                          /* map MacBeth colors to RGB space */
238 <        for (i = 0; i < 24; i++) {
239 <                d = mbxyY[i][2] / mbxyY[i][1];
161 <                ctmp[0] = mbxyY[i][0] * d;
162 <                ctmp[1] = mbxyY[i][2];
163 <                ctmp[2] = (1. - mbxyY[i][0] - mbxyY[i][1]) * d;
164 <                cie_rgb(mbRGB[i], ctmp);
165 <        }
238 >        for (i = 0; i < 24; i++)
239 >                xyY2RGB(mbRGB[i], mbxyY[i]);
240   }
241  
242  
243   int
244 < chartndx(x, y)                          /* find color number for position */
244 > chartndx(x, y, np)                      /* find color number for position */
245   int     x, y;
246 + int     *np;
247   {
248          double  ipos[3], cpos[3];
249          int     ix, iy;
# Line 181 | Line 256 | int    x, y;
256          cpos[0] /= cpos[2];
257          cpos[1] /= cpos[2];
258          if (cpos[0] < 0. || cpos[0] >= 6. || cpos[1] < 0. || cpos[1] >= 4.)
259 <                return(-1);
259 >                return(RG_BORD);
260          ix = cpos[0];
261          iy = cpos[1];
262          fx = cpos[0] - ix;
263          fy = cpos[1] - iy;
264 <        if (fx < .35 || fx >= .65 || fy < .35 || fy >= .65)
265 <                return(-1);
266 <        return(iy*6 + ix);
264 >        *np = iy*6 + ix;
265 >        if (fx >= 0.35 && fx < 0.65 && fy >= 0.35 && fy < 0.65)
266 >                return(RG_CENT);
267 >        if (fx < 0.05 || fx >= 0.95 || fy < 0.05 || fy >= 0.95)
268 >                return(RG_BORD);
269 >        if (fx >= 0.5)                  /* right side is corrected */
270 >                return(RG_CORR);
271 >        return(RG_ORIG);                /* left side is original */
272   }
273  
274  
275 < getcolors()                             /* load in picture colors */
275 > getpicture()                            /* load in picture colors */
276   {
277          COLR    *scanln;
278          COLOR   pval;
279          int     ccount[24];
280          double  d;
281 <        int     y;
282 <        register int    x, i;
281 >        int     y, i;
282 >        register int    x;
283  
284          scanln = (COLR *)malloc(xmax*sizeof(COLR));
285          if (scanln == NULL) {
# Line 207 | Line 287 | getcolors()                            /* load in picture colors */
287                  exit(1);
288          }
289          for (i = 0; i < 24; i++) {
290 <                setcolor(picRGB[i], 0., 0., 0.);
290 >                setcolor(inpRGB[i], 0., 0., 0.);
291                  ccount[i] = 0;
292          }
293          for (y = ymax-1; y >= 0; y--) {
# Line 216 | Line 296 | getcolors()                            /* load in picture colors */
296                                          progname);
297                          exit(1);
298                  }
299 <                for (x = 0; x < xmax; x++) {
300 <                        i = chartndx(x, y);
221 <                        if (i >= 0) {
299 >                for (x = 0; x < xmax; x++)
300 >                        if (chartndx(x, y, &i) == RG_CENT) {
301                                  colr_color(pval, scanln[x]);
302 <                                addcolor(picRGB[i], pval);
302 >                                addcolor(inpRGB[i], pval);
303                                  ccount[i]++;
304                          }
305 +        }
306 +        for (i = 0; i < 24; i++) {              /* compute averages */
307 +                if (ccount[i] == 0)
308 +                        continue;
309 +                d = 1./ccount[i];
310 +                scalecolor(inpRGB[i], d);
311 +                inpflags |= 1L<<i;
312 +        }
313 +        free((char *)scanln);
314 + }
315 +
316 +
317 + getcolors()                     /* get xyY colors from standard input */
318 + {
319 +        int     gotwhite = 0;
320 +        COLOR   whiteclr;
321 +        int     n;
322 +        float   xyYin[3];
323 +
324 +        while (fgetval(stdin, 'i', &n) == 1) {          /* read colors */
325 +                if (n < 0 | n > 24 ||
326 +                                fgetval(stdin, 'f', &xyYin[0]) != 1 ||
327 +                                fgetval(stdin, 'f', &xyYin[1]) != 1 ||
328 +                                fgetval(stdin, 'f', &xyYin[2]) != 1 ||
329 +                                xyYin[0] < 0. | xyYin[1] < 0. ||
330 +                                xyYin[0] + xyYin[1] > 1.) {
331 +                        fprintf(stderr, "%s: bad color input data\n",
332 +                                        progname);
333 +                        exit(1);
334                  }
335 +                if (n == 0) {                           /* calibration white */
336 +                        xyY2RGB(whiteclr, xyYin);
337 +                        gotwhite++;
338 +                } else {                                /* standard color */
339 +                        n--;
340 +                        xyY2RGB(inpRGB[n], xyYin);
341 +                        inpflags |= 1L<<n;
342 +                }
343          }
344 <        for (i = 0; i < 24; i++) {
345 <                if (ccount[i] == 0) {
346 <                        fprintf(stderr, "%s: bad chart boundaries\n",
344 >                                        /* normalize colors */
345 >        if (!gotwhite) {
346 >                if (!(inpflags & 1L<<White)) {
347 >                        fprintf(stderr, "%s: missing input for White\n",
348                                          progname);
349                          exit(1);
350                  }
351 <                d = 1.0/ccount[i];
352 <                scalecolor(picRGB[i], d);
351 >                setcolor(whiteclr,
352 >                        colval(inpRGB[White],RED)/colval(mbRGB[White],RED),
353 >                        colval(inpRGB[White],GRN)/colval(mbRGB[White],GRN),
354 >                        colval(inpRGB[White],BLU)/colval(mbRGB[White],BLU));
355          }
356 <        free((char *)scanln);
356 >        for (n = 0; n < 24; n++)
357 >                if (inpflags & 1L<<n)
358 >                        setcolor(inpRGB[n],
359 >                                colval(inpRGB[n],RED)/colval(whiteclr,RED),
360 >                                colval(inpRGB[n],GRN)/colval(whiteclr,GRN),
361 >                                colval(inpRGB[n],BLU)/colval(whiteclr,BLU));
362   }
363  
364  
365 < double
366 < bresp(x)                /* piecewise linear interpolation of brightness */
243 < double  x;
365 > bresp(y, x)             /* piecewise linear interpolation of primaries */
366 > COLOR   y, x;
367   {
368 <        register int    n = NMBNEU;
368 >        register int    i, n;
369  
370 <        while (n > 0 && x < bramp[--n][0])
371 <                ;
372 <        return( ((bramp[n+1][0] - x)*bramp[n][1] +
373 <                                (x - bramp[n][0])*bramp[n+1][1]) /
374 <                        (bramp[n+1][0] - bramp[n][0]) );
370 >        for (i = 0; i < 3; i++) {
371 >                for (n = 0; n < NMBNEU-2; n++)
372 >                        if (colval(x,i) < colval(bramp[n+1][0],i))
373 >                                break;
374 >                colval(y,i) = ((colval(bramp[n+1][0],i) - colval(x,i)) *
375 >                                                colval(bramp[n][1],i) +
376 >                                (colval(x,i) - colval(bramp[n][0],i)) *
377 >                                                colval(bramp[n+1][1],i)) /
378 >                        (colval(bramp[n+1][0],i) - colval(bramp[n][0],i));
379 >        }
380   }
381  
382  
383 < putmapping()                    /* compute and print mapping for pcomb -f */
383 > compute()                       /* compute color mapping */
384   {
385 <        float   clrin[NMBMOD][3], clrout[NMBMOD][3];
386 <        register int    i, j;
385 >        COLOR   clrin[24], clrout[24];
386 >        long    cflags;
387 >        COLOR   ctmp;
388 >        register int    i, j, n;
389 >                                        /* did we get what we need? */
390 >        if ((inpflags & REQFLGS) != REQFLGS) {
391 >                fprintf(stderr, "%s: missing required input colors\n",
392 >                                progname);
393 >                exit(1);
394 >        }
395                                          /* compute piecewise luminance curve */
396          for (i = 0; i < NMBNEU; i++) {
397 <                bramp[i][0] = bright(picRGB[mbneu[i]]);
398 <                bramp[i][1] = bright(mbRGB[mbneu[i]]);
397 >                copycolor(bramp[i][0], inpRGB[mbneu[i]]);
398 >                copycolor(bramp[i][1], mbRGB[mbneu[i]]);
399          }
400 <                                        /* compute color matrix */
401 <        for (i = 0; i < NMBMOD; i++)
402 <                for (j = 0; j < 3; j++) {
403 <                        clrin[i][j] = bresp(picRGB[mbmod[i]][j]);
404 <                        clrout[i][j] = mbRGB[mbmod[i]][j];
405 <                }
406 <        compsoln(clrin, clrout, NMBMOD);
400 >                                        /* compute color mapping */
401 >        do {
402 >                cflags = inpflags & ~gmtflags;
403 >                n = 0;                          /* compute transform matrix */
404 >                for (i = 0; i < 24; i++)
405 >                        if (cflags & 1L<<i) {
406 >                                bresp(clrin[n], inpRGB[i]);
407 >                                copycolor(clrout[n], mbRGB[i]);
408 >                                n++;
409 >                        }
410 >                compsoln(clrin, clrout, n);
411 >                                                /* check out-of-gamut colors */
412 >                for (i = 0; i < 24; i++)
413 >                        if (cflags & 1L<<i) {
414 >                                cvtcolor(ctmp, mbRGB[i]);
415 >                                for (j = 0; j < 3; j++)
416 >                                        if (colval(ctmp,j) <= 1e-6 ||
417 >                                                colval(ctmp,j) >= 1.-1e-6) {
418 >                                                gmtflags |= 1L<<i;
419 >                                                break;
420 >                                        }
421 >                        }
422 >        } while (cflags & gmtflags);
423 >        if (gmtflags & MODFLGS)
424 >                fprintf(stderr,
425 >                "%s: warning - some moderate colors are out of gamut\n",
426 >                                progname);
427 > }
428 >
429 >
430 > putmapping()                    /* put out color mapping for pcomb -f */
431 > {
432 >        static char     cchar[3] = {'r', 'g', 'b'};
433 >        register int    i, j;
434                                          /* print brightness mapping */
435 <        printf("xval(i) : select(i, 0");
436 <        for (i = 0; i < NMBNEU; i++)
437 <                printf(", %g", bramp[i][0]);
438 <        printf(");\n");
439 <        printf("yval(i) : select(i, 0");
440 <        for (i = 0; i < NMBNEU; i++)
441 <                printf(", %g", bramp[i][1]);
442 <        printf(");\n");
443 <        printf("ifind(x,f,n) : if(1.5-n, 1, if(x-f(n), n, ifind(x,f,n-1)));\n");
444 <        printf("binterp(i,x) : ((xval(i+1)-x)*yval(i) + (x-xval(i))*yval(i+1))/\n");
445 <        printf("\t\t(xval(i+1) - xval(i));\n");
446 <        printf("bresp(x) : binterp(ifind(x,xval,xval(0)-1), x);\n");
447 <        printf("nred = bresp(ri(1));\n");
448 <        printf("ngrn = bresp(gi(1));\n");
449 <        printf("nblu = bresp(bi(1));\n");
435 >        for (j = 0; j < 3; j++) {
436 >                printf("%cxa(i) : select(i", cchar[j]);
437 >                for (i = 0; i < NMBNEU; i++)
438 >                        printf(",%g", colval(bramp[i][0],j));
439 >                printf(");\n");
440 >                printf("%cya(i) : select(i", cchar[j]);
441 >                for (i = 0; i < NMBNEU; i++)
442 >                        printf(",%g", colval(bramp[i][1],j));
443 >                printf(");\n");
444 >                printf("%cfi(n) = if(n-%g, %d, if(%cxa(n+1)-%c, n, %cfi(n+1)));\n",
445 >                                cchar[j], NMBNEU-1.5, NMBNEU-1, cchar[j],
446 >                                cchar[j], cchar[j]);
447 >                printf("%cndx = %cfi(1);\n", cchar[j], cchar[j]);
448 >                printf("%c%c = ((%cxa(%cndx+1)-%c)*%cya(%cndx) + ",
449 >                                cchar[j], scanning?'n':'o', cchar[j],
450 >                                cchar[j], cchar[j], cchar[j], cchar[j]);
451 >                printf("(%c-%cxa(%cndx))*%cya(%cndx+1)) /\n",
452 >                                cchar[j], cchar[j], cchar[j],
453 >                                cchar[j], cchar[j]);
454 >                printf("\t\t(%cxa(%cndx+1) - %cxa(%cndx)) ;\n",
455 >                                cchar[j], cchar[j], cchar[j], cchar[j]);
456 >        }
457                                          /* print color mapping */
458 <        printf("ro = %g*nred + %g*ngrn + %g*nblu\n",
459 <                        solmat[0][0], solmat[1][0], solmat[2][0]);
460 <        printf("go = %g*nred + %g*ngrn + %g*nblu\n",
461 <                        solmat[0][1], solmat[1][1], solmat[2][1]);
462 <        printf("bo = %g*nred + %g*ngrn + %g*nblu\n",
463 <                        solmat[0][2], solmat[1][2], solmat[2][2]);
458 >        if (scanning) {
459 >                printf("r = ri(1); g = gi(1); b = bi(1);\n");
460 >                printf("ro = %g*rn + %g*gn + %g*bn ;\n",
461 >                                solmat[0][0], solmat[0][1], solmat[0][2]);
462 >                printf("go = %g*rn + %g*gn + %g*bn ;\n",
463 >                                solmat[1][0], solmat[1][1], solmat[1][2]);
464 >                printf("bo = %g*rn + %g*gn + %g*bn ;\n",
465 >                                solmat[2][0], solmat[2][1], solmat[2][2]);
466 >        } else {
467 >                printf("r1 = ri(1); g1 = gi(1); b1 = bi(1);\n");
468 >                printf("r = %g*r1 + %g*g1 + %g*b1 ;\n",
469 >                                solmat[0][0], solmat[0][1], solmat[0][2]);
470 >                printf("g = %g*r1 + %g*g1 + %g*b1 ;\n",
471 >                                solmat[1][0], solmat[1][1], solmat[1][2]);
472 >                printf("b = %g*r1 + %g*g1 + %g*b1 ;\n",
473 >                                solmat[2][0], solmat[2][1], solmat[2][2]);
474 >        }
475   }
476  
477  
478 < compsoln(cin, cout, n)          /* solve 3x3 system */
479 < float   cin[][3], cout[][3];
478 > compsoln(cin, cout, n)          /* solve 3xN system using least-squares */
479 > COLOR   cin[], cout[];
480   int     n;
481   {
482          extern double   mx3d_adjoint(), fabs();
483          double  mat[3][3], invmat[3][3];
484          double  det;
485          double  colv[3], rowv[3];
486 <        register int    i, j;
486 >        register int    i, j, k;
487  
488 <        if (n != 3) {
489 <                fprintf(stderr, "%s: inconsistent code!\n", progname);
488 >        if (n < 3) {
489 >                fprintf(stderr, "%s: too few colors to match!\n", progname);
490                  exit(1);
491          }
492 <        for (i = 0; i < 3; i++)
493 <                for (j = 0; j < 3; j++)
494 <                        mat[i][j] = cin[j][i];
492 >        if (n == 3)
493 >                for (i = 0; i < 3; i++)
494 >                        for (j = 0; j < 3; j++)
495 >                                mat[i][j] = colval(cin[j],i);
496 >        else {                          /* compute A^t A */
497 >                for (i = 0; i < 3; i++)
498 >                        for (j = i; j < 3; j++) {
499 >                                mat[i][j] = 0.;
500 >                                for (k = 0; k < n; k++)
501 >                                        mat[i][j] += colval(cin[k],i) *
502 >                                                        colval(cin[k],j);
503 >                        }
504 >                for (i = 1; i < 3; i++)         /* using symmetry */
505 >                        for (j = 0; j < i; j++)
506 >                                mat[i][j] = mat[j][i];
507 >        }
508          det = mx3d_adjoint(mat, invmat);
509          if (fabs(det) < 1e-4) {
510                  fprintf(stderr, "%s: cannot compute color mapping\n",
# Line 324 | Line 518 | int    n;
518                  for (j = 0; j < 3; j++)
519                          invmat[i][j] /= det;
520          for (i = 0; i < 3; i++) {
521 +                if (n == 3)
522 +                        for (j = 0; j < 3; j++)
523 +                                colv[j] = colval(cout[j],i);
524 +                else
525 +                        for (j = 0; j < 3; j++) {
526 +                                colv[j] = 0.;
527 +                                for (k = 0; k < n; k++)
528 +                                        colv[j] += colval(cout[k],i) *
529 +                                                        colval(cin[k],j);
530 +                        }
531 +                mx3d_transform(colv, invmat, rowv);
532                  for (j = 0; j < 3; j++)
533 <                        rowv[j] = cout[j][i];
329 <                mx3d_transform(rowv, invmat, colv);
330 <                for (j = 0; j < 3; j++)
331 <                        solmat[j][i] = colv[j];
533 >                        solmat[i][j] = rowv[j];
534          }
535   }
536  
# Line 336 | Line 538 | int    n;
538   cvtcolor(cout, cin)             /* convert color according to our mapping */
539   COLOR   cout, cin;
540   {
541 +        COLOR   ctmp;
542 +
543 +        if (scanning) {
544 +                bresp(ctmp, cin);
545 +                cresp(cout, ctmp);
546 +        } else {
547 +                cresp(ctmp, cin);
548 +                bresp(cout, ctmp);
549 +        }
550 +        if (colval(cout,RED) < 0.)
551 +                colval(cout,RED) = 0.;
552 +        if (colval(cout,GRN) < 0.)
553 +                colval(cout,GRN) = 0.;
554 +        if (colval(cout,BLU) < 0.)
555 +                colval(cout,BLU) = 0.;
556 + }
557 +
558 +
559 + cresp(cout, cin)                /* transform color according to matrix */
560 + COLOR   cout, cin;
561 + {
562          double  r, g, b;
340        double  r1, g1, b1;
563  
564 <        r = bresp(colval(cin,RED));
565 <        g = bresp(colval(cin,GRN));
566 <        b = bresp(colval(cin,BLU));
567 <        r1 = r*solmat[0][0] + g*solmat[1][0] + b*solmat[2][0];
568 <        if (r1 < 0) r1 = 0;
569 <        g1 = r*solmat[0][1] + g*solmat[1][1] + b*solmat[2][1];
570 <        if (g1 < 0) g1 = 0;
349 <        b1 = r*solmat[0][2] + g*solmat[1][2] + b*solmat[2][2];
350 <        if (b1 < 0) b1 = 0;
351 <        setcolor(cout, r1, g1, b1);
564 >        r = colval(cin,0)*solmat[0][0] + colval(cin,1)*solmat[0][1]
565 >                        + colval(cin,2)*solmat[0][2];
566 >        g = colval(cin,0)*solmat[1][0] + colval(cin,1)*solmat[1][1]
567 >                        + colval(cin,2)*solmat[1][2];
568 >        b = colval(cin,0)*solmat[2][0] + colval(cin,1)*solmat[2][1]
569 >                        + colval(cin,2)*solmat[2][2];
570 >        setcolor(cout, r, g, b);
571   }
572  
573  
574 < putdebug()                      /* put out debugging picture */
574 > xyY2RGB(rgbout, xyYin)          /* convert xyY to RGB */
575 > COLOR   rgbout;
576 > register float  xyYin[3];
577   {
578 +        COLOR   ctmp;
579 +        double  d;
580 +
581 +        d = xyYin[2] / xyYin[1];
582 +        ctmp[0] = xyYin[0] * d;
583 +        ctmp[1] = xyYin[2];
584 +        ctmp[2] = (1. - xyYin[0] - xyYin[1]) * d;
585 +        cie_rgb(rgbout, ctmp);
586 +                                /* allow negative values */
587 +        colortrans(rgbout, xyz2rgbmat, ctmp, 0);
588 + }
589 +
590 +
591 + picdebug()                      /* put out debugging picture */
592 + {
593 +        static COLOR    blkcol = BLKCOLOR;
594          COLOR   *scan;
595 <        int     y;
596 <        register int    x, i;
595 >        int     y, i;
596 >        register int    x, rg;
597  
361        if (debugfp == NULL)
362                return;
598          if (fseek(stdin, 0L, 0) == EOF) {
599                  fprintf(stderr, "%s: cannot seek on input picture\n", progname);
600                  exit(1);
# Line 373 | Line 608 | putdebug()                     /* put out debugging picture */
608                  exit(1);
609          }
610                                                  /* finish debug header */
611 +        fputformat(COLRFMT, debugfp);
612          putc('\n', debugfp);
613          fprtresolu(xmax, ymax, debugfp);
614 +                                                /* write debug picture */
615          for (y = ymax-1; y >= 0; y--) {
616                  if (freadscan(scan, xmax, stdin) < 0) {
617                          fprintf(stderr, "%s: error rereading input picture\n",
# Line 382 | Line 619 | putdebug()                     /* put out debugging picture */
619                          exit(1);
620                  }
621                  for (x = 0; x < xmax; x++) {
622 <                        i = chartndx(x, y);
623 <                        if (i < 0)
622 >                        rg = chartndx(x, y, &i);
623 >                        if (rg == RG_CENT) {
624 >                                if (!(1L<<i & gmtflags) || (x+y)&07)
625 >                                        copycolor(scan[x], mbRGB[i]);
626 >                                else
627 >                                        copycolor(scan[x], blkcol);
628 >                        } else if (rg == RG_CORR)
629                                  cvtcolor(scan[x], scan[x]);
630 <                        else
631 <                                copycolor(scan[x], mbRGB[i]);
630 >                        else if (rg != RG_ORIG)
631 >                                copycolor(scan[x], blkcol);
632                  }
633                  if (fwritescan(scan, xmax, debugfp) < 0) {
634                          fprintf(stderr, "%s: error writing debugging picture\n",
# Line 394 | Line 636 | putdebug()                     /* put out debugging picture */
636                          exit(1);
637                  }
638          }
639 +                                                /* clean up */
640 +        fclose(debugfp);
641 +        free((char *)scan);
642 + }
643 +
644 +
645 + clrdebug()                      /* put out debug picture from color input */
646 + {
647 +        static COLR     blkclr = BLKCOLR;
648 +        COLR    mbclr[24], cvclr[24], orclr[24];
649 +        COLR    *scan;
650 +        COLOR   ctmp;
651 +        int     y, i;
652 +        register int    x, rg;
653 +                                                /* convert colors */
654 +        for (i = 0; i < 24; i++) {
655 +                setcolr(mbclr[i], colval(mbRGB[i],RED),
656 +                                colval(mbRGB[i],GRN), colval(mbRGB[i],BLU));
657 +                if (inpflags & 1L<<i) {
658 +                        setcolr(orclr[i], colval(inpRGB[i],RED),
659 +                                        colval(inpRGB[i],GRN),
660 +                                        colval(inpRGB[i],BLU));
661 +                        cvtcolor(ctmp, inpRGB[i]);
662 +                        setcolr(cvclr[i], colval(ctmp,RED),
663 +                                        colval(ctmp,GRN), colval(ctmp,BLU));
664 +                }
665 +        }
666 +                                                /* allocate scanline */
667 +        scan = (COLR *)malloc(xmax*sizeof(COLR));
668 +        if (scan == NULL) {
669 +                perror(progname);
670 +                exit(1);
671 +        }
672 +                                                /* finish debug header */
673 +        fputformat(COLRFMT, debugfp);
674 +        putc('\n', debugfp);
675 +        fprtresolu(xmax, ymax, debugfp);
676 +                                                /* write debug picture */
677 +        for (y = ymax-1; y >= 0; y--) {
678 +                for (x = 0; x < xmax; x++) {
679 +                        rg = chartndx(x, y, &i);
680 +                        if (rg == RG_CENT) {
681 +                                if (!(1L<<i & gmtflags) || (x+y)&07)
682 +                                        copycolr(scan[x], mbclr[i]);
683 +                                else
684 +                                        copycolr(scan[x], blkclr);
685 +                        } else if (rg == RG_BORD || !(1L<<i & inpflags))
686 +                                copycolr(scan[x], blkclr);
687 +                        else if (rg == RG_ORIG)
688 +                                copycolr(scan[x], orclr[i]);
689 +                        else /* rg == RG_CORR */
690 +                                copycolr(scan[x], cvclr[i]);
691 +                }
692 +                if (fwritecolrs(scan, xmax, debugfp) < 0) {
693 +                        fprintf(stderr, "%s: error writing debugging picture\n",
694 +                                        progname);
695 +                        exit(1);
696 +                }
697 +        }
698 +                                                /* clean up */
699 +        fclose(debugfp);
700          free((char *)scan);
701   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines