src/rt/ambient.c

/* Copyright (c) 1992 Regents of the University of California */

#ifndef lint
static char SCCSid[] = "$SunId$ LBL";
#endif

/*
 *  ambient.c - routines dealing with ambient (inter-reflected) component.
 *
 *     5/9/86
 */

#include  "ray.h"

#include  "octree.h"

#include  "otypes.h"

#include  "ambient.h"

#include  "random.h"

#define  OCTSCALE       0.5     /* ceil((valid rad.)/(cube size)) */

typedef struct ambtree {
        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 */

double  maxarad;                /* maximum ambient radius */
double  minarad;                /* minimum ambient radius */

static AMBTREE  atrunk;         /* our ambient trunk node */

static FILE  *ambfp = NULL;     /* ambient file pointer */

#define  AMBFLUSH       (BUFSIZ/AMBVALSIZ)

#define  newambval()    (AMBVAL *)bmalloc(sizeof(AMBVAL))

#define  newambtree()   (AMBTREE *)calloc(8, sizeof(AMBTREE))

extern long  ftell(), lseek();
static int  initambfile(), avsave(), avinsert(), ambsync();


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  headlen;
        AMBVAL  amb;
                                                /* init ambient limits */
        setambres(ambres);
                                                /* open ambient file */
        if (afile != NULL)
                if ((ambfp = fopen(afile, "r+")) != NULL) {
                        initambfile(0);
                        headlen = ftell(ambfp);
                        while (readambval(&amb, ambfp))
                                avinsert(&amb, &atrunk, thescene.cuorg,
                                                thescene.cusize);
                                                        /* align */
                        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);
                }
}


ambnotify(obj)                  /* record new modifier */
OBJECT  obj;
{
        static int  hitlimit = 0;
        register OBJREC  *o = objptr(obj);
        register char  **amblp;

        if (hitlimit || !ismodifier(o->otype))
                return;
        for (amblp = amblist; *amblp != NULL; amblp++)
                if (!strcmp(o->oname, *amblp)) {
                        if (ambset[0] >= MAXASET) {
                                error(WARNING, "too many modifiers in ambient list");
                                hitlimit++;
                                return;         /* should this be fatal? */
                        }
                        insertelem(ambset, obj);
                        return;
                }
}


ambient(acol, r)                /* compute ambient component for ray */
COLOR  acol;
register RAY  *r;
{
        static int  rdepth = 0;                 /* ambient recursion */
        double  d;

        if (ambdiv <= 0)                        /* no ambient calculation */
                goto dumbamb;
                                                /* check number of bounces */
        if (rdepth >= ambounce)
                goto dumbamb;
                                                /* check ambient list */
        if (ambincl != -1 && r->ro != NULL &&
                        ambincl != inset(ambset, r->ro->omod))
                goto dumbamb;

        if (ambacc <= FTINY) {                  /* no ambient storage */
                rdepth++;
                d = doambient(acol, r, r->rweight, NULL, NULL);
                rdepth--;
                if (d == 0.0)
                        goto dumbamb;
                return;
        }
                                                /* get ambient value */
        setcolor(acol, 0.0, 0.0, 0.0);
        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);
}


double
sumambient(acol, r, al, at, c0, s)      /* get interpolated ambient value */
COLOR  acol;
register RAY  *r;
int  al;
AMBTREE  *at;
FVECT  c0;
double  s;
{
        extern double  sqrt();
        double  d, e1, e2, wt, wsum;
        COLOR  ct;
        FVECT  ck0;
        int  i;
        register int  j;
        register AMBVAL  *av;
                                        /* do this node */
        wsum = 0.0;
        for (av = at->alist; av != NULL; av = av->next) {
                /*
                 *  Ambient level test.
                 */
                if (av->lvl > al || av->weight < r->rweight-FTINY)
                        continue;
                /*
                 *  Ambient radius test.
                 */
                e1 = 0.0;
                for (j = 0; j < 3; j++) {
                        d = av->pos[j] - r->rop[j];
                        e1 += d * d;
                }
                e1 /= av->rad * av->rad;
                if (e1 > ambacc*ambacc*1.21)
                        continue;
                /*
                 *  Normal direction test.
                 */
                e2 = (1.0 - DOT(av->dir, r->ron)) * r->rweight;
                if (e2 < 0.0) e2 = 0.0;
                if (e1 + e2 > ambacc*ambacc*1.21)
                        continue;
                /*
                 *  Ray behind test.
                 */
                d = 0.0;
                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-.001)
                        continue;
                /*
                 *  Jittering final test reduces image artifacts.
                 */
                wt = sqrt(e1) + sqrt(e2);
                wt *= .9 + .2*urand(9015+samplendx);
                if (wt > ambacc)
                        continue;
                if (wt <= 1e-3)
                        wt = 1e3;
                else
                        wt = 1.0 / wt;
                wsum += wt;
                extambient(ct, av, r->rop, r->ron);
                scalecolor(ct, wt);
                addcolor(acol, ct);
        }
        if (at->kid == NULL)
                return(wsum);
                                        /* do children */
        s *= 0.5;
        for (i = 0; i < 8; i++) {
                for (j = 0; j < 3; j++) {
                        ck0[j] = c0[j];
                        if (1<<j & i)
                                ck0[j] += s;
                        if (r->rop[j] < ck0[j] - OCTSCALE*s)
                                break;
                        if (r->rop[j] > ck0[j] + (1.0+OCTSCALE)*s)
                                break;
                }
                if (j == 3)
                        wsum += sumambient(acol, r, al, at->kid+i, ck0, s);
        }
        return(wsum);
}


double
makeambient(acol, r, al)        /* make a new ambient value */
COLOR  acol;
register RAY  *r;
int  al;
{
        AMBVAL  amb;
        FVECT   gp, gd;
                                                /* 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 = al;
        copycolor(amb.val, acol);
        VCOPY(amb.gpos, gp);
        VCOPY(amb.gdir, gd);
                                                /* insert into tree */
        avsave(&amb);                           /* and save to file */
        return(amb.rad);
}


extambient(cr, ap, pv, nv)              /* extrapolate value at pv, nv */
COLOR  cr;
register AMBVAL  *ap;
FVECT  pv, nv;
{
        FVECT  v1, v2;
        register int  i;
        double  d;

        d = 1.0;                        /* zeroeth order */
                                        /* gradient due to translation */
        for (i = 0; i < 3; i++)
                d += ap->gpos[i]*(pv[i]-ap->pos[i]);
                                        /* gradient due to rotation */
        VCOPY(v1, ap->dir);
        fcross(v2, v1, nv);
        d += DOT(ap->gdir, v2);
        if (d <= 0.0) {
                setcolor(cr, 0.0, 0.0, 0.0);
                return;
        }
        copycolor(cr, ap->val);
        scalecolor(cr, d);
}


static
initambfile(creat)              /* initialize ambient file */
int  creat;
{
        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);
                fflush(ambfp);
#ifndef  NIX
                sync();                 /* protect against NFS buffering */
#endif
        } 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;
{
        static int  nunflshed = 0;

        avinsert(av, &atrunk, thescene.cuorg, thescene.cusize);
        if (ambfp == NULL)
                return;
        if (writambval(av, ambfp) < 0)
                goto writerr;
        if (++nunflshed >= AMBFLUSH) {
                if (ambsync() == EOF)
                        goto writerr;
                nunflshed = 0;
        }
        return;
writerr:
        error(SYSTEM, "error writing ambient file");
}


static
avinsert(aval, at, c0, s)               /* insert ambient value in a tree */
AMBVAL  *aval;
register AMBTREE  *at;
FVECT  c0;
double  s;
{
        FVECT  ck0;
        int  branch;
        register AMBVAL  *av;
        register int  i;

        if ((av = newambval()) == NULL)
                goto memerr;
        copystruct(av, aval);
        VCOPY(ck0, c0);
        while (s*(OCTSCALE/2) > av->rad*ambacc) {
                if (at->kid == NULL)
                        if ((at->kid = newambtree()) == NULL)
                                goto memerr;
                s *= 0.5;
                branch = 0;
                for (i = 0; i < 3; i++)
                        if (av->pos[i] > ck0[i] + s) {
                                ck0[i] += s;
                                branch |= 1 << i;
                        }
                at = at->kid + branch;
        }
        av->next = at->alist;
        at->alist = av;
        return;
memerr:
        error(SYSTEM, "out of memory in avinsert");
}


#ifdef  NIX

static
ambsync()                       /* flush ambient file */
{
        return(fflush(ambfp));
}

#else

#include  <fcntl.h>
#include  <sys/types.h>
#include  <sys/stat.h>

static
ambsync()                       /* synchronize ambient file */
{
        static FILE  *ambinp = NULL;
        struct flock  fls;
        struct stat  sts;
        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, 1);  /* get previous position */
        if (fstat(fileno(ambfp), &sts) < 0)     /* get current length */
                error(SYSTEM, "cannot stat ambient file");
        flen = sts.st_size;
        if (n = (flen - lastpos)/AMBVALSIZ) {   /* file has grown */
                if (ambinp == NULL)             /* use duplicate file */
                        ambinp = fdopen(dup(fileno(ambfp)), "r");
                while (n--) {                   /* load contributed values */
                        readambval(&avs, ambinp);
                        avinsert(&avs,&atrunk,thescene.cuorg,thescene.cusize);
                }                               /* moves shared file pointer */
                if (n = (flen - lastpos)%AMBVALSIZ)     /* alignment */
                        lseek(fileno(ambfp), flen-n, 0);
        }
        n = fflush(ambfp);                      /* calls write() at last */
        fls.l_type = F_UNLCK;                   /* release file */
        fcntl(fileno(ambfp), F_SETLKW, &fls);
        return(n);
}

#endif
Revision:	2.12
Committed:	Mon Sep 21 12:07:31 1992 UTC (31 years, 7 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 2.11:	+33 -30 lines
Log Message:	Changes for PC port
#	User	Rev	Content
1	greg	2.12	/* Copyright (c) 1992 Regents of the University of California */
2	greg	1.1
3			#ifndef lint
4			static char SCCSid[] = "$SunId$ LBL";
5			#endif
6
7			/*
8			* ambient.c - routines dealing with ambient (inter-reflected) component.
9			*
10			* 5/9/86
11			*/
12
13			#include "ray.h"
14
15			#include "octree.h"
16
17	greg	1.8	#include "otypes.h"
18
19	greg	1.14	#include "ambient.h"
20
21	greg	1.1	#include "random.h"
22
23	greg	2.12	#define OCTSCALE 0.5 /* ceil((valid rad.)/(cube size)) */
24	greg	1.1
25	greg	1.14	typedef struct ambtree {
26	greg	2.12	AMBVAL alist; / ambient value list */
27			struct ambtree kid; / 8 child nodes */
28	greg	1.14	} AMBTREE; /* ambient octree */
29
30	greg	1.1	extern CUBE thescene; /* contains space boundaries */
31
32	greg	2.12	#define MAXASET 511 /* maximum number of elements in ambient set */
33			OBJECT ambset[MAXASET+1]={0}; /* ambient include/exclude set */
34	greg	1.1
35	greg	2.12	double maxarad; /* maximum ambient radius */
36			double minarad; /* minimum ambient radius */
37	greg	1.1
38	greg	2.12	static AMBTREE atrunk; /* our ambient trunk node */
39	greg	1.1
40			static FILE ambfp = NULL; / ambient file pointer */
41
42	greg	2.12	#define AMBFLUSH (BUFSIZ/AMBVALSIZ)
43	greg	2.6
44	greg	2.12	#define newambval() (AMBVAL *)bmalloc(sizeof(AMBVAL))
45	greg	1.1
46	greg	2.12	#define newambtree() (AMBTREE *)calloc(8, sizeof(AMBTREE))
47	greg	1.1
48	greg	2.7	extern long ftell(), lseek();
49	greg	2.10	static int initambfile(), avsave(), avinsert(), ambsync();
50	greg	1.1
51	greg	2.7
52	greg	2.3	setambres(ar) /* set ambient resolution */
53			int ar;
54			{
55			/* set min & max radii */
56			if (ar <= 0) {
57			minarad = 0.0;
58			maxarad = thescene.cusize / 2.0;
59			} else {
60			minarad = thescene.cusize / ar;
61			maxarad = 16.0 * minarad; /* heuristic */
62			if (maxarad > thescene.cusize / 2.0)
63			maxarad = thescene.cusize / 2.0;
64			}
65	greg	2.4	if (maxarad <= FTINY)
66			maxarad = .001;
67	greg	2.3	}
68
69
70	greg	1.1	setambient(afile) /* initialize calculation */
71			char *afile;
72			{
73	greg	2.9	long headlen;
74	greg	2.12	AMBVAL amb;
75	greg	2.3	/* init ambient limits */
76			setambres(ambres);
77			/* open ambient file */
78	greg	2.10	if (afile != NULL)
79	greg	1.1	if ((ambfp = fopen(afile, "r+")) != NULL) {
80	greg	2.6	initambfile(0);
81	greg	2.9	headlen = ftell(ambfp);
82	greg	2.5	while (readambval(&amb, ambfp))
83	greg	1.1	avinsert(&amb, &atrunk, thescene.cuorg,
84			thescene.cusize);
85			/* align */
86	greg	2.9	fseek(ambfp, -((ftell(ambfp)-headlen)%AMBVALSIZ), 1);
87	greg	2.10	} else if ((ambfp = fopen(afile, "w+")) != NULL)
88	greg	2.6	initambfile(1);
89			else {
90	greg	1.1	sprintf(errmsg, "cannot open ambient file \"%s\"",
91			afile);
92			error(SYSTEM, errmsg);
93	greg	1.8	}
94			}
95
96
97			ambnotify(obj) /* record new modifier */
98	greg	2.12	OBJECT obj;
99	greg	1.8	{
100	greg	1.11	static int hitlimit = 0;
101	greg	2.12	register OBJREC *o = objptr(obj);
102	greg	1.8	register char **amblp;
103
104	greg	1.11	if (hitlimit \|\| !ismodifier(o->otype))
105	greg	1.8	return;
106			for (amblp = amblist; *amblp != NULL; amblp++)
107			if (!strcmp(o->oname, *amblp)) {
108	greg	1.11	if (ambset[0] >= MAXASET) {
109			error(WARNING, "too many modifiers in ambient list");
110			hitlimit++;
111			return; /* should this be fatal? */
112			}
113	greg	1.8	insertelem(ambset, obj);
114			return;
115	greg	1.1	}
116			}
117
118
119			ambient(acol, r) /* compute ambient component for ray */
120			COLOR acol;
121			register RAY *r;
122			{
123			static int rdepth = 0; /* ambient recursion */
124	greg	2.12	double d;
125	greg	1.1
126			if (ambdiv <= 0) /* no ambient calculation */
127			goto dumbamb;
128			/* check number of bounces */
129	greg	1.16	if (rdepth >= ambounce)
130	greg	1.1	goto dumbamb;
131			/* check ambient list */
132			if (ambincl != -1 && r->ro != NULL &&
133			ambincl != inset(ambset, r->ro->omod))
134			goto dumbamb;
135
136			if (ambacc <= FTINY) { /* no ambient storage */
137	greg	1.16	rdepth++;
138			d = doambient(acol, r, r->rweight, NULL, NULL);
139			rdepth--;
140			if (d == 0.0)
141	greg	1.1	goto dumbamb;
142	greg	1.16	return;
143	greg	1.1	}
144			/* get ambient value */
145			setcolor(acol, 0.0, 0.0, 0.0);
146	greg	1.16	d = sumambient(acol, r, rdepth,
147			&atrunk, thescene.cuorg, thescene.cusize);
148			if (d > FTINY)
149			scalecolor(acol, 1.0/d);
150			else {
151			d = makeambient(acol, r, rdepth++);
152			rdepth--;
153			}
154			if (d > FTINY)
155			return;
156	greg	1.1	dumbamb: /* return global value */
157			copycolor(acol, ambval);
158			}
159
160
161			double
162	greg	1.16	sumambient(acol, r, al, at, c0, s) /* get interpolated ambient value */
163	greg	1.1	COLOR acol;
164			register RAY *r;
165	greg	1.16	int al;
166	greg	2.12	AMBTREE *at;
167	greg	1.1	FVECT c0;
168	greg	2.12	double s;
169	greg	1.1	{
170			extern double sqrt();
171	greg	2.12	double d, e1, e2, wt, wsum;
172	greg	1.1	COLOR ct;
173			FVECT ck0;
174			int i;
175			register int j;
176	greg	2.12	register AMBVAL *av;
177	greg	1.7	/* do this node */
178	greg	1.1	wsum = 0.0;
179			for (av = at->alist; av != NULL; av = av->next) {
180			/*
181	greg	1.16	* Ambient level test.
182	greg	1.1	*/
183	greg	1.16	if (av->lvl > al \|\| av->weight < r->rweight-FTINY)
184	greg	1.1	continue;
185			/*
186			* Ambient radius test.
187			*/
188			e1 = 0.0;
189			for (j = 0; j < 3; j++) {
190			d = av->pos[j] - r->rop[j];
191			e1 += d * d;
192			}
193			e1 /= av->rad * av->rad;
194			if (e1 > ambaccambacc1.21)
195			continue;
196			/*
197			* Normal direction test.
198			*/
199			e2 = (1.0 - DOT(av->dir, r->ron)) * r->rweight;
200			if (e2 < 0.0) e2 = 0.0;
201			if (e1 + e2 > ambaccambacc1.21)
202			continue;
203			/*
204			* Ray behind test.
205			*/
206			d = 0.0;
207			for (j = 0; j < 3; j++)
208			d += (r->rop[j] - av->pos[j]) *
209			(av->dir[j] + r->ron[j]);
210	greg	1.18	if (d0.5 < -minaradambacc-.001)
211	greg	1.1	continue;
212			/*
213			* Jittering final test reduces image artifacts.
214			*/
215			wt = sqrt(e1) + sqrt(e2);
216	greg	2.2	wt = .9 + .2urand(9015+samplendx);
217	greg	1.6	if (wt > ambacc)
218	greg	1.1	continue;
219			if (wt <= 1e-3)
220			wt = 1e3;
221			else
222			wt = 1.0 / wt;
223			wsum += wt;
224	greg	1.15	extambient(ct, av, r->rop, r->ron);
225	greg	1.1	scalecolor(ct, wt);
226			addcolor(acol, ct);
227	greg	1.7	}
228			if (at->kid == NULL)
229			return(wsum);
230			/* do children */
231			s *= 0.5;
232			for (i = 0; i < 8; i++) {
233			for (j = 0; j < 3; j++) {
234			ck0[j] = c0[j];
235			if (1<<j & i)
236			ck0[j] += s;
237			if (r->rop[j] < ck0[j] - OCTSCALE*s)
238			break;
239			if (r->rop[j] > ck0[j] + (1.0+OCTSCALE)*s)
240			break;
241			}
242			if (j == 3)
243	greg	1.16	wsum += sumambient(acol, r, al, at->kid+i, ck0, s);
244	greg	1.1	}
245			return(wsum);
246			}
247
248
249			double
250	greg	1.16	makeambient(acol, r, al) /* make a new ambient value */
251	greg	1.1	COLOR acol;
252			register RAY *r;
253	greg	1.16	int al;
254	greg	1.1	{
255	greg	2.12	AMBVAL amb;
256	greg	1.14	FVECT gp, gd;
257	greg	1.16	/* compute weight */
258			amb.weight = pow(AVGREFL, (double)al);
259	greg	1.17	if (r->rweight < 0.2amb.weight) / heuristic */
260	greg	1.16	amb.weight = r->rweight;
261			/* compute ambient */
262			amb.rad = doambient(acol, r, amb.weight, gp, gd);
263	greg	1.1	if (amb.rad == 0.0)
264			return(0.0);
265			/* store it */
266			VCOPY(amb.pos, r->rop);
267			VCOPY(amb.dir, r->ron);
268	greg	1.16	amb.lvl = al;
269	greg	1.1	copycolor(amb.val, acol);
270	greg	1.14	VCOPY(amb.gpos, gp);
271			VCOPY(amb.gdir, gd);
272	greg	1.1	/* insert into tree */
273	greg	2.7	avsave(&amb); /* and save to file */
274	greg	1.1	return(amb.rad);
275	greg	1.15	}
276
277
278			extambient(cr, ap, pv, nv) /* extrapolate value at pv, nv */
279			COLOR cr;
280	greg	2.12	register AMBVAL *ap;
281	greg	1.15	FVECT pv, nv;
282			{
283			FVECT v1, v2;
284			register int i;
285	greg	2.12	double d;
286	greg	1.15
287			d = 1.0; /* zeroeth order */
288			/* gradient due to translation */
289			for (i = 0; i < 3; i++)
290			d += ap->gpos[i]*(pv[i]-ap->pos[i]);
291			/* gradient due to rotation */
292			VCOPY(v1, ap->dir);
293			fcross(v2, v1, nv);
294			d += DOT(ap->gdir, v2);
295			if (d <= 0.0) {
296			setcolor(cr, 0.0, 0.0, 0.0);
297			return;
298			}
299			copycolor(cr, ap->val);
300			scalecolor(cr, d);
301	greg	1.1	}
302
303
304			static
305	greg	2.9	initambfile(creat) /* initialize ambient file */
306			int creat;
307			{
308			extern char progname, octname, VersionID[];
309
310	greg	2.12	#ifdef MSDOS
311			setmode(fileno(ambfp), O_BINARY);
312			#endif
313	greg	2.9	setbuf(ambfp, bmalloc(BUFSIZ));
314			if (creat) { /* new file */
315			fprintf(ambfp, "%s -av %g %g %g -ab %d -aa %g ",
316			progname, colval(ambval,RED),
317			colval(ambval,GRN), colval(ambval,BLU),
318			ambounce, ambacc);
319			fprintf(ambfp, "-ad %d -as %d -ar %d %s\n",
320			ambdiv, ambssamp, ambres,
321			octname==NULL ? "" : octname);
322			fprintf(ambfp, "SOFTWARE= %s\n", VersionID);
323			fputformat(AMBFMT, ambfp);
324			putc('\n', ambfp);
325			putambmagic(ambfp);
326			fflush(ambfp);
327	greg	2.12	#ifndef NIX
328	greg	2.11	sync(); /* protect against NFS buffering */
329			#endif
330	greg	2.10	} else if (checkheader(ambfp, AMBFMT, NULL) < 0 \|\| !hasambmagic(ambfp))
331			error(USER, "bad ambient file");
332	greg	2.9	}
333
334
335			static
336	greg	2.7	avsave(av) /* insert and save an ambient value */
337	greg	2.12	AMBVAL *av;
338	greg	1.1	{
339			static int nunflshed = 0;
340	greg	2.6
341	greg	2.7	avinsert(av, &atrunk, thescene.cuorg, thescene.cusize);
342	greg	1.1	if (ambfp == NULL)
343			return;
344	greg	2.5	if (writambval(av, ambfp) < 0)
345	greg	1.1	goto writerr;
346			if (++nunflshed >= AMBFLUSH) {
347	greg	2.7	if (ambsync() == EOF)
348	greg	1.1	goto writerr;
349			nunflshed = 0;
350			}
351			return;
352			writerr:
353			error(SYSTEM, "error writing ambient file");
354			}
355
356
357			static
358			avinsert(aval, at, c0, s) /* insert ambient value in a tree */
359	greg	2.12	AMBVAL *aval;
360	greg	1.1	register AMBTREE *at;
361			FVECT c0;
362	greg	2.12	double s;
363	greg	1.1	{
364			FVECT ck0;
365			int branch;
366	greg	2.12	register AMBVAL *av;
367	greg	1.1	register int i;
368
369			if ((av = newambval()) == NULL)
370			goto memerr;
371	greg	1.9	copystruct(av, aval);
372	greg	1.1	VCOPY(ck0, c0);
373			while (s(OCTSCALE/2) > av->radambacc) {
374			if (at->kid == NULL)
375			if ((at->kid = newambtree()) == NULL)
376			goto memerr;
377			s *= 0.5;
378			branch = 0;
379			for (i = 0; i < 3; i++)
380			if (av->pos[i] > ck0[i] + s) {
381			ck0[i] += s;
382			branch \|= 1 << i;
383			}
384			at = at->kid + branch;
385			}
386			av->next = at->alist;
387			at->alist = av;
388			return;
389			memerr:
390			error(SYSTEM, "out of memory in avinsert");
391	greg	2.7	}
392
393
394	greg	2.12	#ifdef NIX
395	greg	2.10
396			static
397			ambsync() /* flush ambient file */
398			{
399			return(fflush(ambfp));
400			}
401
402			#else
403
404	greg	2.7	#include <fcntl.h>
405	greg	2.10	#include <sys/types.h>
406			#include <sys/stat.h>
407	greg	2.7
408			static
409			ambsync() /* synchronize ambient file */
410			{
411			static FILE *ambinp = NULL;
412			struct flock fls;
413	greg	2.10	struct stat sts;
414	greg	2.12	AMBVAL avs;
415	greg	2.7	long lastpos, flen;
416			register int n;
417			/* gain exclusive access */
418			fls.l_type = F_WRLCK;
419			fls.l_whence = 0;
420			fls.l_start = 0L;
421			fls.l_len = 0L;
422			if (fcntl(fileno(ambfp), F_SETLKW, &fls) < 0)
423			error(SYSTEM, "cannot lock ambient file");
424			/* see if file has grown */
425	greg	2.9	lastpos = lseek(fileno(ambfp), 0L, 1); /* get previous position */
426	greg	2.10	if (fstat(fileno(ambfp), &sts) < 0) /* get current length */
427			error(SYSTEM, "cannot stat ambient file");
428			flen = sts.st_size;
429	greg	2.7	if (n = (flen - lastpos)/AMBVALSIZ) { /* file has grown */
430	greg	2.10	if (ambinp == NULL) /* use duplicate file */
431			ambinp = fdopen(dup(fileno(ambfp)), "r");
432	greg	2.7	while (n--) { /* load contributed values */
433			readambval(&avs, ambinp);
434			avinsert(&avs,&atrunk,thescene.cuorg,thescene.cusize);
435	greg	2.10	} /* moves shared file pointer */
436			if (n = (flen - lastpos)%AMBVALSIZ) /* alignment */
437			lseek(fileno(ambfp), flen-n, 0);
438	greg	2.7	}
439			n = fflush(ambfp); /* calls write() at last */
440			fls.l_type = F_UNLCK; /* release file */
441			fcntl(fileno(ambfp), F_SETLKW, &fls);
442			return(n);
443	greg	1.1	}
444	greg	2.10
445			#endif