--- ray/src/rt/ambient.c 1991/08/20 10:05:18 1.16 +++ ray/src/rt/ambient.c 1993/08/04 14:22:11 2.19 @@ -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,61 +18,91 @@ 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)) -setambient(afile) /* initialize calculation */ -char *afile; +extern long ftell(), lseek(); +static int initambfile(), avsave(), avinsert(); +#ifdef F_SETLKW +static aflock(); +#endif + + +setambres(ar) /* set ambient resolution */ +int ar; { - long ftell(); - AMBVAL amb; + /* 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; +} - maxarad = thescene.cusize / 2.0; /* maximum radius */ - /* minimum radius */ - minarad = ambres > 0 ? thescene.cusize/ambres : 0.0; - /* open ambient file */ - if (afile != NULL) - if ((ambfp = fopen(afile, "r+")) != NULL) { - while (fread((char *)&amb,sizeof(AMBVAL),1,ambfp) == 1) - avinsert(&amb, &atrunk, thescene.cuorg, - thescene.cusize); - /* align */ - fseek(ambfp, -(ftell(ambfp)%sizeof(AMBVAL)), 1); - } else if ((ambfp = fopen(afile, "w")) == NULL) { - sprintf(errmsg, "cannot open ambient file \"%s\"", - afile); - error(SYSTEM, errmsg); - } +setambient(afile) /* initialize calculation */ +char *afile; +{ + long headlen; + AMBVAL amb; + /* init ambient limits */ + setambres(ambres); + if (afile == NULL) + return; + /* open ambient file */ + 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); + } + 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)) @@ -100,7 +125,7 @@ COLOR acol; register RAY *r; { static int rdepth = 0; /* ambient recursion */ - double d; + double d; if (ambdiv <= 0) /* no ambient calculation */ goto dumbamb; @@ -142,17 +167,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) { @@ -186,13 +210,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) @@ -231,11 +255,11 @@ COLOR acol; register RAY *r; int al; { - AMBVAL amb; + AMBVAL amb; FVECT gp, gd; /* compute weight */ amb.weight = pow(AVGREFL, (double)al); - if (r->rweight < 0.1*amb.weight) /* heuristic */ + if (r->rweight < 0.2*amb.weight) /* heuristic */ amb.weight = r->rweight; /* compute ambient */ amb.rad = doambient(acol, r, amb.weight, gp, gd); @@ -249,20 +273,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 */ @@ -282,23 +305,47 @@ 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 F_SETLKW + aflock(creat ? F_WRLCK : F_RDLCK); #endif +#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"); @@ -307,14 +354,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) @@ -340,3 +387,73 @@ double s; memerr: error(SYSTEM, "out of memory in avinsert"); } + + +#ifdef F_SETLKW + +static +aflock(typ) /* lock/unlock ambient file */ +int typ; +{ + static struct flock fls; /* static so initialized to zeroes */ + + fls.l_type = typ; + if (fcntl(fileno(ambfp), F_SETLKW, &fls) < 0) + error(SYSTEM, "cannot (un)lock ambient file"); +} + + +int +ambsync() /* synchronize ambient file */ +{ + static FILE *ambinp = NULL; + static long lastpos = -1; + long flen; + AMBVAL avs; + register int n; + + if (nunflshed == 0) + return(0); + if (lastpos < 0) /* initializing (locked in initambfile) */ + goto syncend; + /* gain exclusive access */ + aflock(F_WRLCK); + /* 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); + aflock(F_UNLCK); /* release file */ + nunflshed = 0; + return(n); +} + +#else + +int +ambsync() /* flush ambient file */ +{ + if (nunflshed == 0) + return(0); + nunflshed = 0; + return(fflush(ambfp)); +} + +#endif