ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/px/pfilt.c
Revision: 2.36
Committed: Fri Dec 8 17:56:26 2023 UTC (4 months, 2 weeks ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.35: +45 -28 lines
Log Message:
feat(pfilt): Added support for hyperspectral image filtering

File Contents

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