ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pcomb.c
Revision: 1.18
Committed: Mon May 27 09:01:05 1991 UTC (32 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 1.17: +91 -68 lines
Log Message:
changed code for output resolution to allow changes from input

File Contents

# User Rev Content
1 greg 1.8 /* Copyright (c) 1991 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     #include <stdio.h>
14    
15     #include <errno.h>
16    
17     #include "color.h"
18    
19     #include "calcomp.h"
20    
21 greg 1.6 #define MAXINP 32 /* maximum number of input files */
22 greg 1.14 #define WINSIZ 9 /* scanline window size */
23     #define MIDSCN 4 /* current scan position */
24 greg 1.1
25 greg 1.14 #define BRT (-1) /* special index for brightness */
26    
27 greg 1.1 struct {
28     char *name; /* file name */
29     FILE *fp; /* stream pointer */
30 greg 1.14 COLOR *scan[WINSIZ]; /* input scanline window */
31 greg 1.3 COLOR coef; /* coefficient */
32 greg 1.17 COLOR expos; /* recorded exposure */
33 greg 1.1 } input[MAXINP]; /* input pictures */
34    
35     int nfiles; /* number of input files */
36    
37     char *vcolin[3] = {"ri", "gi", "bi"};
38     char *vcolout[3] = {"ro", "go", "bo"};
39 greg 1.15 char vbrtin[] = "li";
40     char vbrtout[] = "lo";
41 greg 1.17 char *vcolexp[3] = {"re", "ge", "be"};
42     char vbrtexp[] = "le";
43 greg 1.1
44 greg 1.18 char vnfiles[] = "nfiles";
45     char vxmax[] = "xmax";
46     char vymax[] = "ymax";
47     char vxres[] = "xres";
48     char vyres[] = "yres";
49     char vxpos[] = "x";
50     char vypos[] = "y";
51 greg 1.1
52     int nowarn = 0; /* no warning messages? */
53    
54 greg 1.18 int xmax = 0, ymax = 0; /* input resolution */
55 greg 1.9
56 greg 1.18 int xscan, yscan; /* input position */
57 greg 1.1
58 greg 1.18 int xres, yres; /* output resolution */
59 greg 1.1
60 greg 1.18 int xpos, ypos; /* output position */
61    
62 greg 1.10 int wrongformat = 0;
63 greg 1.1
64 greg 1.10
65 greg 1.1 main(argc, argv)
66     int argc;
67     char *argv[];
68     {
69 greg 1.18 char buf[128];
70     int original;
71 greg 1.3 double f;
72 greg 1.14 int a, i;
73 greg 1.15 /* scan options */
74     for (a = 1; a < argc; a++) {
75 greg 1.1 if (argv[a][0] == '-')
76     switch (argv[a][1]) {
77     case 'x':
78     case 'y':
79 greg 1.18 a++;
80 greg 1.15 continue;
81 greg 1.1 case 'w':
82     nowarn = !nowarn;
83 greg 1.15 continue;
84 greg 1.1 case 'f':
85     case 'e':
86 greg 1.16 a++;
87 greg 1.15 continue;
88 greg 1.1 }
89 greg 1.15 break;
90     }
91     /* process files */
92 greg 1.17 for (nfiles = 0; nfiles < MAXINP; nfiles++) {
93 greg 1.3 setcolor(input[nfiles].coef, 1.0, 1.0, 1.0);
94 greg 1.17 setcolor(input[nfiles].expos, 1.0, 1.0, 1.0);
95     }
96 greg 1.1 nfiles = 0;
97 greg 1.18 original = 0;
98 greg 1.1 for ( ; a < argc; a++) {
99     if (nfiles >= MAXINP) {
100     eputs(argv[0]);
101     eputs(": too many picture files\n");
102     quit(1);
103     }
104 greg 1.3 if (argv[a][0] == '-')
105     switch (argv[a][1]) {
106     case '\0':
107     input[nfiles].name = "<stdin>";
108     input[nfiles].fp = stdin;
109     break;
110 greg 1.17 case 'o':
111     original++;
112     break;
113 greg 1.3 case 's':
114     f = atof(argv[++a]);
115     scalecolor(input[nfiles].coef, f);
116     continue;
117     case 'c':
118     colval(input[nfiles].coef,RED)*=atof(argv[++a]);
119     colval(input[nfiles].coef,GRN)*=atof(argv[++a]);
120     colval(input[nfiles].coef,BLU)*=atof(argv[++a]);
121     continue;
122     default:
123     goto usage;
124     }
125     else {
126 greg 1.1 input[nfiles].name = argv[a];
127     input[nfiles].fp = fopen(argv[a], "r");
128     if (input[nfiles].fp == NULL) {
129 greg 1.6 perror(argv[a]);
130 greg 1.1 quit(1);
131     }
132     }
133 greg 1.16 checkfile();
134 greg 1.18 if (original) {
135     colval(input[nfiles].coef,RED) /=
136     colval(input[nfiles].expos,RED);
137     colval(input[nfiles].coef,GRN) /=
138     colval(input[nfiles].expos,GRN);
139     colval(input[nfiles].coef,BLU) /=
140     colval(input[nfiles].expos,BLU);
141     }
142     nfiles++;
143 greg 1.17 original = 0;
144 greg 1.15 }
145 greg 1.18 init(); /* set constants */
146 greg 1.15 /* go back and get expressions */
147     for (a = 1; a < argc; a++) {
148     if (argv[a][0] == '-')
149     switch (argv[a][1]) {
150     case 'x':
151 greg 1.18 strcpy(buf, vxres);
152     strcat(buf, ":");
153     strcat(buf, argv[++a]);
154     scompile(buf, NULL, 0);
155     continue;
156 greg 1.15 case 'y':
157 greg 1.18 strcpy(buf, vyres);
158     strcat(buf, ":");
159     strcat(buf, argv[++a]);
160     scompile(buf, NULL, 0);
161 greg 1.16 continue;
162 greg 1.15 case 'w':
163     continue;
164     case 'f':
165     fcompile(argv[++a]);
166     continue;
167     case 'e':
168     scompile(argv[++a], NULL, 0);
169     continue;
170     }
171     break;
172     }
173 greg 1.18 /* set/get output resolution */
174     if (!vardefined(vxres))
175     varset(vxres, ':', (double)xmax);
176     if (!vardefined(vyres))
177     varset(vyres, ':', (double)ymax);
178     xres = varvalue(vxres) + .5;
179     yres = varvalue(vyres) + .5;
180     if (xres <= 0 || yres <= 0) {
181     eputs(argv[0]);
182     eputs(": illegal output resolution\n");
183     quit(1);
184     }
185 greg 1.15 /* complete header */
186     printargs(argc, argv, stdout);
187     fputformat(COLRFMT, stdout);
188     putchar('\n');
189     fputresolu(YMAJOR|YDECR, xres, yres, stdout);
190     /* combine pictures */
191     combine();
192     quit(0);
193     usage:
194     eputs("Usage: ");
195     eputs(argv[0]);
196     eputs(
197     " [-w][-h][-x xr][-y yr][-e expr][-f file] [ [-s f][-c r g b] pic ..]\n");
198     quit(1);
199     }
200    
201    
202     tputs(s) /* put out string preceded by a tab */
203     char *s;
204     {
205     char fmt[32];
206     double d;
207     COLOR ctmp;
208    
209     if (isformat(s)) { /* check format */
210     formatval(fmt, s);
211     wrongformat = strcmp(fmt, COLRFMT);
212 greg 1.17 return; /* don't echo */
213     }
214     if (isexpos(s)) { /* exposure */
215     d = exposval(s);
216     scalecolor(input[nfiles].expos, d);
217     } else if (iscolcor(s)) { /* color correction */
218 greg 1.15 colcorval(ctmp, s);
219 greg 1.17 multcolor(input[nfiles].expos, ctmp);
220 greg 1.15 }
221 greg 1.17 /* echo line */
222     putchar('\t');
223     fputs(s, stdout);
224 greg 1.16 }
225 greg 1.15
226 greg 1.16
227     checkfile() /* ready a file */
228     {
229 greg 1.18 int xinp, yinp;
230 greg 1.16 register int i;
231     /* process header */
232     fputs(input[nfiles].name, stdout);
233     fputs(":\n", stdout);
234     getheader(input[nfiles].fp, tputs, NULL);
235     if (wrongformat) {
236     eputs(input[nfiles].name);
237     eputs(": not in Radiance picture format\n");
238     quit(1);
239     }
240 greg 1.18 if (fgetresolu(&xinp, &yinp, input[nfiles].fp) != (YMAJOR|YDECR)) {
241 greg 1.16 eputs(input[nfiles].name);
242     eputs(": bad picture size\n");
243     quit(1);
244     }
245 greg 1.18 if (xmax == 0 && ymax == 0) {
246     xmax = xinp;
247     ymax = yinp;
248     } else if (xinp != xmax || yinp != ymax) {
249 greg 1.16 eputs(input[nfiles].name);
250     eputs(": resolution mismatch\n");
251     quit(1);
252     }
253     /* allocate scanlines */
254     for (i = 0; i < WINSIZ; i++)
255 greg 1.18 input[nfiles].scan[i] = (COLOR *)emalloc(xmax*sizeof(COLOR));
256 greg 1.15 }
257    
258    
259 greg 1.18 init() /* perform final setup */
260 greg 1.15 {
261 greg 1.17 double l_colin(), l_expos();
262 greg 1.16 register int i;
263 greg 1.15 /* define constants */
264     varset(vnfiles, ':', (double)nfiles);
265 greg 1.18 varset(vxmax, ':', (double)xmax);
266     varset(vymax, ':', (double)ymax);
267 greg 1.15 /* set functions */
268     for (i = 0; i < 3; i++) {
269 greg 1.17 funset(vcolexp[i], 1, ':', l_expos);
270 greg 1.15 funset(vcolin[i], 1, '=', l_colin);
271     }
272 greg 1.17 funset(vbrtexp, 1, ':', l_expos);
273 greg 1.15 funset(vbrtin, 1, '=', l_colin);
274 greg 1.1 }
275    
276    
277     combine() /* combine pictures */
278     {
279 greg 1.8 EPNODE *coldef[3], *brtdef;
280 greg 1.1 COLOR *scanout;
281 greg 1.8 double d;
282 greg 1.1 register int i, j;
283     /* check defined variables */
284 greg 1.4 for (j = 0; j < 3; j++) {
285     if (vardefined(vcolout[j]))
286     coldef[j] = eparse(vcolout[j]);
287     else
288     coldef[j] = NULL;
289     }
290 greg 1.8 if (vardefined(vbrtout))
291     brtdef = eparse(vbrtout);
292     else
293     brtdef = NULL;
294 greg 1.1 /* allocate scanline */
295     scanout = (COLOR *)emalloc(xres*sizeof(COLOR));
296 greg 1.18 /* set input position */
297     yscan = ymax+MIDSCN;
298 greg 1.1 /* combine files */
299     for (ypos = yres-1; ypos >= 0; ypos--) {
300 greg 1.14 advance();
301 greg 1.11 varset(vypos, '=', (double)ypos);
302 greg 1.8 for (xpos = 0; xpos < xres; xpos++) {
303 greg 1.18 xscan = (long)xpos*xmax/xres;
304 greg 1.11 varset(vxpos, '=', (double)xpos);
305 greg 1.8 eclock++;
306     if (brtdef != NULL) {
307     d = evalue(brtdef);
308     if (d < 0.0)
309     d = 0.0;
310     setcolor(scanout[xpos], d, d, d);
311     } else {
312     for (j = 0; j < 3; j++) {
313     if (coldef[j] != NULL) {
314 greg 1.13 d = evalue(coldef[j]);
315 greg 1.8 } else {
316 greg 1.13 d = 0.0;
317 greg 1.8 for (i = 0; i < nfiles; i++)
318 greg 1.18 d += colval(input[i].scan[MIDSCN][xscan],j);
319 greg 1.1 }
320 greg 1.13 if (d < 0.0)
321     d = 0.0;
322     colval(scanout[xpos],j) = d;
323 greg 1.8 }
324 greg 1.1 }
325 greg 1.8 }
326     if (fwritescan(scanout, xres, stdout) < 0) {
327     perror("write error");
328     quit(1);
329     }
330 greg 1.1 }
331     efree(scanout);
332     }
333    
334    
335 greg 1.18 advance() /* read in data for next scanline */
336 greg 1.14 {
337 greg 1.18 int ytarget;
338 greg 1.14 register COLOR *st;
339     register int i, j;
340    
341 greg 1.18 for (ytarget = (long)ypos*ymax/yres; yscan > ytarget; yscan--)
342     for (i = 0; i < nfiles; i++) {
343     st = input[i].scan[WINSIZ-1];
344     for (j = WINSIZ-1; j > 0; j--) /* rotate window */
345     input[i].scan[j] = input[i].scan[j-1];
346     input[i].scan[0] = st;
347     if (yscan <= MIDSCN) /* hit bottom? */
348     continue;
349     if (freadscan(st, xmax, input[i].fp) < 0) { /* read */
350     eputs(input[i].name);
351     eputs(": read error\n");
352     quit(1);
353     }
354     for (j = 0; j < xmax; j++) /* adjust color */
355     multcolor(st[j], input[i].coef);
356 greg 1.14 }
357     }
358    
359    
360 greg 1.1 double
361 greg 1.17 l_expos(nam) /* return picture exposure */
362 greg 1.15 register char *nam;
363 greg 1.1 {
364 greg 1.15 register int fn, n;
365 greg 1.14
366 greg 1.18 fn = argument(1) - .5;
367     if (fn < 0 || fn >= nfiles)
368     return(1.0);
369 greg 1.17 if (nam == vbrtexp)
370     return(bright(input[fn].expos));
371 greg 1.15 n = 3;
372     while (n--)
373 greg 1.17 if (nam == vcolexp[n])
374     return(colval(input[fn].expos,n));
375     eputs("Bad call to l_expos()!\n");
376 greg 1.15 quit(1);
377     }
378    
379    
380     double
381     l_colin(nam) /* return color value for picture */
382     register char *nam;
383     {
384     int fn;
385     register int n, xoff, yoff;
386     double d;
387    
388 greg 1.16 d = argument(1);
389     if (d > -.5 && d < .5)
390     return((double)nfiles);
391     fn = d - .5;
392     if (fn < 0 || fn >= nfiles) {
393 greg 1.15 errno = EDOM;
394     return(0.0);
395     }
396 greg 1.14 xoff = yoff = 0;
397     n = nargum();
398     if (n >= 2) {
399     d = argument(2);
400     if (d < 0.0) {
401     xoff = d-.5;
402 greg 1.18 if (xscan+xoff < 0)
403     xoff = -xscan;
404 greg 1.14 } else {
405     xoff = d+.5;
406 greg 1.18 if (xscan+xoff >= xmax)
407     xoff = xmax-1-xscan;
408 greg 1.14 }
409     }
410     if (n >= 3) {
411     d = argument(3);
412     if (d < 0.0) {
413     yoff = d-.5;
414     if (yoff+MIDSCN < 0)
415     yoff = -MIDSCN;
416 greg 1.18 if (yscan+yoff < 0)
417     yoff = -yscan;
418 greg 1.14 } else {
419     yoff = d+.5;
420     if (yoff+MIDSCN >= WINSIZ)
421     yoff = WINSIZ-1-MIDSCN;
422 greg 1.18 if (yscan+yoff >= ymax)
423     yoff = ymax-1-yscan;
424 greg 1.14 }
425     }
426 greg 1.15 if (nam == vbrtin)
427 greg 1.18 return(bright(input[fn].scan[MIDSCN+yoff][xscan+xoff]));
428 greg 1.15 n = 3;
429     while (n--)
430     if (nam == vcolin[n])
431 greg 1.18 return(colval(input[fn].scan[MIDSCN+yoff][xscan+xoff],n));
432 greg 1.15 eputs("Bad call to l_colin()!\n");
433     quit(1);
434 greg 1.1 }
435    
436    
437     wputs(msg)
438     char *msg;
439     {
440     if (!nowarn)
441     eputs(msg);
442     }
443    
444    
445     eputs(msg)
446     char *msg;
447     {
448     fputs(msg, stderr);
449     }
450    
451    
452     quit(code)
453     int code;
454     {
455     exit(code);
456     }