src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.54 2006/04/05 14:30:38 greg Exp $";
#endif
/*
 *  rtrace.c - program and variables for individual ray tracing.
 */

#include "copyright.h"

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

#include  <time.h>

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

CUBE  thescene;                         /* our scene */
OBJECT  nsceneobjs;                     /* number of objects in our scene */

int  dimlist[MAXDIM];                   /* sampling dimensions */
int  ndims = 0;                         /* number of sampling dimensions */
int  samplendx = 0;                     /* index for this sample */

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

int  inform = 'a';                      /* input format */
int  outform = 'a';                     /* output format */
char  *outvals = "v";                   /* output specification */

int  do_irrad = 0;                      /* compute irradiance? */

int  rand_samp = 1;                     /* pure Monte Carlo sampling? */

void  (*trace)() = NULL;                /* trace call */

#ifndef MAXMODLIST
#define MAXMODLIST      1024            /* maximum modifiers we'll track */
#endif

char  *tralist[MAXMODLIST];             /* list of modifers to trace (or no) */
int  traincl = -1;                      /* include == 1, exclude == 0 */
#ifndef  MAXTSET
#define  MAXTSET        8191            /* maximum number in trace set */
#endif
OBJECT  traset[MAXTSET+1]={0};          /* trace include/exclude set */

int  hresolu = 0;                       /* horizontal (scan) size */
int  vresolu = 0;                       /* vertical resolution */

double  dstrsrc = 0.0;                  /* square source distribution */
double  shadthresh = .03;               /* shadow threshold */
double  shadcert = .75;                 /* shadow certainty */
int  directrelay = 2;                   /* number of source relays */
int  vspretest = 512;                   /* virtual source pretest density */
int  directvis = 1;                     /* sources visible? */
double  srcsizerat = .2;                /* maximum ratio source size/dist. */

COLOR  cextinction = BLKCOLOR;          /* global extinction coefficient */
COLOR  salbedo = BLKCOLOR;              /* global scattering albedo */
double  seccg = 0.;                     /* global scattering eccentricity */
double  ssampdist = 0.;                 /* scatter sampling distance */

double  specthresh = .15;               /* specular sampling threshold */
double  specjitter = 1.;                /* specular sampling jitter */

int  backvis = 1;                       /* back face visibility */

int  maxdepth = -10;                    /* maximum recursion depth */
double  minweight = 2e-3;               /* minimum ray weight */

char  *ambfile = NULL;                  /* ambient file name */
COLOR  ambval = BLKCOLOR;               /* ambient value */
int  ambvwt = 0;                        /* initial weight for ambient value */
double  ambacc = 0.15;                  /* ambient accuracy */
int  ambres = 256;                      /* ambient resolution */
int  ambdiv = 1024;                     /* ambient divisions */
int  ambssamp = 512;                    /* ambient super-samples */
int  ambounce = 0;                      /* ambient bounces */
char  *amblist[AMBLLEN];                /* ambient include/exclude list */
int  ambincl = -1;                      /* include == 1, exclude == 0 */

static int  castonly = 0;

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);
static void tranotify(OBJECT obj);
static void bogusray(void);
static void rad(FVECT  org, FVECT  dir, double  dmax);
static void irrad(FVECT  org, FVECT  dir);
static void 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  (*addobjnotify[])() = {ambnotify, tranotify, NULL};


void
quit(                   /* quit program */
        int  code
)
{
#ifndef  NON_POSIX /* XXX we don't clean up elsewhere? */
        headclean();            /* delete header file */
        pfclean();              /* clean up persist files */
#endif
        exit(code);
}


extern 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
)
{
        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);
        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 (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 */
                        bogusray();
                        if (--nextflush <= 0 || !vcount) {
                                fflush(stdout);
                                nextflush = hresolu;
                        }
                } else {
                        samplendx++;
                                                        /* compute and print */
                        if (imm_irrad)
                                irrad(orig, direc);
                        else
                                rad(orig, direc, lim_dist ? d : 0.0);
                                                        /* flush if time */
                        if (!--nextflush) {
                                fflush(stdout);
                                nextflush = hresolu;
                        }
                }
                if (ferror(stdout))
                        error(SYSTEM, "write error");
                if (vcount && !--vcount)                /* check for end */
                        break;
        }
        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 */
        register char  *vs
)
{
        register 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
rad(            /* compute and print ray value(s) */
        FVECT  org,
        FVECT  dir,
        double  dmax
)
{
        VCOPY(thisray.rorg, org);
        VCOPY(thisray.rdir, dir);
        thisray.rmax = dmax;
        rayorigin(&thisray, PRIMARY, NULL, NULL);
        if (castonly) {
                if (!localhit(&thisray, &thescene)) {
                        if (thisray.ro == &Aftplane) {  /* clipped */
                                thisray.ro = NULL;
                                thisray.rot = FHUGE;
                        } else
                                sourcehit(&thisray);
                }
        } else
                rayvalue(&thisray);
        printvals(&thisray);
}


static void
irrad(                  /* compute immediate irradiance value */
        FVECT  org,
        FVECT  dir
)
{
        register int  i;

        for (i = 0; i < 3; i++) {
                thisray.rorg[i] = org[i] + dir[i];
                thisray.rdir[i] = -dir[i];
        }
        thisray.rmax = 0.0;
        rayorigin(&thisray, PRIMARY, NULL, NULL);
                                        /* pretend we hit surface */
        thisray.rot = 1.0-1e-4;
        thisray.rod = 1.0;
        VCOPY(thisray.ron, dir);
        for (i = 0; i < 3; i++)         /* fudge factor */
                thisray.rop[i] = org[i] + 1e-4*dir[i];
                                        /* compute and print */
        (*ofun[Lamb.otype].funp)(&Lamb, &thisray);
        printvals(&thisray);
}


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

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


static int
getvec(         /* get a vector from fp */
        register FVECT  vec,
        int  fmt,
        FILE  *fp
)
{
        static float  vf[3];
        static double  vd[3];
        char  buf[32];
        register 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);
                vec[0] = vf[0]; vec[1] = vf[1]; vec[2] = vf[2];
                break;
        case 'd':                                       /* binary double */
                if (fread((char *)vd, sizeof(double), 3, fp) != 3)
                        return(-1);
                vec[0] = vd[0]; vec[1] = vd[1]; vec[2] = vd[2];
                break;
        default:
                error(CONSISTENCY, "botched input format");
        }
        return(0);
}


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