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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines