--- ray/src/px/pcomb.c 1989/07/11 18:05:05 1.4 +++ ray/src/px/pcomb.c 2007/11/07 05:40:06 2.42 @@ -1,98 +1,137 @@ -/* Copyright (c) 1989 Regents of the University of California */ - #ifndef lint -static char SCCSid[] = "$SunId$ LBL"; +static const char RCSid[] = "$Id: pcomb.c,v 2.42 2007/11/07 05:40:06 greg Exp $"; #endif - /* * Combine picture files according to calcomp functions. * * 1/4/89 */ -#include - -#include - +#include "platform.h" +#include "standard.h" +#include "rtprocess.h" #include "color.h" - #include "calcomp.h" +#include "view.h" -#define MAXINP 16 /* maximum number of input files */ +#define MAXINP 512 /* maximum number of input files */ +#define WINSIZ 64 /* scanline window size */ +#define MIDSCN ((WINSIZ-1)/2+1) struct { - char *name; /* file name */ + char *name; /* file or command name */ FILE *fp; /* stream pointer */ - COLOR *scan; /* input scanline */ + VIEW vw; /* view for picture */ + RESOLU rs; /* image resolution and orientation */ + float pa; /* pixel aspect ratio */ + COLOR *scan[WINSIZ]; /* input scanline window */ COLOR coef; /* coefficient */ + COLOR expos; /* recorded exposure */ } input[MAXINP]; /* input pictures */ int nfiles; /* number of input files */ -char *vcolin[3] = {"ri", "gi", "bi"}; -char *vcolout[3] = {"ro", "go", "bo"}; +char ourfmt[LPICFMT+1] = PICFMT; /* input picture format */ -#define vxpos "x" -#define vypos "y" +char StandardInput[] = ""; +char Command[] = ""; +char vcolin[3][4] = {"ri", "gi", "bi"}; +char vcolout[3][4] = {"ro", "go", "bo"}; +char vbrtin[] = "li"; +char vbrtout[] = "lo"; +char vcolexp[3][4] = {"re", "ge", "be"}; +char vbrtexp[] = "le"; +char vpixaspect[] = "pa"; +char vray[7][4] = {"Ox", "Oy", "Oz", "Dx", "Dy", "Dz", "T"}; + +char vpsize[] = "S"; + +char vnfiles[] = "nfiles"; +char vwhteff[] = "WE"; +char vxmax[] = "xmax"; +char vymax[] = "ymax"; +char vxres[] = "xres"; +char vyres[] = "yres"; +char vxpos[] = "x"; +char vypos[] = "y"; + int nowarn = 0; /* no warning messages? */ -int xres=0, yres=0; /* picture resolution */ +int xmax = 0, ymax = 0; /* input resolution */ -int xpos, ypos; /* picture position */ +int xscan, yscan; /* input position */ +int xres, yres; /* output resolution */ -tputs(s) /* put out string preceded by a tab */ -char *s; -{ - putchar('\t'); - fputs(s, stdout); -} +int xpos, ypos; /* output position */ +char *progname; /* global argv[0] */ -main(argc, argv) -int argc; -char *argv[]; +int echoheader = 1; +int wrongformat = 0; +int gotview; + + +static gethfunc headline; +static void checkfile(void); +static double rgb_bright(COLOR clr); +static double xyz_bright(COLOR clr); +static void init(void); +static void combine(void); +static void advance(void); +static double l_expos(char *nam); +static double l_pixaspect(char *nm); +static double l_colin(char *nam); +static double l_ray(char *nam); +static double l_psize(char *nm); + + +int +main( + int argc, + char *argv[] +) { - extern double l_redin(), l_grnin(), l_bluin(), atof(); + int original; double f; int a; - - funset(vcolin[RED], 1, l_redin); - funset(vcolin[GRN], 1, l_grnin); - funset(vcolin[BLU], 1, l_bluin); - - for (a = 1; a < argc; a++) + SET_DEFAULT_BINARY(); + SET_FILE_BINARY(stdin); + SET_FILE_BINARY(stdout); + progname = argv[0]; + /* scan options */ + for (a = 1; a < argc; a++) { if (argv[a][0] == '-') switch (argv[a][1]) { - case '\0': - case 's': - case 'c': - goto getfiles; case 'x': - xres = atoi(argv[++a]); - break; case 'y': - yres = atoi(argv[++a]); - break; + a++; + continue; case 'w': nowarn = !nowarn; - break; + continue; + case 'h': + echoheader = !echoheader; + continue; case 'f': - fcompile(argv[++a]); - break; case 'e': - scompile(NULL, argv[++a]); - break; - default: - goto usage; + a++; + continue; } - else - break; -getfiles: - for (nfiles = 0; nfiles < MAXINP; nfiles++) + break; + } + newheader("RADIANCE", stdout); /* start header */ + fputnow(stdout); + /* process files */ + for (nfiles = 0; nfiles < MAXINP; nfiles++) { setcolor(input[nfiles].coef, 1.0, 1.0, 1.0); + setcolor(input[nfiles].expos, 1.0, 1.0, 1.0); + input[nfiles].vw = stdview; + input[nfiles].pa = 1.0; + } nfiles = 0; + original = 0; for ( ; a < argc; a++) { if (nfiles >= MAXINP) { eputs(argv[0]); @@ -102,9 +141,12 @@ getfiles: if (argv[a][0] == '-') switch (argv[a][1]) { case '\0': - input[nfiles].name = ""; + input[nfiles].name = StandardInput; input[nfiles].fp = stdin; break; + case 'o': + original++; + continue; case 's': f = atof(argv[++a]); scalecolor(input[nfiles].coef, f); @@ -118,50 +160,223 @@ getfiles: goto usage; } else { - input[nfiles].name = argv[a]; - input[nfiles].fp = fopen(argv[a], "r"); + if (argv[a][0] == '!') { + input[nfiles].name = Command; + input[nfiles].fp = popen(argv[a]+1, "r"); + } else { + input[nfiles].name = argv[a]; + input[nfiles].fp = fopen(argv[a], "r"); + } if (input[nfiles].fp == NULL) { - eputs(argv[a]); - eputs(": cannot open\n"); + perror(argv[a]); quit(1); } } - fputs(input[nfiles].name, stdout); - fputs(":\n", stdout); - getheader(input[nfiles].fp, tputs); - if (fscanf(input[nfiles].fp, "-Y %d +X %d\n", &ypos, &xpos) != 2) { - eputs(input[nfiles].name); - eputs(": bad picture size\n"); - quit(1); + checkfile(); + if (original) { + colval(input[nfiles].coef,RED) /= + colval(input[nfiles].expos,RED); + colval(input[nfiles].coef,GRN) /= + colval(input[nfiles].expos,GRN); + colval(input[nfiles].coef,BLU) /= + colval(input[nfiles].expos,BLU); + setcolor(input[nfiles].expos, 1.0, 1.0, 1.0); } - if (xres == 0 && yres == 0) { - xres = xpos; - yres = ypos; - } else if (xpos != xres || ypos != yres) { - eputs(argv[0]); - eputs(": resolution mismatch\n"); - quit(1); - } - input[nfiles].scan = (COLOR *)emalloc(xres*sizeof(COLOR)); nfiles++; + original = 0; } + init(); /* set constants */ + /* go back and get expressions */ + for (a = 1; a < argc; a++) { + if (argv[a][0] == '-') + switch (argv[a][1]) { + case 'x': + varset(vxres, ':', eval(argv[++a])); + continue; + case 'y': + varset(vyres, ':', eval(argv[++a])); + continue; + case 'w': + continue; + case 'h': + continue; + case 'f': + fcompile(argv[++a]); + continue; + case 'e': + scompile(argv[++a], NULL, 0); + continue; + } + break; + } + /* set/get output resolution */ + if (!vardefined(vxres)) + varset(vxres, ':', (double)xmax); + if (!vardefined(vyres)) + varset(vyres, ':', (double)ymax); + xres = varvalue(vxres) + .5; + yres = varvalue(vyres) + .5; + if (xres <= 0 || yres <= 0) { + eputs(argv[0]); + eputs(": illegal output resolution\n"); + quit(1); + } + /* complete header */ printargs(argc, argv, stdout); + if (strcmp(ourfmt, PICFMT)) + fputformat(ourfmt, stdout); /* print format if known */ putchar('\n'); - printf("-Y %d +X %d\n", yres, xres); + fprtresolu(xres, yres, stdout); + /* combine pictures */ combine(); quit(0); usage: eputs("Usage: "); eputs(argv[0]); -eputs(" [-w][-x xr][-y yr][-e expr][-f file] [ [-s f][-c r g b] picture ..]\n"); + eputs( +" [-w][-h][-x xr][-y yr][-e expr][-f file] [ [-o][-s f][-c r g b] pic ..]\n"); quit(1); + return 1; /* pro forma return */ } -combine() /* combine pictures */ +static int +headline( /* check header line & echo if requested */ + char *s, + void *p +) { - EPNODE *coldef[3]; + char fmt[32]; + double d; + COLOR ctmp; + + if (isheadid(s)) /* header id */ + return(0); /* don't echo */ + if (formatval(fmt, s)) { /* check format */ + if (globmatch(ourfmt, fmt)) { + wrongformat = 0; + strcpy(ourfmt, fmt); + } else + wrongformat = globmatch(PICFMT, fmt) ? 1 : -1; + return(0); /* don't echo */ + } + if (isexpos(s)) { /* exposure */ + d = exposval(s); + scalecolor(input[nfiles].expos, d); + } else if (iscolcor(s)) { /* color correction */ + colcorval(ctmp, s); + multcolor(input[nfiles].expos, ctmp); + } else if (isaspect(s)) + input[nfiles].pa *= aspectval(s); + else if (isview(s) && sscanview(&input[nfiles].vw, s) > 0) + gotview++; + + if (echoheader) { /* echo line */ + putchar('\t'); + return(fputs(s, stdout)); + } + return(0); +} + + +static void +checkfile(void) /* ready a file */ +{ + register int i; + /* process header */ + gotview = 0; + if (echoheader) { + fputs(input[nfiles].name, stdout); + fputs(":\n", stdout); + } + getheader(input[nfiles].fp, headline, NULL); + if (wrongformat < 0) { + eputs(input[nfiles].name); + eputs(": not a Radiance picture\n"); + quit(1); + } + if (wrongformat > 0) { + wputs(input[nfiles].name); + wputs(": warning -- incompatible picture format\n"); + } + if (!gotview || setview(&input[nfiles].vw) != NULL) + input[nfiles].vw.type = 0; + if (!fgetsresolu(&input[nfiles].rs, input[nfiles].fp)) { + eputs(input[nfiles].name); + eputs(": bad picture size\n"); + quit(1); + } + if (xmax == 0 && ymax == 0) { + xmax = scanlen(&input[nfiles].rs); + ymax = numscans(&input[nfiles].rs); + } else if (scanlen(&input[nfiles].rs) != xmax || + numscans(&input[nfiles].rs) != ymax) { + eputs(input[nfiles].name); + eputs(": resolution mismatch\n"); + quit(1); + } + /* allocate scanlines */ + for (i = 0; i < WINSIZ; i++) + input[nfiles].scan[i] = (COLOR *)emalloc(xmax*sizeof(COLOR)); +} + + +static double +rgb_bright( + COLOR clr +) +{ + return(bright(clr)); +} + + +static double +xyz_bright( + COLOR clr +) +{ + return(clr[CIEY]); +} + + +double (*ourbright)() = rgb_bright; + + +static void +init(void) /* perform final setup */ +{ + register int i; + /* define constants */ + varset("PI", ':', PI); + varset(vnfiles, ':', (double)nfiles); + varset(vxmax, ':', (double)xmax); + varset(vymax, ':', (double)ymax); + /* set functions */ + for (i = 0; i < 3; i++) { + funset(vcolexp[i], 1, ':', l_expos); + funset(vcolin[i], 1, '=', l_colin); + } + funset(vbrtexp, 1, ':', l_expos); + funset(vbrtin, 1, '=', l_colin); + funset(vpixaspect, 1, ':', l_pixaspect); + for (i = 0; i < 7; i++) + funset(vray[i], 1, '=', l_ray); + funset(vpsize, 1, '=', l_psize); + /* set brightness function */ + if (!strcmp(ourfmt, CIEFMT)) { + varset(vwhteff, ':', 1.0); + ourbright = xyz_bright; + } else + varset(vwhteff, ':', WHTEFFICACY); +} + + +static void +combine(void) /* combine pictures */ +{ + EPNODE *coldef[3], *brtdef; COLOR *scanout; + double d; register int i, j; /* check defined variables */ for (j = 0; j < 3; j++) { @@ -170,96 +385,299 @@ combine() /* combine pictures */ else coldef[j] = NULL; } + if (vardefined(vbrtout)) + brtdef = eparse(vbrtout); + else + brtdef = NULL; /* allocate scanline */ scanout = (COLOR *)emalloc(xres*sizeof(COLOR)); + /* set input position */ + yscan = ymax+MIDSCN; /* combine files */ for (ypos = yres-1; ypos >= 0; ypos--) { - for (i = 0; i < nfiles; i++) - if (freadscan(input[i].scan, xres, input[i].fp) < 0) { + advance(); + varset(vypos, '=', (double)ypos); + for (xpos = 0; xpos < xres; xpos++) { + xscan = (xpos+.5)*xmax/xres; + varset(vxpos, '=', (double)xpos); + eclock++; + if (brtdef != NULL) { + d = evalue(brtdef); + if (d < 0.0) + d = 0.0; + setcolor(scanout[xpos], d, d, d); + } else { + for (j = 0; j < 3; j++) { + if (coldef[j] != NULL) { + d = evalue(coldef[j]); + } else { + d = 0.0; + for (i = 0; i < nfiles; i++) + d += colval(input[i].scan[MIDSCN][xscan],j); + } + if (d < 0.0) + d = 0.0; + colval(scanout[xpos],j) = d; + } + } + } + if (fwritescan(scanout, xres, stdout) < 0) { + perror("write error"); + quit(1); + } + } + efree((char *)scanout); +} + + +static void +advance(void) /* read in data for next scanline */ +{ + int ytarget; + register COLOR *st; + register int i, j; + + for (ytarget = (ypos+.5)*ymax/yres; yscan > ytarget; yscan--) + for (i = 0; i < nfiles; i++) { + st = input[i].scan[WINSIZ-1]; + for (j = WINSIZ-1; j > 0; j--) /* rotate window */ + input[i].scan[j] = input[i].scan[j-1]; + input[i].scan[0] = st; + if (yscan <= MIDSCN) /* hit bottom? */ + continue; + if (freadscan(st, xmax, input[i].fp) < 0) { /* read */ eputs(input[i].name); eputs(": read error\n"); quit(1); } - varset(vypos, (double)ypos); - for (xpos = 0; xpos < xres; xpos++) { - varset(vxpos, (double)xpos); - eclock++; - for (j = 0; j < 3; j++) { - if (coldef[j] != NULL) { - colval(scanout[xpos],j) = - evalue(coldef[j]); - } else { - colval(scanout[xpos],j) = 0.0; - for (i = 0; i < nfiles; i++) - colval(scanout[xpos],j) += - colval(input[i].coef,j) * - colval(input[i].scan[xpos],j); - } - if (colval(scanout[xpos],j) < 0.0) - colval(scanout[xpos],j) = 0.0; - } + for (j = 0; j < xmax; j++) /* adjust color */ + multcolor(st[j], input[i].coef); } - if (fwritescan(scanout, xres, stdout) < 0) { - eputs("write error\n"); - quit(1); - } - } - efree(scanout); } -double -colin(fn, ci) /* return color value for picture */ -register int fn, ci; +static double +l_expos( /* return picture exposure */ + char *nam +) { - if (fn == 0) + int fn, n; + double d; + + d = argument(1); + if (d <= -0.5 || d >= nfiles+0.5) { + errno = EDOM; + return(0.0); + } + if (d < 0.5) return((double)nfiles); - if (fn < 1 || fn > nfiles) { + fn = d - 0.5; + if (nam == vbrtexp) + return((*ourbright)(input[fn].expos)); + n = 3; + while (n--) + if (nam == vcolexp[n]) + return(colval(input[fn].expos,n)); + eputs("Bad call to l_expos()!\n"); + quit(1); + return 1; /* pro forma return */ +} + + +static double +l_pixaspect(char *nm) /* return pixel aspect ratio */ +{ + int fn; + double d; + + d = argument(1); + if (d <= -0.5 || d >= nfiles+0.5) { errno = EDOM; return(0.0); } - return(colval(input[fn-1].scan[xpos],ci)); + if (d < 0.5) + return((double)nfiles); + fn = d - 0.5; + return(input[fn].pa); } -double -l_redin() /* get red color */ +static double +l_colin( /* return color value for picture */ + register char *nam +) { - return(colin((int)(argument(1)+.5), RED)); + int fn; + register int n, xoff, yoff; + double d; + + d = argument(1); + if (d <= -0.5 || d >= nfiles+0.5) { + errno = EDOM; + return(0.0); + } + if (d < 0.5) + return((double)nfiles); + fn = d - 0.5; + xoff = yoff = 0; + n = nargum(); + if (n >= 2) { + d = argument(2); + if (d < 0.0) { + xoff = d-.5; + if (xscan+xoff < 0) + xoff = -xscan; + } else { + xoff = d+.5; + if (xscan+xoff >= xmax) + xoff = xmax-1-xscan; + } + } + if (n >= 3) { + d = argument(3); + if (d < 0.0) { + yoff = d-.5; + if (yoff+MIDSCN < 0) + yoff = -MIDSCN; + if (yscan+yoff < 0) + yoff = -yscan; + } else { + yoff = d+.5; + if (yoff+MIDSCN >= WINSIZ) + yoff = WINSIZ-1-MIDSCN; + if (yscan+yoff >= ymax) + yoff = ymax-1-yscan; + } + } + if (nam == vbrtin) + return((*ourbright)(input[fn].scan[MIDSCN+yoff][xscan+xoff])); + n = 3; + while (n--) + if (nam == vcolin[n]) + return(colval(input[fn].scan[MIDSCN+yoff][xscan+xoff],n)); + eputs("Bad call to l_colin()!\n"); + quit(1); + return 1; /* pro forma return */ } -double -l_grnin() /* get green color */ +static double +l_ray( /* return ray origin or direction */ + register char *nam +) { - return(colin((int)(argument(1)+.5), GRN)); + static unsigned long ltick[MAXINP]; + static FVECT lorg[MAXINP], ldir[MAXINP]; + static double ldist[MAXINP]; + RREAL loc[2]; + double d; + int fn; + register int i; + + d = argument(1); + if (d <= -0.5 || d >= nfiles+0.5) { + errno = EDOM; + return(0.0); + } + if (d < 0.5) + return((double)nfiles); + fn = d - 0.5; + if (ltick[fn] != eclock) { /* need to compute? */ + lorg[fn][0] = lorg[fn][1] = lorg[fn][2] = 0.0; + ldir[fn][0] = ldir[fn][1] = ldir[fn][2] = 0.0; + ldist[fn] = -1.0; + if (input[fn].vw.type == 0) + errno = EDOM; + else { + pix2loc(loc, &input[fn].rs, xscan, ymax-1-yscan); + ldist[fn] = viewray(lorg[fn], ldir[fn], + &input[fn].vw, loc[0], loc[1]); + } + ltick[fn] = eclock; + } + if (nam == vray[i=6]) + return(ldist[fn]); + while (i--) + if (nam == vray[i]) + return(i < 3 ? lorg[fn][i] : ldir[fn][i-3]); + eputs("Bad call to l_ray()!\n"); + quit(1); + return 1; /* pro forma return */ } -double -l_bluin() /* get blue color */ +static double +l_psize(char *nm) /* compute pixel size in steradians */ { - return(colin((int)(argument(1)+.5), BLU)); + static unsigned long ltick[MAXINP]; + static double psize[MAXINP]; + FVECT dir0, org, dirx, diry; + RREAL locx[2], locy[2]; + double d; + int fn; + register int i; + + d = argument(1); + if (d <= -0.5 || d >= nfiles+0.5) { + errno = EDOM; + return(0.0); + } + if (d < 0.5) + return((double)nfiles); + fn = d - 0.5; + if (ltick[fn] != eclock) { /* need to compute? */ + psize[fn] = 0.0; + if (input[fn].vw.type == 0) + errno = EDOM; + else if (input[fn].vw.type != VT_PAR && + funvalue(vray[6], 1, &d) >= -FTINY) { + for (i = 0; i < 3; i++) + dir0[i] = funvalue(vray[3+i], 1, &d); + pix2loc(locx, &input[fn].rs, xscan+1, ymax-1-yscan); + pix2loc(locy, &input[fn].rs, xscan, ymax-yscan); + if (viewray(org, dirx, &input[fn].vw, + locx[0], locx[1]) >= -FTINY && + viewray(org, diry, &input[fn].vw, + locy[0], locy[1]) >= -FTINY) { + /* approximate solid angle */ + for (i = 0; i < 3; i++) { + dirx[i] -= dir0[i]; + diry[i] -= dir0[i]; + } + fcross(dir0, dirx, diry); + psize[fn] = VLEN(dir0); + } + } + ltick[fn] = eclock; + } + return(psize[fn]); } -wputs(msg) -char *msg; +extern void +wputs(char *msg) { if (!nowarn) eputs(msg); } -eputs(msg) -char *msg; +extern void +eputs(char *msg) { fputs(msg, stderr); } -quit(code) -int code; +extern void +quit(int code) /* exit gracefully */ { + register int i; + /* close input files */ + for (i = 0; i < nfiles; i++) + if (input[i].name == Command) + pclose(input[i].fp); + else + fclose(input[i].fp); exit(code); }