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

Comparing ray/src/px/pcomb.c (file contents):
Revision 2.47 by greg, Fri Jan 12 00:50:17 2018 UTC vs.
Revision 2.60 by greg, Thu Nov 7 20:07:08 2024 UTC

# Line 23 | Line 23 | struct {
23          FILE    *fp;            /* stream pointer */
24          VIEW    vw;             /* view for picture */
25          RESOLU  rs;             /* image resolution and orientation */
26 +        int     infloat;        /* input is floating point (#comp)? */
27          float   pa;             /* pixel aspect ratio */
28          COLOR   *scan[WINSIZ];  /* input scanline window */
29          COLOR   coef;           /* coefficient */
# Line 31 | Line 32 | struct {
32  
33   int     nfiles;                         /* number of input files */
34  
35 < char    ourfmt[LPICFMT+1] = PICFMT;     /* input picture format */
35 > VIEW    *commvp = NULL;                 /* common view parameters */
36  
37 + char    ourfmt[MAXFMTLEN] = PICFMT;     /* input picture format */
38 + int     outfloat = 0;                   /* #component float output? */
39 +
40   char    StandardInput[] = "<stdin>";
41   char    Command[] = "<Command>";
42   char    vcolin[3][4] = {"ri", "gi", "bi"};
# Line 74 | Line 78 | int    gotview;
78  
79  
80   static gethfunc headline;
81 < static void checkfile(void);
81 > static void checkfile(int orig);
82   static double rgb_bright(COLOR  clr);
83   static double xyz_bright(COLOR  clr);
84   static void init(void);
# Line 118 | Line 122 | main(
122                                  echoheader = !echoheader;
123                                  continue;
124                          case 'f':
125 +                                if (argv[a][2]) {
126 +                                        outfloat = (argv[a][2] == 'f');
127 +                                        continue;
128 +                                }
129 +                                /* fall through */
130                          case 'e':
131                                  a++;
132                                  continue;
# Line 175 | Line 184 | main(
184                                  quit(1);
185                          }
186                  }
187 <                checkfile();
179 <                if (original) {
180 <                        colval(input[nfiles].coef,RED) /=
181 <                                        colval(input[nfiles].expos,RED);
182 <                        colval(input[nfiles].coef,GRN) /=
183 <                                        colval(input[nfiles].expos,GRN);
184 <                        colval(input[nfiles].coef,BLU) /=
185 <                                        colval(input[nfiles].expos,BLU);
186 <                        setcolor(input[nfiles].expos, 1.0, 1.0, 1.0);
187 <                }
187 >                checkfile(original);
188                  nfiles++;
189                  original = 0;
190          }
191          init();                         /* set constants */
192                                          /* go back and get expressions */
193          for (a = 1; a < argc; a++) {
194 +                char    *fpath;
195                  if (argv[a][0] == '-')
196                          switch (argv[a][1]) {
197                          case 'x':
# Line 204 | Line 205 | main(
205                          case 'h':
206                                  continue;
207                          case 'f':
208 <                                fcompile(argv[++a]);
208 >                                if (argv[a][2])
209 >                                        continue;
210 >                                fpath = getpath(argv[++a], getrlibpath(), 0);
211 >                                if (fpath == NULL) {
212 >                                        eputs(argv[0]);
213 >                                        eputs(": cannot find file '");
214 >                                        eputs(argv[a]);
215 >                                        eputs("'\n");
216 >                                        quit(1);
217 >                                }
218 >                                fcompile(fpath);
219                                  continue;
220                          case 'e':
221                                  scompile(argv[++a], NULL, 0);
# Line 212 | Line 223 | main(
223                          }
224                  break;
225          }
226 <                                                /* set/get output resolution */
216 <        if (!vardefined(vxres))
217 <                varset(vxres, ':', (double)xmax);
218 <        if (!vardefined(vyres))
219 <                varset(vyres, ':', (double)ymax);
226 >                                                /* get output resolution */
227          xres = varvalue(vxres) + .5;
228          yres = varvalue(vyres) + .5;
229          if (xres <= 0 || yres <= 0) {
# Line 224 | Line 231 | main(
231                  eputs(": illegal output resolution\n");
232                  quit(1);
233          }
234 <                                                /* complete header */
235 <        printargs(argc, argv, stdout);
236 <        if (strcmp(ourfmt, PICFMT))
237 <                fputformat(ourfmt, stdout);     /* print format if known */
238 <        putchar('\n');
239 <        fprtresolu(xres, yres, stdout);
240 <                                                /* combine pictures */
241 <        combine();
234 >        if (!vardefined(vbrtout))               /* single or 3-channel? */
235 >                outfloat *= 3;
236 >
237 >        printargs(argc, argv, stdout);          /* complete header */
238 >        if (commvp != NULL) {
239 >                fputs(VIEWSTR, stdout);
240 >                fprintview(commvp, stdout);
241 >                fputc('\n', stdout);
242 >        }
243 >        if (outfloat) {                         /* print output format */
244 >                printf("NROWS=%d\nNCOLS=%d\n", yres, xres);
245 >                fputncomp(outfloat, stdout);
246 >                fputendian(stdout);
247 >                fputformat("float", stdout);
248 >        } else if (strcmp(ourfmt, PICFMT))
249 >                fputformat(ourfmt, stdout);
250 >        else
251 >                fputformat(COLRFMT, stdout);
252 >        fputc('\n', stdout);                    /* end header */
253 >        if (!outfloat)
254 >                fprtresolu(xres, yres, stdout);
255 >
256 >        doptimize(1);                           /* optimize definitions */
257 >        combine();                              /* combine pictures */
258          quit(0);
259   usage:
260          eputs("Usage: ");
261          eputs(argv[0]);
262          eputs(
263 < " [-w][-h][-x xr][-y yr][-e expr][-f file] [ [-o][-s f][-c r g b] hdr ..]\n");
263 > " [-w][-h][-ff][-x xr][-y yr][-e expr][-f file] [ [-o][-s f][-c r g b] hdr ..]\n");
264          quit(1);
265          return 1; /* pro forma return */
266   }
# Line 249 | Line 272 | headline(                      /* check header line & echo if requested *
272          void    *p
273   )
274   {
275 <        char    fmt[32];
275 >        int     orig = *(int *)p;
276 >        char    fmt[MAXFMTLEN];
277          double  d;
278 +        int     bigend;
279          COLOR   ctmp;
280  
281          if (isheadid(s))                        /* header id */
# Line 259 | Line 284 | headline(                      /* check header line & echo if requested *
284                  if (globmatch(ourfmt, fmt)) {
285                          wrongformat = 0;
286                          strcpy(ourfmt, fmt);
287 <                } else
287 >                } else if (!strcmp("float", fmt))
288 >                        input[nfiles].infloat *= -1;
289 >                else
290                          wrongformat = globmatch(PICFMT, fmt) ? 1 : -1;
291                  return(0);      /* don't echo */
292          }
293 <        if (isexpos(s)) {                       /* exposure */
293 >        if ((bigend = isbigendian(s)) >= 0) {
294 >                if (bigend != nativebigendian()) {
295 >                        eputs(input[nfiles].name);
296 >                        eputs(": unsupported input byte ordering\n");
297 >                        quit(1);
298 >                }
299 >                return(0);      /* don't echo */
300 >        }
301 >        if (!strncmp("NROWS=", s, 6)) {         /* X-resolution */
302 >                input[nfiles].rs.yr = atoi(s+6);
303 >                return(0);      /* don't echo */
304 >        }
305 >        if (!strncmp("NCOLS=", s, 6)) {         /* Y-resolution */
306 >                input[nfiles].rs.xr = atoi(s+6);
307 >                return(0);      /* don't echo */
308 >        }
309 >        if (isncomp(s))                         /* # components */
310 >                input[nfiles].infloat *= ncompval(s);
311 >
312 >        if (orig) {                             /* undo exposure? */
313 >                if (isexpos(s)) {
314 >                        d = 1./exposval(s);
315 >                        scalecolor(input[nfiles].coef, d);
316 >                        return(0);      /* don't echo */
317 >                }
318 >                if (iscolcor(s)) {
319 >                        int     i;
320 >                        colcorval(ctmp, s);
321 >                        for (i = 3; i--; )
322 >                                colval(input[nfiles].coef,i) /= colval(ctmp,i);
323 >                        return(0);      /* don't echo */
324 >                }
325 >        } else if (isexpos(s)) {                /* record exposure? */
326                  d = exposval(s);
327                  scalecolor(input[nfiles].expos, d);
328 <        } else if (iscolcor(s)) {               /* color correction */
328 >        } else if (iscolcor(s)) {
329                  colcorval(ctmp, s);
330                  multcolor(input[nfiles].expos, ctmp);
331 <        } else if (isaspect(s))
331 >        }
332 >        if (isaspect(s))
333                  input[nfiles].pa *= aspectval(s);
334          else if (isview(s) && sscanview(&input[nfiles].vw, s) > 0)
335                  gotview++;
# Line 283 | Line 343 | headline(                      /* check header line & echo if requested *
343  
344  
345   static void
346 < checkfile(void)                 /* ready a file */
346 > checkfile(int orig)                     /* ready a file */
347   {
348          int     i;
349                                          /* process header */
350          gotview = 0;
351 +        input[nfiles].infloat = -1;
352 +        input[nfiles].rs.rt = PIXSTANDARD;
353 +        input[nfiles].rs.xr = input[nfiles].rs.yr = 0;
354          if (echoheader) {
355                  fputs(input[nfiles].name, stdout);
356                  fputs(":\n", stdout);
357          }
358 <        getheader(input[nfiles].fp, headline, NULL);
358 >        getheader(input[nfiles].fp, headline, &orig);
359 >
360 >        if (input[nfiles].infloat <= 0)
361 >                input[nfiles].infloat = 0;
362 >        else if ((input[nfiles].infloat != 3) &
363 >                                (input[nfiles].infloat != 1)) {
364 >                eputs(input[nfiles].name);
365 >                eputs(": unsupported number of components\n");
366 >                quit(1);
367 >        }
368          if (wrongformat < 0) {
369                  eputs(input[nfiles].name);
370 <                eputs(": not a Radiance picture\n");
370 >                eputs(": not a Radiance picture or float matrix\n");
371                  quit(1);
372          }
373          if (wrongformat > 0) {
# Line 304 | Line 376 | checkfile(void)                        /* ready a file */
376          }
377          if (!gotview || setview(&input[nfiles].vw) != NULL)
378                  input[nfiles].vw.type = 0;
379 <        if (!fgetsresolu(&input[nfiles].rs, input[nfiles].fp)) {
379 >        else if (commvp == NULL)
380 >                commvp = &input[nfiles].vw;
381 >        if ((input[nfiles].rs.xr <= 0) | (input[nfiles].rs.yr <= 0) &&
382 >                        !fgetsresolu(&input[nfiles].rs, input[nfiles].fp)) {
383                  eputs(input[nfiles].name);
384                  eputs(": bad picture size\n");
385                  quit(1);
# Line 371 | Line 446 | init(void)                                     /* perform final setup */
446                  ourbright = xyz_bright;
447          } else
448                  varset(vwhteff, ':', WHTEFFICACY);
449 +                                                /* these may be overridden */
450 +        varset(vxres, ':', (double)xmax);
451 +        varset(vyres, ':', (double)ymax);
452   }
453  
454  
# Line 378 | Line 456 | static void
456   combine(void)                   /* combine pictures */
457   {
458          EPNODE  *coldef[3], *brtdef;
459 +        int     set_x, set_y;
460          COLOR   *scanout;
461          double  d;
462          int     i, j;
# Line 392 | Line 471 | combine(void)                  /* combine pictures */
471                  brtdef = eparse(vbrtout);
472          else
473                  brtdef = NULL;
474 +                                                /* what to set */
475 +        set_x = varlookup(vxpos) != NULL && !vardefined(vxpos);
476 +        set_y = varlookup(vypos) != NULL && !vardefined(vypos);
477                                                  /* allocate scanline */
478          scanout = (COLOR *)emalloc(xres*sizeof(COLOR));
479                                                  /* set input position */
# Line 399 | Line 481 | combine(void)                  /* combine pictures */
481                                                  /* combine files */
482          for (ypos = yres-1; ypos >= 0; ypos--) {
483              advance();
484 <            varset(vypos, '=', (double)ypos);
484 >            if (set_y) varset(vypos, '=', (double)ypos);
485              for (xpos = 0; xpos < xres; xpos++) {
486                  xscan = (xpos+.5)*xmax/xres;
487 <                varset(vxpos, '=', (double)xpos);
487 >                if (set_x) varset(vxpos, '=', (double)xpos);
488                  eclock++;
489                  if (brtdef != NULL) {
490                      d = evalue(brtdef);
491 <                    if (d < 0.0)
410 <                        d = 0.0;
491 >                    d *= (outfloat > 0) | (d >= 0);
492                      setcolor(scanout[xpos], d, d, d);
493                  } else {
494                      for (j = 0; j < 3; j++) {
# Line 418 | Line 499 | combine(void)                  /* combine pictures */
499                              for (i = 0; i < nfiles; i++)
500                                  d += colval(input[i].scan[MIDSCN][xscan],j);
501                          }
502 <                        if (d < 0.0)
422 <                            d = 0.0;
502 >                        d *= (outfloat > 0) | (d >= 0);
503                          colval(scanout[xpos],j) = d;
504                      }
505                  }
506              }
507 <            if (fwritescan(scanout, xres, stdout) < 0) {
508 <                    perror("write error");
509 <                    quit(1);
507 >            switch (outfloat) {
508 >            case 3:                     /* writing out float triplets */
509 >                if (fwrite(scanout, sizeof(float)*3, xres, stdout) != xres)
510 >                    goto writerr;
511 >                break;
512 >            case 1:                     /* writing out gray float values */
513 >                for (xpos = 0; xpos < xres; xpos++)
514 >                    if (putbinary(&scanout[xpos][CIEY],
515 >                                sizeof(float), 1, stdout) != 1)
516 >                        goto writerr;
517 >                break;
518 >            default:                    /* writing out Radiance picture */
519 >                if (fwritescan(scanout, xres, stdout) < 0)
520 >                    goto writerr;
521 >                break;
522              }
523          }
524          efree((char *)scanout);
525 +        return;
526 + writerr:
527 +    perror("write error");
528 +    quit(1);
529   }
530  
531  
# Line 448 | Line 544 | advance(void)                  /* read in data for next scanline */
544                          input[i].scan[0] = st;
545                          if (yscan <= MIDSCN)            /* hit bottom? */
546                                  continue;
547 <                        if (freadscan(st, xmax, input[i].fp) < 0) {  /* read */
548 <                                eputs(input[i].name);
549 <                                eputs(": read error\n");
550 <                                quit(1);
551 <                        }
547 >                                                         /* read scanline */
548 >                        if (input[i].infloat) {
549 >                                if (fread(st, sizeof(float)*input[i].infloat,
550 >                                                xmax, input[i].fp) != xmax)
551 >                                        goto readerr;
552 >                                if (input[i].infloat == 1) {
553 >                                        const COLORV    *f = st[0] + xmax;
554 >                                        int             x = xmax;
555 >                                        while (x-- > 0)
556 >                                                st[x][BLU] = st[x][GRN] =
557 >                                                        st[x][RED] = *--f;
558 >                                }
559 >                        } else if (freadscan(st, xmax, input[i].fp) < 0)
560 >                                goto readerr;
561                          for (j = 0; j < xmax; j++)      /* adjust color */
562                                  multcolor(st[j], input[i].coef);
563                  }
564 +        return;
565 + readerr:
566 +        eputs(input[i].name);
567 +        eputs(": read error\n");
568 +        quit(1);
569   }
570  
571  
# Line 586 | Line 696 | l_ray(         /* return ray origin or direction */
696                  return((double)nfiles);
697          fn = d - 0.5;
698          if (ltick[fn] != eclock) {              /* need to compute? */
699 <                lorg[fn][0] = lorg[fn][1] = lorg[fn][2] = 0.0;
700 <                ldir[fn][0] = ldir[fn][1] = ldir[fn][2] = 0.0;
701 <                ldist[fn] = -1.0;
702 <                if (input[fn].vw.type == 0)
699 >                if (input[fn].vw.type == 0) {
700 >                        lorg[fn][0] = lorg[fn][1] = lorg[fn][2] = 0.0;
701 >                        ldir[fn][0] = ldir[fn][1] = ldir[fn][2] = 0.0;
702 >                        ldist[fn] = -1.0;
703                          errno = EDOM;
704 <                else {
704 >                } else {
705                          pix2loc(loc, &input[fn].rs, xscan, ymax-1-yscan);
706                          ldist[fn] = viewray(lorg[fn], ldir[fn],
707                                          &input[fn].vw, loc[0], loc[1]);
708 +                        if (ldist[fn] < -FTINY)
709 +                                errno = EDOM;
710                  }
711                  ltick[fn] = eclock;
712          }
# Line 658 | Line 770 | l_psize(char *nm)              /* compute pixel size in steradians
770  
771  
772   extern void
773 < wputs(char      *msg)
773 > wputs(const char *msg)
774   {
775          if (!nowarn)
776                  eputs(msg);
# Line 666 | Line 778 | wputs(char     *msg)
778  
779  
780   extern void
781 < eputs(char *msg)
781 > eputs(const char *msg)
782   {
783          fputs(msg, stderr);
784   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines