ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pcomb.c
Revision: 2.21
Committed: Tue Oct 27 09:08:25 1998 UTC (25 years, 6 months ago) by gwlarson
Content type: text/plain
Branch: MAIN
Changes since 2.20: +3 -3 lines
Log Message:
changed getheader() to listen to return value of passed function

File Contents

# User Rev Content
1 greg 2.6 /* Copyright (c) 1992 Regents of the University of California */
2 greg 1.1
3     #ifndef lint
4     static char SCCSid[] = "$SunId$ LBL";
5     #endif
6    
7     /*
8     * Combine picture files according to calcomp functions.
9     *
10     * 1/4/89
11     */
12    
13 greg 2.10 #include "standard.h"
14 greg 1.1
15     #include "color.h"
16    
17 greg 2.8 #include "resolu.h"
18 greg 1.23
19 greg 1.1 #include "calcomp.h"
20    
21 greg 2.10 #include "view.h"
22    
23 greg 1.6 #define MAXINP 32 /* maximum number of input files */
24 greg 2.13 #define WINSIZ 17 /* scanline window size */
25     #define MIDSCN ((WINSIZ-1)/2+1)
26 greg 1.1
27     struct {
28 greg 1.19 char *name; /* file or command name */
29 greg 1.1 FILE *fp; /* stream pointer */
30 greg 2.10 VIEW vw; /* view for picture */
31     RESOLU rs; /* image resolution and orientation */
32 greg 2.17 float pa; /* pixel aspect ratio */
33 greg 1.14 COLOR *scan[WINSIZ]; /* input scanline window */
34 greg 1.3 COLOR coef; /* coefficient */
35 greg 1.17 COLOR expos; /* recorded exposure */
36 greg 1.1 } input[MAXINP]; /* input pictures */
37    
38     int nfiles; /* number of input files */
39    
40 greg 2.16 char ourfmt[LPICFMT+1] = PICFMT; /* input picture format */
41    
42 greg 2.6 char Command[] = "<Command>";
43 greg 1.21 char vcolin[3][4] = {"ri", "gi", "bi"};
44     char vcolout[3][4] = {"ro", "go", "bo"};
45 greg 1.15 char vbrtin[] = "li";
46     char vbrtout[] = "lo";
47 greg 1.21 char vcolexp[3][4] = {"re", "ge", "be"};
48 greg 1.17 char vbrtexp[] = "le";
49 greg 2.18 char vpixaspect[] = "pa";
50 greg 1.1
51 greg 2.18 char vray[7][4] = {"Ox", "Oy", "Oz", "Dx", "Dy", "Dz", "T"};
52 greg 2.10
53 greg 2.17 char vpsize[] = "S";
54    
55 greg 1.18 char vnfiles[] = "nfiles";
56 greg 2.18 char vwhteff[] = "WE";
57 greg 1.18 char vxmax[] = "xmax";
58     char vymax[] = "ymax";
59     char vxres[] = "xres";
60     char vyres[] = "yres";
61     char vxpos[] = "x";
62     char vypos[] = "y";
63 greg 1.1
64     int nowarn = 0; /* no warning messages? */
65    
66 greg 1.18 int xmax = 0, ymax = 0; /* input resolution */
67 greg 1.9
68 greg 1.18 int xscan, yscan; /* input position */
69 greg 1.1
70 greg 1.18 int xres, yres; /* output resolution */
71 greg 1.1
72 greg 1.18 int xpos, ypos; /* output position */
73    
74 greg 2.10 char *progname; /* global argv[0] */
75    
76 greg 1.10 int wrongformat = 0;
77 greg 2.10 int gotview;
78 greg 1.1
79 greg 1.19 FILE *popen();
80 greg 1.10
81 greg 2.7 extern char *emalloc();
82 greg 1.19
83 greg 2.7
84 greg 1.1 main(argc, argv)
85     int argc;
86     char *argv[];
87     {
88 greg 1.18 int original;
89 greg 1.3 double f;
90 greg 1.14 int a, i;
91 greg 2.7 #ifdef MSDOS
92     extern int _fmode;
93     _fmode = O_BINARY;
94     setmode(fileno(stdin), O_BINARY);
95     setmode(fileno(stdout), O_BINARY);
96     #endif
97 greg 2.10 progname = argv[0];
98 greg 1.15 /* scan options */
99     for (a = 1; a < argc; a++) {
100 greg 1.1 if (argv[a][0] == '-')
101     switch (argv[a][1]) {
102     case 'x':
103     case 'y':
104 greg 1.18 a++;
105 greg 1.15 continue;
106 greg 1.1 case 'w':
107     nowarn = !nowarn;
108 greg 1.15 continue;
109 greg 1.1 case 'f':
110     case 'e':
111 greg 1.16 a++;
112 greg 1.15 continue;
113 greg 1.1 }
114 greg 1.15 break;
115     }
116 greg 2.14 newheader("RADIANCE", stdout); /* start header */
117 greg 1.15 /* process files */
118 greg 1.17 for (nfiles = 0; nfiles < MAXINP; nfiles++) {
119 greg 1.3 setcolor(input[nfiles].coef, 1.0, 1.0, 1.0);
120 greg 1.17 setcolor(input[nfiles].expos, 1.0, 1.0, 1.0);
121 greg 2.10 copystruct(&input[nfiles].vw, &stdview);
122 greg 2.17 input[nfiles].pa = 1.0;
123 greg 1.17 }
124 greg 1.1 nfiles = 0;
125 greg 1.18 original = 0;
126 greg 1.1 for ( ; a < argc; a++) {
127     if (nfiles >= MAXINP) {
128     eputs(argv[0]);
129     eputs(": too many picture files\n");
130     quit(1);
131     }
132 greg 1.3 if (argv[a][0] == '-')
133     switch (argv[a][1]) {
134     case '\0':
135     input[nfiles].name = "<stdin>";
136     input[nfiles].fp = stdin;
137     break;
138 greg 1.17 case 'o':
139     original++;
140 greg 1.22 continue;
141 greg 1.3 case 's':
142     f = atof(argv[++a]);
143     scalecolor(input[nfiles].coef, f);
144     continue;
145     case 'c':
146     colval(input[nfiles].coef,RED)*=atof(argv[++a]);
147     colval(input[nfiles].coef,GRN)*=atof(argv[++a]);
148     colval(input[nfiles].coef,BLU)*=atof(argv[++a]);
149     continue;
150     default:
151     goto usage;
152     }
153     else {
154 greg 2.2 if (argv[a][0] == '!') {
155 greg 2.6 input[nfiles].name = Command;
156 greg 2.2 input[nfiles].fp = popen(argv[a]+1, "r");
157     } else {
158     input[nfiles].name = argv[a];
159     input[nfiles].fp = fopen(argv[a], "r");
160     }
161 greg 1.1 if (input[nfiles].fp == NULL) {
162 greg 1.6 perror(argv[a]);
163 greg 1.1 quit(1);
164     }
165     }
166 greg 1.16 checkfile();
167 greg 1.18 if (original) {
168     colval(input[nfiles].coef,RED) /=
169     colval(input[nfiles].expos,RED);
170     colval(input[nfiles].coef,GRN) /=
171     colval(input[nfiles].expos,GRN);
172     colval(input[nfiles].coef,BLU) /=
173     colval(input[nfiles].expos,BLU);
174     }
175     nfiles++;
176 greg 1.17 original = 0;
177 greg 1.15 }
178 greg 1.18 init(); /* set constants */
179 greg 1.15 /* go back and get expressions */
180     for (a = 1; a < argc; a++) {
181     if (argv[a][0] == '-')
182     switch (argv[a][1]) {
183     case 'x':
184 greg 1.21 varset(vxres, ':', eval(argv[++a]));
185 greg 1.18 continue;
186 greg 1.15 case 'y':
187 greg 1.21 varset(vyres, ':', eval(argv[++a]));
188 greg 1.16 continue;
189 greg 1.15 case 'w':
190     continue;
191     case 'f':
192     fcompile(argv[++a]);
193     continue;
194     case 'e':
195     scompile(argv[++a], NULL, 0);
196     continue;
197     }
198     break;
199     }
200 greg 1.18 /* set/get output resolution */
201     if (!vardefined(vxres))
202     varset(vxres, ':', (double)xmax);
203     if (!vardefined(vyres))
204     varset(vyres, ':', (double)ymax);
205     xres = varvalue(vxres) + .5;
206     yres = varvalue(vyres) + .5;
207     if (xres <= 0 || yres <= 0) {
208     eputs(argv[0]);
209     eputs(": illegal output resolution\n");
210     quit(1);
211     }
212 greg 1.15 /* complete header */
213     printargs(argc, argv, stdout);
214 greg 2.16 if (strcmp(ourfmt, PICFMT))
215     fputformat(ourfmt, stdout); /* print format if known */
216 greg 1.15 putchar('\n');
217 greg 1.23 fprtresolu(xres, yres, stdout);
218 greg 1.15 /* combine pictures */
219     combine();
220     quit(0);
221     usage:
222     eputs("Usage: ");
223     eputs(argv[0]);
224     eputs(
225 greg 2.9 " [-w][-x xr][-y yr][-e expr][-f file] [ [-o][-s f][-c r g b] pic ..]\n");
226 greg 1.15 quit(1);
227     }
228    
229    
230     tputs(s) /* put out string preceded by a tab */
231     char *s;
232     {
233     char fmt[32];
234     double d;
235     COLOR ctmp;
236    
237 greg 2.14 if (isheadid(s)) /* header id */
238 gwlarson 2.21 return(0); /* don't echo */
239 greg 2.14 if (formatval(fmt, s)) { /* check format */
240 greg 2.16 if (globmatch(ourfmt, fmt)) {
241     wrongformat = 0;
242     strcpy(ourfmt, fmt);
243     } else
244     wrongformat = 1;
245 gwlarson 2.21 return(0); /* don't echo */
246 greg 1.17 }
247     if (isexpos(s)) { /* exposure */
248     d = exposval(s);
249     scalecolor(input[nfiles].expos, d);
250     } else if (iscolcor(s)) { /* color correction */
251 greg 1.15 colcorval(ctmp, s);
252 greg 1.17 multcolor(input[nfiles].expos, ctmp);
253 greg 2.18 } else if (isaspect(s))
254 greg 2.17 input[nfiles].pa *= aspectval(s);
255 greg 2.18 else if (isview(s) && sscanview(&input[nfiles].vw, s) > 0)
256 greg 2.10 gotview++;
257 greg 1.17 /* echo line */
258     putchar('\t');
259 gwlarson 2.21 return(fputs(s, stdout));
260 greg 1.16 }
261 greg 1.15
262 greg 1.16
263     checkfile() /* ready a file */
264     {
265     register int i;
266     /* process header */
267 greg 2.10 gotview = 0;
268 greg 1.16 fputs(input[nfiles].name, stdout);
269     fputs(":\n", stdout);
270     getheader(input[nfiles].fp, tputs, NULL);
271     if (wrongformat) {
272     eputs(input[nfiles].name);
273     eputs(": not in Radiance picture format\n");
274     quit(1);
275     }
276 greg 2.10 if (!gotview || setview(&input[nfiles].vw) != NULL)
277     input[nfiles].vw.type = 0;
278     if (!fgetsresolu(&input[nfiles].rs, input[nfiles].fp)) {
279 greg 1.16 eputs(input[nfiles].name);
280     eputs(": bad picture size\n");
281     quit(1);
282     }
283 greg 1.18 if (xmax == 0 && ymax == 0) {
284 greg 2.10 xmax = scanlen(&input[nfiles].rs);
285     ymax = numscans(&input[nfiles].rs);
286     } else if (scanlen(&input[nfiles].rs) != xmax ||
287     numscans(&input[nfiles].rs) != ymax) {
288 greg 1.16 eputs(input[nfiles].name);
289     eputs(": resolution mismatch\n");
290     quit(1);
291     }
292     /* allocate scanlines */
293     for (i = 0; i < WINSIZ; i++)
294 greg 1.18 input[nfiles].scan[i] = (COLOR *)emalloc(xmax*sizeof(COLOR));
295 greg 1.15 }
296    
297    
298 greg 2.16 double
299     rgb_bright(clr)
300     COLOR clr;
301     {
302     return(bright(clr));
303     }
304    
305    
306     double
307     xyz_bright(clr)
308     COLOR clr;
309     {
310     return(clr[CIEY]);
311     }
312    
313    
314     double (*ourbright)() = rgb_bright;
315    
316    
317 greg 1.18 init() /* perform final setup */
318 greg 1.15 {
319 greg 2.18 double l_colin(), l_expos(), l_pixaspect(), l_ray(), l_psize();
320 greg 1.16 register int i;
321 greg 1.15 /* define constants */
322 greg 2.19 varset("PI", ':', PI);
323 greg 1.15 varset(vnfiles, ':', (double)nfiles);
324 greg 1.18 varset(vxmax, ':', (double)xmax);
325     varset(vymax, ':', (double)ymax);
326 greg 1.15 /* set functions */
327     for (i = 0; i < 3; i++) {
328 greg 1.17 funset(vcolexp[i], 1, ':', l_expos);
329 greg 1.15 funset(vcolin[i], 1, '=', l_colin);
330     }
331 greg 1.17 funset(vbrtexp, 1, ':', l_expos);
332 greg 1.15 funset(vbrtin, 1, '=', l_colin);
333 greg 2.18 funset(vpixaspect, 1, ':', l_pixaspect);
334     for (i = 0; i < 7; i++)
335 greg 2.10 funset(vray[i], 1, '=', l_ray);
336 greg 2.17 funset(vpsize, 1, '=', l_psize);
337 greg 2.16 /* set brightness function */
338 greg 2.18 if (!strcmp(ourfmt, CIEFMT)) {
339     varset(vwhteff, ':', 1.0);
340 greg 2.16 ourbright = xyz_bright;
341 greg 2.18 } else
342     varset(vwhteff, ':', WHTEFFICACY);
343 greg 1.1 }
344    
345    
346     combine() /* combine pictures */
347     {
348 greg 1.8 EPNODE *coldef[3], *brtdef;
349 greg 1.1 COLOR *scanout;
350 greg 1.8 double d;
351 greg 1.1 register int i, j;
352     /* check defined variables */
353 greg 1.4 for (j = 0; j < 3; j++) {
354     if (vardefined(vcolout[j]))
355     coldef[j] = eparse(vcolout[j]);
356     else
357     coldef[j] = NULL;
358     }
359 greg 1.8 if (vardefined(vbrtout))
360     brtdef = eparse(vbrtout);
361     else
362     brtdef = NULL;
363 greg 1.1 /* allocate scanline */
364     scanout = (COLOR *)emalloc(xres*sizeof(COLOR));
365 greg 1.18 /* set input position */
366     yscan = ymax+MIDSCN;
367 greg 1.1 /* combine files */
368     for (ypos = yres-1; ypos >= 0; ypos--) {
369 greg 1.14 advance();
370 greg 1.11 varset(vypos, '=', (double)ypos);
371 greg 1.8 for (xpos = 0; xpos < xres; xpos++) {
372 greg 1.18 xscan = (long)xpos*xmax/xres;
373 greg 1.11 varset(vxpos, '=', (double)xpos);
374 greg 1.8 eclock++;
375     if (brtdef != NULL) {
376     d = evalue(brtdef);
377     if (d < 0.0)
378     d = 0.0;
379     setcolor(scanout[xpos], d, d, d);
380     } else {
381     for (j = 0; j < 3; j++) {
382     if (coldef[j] != NULL) {
383 greg 1.13 d = evalue(coldef[j]);
384 greg 1.8 } else {
385 greg 1.13 d = 0.0;
386 greg 1.8 for (i = 0; i < nfiles; i++)
387 greg 1.18 d += colval(input[i].scan[MIDSCN][xscan],j);
388 greg 1.1 }
389 greg 1.13 if (d < 0.0)
390     d = 0.0;
391     colval(scanout[xpos],j) = d;
392 greg 1.8 }
393 greg 1.1 }
394 greg 1.8 }
395     if (fwritescan(scanout, xres, stdout) < 0) {
396     perror("write error");
397     quit(1);
398     }
399 greg 1.1 }
400     efree(scanout);
401     }
402    
403    
404 greg 1.18 advance() /* read in data for next scanline */
405 greg 1.14 {
406 greg 1.18 int ytarget;
407 greg 1.14 register COLOR *st;
408     register int i, j;
409    
410 greg 1.18 for (ytarget = (long)ypos*ymax/yres; yscan > ytarget; yscan--)
411     for (i = 0; i < nfiles; i++) {
412     st = input[i].scan[WINSIZ-1];
413     for (j = WINSIZ-1; j > 0; j--) /* rotate window */
414     input[i].scan[j] = input[i].scan[j-1];
415     input[i].scan[0] = st;
416     if (yscan <= MIDSCN) /* hit bottom? */
417     continue;
418 greg 2.4 if (freadscan(st, xmax, input[i].fp) < 0) { /* read */
419 greg 1.18 eputs(input[i].name);
420     eputs(": read error\n");
421     quit(1);
422     }
423 greg 2.4 if (fabs(colval(input[i].coef,RED)-1.0) > 1e-3 ||
424     fabs(colval(input[i].coef,GRN)-1.0) > 1e-3 ||
425     fabs(colval(input[i].coef,BLU)-1.0) > 1e-3)
426     for (j = 0; j < xmax; j++) /* adjust color */
427     multcolor(st[j], input[i].coef);
428 greg 1.14 }
429     }
430    
431    
432 greg 1.1 double
433 greg 1.17 l_expos(nam) /* return picture exposure */
434 greg 1.15 register char *nam;
435 greg 1.1 {
436 greg 1.15 register int fn, n;
437 greg 1.14
438 greg 1.18 fn = argument(1) - .5;
439     if (fn < 0 || fn >= nfiles)
440     return(1.0);
441 greg 1.17 if (nam == vbrtexp)
442 greg 2.16 return((*ourbright)(input[fn].expos));
443 greg 1.15 n = 3;
444     while (n--)
445 greg 1.17 if (nam == vcolexp[n])
446     return(colval(input[fn].expos,n));
447     eputs("Bad call to l_expos()!\n");
448 greg 1.15 quit(1);
449     }
450    
451    
452     double
453 greg 2.18 l_pixaspect() /* return pixel aspect ratio */
454     {
455     register int fn;
456    
457     fn = argument(1) - .5;
458     if (fn < 0 || fn >= nfiles)
459     return(1.0);
460     return(input[fn].pa);
461     }
462    
463    
464     double
465 greg 1.15 l_colin(nam) /* return color value for picture */
466     register char *nam;
467     {
468     int fn;
469     register int n, xoff, yoff;
470     double d;
471    
472 greg 2.18 fn = argument(1) - .5;
473 greg 1.16 if (fn < 0 || fn >= nfiles) {
474 greg 1.15 errno = EDOM;
475     return(0.0);
476     }
477 greg 1.14 xoff = yoff = 0;
478     n = nargum();
479     if (n >= 2) {
480     d = argument(2);
481     if (d < 0.0) {
482     xoff = d-.5;
483 greg 1.18 if (xscan+xoff < 0)
484     xoff = -xscan;
485 greg 1.14 } else {
486     xoff = d+.5;
487 greg 1.18 if (xscan+xoff >= xmax)
488     xoff = xmax-1-xscan;
489 greg 1.14 }
490     }
491     if (n >= 3) {
492     d = argument(3);
493     if (d < 0.0) {
494     yoff = d-.5;
495     if (yoff+MIDSCN < 0)
496     yoff = -MIDSCN;
497 greg 1.18 if (yscan+yoff < 0)
498     yoff = -yscan;
499 greg 1.14 } else {
500     yoff = d+.5;
501     if (yoff+MIDSCN >= WINSIZ)
502     yoff = WINSIZ-1-MIDSCN;
503 greg 1.18 if (yscan+yoff >= ymax)
504     yoff = ymax-1-yscan;
505 greg 1.14 }
506     }
507 greg 1.15 if (nam == vbrtin)
508 greg 2.16 return((*ourbright)(input[fn].scan[MIDSCN+yoff][xscan+xoff]));
509 greg 1.15 n = 3;
510     while (n--)
511     if (nam == vcolin[n])
512 greg 1.18 return(colval(input[fn].scan[MIDSCN+yoff][xscan+xoff],n));
513 greg 1.15 eputs("Bad call to l_colin()!\n");
514 greg 2.10 quit(1);
515     }
516    
517    
518     double
519     l_ray(nam) /* return ray origin or direction */
520     register char *nam;
521     {
522 greg 2.15 static unsigned long ltick[MAXINP];
523 greg 2.10 static FVECT lorg[MAXINP], ldir[MAXINP];
524 greg 2.18 static double ldist[MAXINP];
525 greg 2.10 FLOAT loc[2];
526     int fn;
527     register int i;
528    
529 greg 2.18 fn = argument(1) - .5;
530 greg 2.10 if (fn < 0 || fn >= nfiles) {
531     errno = EDOM;
532     return(0.0);
533     }
534     if (ltick[fn] < eclock) { /* need to compute? */
535     lorg[fn][0] = lorg[fn][1] = lorg[fn][2] = 0.0;
536     ldir[fn][0] = ldir[fn][1] = ldir[fn][2] = 0.0;
537 greg 2.20 ldist[fn] = -1.0;
538 greg 2.10 if (input[fn].vw.type == 0)
539 greg 2.12 errno = EDOM;
540 greg 2.10 else {
541 greg 2.11 pix2loc(loc, &input[fn].rs, xscan, ymax-1-yscan);
542 greg 2.18 ldist[fn] = viewray(lorg[fn], ldir[fn],
543     &input[fn].vw, loc[0], loc[1]);
544 greg 2.10 }
545 greg 2.18 ltick[fn] = eclock;
546 greg 2.10 }
547 greg 2.18 if (nam == vray[i=6])
548     return(ldist[fn]);
549 greg 2.10 while (i--)
550     if (nam == vray[i])
551     return(i < 3 ? lorg[fn][i] : ldir[fn][i-3]);
552     eputs("Bad call to l_ray()!\n");
553 greg 1.15 quit(1);
554 greg 2.17 }
555    
556    
557     double
558     l_psize() /* compute pixel size in steradians */
559     {
560     static unsigned long ltick[MAXINP];
561     static double psize[MAXINP];
562 greg 2.18 FVECT dir0, org, dirx, diry;
563     FLOAT locx[2], locy[2];
564 greg 2.17 double d;
565     int fn;
566 greg 2.18 register int i;
567 greg 2.17
568     d = argument(1);
569 greg 2.18 if (d < .5 || d >= nfiles+.5) {
570 greg 2.17 errno = EDOM;
571     return(0.0);
572     }
573 greg 2.18 fn = d - .5;
574     if (ltick[fn] < eclock) { /* need to compute? */
575     psize[fn] = 0.0;
576     if (input[fn].vw.type == 0)
577 greg 2.17 errno = EDOM;
578 greg 2.18 else if (input[fn].vw.type != VT_PAR &&
579     funvalue(vray[6], 1, &d) >= 0) {
580     for (i = 0; i < 3; i++)
581     dir0[i] = funvalue(vray[3+i], 1, &d);
582     pix2loc(locx, &input[fn].rs, xscan+1, ymax-1-yscan);
583     pix2loc(locy, &input[fn].rs, xscan, ymax-yscan);
584     if (viewray(org, dirx, &input[fn].vw,
585     locx[0], locx[1]) >= 0 &&
586     viewray(org, diry, &input[fn].vw,
587     locy[0], locy[1]) >= 0) {
588 greg 2.17 /* approximate solid angle */
589 greg 2.18 for (i = 0; i < 3; i++) {
590     dirx[i] -= dir0[i];
591     diry[i] -= dir0[i];
592     }
593     fcross(dir0, dirx, diry);
594     psize[fn] = 0.5 * sqrt(DOT(dir0,dir0));
595 greg 2.17 }
596     }
597 greg 2.18 ltick[fn] = eclock;
598     }
599 greg 2.17 return(psize[fn]);
600 greg 1.1 }
601    
602    
603     wputs(msg)
604     char *msg;
605     {
606     if (!nowarn)
607     eputs(msg);
608     }
609    
610    
611     eputs(msg)
612     char *msg;
613     {
614     fputs(msg, stderr);
615     }
616    
617    
618 greg 2.4 quit(code) /* exit gracefully */
619     int code;
620     {
621 greg 2.5 register int i;
622     /* close input files */
623     for (i = 0; i < nfiles; i++)
624 greg 2.6 if (input[i].name == Command)
625     pclose(input[i].fp);
626     else
627     fclose(input[i].fp);
628 greg 1.1 exit(code);
629     }