src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.79 2019/05/06 16:49:38 greg Exp $";
#endif
/*
 *  rtrace.c - program and variables for individual ray tracing.
 */

#include "copyright.h"

/*
 *  Input is in the form:
 *
 *      xorg    yorg    zorg    xdir    ydir    zdir
 *
 *  The direction need not be normalized.  Output is flexible.
 *  If the direction vector is (0,0,0), then the output is flushed.
 *  All values default to ascii representation of real
 *  numbers.  Binary representations can be selected
 *  with '-ff' for float or '-fd' for double.  By default,
 *  radiance is computed.  The '-i' or '-I' options indicate that
 *  irradiance values are desired.
 */

#include  <time.h>

#include  "platform.h"
#include  "ray.h"
#include  "ambient.h"
#include  "source.h"
#include  "otypes.h"
#include  "otspecial.h"
#include  "resolu.h"
#include  "random.h"

extern int  inform;                     /* input format */
extern int  outform;                    /* output format */
extern char  *outvals;                  /* output values */

extern int  imm_irrad;                  /* compute immediate irradiance? */
extern int  lim_dist;                   /* limit distance? */

extern char  *tralist[];                /* list of modifers to trace (or no) */
extern int  traincl;                    /* include == 1, exclude == 0 */

extern int  hresolu;                    /* horizontal resolution */
extern int  vresolu;                    /* vertical resolution */

static int  castonly = 0;

#ifndef  MAXTSET
#define  MAXTSET        8191            /* maximum number in trace set */
#endif
OBJECT  traset[MAXTSET+1]={0};          /* trace include/exclude set */

static RAY  thisray;                    /* for our convenience */

typedef void putf_t(RREAL *v, int n);
static putf_t puta, putd, putf, putrgbe;

typedef void oputf_t(RAY *r);
static oputf_t  oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
                oputr, oputR, oputx, oputX, oputn, oputN, oputs,
                oputw, oputW, oputm, oputM, oputtilde;

static void setoutput(char *vs);
extern void tranotify(OBJECT obj);
static void bogusray(void);
static void raycast(RAY *r);
static void rayirrad(RAY *r);
static void rtcompute(FVECT org, FVECT dir, double dmax);
static int printvals(RAY *r);
static int getvec(FVECT vec, int fmt, FILE *fp);
static void tabin(RAY *r);
static void ourtrace(RAY *r);

static oputf_t *ray_out[32], *every_out[32];
static putf_t *putreal;


void
quit(                   /* quit program */
        int  code
)
{
        if (ray_pnprocs > 0)    /* close children if any */
                ray_pclose(0);          
#ifndef  NON_POSIX
        else if (!ray_pnprocs) {
                headclean();    /* delete header file */
                pfclean();      /* clean up persist files */
        }
#endif
        exit(code);
}


char *
formstr(                                /* return format identifier */
        int  f
)
{
        switch (f) {
        case 'a': return("ascii");
        case 'f': return("float");
        case 'd': return("double");
        case 'c': return(COLRFMT);
        }
        return("unknown");
}


extern void
rtrace(                         /* trace rays from file */
        char  *fname,
        int  nproc
)
{
        unsigned long  vcount = (hresolu > 1) ? (unsigned long)hresolu*vresolu
                                              : (unsigned long)vresolu;
        long  nextflush = (vresolu > 0) & (hresolu > 1) ? 0 : hresolu;
        int  something2flush = 0;
        FILE  *fp;
        double  d;
        FVECT  orig, direc;
                                        /* set up input */
        if (fname == NULL)
                fp = stdin;
        else if ((fp = fopen(fname, "r")) == NULL) {
                sprintf(errmsg, "cannot open input file \"%s\"", fname);
                error(SYSTEM, errmsg);
        }
        if (inform != 'a')
                SET_FILE_BINARY(fp);
                                        /* set up output */
        setoutput(outvals);
        if (imm_irrad)
                castonly = 0;
        else if (castonly)
                nproc = 1;              /* don't bother multiprocessing */
        if ((nextflush > 0) & (nproc > nextflush)) {
                error(WARNING, "reducing number of processes to match flush interval");
                nproc = nextflush;
        }
        switch (outform) {
        case 'a': putreal = puta; break;
        case 'f': putreal = putf; break;
        case 'd': putreal = putd; break;
        case 'c': 
                if (outvals[0] && (outvals[1] || !strchr("vVWrx", outvals[0])))
                        error(USER, "color format only with one of 'vVWrx'");
                putreal = putrgbe; break;
        default:
                error(CONSISTENCY, "botched output format");
        }
        if (nproc > 1) {                /* start multiprocessing */
                ray_popen(nproc);
                ray_fifo_out = printvals;
        }
        if (hresolu > 0) {
                if (vresolu > 0)
                        fprtresolu(hresolu, vresolu, stdout);
                else
                        fflush(stdout);
        }
                                        /* process file */
        while (getvec(orig, inform, fp) == 0 &&
                        getvec(direc, inform, fp) == 0) {

                d = normalize(direc);
                if (d == 0.0) {                         /* zero ==> flush */
                        if ((--nextflush <= 0) | !vcount && something2flush) {
                                if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
                                        error(USER, "child(ren) died");
                                bogusray();
                                fflush(stdout);
                                something2flush = 0;
                                nextflush = (vresolu > 0) & (hresolu > 1) ? 0 :
                                                                hresolu;
                        } else
                                bogusray();
                } else {                                /* compute and print */
                        rtcompute(orig, direc, lim_dist ? d : 0.0);
                                                        /* flush if time */
                        if (!--nextflush) {
                                if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
                                        error(USER, "child(ren) died");
                                fflush(stdout);
                                nextflush = hresolu;
                        } else
                                something2flush = 1;
                }
                if (ferror(stdout))
                        error(SYSTEM, "write error");
                if (vcount && !--vcount)                /* check for end */
                        break;
        }
        if (ray_pnprocs > 1) {                          /* clean up children */
                if (ray_fifo_flush() < 0)
                        error(USER, "unable to complete processing");
                ray_pclose(0);
        }
        if (fflush(stdout) < 0)
                error(SYSTEM, "write error");
        if (vcount)
                error(USER, "unexpected EOF on input");
        if (fname != NULL)
                fclose(fp);
}


static void
trace_sources(void)                     /* trace rays to light sources, also */
{
        int     sn;
        
        for (sn = 0; sn < nsources; sn++)
                source[sn].sflags |= SFOLLOW;
}


static void
setoutput(                              /* set up output tables */
        char  *vs
)
{
        oputf_t **table = ray_out;

        castonly = 1;
        while (*vs)
                switch (*vs++) {
                case 'T':                               /* trace sources */
                        if (!*vs) break;
                        trace_sources();
                        /* fall through */
                case 't':                               /* trace */
                        if (!*vs) break;
                        *table = NULL;
                        table = every_out;
                        trace = ourtrace;
                        castonly = 0;
                        break;
                case 'o':                               /* origin */
                        *table++ = oputo;
                        break;
                case 'd':                               /* direction */
                        *table++ = oputd;
                        break;
                case 'r':                               /* reflected contrib. */
                        *table++ = oputr;
                        castonly = 0;
                        break;
                case 'R':                               /* reflected distance */
                        *table++ = oputR;
                        castonly = 0;
                        break;
                case 'x':                               /* xmit contrib. */
                        *table++ = oputx;
                        castonly = 0;
                        break;
                case 'X':                               /* xmit distance */
                        *table++ = oputX;
                        castonly = 0;
                        break;
                case 'v':                               /* value */
                        *table++ = oputv;
                        castonly = 0;
                        break;
                case 'V':                               /* contribution */
                        *table++ = oputV;
                        castonly = 0;
                        if (ambounce > 0 && (ambacc > FTINY || ambssamp > 0))
                                error(WARNING,
                                        "-otV accuracy depends on -aa 0 -as 0");
                        break;
                case 'l':                               /* effective distance */
                        *table++ = oputl;
                        castonly = 0;
                        break;
                case 'c':                               /* local coordinates */
                        *table++ = oputc;
                        break;
                case 'L':                               /* single ray length */
                        *table++ = oputL;
                        break;
                case 'p':                               /* point */
                        *table++ = oputp;
                        break;
                case 'n':                               /* perturbed normal */
                        *table++ = oputn;
                        castonly = 0;
                        break;
                case 'N':                               /* unperturbed normal */
                        *table++ = oputN;
                        break;
                case 's':                               /* surface */
                        *table++ = oputs;
                        break;
                case 'w':                               /* weight */
                        *table++ = oputw;
                        break;
                case 'W':                               /* coefficient */
                        *table++ = oputW;
                        castonly = 0;
                        if (ambounce > 0 && (ambacc > FTINY) | (ambssamp > 0))
                                error(WARNING,
                                        "-otW accuracy depends on -aa 0 -as 0");
                        break;
                case 'm':                               /* modifier */
                        *table++ = oputm;
                        break;
                case 'M':                               /* material */
                        *table++ = oputM;
                        break;
                case '~':                               /* tilde */
                        *table++ = oputtilde;
                        break;
                }
        *table = NULL;
        if (do_irrad | imm_irrad)                       /* compatibile? */
                for (table = ray_out; *table != NULL; table++)
                        if ((*table == oputr) | (*table == oputR) |
                                        (*table == oputx) | (*table == oputX)) {
                                error(WARNING, "-orRxX options incompatible with -I+ and -i+");
                                break;
                        }
}


static void
bogusray(void)                  /* print out empty record */
{
        rayorigin(&thisray, PRIMARY, NULL, NULL);
        printvals(&thisray);
}


static void
raycast(                        /* compute first ray intersection only */
        RAY *r
)
{
        if (!localhit(r, &thescene)) {
                if (r->ro == &Aftplane) {       /* clipped */
                        r->ro = NULL;
                        r->rot = FHUGE;
                } else
                        sourcehit(r);
        }
}


static void
rayirrad(                       /* compute irradiance rather than radiance */
        RAY *r
)
{
        void    (*old_revf)(RAY *) = r->revf;
                                        /* pretend we hit surface */
        r->rxt = r->rot = 1e-5;
        VSUM(r->rop, r->rorg, r->rdir, r->rot);
        r->ron[0] = -r->rdir[0];
        r->ron[1] = -r->rdir[1];
        r->ron[2] = -r->rdir[2];
        r->rod = 1.0;
                                        /* compute result */
        r->revf = raytrace;
        (*ofun[Lamb.otype].funp)(&Lamb, r);
        r->revf = old_revf;
}


static void
rtcompute(                      /* compute and print ray value(s) */
        FVECT  org,
        FVECT  dir,
        double  dmax
)
{
                                        /* set up ray */
        rayorigin(&thisray, PRIMARY, NULL, NULL);
        if (imm_irrad) {
                VSUM(thisray.rorg, org, dir, 1.1e-4);
                thisray.rdir[0] = -dir[0];
                thisray.rdir[1] = -dir[1];
                thisray.rdir[2] = -dir[2];
                thisray.rmax = 0.0;
                thisray.revf = rayirrad;
        } else {
                VCOPY(thisray.rorg, org);
                VCOPY(thisray.rdir, dir);
                thisray.rmax = dmax;
                if (castonly)
                        thisray.revf = raycast;
        }
        if (ray_pnprocs > 1) {          /* multiprocessing FIFO? */
                if (ray_fifo_in(&thisray) < 0)
                        error(USER, "lost children");
                return;
        }
        samplendx++;                    /* else do it ourselves */
        rayvalue(&thisray);
        printvals(&thisray);
}


static int
printvals(                      /* print requested ray values */
        RAY  *r
)
{
        oputf_t **tp;

        if (ray_out[0] == NULL)
                return(0);
        for (tp = ray_out; *tp != NULL; tp++)
                (**tp)(r);
        if (outform == 'a')
                putchar('\n');
        return(1);
}


static int
getvec(         /* get a vector from fp */
        FVECT  vec,
        int  fmt,
        FILE  *fp
)
{
        static float  vf[3];
        static double  vd[3];
        char  buf[32];
        int  i;

        switch (fmt) {
        case 'a':                                       /* ascii */
                for (i = 0; i < 3; i++) {
                        if (fgetword(buf, sizeof(buf), fp) == NULL ||
                                        !isflt(buf))
                                return(-1);
                        vec[i] = atof(buf);
                }
                break;
        case 'f':                                       /* binary float */
                if (getbinary(vf, sizeof(float), 3, fp) != 3)
                        return(-1);
                VCOPY(vec, vf);
                break;
        case 'd':                                       /* binary double */
                if (getbinary(vd, sizeof(double), 3, fp) != 3)
                        return(-1);
                VCOPY(vec, vd);
                break;
        default:
                error(CONSISTENCY, "botched input format");
        }
        return(0);
}


void
tranotify(                      /* record new modifier */
        OBJECT  obj
)
{
        static int  hitlimit = 0;
        OBJREC   *o = objptr(obj);
        char  **tralp;

        if (obj == OVOID) {             /* starting over */
                traset[0] = 0;
                hitlimit = 0;
                return;
        }
        if (hitlimit || !ismodifier(o->otype))
                return;
        for (tralp = tralist; *tralp != NULL; tralp++)
                if (!strcmp(o->oname, *tralp)) {
                        if (traset[0] >= MAXTSET) {
                                error(WARNING, "too many modifiers in trace list");
                                hitlimit++;
                                return;         /* should this be fatal? */
                        }
                        insertelem(traset, obj);
                        return;
                }
}


static void
ourtrace(                               /* print ray values */
        RAY  *r
)
{
        oputf_t **tp;

        if (every_out[0] == NULL)
                return;
        if (r->ro == NULL) {
                if (traincl == 1)
                        return;
        } else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
                return;
        tabin(r);
        for (tp = every_out; *tp != NULL; tp++)
                (**tp)(r);
        if (outform == 'a')
                putchar('\n');
}


static void
tabin(                          /* tab in appropriate amount */
        RAY  *r
)
{
        const RAY  *rp;

        for (rp = r->parent; rp != NULL; rp = rp->parent)
                putchar('\t');
}


static void
oputo(                          /* print origin */
        RAY  *r
)
{
        (*putreal)(r->rorg, 3);
}


static void
oputd(                          /* print direction */
        RAY  *r
)
{
        (*putreal)(r->rdir, 3);
}


static void
oputr(                          /* print mirrored contribution */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->mcol,RED);
        cval[1] = colval(r->mcol,GRN);
        cval[2] = colval(r->mcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputR(                          /* print mirrored distance */
        RAY  *r
)
{
        (*putreal)(&r->rmt, 1);
}


static void
oputx(                          /* print unmirrored contribution */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->rcol,RED) - colval(r->mcol,RED);
        cval[1] = colval(r->rcol,GRN) - colval(r->mcol,GRN);
        cval[2] = colval(r->rcol,BLU) - colval(r->mcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputX(                          /* print unmirrored distance */
        RAY  *r
)
{
        (*putreal)(&r->rxt, 1);
}


static void
oputv(                          /* print value */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->rcol,RED);
        cval[1] = colval(r->rcol,GRN);
        cval[2] = colval(r->rcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputV(                          /* print value contribution */
        RAY *r
)
{
        RREAL   contr[3];

        raycontrib(contr, r, PRIMARY);
        multcolor(contr, r->rcol);
        (*putreal)(contr, 3);
}


static void
oputl(                          /* print effective distance */
        RAY  *r
)
{
        RREAL   d = raydistance(r);

        (*putreal)(&d, 1);
}


static void
oputL(                          /* print single ray length */
        RAY  *r
)
{
        (*putreal)(&r->rot, 1);
}


static void
oputc(                          /* print local coordinates */
        RAY  *r
)
{
        (*putreal)(r->uv, 2);
}


static RREAL    vdummy[3] = {0.0, 0.0, 0.0};


static void
oputp(                          /* print point */
        RAY  *r
)
{
        if (r->rot < FHUGE)
                (*putreal)(r->rop, 3);
        else
                (*putreal)(vdummy, 3);
}


static void
oputN(                          /* print unperturbed normal */
        RAY  *r
)
{
        if (r->rot < FHUGE)
                (*putreal)(r->ron, 3);
        else
                (*putreal)(vdummy, 3);
}


static void
oputn(                          /* print perturbed normal */
        RAY  *r
)
{
        FVECT  pnorm;

        if (r->rot >= FHUGE) {
                (*putreal)(vdummy, 3);
                return;
        }
        raynormal(pnorm, r);
        (*putreal)(pnorm, 3);
}


static void
oputs(                          /* print name */
        RAY  *r
)
{
        if (r->ro != NULL)
                fputs(r->ro->oname, stdout);
        else
                putchar('*');
        putchar('\t');
}


static void
oputw(                          /* print weight */
        RAY  *r
)
{
        RREAL   rwt = r->rweight;
        
        (*putreal)(&rwt, 1);
}


static void
oputW(                          /* print coefficient */
        RAY  *r
)
{
        RREAL   contr[3];
                                /* shadow ray not on source? */
        if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
                setcolor(contr, 0.0, 0.0, 0.0);
        else
                raycontrib(contr, r, PRIMARY);

        (*putreal)(contr, 3);
}


static void
oputm(                          /* print modifier */
        RAY  *r
)
{
        if (r->ro != NULL)
                if (r->ro->omod != OVOID)
                        fputs(objptr(r->ro->omod)->oname, stdout);
                else
                        fputs(VOIDID, stdout);
        else
                putchar('*');
        putchar('\t');
}


static void
oputM(                          /* print material */
        RAY  *r
)
{
        OBJREC  *mat;

        if (r->ro != NULL) {
                if ((mat = findmaterial(r->ro)) != NULL)
                        fputs(mat->oname, stdout);
                else
                        fputs(VOIDID, stdout);
        } else
                putchar('*');
        putchar('\t');
}


static void
oputtilde(                      /* output tilde (spacer) */
        RAY  *r
)
{
        fputs("~\t", stdout);
}


static void
puta(                           /* print ascii value(s) */
        RREAL *v, int n
)
{
        if (n == 3) {
                printf("%e\t%e\t%e\t", v[0], v[1], v[2]);
                return;
        }
        while (n--)
                printf("%e\t", *v++);
}


static void
putd(RREAL *v, int n)           /* output binary double(s) */
{
#ifdef  SMLFLT
        double  da[3];
        int     i;

        if (n > 3)
                error(INTERNAL, "code error in putd()");
        for (i = n; i--; )
                da[i] = v[i];
        putbinary(da, sizeof(double), n, stdout);
#else
        putbinary(v, sizeof(RREAL), n, stdout);
#endif
}


static void
putf(RREAL *v, int n)           /* output binary float(s) */
{
#ifndef SMLFLT
        float   fa[3];
        int     i;

        if (n > 3)
                error(INTERNAL, "code error in putf()");
        for (i = n; i--; )
                fa[i] = v[i];
        putbinary(fa, sizeof(float), n, stdout);
#else
        putbinary(v, sizeof(RREAL), n, stdout);
#endif
}


static void
putrgbe(RREAL *v, int n)        /* output RGBE color */
{
        COLR  cout;

        if (n != 3)
                error(INTERNAL, "putrgbe() not called with 3 components");
        setcolr(cout, v[0], v[1], v[2]);
        putbinary(cout, sizeof(cout), 1, stdout);
}
Revision:	2.80
Committed:	Thu Jul 4 01:06:35 2019 UTC (4 years, 10 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.79:	+27 -16 lines
Log Message:	Made -f?c RGBE output compatible with any of 'vVWrx' options and added warning
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.80	static const char RCSid[] = "$Id: rtrace.c,v 2.79 2019/05/06 16:49:38 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* rtrace.c - program and variables for individual ray tracing.
6	greg	2.27	*/
7
8	greg	2.28	#include "copyright.h"
9	greg	1.1
10			/*
11			* Input is in the form:
12			*
13			* xorg yorg zorg xdir ydir zdir
14			*
15			* The direction need not be normalized. Output is flexible.
16	greg	1.7	* If the direction vector is (0,0,0), then the output is flushed.
17	greg	1.1	* All values default to ascii representation of real
18			* numbers. Binary representations can be selected
19			* with '-ff' for float or '-fd' for double. By default,
20	greg	1.13	* radiance is computed. The '-i' or '-I' options indicate that
21	greg	1.1	* irradiance values are desired.
22			*/
23	greg	2.32
24			#include <time.h>
25	greg	1.1
26	schorsch	2.30	#include "platform.h"
27	greg	1.1	#include "ray.h"
28	schorsch	2.37	#include "ambient.h"
29			#include "source.h"
30	greg	1.1	#include "otypes.h"
31	greg	2.72	#include "otspecial.h"
32	greg	2.6	#include "resolu.h"
33	schorsch	2.49	#include "random.h"
34	greg	2.6
35	greg	2.57	extern int inform; /* input format */
36			extern int outform; /* output format */
37			extern char outvals; / output values */
38	greg	2.27
39	greg	2.57	extern int imm_irrad; /* compute immediate irradiance? */
40			extern int lim_dist; /* limit distance? */
41	greg	1.14
42	greg	2.57	extern char tralist[]; / list of modifers to trace (or no) */
43			extern int traincl; /* include == 1, exclude == 0 */
44	greg	1.13
45	greg	2.57	extern int hresolu; /* horizontal resolution */
46			extern int vresolu; /* vertical resolution */
47	greg	1.1
48	greg	2.57	static int castonly = 0;
49	greg	2.55
50	greg	2.46	#ifndef MAXTSET
51	greg	2.55	#define MAXTSET 8191 /* maximum number in trace set */
52	greg	2.46	#endif
53	greg	2.15	OBJECT traset[MAXTSET+1]={0}; /* trace include/exclude set */
54
55	greg	1.1	static RAY thisray; /* for our convenience */
56
57	greg	2.65	typedef void putf_t(RREAL *v, int n);
58	greg	2.80	static putf_t puta, putd, putf, putrgbe;
59	greg	1.1
60	schorsch	2.37	typedef void oputf_t(RAY *r);
61	greg	2.51	static oputf_t oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
62	greg	2.73	oputr, oputR, oputx, oputX, oputn, oputN, oputs,
63			oputw, oputW, oputm, oputM, oputtilde;
64	schorsch	2.37
65			static void setoutput(char *vs);
66	greg	2.57	extern void tranotify(OBJECT obj);
67	schorsch	2.37	static void bogusray(void);
68	greg	2.59	static void raycast(RAY *r);
69			static void rayirrad(RAY *r);
70			static void rtcompute(FVECT org, FVECT dir, double dmax);
71	greg	2.57	static int printvals(RAY *r);
72			static int getvec(FVECT vec, int fmt, FILE *fp);
73			static void tabin(RAY *r);
74			static void ourtrace(RAY *r);
75	greg	1.1
76	greg	2.73	static oputf_t ray_out[32], every_out[32];
77	schorsch	2.37	static putf_t *putreal;
78	greg	2.27
79	greg	1.1
80	greg	2.27	void
81	schorsch	2.37	quit( /* quit program */
82			int code
83			)
84	greg	1.1	{
85	greg	2.57	if (ray_pnprocs > 0) /* close children if any */
86			ray_pclose(0);
87			#ifndef NON_POSIX
88	greg	2.58	else if (!ray_pnprocs) {
89			headclean(); /* delete header file */
90			pfclean(); /* clean up persist files */
91			}
92	greg	2.11	#endif
93	greg	1.1	exit(code);
94			}
95
96
97	greg	2.57	char *
98	schorsch	2.37	formstr( /* return format identifier */
99			int f
100			)
101	greg	2.6	{
102			switch (f) {
103			case 'a': return("ascii");
104			case 'f': return("float");
105			case 'd': return("double");
106			case 'c': return(COLRFMT);
107			}
108			return("unknown");
109			}
110
111
112	schorsch	2.37	extern void
113			rtrace( /* trace rays from file */
114	greg	2.59	char *fname,
115			int nproc
116	schorsch	2.37	)
117	greg	1.1	{
118	greg	2.45	unsigned long vcount = (hresolu > 1) ? (unsigned long)hresolu*vresolu
119	greg	2.66	: (unsigned long)vresolu;
120	greg	2.64	long nextflush = (vresolu > 0) & (hresolu > 1) ? 0 : hresolu;
121	greg	2.79	int something2flush = 0;
122	greg	1.1	FILE *fp;
123	gregl	2.25	double d;
124	greg	1.2	FVECT orig, direc;
125	greg	1.1	/* set up input */
126			if (fname == NULL)
127			fp = stdin;
128			else if ((fp = fopen(fname, "r")) == NULL) {
129			sprintf(errmsg, "cannot open input file \"%s\"", fname);
130			error(SYSTEM, errmsg);
131			}
132	greg	2.8	if (inform != 'a')
133	schorsch	2.30	SET_FILE_BINARY(fp);
134	greg	1.1	/* set up output */
135	greg	2.16	setoutput(outvals);
136	greg	2.59	if (imm_irrad)
137			castonly = 0;
138			else if (castonly)
139			nproc = 1; /* don't bother multiprocessing */
140	greg	2.79	if ((nextflush > 0) & (nproc > nextflush)) {
141			error(WARNING, "reducing number of processes to match flush interval");
142			nproc = nextflush;
143			}
144	greg	1.1	switch (outform) {
145			case 'a': putreal = puta; break;
146			case 'f': putreal = putf; break;
147			case 'd': putreal = putd; break;
148	greg	2.6	case 'c':
149	greg	2.80	if (outvals[0] && (outvals[1] \|\| !strchr("vVWrx", outvals[0])))
150			error(USER, "color format only with one of 'vVWrx'");
151			putreal = putrgbe; break;
152	greg	2.6	default:
153			error(CONSISTENCY, "botched output format");
154	greg	1.1	}
155	greg	2.59	if (nproc > 1) { /* start multiprocessing */
156			ray_popen(nproc);
157	greg	2.58	ray_fifo_out = printvals;
158	greg	2.59	}
159	greg	2.12	if (hresolu > 0) {
160			if (vresolu > 0)
161			fprtresolu(hresolu, vresolu, stdout);
162	greg	2.78	else
163			fflush(stdout);
164	greg	2.12	}
165	greg	1.1	/* process file */
166			while (getvec(orig, inform, fp) == 0 &&
167			getvec(direc, inform, fp) == 0) {
168
169	gregl	2.25	d = normalize(direc);
170			if (d == 0.0) { /* zero ==> flush */
171	greg	2.79	if ((--nextflush <= 0) \| !vcount && something2flush) {
172	greg	2.74	if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
173	greg	2.71	error(USER, "child(ren) died");
174	greg	2.63	bogusray();
175	gregl	2.24	fflush(stdout);
176	greg	2.79	something2flush = 0;
177	greg	2.64	nextflush = (vresolu > 0) & (hresolu > 1) ? 0 :
178			hresolu;
179	greg	2.63	} else
180			bogusray();
181	greg	2.57	} else { /* compute and print */
182	greg	2.59	rtcompute(orig, direc, lim_dist ? d : 0.0);
183	greg	1.7	/* flush if time */
184	greg	2.45	if (!--nextflush) {
185	greg	2.74	if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
186	greg	2.71	error(USER, "child(ren) died");
187	gregl	2.24	fflush(stdout);
188			nextflush = hresolu;
189	greg	2.78	} else
190	greg	2.79	something2flush = 1;
191	greg	1.1	}
192			if (ferror(stdout))
193			error(SYSTEM, "write error");
194	greg	2.45	if (vcount && !--vcount) /* check for end */
195	greg	1.1	break;
196			}
197	greg	2.74	if (ray_pnprocs > 1) { /* clean up children */
198	greg	2.59	if (ray_fifo_flush() < 0)
199			error(USER, "unable to complete processing");
200			ray_pclose(0);
201			}
202	greg	2.50	if (fflush(stdout) < 0)
203			error(SYSTEM, "write error");
204	greg	2.45	if (vcount)
205	greg	2.42	error(USER, "unexpected EOF on input");
206	greg	2.12	if (fname != NULL)
207			fclose(fp);
208	greg	1.1	}
209
210
211	schorsch	2.37	static void
212	greg	2.40	trace_sources(void) /* trace rays to light sources, also */
213			{
214			int sn;
215
216			for (sn = 0; sn < nsources; sn++)
217			source[sn].sflags \|= SFOLLOW;
218			}
219
220
221			static void
222	schorsch	2.37	setoutput( /* set up output tables */
223	greg	2.56	char *vs
224	schorsch	2.37	)
225	greg	1.1	{
226	greg	2.56	oputf_t **table = ray_out;
227	greg	1.1
228	greg	1.11	castonly = 1;
229	greg	1.1	while (*vs)
230			switch (*vs++) {
231	greg	2.40	case 'T': /* trace sources */
232	greg	2.42	if (!*vs) break;
233	greg	2.40	trace_sources();
234			/* fall through */
235	greg	1.1	case 't': /* trace */
236	greg	2.42	if (!*vs) break;
237	greg	1.1	*table = NULL;
238			table = every_out;
239			trace = ourtrace;
240	greg	1.11	castonly = 0;
241	greg	1.1	break;
242			case 'o': /* origin */
243			*table++ = oputo;
244			break;
245			case 'd': /* direction */
246			*table++ = oputd;
247			break;
248	greg	2.73	case 'r': /* reflected contrib. */
249			*table++ = oputr;
250			castonly = 0;
251			break;
252			case 'R': /* reflected distance */
253			*table++ = oputR;
254			castonly = 0;
255			break;
256			case 'x': /* xmit contrib. */
257			*table++ = oputx;
258			castonly = 0;
259			break;
260			case 'X': /* xmit distance */
261			*table++ = oputX;
262			castonly = 0;
263			break;
264	greg	1.1	case 'v': /* value */
265			*table++ = oputv;
266	greg	1.11	castonly = 0;
267	greg	1.1	break;
268	greg	2.51	case 'V': /* contribution */
269			*table++ = oputV;
270	greg	2.73	castonly = 0;
271	greg	2.51	if (ambounce > 0 && (ambacc > FTINY \|\| ambssamp > 0))
272			error(WARNING,
273			"-otV accuracy depends on -aa 0 -as 0");
274			break;
275	greg	2.5	case 'l': /* effective distance */
276	greg	1.1	*table++ = oputl;
277	greg	1.11	castonly = 0;
278	greg	1.1	break;
279	greg	2.29	case 'c': /* local coordinates */
280			*table++ = oputc;
281			break;
282	greg	2.5	case 'L': /* single ray length */
283			*table++ = oputL;
284			break;
285	greg	1.1	case 'p': /* point */
286			*table++ = oputp;
287			break;
288	greg	2.9	case 'n': /* perturbed normal */
289	greg	1.1	*table++ = oputn;
290	greg	2.9	castonly = 0;
291	greg	1.1	break;
292	greg	2.9	case 'N': /* unperturbed normal */
293			*table++ = oputN;
294			break;
295	greg	1.1	case 's': /* surface */
296			*table++ = oputs;
297			break;
298			case 'w': /* weight */
299			*table++ = oputw;
300			break;
301	greg	2.40	case 'W': /* coefficient */
302			*table++ = oputW;
303	greg	2.73	castonly = 0;
304	greg	2.80	if (ambounce > 0 && (ambacc > FTINY) \| (ambssamp > 0))
305	greg	2.40	error(WARNING,
306	greg	2.41	"-otW accuracy depends on -aa 0 -as 0");
307	greg	2.40	break;
308	greg	1.1	case 'm': /* modifier */
309			*table++ = oputm;
310			break;
311	greg	2.39	case 'M': /* material */
312			*table++ = oputM;
313			break;
314	greg	2.42	case '~': /* tilde */
315			*table++ = oputtilde;
316	greg	2.41	break;
317	greg	1.1	}
318			*table = NULL;
319	greg	2.80	if (do_irrad \| imm_irrad) /* compatibile? */
320			for (table = ray_out; *table != NULL; table++)
321			if ((table == oputr) \| (table == oputR) \|
322			(table == oputx) \| (table == oputX)) {
323			error(WARNING, "-orRxX options incompatible with -I+ and -i+");
324			break;
325			}
326	gregl	2.24	}
327
328
329	schorsch	2.37	static void
330			bogusray(void) /* print out empty record */
331	gregl	2.24	{
332	greg	2.40	rayorigin(&thisray, PRIMARY, NULL, NULL);
333	gregl	2.24	printvals(&thisray);
334	greg	1.1	}
335
336
337	schorsch	2.37	static void
338	greg	2.59	raycast( /* compute first ray intersection only */
339			RAY *r
340	schorsch	2.37	)
341	greg	1.1	{
342	greg	2.59	if (!localhit(r, &thescene)) {
343			if (r->ro == &Aftplane) { /* clipped */
344			r->ro = NULL;
345			r->rot = FHUGE;
346			} else
347			sourcehit(r);
348	greg	2.57	}
349	greg	1.1	}
350
351
352	schorsch	2.37	static void
353	greg	2.59	rayirrad( /* compute irradiance rather than radiance */
354			RAY *r
355			)
356			{
357	greg	2.62	void (old_revf)(RAY ) = r->revf;
358	greg	2.75	/* pretend we hit surface */
359	greg	2.76	r->rxt = r->rot = 1e-5;
360	greg	2.59	VSUM(r->rop, r->rorg, r->rdir, r->rot);
361			r->ron[0] = -r->rdir[0];
362			r->ron[1] = -r->rdir[1];
363			r->ron[2] = -r->rdir[2];
364			r->rod = 1.0;
365	greg	2.62	/* compute result */
366	greg	2.61	r->revf = raytrace;
367	greg	2.59	(*ofun[Lamb.otype].funp)(&Lamb, r);
368	greg	2.62	r->revf = old_revf;
369	greg	2.59	}
370
371
372			static void
373			rtcompute( /* compute and print ray value(s) */
374	schorsch	2.37	FVECT org,
375	greg	2.59	FVECT dir,
376			double dmax
377	schorsch	2.37	)
378	greg	1.1	{
379	greg	2.60	/* set up ray */
380			rayorigin(&thisray, PRIMARY, NULL, NULL);
381			if (imm_irrad) {
382	greg	2.59	VSUM(thisray.rorg, org, dir, 1.1e-4);
383			thisray.rdir[0] = -dir[0];
384			thisray.rdir[1] = -dir[1];
385			thisray.rdir[2] = -dir[2];
386			thisray.rmax = 0.0;
387	greg	2.60	thisray.revf = rayirrad;
388	greg	2.59	} else {
389			VCOPY(thisray.rorg, org);
390			VCOPY(thisray.rdir, dir);
391			thisray.rmax = dmax;
392	greg	2.60	if (castonly)
393			thisray.revf = raycast;
394	greg	2.59	}
395			if (ray_pnprocs > 1) { /* multiprocessing FIFO? */
396			if (ray_fifo_in(&thisray) < 0)
397			error(USER, "lost children");
398			return;
399			}
400			samplendx++; /* else do it ourselves */
401			rayvalue(&thisray);
402	greg	2.16	printvals(&thisray);
403			}
404
405
406	greg	2.57	static int
407	schorsch	2.37	printvals( /* print requested ray values */
408			RAY *r
409			)
410	greg	2.16	{
411	greg	2.56	oputf_t **tp;
412	greg	2.16
413			if (ray_out[0] == NULL)
414	greg	2.57	return(0);
415	greg	2.16	for (tp = ray_out; *tp != NULL; tp++)
416			(**tp)(r);
417	greg	1.1	if (outform == 'a')
418			putchar('\n');
419	greg	2.57	return(1);
420	greg	1.1	}
421
422
423	schorsch	2.37	static int
424			getvec( /* get a vector from fp */
425	greg	2.56	FVECT vec,
426	schorsch	2.37	int fmt,
427			FILE *fp
428			)
429	greg	1.1	{
430			static float vf[3];
431	greg	2.5	static double vd[3];
432	greg	1.19	char buf[32];
433	greg	2.56	int i;
434	greg	1.1
435			switch (fmt) {
436			case 'a': /* ascii */
437	greg	1.19	for (i = 0; i < 3; i++) {
438			if (fgetword(buf, sizeof(buf), fp) == NULL \|\|
439			!isflt(buf))
440			return(-1);
441			vec[i] = atof(buf);
442			}
443	greg	1.1	break;
444			case 'f': /* binary float */
445	greg	2.70	if (getbinary(vf, sizeof(float), 3, fp) != 3)
446	greg	1.1	return(-1);
447	greg	2.58	VCOPY(vec, vf);
448	greg	1.1	break;
449			case 'd': /* binary double */
450	greg	2.70	if (getbinary(vd, sizeof(double), 3, fp) != 3)
451	greg	1.1	return(-1);
452	greg	2.58	VCOPY(vec, vd);
453	greg	1.1	break;
454	greg	2.6	default:
455			error(CONSISTENCY, "botched input format");
456	greg	1.1	}
457			return(0);
458			}
459
460
461	greg	2.57	void
462	schorsch	2.37	tranotify( /* record new modifier */
463			OBJECT obj
464			)
465	greg	2.15	{
466			static int hitlimit = 0;
467	greg	2.56	OBJREC *o = objptr(obj);
468			char **tralp;
469	greg	2.15
470	greg	2.27	if (obj == OVOID) { /* starting over */
471			traset[0] = 0;
472			hitlimit = 0;
473			return;
474			}
475	greg	2.15	if (hitlimit \|\| !ismodifier(o->otype))
476			return;
477			for (tralp = tralist; *tralp != NULL; tralp++)
478			if (!strcmp(o->oname, *tralp)) {
479			if (traset[0] >= MAXTSET) {
480			error(WARNING, "too many modifiers in trace list");
481			hitlimit++;
482			return; /* should this be fatal? */
483			}
484			insertelem(traset, obj);
485			return;
486			}
487			}
488
489
490	greg	2.27	static void
491	schorsch	2.37	ourtrace( /* print ray values */
492			RAY *r
493			)
494	greg	1.1	{
495	greg	2.56	oputf_t **tp;
496	greg	1.1
497			if (every_out[0] == NULL)
498	greg	2.15	return;
499	greg	2.16	if (r->ro == NULL) {
500			if (traincl == 1)
501			return;
502			} else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
503	greg	1.1	return;
504			tabin(r);
505			for (tp = every_out; *tp != NULL; tp++)
506			(**tp)(r);
507	greg	2.41	if (outform == 'a')
508			putchar('\n');
509	greg	1.1	}
510
511
512	greg	2.27	static void
513	schorsch	2.37	tabin( /* tab in appropriate amount */
514			RAY *r
515			)
516	greg	1.1	{
517	greg	2.40	const RAY *rp;
518	greg	1.1
519			for (rp = r->parent; rp != NULL; rp = rp->parent)
520			putchar('\t');
521			}
522
523
524	greg	2.27	static void
525	schorsch	2.37	oputo( /* print origin */
526			RAY *r
527			)
528	greg	1.1	{
529	greg	2.65	(*putreal)(r->rorg, 3);
530	greg	1.1	}
531
532
533	greg	2.27	static void
534	schorsch	2.37	oputd( /* print direction */
535			RAY *r
536			)
537	greg	1.1	{
538	greg	2.65	(*putreal)(r->rdir, 3);
539	greg	1.1	}
540
541
542	greg	2.27	static void
543	greg	2.73	oputr( /* print mirrored contribution */
544			RAY *r
545			)
546			{
547			RREAL cval[3];
548
549			cval[0] = colval(r->mcol,RED);
550			cval[1] = colval(r->mcol,GRN);
551			cval[2] = colval(r->mcol,BLU);
552			(*putreal)(cval, 3);
553			}
554
555
556
557			static void
558			oputR( /* print mirrored distance */
559			RAY *r
560			)
561			{
562			(*putreal)(&r->rmt, 1);
563			}
564
565
566			static void
567			oputx( /* print unmirrored contribution */
568			RAY *r
569			)
570			{
571			RREAL cval[3];
572
573			cval[0] = colval(r->rcol,RED) - colval(r->mcol,RED);
574			cval[1] = colval(r->rcol,GRN) - colval(r->mcol,GRN);
575			cval[2] = colval(r->rcol,BLU) - colval(r->mcol,BLU);
576			(*putreal)(cval, 3);
577			}
578
579
580			static void
581			oputX( /* print unmirrored distance */
582			RAY *r
583			)
584			{
585			(*putreal)(&r->rxt, 1);
586			}
587
588
589			static void
590	schorsch	2.37	oputv( /* print value */
591			RAY *r
592			)
593	greg	1.1	{
594	greg	2.65	RREAL cval[3];
595
596			cval[0] = colval(r->rcol,RED);
597			cval[1] = colval(r->rcol,GRN);
598			cval[2] = colval(r->rcol,BLU);
599			(*putreal)(cval, 3);
600	greg	1.1	}
601
602
603	greg	2.27	static void
604	greg	2.51	oputV( /* print value contribution */
605			RAY *r
606			)
607			{
608	greg	2.65	RREAL contr[3];
609	greg	2.51
610			raycontrib(contr, r, PRIMARY);
611			multcolor(contr, r->rcol);
612	greg	2.65	(*putreal)(contr, 3);
613	greg	2.51	}
614
615
616			static void
617	schorsch	2.37	oputl( /* print effective distance */
618			RAY *r
619			)
620	greg	1.1	{
621	greg	2.73	RREAL d = raydistance(r);
622
623			(*putreal)(&d, 1);
624	greg	2.5	}
625
626
627	greg	2.27	static void
628	schorsch	2.37	oputL( /* print single ray length */
629			RAY *r
630			)
631	greg	2.5	{
632	greg	2.65	(*putreal)(&r->rot, 1);
633	greg	1.1	}
634
635
636	greg	2.27	static void
637	schorsch	2.37	oputc( /* print local coordinates */
638			RAY *r
639			)
640	greg	2.29	{
641	greg	2.65	(*putreal)(r->uv, 2);
642	greg	2.29	}
643
644
645	greg	2.65	static RREAL vdummy[3] = {0.0, 0.0, 0.0};
646
647
648	greg	2.29	static void
649	schorsch	2.37	oputp( /* print point */
650			RAY *r
651			)
652	greg	1.1	{
653	greg	2.65	if (r->rot < FHUGE)
654			(*putreal)(r->rop, 3);
655			else
656			(*putreal)(vdummy, 3);
657	greg	1.1	}
658
659
660	greg	2.27	static void
661	schorsch	2.37	oputN( /* print unperturbed normal */
662			RAY *r
663			)
664	greg	1.1	{
665	greg	2.65	if (r->rot < FHUGE)
666			(*putreal)(r->ron, 3);
667			else
668			(*putreal)(vdummy, 3);
669	greg	2.9	}
670
671
672	greg	2.27	static void
673	schorsch	2.37	oputn( /* print perturbed normal */
674			RAY *r
675			)
676	greg	2.9	{
677			FVECT pnorm;
678
679			if (r->rot >= FHUGE) {
680	greg	2.65	(*putreal)(vdummy, 3);
681	greg	2.9	return;
682			}
683			raynormal(pnorm, r);
684	greg	2.65	(*putreal)(pnorm, 3);
685	greg	1.1	}
686
687
688	greg	2.27	static void
689	schorsch	2.37	oputs( /* print name */
690			RAY *r
691			)
692	greg	1.1	{
693			if (r->ro != NULL)
694			fputs(r->ro->oname, stdout);
695			else
696			putchar('*');
697			putchar('\t');
698			}
699
700
701	greg	2.27	static void
702	schorsch	2.37	oputw( /* print weight */
703			RAY *r
704			)
705	greg	1.1	{
706	greg	2.65	RREAL rwt = r->rweight;
707
708			(*putreal)(&rwt, 1);
709	greg	1.1	}
710
711
712	greg	2.27	static void
713	greg	2.51	oputW( /* print coefficient */
714	greg	2.40	RAY *r
715			)
716			{
717	greg	2.65	RREAL contr[3];
718	greg	2.67	/* shadow ray not on source? */
719			if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
720			setcolor(contr, 0.0, 0.0, 0.0);
721			else
722			raycontrib(contr, r, PRIMARY);
723	greg	2.40
724	greg	2.65	(*putreal)(contr, 3);
725	greg	2.40	}
726
727
728			static void
729	schorsch	2.37	oputm( /* print modifier */
730			RAY *r
731			)
732	greg	1.1	{
733			if (r->ro != NULL)
734	greg	2.23	if (r->ro->omod != OVOID)
735			fputs(objptr(r->ro->omod)->oname, stdout);
736			else
737			fputs(VOIDID, stdout);
738	greg	1.1	else
739			putchar('*');
740			putchar('\t');
741			}
742
743
744	greg	2.27	static void
745	greg	2.39	oputM( /* print material */
746			RAY *r
747			)
748			{
749			OBJREC *mat;
750
751			if (r->ro != NULL) {
752			if ((mat = findmaterial(r->ro)) != NULL)
753			fputs(mat->oname, stdout);
754			else
755			fputs(VOIDID, stdout);
756			} else
757			putchar('*');
758			putchar('\t');
759			}
760
761
762			static void
763	greg	2.42	oputtilde( /* output tilde (spacer) */
764	greg	2.41	RAY *r
765			)
766			{
767	greg	2.42	fputs("~\t", stdout);
768	greg	2.41	}
769
770
771			static void
772	greg	2.65	puta( /* print ascii value(s) */
773			RREAL *v, int n
774	schorsch	2.37	)
775	greg	1.1	{
776	greg	2.65	if (n == 3) {
777			printf("%e\t%e\t%e\t", v[0], v[1], v[2]);
778			return;
779			}
780			while (n--)
781			printf("%e\t", *v++);
782	greg	1.1	}
783
784
785	greg	2.27	static void
786	greg	2.80	putd(RREAL v, int n) / output binary double(s) */
787	greg	1.1	{
788	greg	2.68	#ifdef SMLFLT
789			double da[3];
790			int i;
791
792			if (n > 3)
793	greg	2.65	error(INTERNAL, "code error in putd()");
794	greg	2.68	for (i = n; i--; )
795			da[i] = v[i];
796	greg	2.69	putbinary(da, sizeof(double), n, stdout);
797	greg	2.68	#else
798	greg	2.69	putbinary(v, sizeof(RREAL), n, stdout);
799	greg	2.68	#endif
800	greg	1.1	}
801
802
803	greg	2.27	static void
804	greg	2.80	putf(RREAL v, int n) / output binary float(s) */
805	greg	1.1	{
806	greg	2.68	#ifndef SMLFLT
807	greg	2.65	float fa[3];
808			int i;
809	greg	1.1
810	greg	2.65	if (n > 3)
811			error(INTERNAL, "code error in putf()");
812			for (i = n; i--; )
813			fa[i] = v[i];
814	greg	2.69	putbinary(fa, sizeof(float), n, stdout);
815	greg	2.68	#else
816	greg	2.69	putbinary(v, sizeof(RREAL), n, stdout);
817	greg	2.68	#endif
818	greg	1.1	}
819	greg	2.80
820
821			static void
822			putrgbe(RREAL v, int n) / output RGBE color */
823			{
824			COLR cout;
825
826			if (n != 3)
827			error(INTERNAL, "putrgbe() not called with 3 components");
828			setcolr(cout, v[0], v[1], v[2]);
829			putbinary(cout, sizeof(cout), 1, stdout);
830			}