src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.59 2009/12/13 19:13:04 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
)
{
                                        /* 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 (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.60
Committed:	Mon Dec 14 03:15:45 2009 UTC (14 years, 4 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.59:	+7 -9 lines
Log Message:	Minor cosmetic change
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.60	static const char RCSid[] = "$Id: rtrace.c,v 2.59 2009/12/13 19:13:04 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.60	/* set up ray */
342			rayorigin(&thisray, PRIMARY, NULL, NULL);
343			if (imm_irrad) {
344	greg	2.59	VSUM(thisray.rorg, org, dir, 1.1e-4);
345			thisray.rdir[0] = -dir[0];
346			thisray.rdir[1] = -dir[1];
347			thisray.rdir[2] = -dir[2];
348			thisray.rmax = 0.0;
349	greg	2.60	thisray.revf = rayirrad;
350	greg	2.59	} else {
351			VCOPY(thisray.rorg, org);
352			VCOPY(thisray.rdir, dir);
353			thisray.rmax = dmax;
354	greg	2.60	if (castonly)
355			thisray.revf = raycast;
356	greg	2.59	}
357			if (ray_pnprocs > 1) { /* multiprocessing FIFO? */
358			if (ray_fifo_in(&thisray) < 0)
359			error(USER, "lost children");
360			return;
361			}
362			samplendx++; /* else do it ourselves */
363			rayvalue(&thisray);
364	greg	2.16	printvals(&thisray);
365			}
366
367
368	greg	2.57	static int
369	schorsch	2.37	printvals( /* print requested ray values */
370			RAY *r
371			)
372	greg	2.16	{
373	greg	2.56	oputf_t **tp;
374	greg	2.16
375			if (ray_out[0] == NULL)
376	greg	2.57	return(0);
377	greg	2.16	for (tp = ray_out; *tp != NULL; tp++)
378			(**tp)(r);
379	greg	1.1	if (outform == 'a')
380			putchar('\n');
381	greg	2.57	return(1);
382	greg	1.1	}
383
384
385	schorsch	2.37	static int
386			getvec( /* get a vector from fp */
387	greg	2.56	FVECT vec,
388	schorsch	2.37	int fmt,
389			FILE *fp
390			)
391	greg	1.1	{
392			static float vf[3];
393	greg	2.5	static double vd[3];
394	greg	1.19	char buf[32];
395	greg	2.56	int i;
396	greg	1.1
397			switch (fmt) {
398			case 'a': /* ascii */
399	greg	1.19	for (i = 0; i < 3; i++) {
400			if (fgetword(buf, sizeof(buf), fp) == NULL \|\|
401			!isflt(buf))
402			return(-1);
403			vec[i] = atof(buf);
404			}
405	greg	1.1	break;
406			case 'f': /* binary float */
407	greg	1.8	if (fread((char *)vf, sizeof(float), 3, fp) != 3)
408	greg	1.1	return(-1);
409	greg	2.58	VCOPY(vec, vf);
410	greg	1.1	break;
411			case 'd': /* binary double */
412	greg	2.5	if (fread((char *)vd, sizeof(double), 3, fp) != 3)
413	greg	1.1	return(-1);
414	greg	2.58	VCOPY(vec, vd);
415	greg	1.1	break;
416	greg	2.6	default:
417			error(CONSISTENCY, "botched input format");
418	greg	1.1	}
419			return(0);
420			}
421
422
423	greg	2.57	void
424	schorsch	2.37	tranotify( /* record new modifier */
425			OBJECT obj
426			)
427	greg	2.15	{
428			static int hitlimit = 0;
429	greg	2.56	OBJREC *o = objptr(obj);
430			char **tralp;
431	greg	2.15
432	greg	2.27	if (obj == OVOID) { /* starting over */
433			traset[0] = 0;
434			hitlimit = 0;
435			return;
436			}
437	greg	2.15	if (hitlimit \|\| !ismodifier(o->otype))
438			return;
439			for (tralp = tralist; *tralp != NULL; tralp++)
440			if (!strcmp(o->oname, *tralp)) {
441			if (traset[0] >= MAXTSET) {
442			error(WARNING, "too many modifiers in trace list");
443			hitlimit++;
444			return; /* should this be fatal? */
445			}
446			insertelem(traset, obj);
447			return;
448			}
449			}
450
451
452	greg	2.27	static void
453	schorsch	2.37	ourtrace( /* print ray values */
454			RAY *r
455			)
456	greg	1.1	{
457	greg	2.56	oputf_t **tp;
458	greg	1.1
459			if (every_out[0] == NULL)
460	greg	2.15	return;
461	greg	2.16	if (r->ro == NULL) {
462			if (traincl == 1)
463			return;
464			} else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
465	greg	1.1	return;
466			tabin(r);
467			for (tp = every_out; *tp != NULL; tp++)
468			(**tp)(r);
469	greg	2.41	if (outform == 'a')
470			putchar('\n');
471	greg	1.1	}
472
473
474	greg	2.27	static void
475	schorsch	2.37	tabin( /* tab in appropriate amount */
476			RAY *r
477			)
478	greg	1.1	{
479	greg	2.40	const RAY *rp;
480	greg	1.1
481			for (rp = r->parent; rp != NULL; rp = rp->parent)
482			putchar('\t');
483			}
484
485
486	greg	2.27	static void
487	schorsch	2.37	oputo( /* print origin */
488			RAY *r
489			)
490	greg	1.1	{
491			(*putreal)(r->rorg[0]);
492			(*putreal)(r->rorg[1]);
493			(*putreal)(r->rorg[2]);
494			}
495
496
497	greg	2.27	static void
498	schorsch	2.37	oputd( /* print direction */
499			RAY *r
500			)
501	greg	1.1	{
502			(*putreal)(r->rdir[0]);
503			(*putreal)(r->rdir[1]);
504			(*putreal)(r->rdir[2]);
505			}
506
507
508	greg	2.27	static void
509	schorsch	2.37	oputv( /* print value */
510			RAY *r
511			)
512	greg	1.1	{
513	greg	2.6	if (outform == 'c') {
514	greg	2.41	COLR cout;
515	greg	2.6	setcolr(cout, colval(r->rcol,RED),
516			colval(r->rcol,GRN),
517			colval(r->rcol,BLU));
518			fwrite((char *)cout, sizeof(cout), 1, stdout);
519			return;
520			}
521	greg	1.1	(*putreal)(colval(r->rcol,RED));
522			(*putreal)(colval(r->rcol,GRN));
523			(*putreal)(colval(r->rcol,BLU));
524			}
525
526
527	greg	2.27	static void
528	greg	2.51	oputV( /* print value contribution */
529			RAY *r
530			)
531			{
532			double contr[3];
533
534			raycontrib(contr, r, PRIMARY);
535			multcolor(contr, r->rcol);
536			(*putreal)(contr[RED]);
537			(*putreal)(contr[GRN]);
538			(*putreal)(contr[BLU]);
539			}
540
541
542			static void
543	schorsch	2.37	oputl( /* print effective distance */
544			RAY *r
545			)
546	greg	1.1	{
547	greg	1.9	(*putreal)(r->rt);
548	greg	2.5	}
549
550
551	greg	2.27	static void
552	schorsch	2.37	oputL( /* print single ray length */
553			RAY *r
554			)
555	greg	2.5	{
556			(*putreal)(r->rot);
557	greg	1.1	}
558
559
560	greg	2.27	static void
561	schorsch	2.37	oputc( /* print local coordinates */
562			RAY *r
563			)
564	greg	2.29	{
565			(*putreal)(r->uv[0]);
566			(*putreal)(r->uv[1]);
567			}
568
569
570			static void
571	schorsch	2.37	oputp( /* print point */
572			RAY *r
573			)
574	greg	1.1	{
575			if (r->rot < FHUGE) {
576			(*putreal)(r->rop[0]);
577			(*putreal)(r->rop[1]);
578			(*putreal)(r->rop[2]);
579			} else {
580			(*putreal)(0.0);
581			(*putreal)(0.0);
582			(*putreal)(0.0);
583			}
584			}
585
586
587	greg	2.27	static void
588	schorsch	2.37	oputN( /* print unperturbed normal */
589			RAY *r
590			)
591	greg	1.1	{
592			if (r->rot < FHUGE) {
593			(*putreal)(r->ron[0]);
594			(*putreal)(r->ron[1]);
595			(*putreal)(r->ron[2]);
596			} else {
597			(*putreal)(0.0);
598			(*putreal)(0.0);
599			(*putreal)(0.0);
600			}
601	greg	2.9	}
602
603
604	greg	2.27	static void
605	schorsch	2.37	oputn( /* print perturbed normal */
606			RAY *r
607			)
608	greg	2.9	{
609			FVECT pnorm;
610
611			if (r->rot >= FHUGE) {
612			(*putreal)(0.0);
613			(*putreal)(0.0);
614			(*putreal)(0.0);
615			return;
616			}
617			raynormal(pnorm, r);
618			(*putreal)(pnorm[0]);
619			(*putreal)(pnorm[1]);
620			(*putreal)(pnorm[2]);
621	greg	1.1	}
622
623
624	greg	2.27	static void
625	schorsch	2.37	oputs( /* print name */
626			RAY *r
627			)
628	greg	1.1	{
629			if (r->ro != NULL)
630			fputs(r->ro->oname, stdout);
631			else
632			putchar('*');
633			putchar('\t');
634			}
635
636
637	greg	2.27	static void
638	schorsch	2.37	oputw( /* print weight */
639			RAY *r
640			)
641	greg	1.1	{
642			(*putreal)(r->rweight);
643			}
644
645
646	greg	2.27	static void
647	greg	2.51	oputW( /* print coefficient */
648	greg	2.40	RAY *r
649			)
650			{
651	greg	2.48	double contr[3];
652	greg	2.40
653			raycontrib(contr, r, PRIMARY);
654	greg	2.48	(*putreal)(contr[RED]);
655			(*putreal)(contr[GRN]);
656			(*putreal)(contr[BLU]);
657	greg	2.40	}
658
659
660			static void
661	schorsch	2.37	oputm( /* print modifier */
662			RAY *r
663			)
664	greg	1.1	{
665			if (r->ro != NULL)
666	greg	2.23	if (r->ro->omod != OVOID)
667			fputs(objptr(r->ro->omod)->oname, stdout);
668			else
669			fputs(VOIDID, stdout);
670	greg	1.1	else
671			putchar('*');
672			putchar('\t');
673			}
674
675
676	greg	2.27	static void
677	greg	2.39	oputM( /* print material */
678			RAY *r
679			)
680			{
681			OBJREC *mat;
682
683			if (r->ro != NULL) {
684			if ((mat = findmaterial(r->ro)) != NULL)
685			fputs(mat->oname, stdout);
686			else
687			fputs(VOIDID, stdout);
688			} else
689			putchar('*');
690			putchar('\t');
691			}
692
693
694			static void
695	greg	2.42	oputtilde( /* output tilde (spacer) */
696	greg	2.41	RAY *r
697			)
698			{
699	greg	2.42	fputs("~\t", stdout);
700	greg	2.41	}
701
702
703			static void
704	schorsch	2.37	puta( /* print ascii value */
705			double v
706			)
707	greg	1.1	{
708			printf("%e\t", v);
709			}
710
711
712	greg	2.27	static void
713	greg	1.1	putd(v) /* print binary double */
714			double v;
715			{
716	greg	1.8	fwrite((char *)&v, sizeof(v), 1, stdout);
717	greg	1.1	}
718
719
720	greg	2.27	static void
721	greg	1.1	putf(v) /* print binary float */
722			double v;
723			{
724			float f = v;
725
726	greg	1.8	fwrite((char *)&f, sizeof(f), 1, stdout);
727	greg	1.1	}