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.3 by greg, Wed Oct 11 18:52:00 1995 UTC vs.
Revision 2.28 by greg, Sat Dec 28 18:05:14 2019 UTC

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

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines