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.13 by greg, Fri Jan 31 15:56:17 1997 UTC vs.
Revision 2.23 by schorsch, Sun Mar 28 20:33:13 2004 UTC

# Line 1 | Line 1
1 – /* Copyright (c) 1997 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
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
# Line 95 | Line 100 | short  mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutra
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 */
# Line 109 | Line 120 | COLOR  bramp[NMBNEU][2];       /* brightness ramp (per primar
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 < extern char     *malloc();
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 < main(argc, argv)
147 < int     argc;
148 < char    **argv;
146 > int
147 > main(
148 >        int     argc,
149 >        char    **argv
150 > )
151   {
152          int     i;
153  
# Line 132 | Line 162 | char   **argv;
162                                  perror(argv[i]);
163                                  exit(1);
164                          }
165 < #ifdef MSDOS
136 <                        setmode(fileno(debugfp), O_BINARY);
137 < #endif
165 >                        SET_FILE_BINARY(debugfp);
166                          newheader("RADIANCE", debugfp);         /* start */
167                          printargs(argc, argv, debugfp);         /* header */
168                          break;
# Line 151 | Line 179 | char   **argv;
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;
# Line 167 | Line 207 | char   **argv;
207                  exit(1);
208          }
209          if (scanning) {                 /* load input picture header */
210 < #ifdef MSDOS
171 <                setmode(fileno(stdin), O_BINARY);
172 < #endif
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;
# Line 191 | Line 235 | char   **argv;
235          else
236                  getcolors();
237          compute();                      /* compute color mapping */
238 <                                        /* print comment */
239 <        printf("{\n\tColor correction file computed by:\n\t\t");
240 <        printargs(argc, argv, stdout);
241 <        printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
242 <                        i+1 < argc ? argv[i+1] : "{this_file}");
243 <        if (!scanning)
200 <                printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
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.pic > corrected.pic\n",
244                                  i+1 < argc ? argv[i+1] : "{this_file}");
245 <        printf("}\n");
246 <        putmapping();                   /* put out color mapping */
247 <        if (debugfp != NULL)            /* put out debug picture */
245 >                printf("#\t   Or: pcond [options] -m %s orig.pic > output.pic\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.pic > corrected.pic\n",
259 >                                i+1 < argc ? argv[i+1] : "{this_file}");
260 >                if (!scanning)
261 >                        printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\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.pic][-p xul yul xur yur xll yll xlr ylr] input.pic [output.cal]\n",
275 > "Usage: %s [-d dbg.pic][-P | -p xul yul xur yur xll yll xlr ylr][-i irrad][-m] input.pic [output.{cal|cwp}]\n",
276                          progname);
277 <        fprintf(stderr, "   or: %s [-d dbg.pic] -c [xyY.dat [output.cal]]\n",
277 >        fprintf(stderr, "   or: %s [-d dbg.pic][-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 < init()                          /* initialize */
284 > static void
285 > init(void)                              /* initialize */
286   {
287          double  quad[4][2];
288          register int    i;
# Line 236 | Line 301 | init()                         /* initialize */
301                  exit(1);
302          }
303                                          /* map MacBeth colors to RGB space */
304 <        for (i = 0; i < 24; i++)
304 >        for (i = 0; i < 24; i++) {
305                  xyY2RGB(mbRGB[i], mbxyY[i]);
306 +                scalecolor(mbRGB[i], irrad);
307 +        }
308   }
309  
310  
311 < int
312 < chartndx(x, y, np)                      /* find color number for position */
313 < int     x, y;
314 < int     *np;
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;
# Line 273 | Line 342 | int    *np;
342   }
343  
344  
345 < getpicture()                            /* load in picture colors */
345 > static void
346 > getpicture(void)                                /* load in picture colors */
347   {
348          COLR    *scanln;
349          COLOR   pval;
# Line 311 | Line 381 | getpicture()                           /* load in picture colors */
381                  scalecolor(inpRGB[i], d);
382                  inpflags |= 1L<<i;
383          }
384 <        free((char *)scanln);
384 >        free((void *)scanln);
385   }
386  
387  
388 < getcolors()                     /* get xyY colors from standard input */
388 > static void
389 > getcolors(void)                 /* get xyY colors from standard input */
390   {
391          int     gotwhite = 0;
392          COLOR   whiteclr;
# Line 323 | Line 394 | getcolors()                    /* get xyY colors from standard input */
394          float   xyYin[3];
395  
396          while (fgetval(stdin, 'i', &n) == 1) {          /* read colors */
397 <                if (n < 0 | n > 24 ||
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. ||
401 >                                (xyYin[0] < 0.) | (xyYin[1] < 0.) ||
402                                  xyYin[0] + xyYin[1] > 1.) {
403                          fprintf(stderr, "%s: bad color input data\n",
404                                          progname);
# Line 363 | Line 434 | getcolors()                    /* get xyY colors from standard input */
434   }
435  
436  
437 < bresp(y, x)             /* piecewise linear interpolation of primaries */
438 < COLOR   y, x;
437 > static void
438 > bresp(          /* piecewise linear interpolation of primaries */
439 >        COLOR   y,
440 >        COLOR   x
441 > )
442   {
443          register int    i, n;
444  
# Line 381 | Line 455 | COLOR  y, x;
455   }
456  
457  
458 < compute()                       /* compute color mapping */
458 > static void
459 > compute(void)                   /* compute color mapping */
460   {
461          COLOR   clrin[24], clrout[24];
462          long    cflags;
# Line 402 | Line 477 | compute()                      /* compute color mapping */
477          if (scanning) {
478                  copycolor(colmin, cblack);
479                  copycolor(colmax, cwhite);
480 +                scalecolor(colmax, irrad);
481          } else
482                  for (i = 0; i < 3; i++) {
483                          colval(colmin,i) = colval(bramp[0][0],i) -
# Line 426 | Line 502 | compute()                      /* compute color mapping */
502                                  n++;
503                          }
504                  compsoln(clrin, clrout, n);
505 <                                                /* check out-of-gamut colors */
506 <                for (i = 0; i < 24; i++)
507 <                        if (cflags & 1L<<i && cvtcolor(ctmp, mbRGB[i]))
508 <                                gmtflags |= 1L<<i;
505 >                if (irrad > 0.99 && irrad < 1.01)       /* check gamut */
506 >                        for (i = 0; i < 24; i++)
507 >                                if (cflags & 1L<<i && cvtcolor(ctmp, mbRGB[i]))
508 >                                        gmtflags |= 1L<<i;
509          } while (cflags & gmtflags);
510          if (gmtflags & MODFLGS)
511                  fprintf(stderr,
# Line 438 | Line 514 | compute()                      /* compute color mapping */
514   }
515  
516  
517 < putmapping()                    /* put out color mapping for pcomb -f */
517 > static void
518 > putmapping(void)                        /* put out color mapping */
519   {
520          static char     cchar[3] = {'r', 'g', 'b'};
521          register int    i, j;
# Line 486 | Line 563 | putmapping()                   /* put out color mapping for pcomb -f *
563   }
564  
565  
566 < compsoln(cin, cout, n)          /* solve 3xN system using least-squares */
567 < COLOR   cin[], cout[];
568 < int     n;
566 > static void
567 > compsoln(               /* solve 3xN system using least-squares */
568 >        COLOR   cin[],
569 >        COLOR   cout[],
570 >        int     n
571 > )
572   {
573          extern double   mx3d_adjoint(), fabs();
574          double  mat[3][3], invmat[3][3];
# Line 546 | Line 626 | int    n;
626   }
627  
628  
629 < int
630 < cvtcolor(cout, cin)             /* convert color according to our mapping */
551 < COLOR   cout, cin;
629 > static void
630 > cwarp(void)                             /* compute color warp map */
631   {
632 +        register int    i;
633 +
634 +        if ((wcor = new3dw(W3EXACT)) == NULL)
635 +                goto memerr;
636 +        for (i = 0; i < 24; i++)
637 +                if (!add3dpt(wcor, inpRGB[i], mbRGB[i]))
638 +                        goto memerr;
639 +        return;
640 + memerr:
641 +        perror(progname);
642 +        exit(1);
643 + }
644 +
645 +
646 + static int
647 + cvtcolor(               /* convert color according to our mapping */
648 +        COLOR   cout,
649 +        COLOR   cin
650 + )
651 + {
652          COLOR   ctmp;
653          int     clipped;
654  
655 <        if (scanning) {
655 >        if (wcor != NULL) {
656 >                clipped = warp3d(cout, cin, wcor);
657 >                clipped |= clipgamut(cout,bright(cout),CGAMUT,colmin,colmax);
658 >        } else if (scanning) {
659                  bresp(ctmp, cin);
660                  clipped = cresp(cout, ctmp);
661          } else {
# Line 564 | Line 666 | COLOR  cout, cin;
666   }
667  
668  
669 < int
670 < cresp(cout, cin)                /* transform color according to matrix */
671 < COLOR   cout, cin;
669 > static int
670 > cresp(          /* transform color according to matrix */
671 >        COLOR   cout,
672 >        COLOR   cin
673 > )
674   {
675          colortrans(cout, solmat, cin);
676          return(clipgamut(cout, bright(cout), CGAMUT, colmin, colmax));
677   }
678  
679  
680 < xyY2RGB(rgbout, xyYin)          /* convert xyY to RGB */
681 < COLOR   rgbout;
682 < register float  xyYin[3];
680 > static void
681 > xyY2RGB(                /* convert xyY to RGB */
682 >        COLOR   rgbout,
683 >        register float  xyYin[3]
684 > )
685   {
686          COLOR   ctmp;
687          double  d;
# Line 589 | Line 695 | register float xyYin[3];
695   }
696  
697  
698 < picdebug()                      /* put out debugging picture */
698 > static void
699 > picdebug(void)                  /* put out debugging picture */
700   {
701          static COLOR    blkcol = BLKCOLOR;
702          COLOR   *scan;
# Line 625 | Line 732 | picdebug()                     /* put out debugging picture */
732                                  if (!(1L<<i & gmtflags) || (x+y)&07) {
733                                          copycolor(scan[x], mbRGB[i]);
734                                          clipgamut(scan[x], bright(scan[x]),
735 <                                                CGAMUT, cblack, cwhite);
735 >                                                CGAMUT, colmin, colmax);
736                                  } else
737                                          copycolor(scan[x], blkcol);
738                          } else if (rg == RG_CORR)
# Line 641 | Line 748 | picdebug()                     /* put out debugging picture */
748          }
749                                                  /* clean up */
750          fclose(debugfp);
751 <        free((char *)scan);
751 >        free((void *)scan);
752   }
753  
754  
755 < clrdebug()                      /* put out debug picture from color input */
755 > static void
756 > clrdebug(void)                  /* put out debug picture from color input */
757   {
758          static COLR     blkclr = BLKCOLR;
759          COLR    mbclr[24], cvclr[24], orclr[24];
# Line 664 | Line 772 | clrdebug()                     /* put out debug picture from color input
772                          clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
773                          setcolr(orclr[i], colval(ctmp,RED),
774                                          colval(ctmp,GRN), colval(ctmp,BLU));
775 <                        bresp(ctmp, inpRGB[i]);
776 <                        colortrans(ct2, solmat, ctmp);
777 <                        clipgamut(ct2, bright(ct2), CGAMUT, cblack, cwhite);
778 <                        setcolr(cvclr[i], colval(ct2,RED),
779 <                                        colval(ct2,GRN), colval(ct2,BLU));
775 >                        if (rawmap)
776 >                                copycolr(cvclr[i], mbclr[i]);
777 >                        else {
778 >                                bresp(ctmp, inpRGB[i]);
779 >                                colortrans(ct2, solmat, ctmp);
780 >                                clipgamut(ct2, bright(ct2), CGAMUT,
781 >                                                cblack, cwhite);
782 >                                setcolr(cvclr[i], colval(ct2,RED),
783 >                                                colval(ct2,GRN),
784 >                                                colval(ct2,BLU));
785 >                        }
786                  }
787          }
788                                                  /* allocate scanline */
# Line 705 | Line 819 | clrdebug()                     /* put out debug picture from color input
819          }
820                                                  /* clean up */
821          fclose(debugfp);
822 <        free((char *)scan);
822 >        free((void *)scan);
823 > }
824 >
825 >
826 > static void
827 > getpos(         /* get boundary position */
828 >        char    *name,
829 >        int     bnds[2],
830 >        FILE    *fp
831 > )
832 > {
833 >        char    buf[64];
834 >
835 >        fprintf(stderr, "\tSelect corner: %s\n", name);
836 >        if (fgets(buf, sizeof(buf), fp) == NULL ||
837 >                        sscanf(buf, "%d %d", &bnds[0], &bnds[1]) != 2) {
838 >                fprintf(stderr, "%s: read error from display process\n",
839 >                                progname);
840 >                exit(1);
841 >        }
842 > }
843 >
844 >
845 > static void
846 > pickchartpos(           /* display picture and pick chart location */
847 >        char    *pfn
848 > )
849 > {
850 >        char    combuf[PATH_MAX];
851 >        FILE    *pfp;
852 >
853 >        sprintf(combuf, DISPCOM, pfn);
854 >        if ((pfp = popen(combuf, "r")) == NULL) {
855 >                perror(combuf);
856 >                exit(1);
857 >        }
858 >        fputs("Use middle mouse button to select chart corners:\n", stderr);
859 >        getpos("upper left (dark skin)", bounds[0], pfp);
860 >        getpos("upper right (bluish green)", bounds[1], pfp);
861 >        getpos("lower left (white)", bounds[2], pfp);
862 >        getpos("lower right (black)", bounds[3], pfp);
863 >        fputs("Got it -- quit display program.\n", stderr);
864 >        pclose(pfp);
865   }

Diff Legend

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