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.11 by greg, Thu Jan 30 17:07:42 1997 UTC vs.
Revision 2.21 by schorsch, Mon Oct 27 10:24:51 2003 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
14 > #include <math.h>
15 > #include <time.h>
16 >
17 > #include "platform.h"
18 > #include "rtprocess.h"
19   #include "color.h"
20   #include "resolu.h"
21   #include "pmap.h"
22 + #include "warp3d.h"
23  
24                                  /* MacBeth colors */
25   #define DarkSkin        0
# Line 95 | Line 98 | short  mbneu[NMBNEU] = {Black,Neutral35,Neutral5,Neutra
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 */
# Line 106 | Line 115 | long   inpflags = 0;           /* flags of which colors were inpu
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 + WARP3D  *wcor = NULL;           /* color space warp */
122 +
123   FILE    *debugfp = NULL;        /* debug output picture */
124   char    *progname;
125  
114 extern char     *malloc();
126  
116
127   main(argc, argv)
128   int     argc;
129   char    **argv;
# Line 131 | Line 141 | char   **argv;
141                                  perror(argv[i]);
142                                  exit(1);
143                          }
144 < #ifdef MSDOS
135 <                        setmode(fileno(debugfp), O_BINARY);
136 < #endif
144 >                        SET_FILE_BINARY(debugfp);
145                          newheader("RADIANCE", debugfp);         /* start */
146                          printargs(argc, argv, debugfp);         /* header */
147                          break;
# Line 150 | Line 158 | char   **argv;
158                          bounds[3][1] = atoi(argv[++i]);
159                          scanning = 2;
160                          break;
161 +                case 'P':                               /* pick position */
162 +                        scanning = 3;
163 +                        break;
164 +                case 'i':                               /* irradiance factor */
165 +                        i++;
166 +                        if (badarg(argc-i, argv+i, "f"))
167 +                                goto userr;
168 +                        irrad = atof(argv[i]);
169 +                        break;
170 +                case 'm':                               /* raw map output */
171 +                        rawmap = 1;
172 +                        break;
173                  case 'c':                               /* color input */
174                          scanning = 0;
175                          break;
# Line 158 | Line 178 | char   **argv;
178                  }
179                                                          /* open files */
180          if (i < argc && freopen(argv[i], "r", stdin) == NULL) {
181 <                perror(argv[1]);
181 >                perror(argv[i]);
182                  exit(1);
183          }
184          if (i+1 < argc && freopen(argv[i+1], "w", stdout) == NULL) {
185 <                perror(argv[2]);
185 >                perror(argv[i+1]);
186                  exit(1);
187          }
188          if (scanning) {                 /* load input picture header */
189 < #ifdef MSDOS
170 <                setmode(fileno(stdin), O_BINARY);
171 < #endif
189 >                SET_FILE_BINARY(stdin);
190                  if (checkheader(stdin, COLRFMT, NULL) < 0 ||
191                                  fgetresolu(&xmax, &ymax, stdin) < 0) {
192                          fprintf(stderr, "%s: bad input picture\n", progname);
193                          exit(1);
194                  }
195 +                if (scanning == 3) {
196 +                        if (i >= argc)
197 +                                goto userr;
198 +                        pickchartpos(argv[i]);
199 +                        scanning = 2;
200 +                }
201          } else {                        /* else set default xmax and ymax */
202                  xmax = 512;
203                  ymax = 2*512/3;
# Line 190 | Line 214 | char   **argv;
214          else
215                  getcolors();
216          compute();                      /* compute color mapping */
217 <                                        /* print comment */
218 <        printf("{\n\tColor correction file computed by:\n\t\t");
219 <        printargs(argc, argv, stdout);
220 <        printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
221 <                        i+1 < argc ? argv[i+1] : "{this_file}");
222 <        if (!scanning)
199 <                printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
217 >        if (rawmap) {                   /* print out raw correspondence */
218 >                register int    j;
219 >
220 >                printf("# Color correspondence produced by:\n#\t\t");
221 >                printargs(argc, argv, stdout);
222 >                printf("#\tUsage: pcwarp %s uncorrected.pic > corrected.pic\n",
223                                  i+1 < argc ? argv[i+1] : "{this_file}");
224 <        printf("}\n");
225 <        putmapping();                   /* put out color mapping */
226 <        if (debugfp != NULL)            /* put out debug picture */
224 >                printf("#\t   Or: pcond [options] -m %s orig.pic > output.pic\n",
225 >                                i+1 < argc ? argv[i+1] : "{this_file}");
226 >                for (j = 0; j < 24; j++)
227 >                        printf("%f %f %f    %f %f %f\n",
228 >                                colval(inpRGB[j],RED), colval(inpRGB[j],GRN),
229 >                                colval(inpRGB[j],BLU), colval(mbRGB[j],RED),
230 >                                colval(mbRGB[j],GRN), colval(mbRGB[j],BLU));
231 >                if (scanning && debugfp != NULL)
232 >                        cwarp();                /* color warp for debugging */
233 >        } else {                        /* print color mapping */
234 >                                                /* print header */
235 >                printf("{\n\tColor correction file computed by:\n\t\t");
236 >                printargs(argc, argv, stdout);
237 >                printf("\n\tUsage: pcomb -f %s uncorrected.pic > corrected.pic\n",
238 >                                i+1 < argc ? argv[i+1] : "{this_file}");
239 >                if (!scanning)
240 >                        printf("\t   Or: pcond [options] -f %s orig.pic > output.pic\n",
241 >                                        i+1 < argc ? argv[i+1] : "{this_file}");
242 >                printf("}\n");
243 >                putmapping();                   /* put out color mapping */
244 >        }
245 >        if (debugfp != NULL) {          /* put out debug picture */
246                  if (scanning)
247                          picdebug();
248                  else
249                          clrdebug();
250 +        }
251          exit(0);
252   userr:
253          fprintf(stderr,
254 < "Usage: %s [-d dbg.pic][-p xul yul xur yur xll yll xlr ylr] input.pic [output.cal]\n",
254 > "Usage: %s [-d dbg.pic][-P | -p xul yul xur yur xll yll xlr ylr][-i irrad][-m] input.pic [output.{cal|cwp}]\n",
255                          progname);
256 <        fprintf(stderr, "   or: %s [-d dbg.pic] -c [xyY.dat [output.cal]]\n",
256 >        fprintf(stderr, "   or: %s [-d dbg.pic][-i irrad][-m] -c [xyY.dat [output.{cal|cwp}]]\n",
257                          progname);
258          exit(1);
259   }
# Line 235 | Line 278 | init()                         /* initialize */
278                  exit(1);
279          }
280                                          /* map MacBeth colors to RGB space */
281 <        for (i = 0; i < 24; i++)
281 >        for (i = 0; i < 24; i++) {
282                  xyY2RGB(mbRGB[i], mbxyY[i]);
283 +                scalecolor(mbRGB[i], irrad);
284 +        }
285   }
286  
287  
# Line 310 | Line 355 | getpicture()                           /* load in picture colors */
355                  scalecolor(inpRGB[i], d);
356                  inpflags |= 1L<<i;
357          }
358 <        free((char *)scanln);
358 >        free((void *)scanln);
359   }
360  
361  
# Line 322 | Line 367 | getcolors()                    /* get xyY colors from standard input */
367          float   xyYin[3];
368  
369          while (fgetval(stdin, 'i', &n) == 1) {          /* read colors */
370 <                if (n < 0 | n > 24 ||
370 >                if ((n < 0) | (n > 24) ||
371                                  fgetval(stdin, 'f', &xyYin[0]) != 1 ||
372                                  fgetval(stdin, 'f', &xyYin[1]) != 1 ||
373                                  fgetval(stdin, 'f', &xyYin[2]) != 1 ||
374 <                                xyYin[0] < 0. | xyYin[0] > 1. |
375 <                                xyYin[1] < 0. | xyYin[1] > 1.) {
374 >                                (xyYin[0] < 0.) | (xyYin[1] < 0.) ||
375 >                                xyYin[0] + xyYin[1] > 1.) {
376                          fprintf(stderr, "%s: bad color input data\n",
377                                          progname);
378                          exit(1);
# Line 385 | Line 430 | compute()                      /* compute color mapping */
430          COLOR   clrin[24], clrout[24];
431          long    cflags;
432          COLOR   ctmp;
433 <        register int    i, j, n;
433 >        register int    i, n;
434                                          /* did we get what we need? */
435          if ((inpflags & REQFLGS) != REQFLGS) {
436                  fprintf(stderr, "%s: missing required input colors\n",
# Line 397 | Line 442 | compute()                      /* compute color mapping */
442                  copycolor(bramp[i][0], inpRGB[mbneu[i]]);
443                  copycolor(bramp[i][1], mbRGB[mbneu[i]]);
444          }
445 +                                        /* compute color space gamut */
446 +        if (scanning) {
447 +                copycolor(colmin, cblack);
448 +                copycolor(colmax, cwhite);
449 +                scalecolor(colmax, irrad);
450 +        } else
451 +                for (i = 0; i < 3; i++) {
452 +                        colval(colmin,i) = colval(bramp[0][0],i) -
453 +                                colval(bramp[0][1],i) *
454 +                                (colval(bramp[1][0],i)-colval(bramp[0][0],i)) /
455 +                                (colval(bramp[1][1],i)-colval(bramp[1][0],i));
456 +                        colval(colmax,i) = colval(bramp[NMBNEU-2][0],i) +
457 +                                (1.-colval(bramp[NMBNEU-2][1],i)) *
458 +                                (colval(bramp[NMBNEU-1][0],i) -
459 +                                        colval(bramp[NMBNEU-2][0],i)) /
460 +                                (colval(bramp[NMBNEU-1][1],i) -
461 +                                        colval(bramp[NMBNEU-2][1],i));
462 +                }
463                                          /* compute color mapping */
464          do {
465                  cflags = inpflags & ~gmtflags;
# Line 408 | Line 471 | compute()                      /* compute color mapping */
471                                  n++;
472                          }
473                  compsoln(clrin, clrout, n);
474 <                                                /* check out-of-gamut colors */
475 <                for (i = 0; i < 24; i++)
476 <                        if (cflags & 1L<<i) {
477 <                                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 <                        }
474 >                if (irrad > 0.99 && irrad < 1.01)       /* check gamut */
475 >                        for (i = 0; i < 24; i++)
476 >                                if (cflags & 1L<<i && cvtcolor(ctmp, mbRGB[i]))
477 >                                        gmtflags |= 1L<<i;
478          } while (cflags & gmtflags);
479          if (gmtflags & MODFLGS)
480                  fprintf(stderr,
# Line 427 | Line 483 | compute()                      /* compute color mapping */
483   }
484  
485  
486 < putmapping()                    /* put out color mapping for pcomb -f */
486 > putmapping()                    /* put out color mapping */
487   {
488          static char     cchar[3] = {'r', 'g', 'b'};
489          register int    i, j;
# Line 535 | Line 591 | int    n;
591   }
592  
593  
594 + cwarp()                         /* compute color warp map */
595 + {
596 +        register int    i;
597 +
598 +        if ((wcor = new3dw(W3EXACT)) == NULL)
599 +                goto memerr;
600 +        for (i = 0; i < 24; i++)
601 +                if (!add3dpt(wcor, inpRGB[i], mbRGB[i]))
602 +                        goto memerr;
603 +        return;
604 + memerr:
605 +        perror(progname);
606 +        exit(1);
607 + }
608 +
609 +
610 + int
611   cvtcolor(cout, cin)             /* convert color according to our mapping */
612   COLOR   cout, cin;
613   {
614          COLOR   ctmp;
615 +        int     clipped;
616  
617 <        if (scanning) {
617 >        if (wcor != NULL) {
618 >                clipped = warp3d(cout, cin, wcor);
619 >                clipped |= clipgamut(cout,bright(cout),CGAMUT,colmin,colmax);
620 >        } else if (scanning) {
621                  bresp(ctmp, cin);
622 <                cresp(cout, ctmp);
622 >                clipped = cresp(cout, ctmp);
623          } else {
624 <                cresp(ctmp, cin);
624 >                clipped = cresp(ctmp, cin);
625                  bresp(cout, ctmp);
626          }
627 <        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.;
627 >        return(clipped);
628   }
629  
630  
631 + int
632   cresp(cout, cin)                /* transform color according to matrix */
633   COLOR   cout, cin;
634   {
635 <        double  r, g, b;
636 <
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);
635 >        colortrans(cout, solmat, cin);
636 >        return(clipgamut(cout, bright(cout), CGAMUT, colmin, colmax));
637   }
638  
639  
# Line 582 | Line 648 | register float xyYin[3];
648          ctmp[0] = xyYin[0] * d;
649          ctmp[1] = xyYin[2];
650          ctmp[2] = (1. - xyYin[0] - xyYin[1]) * d;
651 <        cie_rgb(rgbout, ctmp);
651 >                                /* allow negative values */
652 >        colortrans(rgbout, xyz2rgbmat, ctmp);
653   }
654  
655  
# Line 619 | Line 686 | picdebug()                     /* put out debugging picture */
686                  for (x = 0; x < xmax; x++) {
687                          rg = chartndx(x, y, &i);
688                          if (rg == RG_CENT) {
689 <                                if (!(1L<<i & gmtflags) || (x+y)&07)
689 >                                if (!(1L<<i & gmtflags) || (x+y)&07) {
690                                          copycolor(scan[x], mbRGB[i]);
691 <                                else
691 >                                        clipgamut(scan[x], bright(scan[x]),
692 >                                                CGAMUT, colmin, colmax);
693 >                                } else
694                                          copycolor(scan[x], blkcol);
695                          } else if (rg == RG_CORR)
696                                  cvtcolor(scan[x], scan[x]);
# Line 636 | Line 705 | picdebug()                     /* put out debugging picture */
705          }
706                                                  /* clean up */
707          fclose(debugfp);
708 <        free((char *)scan);
708 >        free((void *)scan);
709   }
710  
711  
# Line 645 | Line 714 | clrdebug()                     /* put out debug picture from color input
714          static COLR     blkclr = BLKCOLR;
715          COLR    mbclr[24], cvclr[24], orclr[24];
716          COLR    *scan;
717 <        COLOR   ctmp;
717 >        COLOR   ctmp, ct2;
718          int     y, i;
719          register int    x, rg;
720                                                  /* convert colors */
721          for (i = 0; i < 24; i++) {
722 <                setcolr(mbclr[i], colval(mbRGB[i],RED),
723 <                                colval(mbRGB[i],GRN), colval(mbRGB[i],BLU));
722 >                copycolor(ctmp, mbRGB[i]);
723 >                clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
724 >                setcolr(mbclr[i], colval(ctmp,RED),
725 >                                colval(ctmp,GRN), colval(ctmp,BLU));
726                  if (inpflags & 1L<<i) {
727 <                        setcolr(orclr[i], colval(inpRGB[i],RED),
728 <                                        colval(inpRGB[i],GRN),
729 <                                        colval(inpRGB[i],BLU));
659 <                        cvtcolor(ctmp, inpRGB[i]);
660 <                        setcolr(cvclr[i], colval(ctmp,RED),
727 >                        copycolor(ctmp, inpRGB[i]);
728 >                        clipgamut(ctmp, bright(ctmp), CGAMUT, cblack, cwhite);
729 >                        setcolr(orclr[i], colval(ctmp,RED),
730                                          colval(ctmp,GRN), colval(ctmp,BLU));
731 +                        if (rawmap)
732 +                                copycolr(cvclr[i], mbclr[i]);
733 +                        else {
734 +                                bresp(ctmp, inpRGB[i]);
735 +                                colortrans(ct2, solmat, ctmp);
736 +                                clipgamut(ct2, bright(ct2), CGAMUT,
737 +                                                cblack, cwhite);
738 +                                setcolr(cvclr[i], colval(ct2,RED),
739 +                                                colval(ct2,GRN),
740 +                                                colval(ct2,BLU));
741 +                        }
742                  }
743          }
744                                                  /* allocate scanline */
# Line 695 | Line 775 | clrdebug()                     /* put out debug picture from color input
775          }
776                                                  /* clean up */
777          fclose(debugfp);
778 <        free((char *)scan);
778 >        free((void *)scan);
779 > }
780 >
781 >
782 > getpos(name, bnds, fp)          /* get boundary position */
783 > char    *name;
784 > int     bnds[2];
785 > FILE    *fp;
786 > {
787 >        char    buf[64];
788 >
789 >        fprintf(stderr, "\tSelect corner: %s\n", name);
790 >        if (fgets(buf, sizeof(buf), fp) == NULL ||
791 >                        sscanf(buf, "%d %d", &bnds[0], &bnds[1]) != 2) {
792 >                fprintf(stderr, "%s: read error from display process\n",
793 >                                progname);
794 >                exit(1);
795 >        }
796 > }
797 >
798 >
799 > pickchartpos(pfn)               /* display picture and pick chart location */
800 > char    *pfn;
801 > {
802 >        char    combuf[512];
803 >        FILE    *pfp;
804 >
805 >        sprintf(combuf, DISPCOM, pfn);
806 >        if ((pfp = popen(combuf, "r")) == NULL) {
807 >                perror(combuf);
808 >                exit(1);
809 >        }
810 >        fputs("Use middle mouse button to select chart corners:\n", stderr);
811 >        getpos("upper left (dark skin)", bounds[0], pfp);
812 >        getpos("upper right (bluish green)", bounds[1], pfp);
813 >        getpos("lower left (white)", bounds[2], pfp);
814 >        getpos("lower right (black)", bounds[3], pfp);
815 >        fputs("Got it -- quit display program.\n", stderr);
816 >        pclose(pfp);
817   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines