src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.62 2009/12/14 07:31:37 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 (--nextflush <= 0 || !vcount) {
                                if (nproc > 1 && ray_fifo_flush() < 0)
                                        error(USER, "lost children");
                                bogusray();
                                fflush(stdout);
                                nextflush = hresolu;
                        } else
                                bogusray();
                } 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
)
{
        void    (*old_revf)(RAY *) = r->revf;

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