src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.58 2009/12/12 23:08:13 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  "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(double v);
static putf_t puta, putd, putf;

typedef void oputf_t(RAY *r);
static oputf_t  oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
                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[16], *every_out[16];
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
                                              : vresolu;
        long  nextflush = hresolu;
        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 */
        switch (outform) {
        case 'a': putreal = puta; break;
        case 'f': putreal = putf; break;
        case 'd': putreal = putd; break;
        case 'c': 
                if (strcmp(outvals, "v"))
                        error(USER, "color format with value output only");
                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);
                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 (nproc > 1 && ray_fifo_flush() < 0)
                                error(USER, "lost children");
                        bogusray();
                        if (--nextflush <= 0 || !vcount) {
                                fflush(stdout);
                                nextflush = hresolu;
                        }
                } else {                                /* compute and print */
                        rtcompute(orig, direc, lim_dist ? d : 0.0);
                                                        /* flush if time */
                        if (!--nextflush) {
                                if (nproc > 1 && ray_fifo_flush() < 0)
                                        error(USER, "lost children");
                                fflush(stdout);
                                nextflush = hresolu;
                        }
                }
                if (ferror(stdout))
                        error(SYSTEM, "write error");
                if (vcount && !--vcount)                /* check for end */
                        break;
        }
        if (nproc > 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 'v':                               /* value */
                        *table++ = oputv;
                        castonly = 0;
                        break;
                case 'V':                               /* contribution */
                        *table++ = oputV;
                        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;
                        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;
}


static void
bogusray(void)                  /* print out empty record */
{
        thisray.rorg[0] = thisray.rorg[1] = thisray.rorg[2] =
        thisray.rdir[0] = thisray.rdir[1] = thisray.rdir[2] = 0.0;
        thisray.rmax = 0.0;
        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
)
{
        r->rot = 1e-5;                  /* pretend we hit surface */
        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 */
        (*ofun[Lamb.otype].funp)(&Lamb, r);
}


static void
rtcompute(                      /* compute and print ray value(s) */
        FVECT  org,
        FVECT  dir,
        double  dmax
)
{
        if (imm_irrad) {                /* set up ray */
                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;
        } else {
                VCOPY(thisray.rorg, org);
                VCOPY(thisray.rdir, dir);
                thisray.rmax = dmax;
        }
        rayorigin(&thisray, PRIMARY, NULL, NULL);
                                        /* special case evaluators */
        if (castonly)
                thisray.revf = raycast;
        else if (imm_irrad)
                thisray.revf = rayirrad;

        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 (fread((char *)vf, sizeof(float), 3, fp) != 3)
                        return(-1);
                VCOPY(vec, vf);
                break;
        case 'd':                                       /* binary double */
                if (fread((char *)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[0]);
        (*putreal)(r->rorg[1]);
        (*putreal)(r->rorg[2]);
}


static void
oputd(                          /* print direction */
        RAY  *r
)
{
        (*putreal)(r->rdir[0]);
        (*putreal)(r->rdir[1]);
        (*putreal)(r->rdir[2]);
}


static void
oputv(                          /* print value */
        RAY  *r
)
{
        if (outform == 'c') {
                COLR  cout;
                setcolr(cout,   colval(r->rcol,RED),
                                colval(r->rcol,GRN),
                                colval(r->rcol,BLU));
                fwrite((char *)cout, sizeof(cout), 1, stdout);
                return;
        }
        (*putreal)(colval(r->rcol,RED));
        (*putreal)(colval(r->rcol,GRN));
        (*putreal)(colval(r->rcol,BLU));
}


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

        raycontrib(contr, r, PRIMARY);
        multcolor(contr, r->rcol);
        (*putreal)(contr[RED]);
        (*putreal)(contr[GRN]);
        (*putreal)(contr[BLU]);
}


static void
oputl(                          /* print effective distance */
        RAY  *r
)
{
        (*putreal)(r->rt);
}


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


static void
oputc(                          /* print local coordinates */
        RAY  *r
)
{
        (*putreal)(r->uv[0]);
        (*putreal)(r->uv[1]);
}


static void
oputp(                          /* print point */
        RAY  *r
)
{
        if (r->rot < FHUGE) {
                (*putreal)(r->rop[0]);
                (*putreal)(r->rop[1]);
                (*putreal)(r->rop[2]);
        } else {
                (*putreal)(0.0);
                (*putreal)(0.0);
                (*putreal)(0.0);
        }
}


static void
oputN(                          /* print unperturbed normal */
        RAY  *r
)
{
        if (r->rot < FHUGE) {
                (*putreal)(r->ron[0]);
                (*putreal)(r->ron[1]);
                (*putreal)(r->ron[2]);
        } else {
                (*putreal)(0.0);
                (*putreal)(0.0);
                (*putreal)(0.0);
        }
}


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

        if (r->rot >= FHUGE) {
                (*putreal)(0.0);
                (*putreal)(0.0);
                (*putreal)(0.0);
                return;
        }
        raynormal(pnorm, r);
        (*putreal)(pnorm[0]);
        (*putreal)(pnorm[1]);
        (*putreal)(pnorm[2]);
}


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
)
{
        (*putreal)(r->rweight);
}


static void
oputW(                          /* print coefficient */
        RAY  *r
)
{
        double  contr[3];

        raycontrib(contr, r, PRIMARY);
        (*putreal)(contr[RED]);
        (*putreal)(contr[GRN]);
        (*putreal)(contr[BLU]);
}


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 */
        double  v
)
{
        printf("%e\t", v);
}


static void
putd(v)                         /* print binary double */
double  v;
{
        fwrite((char *)&v, sizeof(v), 1, stdout);
}


static void
putf(v)                         /* print binary float */
double  v;
{
        float f = v;

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