--- ray/src/hd/rholo.c 1997/11/10 18:05:24 3.8 +++ ray/src/hd/rholo.c 1997/12/08 18:51:15 3.17 @@ -9,8 +9,11 @@ static char SCCSid[] = "$SunId$ SGI"; */ #include "rholo.h" +#include "random.h" #include "paths.h" +#include #include +#include /* the following must be consistent with rholo.h */ int NVARS = NRHVARS; /* total number of variables */ @@ -23,12 +26,12 @@ char froot[MAXPATH]; /* root file name */ int nowarn = 0; /* turn warnings off? */ -double expval = 1.; /* global exposure value */ - int ncprocs = 0; /* desired number of compute processes */ char *outdev = NULL; /* output device name */ +int readinp = 0; /* read commands from stdin */ + time_t starttime; /* time we got started */ time_t endtime; /* time we should end by */ time_t reporttime; /* time for next report */ @@ -38,11 +41,15 @@ long maxdisk; /* maximum file space (bytes) */ int rtargc = 1; /* rtrace command */ char *rtargv[128] = {"rtrace", NULL}; +int orig_mode = -1; /* original file mode (-1 if unchanged) */ + long nraysdone = 0L; /* number of rays done */ long npacksdone = 0L; /* number of packets done */ PACKET *freepacks; /* available packets */ +char *sigerr[NSIG]; /* signal error messages */ + extern time_t time(); @@ -55,6 +62,7 @@ char *argv[]; int force = 0; /* mark start time */ starttime = time(NULL); + initurand(10240); /* initialize urand */ progname = argv[0]; /* get arguments */ for (i = 1; i < argc && argv[i][0] == '-'; i++) switch (argv[i][1]) { @@ -64,6 +72,9 @@ char *argv[]; case 'f': /* force overwrite */ force++; break; + case 'i': /* read input from stdin */ + readinp++; + break; case 'n': /* compute processes */ if (i >= argc-2) goto userr; @@ -107,12 +118,17 @@ char *argv[]; creatholo(&hdg); } else { /* else load holodeck */ loadholo(); - if (vdef(RIF)) /* load RIF if any */ + /* check settings */ + checkvalues(); + /* load RIF if any */ + if (vdef(RIF)) getradfile(vval(RIF)); + /* set defaults */ + setdefaults(NULL); } /* initialize */ initrholo(); - /* run */ + /* main loop */ while (rholo()) ; /* done */ @@ -125,10 +141,59 @@ userr: } +onsig(signo) /* fatal signal */ +int signo; +{ + static int gotsig = 0; + + if (gotsig++) /* two signals and we're gone! */ + _exit(signo); + + alarm(30); /* allow 30 seconds to clean up */ + signal(SIGALRM, SIG_DFL); /* make certain we do die */ + eputs("signal - "); + eputs(sigerr[signo]); + eputs("\n"); + quit(3); +} + + +sigdie(signo, msg) /* set fatal signal */ +int signo; +char *msg; +{ + if (signal(signo, onsig) == SIG_IGN) + signal(signo, SIG_IGN); + sigerr[signo] = msg; +} + + +int +resfmode(fd, mod) /* restrict open file access mode */ +int fd, mod; +{ + struct stat stbuf; + /* get original mode */ + if (fstat(fd, &stbuf) < 0) + error(SYSTEM, "cannot stat open holodeck file"); + mod &= stbuf.st_mode; /* always more restrictive */ + if (mod == stbuf.st_mode) + return(-1); /* already set */ + /* else change it */ + if (fchmod(fd, mod) < 0) { + error(WARNING, "cannot change holodeck file access mode"); + return(-1); + } + return(stbuf.st_mode); /* return original mode */ +} + + initrholo() /* get our holodeck running */ { extern int global_packet(); register int i; + /* close holodeck on exec() */ + fcntl(hdlist[0]->fd, F_SETFD, FD_CLOEXEC); if (outdev != NULL) /* open output device */ disp_open(outdev); @@ -138,7 +203,7 @@ initrholo() /* get our holodeck running */ if (!vdef(DISKSPACE)) maxdisk = 0; else - maxdisk = 1024.*1024.*vflt(DISKSPACE)); + maxdisk = 1024.*1024.*vflt(DISKSPACE); /* record end time */ if (!vdef(TIME) || vflt(TIME) <= FTINY) endtime = 0; @@ -184,6 +249,18 @@ initrholo() /* get our holodeck running */ freepacks[i].next = &freepacks[i+1]; } } + /* set up signal handling */ + sigdie(SIGINT, "Interrupt"); + sigdie(SIGHUP, "Hangup"); + sigdie(SIGTERM, "Terminate"); + sigdie(SIGPIPE, "Broken pipe"); + sigdie(SIGALRM, "Alarm clock"); +#ifdef SIGXCPU + sigdie(SIGXCPU, "CPU limit exceeded"); + sigdie(SIGXFSZ, "File size exceeded"); +#endif + /* protect holodeck file */ + orig_mode = resfmode(hdlist[0]->fd, ncprocs>0 ? 0 : 0444); return; memerr: error(SYSTEM, "out of memory in initrholo"); @@ -205,7 +282,7 @@ rholo() /* holodeck main loop */ if (ncprocs <= 0) return(1); /* check file size */ - if (maxdisk > 0 && hdfiluse(hdlist[0]->fd,0)+hdmemuse(0) >= maxdisk) { + if (maxdisk > 0 && hdfilen(hdlist[0]->fd) >= maxdisk) { error(WARNING, "file limit exceeded"); return(0); } @@ -268,32 +345,6 @@ register HDGRID *gp; sprintf(errmsg, "ignoring all but first %s", vnam(SECTION)); error(WARNING, errmsg); } - if (sscanf(vval(SECTION), - "%lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf", - &gp->orig[0], &gp->orig[1], &gp->orig[2], - &gp->xv[0][0], &gp->xv[0][1], &gp->xv[0][2], - &gp->xv[1][0], &gp->xv[1][1], &gp->xv[1][2], - &gp->xv[2][0], &gp->xv[2][1], &gp->xv[2][2]) != 12) - badvalue(SECTION); - maxlen = 0.; - for (i = 0; i < 3; i++) - if ((len[i] = VLEN(gp->xv[i])) > maxlen) - maxlen = len[i]; - if (!vdef(GRID)) { - sprintf(buf, "%.4f", maxlen/8.); - vval(GRID) = savqstr(buf); - vdef(GRID)++; - } - if ((d = vflt(GRID)) <= FTINY) - badvalue(GRID); - for (i = 0; i < 3; i++) - gp->grid[i] = len[i]/d + (1.-FTINY); - if (!vdef(EXPOSURE)) { - sprintf(errmsg, "%s must be defined", vnam(EXPOSURE)); - error(USER, errmsg); - } - expval = vval(EXPOSURE)[0] == '-' || vval(EXPOSURE)[0] == '+' ? - pow(2., vflt(EXPOSURE)) : vflt(EXPOSURE); if (!vdef(OCTREE)) { if ((vval(OCTREE) = bmalloc(strlen(froot)+5)) == NULL) error(SYSTEM, "out of memory"); @@ -304,6 +355,10 @@ register HDGRID *gp; vval(OBSTRUCTIONS) = "T"; vdef(OBSTRUCTIONS)++; } + if (!vdef(VDIST)) { + vval(VDIST) = "F"; + vdef(VDIST)++; + } if (!vdef(OCCUPANCY)) { vval(OCCUPANCY) = "U"; vdef(OCCUPANCY)++; @@ -311,13 +366,42 @@ register HDGRID *gp; /* append rendering options */ if (vdef(RENDER)) rtargc += wordstring(rtargv+rtargc, vval(RENDER)); + + if (gp == NULL) /* already initialized? */ + return; + /* set grid parameters */ + gp->grid[0] = gp->grid[1] = gp->grid[2] = 0; + if (sscanf(vval(SECTION), + "%lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %hd %hd %hd", + &gp->orig[0], &gp->orig[1], &gp->orig[2], + &gp->xv[0][0], &gp->xv[0][1], &gp->xv[0][2], + &gp->xv[1][0], &gp->xv[1][1], &gp->xv[1][2], + &gp->xv[2][0], &gp->xv[2][1], &gp->xv[2][2], + &gp->grid[0], &gp->grid[1], &gp->grid[2]) < 12) + badvalue(SECTION); + maxlen = 0.; + for (i = 0; i < 3; i++) + if ((len[i] = VLEN(gp->xv[i])) > maxlen) + maxlen = len[i]; + if (!vdef(GRID)) { + sprintf(buf, "%.4f", maxlen/8.); + vval(GRID) = savqstr(buf); + vdef(GRID)++; + } + if ((d = vflt(GRID)) <= FTINY) + badvalue(GRID); + for (i = 0; i < 3; i++) + if (gp->grid[i] <= 0) + gp->grid[i] = len[i]/d + (1.-FTINY); } creatholo(gp) /* create a holodeck output file */ HDGRID *gp; { + extern char VersionID[]; long endloc = 0; + int fd; FILE *fp; /* open & truncate file */ if ((fp = fopen(hdkfile, "w+")) == NULL) { @@ -326,14 +410,15 @@ HDGRID *gp; } /* write information header */ newheader("RADIANCE", fp); + fprintf(fp, "SOFTWARE= %s\n", VersionID); printvars(fp); fputformat(HOLOFMT, fp); fputc('\n', fp); putw(HOLOMAGIC, fp); /* put magic number & terminus */ fwrite(&endloc, sizeof(long), 1, fp); - fflush(fp); /* flush buffer */ - hdinit(fileno(fp), gp); /* allocate and initialize index */ - /* we're dropping fp here.... */ + fd = dup(fileno(fp)); + fclose(fp); /* flush and close stdio stream */ + hdinit(fd, gp); /* allocate and initialize index */ } @@ -363,8 +448,10 @@ char *s; loadholo() /* start loading a holodeck from fname */ { + extern long ftell(); FILE *fp; - long endloc; + int fd; + long fpos; /* open holodeck file */ if ((fp = fopen(hdkfile, ncprocs>0 ? "r+" : "r")) == NULL) { sprintf(errmsg, "cannot open \"%s\" for %s", hdkfile, @@ -379,19 +466,21 @@ loadholo() /* start loading a holodeck from fname */ hdkfile); error(USER, errmsg); } - fread(&endloc, sizeof(long), 1, fp); - if (endloc != 0) + fread(&fpos, sizeof(long), 1, fp); + if (fpos != 0) error(WARNING, "ignoring multiple sections in holodeck file"); - fseek(fp, 0L, 1); /* align system file pointer */ - hdinit(fileno(fp), NULL); /* allocate and load index */ - /* we're dropping fp here.... */ + fpos = ftell(fp); /* get stdio position */ + fd = dup(fileno(fp)); + fclose(fp); /* done with stdio */ + lseek(fd, fpos, 0); /* align system file pointer */ + hdinit(fd, NULL); /* allocate and load index */ } done_packets(pl) /* handle finished packets */ PACKET *pl; { - static int nunflushed = 0; + static int n2flush = 0; register PACKET *p; while (pl != NULL) { @@ -402,8 +491,8 @@ PACKET *pl; p->nr*sizeof(RAYVAL)); if (outdev != NULL) /* display it */ disp_packet((PACKHEAD *)p); - else - nunflushed += p->nr; + if (hdcachesize <= 0) /* manual flushing */ + n2flush += p->nr; nraysdone += p->nr; npacksdone++; } @@ -411,9 +500,9 @@ PACKET *pl; p->next = freepacks; freepacks = p; } - if (nunflushed >= 256*RPACKSIZ) { + if (n2flush > 512*RPACKSIZ*ncprocs) { hdflush(NULL); /* flush holodeck buffers */ - nunflushed = 0; + n2flush = 0; } } @@ -421,7 +510,7 @@ PACKET *pl; getradfile(rfargs) /* run rad and get needed variables */ char *rfargs; { - static short mvar[] = {VIEW,OCTREE,EXPOSURE,-1}; + static short mvar[] = {OCTREE,-1}; static char tf1[] = TEMPLATE; char tf2[64]; char combuf[256]; @@ -520,8 +609,6 @@ int ec; { int status = 0; - if (outdev != NULL) /* close display */ - disp_close(); if (hdlist[0] != NULL) { /* flush holodeck */ if (ncprocs > 0) { done_packets(flush_queue()); @@ -530,7 +617,7 @@ int ec; if (vdef(REPORT)) { long fsiz, fuse; report(0); - fsiz = lseek(hdlist[0]->fd, 0L, 2); + fsiz = hdfilen(hdlist[0]->fd); fuse = hdfiluse(hdlist[0]->fd, 1); fprintf(stderr, "%s: %.1f Mbyte holodeck file, %.1f%% fragmentation\n", @@ -540,5 +627,9 @@ int ec; } else hdflush(NULL); } + if (orig_mode >= 0) /* reset holodeck access mode */ + fchmod(hdlist[0]->fd, orig_mode); + if (outdev != NULL) /* close display */ + disp_close(); exit(ec ? ec : status); /* exit */ }