src/rt/rtrace.c

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