--- ray/src/rt/source.c	1990/12/15 15:03:34	1.26
+++ ray/src/rt/source.c	1991/10/21 12:58:06	1.45
@@ -1,4 +1,4 @@
-/* Copyright (c) 1990 Regents of the University of California */
+/* Copyright (c) 1991 Regents of the University of California */
 
 #ifndef lint
 static char SCCSid[] = "$SunId$ LBL";
@@ -14,38 +14,44 @@ static char SCCSid[] = "$SunId$ LBL";
 
 #include  "octree.h"
 
-#include  "source.h"
-
 #include  "otypes.h"
 
-#include  "cone.h"
+#include  "source.h"
 
-#include  "face.h"
+/*
+ * Structures used by direct()
+ */
 
-#include  "random.h"
+typedef struct {
+	int  sno;		/* source number */
+	FVECT  dir;		/* source direction */
+	COLOR  coef;		/* material coefficient */
+	COLOR  val;		/* contribution */
+}  CONTRIB;		/* direct contribution */
 
+typedef struct {
+	int  sndx;		/* source index (to CONTRIB array) */
+	float  brt;		/* brightness (for comparison) */
+}  CNTPTR;		/* contribution pointer */
 
-extern double  dstrsrc;			/* source distribution amount */
-extern double  shadthresh;		/* relative shadow threshold */
-extern double  shadcert;		/* shadow testing certainty */
-
-SRCREC  *source = NULL;			/* our list of sources */
-int  nsources = 0;			/* the number of sources */
-
 static CONTRIB  *srccnt;		/* source contributions in direct() */
 static CNTPTR  *cntord;			/* source ordering in direct() */
+static int  maxcntr = 0;		/* size of contribution arrays */
 
 
 marksources()			/* find and mark source objects */
 {
+	int  i;
 	register OBJREC  *o, *m;
-	register int  i;
-
+	register int  ns;
+					/* initialize dispatch table */
+	initstypes();
+					/* find direct sources */
 	for (i = 0; i < nobjects; i++) {
 	
 		o = objptr(i);
 
-		if (o->omod == OVOID)
+		if (!issurface(o->otype) || o->omod == OVOID)
 			continue;
 
 		m = objptr(o->omod);
@@ -62,236 +68,126 @@ marksources()			/* find and mark source objects */
 				m->oargs.farg[3] <= FTINY)
 			continue;			/* don't bother */
 
-		if (source == NULL)
-			source = (SRCREC *)malloc(sizeof(SRCREC));
-		else
-			source = (SRCREC *)realloc((char *)source,
-					(unsigned)(nsources+1)*sizeof(SRCREC));
-		if (source == NULL)
+		if (sfun[o->otype].of == NULL ||
+				sfun[o->otype].of->setsrc == NULL)
+			objerror(o, USER, "illegal material");
+
+		if ((ns = newsource()) < 0)
 			goto memerr;
 
-		newsource(&source[nsources], o);
+		setsource(&source[ns], o);
 
 		if (m->otype == MAT_GLOW) {
-			source[nsources].sflags |= SPROX;
-			source[nsources].sl.prox = m->oargs.farg[3];
+			source[ns].sflags |= SPROX;
+			source[ns].sl.prox = m->oargs.farg[3];
 			if (o->otype == OBJ_SOURCE)
-				source[nsources].sflags |= SSKIP;
+				source[ns].sflags |= SSKIP;
 		} else if (m->otype == MAT_SPOT) {
-			source[nsources].sflags |= SSPOT;
-			source[nsources].sl.s = makespot(m);
+			source[ns].sflags |= SSPOT;
+			if ((source[ns].sl.s = makespot(m)) == NULL)
+				goto memerr;
+			if (source[ns].sflags & SFLAT &&
+				!checkspot(source[ns].sl.s,source[ns].snorm)) {
+				objerror(o, WARNING,
+					"invalid spotlight direction");
+				source[ns].sflags |= SSKIP;
+			}
 		}
-		nsources++;
 	}
 	if (nsources <= 0) {
 		error(WARNING, "no light sources found");
 		return;
 	}
-	srccnt = (CONTRIB *)malloc(nsources*sizeof(CONTRIB));
-	cntord = (CNTPTR *)malloc(nsources*sizeof(CNTPTR));
-	if (srccnt != NULL && cntord != NULL)
-		return;
-	/* fall through */
+	markvirtuals();			/* find and add virtual sources */
+				/* allocate our contribution arrays */
+	maxcntr = nsources + MAXSPART*2;	/* start with this many */
+	srccnt = (CONTRIB *)malloc(maxcntr*sizeof(CONTRIB));
+	cntord = (CNTPTR *)malloc(maxcntr*sizeof(CNTPTR));
+	if (srccnt == NULL || cntord == NULL)
+		goto memerr;
+	return;
 memerr:
 	error(SYSTEM, "out of memory in marksources");
 }
 
 
-newsource(src, so)			/* add a source to the array */
-register SRCREC  *src;
-register OBJREC  *so;
-{
-	double  cos(), tan(), sqrt();
-	double  theta;
-	FACE  *f;
-	CONE  *co;
-	int  j;
-	register int  i;
-	
-	src->sflags = 0;
-	src->nhits = 1; src->ntests = 2;	/* start probability = 1/2 */
-	src->so = so;
-
-	switch (so->otype) {
-	case OBJ_SOURCE:
-		if (so->oargs.nfargs != 4)
-			objerror(so, USER, "bad arguments");
-		src->sflags |= SDISTANT;
-		VCOPY(src->sloc, so->oargs.farg);
-		if (normalize(src->sloc) == 0.0)
-			objerror(so, USER, "zero direction");
-		theta = PI/180.0/2.0 * so->oargs.farg[3];
-		if (theta <= FTINY)
-			objerror(so, USER, "zero size");
-		src->ss = theta >= PI/4 ? 1.0 : tan(theta);
-		src->ss2 = 2.0*PI * (1.0 - cos(theta));
-		break;
-	case OBJ_SPHERE:
-		VCOPY(src->sloc, so->oargs.farg);
-		src->ss = so->oargs.farg[3];
-		src->ss2 = PI * src->ss * src->ss;
-		break;
-	case OBJ_FACE:
-						/* get the face */
-		f = getface(so);
-						/* find the center */
-		for (j = 0; j < 3; j++) {
-			src->sloc[j] = 0.0;
-			for (i = 0; i < f->nv; i++)
-				src->sloc[j] += VERTEX(f,i)[j];
-			src->sloc[j] /= f->nv;
-		}
-		if (!inface(src->sloc, f))
-			objerror(so, USER, "cannot hit center");
-		src->ss = sqrt(f->area / PI);
-		src->ss2 = f->area;
-		break;
-	case OBJ_RING:
-						/* get the ring */
-		co = getcone(so, 0);
-		VCOPY(src->sloc, CO_P0(co));
-		if (CO_R0(co) > 0.0)
-			objerror(so, USER, "cannot hit center");
-		src->ss = CO_R1(co);
-		src->ss2 = PI * src->ss * src->ss;
-		break;
-	default:
-		objerror(so, USER, "illegal material");
-	}
-}
-
-
-SPOT *
-makespot(m)			/* make a spotlight */
-register OBJREC  *m;
-{
-	extern double  cos();
-	register SPOT  *ns;
-
-	if ((ns = (SPOT *)malloc(sizeof(SPOT))) == NULL)
-		error(SYSTEM, "out of memory in makespot");
-	ns->siz = 2.0*PI * (1.0 - cos(PI/180.0/2.0 * m->oargs.farg[3]));
-	VCOPY(ns->aim, m->oargs.farg+4);
-	if ((ns->flen = normalize(ns->aim)) == 0.0)
-		objerror(m, USER, "zero focus vector");
-	return(ns);
-}
-
-
-double
-srcray(sr, r, sn)		/* send a ray to a source, return domega */
+srcray(sr, r, si)		/* send a ray to a source, return domega */
 register RAY  *sr;		/* returned source ray */
 RAY  *r;			/* ray which hit object */
-register int  sn;		/* source number */
+SRCINDEX  *si;			/* source sample index */
 {
-	register double  *norm = NULL;	/* plane normal */
-	double  ddot;			/* (distance times) cosine */
-	FVECT  vd;
-	double  d;
-	register int  i;
+    double  d;				/* distance to source */
+    FVECT  vd;
+    register SRCREC  *srcp;
+    register int  i;
 
-	if (source[sn].sflags & SSKIP)
-		return(0.0);			/* skip this source */
+    rayorigin(sr, r, SHADOW, 1.0);		/* ignore limits */
 
-	rayorigin(sr, r, SHADOW, 1.0);		/* ignore limits */
-
-	sr->rsrc = sn;				/* remember source */
-						/* get source direction */
-	if (source[sn].sflags & SDISTANT)
-						/* constant direction */
-		VCOPY(sr->rdir, source[sn].sloc);
-	else {					/* compute direction */
-		for (i = 0; i < 3; i++)
-			sr->rdir[i] = source[sn].sloc[i] - sr->rorg[i];
-
-		if (source[sn].so->otype == OBJ_FACE)
- 			norm = getface(source[sn].so)->norm;
-		else if (source[sn].so->otype == OBJ_RING)
-			norm = getcone(source[sn].so,0)->ad;
-
-		if (norm != NULL && (ddot = -DOT(sr->rdir, norm)) <= FTINY)
-			return(0.0);		/* behind surface! */
-	}
-	if (dstrsrc > FTINY) {
-					/* distribute source direction */
-		for (i = 0; i < 3; i++)
-			vd[i] = dstrsrc * source[sn].ss * (1.0 - 2.0*frandom());
-
-		if (norm != NULL) {		/* project offset */
-			d = DOT(vd, norm);
+    while ((d = nextssamp(sr->rorg, sr->rdir, si)) != 0.0) {
+	sr->rsrc = si->sn;			/* remember source */
+	srcp = source + si->sn;
+	if (srcp->sflags & SDISTANT) {
+		if (srcp->sflags & SSPOT) {	/* check location */
 			for (i = 0; i < 3; i++)
-				vd[i] -= d * norm[i];
+			    vd[i] = srcp->sl.s->aim[i] - sr->rorg[i];
+			d = DOT(sr->rdir,vd);
+			if (d <= FTINY)
+				continue;
+			d = DOT(vd,vd) - d*d;
+			if (PI*d > srcp->sl.s->siz)
+				continue;
 		}
-		for (i = 0; i < 3; i++)		/* offset source direction */
-			sr->rdir[i] += vd[i];
-
-	} else if (source[sn].sflags & SDISTANT)
-						/* already normalized */
-		return(source[sn].ss2);
-
-	if ((d = normalize(sr->rdir)) == 0.0)
-						/* at source! */
-		return(0.0);
-	
-	if (source[sn].sflags & SDISTANT)
-						/* domega constant */
-		return(source[sn].ss2);
-
-	else {
+		return(1);		/* sample OK */
+	}
+				/* local source */
 						/* check proximity */
-		if (source[sn].sflags & SPROX &&
-				d > source[sn].sl.prox)
-			return(0.0);
-
-		if (norm != NULL)
-			ddot /= d;
-		else
-			ddot = 1.0;
+	if (srcp->sflags & SPROX && d > srcp->sl.prox)
+		continue;
 						/* check angle */
-		if (source[sn].sflags & SSPOT) {
-			if (source[sn].sl.s->siz < 2.0*PI *
-				(1.0 + DOT(source[sn].sl.s->aim,sr->rdir)))
-				return(0.0);
-			d += source[sn].sl.s->flen;
-		}
-						/* return domega */
-		return(ddot*source[sn].ss2/(d*d));
+	if (srcp->sflags & SSPOT) {
+		if (srcp->sl.s->siz < 2.0*PI *
+				(1.0 + DOT(srcp->sl.s->aim,sr->rdir)))
+			continue;
+					/* adjust solid angle */
+		si->dom *= d*d;
+		d += srcp->sl.s->flen;
+		si->dom /= d*d;
 	}
+	return(1);			/* sample OK */
+    }
+    return(0);			/* no more samples */
 }
 
 
-sourcehit(r)			/* check to see if ray hit distant source */
-register RAY  *r;
+srcvalue(r)			/* punch ray to source and compute value */
+RAY  *r;
 {
-	int  first, last;
-	register int  i;
+	register SRCREC  *sp;
 
-	if (r->rsrc >= 0) {		/* check only one if aimed */
-		first = last = r->rsrc;
-	} else {			/* otherwise check all */
-		first = 0; last = nsources-1;
+	sp = &source[r->rsrc];
+	if (sp->sflags & SVIRTUAL) {	/* virtual source */
+					/* check intersection */
+		if (!(*ofun[sp->so->otype].funp)(sp->so, r))
+			return;
+		raycont(r);		/* compute contribution */
+		return;
 	}
-	for (i = first; i <= last; i++)
-		if (source[i].sflags & SDISTANT)
-			/*
-			 * Check to see if ray is within
-			 * solid angle of source.
-			 */
-			if (2.0*PI * (1.0 - DOT(source[i].sloc,r->rdir))
-					<= source[i].ss2) {
-				r->ro = source[i].so;
-				if (!(source[i].sflags & SSKIP))
-					break;
-			}
-
-	if (r->ro != NULL) {
-		for (i = 0; i < 3; i++)
-			r->ron[i] = -r->rdir[i];
-		r->rod = 1.0;
-		r->rox = NULL;
-		return(1);
+					/* compute intersection */
+	if (sp->sflags & SDISTANT ? sourcehit(r) :
+			(*ofun[sp->so->otype].funp)(sp->so, r)) {
+		if (sp->sa.success >= 0)
+			sp->sa.success++;
+		raycont(r);		/* compute contribution */
+		return;
 	}
-	return(0);
+	if (sp->sa.success < 0)
+		return;			/* bitched already */
+	sp->sa.success -= AIMREQT;
+	if (sp->sa.success >= 0)
+		return;			/* leniency */
+	sprintf(errmsg, "aiming failure for light source \"%s\"",
+			sp->so->oname);
+	error(WARNING, errmsg);		/* issue warning */
 }
 
 
@@ -312,53 +208,51 @@ RAY  *r;			/* ray that hit surface */
 int  (*f)();			/* direct component coefficient function */
 char  *p;			/* data for f */
 {
+	extern int  (*trace)();
 	extern double  pow();
 	register int  sn;
+	SRCINDEX  si;
 	int  nshadcheck, ncnts;
-	double  prob, ourthresh, hwt, test2, hit2;
+	int  nhits;
+	double  prob, ourthresh, hwt;
 	RAY  sr;
 			/* NOTE: srccnt and cntord global so no recursion */
 	if (nsources <= 0)
 		return;		/* no sources?! */
-						/* compute number to check */
-	nshadcheck = pow((double)nsources, shadcert) + .5;
-						/* modify threshold */
-	ourthresh = shadthresh / r->rweight;
 						/* potential contributions */
-	for (sn = 0; sn < nsources; sn++) {
-		cntord[sn].sno = sn;
+	initsrcindex(&si);
+	for (sn = 0; srcray(&sr, r, &si); sn++) {
+		if (sn >= maxcntr) {
+			maxcntr = sn + MAXSPART;
+			srccnt = (CONTRIB *)realloc((char *)srccnt,
+					maxcntr*sizeof(CONTRIB));
+			cntord = (CNTPTR *)realloc((char *)cntord,
+					maxcntr*sizeof(CNTPTR));
+			if (srccnt == NULL || cntord == NULL)
+				error(SYSTEM, "out of memory in direct");
+		}
+		cntord[sn].sndx = sn;
 		cntord[sn].brt = 0.0;
-						/* get source ray */
-		if ((srccnt[sn].dom = srcray(&sr, r, sn)) == 0.0)
-			continue;
-		VCOPY(srccnt[sn].dir, sr.rdir);
 						/* compute coefficient */
-		(*f)(srccnt[sn].val, p, srccnt[sn].dir, srccnt[sn].dom);
-		cntord[sn].brt = bright(srccnt[sn].val);
+		(*f)(srccnt[sn].coef, p, srccnt[sn].dir, si.dom);
+		cntord[sn].brt = bright(srccnt[sn].coef);
 		if (cntord[sn].brt <= 0.0)
 			continue;
-						/* compute intersection */
-		if (!( source[sn].sflags & SDISTANT ?
-				sourcehit(&sr) :
-				(*ofun[source[sn].so->otype].funp)
-				(source[sn].so, &sr) )) {
-			sprintf(errmsg,
-				"aiming failure for light source \"%s\"",
-					source[sn].so->oname);
-			error(WARNING, errmsg);
-			continue;
-		}
-						/* compute contribution */
-		raycont(&sr);
-		multcolor(srccnt[sn].val, sr.rcol);
+		VCOPY(srccnt[sn].dir, sr.rdir);
+						/* compute potential */
+		sr.revf = srcvalue;
+		rayvalue(&sr);
+		copycolor(srccnt[sn].val, sr.rcol);
+		multcolor(srccnt[sn].val, srccnt[sn].coef);
 		cntord[sn].brt = bright(srccnt[sn].val);
 	}
 						/* sort contributions */
-	qsort(cntord, nsources, sizeof(CNTPTR), cntcmp);
+	qsort(cntord, sn, sizeof(CNTPTR), cntcmp);
 	{					/* find last */
 		register int  l, m;
 
-		sn = 0; ncnts = l = nsources;
+		ncnts = l = sn;
+		sn = 0;
 		while ((m = (sn + ncnts) >> 1) != l) {
 			if (cntord[m].brt > 0.0)
 				sn = m;
@@ -370,42 +264,46 @@ char  *p;			/* data for f */
                                                 /* accumulate tail */
         for (sn = ncnts-1; sn > 0; sn--)
                 cntord[sn-1].brt += cntord[sn].brt;
-						/* start with prob=.5 */
-	hit2 = 0.5; test2 = 1.0;
+						/* compute number to check */
+	nshadcheck = pow((double)ncnts, shadcert) + .5;
+						/* modify threshold */
+	ourthresh = shadthresh / r->rweight;
 						/* test for shadows */
+	nhits = 0;
 	for (sn = 0; sn < ncnts; sn++) {
 						/* check threshold */
 		if ((sn+nshadcheck>=ncnts ? cntord[sn].brt :
-				cntord[sn].brt-cntord[sn+nshadcheck].brt) <
-				ourthresh*bright(r->rcol))
+				cntord[sn].brt-cntord[sn+nshadcheck].brt)
+				< ourthresh*bright(r->rcol))
 			break;
-						/* get statistics */
-		hwt = (double)source[cntord[sn].sno].nhits /
-				(double)source[cntord[sn].sno].ntests;
-		test2 += hwt;
-		source[cntord[sn].sno].ntests++;
 						/* test for hit */
 		rayorigin(&sr, r, SHADOW, 1.0);
-		VCOPY(sr.rdir, srccnt[cntord[sn].sno].dir);
-		sr.rsrc = cntord[sn].sno;
+		VCOPY(sr.rdir, srccnt[cntord[sn].sndx].dir);
+		sr.rsrc = srccnt[cntord[sn].sndx].sno;
+		source[sr.rsrc].ntests++;	/* keep statistics */
 		if (localhit(&sr, &thescene) &&
-				sr.ro != source[cntord[sn].sno].so) {
-						/* check for transmission */
+				( sr.ro != source[sr.rsrc].so ||
+				source[sr.rsrc].sflags & SFOLLOW )) {
+						/* follow entire path */
 			raycont(&sr);
+			if (trace != NULL)
+				(*trace)(&sr);	/* trace execution */
 			if (bright(sr.rcol) <= FTINY)
 				continue;	/* missed! */
-			(*f)(srccnt[cntord[sn].sno].val, p,
-					srccnt[cntord[sn].sno].dir,
-					srccnt[cntord[sn].sno].dom);
-			multcolor(srccnt[cntord[sn].sno].val, sr.rcol);
+			copycolor(srccnt[cntord[sn].sndx].val, sr.rcol);
+			multcolor(srccnt[cntord[sn].sndx].val,
+					srccnt[cntord[sn].sndx].coef);
 		}
 						/* add contribution if hit */
-		addcolor(r->rcol, srccnt[cntord[sn].sno].val);
-		hit2 += hwt;
-		source[cntord[sn].sno].nhits++;
+		addcolor(r->rcol, srccnt[cntord[sn].sndx].val);
+		nhits++;
+		source[sr.rsrc].nhits++;
 	}
-					/* weighted hit rate */
-	hwt = hit2 / test2;
+					/* surface hit rate */
+	if (sn > 0)
+		hwt = (double)nhits / (double)sn;
+	else
+		hwt = 0.5;
 #ifdef DEBUG
 	sprintf(errmsg, "%d tested, %d untested, %f hit rate\n",
 			sn, ncnts-sn, hwt);
@@ -413,55 +311,10 @@ char  *p;			/* data for f */
 #endif
 					/* add in untested sources */
 	for ( ; sn < ncnts; sn++) {
-		prob = hwt * (double)source[cntord[sn].sno].nhits /
-				(double)source[cntord[sn].sno].ntests;
-		scalecolor(srccnt[cntord[sn].sno].val, prob);
-		addcolor(r->rcol, srccnt[cntord[sn].sno].val);
-	}
-}
-
-
-#define  wrongsource(m, r)	(m->otype!=MAT_ILLUM && \
-				r->rsrc>=0 && \
-				source[r->rsrc].so!=r->ro)
-
-#define  badambient(m, r)	((r->crtype&(AMBIENT|SHADOW))==AMBIENT && \
-				!(r->rtype&REFLECTED) && 	/* hack! */\
-				!(m->otype==MAT_GLOW&&r->rot>m->oargs.farg[3]))
-
-#define  passillum(m, r)	(m->otype==MAT_ILLUM && \
-				!(r->rsrc>=0&&source[r->rsrc].so==r->ro))
-
-
-m_light(m, r)			/* ray hit a light source */
-register OBJREC  *m;
-register RAY  *r;
-{
-						/* check for over-counting */
-	if (wrongsource(m, r) || badambient(m, r))
-		return;
-						/* check for passed illum */
-	if (passillum(m, r)) {
-
-		if (m->oargs.nsargs < 1 || !strcmp(m->oargs.sarg[0], VOIDID))
-			raytrans(r);
-		else
-			rayshade(r, modifier(m->oargs.sarg[0]));
-
-						/* otherwise treat as source */
-	} else {
-						/* check for behind */
-		if (r->rod < 0.0)
-			return;
-						/* get distribution pattern */
-		raytexture(r, m->omod);
-						/* get source color */
-		setcolor(r->rcol, m->oargs.farg[0],
-				  m->oargs.farg[1],
-				  m->oargs.farg[2]);
-						/* modify value */
-		multcolor(r->rcol, r->pcol);
-						/* assign distance */
-		r->rt = r->rot;
+		sr.rsrc = srccnt[cntord[sn].sndx].sno;
+		prob = hwt * (double)source[sr.rsrc].nhits /
+				(double)source[sr.rsrc].ntests;
+		scalecolor(srccnt[cntord[sn].sndx].val, prob);
+		addcolor(r->rcol, srccnt[cntord[sn].sndx].val);
 	}
 }