src/px/pinterp.c

#ifndef lint
static char SCCSid[] = "$SunId$ LBL";
#endif

/*
 * Interpolate and extrapolate pictures with different view parameters.
 *
 *      Greg Ward       09Dec89
 */

#include "standard.h"

#include "view.h"

#include "color.h"

#define pscan(y)        (ourpict+(y)*ourview.hresolu)
#define zscan(y)        (ourzbuf+(y)*ourview.hresolu)

#define F_FORE          1               /* fill foreground */
#define F_BACK          2               /* fill background */

#define PACKSIZ         42              /* calculation packet size */

#define RTCOM           "rtrace -h -ovl -fff -x %d %s"

#define ABS(x)          ((x)>0?(x):-(x))

VIEW    ourview = STDVIEW(512);         /* desired view */

double  zeps = .02;                     /* allowed z epsilon */

COLR    *ourpict;                       /* output picture */
float   *ourzbuf;                       /* corresponding z-buffer */

char    *progname;

VIEW    theirview = STDVIEW(512);       /* input view */
int     gotview;                        /* got input view? */

int     fill = F_FORE|F_BACK;           /* selected fill algorithm */
extern int      backfill(), rcalfill(); /* fill functions */
int     (*deffill)() = backfill;        /* selected fill function */
COLR    backcolr = BLKCOLR;             /* background color */
double  backz = 0.0;                    /* background z value */

double  theirs2ours[4][4];              /* transformation matrix */
int     normdist = 1;                   /* normalized distance? */

int     childpid = -1;                  /* id of fill process */
FILE    *psend, *precv;                 /* pipes to/from fill calculation */
int     queue[PACKSIZ][2];              /* pending pixels */
int     queuesiz;                       /* number of pixels pending */


main(argc, argv)                        /* interpolate pictures */
int     argc;
char    *argv[];
{
#define check(olen,narg)        if (argv[i][olen] || narg >= argc-i) goto badopt
        extern double   atof();
        int     gotvfile = 0;
        char    *zfile = NULL;
        char    *err;
        int     i;

        progname = argv[0];

        for (i = 1; i < argc && argv[i][0] == '-'; i++)
                switch (argv[i][1]) {
                case 't':                               /* threshold */
                        check(2,1);
                        zeps = atof(argv[++i]);
                        break;
                case 'n':                               /* dist. normalized? */
                        check(2,0);
                        normdist = !normdist;
                        break;
                case 'f':                               /* fill type */
                        switch (argv[i][2]) {
                        case '0':                               /* none */
                                check(3,0);
                                fill = 0;
                                break;
                        case 'f':                               /* foreground */
                                check(3,0);
                                fill = F_FORE;
                                break;
                        case 'b':                               /* background */
                                check(3,0);
                                fill = F_BACK;
                                break;
                        case 'a':                               /* all */
                                check(3,0);
                                fill = F_FORE|F_BACK;
                                break;
                        case 'c':                               /* color */
                                check(3,3);
                                deffill = backfill;
                                setcolr(backcolr, atof(argv[i+1]),
                                        atof(argv[i+2]), atof(argv[i+3]));
                                i += 3;
                                break;
                        case 'z':                               /* z value */
                                check(3,1);
                                deffill = backfill;
                                backz = atof(argv[++i]);
                                break;
                        case 'r':                               /* rtrace */
                                check(3,1);
                                deffill = rcalfill;
                                calstart(RTCOM, argv[++i]);
                                break;
                        default:
                                goto badopt;
                        }
                        break;
                case 'z':                               /* z file */
                        check(2,1);
                        zfile = argv[++i];
                        break;
                case 'x':                               /* x resolution */
                        check(2,1);
                        ourview.hresolu = atoi(argv[++i]);
                        break;
                case 'y':                               /* y resolution */
                        check(2,1);
                        ourview.vresolu = atoi(argv[++i]);
                        break;
                case 'v':                               /* view */
                        switch (argv[i][2]) {
                        case 't':                               /* type */
                                check(4,0);
                                ourview.type = argv[i][3];
                                break;
                        case 'p':                               /* point */
                                check(3,3);
                                ourview.vp[0] = atof(argv[++i]);
                                ourview.vp[1] = atof(argv[++i]);
                                ourview.vp[2] = atof(argv[++i]);
                                break;
                        case 'd':                               /* direction */
                                check(3,3);
                                ourview.vdir[0] = atof(argv[++i]);
                                ourview.vdir[1] = atof(argv[++i]);
                                ourview.vdir[2] = atof(argv[++i]);
                                break;
                        case 'u':                               /* up */
                                check(3,3);
                                ourview.vup[0] = atof(argv[++i]);
                                ourview.vup[1] = atof(argv[++i]);
                                ourview.vup[2] = atof(argv[++i]);
                                break;
                        case 'h':                               /* horizontal */
                                check(3,1);
                                ourview.horiz = atof(argv[++i]);
                                break;
                        case 'v':                               /* vertical */
                                check(3,1);
                                ourview.vert = atof(argv[++i]);
                                break;
                        case 'f':                               /* file */
                                check(3,1);
                                gotvfile = viewfile(argv[++i], &ourview);
                                if (gotvfile < 0) {
                                        perror(argv[i]);
                                        exit(1);
                                } else if (gotvfile == 0) {
                                        fprintf(stderr, "%s: bad view file\n",
                                                        argv[i]);
                                        exit(1);
                                }
                                break;
                        default:
                                goto badopt;
                        }
                        break;
                default:
                badopt:
                        fprintf(stderr, "%s: command line error at '%s'\n",
                                        progname, argv[i]);
                        goto userr;
                }
                                                /* check arguments */
        if ((argc-i)%2)
                goto userr;
                                                /* set view */
        if (err = setview(&ourview)) {
                fprintf(stderr, "%s: %s\n", progname, err);
                exit(1);
        }
                                                /* allocate frame */
        ourpict = (COLR *)malloc(ourview.hresolu*ourview.vresolu*sizeof(COLR));
        ourzbuf = (float *)calloc(ourview.hresolu*ourview.vresolu,sizeof(float));
        if (ourpict == NULL || ourzbuf == NULL) {
                perror(progname);
                exit(1);
        }
                                                        /* get input */
        for ( ; i < argc; i += 2)
                addpicture(argv[i], argv[i+1]);
                                                        /* fill in spaces */
        if (fill&F_BACK)
                backpicture();
        else
                fillpicture();
                                                        /* close calculation */
        caldone();
                                                        /* add to header */
        printargs(argc, argv, stdout);
        if (gotvfile) {
                printf(VIEWSTR);
                fprintview(&ourview, stdout);
                printf("\n");
        }
        printf("\n");
                                                        /* write picture */
        writepicture();
                                                        /* write z file */
        if (zfile != NULL)
                writedistance(zfile);

        exit(0);
userr:
        fprintf(stderr,
        "Usage: %s [view opts][-t zthresh][-z zout][-fT][-n] pfile zspec ..\n",
                        progname);
        exit(1);
#undef check
}


headline(s)                             /* process header string */
char    *s;
{
        static char     *altname[] = {"rview","rpict","pinterp",VIEWSTR,NULL};
        register char   **an;

        printf("\t%s", s);

        for (an = altname; *an != NULL; an++)
                if (!strncmp(*an, s, strlen(*an))) {
                        if (sscanview(&theirview, s+strlen(*an)) == 0)
                                gotview++;
                        break;
                }
}


addpicture(pfile, zspec)                /* add picture to output */
char    *pfile, *zspec;
{
        extern double   atof();
        FILE    *pfp, *zfp;
        char    *err;
        COLR    *scanin;
        float   *zin, *zlast;
        int     *plast;
        int     y;
                                        /* open picture file */
        if ((pfp = fopen(pfile, "r")) == NULL) {
                perror(pfile);
                exit(1);
        }
                                        /* get header and view */
        printf("%s:\n", pfile);
        gotview = 0;
        getheader(pfp, headline);
        if (!gotview || fgetresolu(&theirview.hresolu, &theirview.vresolu, pfp)
                        != (YMAJOR|YDECR)) {
                fprintf(stderr, "%s: picture view error\n", pfile);
                exit(1);
        }
        if (err = setview(&theirview)) {
                fprintf(stderr, "%s: %s\n", pfile, err);
                exit(1);
        }
                                        /* compute transformation */
        pixform(theirs2ours, &theirview, &ourview);
                                        /* allocate scanlines */
        scanin = (COLR *)malloc(theirview.hresolu*sizeof(COLR));
        zin = (float *)malloc(theirview.hresolu*sizeof(float));
        plast = (int *)calloc(theirview.hresolu, sizeof(int));
        zlast = (float *)calloc(theirview.hresolu, sizeof(float));
        if (scanin == NULL || zin == NULL || plast == NULL || zlast == NULL) {
                perror(progname);
                exit(1);
        }
                                        /* get z specification or file */
        if ((zfp = fopen(zspec, "r")) == NULL) {
                double  zvalue;
                register int    x;
                if (!isfloat(zspec) || (zvalue = atof(zspec)) <= 0.0) {
                        perror(zspec);
                        exit(1);
                }
                for (x = 0; x < theirview.hresolu; x++)
                        zin[x] = zvalue;
        }
                                        /* load image */
        for (y = theirview.vresolu-1; y >= 0; y--) {
                if (freadcolrs(scanin, theirview.hresolu, pfp) < 0) {
                        fprintf(stderr, "%s: read error\n", pfile);
                        exit(1);
                }
                if (zfp != NULL
                        && fread(zin,sizeof(float),theirview.hresolu,zfp)
                                < theirview.hresolu) {
                        fprintf(stderr, "%s: read error\n", zspec);
                        exit(1);
                }
                addscanline(y, scanin, zin, plast, zlast);
        }
                                        /* clean up */
        free((char *)scanin);
        free((char *)zin);
        free((char *)plast);
        free((char *)zlast);
        fclose(pfp);
        if (zfp != NULL)
                fclose(zfp);
}


pixform(xfmat, vw1, vw2)                /* compute view1 to view2 matrix */
register double xfmat[4][4];
register VIEW   *vw1, *vw2;
{
        double  m4t[4][4];

        setident4(xfmat);
        xfmat[0][0] = vw1->vhinc[0];
        xfmat[0][1] = vw1->vhinc[1];
        xfmat[0][2] = vw1->vhinc[2];
        xfmat[1][0] = vw1->vvinc[0];
        xfmat[1][1] = vw1->vvinc[1];
        xfmat[1][2] = vw1->vvinc[2];
        xfmat[2][0] = vw1->vdir[0];
        xfmat[2][1] = vw1->vdir[1];
        xfmat[2][2] = vw1->vdir[2];
        xfmat[3][0] = vw1->vp[0];
        xfmat[3][1] = vw1->vp[1];
        xfmat[3][2] = vw1->vp[2];
        setident4(m4t);
        m4t[0][0] = vw2->vhinc[0]/vw2->vhn2;
        m4t[1][0] = vw2->vhinc[1]/vw2->vhn2;
        m4t[2][0] = vw2->vhinc[2]/vw2->vhn2;
        m4t[3][0] = -DOT(vw2->vp,vw2->vhinc)/vw2->vhn2;
        m4t[0][1] = vw2->vvinc[0]/vw2->vvn2;
        m4t[1][1] = vw2->vvinc[1]/vw2->vvn2;
        m4t[2][1] = vw2->vvinc[2]/vw2->vvn2;
        m4t[3][1] = -DOT(vw2->vp,vw2->vvinc)/vw2->vvn2;
        m4t[0][2] = vw2->vdir[0];
        m4t[1][2] = vw2->vdir[1];
        m4t[2][2] = vw2->vdir[2];
        m4t[3][2] = -DOT(vw2->vp,vw2->vdir);
        multmat4(xfmat, xfmat, m4t);
}


addscanline(y, pline, zline, lasty, lastyz)     /* add scanline to output */
int     y;
COLR    *pline;
float   *zline;
int     *lasty;                 /* input/output */
float   *lastyz;                /* input/output */
{
        extern double   sqrt();
        double  pos[3];
        int     lastx = 0;
        double  lastxz = 0;
        double  zt;
        int     xpos, ypos;
        register int    x;

        for (x = theirview.hresolu-1; x >= 0; x--) {
                pos[0] = x - .5*(theirview.hresolu-1);
                pos[1] = y - .5*(theirview.vresolu-1);
                pos[2] = zline[x];
                if (theirview.type == VT_PER) {
                        if (normdist)   /* adjust for eye-ray distance */
                                pos[2] /= sqrt( 1.
                                        + pos[0]*pos[0]*theirview.vhn2
                                        + pos[1]*pos[1]*theirview.vvn2 );
                        pos[0] *= pos[2];
                        pos[1] *= pos[2];
                }
                multp3(pos, pos, theirs2ours);
                if (pos[2] <= 0)
                        continue;
                if (ourview.type == VT_PER) {
                        pos[0] /= pos[2];
                        pos[1] /= pos[2];
                }
                pos[0] += .5*ourview.hresolu;
                pos[1] += .5*ourview.vresolu;
                if (pos[0] < 0 || (xpos = pos[0]) >= ourview.hresolu
                        || pos[1] < 0 || (ypos = pos[1]) >= ourview.vresolu)
                        continue;
                                        /* add pixel to our image */
                zt = 2.*zeps*zline[x];
                addpixel(xpos, ypos,
                        (fill&F_FORE && ABS(zline[x]-lastxz) <= zt)
                                ? lastx - xpos : 1,
                        (fill&F_FORE && ABS(zline[x]-lastyz[x]) <= zt)
                                ? lasty[x] - ypos : 1,
                        pline[x], pos[2]);
                lastx = xpos;
                lasty[x] = ypos;
                lastxz = lastyz[x] = zline[x];
        }
}


addpixel(xstart, ystart, width, height, pix, z) /* fill in area for pixel */
int     xstart, ystart;
int     width, height;
COLR    pix;
double  z;
{
        register int    x, y;
                                        /* make width and height positive */
        if (width < 0) {
                width = -width;
                xstart = xstart-width+1;
        } else if (width == 0)
                width = 1;
        if (height < 0) {
                height = -height;
                ystart = ystart-height+1;
        } else if (height == 0)
                height = 1;
                                        /* fill pixel(s) within rectangle */
        for (y = ystart; y < ystart+height; y++)
                for (x = xstart; x < xstart+width; x++)
                        if (zscan(y)[x] <= 0
                                        || zscan(y)[x]-z > zeps*zscan(y)[x]) {
                                zscan(y)[x] = z;
                                copycolr(pscan(y)[x], pix);
                        }
}


backpicture()                           /* background fill algorithm */
{
        int     *yback, xback;
        int     y;
        COLR    pfill;
        register int    x, i;
                                                        /* get back buffer */
        yback = (int *)malloc(ourview.hresolu*sizeof(int));
        if (yback == NULL) {
                perror(progname);
                return;
        }
        for (x = 0; x < ourview.hresolu; x++)
                yback[x] = -2;
        /*
         * Xback and yback are the pixel locations of suitable
         * background values in each direction.
         * A value of -2 means unassigned, and -1 means
         * that there is no suitable background in this direction.
         */
                                                        /* fill image */
        for (y = 0; y < ourview.vresolu; y++) {
                xback = -2;
                for (x = 0; x < ourview.hresolu; x++)
                        if (zscan(y)[x] <= 0) {         /* empty pixel */
                                /*
                                 * First, find background from above or below.
                                 * (farthest assigned pixel)
                                 */
                                if (yback[x] == -2) {
                                        for (i = y+1; i < ourview.vresolu; i++)
                                                if (zscan(i)[x] > 0)
                                                        break;
                                        if (i < ourview.vresolu
                                && (y <= 0 || zscan(y-1)[x] < zscan(i)[x]))
                                                yback[x] = i;
                                        else
                                                yback[x] = y-1;
                                }
                                /*
                                 * Next, find background from left or right.
                                 */
                                if (xback == -2) {
                                        for (i = x+1; i < ourview.hresolu; i++)
                                                if (zscan(y)[i] > 0)
                                                        break;
                                        if (i < ourview.hresolu
                                && (x <= 0 || zscan(y)[x-1] < zscan(y)[i]))
                                                xback = i;
                                        else
                                                xback = x-1;
                                }
                                /*
                                 * Check to see if we have no background for
                                 * this pixel.  If not, use background color.
                                 */
                                if (xback < 0 && yback[x] < 0) {
                                        (*deffill)(x,y);
                                        continue;
                                }
                                /*
                                 * Compare, and use the background that is
                                 * farther, unless one of them is next to us.
                                 */
                                if ( yback[x] < 0
                                        || (xback >= 0 && ABS(x-xback) <= 1)
                                        || ( ABS(y-yback[x]) > 1
                                                && zscan(yback[x])[x]
                                                < zscan(y)[xback] ) ) {
                                        copycolr(pscan(y)[x],pscan(y)[xback]);
                                        zscan(y)[x] = zscan(y)[xback];
                                } else {
                                        copycolr(pscan(y)[x],pscan(yback[x])[x]);
                                        zscan(y)[x] = zscan(yback[x])[x];
                                }
                        } else {                                /* full pixel */
                                yback[x] = -2;
                                xback = -2;
                        }
        }
        free((char *)yback);
}


fillpicture()                           /* paint in empty pixels with default */
{
        register int    x, y;

        for (y = 0; y < ourview.vresolu; y++)
                for (x = 0; x < ourview.hresolu; x++)
                        if (zscan(y)[x] <= 0)
                                (*deffill)(x,y);
}


writepicture()                          /* write out picture */
{
        int     y;

        fputresolu(YMAJOR|YDECR, ourview.hresolu, ourview.vresolu, stdout);
        for (y = ourview.vresolu-1; y >= 0; y--)
                if (fwritecolrs(pscan(y), ourview.hresolu, stdout) < 0) {
                        perror(progname);
                        exit(1);
                }
}


writedistance(fname)                    /* write out z file */
char    *fname;
{
        extern double   sqrt();
        int     donorm = normdist && ourview.type == VT_PER;
        FILE    *fp;
        int     y;
        float   *zout;

        if ((fp = fopen(fname, "w")) == NULL) {
                perror(fname);
                exit(1);
        }
        if (donorm
        && (zout = (float *)malloc(ourview.hresolu*sizeof(float))) == NULL) {
                perror(progname);
                exit(1);
        }
        for (y = ourview.vresolu-1; y >= 0; y--) {
                if (donorm) {
                        double  vx, yzn2;
                        register int    x;
                        yzn2 = y - .5*(ourview.vresolu-1);
                        yzn2 = 1. + yzn2*yzn2*ourview.vvn2;
                        for (x = 0; x < ourview.hresolu; x++) {
                                vx = x - .5*(ourview.hresolu-1);
                                zout[x] = zscan(y)[x]
                                        * sqrt(vx*vx*ourview.vhn2 + yzn2);
                        }
                } else
                        zout = zscan(y);
                if (fwrite(zout, sizeof(float), ourview.hresolu, fp)
                                < ourview.hresolu) {
                        perror(fname);
                        exit(1);
                }
        }
        if (donorm)
                free((char *)zout);
        fclose(fp);
}


isfloat(s)                              /* see if string is floating number */
register char   *s;
{
        for ( ; *s; s++)
                if ((*s < '0' || *s > '9') && *s != '.' && *s != '-'
                                && *s != 'e' && *s != 'E' && *s != '+')
                        return(0);
        return(1);
}


backfill(x, y)                          /* fill pixel with background */
int     x, y;
{
        register BYTE   *dest = pscan(y)[x];

        copycolr(dest, backcolr);
        zscan(y)[x] = backz;
}


calstart(prog, args)                    /* start fill calculation */
char    *prog, *args;
{
        char    combuf[512];
        int     p0[2], p1[2];

        if (childpid != -1) {
                fprintf(stderr, "%s: too many calculations\n", progname);
                exit(1);
        }
        sprintf(combuf, prog, PACKSIZ, args);
        if (pipe(p0) < 0 || pipe(p1) < 0)
                syserror();
        if ((childpid = vfork()) == 0) {        /* fork calculation */
                close(p0[1]);
                close(p1[0]);
                if (p0[0] != 0) {
                        dup2(p0[0], 0);
                        close(p0[0]);
                }
                if (p1[1] != 1) {
                        dup2(p1[1], 1);
                        close(p1[1]);
                }
                execl("/bin/sh", "sh", "-c", combuf, 0);
                perror("/bin/sh");
                _exit(127);
        }
        if (childpid == -1)
                syserror();
        close(p0[0]);
        close(p1[1]);
        if ((psend = fdopen(p0[1], "w")) == NULL)
                syserror();
        if ((precv = fdopen(p1[0], "r")) == NULL)
                syserror();
        queuesiz = 0;
}


caldone()                               /* done with calculation */
{
        int     pid;

        if (childpid == -1)
                return;
        if (fclose(psend) == EOF)
                syserror();
        clearqueue();
        fclose(precv);
        while ((pid = wait(0)) != -1 && pid != childpid)
                ;
        childpid = -1;
}


rcalfill(x, y)                          /* fill with ray-calculated pixel */
int     x, y;
{
        FVECT   orig, dir;
        float   outbuf[6];

        if (queuesiz >= PACKSIZ) {      /* flush queue */
                if (fflush(psend) == EOF)
                        syserror();
                clearqueue();
        }
                                        /* send new ray */
        rayview(orig, dir, &ourview, x+.5, y+.5);
        outbuf[0] = orig[0]; outbuf[1] = orig[1]; outbuf[2] = orig[2];
        outbuf[3] = dir[0]; outbuf[4] = dir[1]; outbuf[5] = dir[2];
        if (fwrite(outbuf, sizeof(float), 6, psend) < 6)
                syserror();
                                        /* remember it */
        queue[queuesiz][0] = x;
        queue[queuesiz][1] = y;
        queuesiz++;
}


clearqueue()                            /* get results from queue */
{
        float   inbuf[4];
        register int    i;

        for (i = 0; i < queuesiz; i++) {
                if (fread(inbuf, sizeof(float), 4, precv) < 4) {
                        fprintf(stderr, "%s: read error in clearqueue\n",
                                        progname);
                        exit(1);
                }
                setcolr(pscan(queue[i][1])[queue[i][0]],
                                inbuf[0], inbuf[1], inbuf[2]);
                zscan(queue[i][1])[queue[i][0]] = inbuf[3];
        }
        queuesiz = 0;
}


syserror()                      /* report error and exit */
{
        perror(progname);
        exit(1);
}
Revision:	1.14
Committed:	Thu Jan 4 17:47:37 1990 UTC (34 years, 4 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 1.13:	+35 -19 lines
Log Message:	added error checking on calculation I/O
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2			static char SCCSid[] = "$SunId$ LBL";
3			#endif
4
5			/*
6			* Interpolate and extrapolate pictures with different view parameters.
7			*
8			* Greg Ward 09Dec89
9			*/
10
11			#include "standard.h"
12
13			#include "view.h"
14
15			#include "color.h"
16
17			#define pscan(y) (ourpict+(y)*ourview.hresolu)
18			#define zscan(y) (ourzbuf+(y)*ourview.hresolu)
19
20	greg	1.13	#define F_FORE 1 /* fill foreground */
21			#define F_BACK 2 /* fill background */
22	greg	1.11
23	greg	1.13	#define PACKSIZ 42 /* calculation packet size */
24
25			#define RTCOM "rtrace -h -ovl -fff -x %d %s"
26
27	greg	1.3	#define ABS(x) ((x)>0?(x):-(x))
28
29	greg	1.1	VIEW ourview = STDVIEW(512); /* desired view */
30
31	greg	1.8	double zeps = .02; /* allowed z epsilon */
32	greg	1.1
33			COLR ourpict; / output picture */
34			float ourzbuf; / corresponding z-buffer */
35
36			char *progname;
37
38			VIEW theirview = STDVIEW(512); /* input view */
39			int gotview; /* got input view? */
40
41	greg	1.12	int fill = F_FORE\|F_BACK; /* selected fill algorithm */
42	greg	1.14	extern int backfill(), rcalfill(); /* fill functions */
43	greg	1.12	int (deffill)() = backfill; / selected fill function */
44	greg	1.11	COLR backcolr = BLKCOLR; /* background color */
45	greg	1.13	double backz = 0.0; /* background z value */
46	greg	1.11
47	greg	1.5	double theirs2ours[4][4]; /* transformation matrix */
48	greg	1.11	int normdist = 1; /* normalized distance? */
49	greg	1.1
50	greg	1.14	int childpid = -1; /* id of fill process */
51	greg	1.13	FILE psend, precv; /* pipes to/from fill calculation */
52			int queue[PACKSIZ][2]; /* pending pixels */
53			int queuesiz; /* number of pixels pending */
54	greg	1.5
55	greg	1.13
56	greg	1.1	main(argc, argv) /* interpolate pictures */
57			int argc;
58			char *argv[];
59			{
60			#define check(olen,narg) if (argv[i][olen] \|\| narg >= argc-i) goto badopt
61	greg	1.11	extern double atof();
62	greg	1.1	int gotvfile = 0;
63	greg	1.11	char *zfile = NULL;
64	greg	1.1	char *err;
65			int i;
66
67			progname = argv[0];
68
69			for (i = 1; i < argc && argv[i][0] == '-'; i++)
70			switch (argv[i][1]) {
71			case 't': /* threshold */
72			check(2,1);
73			zeps = atof(argv[++i]);
74			break;
75	greg	1.11	case 'n': /* dist. normalized? */
76	greg	1.10	check(2,0);
77	greg	1.11	normdist = !normdist;
78	greg	1.10	break;
79	greg	1.11	case 'f': /* fill type */
80			switch (argv[i][2]) {
81			case '0': /* none */
82			check(3,0);
83			fill = 0;
84			break;
85			case 'f': /* foreground */
86			check(3,0);
87			fill = F_FORE;
88			break;
89			case 'b': /* background */
90			check(3,0);
91			fill = F_BACK;
92			break;
93			case 'a': /* all */
94			check(3,0);
95			fill = F_FORE\|F_BACK;
96			break;
97			case 'c': /* color */
98			check(3,3);
99	greg	1.12	deffill = backfill;
100	greg	1.11	setcolr(backcolr, atof(argv[i+1]),
101			atof(argv[i+2]), atof(argv[i+3]));
102			i += 3;
103			break;
104	greg	1.13	case 'z': /* z value */
105			check(3,1);
106			deffill = backfill;
107			backz = atof(argv[++i]);
108			break;
109			case 'r': /* rtrace */
110			check(3,1);
111	greg	1.14	deffill = rcalfill;
112	greg	1.13	calstart(RTCOM, argv[++i]);
113			break;
114	greg	1.11	default:
115			goto badopt;
116			}
117			break;
118			case 'z': /* z file */
119			check(2,1);
120			zfile = argv[++i];
121			break;
122	greg	1.6	case 'x': /* x resolution */
123			check(2,1);
124			ourview.hresolu = atoi(argv[++i]);
125			break;
126			case 'y': /* y resolution */
127			check(2,1);
128			ourview.vresolu = atoi(argv[++i]);
129			break;
130	greg	1.1	case 'v': /* view */
131			switch (argv[i][2]) {
132			case 't': /* type */
133			check(4,0);
134			ourview.type = argv[i][3];
135			break;
136			case 'p': /* point */
137			check(3,3);
138			ourview.vp[0] = atof(argv[++i]);
139			ourview.vp[1] = atof(argv[++i]);
140			ourview.vp[2] = atof(argv[++i]);
141			break;
142			case 'd': /* direction */
143			check(3,3);
144			ourview.vdir[0] = atof(argv[++i]);
145			ourview.vdir[1] = atof(argv[++i]);
146			ourview.vdir[2] = atof(argv[++i]);
147			break;
148			case 'u': /* up */
149			check(3,3);
150			ourview.vup[0] = atof(argv[++i]);
151			ourview.vup[1] = atof(argv[++i]);
152			ourview.vup[2] = atof(argv[++i]);
153			break;
154			case 'h': /* horizontal */
155			check(3,1);
156			ourview.horiz = atof(argv[++i]);
157			break;
158			case 'v': /* vertical */
159			check(3,1);
160			ourview.vert = atof(argv[++i]);
161			break;
162			case 'f': /* file */
163			check(3,1);
164			gotvfile = viewfile(argv[++i], &ourview);
165			if (gotvfile < 0) {
166			perror(argv[i]);
167			exit(1);
168			} else if (gotvfile == 0) {
169			fprintf(stderr, "%s: bad view file\n",
170			argv[i]);
171			exit(1);
172			}
173			break;
174			default:
175			goto badopt;
176			}
177			break;
178			default:
179			badopt:
180	greg	1.10	fprintf(stderr, "%s: command line error at '%s'\n",
181	greg	1.1	progname, argv[i]);
182	greg	1.10	goto userr;
183	greg	1.1	}
184			/* check arguments */
185	greg	1.13	if ((argc-i)%2)
186	greg	1.10	goto userr;
187	greg	1.1	/* set view */
188			if (err = setview(&ourview)) {
189			fprintf(stderr, "%s: %s\n", progname, err);
190			exit(1);
191			}
192			/* allocate frame */
193	greg	1.12	ourpict = (COLR )malloc(ourview.hresoluourview.vresolu*sizeof(COLR));
194	greg	1.1	ourzbuf = (float )calloc(ourview.hresoluourview.vresolu,sizeof(float));
195			if (ourpict == NULL \|\| ourzbuf == NULL) {
196			perror(progname);
197			exit(1);
198			}
199			/* get input */
200			for ( ; i < argc; i += 2)
201			addpicture(argv[i], argv[i+1]);
202			/* fill in spaces */
203	greg	1.11	if (fill&F_BACK)
204	greg	1.12	backpicture();
205			else
206	greg	1.11	fillpicture();
207	greg	1.13	/* close calculation */
208	greg	1.14	caldone();
209	greg	1.1	/* add to header */
210			printargs(argc, argv, stdout);
211			if (gotvfile) {
212			printf(VIEWSTR);
213			fprintview(&ourview, stdout);
214			printf("\n");
215			}
216			printf("\n");
217	greg	1.11	/* write picture */
218	greg	1.1	writepicture();
219	greg	1.11	/* write z file */
220			if (zfile != NULL)
221			writedistance(zfile);
222	greg	1.1
223			exit(0);
224	greg	1.10	userr:
225			fprintf(stderr,
226	greg	1.11	"Usage: %s [view opts][-t zthresh][-z zout][-fT][-n] pfile zspec ..\n",
227	greg	1.10	progname);
228			exit(1);
229	greg	1.1	#undef check
230			}
231
232
233			headline(s) /* process header string */
234			char *s;
235			{
236			static char *altname[] = {"rview","rpict","pinterp",VIEWSTR,NULL};
237			register char **an;
238
239			printf("\t%s", s);
240
241			for (an = altname; *an != NULL; an++)
242			if (!strncmp(an, s, strlen(an))) {
243			if (sscanview(&theirview, s+strlen(*an)) == 0)
244			gotview++;
245			break;
246			}
247			}
248
249
250	greg	1.10	addpicture(pfile, zspec) /* add picture to output */
251			char pfile, zspec;
252	greg	1.1	{
253	greg	1.10	extern double atof();
254	greg	1.1	FILE pfp, zfp;
255	greg	1.8	char *err;
256	greg	1.1	COLR *scanin;
257	greg	1.9	float zin, zlast;
258			int *plast;
259	greg	1.1	int y;
260	greg	1.10	/* open picture file */
261	greg	1.1	if ((pfp = fopen(pfile, "r")) == NULL) {
262			perror(pfile);
263			exit(1);
264			}
265			/* get header and view */
266			printf("%s:\n", pfile);
267			gotview = 0;
268			getheader(pfp, headline);
269	greg	1.9	if (!gotview \|\| fgetresolu(&theirview.hresolu, &theirview.vresolu, pfp)
270			!= (YMAJOR\|YDECR)) {
271	greg	1.1	fprintf(stderr, "%s: picture view error\n", pfile);
272			exit(1);
273			}
274			if (err = setview(&theirview)) {
275			fprintf(stderr, "%s: %s\n", pfile, err);
276			exit(1);
277			}
278	greg	1.5	/* compute transformation */
279			pixform(theirs2ours, &theirview, &ourview);
280	greg	1.1	/* allocate scanlines */
281	greg	1.9	scanin = (COLR )malloc(theirview.hresolusizeof(COLR));
282			zin = (float )malloc(theirview.hresolusizeof(float));
283	greg	1.10	plast = (int *)calloc(theirview.hresolu, sizeof(int));
284	greg	1.9	zlast = (float *)calloc(theirview.hresolu, sizeof(float));
285	greg	1.10	if (scanin == NULL \|\| zin == NULL \|\| plast == NULL \|\| zlast == NULL) {
286	greg	1.1	perror(progname);
287			exit(1);
288			}
289	greg	1.10	/* get z specification or file */
290			if ((zfp = fopen(zspec, "r")) == NULL) {
291			double zvalue;
292			register int x;
293			if (!isfloat(zspec) \|\| (zvalue = atof(zspec)) <= 0.0) {
294			perror(zspec);
295			exit(1);
296			}
297			for (x = 0; x < theirview.hresolu; x++)
298			zin[x] = zvalue;
299			}
300	greg	1.1	/* load image */
301	greg	1.9	for (y = theirview.vresolu-1; y >= 0; y--) {
302			if (freadcolrs(scanin, theirview.hresolu, pfp) < 0) {
303	greg	1.1	fprintf(stderr, "%s: read error\n", pfile);
304			exit(1);
305			}
306	greg	1.10	if (zfp != NULL
307			&& fread(zin,sizeof(float),theirview.hresolu,zfp)
308	greg	1.9	< theirview.hresolu) {
309	greg	1.10	fprintf(stderr, "%s: read error\n", zspec);
310	greg	1.1	exit(1);
311			}
312	greg	1.9	addscanline(y, scanin, zin, plast, zlast);
313	greg	1.1	}
314			/* clean up */
315			free((char *)scanin);
316			free((char *)zin);
317	greg	1.9	free((char *)plast);
318			free((char *)zlast);
319	greg	1.1	fclose(pfp);
320	greg	1.10	if (zfp != NULL)
321			fclose(zfp);
322	greg	1.1	}
323
324
325	greg	1.5	pixform(xfmat, vw1, vw2) /* compute view1 to view2 matrix */
326			register double xfmat[4][4];
327			register VIEW vw1, vw2;
328			{
329			double m4t[4][4];
330
331			setident4(xfmat);
332			xfmat[0][0] = vw1->vhinc[0];
333			xfmat[0][1] = vw1->vhinc[1];
334			xfmat[0][2] = vw1->vhinc[2];
335			xfmat[1][0] = vw1->vvinc[0];
336			xfmat[1][1] = vw1->vvinc[1];
337			xfmat[1][2] = vw1->vvinc[2];
338			xfmat[2][0] = vw1->vdir[0];
339			xfmat[2][1] = vw1->vdir[1];
340			xfmat[2][2] = vw1->vdir[2];
341			xfmat[3][0] = vw1->vp[0];
342			xfmat[3][1] = vw1->vp[1];
343			xfmat[3][2] = vw1->vp[2];
344			setident4(m4t);
345			m4t[0][0] = vw2->vhinc[0]/vw2->vhn2;
346			m4t[1][0] = vw2->vhinc[1]/vw2->vhn2;
347			m4t[2][0] = vw2->vhinc[2]/vw2->vhn2;
348			m4t[3][0] = -DOT(vw2->vp,vw2->vhinc)/vw2->vhn2;
349			m4t[0][1] = vw2->vvinc[0]/vw2->vvn2;
350			m4t[1][1] = vw2->vvinc[1]/vw2->vvn2;
351			m4t[2][1] = vw2->vvinc[2]/vw2->vvn2;
352			m4t[3][1] = -DOT(vw2->vp,vw2->vvinc)/vw2->vvn2;
353			m4t[0][2] = vw2->vdir[0];
354			m4t[1][2] = vw2->vdir[1];
355			m4t[2][2] = vw2->vdir[2];
356			m4t[3][2] = -DOT(vw2->vp,vw2->vdir);
357			multmat4(xfmat, xfmat, m4t);
358			}
359
360
361	greg	1.8	addscanline(y, pline, zline, lasty, lastyz) /* add scanline to output */
362	greg	1.1	int y;
363			COLR *pline;
364			float *zline;
365	greg	1.9	int lasty; / input/output */
366			float lastyz; / input/output */
367	greg	1.1	{
368	greg	1.11	extern double sqrt();
369	greg	1.5	double pos[3];
370	greg	1.8	int lastx = 0;
371			double lastxz = 0;
372			double zt;
373			int xpos, ypos;
374	greg	1.1	register int x;
375
376	greg	1.8	for (x = theirview.hresolu-1; x >= 0; x--) {
377	greg	1.5	pos[0] = x - .5*(theirview.hresolu-1);
378			pos[1] = y - .5*(theirview.vresolu-1);
379			pos[2] = zline[x];
380			if (theirview.type == VT_PER) {
381	greg	1.11	if (normdist) /* adjust for eye-ray distance */
382	greg	1.10	pos[2] /= sqrt( 1.
383	greg	1.5	+ pos[0]pos[0]theirview.vhn2
384			+ pos[1]pos[1]theirview.vvn2 );
385			pos[0] *= pos[2];
386			pos[1] *= pos[2];
387			}
388			multp3(pos, pos, theirs2ours);
389	greg	1.8	if (pos[2] <= 0)
390	greg	1.1	continue;
391	greg	1.5	if (ourview.type == VT_PER) {
392			pos[0] /= pos[2];
393			pos[1] /= pos[2];
394			}
395			pos[0] += .5*ourview.hresolu;
396			pos[1] += .5*ourview.vresolu;
397	greg	1.8	if (pos[0] < 0 \|\| (xpos = pos[0]) >= ourview.hresolu
398			\|\| pos[1] < 0 \|\| (ypos = pos[1]) >= ourview.vresolu)
399	greg	1.5	continue;
400	greg	1.8	/* add pixel to our image */
401			zt = 2.zepszline[x];
402			addpixel(xpos, ypos,
403	greg	1.11	(fill&F_FORE && ABS(zline[x]-lastxz) <= zt)
404			? lastx - xpos : 1,
405			(fill&F_FORE && ABS(zline[x]-lastyz[x]) <= zt)
406			? lasty[x] - ypos : 1,
407	greg	1.8	pline[x], pos[2]);
408			lastx = xpos;
409			lasty[x] = ypos;
410			lastxz = lastyz[x] = zline[x];
411	greg	1.1	}
412			}
413
414
415	greg	1.8	addpixel(xstart, ystart, width, height, pix, z) /* fill in area for pixel */
416			int xstart, ystart;
417			int width, height;
418			COLR pix;
419			double z;
420			{
421			register int x, y;
422			/* make width and height positive */
423			if (width < 0) {
424			width = -width;
425			xstart = xstart-width+1;
426			} else if (width == 0)
427			width = 1;
428			if (height < 0) {
429			height = -height;
430			ystart = ystart-height+1;
431			} else if (height == 0)
432			height = 1;
433			/* fill pixel(s) within rectangle */
434			for (y = ystart; y < ystart+height; y++)
435			for (x = xstart; x < xstart+width; x++)
436			if (zscan(y)[x] <= 0
437			\|\| zscan(y)[x]-z > zeps*zscan(y)[x]) {
438			zscan(y)[x] = z;
439			copycolr(pscan(y)[x], pix);
440			}
441			}
442
443
444	greg	1.12	backpicture() /* background fill algorithm */
445	greg	1.1	{
446	greg	1.2	int *yback, xback;
447	greg	1.1	int y;
448	greg	1.2	COLR pfill;
449			register int x, i;
450			/* get back buffer */
451			yback = (int )malloc(ourview.hresolusizeof(int));
452			if (yback == NULL) {
453			perror(progname);
454			return;
455			}
456			for (x = 0; x < ourview.hresolu; x++)
457	greg	1.4	yback[x] = -2;
458	greg	1.7	/*
459			* Xback and yback are the pixel locations of suitable
460			* background values in each direction.
461			* A value of -2 means unassigned, and -1 means
462			* that there is no suitable background in this direction.
463			*/
464	greg	1.2	/* fill image */
465	greg	1.4	for (y = 0; y < ourview.vresolu; y++) {
466			xback = -2;
467	greg	1.1	for (x = 0; x < ourview.hresolu; x++)
468	greg	1.8	if (zscan(y)[x] <= 0) { /* empty pixel */
469	greg	1.7	/*
470			* First, find background from above or below.
471			* (farthest assigned pixel)
472			*/
473	greg	1.4	if (yback[x] == -2) {
474			for (i = y+1; i < ourview.vresolu; i++)
475	greg	1.8	if (zscan(i)[x] > 0)
476	greg	1.4	break;
477			if (i < ourview.vresolu
478	greg	1.2	&& (y <= 0 \|\| zscan(y-1)[x] < zscan(i)[x]))
479	greg	1.4	yback[x] = i;
480			else
481			yback[x] = y-1;
482	greg	1.2	}
483	greg	1.7	/*
484			* Next, find background from left or right.
485			*/
486	greg	1.4	if (xback == -2) {
487	greg	1.11	for (i = x+1; i < ourview.hresolu; i++)
488	greg	1.8	if (zscan(y)[i] > 0)
489	greg	1.4	break;
490			if (i < ourview.hresolu
491			&& (x <= 0 \|\| zscan(y)[x-1] < zscan(y)[i]))
492			xback = i;
493			else
494			xback = x-1;
495			}
496	greg	1.7	/*
497	greg	1.11	* Check to see if we have no background for
498			* this pixel. If not, use background color.
499			*/
500			if (xback < 0 && yback[x] < 0) {
501	greg	1.12	(*deffill)(x,y);
502	greg	1.11	continue;
503			}
504			/*
505	greg	1.7	* Compare, and use the background that is
506			* farther, unless one of them is next to us.
507			*/
508	greg	1.11	if ( yback[x] < 0
509			\|\| (xback >= 0 && ABS(x-xback) <= 1)
510	greg	1.4	\|\| ( ABS(y-yback[x]) > 1
511	greg	1.11	&& zscan(yback[x])[x]
512	greg	1.13	< zscan(y)[xback] ) ) {
513	greg	1.4	copycolr(pscan(y)[x],pscan(y)[xback]);
514	greg	1.13	zscan(y)[x] = zscan(y)[xback];
515			} else {
516	greg	1.4	copycolr(pscan(y)[x],pscan(yback[x])[x]);
517	greg	1.13	zscan(y)[x] = zscan(yback[x])[x];
518			}
519	greg	1.4	} else { /* full pixel */
520			yback[x] = -2;
521			xback = -2;
522			}
523			}
524	greg	1.2	free((char *)yback);
525	greg	1.1	}
526
527
528	greg	1.12	fillpicture() /* paint in empty pixels with default */
529	greg	1.11	{
530			register int x, y;
531
532			for (y = 0; y < ourview.vresolu; y++)
533			for (x = 0; x < ourview.hresolu; x++)
534			if (zscan(y)[x] <= 0)
535	greg	1.12	(*deffill)(x,y);
536	greg	1.11	}
537
538
539	greg	1.1	writepicture() /* write out picture */
540			{
541			int y;
542
543			fputresolu(YMAJOR\|YDECR, ourview.hresolu, ourview.vresolu, stdout);
544			for (y = ourview.vresolu-1; y >= 0; y--)
545			if (fwritecolrs(pscan(y), ourview.hresolu, stdout) < 0) {
546			perror(progname);
547			exit(1);
548			}
549	greg	1.11	}
550
551
552			writedistance(fname) /* write out z file */
553			char *fname;
554			{
555			extern double sqrt();
556			int donorm = normdist && ourview.type == VT_PER;
557			FILE *fp;
558			int y;
559			float *zout;
560
561			if ((fp = fopen(fname, "w")) == NULL) {
562			perror(fname);
563			exit(1);
564			}
565			if (donorm
566			&& (zout = (float )malloc(ourview.hresolusizeof(float))) == NULL) {
567			perror(progname);
568			exit(1);
569			}
570			for (y = ourview.vresolu-1; y >= 0; y--) {
571			if (donorm) {
572			double vx, yzn2;
573			register int x;
574			yzn2 = y - .5*(ourview.vresolu-1);
575			yzn2 = 1. + yzn2yzn2ourview.vvn2;
576			for (x = 0; x < ourview.hresolu; x++) {
577			vx = x - .5*(ourview.hresolu-1);
578			zout[x] = zscan(y)[x]
579			* sqrt(vxvxourview.vhn2 + yzn2);
580			}
581			} else
582			zout = zscan(y);
583			if (fwrite(zout, sizeof(float), ourview.hresolu, fp)
584			< ourview.hresolu) {
585			perror(fname);
586			exit(1);
587			}
588			}
589			if (donorm)
590			free((char *)zout);
591			fclose(fp);
592	greg	1.10	}
593
594
595			isfloat(s) /* see if string is floating number */
596			register char *s;
597			{
598			for ( ; *s; s++)
599			if ((s < '0' \|\| s > '9') && s != '.' && s != '-'
600			&& s != 'e' && s != 'E' && *s != '+')
601			return(0);
602			return(1);
603	greg	1.12	}
604
605
606			backfill(x, y) /* fill pixel with background */
607			int x, y;
608			{
609			register BYTE *dest = pscan(y)[x];
610
611			copycolr(dest, backcolr);
612	greg	1.13	zscan(y)[x] = backz;
613			}
614
615
616			calstart(prog, args) /* start fill calculation */
617			char prog, args;
618			{
619			char combuf[512];
620			int p0[2], p1[2];
621
622	greg	1.14	if (childpid != -1) {
623			fprintf(stderr, "%s: too many calculations\n", progname);
624			exit(1);
625			}
626	greg	1.13	sprintf(combuf, prog, PACKSIZ, args);
627			if (pipe(p0) < 0 \|\| pipe(p1) < 0)
628	greg	1.14	syserror();
629	greg	1.13	if ((childpid = vfork()) == 0) { /* fork calculation */
630			close(p0[1]);
631			close(p1[0]);
632			if (p0[0] != 0) {
633			dup2(p0[0], 0);
634			close(p0[0]);
635			}
636			if (p1[1] != 1) {
637			dup2(p1[1], 1);
638			close(p1[1]);
639			}
640			execl("/bin/sh", "sh", "-c", combuf, 0);
641			perror("/bin/sh");
642			_exit(127);
643			}
644	greg	1.14	if (childpid == -1)
645			syserror();
646	greg	1.13	close(p0[0]);
647			close(p1[1]);
648			if ((psend = fdopen(p0[1], "w")) == NULL)
649	greg	1.14	syserror();
650	greg	1.13	if ((precv = fdopen(p1[0], "r")) == NULL)
651	greg	1.14	syserror();
652	greg	1.13	queuesiz = 0;
653			}
654
655
656			caldone() /* done with calculation */
657			{
658			int pid;
659
660	greg	1.14	if (childpid == -1)
661			return;
662			if (fclose(psend) == EOF)
663			syserror();
664	greg	1.13	clearqueue();
665			fclose(precv);
666			while ((pid = wait(0)) != -1 && pid != childpid)
667			;
668	greg	1.14	childpid = -1;
669	greg	1.13	}
670
671
672	greg	1.14	rcalfill(x, y) /* fill with ray-calculated pixel */
673	greg	1.13	int x, y;
674			{
675			FVECT orig, dir;
676			float outbuf[6];
677
678			if (queuesiz >= PACKSIZ) { /* flush queue */
679	greg	1.14	if (fflush(psend) == EOF)
680			syserror();
681	greg	1.13	clearqueue();
682			}
683			/* send new ray */
684			rayview(orig, dir, &ourview, x+.5, y+.5);
685			outbuf[0] = orig[0]; outbuf[1] = orig[1]; outbuf[2] = orig[2];
686			outbuf[3] = dir[0]; outbuf[4] = dir[1]; outbuf[5] = dir[2];
687	greg	1.14	if (fwrite(outbuf, sizeof(float), 6, psend) < 6)
688			syserror();
689	greg	1.13	/* remember it */
690			queue[queuesiz][0] = x;
691			queue[queuesiz][1] = y;
692			queuesiz++;
693			}
694
695
696			clearqueue() /* get results from queue */
697			{
698			float inbuf[4];
699			register int i;
700
701			for (i = 0; i < queuesiz; i++) {
702	greg	1.14	if (fread(inbuf, sizeof(float), 4, precv) < 4) {
703			fprintf(stderr, "%s: read error in clearqueue\n",
704			progname);
705			exit(1);
706			}
707	greg	1.13	setcolr(pscan(queue[i][1])[queue[i][0]],
708			inbuf[0], inbuf[1], inbuf[2]);
709			zscan(queue[i][1])[queue[i][0]] = inbuf[3];
710			}
711			queuesiz = 0;
712	greg	1.14	}
713
714
715			syserror() /* report error and exit */
716			{
717			perror(progname);
718			exit(1);
719	greg	1.1	}