--- ray/src/rt/ambient.c 1992/04/10 16:14:34 2.4 +++ ray/src/rt/ambient.c 1992/07/16 14:03:27 2.8 @@ -41,12 +41,18 @@ 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; { @@ -68,19 +74,21 @@ int ar; setambient(afile) /* initialize calculation */ char *afile; { - long ftell(); AMBVAL amb; /* init ambient limits */ setambres(ambres); /* open ambient file */ - if (afile != NULL) + 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); @@ -88,6 +96,34 @@ char *afile; } +initambfile(creat) /* initialize ambient file */ +int creat; +{ + extern char *progname, *octname, VersionID[]; + + 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); + } else if (checkheader(ambfp, AMBFMT, NULL) < 0 + || !hasambmagic(ambfp)) { + sprintf(errmsg, "\"%s\" is not an ambient file", afname); + error(USER, errmsg); + } + ambheadlen = ftell(ambfp); +} + + ambnotify(obj) /* record new modifier */ OBJECT obj; { @@ -264,8 +300,7 @@ 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); } @@ -297,23 +332,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"); @@ -354,4 +387,43 @@ double s; return; memerr: error(SYSTEM, "out of memory in avinsert"); +} + + +#include + + +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); }