src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.50 2005/10/07 03:45:14 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 = 0;                     /* pure Monte Carlo sampling? */

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

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