src/rt/rtrace.c

#ifndef lint
static const char       RCSid[] = "$Id: rtrace.c,v 2.81 2019/07/04 01:25:07 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  "otspecial.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(RREAL *v, int n);
static putf_t puta, putd, putf, putrgbe;

typedef void oputf_t(RAY *r);
static oputf_t  oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
                oputr, oputR, oputx, oputX, 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[32], *every_out[32];
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
                                              : (unsigned long)vresolu;
        long  nextflush = (vresolu > 0) & (hresolu > 1) ? 0 : hresolu;
        int  something2flush = 0;
        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 */
        if ((nextflush > 0) & (nproc > nextflush)) {
                error(WARNING, "reducing number of processes to match flush interval");
                nproc = nextflush;
        }
        switch (outform) {
        case 'a': putreal = puta; break;
        case 'f': putreal = putf; break;
        case 'd': putreal = putd; break;
        case 'c': 
                if (outvals[0] && (outvals[1] || !strchr("vrx", outvals[0])))
                        error(USER, "color format only with -ov, -or, -ox");
                putreal = putrgbe; 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);
                else
                        fflush(stdout);
        }
                                        /* process file */
        while (getvec(orig, inform, fp) == 0 &&
                        getvec(direc, inform, fp) == 0) {

                d = normalize(direc);
                if (d == 0.0) {                         /* zero ==> flush */
                        if ((--nextflush <= 0) | !vcount && something2flush) {
                                if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
                                        error(USER, "child(ren) died");
                                bogusray();
                                fflush(stdout);
                                something2flush = 0;
                                nextflush = (vresolu > 0) & (hresolu > 1) ? 0 :
                                                                hresolu;
                        } else
                                bogusray();
                } else {                                /* compute and print */
                        rtcompute(orig, direc, lim_dist ? d : 0.0);
                                                        /* flush if time */
                        if (!--nextflush) {
                                if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
                                        error(USER, "child(ren) died");
                                fflush(stdout);
                                nextflush = hresolu;
                        } else
                                something2flush = 1;
                }
                if (ferror(stdout))
                        error(SYSTEM, "write error");
                if (vcount && !--vcount)                /* check for end */
                        break;
        }
        if (ray_pnprocs > 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 'r':                               /* reflected contrib. */
                        *table++ = oputr;
                        castonly = 0;
                        break;
                case 'R':                               /* reflected distance */
                        *table++ = oputR;
                        castonly = 0;
                        break;
                case 'x':                               /* xmit contrib. */
                        *table++ = oputx;
                        castonly = 0;
                        break;
                case 'X':                               /* xmit distance */
                        *table++ = oputX;
                        castonly = 0;
                        break;
                case 'v':                               /* value */
                        *table++ = oputv;
                        castonly = 0;
                        break;
                case 'V':                               /* contribution */
                        *table++ = oputV;
                        castonly = 0;
                        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;
                        castonly = 0;
                        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;
                                                        /* compatibility */
        for (table = ray_out; *table != NULL; table++) {
                if ((*table == oputV) | (*table == oputW))
                        error(WARNING, "-oVW options require trace mode");
                if ((do_irrad | imm_irrad) &&
                                (*table == oputr) | (*table == oputR) |
                                (*table == oputx) | (*table == oputX))
                        error(WARNING, "-orRxX options incompatible with -I+ and -i+");
        }
}


static void
bogusray(void)                  /* print out empty record */
{
        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;
                                        /* pretend we hit surface */
        r->rxt = r->rot = 1e-5;
        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 (getbinary(vf, sizeof(float), 3, fp) != 3)
                        return(-1);
                VCOPY(vec, vf);
                break;
        case 'd':                                       /* binary double */
                if (getbinary(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, 3);
}


static void
oputd(                          /* print direction */
        RAY  *r
)
{
        (*putreal)(r->rdir, 3);
}


static void
oputr(                          /* print mirrored contribution */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->mcol,RED);
        cval[1] = colval(r->mcol,GRN);
        cval[2] = colval(r->mcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputR(                          /* print mirrored distance */
        RAY  *r
)
{
        (*putreal)(&r->rmt, 1);
}


static void
oputx(                          /* print unmirrored contribution */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->rcol,RED) - colval(r->mcol,RED);
        cval[1] = colval(r->rcol,GRN) - colval(r->mcol,GRN);
        cval[2] = colval(r->rcol,BLU) - colval(r->mcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputX(                          /* print unmirrored distance */
        RAY  *r
)
{
        (*putreal)(&r->rxt, 1);
}


static void
oputv(                          /* print value */
        RAY  *r
)
{
        RREAL   cval[3];

        cval[0] = colval(r->rcol,RED);
        cval[1] = colval(r->rcol,GRN);
        cval[2] = colval(r->rcol,BLU);
        (*putreal)(cval, 3);
}


static void
oputV(                          /* print value contribution */
        RAY *r
)
{
        RREAL   contr[3];

        raycontrib(contr, r, PRIMARY);
        multcolor(contr, r->rcol);
        (*putreal)(contr, 3);
}


static void
oputl(                          /* print effective distance */
        RAY  *r
)
{
        RREAL   d = raydistance(r);

        (*putreal)(&d, 1);
}


static void
oputL(                          /* print single ray length */
        RAY  *r
)
{
        (*putreal)(&r->rot, 1);
}


static void
oputc(                          /* print local coordinates */
        RAY  *r
)
{
        (*putreal)(r->uv, 2);
}


static RREAL    vdummy[3] = {0.0, 0.0, 0.0};


static void
oputp(                          /* print point */
        RAY  *r
)
{
        if (r->rot < FHUGE)
                (*putreal)(r->rop, 3);
        else
                (*putreal)(vdummy, 3);
}


static void
oputN(                          /* print unperturbed normal */
        RAY  *r
)
{
        if (r->rot < FHUGE) {
                rayreorient(r);
                (*putreal)(r->ron, 3);
        } else
                (*putreal)(vdummy, 3);
}


static void
oputn(                          /* print perturbed normal */
        RAY  *r
)
{
        FVECT  pnorm;

        if (r->rot >= FHUGE) {
                (*putreal)(vdummy, 3);
                return;
        }
        raynormal(pnorm, r);
        (*putreal)(pnorm, 3);
}


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
)
{
        RREAL   rwt = r->rweight;
        
        (*putreal)(&rwt, 1);
}


static void
oputW(                          /* print coefficient */
        RAY  *r
)
{
        RREAL   contr[3];
                                /* shadow ray not on source? */
        if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
                setcolor(contr, 0.0, 0.0, 0.0);
        else
                raycontrib(contr, r, PRIMARY);

        (*putreal)(contr, 3);
}


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(s) */
        RREAL *v, int n
)
{
        if (n == 3) {
                printf("%e\t%e\t%e\t", v[0], v[1], v[2]);
                return;
        }
        while (n--)
                printf("%e\t", *v++);
}


static void
putd(RREAL *v, int n)           /* output binary double(s) */
{
#ifdef  SMLFLT
        double  da[3];
        int     i;

        if (n > 3)
                error(INTERNAL, "code error in putd()");
        for (i = n; i--; )
                da[i] = v[i];
        putbinary(da, sizeof(double), n, stdout);
#else
        putbinary(v, sizeof(RREAL), n, stdout);
#endif
}


static void
putf(RREAL *v, int n)           /* output binary float(s) */
{
#ifndef SMLFLT
        float   fa[3];
        int     i;

        if (n > 3)
                error(INTERNAL, "code error in putf()");
        for (i = n; i--; )
                fa[i] = v[i];
        putbinary(fa, sizeof(float), n, stdout);
#else
        putbinary(v, sizeof(RREAL), n, stdout);
#endif
}


static void
putrgbe(RREAL *v, int n)        /* output RGBE color */
{
        COLR  cout;

        if (n != 3)
                error(INTERNAL, "putrgbe() not called with 3 components");
        setcolr(cout, v[0], v[1], v[2]);
        putbinary(cout, sizeof(cout), 1, stdout);
}
Revision:	2.82
Committed:	Thu Jul 25 16:50:54 2019 UTC (4 years, 9 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.81:	+4 -3 lines
Log Message:	Keep track of ray flips so rayreorient() can put it back as it was
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.82	static const char RCSid[] = "$Id: rtrace.c,v 2.81 2019/07/04 01:25:07 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.72	#include "otspecial.h"
32	greg	2.6	#include "resolu.h"
33	schorsch	2.49	#include "random.h"
34	greg	2.6
35	greg	2.57	extern int inform; /* input format */
36			extern int outform; /* output format */
37			extern char outvals; / output values */
38	greg	2.27
39	greg	2.57	extern int imm_irrad; /* compute immediate irradiance? */
40			extern int lim_dist; /* limit distance? */
41	greg	1.14
42	greg	2.57	extern char tralist[]; / list of modifers to trace (or no) */
43			extern int traincl; /* include == 1, exclude == 0 */
44	greg	1.13
45	greg	2.57	extern int hresolu; /* horizontal resolution */
46			extern int vresolu; /* vertical resolution */
47	greg	1.1
48	greg	2.57	static int castonly = 0;
49	greg	2.55
50	greg	2.46	#ifndef MAXTSET
51	greg	2.55	#define MAXTSET 8191 /* maximum number in trace set */
52	greg	2.46	#endif
53	greg	2.15	OBJECT traset[MAXTSET+1]={0}; /* trace include/exclude set */
54
55	greg	1.1	static RAY thisray; /* for our convenience */
56
57	greg	2.65	typedef void putf_t(RREAL *v, int n);
58	greg	2.80	static putf_t puta, putd, putf, putrgbe;
59	greg	1.1
60	schorsch	2.37	typedef void oputf_t(RAY *r);
61	greg	2.51	static oputf_t oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
62	greg	2.73	oputr, oputR, oputx, oputX, oputn, oputN, oputs,
63			oputw, oputW, oputm, oputM, oputtilde;
64	schorsch	2.37
65			static void setoutput(char *vs);
66	greg	2.57	extern void tranotify(OBJECT obj);
67	schorsch	2.37	static void bogusray(void);
68	greg	2.59	static void raycast(RAY *r);
69			static void rayirrad(RAY *r);
70			static void rtcompute(FVECT org, FVECT dir, double dmax);
71	greg	2.57	static int printvals(RAY *r);
72			static int getvec(FVECT vec, int fmt, FILE *fp);
73			static void tabin(RAY *r);
74			static void ourtrace(RAY *r);
75	greg	1.1
76	greg	2.73	static oputf_t ray_out[32], every_out[32];
77	schorsch	2.37	static putf_t *putreal;
78	greg	2.27
79	greg	1.1
80	greg	2.27	void
81	schorsch	2.37	quit( /* quit program */
82			int code
83			)
84	greg	1.1	{
85	greg	2.57	if (ray_pnprocs > 0) /* close children if any */
86			ray_pclose(0);
87			#ifndef NON_POSIX
88	greg	2.58	else if (!ray_pnprocs) {
89			headclean(); /* delete header file */
90			pfclean(); /* clean up persist files */
91			}
92	greg	2.11	#endif
93	greg	1.1	exit(code);
94			}
95
96
97	greg	2.57	char *
98	schorsch	2.37	formstr( /* return format identifier */
99			int f
100			)
101	greg	2.6	{
102			switch (f) {
103			case 'a': return("ascii");
104			case 'f': return("float");
105			case 'd': return("double");
106			case 'c': return(COLRFMT);
107			}
108			return("unknown");
109			}
110
111
112	schorsch	2.37	extern void
113			rtrace( /* trace rays from file */
114	greg	2.59	char *fname,
115			int nproc
116	schorsch	2.37	)
117	greg	1.1	{
118	greg	2.45	unsigned long vcount = (hresolu > 1) ? (unsigned long)hresolu*vresolu
119	greg	2.66	: (unsigned long)vresolu;
120	greg	2.64	long nextflush = (vresolu > 0) & (hresolu > 1) ? 0 : hresolu;
121	greg	2.79	int something2flush = 0;
122	greg	1.1	FILE *fp;
123	gregl	2.25	double d;
124	greg	1.2	FVECT orig, direc;
125	greg	1.1	/* set up input */
126			if (fname == NULL)
127			fp = stdin;
128			else if ((fp = fopen(fname, "r")) == NULL) {
129			sprintf(errmsg, "cannot open input file \"%s\"", fname);
130			error(SYSTEM, errmsg);
131			}
132	greg	2.8	if (inform != 'a')
133	schorsch	2.30	SET_FILE_BINARY(fp);
134	greg	1.1	/* set up output */
135	greg	2.16	setoutput(outvals);
136	greg	2.59	if (imm_irrad)
137			castonly = 0;
138			else if (castonly)
139			nproc = 1; /* don't bother multiprocessing */
140	greg	2.79	if ((nextflush > 0) & (nproc > nextflush)) {
141			error(WARNING, "reducing number of processes to match flush interval");
142			nproc = nextflush;
143			}
144	greg	1.1	switch (outform) {
145			case 'a': putreal = puta; break;
146			case 'f': putreal = putf; break;
147			case 'd': putreal = putd; break;
148	greg	2.6	case 'c':
149	greg	2.81	if (outvals[0] && (outvals[1] \|\| !strchr("vrx", outvals[0])))
150			error(USER, "color format only with -ov, -or, -ox");
151	greg	2.80	putreal = putrgbe; break;
152	greg	2.6	default:
153			error(CONSISTENCY, "botched output format");
154	greg	1.1	}
155	greg	2.59	if (nproc > 1) { /* start multiprocessing */
156			ray_popen(nproc);
157	greg	2.58	ray_fifo_out = printvals;
158	greg	2.59	}
159	greg	2.12	if (hresolu > 0) {
160			if (vresolu > 0)
161			fprtresolu(hresolu, vresolu, stdout);
162	greg	2.78	else
163			fflush(stdout);
164	greg	2.12	}
165	greg	1.1	/* process file */
166			while (getvec(orig, inform, fp) == 0 &&
167			getvec(direc, inform, fp) == 0) {
168
169	gregl	2.25	d = normalize(direc);
170			if (d == 0.0) { /* zero ==> flush */
171	greg	2.79	if ((--nextflush <= 0) \| !vcount && something2flush) {
172	greg	2.74	if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
173	greg	2.71	error(USER, "child(ren) died");
174	greg	2.63	bogusray();
175	gregl	2.24	fflush(stdout);
176	greg	2.79	something2flush = 0;
177	greg	2.64	nextflush = (vresolu > 0) & (hresolu > 1) ? 0 :
178			hresolu;
179	greg	2.63	} else
180			bogusray();
181	greg	2.57	} else { /* compute and print */
182	greg	2.59	rtcompute(orig, direc, lim_dist ? d : 0.0);
183	greg	1.7	/* flush if time */
184	greg	2.45	if (!--nextflush) {
185	greg	2.74	if (ray_pnprocs > 1 && ray_fifo_flush() < 0)
186	greg	2.71	error(USER, "child(ren) died");
187	gregl	2.24	fflush(stdout);
188			nextflush = hresolu;
189	greg	2.78	} else
190	greg	2.79	something2flush = 1;
191	greg	1.1	}
192			if (ferror(stdout))
193			error(SYSTEM, "write error");
194	greg	2.45	if (vcount && !--vcount) /* check for end */
195	greg	1.1	break;
196			}
197	greg	2.74	if (ray_pnprocs > 1) { /* clean up children */
198	greg	2.59	if (ray_fifo_flush() < 0)
199			error(USER, "unable to complete processing");
200			ray_pclose(0);
201			}
202	greg	2.50	if (fflush(stdout) < 0)
203			error(SYSTEM, "write error");
204	greg	2.45	if (vcount)
205	greg	2.42	error(USER, "unexpected EOF on input");
206	greg	2.12	if (fname != NULL)
207			fclose(fp);
208	greg	1.1	}
209
210
211	schorsch	2.37	static void
212	greg	2.40	trace_sources(void) /* trace rays to light sources, also */
213			{
214			int sn;
215
216			for (sn = 0; sn < nsources; sn++)
217			source[sn].sflags \|= SFOLLOW;
218			}
219
220
221			static void
222	schorsch	2.37	setoutput( /* set up output tables */
223	greg	2.56	char *vs
224	schorsch	2.37	)
225	greg	1.1	{
226	greg	2.56	oputf_t **table = ray_out;
227	greg	1.1
228	greg	1.11	castonly = 1;
229	greg	1.1	while (*vs)
230			switch (*vs++) {
231	greg	2.40	case 'T': /* trace sources */
232	greg	2.42	if (!*vs) break;
233	greg	2.40	trace_sources();
234			/* fall through */
235	greg	1.1	case 't': /* trace */
236	greg	2.42	if (!*vs) break;
237	greg	1.1	*table = NULL;
238			table = every_out;
239			trace = ourtrace;
240	greg	1.11	castonly = 0;
241	greg	1.1	break;
242			case 'o': /* origin */
243			*table++ = oputo;
244			break;
245			case 'd': /* direction */
246			*table++ = oputd;
247			break;
248	greg	2.73	case 'r': /* reflected contrib. */
249			*table++ = oputr;
250			castonly = 0;
251			break;
252			case 'R': /* reflected distance */
253			*table++ = oputR;
254			castonly = 0;
255			break;
256			case 'x': /* xmit contrib. */
257			*table++ = oputx;
258			castonly = 0;
259			break;
260			case 'X': /* xmit distance */
261			*table++ = oputX;
262			castonly = 0;
263			break;
264	greg	1.1	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	greg	2.73	castonly = 0;
271	greg	2.51	if (ambounce > 0 && (ambacc > FTINY \|\| ambssamp > 0))
272			error(WARNING,
273			"-otV accuracy depends on -aa 0 -as 0");
274			break;
275	greg	2.5	case 'l': /* effective distance */
276	greg	1.1	*table++ = oputl;
277	greg	1.11	castonly = 0;
278	greg	1.1	break;
279	greg	2.29	case 'c': /* local coordinates */
280			*table++ = oputc;
281			break;
282	greg	2.5	case 'L': /* single ray length */
283			*table++ = oputL;
284			break;
285	greg	1.1	case 'p': /* point */
286			*table++ = oputp;
287			break;
288	greg	2.9	case 'n': /* perturbed normal */
289	greg	1.1	*table++ = oputn;
290	greg	2.9	castonly = 0;
291	greg	1.1	break;
292	greg	2.9	case 'N': /* unperturbed normal */
293			*table++ = oputN;
294			break;
295	greg	1.1	case 's': /* surface */
296			*table++ = oputs;
297			break;
298			case 'w': /* weight */
299			*table++ = oputw;
300			break;
301	greg	2.40	case 'W': /* coefficient */
302			*table++ = oputW;
303	greg	2.73	castonly = 0;
304	greg	2.80	if (ambounce > 0 && (ambacc > FTINY) \| (ambssamp > 0))
305	greg	2.40	error(WARNING,
306	greg	2.41	"-otW accuracy depends on -aa 0 -as 0");
307	greg	2.40	break;
308	greg	1.1	case 'm': /* modifier */
309			*table++ = oputm;
310			break;
311	greg	2.39	case 'M': /* material */
312			*table++ = oputM;
313			break;
314	greg	2.42	case '~': /* tilde */
315			*table++ = oputtilde;
316	greg	2.41	break;
317	greg	1.1	}
318			*table = NULL;
319	greg	2.81	/* compatibility */
320			for (table = ray_out; *table != NULL; table++) {
321			if ((table == oputV) \| (table == oputW))
322			error(WARNING, "-oVW options require trace mode");
323			if ((do_irrad \| imm_irrad) &&
324			(table == oputr) \| (table == oputR) \|
325			(table == oputx) \| (table == oputX))
326			error(WARNING, "-orRxX options incompatible with -I+ and -i+");
327			}
328	gregl	2.24	}
329
330
331	schorsch	2.37	static void
332			bogusray(void) /* print out empty record */
333	gregl	2.24	{
334	greg	2.40	rayorigin(&thisray, PRIMARY, NULL, NULL);
335	gregl	2.24	printvals(&thisray);
336	greg	1.1	}
337
338
339	schorsch	2.37	static void
340	greg	2.59	raycast( /* compute first ray intersection only */
341			RAY *r
342	schorsch	2.37	)
343	greg	1.1	{
344	greg	2.59	if (!localhit(r, &thescene)) {
345			if (r->ro == &Aftplane) { /* clipped */
346			r->ro = NULL;
347			r->rot = FHUGE;
348			} else
349			sourcehit(r);
350	greg	2.57	}
351	greg	1.1	}
352
353
354	schorsch	2.37	static void
355	greg	2.59	rayirrad( /* compute irradiance rather than radiance */
356			RAY *r
357			)
358			{
359	greg	2.62	void (old_revf)(RAY ) = r->revf;
360	greg	2.75	/* pretend we hit surface */
361	greg	2.76	r->rxt = r->rot = 1e-5;
362	greg	2.59	VSUM(r->rop, r->rorg, r->rdir, r->rot);
363			r->ron[0] = -r->rdir[0];
364			r->ron[1] = -r->rdir[1];
365			r->ron[2] = -r->rdir[2];
366			r->rod = 1.0;
367	greg	2.62	/* compute result */
368	greg	2.61	r->revf = raytrace;
369	greg	2.59	(*ofun[Lamb.otype].funp)(&Lamb, r);
370	greg	2.62	r->revf = old_revf;
371	greg	2.59	}
372
373
374			static void
375			rtcompute( /* compute and print ray value(s) */
376	schorsch	2.37	FVECT org,
377	greg	2.59	FVECT dir,
378			double dmax
379	schorsch	2.37	)
380	greg	1.1	{
381	greg	2.60	/* set up ray */
382			rayorigin(&thisray, PRIMARY, NULL, NULL);
383			if (imm_irrad) {
384	greg	2.59	VSUM(thisray.rorg, org, dir, 1.1e-4);
385			thisray.rdir[0] = -dir[0];
386			thisray.rdir[1] = -dir[1];
387			thisray.rdir[2] = -dir[2];
388			thisray.rmax = 0.0;
389	greg	2.60	thisray.revf = rayirrad;
390	greg	2.59	} else {
391			VCOPY(thisray.rorg, org);
392			VCOPY(thisray.rdir, dir);
393			thisray.rmax = dmax;
394	greg	2.60	if (castonly)
395			thisray.revf = raycast;
396	greg	2.59	}
397			if (ray_pnprocs > 1) { /* multiprocessing FIFO? */
398			if (ray_fifo_in(&thisray) < 0)
399			error(USER, "lost children");
400			return;
401			}
402			samplendx++; /* else do it ourselves */
403			rayvalue(&thisray);
404	greg	2.16	printvals(&thisray);
405			}
406
407
408	greg	2.57	static int
409	schorsch	2.37	printvals( /* print requested ray values */
410			RAY *r
411			)
412	greg	2.16	{
413	greg	2.56	oputf_t **tp;
414	greg	2.16
415			if (ray_out[0] == NULL)
416	greg	2.57	return(0);
417	greg	2.16	for (tp = ray_out; *tp != NULL; tp++)
418			(**tp)(r);
419	greg	1.1	if (outform == 'a')
420			putchar('\n');
421	greg	2.57	return(1);
422	greg	1.1	}
423
424
425	schorsch	2.37	static int
426			getvec( /* get a vector from fp */
427	greg	2.56	FVECT vec,
428	schorsch	2.37	int fmt,
429			FILE *fp
430			)
431	greg	1.1	{
432			static float vf[3];
433	greg	2.5	static double vd[3];
434	greg	1.19	char buf[32];
435	greg	2.56	int i;
436	greg	1.1
437			switch (fmt) {
438			case 'a': /* ascii */
439	greg	1.19	for (i = 0; i < 3; i++) {
440			if (fgetword(buf, sizeof(buf), fp) == NULL \|\|
441			!isflt(buf))
442			return(-1);
443			vec[i] = atof(buf);
444			}
445	greg	1.1	break;
446			case 'f': /* binary float */
447	greg	2.70	if (getbinary(vf, sizeof(float), 3, fp) != 3)
448	greg	1.1	return(-1);
449	greg	2.58	VCOPY(vec, vf);
450	greg	1.1	break;
451			case 'd': /* binary double */
452	greg	2.70	if (getbinary(vd, sizeof(double), 3, fp) != 3)
453	greg	1.1	return(-1);
454	greg	2.58	VCOPY(vec, vd);
455	greg	1.1	break;
456	greg	2.6	default:
457			error(CONSISTENCY, "botched input format");
458	greg	1.1	}
459			return(0);
460			}
461
462
463	greg	2.57	void
464	schorsch	2.37	tranotify( /* record new modifier */
465			OBJECT obj
466			)
467	greg	2.15	{
468			static int hitlimit = 0;
469	greg	2.56	OBJREC *o = objptr(obj);
470			char **tralp;
471	greg	2.15
472	greg	2.27	if (obj == OVOID) { /* starting over */
473			traset[0] = 0;
474			hitlimit = 0;
475			return;
476			}
477	greg	2.15	if (hitlimit \|\| !ismodifier(o->otype))
478			return;
479			for (tralp = tralist; *tralp != NULL; tralp++)
480			if (!strcmp(o->oname, *tralp)) {
481			if (traset[0] >= MAXTSET) {
482			error(WARNING, "too many modifiers in trace list");
483			hitlimit++;
484			return; /* should this be fatal? */
485			}
486			insertelem(traset, obj);
487			return;
488			}
489			}
490
491
492	greg	2.27	static void
493	schorsch	2.37	ourtrace( /* print ray values */
494			RAY *r
495			)
496	greg	1.1	{
497	greg	2.56	oputf_t **tp;
498	greg	1.1
499			if (every_out[0] == NULL)
500	greg	2.15	return;
501	greg	2.16	if (r->ro == NULL) {
502			if (traincl == 1)
503			return;
504			} else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
505	greg	1.1	return;
506			tabin(r);
507			for (tp = every_out; *tp != NULL; tp++)
508			(**tp)(r);
509	greg	2.41	if (outform == 'a')
510			putchar('\n');
511	greg	1.1	}
512
513
514	greg	2.27	static void
515	schorsch	2.37	tabin( /* tab in appropriate amount */
516			RAY *r
517			)
518	greg	1.1	{
519	greg	2.40	const RAY *rp;
520	greg	1.1
521			for (rp = r->parent; rp != NULL; rp = rp->parent)
522			putchar('\t');
523			}
524
525
526	greg	2.27	static void
527	schorsch	2.37	oputo( /* print origin */
528			RAY *r
529			)
530	greg	1.1	{
531	greg	2.65	(*putreal)(r->rorg, 3);
532	greg	1.1	}
533
534
535	greg	2.27	static void
536	schorsch	2.37	oputd( /* print direction */
537			RAY *r
538			)
539	greg	1.1	{
540	greg	2.65	(*putreal)(r->rdir, 3);
541	greg	1.1	}
542
543
544	greg	2.27	static void
545	greg	2.73	oputr( /* print mirrored contribution */
546			RAY *r
547			)
548			{
549			RREAL cval[3];
550
551			cval[0] = colval(r->mcol,RED);
552			cval[1] = colval(r->mcol,GRN);
553			cval[2] = colval(r->mcol,BLU);
554			(*putreal)(cval, 3);
555			}
556
557
558
559			static void
560			oputR( /* print mirrored distance */
561			RAY *r
562			)
563			{
564			(*putreal)(&r->rmt, 1);
565			}
566
567
568			static void
569			oputx( /* print unmirrored contribution */
570			RAY *r
571			)
572			{
573			RREAL cval[3];
574
575			cval[0] = colval(r->rcol,RED) - colval(r->mcol,RED);
576			cval[1] = colval(r->rcol,GRN) - colval(r->mcol,GRN);
577			cval[2] = colval(r->rcol,BLU) - colval(r->mcol,BLU);
578			(*putreal)(cval, 3);
579			}
580
581
582			static void
583			oputX( /* print unmirrored distance */
584			RAY *r
585			)
586			{
587			(*putreal)(&r->rxt, 1);
588			}
589
590
591			static void
592	schorsch	2.37	oputv( /* print value */
593			RAY *r
594			)
595	greg	1.1	{
596	greg	2.65	RREAL cval[3];
597
598			cval[0] = colval(r->rcol,RED);
599			cval[1] = colval(r->rcol,GRN);
600			cval[2] = colval(r->rcol,BLU);
601			(*putreal)(cval, 3);
602	greg	1.1	}
603
604
605	greg	2.27	static void
606	greg	2.51	oputV( /* print value contribution */
607			RAY *r
608			)
609			{
610	greg	2.65	RREAL contr[3];
611	greg	2.51
612			raycontrib(contr, r, PRIMARY);
613			multcolor(contr, r->rcol);
614	greg	2.65	(*putreal)(contr, 3);
615	greg	2.51	}
616
617
618			static void
619	schorsch	2.37	oputl( /* print effective distance */
620			RAY *r
621			)
622	greg	1.1	{
623	greg	2.73	RREAL d = raydistance(r);
624
625			(*putreal)(&d, 1);
626	greg	2.5	}
627
628
629	greg	2.27	static void
630	schorsch	2.37	oputL( /* print single ray length */
631			RAY *r
632			)
633	greg	2.5	{
634	greg	2.65	(*putreal)(&r->rot, 1);
635	greg	1.1	}
636
637
638	greg	2.27	static void
639	schorsch	2.37	oputc( /* print local coordinates */
640			RAY *r
641			)
642	greg	2.29	{
643	greg	2.65	(*putreal)(r->uv, 2);
644	greg	2.29	}
645
646
647	greg	2.65	static RREAL vdummy[3] = {0.0, 0.0, 0.0};
648
649
650	greg	2.29	static void
651	schorsch	2.37	oputp( /* print point */
652			RAY *r
653			)
654	greg	1.1	{
655	greg	2.65	if (r->rot < FHUGE)
656			(*putreal)(r->rop, 3);
657			else
658			(*putreal)(vdummy, 3);
659	greg	1.1	}
660
661
662	greg	2.27	static void
663	schorsch	2.37	oputN( /* print unperturbed normal */
664			RAY *r
665			)
666	greg	1.1	{
667	greg	2.82	if (r->rot < FHUGE) {
668			rayreorient(r);
669	greg	2.65	(*putreal)(r->ron, 3);
670	greg	2.82	} else
671	greg	2.65	(*putreal)(vdummy, 3);
672	greg	2.9	}
673
674
675	greg	2.27	static void
676	schorsch	2.37	oputn( /* print perturbed normal */
677			RAY *r
678			)
679	greg	2.9	{
680			FVECT pnorm;
681
682			if (r->rot >= FHUGE) {
683	greg	2.65	(*putreal)(vdummy, 3);
684	greg	2.9	return;
685			}
686			raynormal(pnorm, r);
687	greg	2.65	(*putreal)(pnorm, 3);
688	greg	1.1	}
689
690
691	greg	2.27	static void
692	schorsch	2.37	oputs( /* print name */
693			RAY *r
694			)
695	greg	1.1	{
696			if (r->ro != NULL)
697			fputs(r->ro->oname, stdout);
698			else
699			putchar('*');
700			putchar('\t');
701			}
702
703
704	greg	2.27	static void
705	schorsch	2.37	oputw( /* print weight */
706			RAY *r
707			)
708	greg	1.1	{
709	greg	2.65	RREAL rwt = r->rweight;
710
711			(*putreal)(&rwt, 1);
712	greg	1.1	}
713
714
715	greg	2.27	static void
716	greg	2.51	oputW( /* print coefficient */
717	greg	2.40	RAY *r
718			)
719			{
720	greg	2.65	RREAL contr[3];
721	greg	2.67	/* shadow ray not on source? */
722			if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
723			setcolor(contr, 0.0, 0.0, 0.0);
724			else
725			raycontrib(contr, r, PRIMARY);
726	greg	2.40
727	greg	2.65	(*putreal)(contr, 3);
728	greg	2.40	}
729
730
731			static void
732	schorsch	2.37	oputm( /* print modifier */
733			RAY *r
734			)
735	greg	1.1	{
736			if (r->ro != NULL)
737	greg	2.23	if (r->ro->omod != OVOID)
738			fputs(objptr(r->ro->omod)->oname, stdout);
739			else
740			fputs(VOIDID, stdout);
741	greg	1.1	else
742			putchar('*');
743			putchar('\t');
744			}
745
746
747	greg	2.27	static void
748	greg	2.39	oputM( /* print material */
749			RAY *r
750			)
751			{
752			OBJREC *mat;
753
754			if (r->ro != NULL) {
755			if ((mat = findmaterial(r->ro)) != NULL)
756			fputs(mat->oname, stdout);
757			else
758			fputs(VOIDID, stdout);
759			} else
760			putchar('*');
761			putchar('\t');
762			}
763
764
765			static void
766	greg	2.42	oputtilde( /* output tilde (spacer) */
767	greg	2.41	RAY *r
768			)
769			{
770	greg	2.42	fputs("~\t", stdout);
771	greg	2.41	}
772
773
774			static void
775	greg	2.65	puta( /* print ascii value(s) */
776			RREAL *v, int n
777	schorsch	2.37	)
778	greg	1.1	{
779	greg	2.65	if (n == 3) {
780			printf("%e\t%e\t%e\t", v[0], v[1], v[2]);
781			return;
782			}
783			while (n--)
784			printf("%e\t", *v++);
785	greg	1.1	}
786
787
788	greg	2.27	static void
789	greg	2.80	putd(RREAL v, int n) / output binary double(s) */
790	greg	1.1	{
791	greg	2.68	#ifdef SMLFLT
792			double da[3];
793			int i;
794
795			if (n > 3)
796	greg	2.65	error(INTERNAL, "code error in putd()");
797	greg	2.68	for (i = n; i--; )
798			da[i] = v[i];
799	greg	2.69	putbinary(da, sizeof(double), n, stdout);
800	greg	2.68	#else
801	greg	2.69	putbinary(v, sizeof(RREAL), n, stdout);
802	greg	2.68	#endif
803	greg	1.1	}
804
805
806	greg	2.27	static void
807	greg	2.80	putf(RREAL v, int n) / output binary float(s) */
808	greg	1.1	{
809	greg	2.68	#ifndef SMLFLT
810	greg	2.65	float fa[3];
811			int i;
812	greg	1.1
813	greg	2.65	if (n > 3)
814			error(INTERNAL, "code error in putf()");
815			for (i = n; i--; )
816			fa[i] = v[i];
817	greg	2.69	putbinary(fa, sizeof(float), n, stdout);
818	greg	2.68	#else
819	greg	2.69	putbinary(v, sizeof(RREAL), n, stdout);
820	greg	2.68	#endif
821	greg	1.1	}
822	greg	2.80
823
824			static void
825			putrgbe(RREAL v, int n) / output RGBE color */
826			{
827			COLR cout;
828
829			if (n != 3)
830			error(INTERNAL, "putrgbe() not called with 3 components");
831			setcolr(cout, v[0], v[1], v[2]);
832			putbinary(cout, sizeof(cout), 1, stdout);
833			}