ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pcomb.c
Revision: 1.16
Committed: Fri May 24 15:47:53 1991 UTC (32 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 1.15: +47 -37 lines
Log Message:
bug fixes after last change

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