ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pfilt.c
Revision: 2.16
Committed: Wed Feb 14 20:19:52 1996 UTC (28 years, 2 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.15: +4 -4 lines
Log Message:
fixed bug in indexing and size of buffers for -m option

File Contents

# User Rev Content
1 greg 2.16 /* Copyright (c) 1996 Regents of the University of California */
2 greg 1.1
3     #ifndef lint
4     static char SCCSid[] = "$SunId$ LBL";
5     #endif
6    
7     /*
8     * pfilt.c - program to post-process picture file.
9     *
10     * 9/26/85
11 greg 2.9 * 6/23/93 Added additional buffers for value spreading
12 greg 1.1 */
13    
14 greg 2.6 #include "standard.h"
15 greg 1.1
16     #include <signal.h>
17    
18     #include "color.h"
19    
20 greg 1.19 #include "resolu.h"
21    
22 greg 2.3 #include "paths.h"
23    
24 greg 1.13 extern float *matchlamp();
25 greg 1.1
26 greg 2.4 #define FEQ(a,b) ((a) >= .98*(b) && (a) <= 1.02*(b))
27 greg 1.15
28 greg 2.9 double CHECKRAD = 1.5; /* radius to check for filtering */
29 greg 1.1
30 greg 2.9 #define THRESHRAD 5.0 /* maximum sample spread in output */
31    
32 greg 1.1 COLOR exposure = WHTCOLOR; /* exposure for the frame */
33    
34 greg 2.4 double rad = 0.0; /* output pixel radius for filtering */
35 greg 1.1
36 greg 2.9 double thresh = 0.0; /* maximum contribution for subpixel */
37    
38 greg 1.1 int nrows = 0; /* number of rows for output */
39     int ncols = 0; /* number of columns for output */
40    
41 greg 2.4 double x_c = 1.0; /* ratio of output x size to input */
42     double y_r = 1.0; /* ratio of output y size to input */
43 greg 1.2
44 greg 1.1 int singlepass = 0; /* true means skip first pass */
45    
46     int avghot = 0; /* true means average in bright spots */
47    
48 greg 2.8 double hotlvl = 100.0; /* level considered "hot" */
49 greg 1.1
50     int npts = 0; /* (half) number of points for stars */
51    
52 greg 2.4 double spread = 1e-4; /* spread for star points */
53 greg 1.1
54     char *tfname = NULL;
55    
56 greg 2.11 char template[] = TEMPLATE;
57    
58 greg 1.13 char *lampdat = "lamp.tab"; /* lamp data file */
59    
60 greg 1.19 int order; /* scanline ordering of input */
61 greg 1.1 int xres, yres; /* resolution of input */
62 greg 2.4 double inpaspect = 1.0; /* pixel aspect ratio of input */
63 greg 1.12 int correctaspect = 0; /* aspect ratio correction? */
64 greg 1.1
65 greg 1.16 int wrongformat = 0;
66    
67 greg 2.9 int xrad; /* x search radius */
68     int yrad; /* y search radius */
69     int xbrad; /* x box size */
70     int ybrad; /* y box size */
71 greg 1.1
72     int barsize; /* size of input scan bar */
73     COLOR **scanin; /* input scan bar */
74     COLOR *scanout; /* output scan line */
75 greg 2.9 COLOR **scoutbar; /* output scan bar (if thresh > 0) */
76     float **greybar; /* grey-averaged input values */
77     int obarsize = 0; /* size of output scan bar */
78     int orad = 0; /* output window radius */
79 greg 1.1
80     char *progname;
81    
82    
83     main(argc, argv)
84     int argc;
85     char **argv;
86     {
87     extern long ftell();
88 greg 1.9 extern int quit(), headline();
89 greg 1.1 FILE *fin;
90 greg 1.13 float *lampcolor;
91     char *lamptype = NULL;
92 greg 1.1 long fpos;
93 greg 2.4 double outaspect = 0.0;
94     double d;
95 greg 2.13 int i, j;
96 greg 2.4 #ifdef MSDOS
97     extern int _fmode;
98     _fmode = O_BINARY;
99     setmode(fileno(stdin), O_BINARY);
100     setmode(fileno(stdout), O_BINARY);
101     #endif
102 greg 1.1 if (signal(SIGINT, quit) == SIG_IGN)
103     signal(SIGINT, SIG_IGN);
104     if (signal(SIGHUP, quit) == SIG_IGN)
105 greg 2.11 signal(SIGHUP, SIG_IGN);
106 greg 1.1 signal(SIGTERM, quit);
107     signal(SIGPIPE, quit);
108 greg 2.4 #ifdef SIGXCPU
109 greg 1.1 signal(SIGXCPU, quit);
110     signal(SIGXFSZ, quit);
111     #endif
112    
113 greg 2.6 progname = argv[0] = fixargv0(argv[0]);
114 greg 1.1
115     for (i = 1; i < argc; i++)
116     if (argv[i][0] == '-')
117     switch (argv[i][1]) {
118     case 'x':
119 greg 1.3 i++;
120 greg 1.5 if (argv[i][0] == '/') {
121 greg 1.4 x_c = 1.0/atof(argv[i]+1);
122 greg 1.5 ncols = 0;
123     } else
124 greg 1.3 ncols = atoi(argv[i]);
125 greg 1.1 break;
126     case 'y':
127 greg 1.3 i++;
128 greg 1.5 if (argv[i][0] == '/') {
129 greg 1.4 y_r = 1.0/atof(argv[i]+1);
130 greg 1.5 nrows = 0;
131     } else
132 greg 1.3 nrows = atoi(argv[i]);
133 greg 1.1 break;
134 greg 1.12 case 'c':
135     correctaspect = !correctaspect;
136     break;
137 greg 1.9 case 'p':
138     i++;
139     outaspect = atof(argv[i]);
140     break;
141 greg 1.1 case 'e':
142     if (argv[i+1][0] == '+' || argv[i+1][0] == '-')
143     d = pow(2.0, atof(argv[i+1]));
144     else
145     d = atof(argv[i+1]);
146 greg 1.15 if (d < 1e-20 || d > 1e20) {
147     fprintf(stderr,
148     "%s: exposure out of range\n",
149     argv[0]);
150     exit(1);
151     }
152 greg 1.1 switch (argv[i][2]) {
153     case '\0':
154     scalecolor(exposure, d);
155     break;
156     case 'r':
157     colval(exposure,RED) *= d;
158     break;
159     case 'g':
160     colval(exposure,GRN) *= d;
161     break;
162     case 'b':
163     colval(exposure,BLU) *= d;
164     break;
165     default:
166     goto badopt;
167     }
168     i++;
169     break;
170 greg 1.13 case 'f':
171     lampdat = argv[++i];
172     break;
173     case 't':
174     lamptype = argv[++i];
175     break;
176 greg 1.1 case '1':
177     singlepass = 1;
178     break;
179 greg 1.6 case '2':
180     singlepass = 0;
181     break;
182 greg 1.9 case 'n':
183 greg 1.1 npts = atoi(argv[++i]) / 2;
184     break;
185     case 's':
186     spread = atof(argv[++i]);
187     break;
188     case 'a':
189     avghot = !avghot;
190     break;
191     case 'h':
192     hotlvl = atof(argv[++i]);
193     break;
194     case 'r':
195     rad = atof(argv[++i]);
196     break;
197 greg 2.9 case 'm':
198     thresh = atof(argv[++i]);
199     if (rad <= FTINY)
200     rad = 1.0;
201     break;
202 greg 1.1 case 'b':
203 greg 2.9 rad = thresh = 0.0;
204 greg 1.1 break;
205     default:;
206     badopt:
207     fprintf(stderr, "%s: unknown option: %s\n",
208     progname, argv[i]);
209     quit(1);
210     break;
211     }
212     else
213     break;
214 greg 1.14 /* get lamp data (if necessary) */
215     if (lamptype != NULL) {
216     if (loadlamps(lampdat) < 0)
217     quit(1);
218     if ((lampcolor = matchlamp(lamptype)) == NULL) {
219     fprintf(stderr, "%s: unknown lamp type\n", lamptype);
220     quit(1);
221     }
222 greg 2.13 for (j = 0; j < 3; j++)
223     if (lampcolor[j] > 1e-4)
224     colval(exposure,j) /= lampcolor[j];
225 greg 1.14 freelamps();
226     }
227     /* open input file */
228 greg 1.1 if (i == argc) {
229     if (singlepass)
230     fin = stdin;
231     else {
232 greg 2.11 tfname = mktemp(template);
233 greg 1.1 if ((fin = fopen(tfname, "w+")) == NULL) {
234     fprintf(stderr, "%s: can't create ", progname);
235     fprintf(stderr, "temp file \"%s\"\n", tfname);
236     quit(1);
237     }
238     copyfile(stdin, fin);
239     if (fseek(fin, 0L, 0) == -1) {
240     fprintf(stderr, "%s: seek fail\n", progname);
241     quit(1);
242     }
243     }
244     } else if (i == argc-1) {
245     if ((fin = fopen(argv[i], "r")) == NULL) {
246     fprintf(stderr, "%s: can't open file \"%s\"\n",
247     progname, argv[i]);
248     quit(1);
249     }
250     } else {
251     fprintf(stderr, "%s: bad # file arguments\n", progname);
252     quit(1);
253     }
254 greg 1.9 /* get header */
255 greg 1.16 getheader(fin, headline, NULL);
256     if (wrongformat) {
257     fprintf(stderr, "%s: input must be a Radiance picture\n",
258     progname);
259     quit(1);
260     }
261 greg 1.1 /* add new header info. */
262     printargs(i, argv, stdout);
263     /* get picture size */
264 greg 1.19 if ((order = fgetresolu(&xres, &yres, fin)) < 0) {
265 greg 1.1 fprintf(stderr, "%s: bad picture size\n", progname);
266     quit(1);
267     }
268 greg 1.19 if (!(order & YMAJOR))
269     inpaspect = 1.0/inpaspect;
270 greg 1.9 /* compute output resolution */
271     if (ncols <= 0)
272 greg 1.4 ncols = x_c*xres + .5;
273 greg 1.9 if (nrows <= 0)
274 greg 1.4 nrows = y_r*yres + .5;
275 greg 1.9 if (outaspect > .01) {
276     d = inpaspect * yres/xres / outaspect;
277     if (d * ncols > nrows)
278     ncols = nrows / d;
279     else
280     nrows = ncols * d;
281     }
282     x_c = (double)ncols/xres;
283     y_r = (double)nrows/yres;
284 greg 1.1
285 greg 1.9 if (singlepass) { /* skip exposure, etc. */
286 greg 1.1 pass1default();
287     pass2(fin);
288     quit(0);
289     }
290    
291     fpos = ftell(fin); /* save input file position */
292    
293     pass1(fin);
294    
295     if (fseek(fin, fpos, 0) == -1) {
296     fprintf(stderr, "%s: seek fail\n", progname);
297     quit(1);
298     }
299     pass2(fin);
300    
301     quit(0);
302     }
303    
304    
305 greg 2.15 double
306     rgb_bright(clr)
307     COLOR clr;
308     {
309     return(bright(clr));
310     }
311    
312    
313     double
314     xyz_bright(clr)
315     COLOR clr;
316     {
317     return(clr[CIEY]);
318     }
319    
320    
321     double (*ourbright)() = rgb_bright;
322    
323    
324 greg 1.9 headline(s) /* process line from header */
325     char *s;
326     {
327 greg 1.16 char fmt[32];
328    
329 greg 1.9 fputs(s, stdout); /* copy to output */
330     if (isaspect(s)) /* get aspect ratio */
331     inpaspect *= aspectval(s);
332 greg 2.15 else if (isexpos(s)) /* get exposure */
333 greg 2.8 hotlvl *= exposval(s);
334 greg 2.15 else if (formatval(fmt, s)) { /* get format */
335     wrongformat = 0;
336     if (!strcmp(COLRFMT, fmt))
337     ourbright = rgb_bright;
338     else if (!strcmp(CIEFMT, fmt))
339     ourbright = xyz_bright;
340     else
341     wrongformat = !globmatch(PICFMT, fmt);
342 greg 1.16 }
343 greg 1.9 }
344    
345    
346 greg 1.1 copyfile(in, out) /* copy a file */
347     register FILE *in, *out;
348     {
349     register int c;
350    
351     while ((c = getc(in)) != EOF)
352     putc(c, out);
353    
354     if (ferror(out)) {
355     fprintf(stderr, "%s: write error in copyfile\n", progname);
356     quit(1);
357     }
358     }
359    
360    
361     pass1(in) /* first pass of picture file */
362     FILE *in;
363     {
364     int i;
365     COLOR *scan;
366    
367     pass1init();
368    
369     scan = (COLOR *)malloc(xres*sizeof(COLOR));
370     if (scan == NULL) {
371     fprintf(stderr, "%s: out of memory\n", progname);
372     quit(1);
373     }
374     for (i = 0; i < yres; i++) {
375     if (freadscan(scan, xres, in) < 0) {
376 greg 2.7 nrows = (long)nrows * i / yres; /* adjust frame */
377 greg 1.1 if (nrows <= 0) {
378     fprintf(stderr, "%s: empty frame\n", progname);
379     quit(1);
380     }
381     fprintf(stderr, "%s: warning - partial frame (%d%%)\n",
382 greg 2.7 progname, (int)(100L*i/yres));
383 greg 1.1 yres = i;
384 greg 1.11 y_r = (double)nrows/yres;
385 greg 1.1 break;
386     }
387     pass1scan(scan, i);
388     }
389     free((char *)scan);
390     }
391    
392    
393     pass2(in) /* last pass on file, write to stdout */
394     FILE *in;
395     {
396     int yread;
397     int ycent, xcent;
398     int r, c;
399 greg 2.4
400 greg 1.1 pass2init();
401     scan2init();
402     yread = 0;
403     for (r = 0; r < nrows; r++) {
404     ycent = (long)r*yres/nrows;
405     while (yread <= ycent+yrad) {
406     if (yread < yres) {
407     if (freadscan(scanin[yread%barsize],
408     xres, in) < 0) {
409     fprintf(stderr,
410 greg 2.14 "%s: truncated input (y=%d)\n",
411 greg 1.1 progname, yres-1-yread);
412     quit(1);
413     }
414     pass2scan(scanin[yread%barsize], yread);
415     }
416     yread++;
417     }
418 greg 2.9 if (obarsize > 0)
419     scan2sync(r);
420 greg 1.1 for (c = 0; c < ncols; c++) {
421     xcent = (long)c*xres/ncols;
422 greg 2.9 if (thresh > FTINY)
423     dothresh(xcent, ycent, c, r);
424     else if (rad > FTINY)
425     dogauss(scanout[c], xcent, ycent, c, r);
426     else
427 greg 1.1 dobox(scanout[c], xcent, ycent, c, r);
428     }
429 greg 2.9 if (scanout != NULL && fwritescan(scanout, ncols, stdout) < 0) {
430 greg 1.1 fprintf(stderr, "%s: write error in pass2\n", progname);
431     quit(1);
432     }
433     }
434 greg 2.9 /* skip leftover input */
435 greg 1.10 while (yread < yres) {
436     if (freadscan(scanin[0], xres, in) < 0)
437     break;
438     yread++;
439     }
440 greg 2.9 scan2flush(); /* flush output */
441 greg 1.1 }
442    
443    
444     scan2init() /* prepare scanline arrays */
445     {
446 greg 1.15 COLOR ctmp;
447 greg 2.4 double d;
448 greg 1.1 register int i;
449    
450 greg 2.9 xbrad = xres/ncols/2 + 1;
451     ybrad = yres/nrows/2 + 1;
452     if (rad > FTINY) {
453 greg 1.1 if (nrows >= yres && ncols >= xres)
454     rad *= (y_r + x_c)/2.0;
455    
456 greg 2.9 if (thresh > FTINY) {
457     orad = CHECKRAD*THRESHRAD*rad + 1;
458 greg 2.16 xrad = orad/x_c + xbrad;
459     yrad = orad/y_r + ybrad;
460 greg 2.9 obarsize = 2*orad + 1;
461     } else {
462     xrad = CHECKRAD*rad/x_c + 1;
463     yrad = CHECKRAD*rad/y_r + 1;
464     }
465 greg 1.1 initmask(); /* initialize filter table */
466 greg 2.9 } else {
467     xrad = xbrad;
468     yrad = ybrad;
469 greg 1.1 }
470 greg 1.17 barsize = 2*yrad + 1;
471 greg 1.1 scanin = (COLOR **)malloc(barsize*sizeof(COLOR *));
472 greg 2.10 if (scanin == NULL)
473     goto memerr;
474 greg 1.1 for (i = 0; i < barsize; i++) {
475     scanin[i] = (COLOR *)malloc(xres*sizeof(COLOR));
476 greg 2.10 if (scanin[i] == NULL)
477     goto memerr;
478 greg 1.1 }
479 greg 2.9 if (obarsize > 0) {
480     scoutbar = (COLOR **)malloc(obarsize*sizeof(COLOR *));
481     greybar = (float **)malloc(obarsize*sizeof(float *));
482 greg 2.10 if (scoutbar == NULL | greybar == NULL)
483     goto memerr;
484 greg 2.9 for (i = 0; i < obarsize; i++) {
485     scoutbar[i] = (COLOR *)malloc(ncols*sizeof(COLOR));
486     greybar[i] = (float *)malloc(ncols*sizeof(float));
487 greg 2.10 if (scoutbar[i] == NULL | greybar[i] == NULL)
488     goto memerr;
489 greg 2.9 }
490     } else {
491     scanout = (COLOR *)malloc(ncols*sizeof(COLOR));
492 greg 2.10 if (scanout == NULL)
493     goto memerr;
494 greg 1.1 }
495 greg 1.15 /* record pixel aspect ratio */
496 greg 1.12 if (!correctaspect) {
497 greg 1.19 d = order & YMAJOR ? x_c/y_r : y_r/x_c ;
498 greg 1.15 if (!FEQ(d,1.0))
499 greg 1.12 fputaspect(d, stdout);
500     }
501 greg 1.15 /* record exposure */
502 greg 2.15 d = (*ourbright)(exposure);
503 greg 1.15 if (!FEQ(d,1.0))
504 greg 1.9 fputexpos(d, stdout);
505 greg 1.15 /* record color correction */
506     copycolor(ctmp, exposure);
507     scalecolor(ctmp, 1.0/d);
508     if (!FEQ(colval(ctmp,RED),colval(ctmp,GRN)) ||
509     !FEQ(colval(ctmp,GRN),colval(ctmp,BLU)))
510     fputcolcor(ctmp, stdout);
511 greg 1.1 printf("\n");
512 greg 1.15 /* write out resolution */
513 greg 1.19 fputresolu(order, ncols, nrows, stdout);
514 greg 2.10 return;
515     memerr:
516     fprintf(stderr, "%s: out of memory\n", progname);
517     quit(1);
518 greg 2.9 }
519    
520    
521     scan2sync(r) /* synchronize grey averages and output scan */
522     int r;
523     {
524     static int nextrow = 0;
525     COLOR ctmp;
526     int ybot;
527     register int c;
528     /* average input scanlines */
529 greg 2.16 while (nextrow <= r+orad && nextrow < nrows) {
530 greg 2.9 ybot = (long)nextrow*yres/nrows;
531     for (c = 0; c < ncols; c++) {
532     dobox(ctmp, (int)((long)c*xres/ncols),ybot, c,nextrow);
533 greg 2.15 greybar[nextrow%obarsize][c] = (*ourbright)(ctmp);
534 greg 2.9 }
535     /* and zero output scanline */
536     bzero((char *)scoutbar[nextrow%obarsize], ncols*sizeof(COLOR));
537     nextrow++;
538     }
539     /* point to top scanline for output */
540     if (r-orad >= 0)
541     scanout = scoutbar[(r-orad)%obarsize];
542     else
543     scanout = NULL;
544     }
545    
546    
547     scan2flush() /* flush output buffer */
548     {
549     register int r;
550    
551     for (r = nrows-orad; r < nrows; r++)
552     if (fwritescan(scoutbar[r%obarsize], ncols, stdout) < 0)
553     break;
554     if (fflush(stdout) < 0) {
555     fprintf(stderr, "%s: write error at end of pass2\n", progname);
556     exit(1);
557     }
558 greg 1.1 }
559    
560    
561     quit(code) /* remove temporary file and exit */
562     int code;
563     {
564     if (tfname != NULL)
565     unlink(tfname);
566     exit(code);
567     }