--- ray/src/px/pfilt.c 1991/04/18 14:35:21 1.16 +++ ray/src/px/pfilt.c 1995/05/31 08:50:27 2.14 @@ -1,4 +1,4 @@ -/* Copyright (c) 1986 Regents of the University of California */ +/* Copyright (c) 1992 Regents of the University of California */ #ifndef lint static char SCCSid[] = "$SunId$ LBL"; @@ -8,59 +8,74 @@ static char SCCSid[] = "$SunId$ LBL"; * pfilt.c - program to post-process picture file. * * 9/26/85 + * 6/23/93 Added additional buffers for value spreading */ -#include +#include "standard.h" #include #include "color.h" -extern char *malloc(); +#include "resolu.h" + +#include "paths.h" + extern float *matchlamp(); -#define FEQ(a,b) ((a) >= .98*(b) && (a) <= 1.02*(b)) +#define FEQ(a,b) ((a) >= .98*(b) && (a) <= 1.02*(b)) -#define CHECKRAD 1.5 /* radius to check for filtering */ +double CHECKRAD = 1.5; /* radius to check for filtering */ +#define THRESHRAD 5.0 /* maximum sample spread in output */ + COLOR exposure = WHTCOLOR; /* exposure for the frame */ -double rad = 0.0; /* output pixel radius for filtering */ +double rad = 0.0; /* output pixel radius for filtering */ +double thresh = 0.0; /* maximum contribution for subpixel */ + int nrows = 0; /* number of rows for output */ int ncols = 0; /* number of columns for output */ -double x_c = 1.0; /* ratio of output x size to input */ -double y_r = 1.0; /* ratio of output y size to input */ +double x_c = 1.0; /* ratio of output x size to input */ +double y_r = 1.0; /* ratio of output y size to input */ int singlepass = 0; /* true means skip first pass */ int avghot = 0; /* true means average in bright spots */ -double hotlvl = 1000.0; /* level considered "hot" */ +double hotlvl = 100.0; /* level considered "hot" */ int npts = 0; /* (half) number of points for stars */ -double spread = 1e-4; /* spread for star points */ +double spread = 1e-4; /* spread for star points */ -#define TEMPLATE "/usr/tmp/pfXXXXXX" - char *tfname = NULL; +char template[] = TEMPLATE; + char *lampdat = "lamp.tab"; /* lamp data file */ +int order; /* scanline ordering of input */ int xres, yres; /* resolution of input */ -double inpaspect = 1.0; /* pixel aspect ratio of input */ +double inpaspect = 1.0; /* pixel aspect ratio of input */ int correctaspect = 0; /* aspect ratio correction? */ int wrongformat = 0; -int xrad; /* x window size */ -int yrad; /* y window size */ +int xrad; /* x search radius */ +int yrad; /* y search radius */ +int xbrad; /* x box size */ +int ybrad; /* y box size */ int barsize; /* size of input scan bar */ COLOR **scanin; /* input scan bar */ COLOR *scanout; /* output scan line */ +COLOR **scoutbar; /* output scan bar (if thresh > 0) */ +float **greybar; /* grey-averaged input values */ +int obarsize = 0; /* size of output scan bar */ +int orad = 0; /* output window radius */ char *progname; @@ -69,30 +84,33 @@ main(argc, argv) int argc; char **argv; { - extern char *mktemp(); - extern double atof(), pow(); extern long ftell(); extern int quit(), headline(); FILE *fin; float *lampcolor; char *lamptype = NULL; long fpos; - double outaspect = 0.0; - double d; - int i; - + double outaspect = 0.0; + double d; + int i, j; +#ifdef MSDOS + extern int _fmode; + _fmode = O_BINARY; + setmode(fileno(stdin), O_BINARY); + setmode(fileno(stdout), O_BINARY); +#endif if (signal(SIGINT, quit) == SIG_IGN) signal(SIGINT, SIG_IGN); if (signal(SIGHUP, quit) == SIG_IGN) - signal(SIGINT, SIG_IGN); + signal(SIGHUP, SIG_IGN); signal(SIGTERM, quit); signal(SIGPIPE, quit); -#ifdef SIGXCPU +#ifdef SIGXCPU signal(SIGXCPU, quit); signal(SIGXFSZ, quit); #endif - progname = argv[0]; + progname = argv[0] = fixargv0(argv[0]); for (i = 1; i < argc; i++) if (argv[i][0] == '-') @@ -176,8 +194,13 @@ char **argv; case 'r': rad = atof(argv[++i]); break; + case 'm': + thresh = atof(argv[++i]); + if (rad <= FTINY) + rad = 1.0; + break; case 'b': - rad = 0.0; + rad = thresh = 0.0; break; default:; badopt: @@ -196,9 +219,9 @@ char **argv; fprintf(stderr, "%s: unknown lamp type\n", lamptype); quit(1); } - colval(exposure,RED) /= lampcolor[0]; - colval(exposure,GRN) /= lampcolor[1]; - colval(exposure,BLU) /= lampcolor[2]; + for (j = 0; j < 3; j++) + if (lampcolor[j] > 1e-4) + colval(exposure,j) /= lampcolor[j]; freelamps(); } /* open input file */ @@ -206,7 +229,7 @@ char **argv; if (singlepass) fin = stdin; else { - tfname = mktemp(TEMPLATE); + tfname = mktemp(template); if ((fin = fopen(tfname, "w+")) == NULL) { fprintf(stderr, "%s: can't create ", progname); fprintf(stderr, "temp file \"%s\"\n", tfname); @@ -238,10 +261,12 @@ char **argv; /* add new header info. */ printargs(i, argv, stdout); /* get picture size */ - if (fgetresolu(&xres, &yres, fin) != (YMAJOR|YDECR)) { + if ((order = fgetresolu(&xres, &yres, fin)) < 0) { fprintf(stderr, "%s: bad picture size\n", progname); quit(1); } + if (!(order & YMAJOR)) + inpaspect = 1.0/inpaspect; /* compute output resolution */ if (ncols <= 0) ncols = x_c*xres + .5; @@ -285,6 +310,8 @@ char *s; fputs(s, stdout); /* copy to output */ if (isaspect(s)) /* get aspect ratio */ inpaspect *= aspectval(s); + else if (isexpos(s)) + hotlvl *= exposval(s); else if (isformat(s)) { formatval(fmt, s); wrongformat = strcmp(fmt, COLRFMT); @@ -322,13 +349,13 @@ FILE *in; } for (i = 0; i < yres; i++) { if (freadscan(scan, xres, in) < 0) { - nrows = nrows * i / yres; /* adjust frame */ + nrows = (long)nrows * i / yres; /* adjust frame */ if (nrows <= 0) { fprintf(stderr, "%s: empty frame\n", progname); quit(1); } fprintf(stderr, "%s: warning - partial frame (%d%%)\n", - progname, 100*i/yres); + progname, (int)(100L*i/yres)); yres = i; y_r = (double)nrows/yres; break; @@ -345,7 +372,7 @@ FILE *in; int yread; int ycent, xcent; int r, c; - + pass2init(); scan2init(); yread = 0; @@ -356,7 +383,7 @@ FILE *in; if (freadscan(scanin[yread%barsize], xres, in) < 0) { fprintf(stderr, - "%s: bad read (y=%d)\n", + "%s: truncated input (y=%d)\n", progname, yres-1-yread); quit(1); } @@ -364,62 +391,86 @@ FILE *in; } yread++; } + if (obarsize > 0) + scan2sync(r); for (c = 0; c < ncols; c++) { xcent = (long)c*xres/ncols; - if (rad <= 0.0) - dobox(scanout[c], xcent, ycent, c, r); - else + if (thresh > FTINY) + dothresh(xcent, ycent, c, r); + else if (rad > FTINY) dogauss(scanout[c], xcent, ycent, c, r); + else + dobox(scanout[c], xcent, ycent, c, r); } - if (fwritescan(scanout, ncols, stdout) < 0) { + if (scanout != NULL && fwritescan(scanout, ncols, stdout) < 0) { fprintf(stderr, "%s: write error in pass2\n", progname); quit(1); } } - /* skip leftovers */ + /* skip leftover input */ while (yread < yres) { if (freadscan(scanin[0], xres, in) < 0) break; yread++; } + scan2flush(); /* flush output */ } scan2init() /* prepare scanline arrays */ { COLOR ctmp; - double d; + double d; register int i; - if (rad <= 0.0) { - xrad = xres/ncols/2 + 1; - yrad = yres/nrows/2 + 1; - } else { + xbrad = xres/ncols/2 + 1; + ybrad = yres/nrows/2 + 1; + if (rad > FTINY) { if (nrows >= yres && ncols >= xres) rad *= (y_r + x_c)/2.0; - xrad = CHECKRAD*rad/x_c + 1; - yrad = CHECKRAD*rad/y_r + 1; - + if (thresh > FTINY) { + xrad = CHECKRAD*THRESHRAD*rad/x_c + xbrad; + yrad = CHECKRAD*THRESHRAD*rad/y_r + ybrad; + orad = CHECKRAD*THRESHRAD*rad + 1; + obarsize = 2*orad + 1; + } else { + xrad = CHECKRAD*rad/x_c + 1; + yrad = CHECKRAD*rad/y_r + 1; + } initmask(); /* initialize filter table */ + } else { + xrad = xbrad; + yrad = ybrad; } - barsize = 2 * yrad; + barsize = 2*yrad + 1; scanin = (COLOR **)malloc(barsize*sizeof(COLOR *)); + if (scanin == NULL) + goto memerr; for (i = 0; i < barsize; i++) { scanin[i] = (COLOR *)malloc(xres*sizeof(COLOR)); - if (scanin[i] == NULL) { - fprintf(stderr, "%s: out of memory\n", progname); - quit(1); + if (scanin[i] == NULL) + goto memerr; + } + if (obarsize > 0) { + scoutbar = (COLOR **)malloc(obarsize*sizeof(COLOR *)); + greybar = (float **)malloc(obarsize*sizeof(float *)); + if (scoutbar == NULL | greybar == NULL) + goto memerr; + for (i = 0; i < obarsize; i++) { + scoutbar[i] = (COLOR *)malloc(ncols*sizeof(COLOR)); + greybar[i] = (float *)malloc(ncols*sizeof(float)); + if (scoutbar[i] == NULL | greybar[i] == NULL) + goto memerr; } + } else { + scanout = (COLOR *)malloc(ncols*sizeof(COLOR)); + if (scanout == NULL) + goto memerr; } - scanout = (COLOR *)malloc(ncols*sizeof(COLOR)); - if (scanout == NULL) { - fprintf(stderr, "%s: out of memory\n", progname); - quit(1); - } /* record pixel aspect ratio */ if (!correctaspect) { - d = x_c / y_r; + d = order & YMAJOR ? x_c/y_r : y_r/x_c ; if (!FEQ(d,1.0)) fputaspect(d, stdout); } @@ -435,7 +486,51 @@ scan2init() /* prepare scanline arrays */ fputcolcor(ctmp, stdout); printf("\n"); /* write out resolution */ - fputresolu(YMAJOR|YDECR, ncols, nrows, stdout); + fputresolu(order, ncols, nrows, stdout); + return; +memerr: + fprintf(stderr, "%s: out of memory\n", progname); + quit(1); +} + + +scan2sync(r) /* synchronize grey averages and output scan */ +int r; +{ + static int nextrow = 0; + COLOR ctmp; + int ybot; + register int c; + /* average input scanlines */ + while (nextrow < r+orad && nextrow < nrows) { + ybot = (long)nextrow*yres/nrows; + for (c = 0; c < ncols; c++) { + dobox(ctmp, (int)((long)c*xres/ncols),ybot, c,nextrow); + greybar[nextrow%obarsize][c] = bright(ctmp); + } + /* and zero output scanline */ + bzero((char *)scoutbar[nextrow%obarsize], ncols*sizeof(COLOR)); + nextrow++; + } + /* point to top scanline for output */ + if (r-orad >= 0) + scanout = scoutbar[(r-orad)%obarsize]; + else + scanout = NULL; +} + + +scan2flush() /* flush output buffer */ +{ + register int r; + + for (r = nrows-orad; r < nrows; r++) + if (fwritescan(scoutbar[r%obarsize], ncols, stdout) < 0) + break; + if (fflush(stdout) < 0) { + fprintf(stderr, "%s: write error at end of pass2\n", progname); + exit(1); + } }