--- ray/src/rt/ambient.c	1991/06/13 09:41:15	1.15
+++ ray/src/rt/ambient.c	1992/07/16 13:37:12	2.7
@@ -41,31 +41,54 @@ double  minarad;		/* minimum ambient radius */
 static AMBTREE  atrunk;		/* our ambient trunk node */
 
 static FILE  *ambfp = NULL;	/* ambient file pointer */
+static char  *afname;		/* ambient file name */
+static long  ambheadlen;	/* length of ambient file header */
 
+#define  AMBFLUSH	(BUFSIZ/AMBVALSIZ)
+
 #define  newambval()	(AMBVAL *)bmalloc(sizeof(AMBVAL))
 
 #define  newambtree()	(AMBTREE *)calloc(8, sizeof(AMBTREE))
 
+extern long  ftell(), lseek();
 
+
+setambres(ar)				/* set ambient resolution */
+int  ar;
+{
+						/* set min & max radii */
+	if (ar <= 0) {
+		minarad = 0.0;
+		maxarad = thescene.cusize / 2.0;
+	} else {
+		minarad = thescene.cusize / ar;
+		maxarad = 16.0 * minarad;		/* heuristic */
+		if (maxarad > thescene.cusize / 2.0)
+			maxarad = thescene.cusize / 2.0;
+	}
+	if (maxarad <= FTINY)
+		maxarad = .001;
+}
+
+
 setambient(afile)			/* initialize calculation */
 char  *afile;
 {
-	long  ftell();
 	AMBVAL  amb;
-
-	maxarad = thescene.cusize / 2.0;		/* maximum radius */
-							/* minimum radius */
-	minarad = ambres > 0 ? thescene.cusize/ambres : 0.0;
-
-					/* open ambient file */
-	if (afile != NULL)
+						/* init ambient limits */
+	setambres(ambres);
+						/* open ambient file */
+	if ((afname = afile) != NULL)
 		if ((ambfp = fopen(afile, "r+")) != NULL) {
-			while (fread((char *)&amb,sizeof(AMBVAL),1,ambfp) == 1)
+			initambfile(0);
+			while (readambval(&amb, ambfp))
 				avinsert(&amb, &atrunk, thescene.cuorg,
 						thescene.cusize);
 							/* align */
-			fseek(ambfp, -(ftell(ambfp)%sizeof(AMBVAL)), 1);
-		} else if ((ambfp = fopen(afile, "w")) == NULL) {
+			fseek(ambfp, -((ftell(ambfp)-ambheadlen)%AMBVALSIZ), 1);
+		} else if ((ambfp = fopen(afile, "w")) != NULL)
+			initambfile(1);
+		else {
 			sprintf(errmsg, "cannot open ambient file \"%s\"",
 					afile);
 			error(SYSTEM, errmsg);
@@ -73,6 +96,33 @@ char  *afile;
 }
 
 
+initambfile(creat)		/* initialize ambient file */
+int  creat;
+{
+	extern char  *progname, *octname;
+
+	setbuf(ambfp, bmalloc(BUFSIZ));
+	if (creat) {			/* new file */
+		fprintf(ambfp, "%s -av %g %g %g -ab %d -aa %g ",
+				progname, colval(ambval,RED),
+				colval(ambval,GRN), colval(ambval,BLU),
+				ambounce, ambacc);
+		fprintf(ambfp, "-ad %d -as %d -ar %d %s\n",
+				ambdiv, ambssamp, ambres,
+				octname==NULL ? "" : octname);
+		fputformat(AMBFMT, ambfp);
+		putc('\n', ambfp);
+		putambmagic(ambfp);
+		fflush(ambfp);
+	} else if (checkheader(ambfp, AMBFMT, NULL) < 0
+			|| !hasambmagic(ambfp)) {
+		sprintf(errmsg, "\"%s\" is not an ambient file", afname);
+		error(USER, afname);
+	}
+	ambheadlen = ftell(ambfp);
+}
+
+
 ambnotify(obj)			/* record new modifier */
 OBJECT  obj;
 {
@@ -100,14 +150,12 @@ COLOR  acol;
 register RAY  *r;
 {
 	static int  rdepth = 0;			/* ambient recursion */
-	double  wsum;
+	double  d;
 
-	rdepth++;				/* increment level */
-
 	if (ambdiv <= 0)			/* no ambient calculation */
 		goto dumbamb;
 						/* check number of bounces */
-	if (rdepth > ambounce)
+	if (rdepth >= ambounce)
 		goto dumbamb;
 						/* check ambient list */
 	if (ambincl != -1 && r->ro != NULL &&
@@ -115,30 +163,35 @@ register RAY  *r;
 		goto dumbamb;
 
 	if (ambacc <= FTINY) {			/* no ambient storage */
-		if (doambient(acol, r, NULL, NULL) == 0.0)
+		rdepth++;
+		d = doambient(acol, r, r->rweight, NULL, NULL);
+		rdepth--;
+		if (d == 0.0)
 			goto dumbamb;
-		goto done;
+		return;
 	}
 						/* get ambient value */
 	setcolor(acol, 0.0, 0.0, 0.0);
-	wsum = sumambient(acol, r, &atrunk, thescene.cuorg, thescene.cusize);
-	if (wsum > FTINY)
-		scalecolor(acol, 1.0/wsum);
-	else if (makeambient(acol, r) == 0.0)
-		goto dumbamb;
-	goto done;
-
+	d = sumambient(acol, r, rdepth,
+			&atrunk, thescene.cuorg, thescene.cusize);
+	if (d > FTINY)
+		scalecolor(acol, 1.0/d);
+	else {
+		d = makeambient(acol, r, rdepth++);
+		rdepth--;
+	}
+	if (d > FTINY)
+		return;
 dumbamb:					/* return global value */
 	copycolor(acol, ambval);
-done:						/* must finish here! */
-	rdepth--;
 }
 
 
 double
-sumambient(acol, r, at, c0, s)		/* get interpolated ambient value */
+sumambient(acol, r, al, at, c0, s)	/* get interpolated ambient value */
 COLOR  acol;
 register RAY  *r;
+int  al;
 AMBTREE  *at;
 FVECT  c0;
 double  s;
@@ -154,9 +207,9 @@ double  s;
 	wsum = 0.0;
 	for (av = at->alist; av != NULL; av = av->next) {
 		/*
-		 *  Ray strength test.
+		 *  Ambient level test.
 		 */
-		if (av->lvl > r->rlvl || av->weight < r->rweight-FTINY)
+		if (av->lvl > al || av->weight < r->rweight-FTINY)
 			continue;
 		/*
 		 *  Ambient radius test.
@@ -183,13 +236,13 @@ double  s;
 		for (j = 0; j < 3; j++)
 			d += (r->rop[j] - av->pos[j]) *
 					(av->dir[j] + r->ron[j]);
-		if (d*0.5 < -minarad*ambacc)
+		if (d*0.5 < -minarad*ambacc-.001)
 			continue;
 		/*
 		 *  Jittering final test reduces image artifacts.
 		 */
 		wt = sqrt(e1) + sqrt(e2);
-		wt *= .9 + .2*frandom();
+		wt *= .9 + .2*urand(9015+samplendx);
 		if (wt > ambacc)
 			continue;
 		if (wt <= 1e-3)
@@ -216,34 +269,37 @@ double  s;
 				break;
 		}
 		if (j == 3)
-			wsum += sumambient(acol, r, at->kid+i, ck0, s);
+			wsum += sumambient(acol, r, al, at->kid+i, ck0, s);
 	}
 	return(wsum);
 }
 
 
 double
-makeambient(acol, r)		/* make a new ambient value */
+makeambient(acol, r, al)	/* make a new ambient value */
 COLOR  acol;
 register RAY  *r;
+int  al;
 {
 	AMBVAL  amb;
 	FVECT	gp, gd;
-
-	amb.rad = doambient(acol, r, gp, gd);	/* compute ambient */
+						/* compute weight */
+	amb.weight = pow(AVGREFL, (double)al);
+	if (r->rweight < 0.2*amb.weight)	/* heuristic */
+		amb.weight = r->rweight;
+						/* compute ambient */
+	amb.rad = doambient(acol, r, amb.weight, gp, gd);
 	if (amb.rad == 0.0)
 		return(0.0);
 						/* store it */
 	VCOPY(amb.pos, r->rop);
 	VCOPY(amb.dir, r->ron);
-	amb.lvl = r->rlvl;
-	amb.weight = r->rweight;
+	amb.lvl = al;
 	copycolor(amb.val, acol);
 	VCOPY(amb.gpos, gp);
 	VCOPY(amb.gdir, gd);
 						/* insert into tree */
-	avinsert(&amb, &atrunk, thescene.cuorg, thescene.cusize);
-	avsave(&amb);				/* write to file */
+	avsave(&amb);				/* and save to file */
 	return(amb.rad);
 }
 
@@ -275,23 +331,21 @@ FVECT  pv, nv;
 
 
 static
-avsave(av)				/* save an ambient value */
+avsave(av)				/* insert and save an ambient value */
 AMBVAL  *av;
 {
-#ifdef  AMBFLUSH
 	static int  nunflshed = 0;
-#endif
+
+	avinsert(av, &atrunk, thescene.cuorg, thescene.cusize);
 	if (ambfp == NULL)
 		return;
-	if (fwrite((char *)av, sizeof(AMBVAL), 1, ambfp) != 1)
+	if (writambval(av, ambfp) < 0)
 		goto writerr;
-#ifdef  AMBFLUSH
 	if (++nunflshed >= AMBFLUSH) {
-		if (fflush(ambfp) == EOF)
+		if (ambsync() == EOF)
 			goto writerr;
 		nunflshed = 0;
 	}
-#endif
 	return;
 writerr:
 	error(SYSTEM, "error writing ambient file");
@@ -332,4 +386,43 @@ double  s;
 	return;
 memerr:
 	error(SYSTEM, "out of memory in avinsert");
+}
+
+
+#include  <fcntl.h>
+
+
+static
+ambsync()			/* synchronize ambient file */
+{
+	static FILE  *ambinp = NULL;
+	struct flock  fls;
+	AMBVAL  avs;
+	long  lastpos, flen;
+	register int  n;
+				/* gain exclusive access */
+	fls.l_type = F_WRLCK;
+	fls.l_whence = 0;
+	fls.l_start = 0L;
+	fls.l_len = 0L;
+	if (fcntl(fileno(ambfp), F_SETLKW, &fls) < 0)
+		error(SYSTEM, "cannot lock ambient file");
+				/* see if file has grown */
+	lastpos = lseek(fileno(ambfp), 0L, 2);	/* may move pointer */
+	flen = lseek(fileno(ambfp), 0L, 1);	/* new(?) file length */
+	if (n = (flen - lastpos)%AMBVALSIZ)	/* assure alignment */
+		lseek(fileno(ambfp), flen -= n, 0);
+	if (n = (flen - lastpos)/AMBVALSIZ) {	/* file has grown */
+		if (ambinp == NULL && (ambinp = fopen(afname, "r")) == NULL)
+			error(SYSTEM, "cannot reopen ambient file");
+		fseek(ambinp, lastpos, 0);	/* go to previous position */
+		while (n--) {			/* load contributed values */
+			readambval(&avs, ambinp);
+			avinsert(&avs,&atrunk,thescene.cuorg,thescene.cusize);
+		}
+	}
+	n = fflush(ambfp);			/* calls write() at last */
+	fls.l_type = F_UNLCK;			/* release file */
+	fcntl(fileno(ambfp), F_SETLKW, &fls);
+	return(n);
 }