--- ray/src/rt/ambient.c	1992/03/04 16:52:13	2.3
+++ ray/src/rt/ambient.c	1993/01/29 12:02:23	2.18
@@ -1,4 +1,4 @@
-/* Copyright (c) 1991 Regents of the University of California */
+/* Copyright (c) 1993 Regents of the University of California */
 
 #ifndef lint
 static char SCCSid[] = "$SunId$ LBL";
@@ -6,11 +6,6 @@ static char SCCSid[] = "$SunId$ LBL";
 
 /*
  *  ambient.c - routines dealing with ambient (inter-reflected) component.
- *
- *  The macro AMBFLUSH (if defined) is the number of ambient values
- *	to wait before flushing to the ambient file.
- *
- *     5/9/86
  */
 
 #include  "ray.h"
@@ -23,30 +18,36 @@ static char SCCSid[] = "$SunId$ LBL";
 
 #include  "random.h"
 
-#define  OCTSCALE	0.5	/* ceil((valid rad.)/(cube size)) */
+#define	 OCTSCALE	0.5	/* ceil((valid rad.)/(cube size)) */
 
 typedef struct ambtree {
-	AMBVAL  *alist;		/* ambient value list */
-	struct ambtree  *kid;	/* 8 child nodes */
+	AMBVAL	*alist;		/* ambient value list */
+	struct ambtree	*kid;	/* 8 child nodes */
 }  AMBTREE;			/* ambient octree */
 
 extern CUBE  thescene;		/* contains space boundaries */
 
-#define  MAXASET	511	/* maximum number of elements in ambient set */
-OBJECT  ambset[MAXASET+1]={0};	/* ambient include/exclude set */
+#define	 MAXASET	511	/* maximum number of elements in ambient set */
+OBJECT	ambset[MAXASET+1]={0};	/* ambient include/exclude set */
 
-double  maxarad;		/* maximum ambient radius */
-double  minarad;		/* minimum ambient radius */
+double	maxarad;		/* maximum ambient radius */
+double	minarad;		/* minimum ambient radius */
 
-static AMBTREE  atrunk;		/* our ambient trunk node */
+static AMBTREE	atrunk;		/* our ambient trunk node */
 
 static FILE  *ambfp = NULL;	/* ambient file pointer */
+static int  nunflshed = 0;	/* number of unflushed ambient values */
 
-#define  newambval()	(AMBVAL *)bmalloc(sizeof(AMBVAL))
+#define	 AMBFLUSH	(BUFSIZ/AMBVALSIZ)
 
-#define  newambtree()	(AMBTREE *)calloc(8, sizeof(AMBTREE))
+#define	 newambval()	(AMBVAL *)bmalloc(sizeof(AMBVAL))
 
+#define	 newambtree()	(AMBTREE *)calloc(8, sizeof(AMBTREE))
 
+extern long  ftell(), lseek();
+static int  initambfile(), avsave(), avinsert();
+
+
 setambres(ar)				/* set ambient resolution */
 int  ar;
 {
@@ -60,37 +61,46 @@ int  ar;
 		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;
+	long  headlen;
+	AMBVAL	amb;
 						/* init ambient limits */
 	setambres(ambres);
 						/* open ambient file */
-	if (afile != NULL)
+	if (afile != NULL) {
 		if ((ambfp = fopen(afile, "r+")) != NULL) {
-			while (fread((char *)&amb,sizeof(AMBVAL),1,ambfp) == 1)
+			initambfile(0);
+			headlen = ftell(ambfp);
+			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)-headlen)%AMBVALSIZ), 1);
+		} else if ((ambfp = fopen(afile, "w+")) != NULL)
+			initambfile(1);
+		else {
 			sprintf(errmsg, "cannot open ambient file \"%s\"",
 					afile);
 			error(SYSTEM, errmsg);
 		}
+		nunflshed++;	/* lie */
+		ambsync();
+	}
 }
 
 
 ambnotify(obj)			/* record new modifier */
-OBJECT  obj;
+OBJECT	obj;
 {
 	static int  hitlimit = 0;
-	register OBJREC  *o = objptr(obj);
+	register OBJREC	 *o = objptr(obj);
 	register char  **amblp;
 
 	if (hitlimit || !ismodifier(o->otype))
@@ -113,7 +123,7 @@ COLOR  acol;
 register RAY  *r;
 {
 	static int  rdepth = 0;			/* ambient recursion */
-	double  d;
+	double	d;
 
 	if (ambdiv <= 0)			/* no ambient calculation */
 		goto dumbamb;
@@ -155,17 +165,16 @@ sumambient(acol, r, al, at, c0, s)	/* get interpolated
 COLOR  acol;
 register RAY  *r;
 int  al;
-AMBTREE  *at;
+AMBTREE	 *at;
 FVECT  c0;
-double  s;
+double	s;
 {
-	extern double  sqrt();
-	double  d, e1, e2, wt, wsum;
+	double	d, e1, e2, wt, wsum;
 	COLOR  ct;
 	FVECT  ck0;
 	int  i;
 	register int  j;
-	register AMBVAL  *av;
+	register AMBVAL	 *av;
 					/* do this node */
 	wsum = 0.0;
 	for (av = at->alist; av != NULL; av = av->next) {
@@ -244,7 +253,7 @@ COLOR  acol;
 register RAY  *r;
 int  al;
 {
-	AMBVAL  amb;
+	AMBVAL	amb;
 	FVECT	gp, gd;
 						/* compute weight */
 	amb.weight = pow(AVGREFL, (double)al);
@@ -262,20 +271,19 @@ int  al;
 	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);
 }
 
 
 extambient(cr, ap, pv, nv)		/* extrapolate value at pv, nv */
 COLOR  cr;
-register AMBVAL  *ap;
+register AMBVAL	 *ap;
 FVECT  pv, nv;
 {
 	FVECT  v1, v2;
 	register int  i;
-	double  d;
+	double	d;
 
 	d = 1.0;			/* zeroeth order */
 					/* gradient due to translation */
@@ -295,23 +303,44 @@ FVECT  pv, nv;
 
 
 static
-avsave(av)				/* save an ambient value */
-AMBVAL  *av;
+initambfile(creat)		/* initialize ambient file */
+int  creat;
 {
-#ifdef  AMBFLUSH
-	static int  nunflshed = 0;
+	extern char  *progname, *octname, VersionID[];
+
+#ifdef MSDOS
+	setmode(fileno(ambfp), O_BINARY);
 #endif
+	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);
+		fprintf(ambfp, "SOFTWARE= %s\n", VersionID);
+		fputformat(AMBFMT, ambfp);
+		putc('\n', ambfp);
+		putambmagic(ambfp);
+	} else if (checkheader(ambfp, AMBFMT, NULL) < 0 || !hasambmagic(ambfp))
+		error(USER, "bad ambient file");
+}
+
+
+static
+avsave(av)				/* insert and save an ambient value */
+AMBVAL	*av;
+{
+	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 (++nunflshed >= AMBFLUSH)
+		if (ambsync() == EOF)
 			goto writerr;
-		nunflshed = 0;
-	}
-#endif
 	return;
 writerr:
 	error(SYSTEM, "error writing ambient file");
@@ -320,14 +349,14 @@ writerr:
 
 static
 avinsert(aval, at, c0, s)		/* insert ambient value in a tree */
-AMBVAL  *aval;
+AMBVAL	*aval;
 register AMBTREE  *at;
 FVECT  c0;
-double  s;
+double	s;
 {
 	FVECT  ck0;
 	int  branch;
-	register AMBVAL  *av;
+	register AMBVAL	 *av;
 	register int  i;
 
 	if ((av = newambval()) == NULL)
@@ -353,3 +382,68 @@ double  s;
 memerr:
 	error(SYSTEM, "out of memory in avinsert");
 }
+
+
+#ifdef	F_SETLKW
+
+int
+ambsync()			/* synchronize ambient file */
+{
+	static FILE  *ambinp = NULL;
+	static long  lastpos = -1;
+	struct flock  fls;
+	long  flen;
+	AMBVAL	avs;
+	register int  n;
+
+	if (nunflshed == 0)
+		return(0);
+				/* 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");
+	if (lastpos < 0)	/* initializing */
+		goto syncend;
+				/* see if file has grown */
+	if ((flen = lseek(fileno(ambfp), 0L, 2)) < 0)
+		error(SYSTEM, "cannot seek on ambient file");
+	if (n = flen - lastpos) {		/* file has grown */
+		if (ambinp == NULL) {		/* use duplicate filedes */
+			ambinp = fdopen(dup(fileno(ambfp)), "r");
+			if (ambinp == NULL)
+				error(SYSTEM, "fdopen failed in ambsync");
+		}
+		if (fseek(ambinp, lastpos, 0) < 0)
+			error(SYSTEM, "fseek failed in ambsync");
+		while (n >= AMBVALSIZ) {	/* load contributed values */
+			readambval(&avs, ambinp);
+			avinsert(&avs,&atrunk,thescene.cuorg,thescene.cusize);
+			n -= AMBVALSIZ;
+		}
+		if (n)				/* alignment */
+			lseek(fileno(ambfp), flen-n, 0);
+	}
+syncend:
+	n = fflush(ambfp);			/* calls write() at last */
+	lastpos = lseek(fileno(ambfp), 0L, 1);
+	fls.l_type = F_UNLCK;			/* release file */
+	fcntl(fileno(ambfp), F_SETLKW, &fls);
+	nunflshed = 0;
+	return(n);
+}
+
+#else
+
+int
+ambsync()			/* flush ambient file */
+{
+	if (nunflshed == 0)
+		return(0);
+	nunflshed = 0;
+	return(fflush(ambfp));
+}
+
+#endif