ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/hd/rholo.c
(Generate patch)

Comparing ray/src/hd/rholo.c (file contents):
Revision 3.3 by gregl, Fri Oct 31 15:49:23 1997 UTC vs.
Revision 3.22 by gregl, Fri Dec 12 18:23:08 1997 UTC

# Line 9 | Line 9 | static char SCCSid[] = "$SunId$ SGI";
9   */
10  
11   #include "rholo.h"
12 < #include "paths.h"
12 > #include "random.h"
13 > #include <signal.h>
14   #include <sys/types.h>
15 + #include <sys/stat.h>
16  
17                          /* the following must be consistent with rholo.h */
18   int     NVARS = NRHVARS;                /* total number of variables */
# Line 19 | Line 21 | VARIABLE       vv[] = RHVINIT;         /* variable-value pairs */
21  
22   char    *progname;              /* our program name */
23   char    *hdkfile;               /* holodeck file name */
24 < char    froot[MAXPATH];         /* root file name */
24 > char    froot[256];             /* root file name */
25  
26   int     nowarn = 0;             /* turn warnings off? */
27  
26 – double  expval = 1.;            /* global exposure value */
27 –
28   int     ncprocs = 0;            /* desired number of compute processes */
29  
30   char    *outdev = NULL;         /* output device name */
31  
32 + int     readinp = 0;            /* read commands from stdin */
33 +
34 + int     force = 0;              /* allow overwrite of holodeck */
35 +
36   time_t  starttime;              /* time we got started */
37   time_t  endtime;                /* time we should end by */
38   time_t  reporttime;             /* time for next report */
39  
40 + long    maxdisk;                /* maximum file space (bytes) */
41 +
42   int     rtargc = 1;             /* rtrace command */
43   char    *rtargv[128] = {"rtrace", NULL};
44  
45 + int     orig_mode = -1;         /* original file mode (-1 if unchanged) */
46 +
47   long    nraysdone = 0L;         /* number of rays done */
48   long    npacksdone = 0L;        /* number of packets done */
49  
50   PACKET  *freepacks;             /* available packets */
51  
52 + char  *sigerr[NSIG];            /* signal error messages */
53 +
54   extern time_t   time();
55  
56  
# Line 50 | Line 60 | char   *argv[];
60   {
61          HDGRID  hdg;
62          int     i;
53 –        int     force = 0;
63                                                  /* mark start time */
64          starttime = time(NULL);
65 +        initurand(16384);                       /* initialize urand */
66          progname = argv[0];                     /* get arguments */
67          for (i = 1; i < argc && argv[i][0] == '-'; i++)
68                  switch (argv[i][1]) {
# Line 62 | Line 72 | char   *argv[];
72                  case 'f':                       /* force overwrite */
73                          force++;
74                          break;
75 +                case 'i':                       /* read input from stdin */
76 +                        readinp++;
77 +                        break;
78                  case 'n':                       /* compute processes */
79                          if (i >= argc-2)
80                                  goto userr;
# Line 93 | Line 106 | char   *argv[];
106                                                          /* check settings */
107                  checkvalues();
108                                                          /* load RIF if any */
109 <                if (vdef(RIF))
97 <                        getradfile(vval(RIF));
109 >                getradfile();
110                                                          /* set defaults */
111                  setdefaults(&hdg);
112                                                          /* holodeck exists? */
# Line 105 | Line 117 | char   *argv[];
117                  creatholo(&hdg);
118          } else {                                /* else load holodeck */
119                  loadholo();
120 <                if (vdef(RIF))                          /* load RIF if any */
121 <                        getradfile(vval(RIF));
120 >                                                        /* check settings */
121 >                checkvalues();
122 >                                                        /* load RIF if any */
123 >                getradfile();
124 >                                                        /* set defaults */
125 >                setdefaults(NULL);
126          }
127                                                  /* initialize */
128          initrholo();
129 <                                                /* run */
129 >                                                /* main loop */
130          while (rholo())
131                  ;
132                                                  /* done */
# Line 123 | Line 139 | userr:
139   }
140  
141  
142 + onsig(signo)                            /* fatal signal */
143 + int  signo;
144 + {
145 +        static int  gotsig = 0;
146 +
147 +        if (gotsig++)                   /* two signals and we're gone! */
148 +                _exit(signo);
149 +
150 +        alarm(30);                      /* allow 30 seconds to clean up */
151 +        signal(SIGALRM, SIG_DFL);       /* make certain we do die */
152 +        eputs("signal - ");
153 +        eputs(sigerr[signo]);
154 +        eputs("\n");
155 +        quit(3);
156 + }
157 +
158 +
159 + sigdie(signo, msg)                      /* set fatal signal */
160 + int  signo;
161 + char  *msg;
162 + {
163 +        if (signal(signo, onsig) == SIG_IGN)
164 +                signal(signo, SIG_IGN);
165 +        sigerr[signo] = msg;
166 + }
167 +
168 +
169 + int
170 + resfmode(fd, mod)               /* restrict open file access mode */
171 + int     fd, mod;
172 + {
173 +        struct stat     stbuf;
174 +                                        /* get original mode */
175 +        if (fstat(fd, &stbuf) < 0)
176 +                error(SYSTEM, "cannot stat open holodeck file");
177 +        mod &= stbuf.st_mode;           /* always more restrictive */
178 +        if (mod == stbuf.st_mode)
179 +                return(-1);             /* already set */
180 +                                        /* else change it */
181 +        if (fchmod(fd, mod) < 0) {
182 +                error(WARNING, "cannot change holodeck file access mode");
183 +                return(-1);
184 +        }
185 +        return(stbuf.st_mode);          /* return original mode */
186 + }
187 +
188 +
189   initrholo()                     /* get our holodeck running */
190   {
191          extern int      global_packet();
192          register int    i;
193 <                                                /* check output device */
194 <        if (outdev != NULL)
195 <                open_display(outdev);
193 >                                                /* close holodeck on exec() */
194 >        fcntl(hdlist[0]->fd, F_SETFD, FD_CLOEXEC);
195 >
196 >        if (outdev != NULL)                     /* open output device */
197 >                disp_open(outdev);
198          else if (ncprocs > 0)                   /* else use global ray feed */
199                  init_global();
200 +                                                /* record disk space limit */
201 +        if (!vdef(DISKSPACE))
202 +                maxdisk = 0;
203 +        else
204 +                maxdisk = 1024.*1024.*vflt(DISKSPACE);
205                                                  /* record end time */
206          if (!vdef(TIME) || vflt(TIME) <= FTINY)
207                  endtime = 0;
208          else
209                  endtime = starttime + vflt(TIME)*3600.;
210                                                  /* set up memory cache */
211 <        hdcachesize = 1024.*1024.*vflt(CACHE);
211 >        if (outdev == NULL)
212 >                hdcachesize = 0;        /* manual flushing */
213 >        else if (vdef(CACHE))
214 >                hdcachesize = 1024.*1024.*vflt(CACHE);
215                                                  /* open report file */
216          if (vdef(REPORT)) {
217                  register char   *s = sskip2(vval(REPORT), 1);
# Line 149 | Line 222 | initrholo()                    /* get our holodeck running */
222          if (ncprocs > 0) {
223                  i = start_rtrace();
224                  if (i < 1)
225 <                        error(USER, "cannot start rtrace process");
225 >                        error(USER, "cannot start rtrace process(es)");
226                  if (vdef(REPORT)) {             /* make first report */
227                          printargs(rtargc, rtargv, stderr);
228                          report(0);
# Line 174 | Line 247 | initrholo()                    /* get our holodeck running */
247                          freepacks[i].next = &freepacks[i+1];
248                  }
249          }
250 +                                        /* set up signal handling */
251 +        sigdie(SIGINT, "Interrupt");
252 +        sigdie(SIGHUP, "Hangup");
253 +        sigdie(SIGTERM, "Terminate");
254 +        sigdie(SIGPIPE, "Broken pipe");
255 +        sigdie(SIGALRM, "Alarm clock");
256 + #ifdef  SIGXCPU
257 +        sigdie(SIGXCPU, "CPU limit exceeded");
258 +        sigdie(SIGXFSZ, "File size exceeded");
259 + #endif
260 +                                                /* protect holodeck file */
261 +        orig_mode = resfmode(hdlist[0]->fd, ncprocs>0 ? 0 : 0444);
262          return;
263   memerr:
264          error(SYSTEM, "out of memory in initrholo");
# Line 187 | Line 272 | rholo()                                /* holodeck main loop */
272          register PACKET *p;
273          time_t  t;
274          long    l;
275 <                                        /* check display */
276 <        if (outdev != NULL && !disp_check(idle))
277 <                return(0);
275 >
276 >        if (outdev != NULL)             /* check display */
277 >                if (!disp_check(idle))
278 >                        return(0);
279                                          /* display only? */
280 <        if (ncprocs <= 0)
281 <                return(1);
280 >        if (nprocs <= 0)
281 >                return(outdev != NULL);
282                                          /* check file size */
283 <        if ((l = 1024.*1024.*vflt(DISKSPACE)) > 0 &&
284 <                        hdfiluse(hdlist[0]->fd, 0) + hdmemuse(0) >= l)
285 <                return(0);
283 >        if (maxdisk > 0 && hdfilen(hdlist[0]->fd) >= maxdisk) {
284 >                error(WARNING, "file limit exceeded");
285 >                done_rtrace();
286 >                idle = 1;
287 >                return(1);      /* comes back */
288 >        }
289                                          /* check time */
290          if (endtime > 0 || reporttime > 0)
291                  t = time(NULL);
292 <        if (endtime > 0 && t >= endtime)
293 <                return(0);
292 >        if (endtime > 0 && t >= endtime) {
293 >                error(WARNING, "time limit exceeded");
294 >                done_rtrace();
295 >                idle = 1;
296 >                return(1);      /* comes back */
297 >        }
298          if (reporttime > 0 && t >= reporttime)
299                  report(t);
300                                          /* get packets to process */
# Line 209 | Line 302 | rholo()                                /* holodeck main loop */
302                  p = freepacks; freepacks = p->next; p->next = NULL;
303                  if (!next_packet(p)) {
304                          p->next = freepacks; freepacks = p;
305 +                        idle = 1;
306                          break;
307                  }
308                  if (pl == NULL) pl = p;
309                  else plend->next = p;
310                  plend = p;
311          }
312 <        idle = pl == NULL && freepacks != NULL;
219 <                                        /* are we out of stuff to do? */
220 <        if (idle && outdev == NULL)
221 <                return(0);
222 <                                        /* else process packets */
312 >                                        /* process packets */
313          done_packets(do_packets(pl));
314          return(1);                      /* and continue */
315   }
# Line 228 | Line 318 | rholo()                                /* holodeck main loop */
318   report(t)                       /* report progress so far */
319   time_t  t;
320   {
321 <        if (t == 0)
321 >        static time_t   seconds2go = 1000000;
322 >
323 >        if (t == 0L)
324                  t = time(NULL);
325 <        fprintf(stderr, "%s: %ld packets (%ld rays) done after %.2f hours\n",
326 <                        progname, npacksdone, nraysdone, (t-starttime)/3600.);
327 <        fflush(stderr);
328 <        if (vdef(REPORT))
329 <                reporttime = t + (time_t)(vflt(REPORT)*60.+.5);
325 >        sprintf(errmsg, "%ld packets (%ld rays) done after %.2f hours\n",
326 >                        npacksdone, nraysdone, (t-starttime)/3600.);
327 >        eputs(errmsg);
328 >        if (seconds2go == 1000000)
329 >                seconds2go = vdef(REPORT) ? (long)(vflt(REPORT)*60. + .5) : 0L;
330 >        if (seconds2go)
331 >                reporttime = t + seconds2go;
332   }
333  
334  
# Line 254 | Line 348 | register HDGRID        *gp;
348                  sprintf(errmsg, "ignoring all but first %s", vnam(SECTION));
349                  error(WARNING, errmsg);
350          }
257 –        if (sscanf(vval(SECTION),
258 –                        "%lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf",
259 –                        &gp->orig[0], &gp->orig[1], &gp->orig[2],
260 –                        &gp->xv[0][0], &gp->xv[0][1], &gp->xv[0][2],
261 –                        &gp->xv[1][0], &gp->xv[1][1], &gp->xv[1][2],
262 –                        &gp->xv[2][0], &gp->xv[2][1], &gp->xv[2][2]) != 12)
263 –                badvalue(SECTION);
264 –        maxlen = 0.;
265 –        for (i = 0; i < 3; i++)
266 –                if ((len[i] = VLEN(gp->xv[i])) > maxlen)
267 –                        maxlen = len[i];
268 –        if (!vdef(GRID)) {
269 –                sprintf(buf, "%.4f", maxlen/8.);
270 –                vval(GRID) = savqstr(buf);
271 –                vdef(GRID)++;
272 –        }
273 –        if ((d = vflt(GRID)) <= FTINY)
274 –                badvalue(GRID);
275 –        for (i = 0; i < 3; i++)
276 –                gp->grid[i] = len[i]/d + (1.-FTINY);
277 –        if (!vdef(EXPOSURE)) {
278 –                sprintf(errmsg, "%s must be defined", vnam(EXPOSURE));
279 –                error(USER, errmsg);
280 –        }
281 –        expval = vval(EXPOSURE)[0] == '-' || vval(EXPOSURE)[0] == '+' ?
282 –                        pow(2., vflt(EXPOSURE)) : vflt(EXPOSURE);
351          if (!vdef(OCTREE)) {
352                  if ((vval(OCTREE) = bmalloc(strlen(froot)+5)) == NULL)
353                          error(SYSTEM, "out of memory");
354                  sprintf(vval(OCTREE), "%s.oct", froot);
355                  vdef(OCTREE)++;
356          }
289 –        if (!vdef(DISKSPACE)) {
290 –                sprintf(errmsg,
291 –                        "no %s setting, assuming 100 Mbytes available",
292 –                                vnam(DISKSPACE));
293 –                error(WARNING, errmsg);
294 –                vval(DISKSPACE) = "100";
295 –                vdef(DISKSPACE)++;
296 –        }
297 –        if (!vdef(CACHE)) {
298 –                sprintf(errmsg,
299 –                        "no %s setting, assuming 10 Mbytes available",
300 –                                vnam(CACHE));
301 –                error(WARNING, errmsg);
302 –                vval(CACHE) = "10";
303 –                vdef(CACHE)++;
304 –        }
357          if (!vdef(OBSTRUCTIONS)) {
358                  vval(OBSTRUCTIONS) = "T";
359                  vdef(OBSTRUCTIONS)++;
360          }
361 +        if (!vdef(VDIST)) {
362 +                vval(VDIST) = "F";
363 +                vdef(VDIST)++;
364 +        }
365          if (!vdef(OCCUPANCY)) {
366                  vval(OCCUPANCY) = "U";
367                  vdef(OCCUPANCY)++;
# Line 313 | Line 369 | register HDGRID        *gp;
369                                  /* append rendering options */
370          if (vdef(RENDER))
371                  rtargc += wordstring(rtargv+rtargc, vval(RENDER));
372 +        
373 +        if (gp == NULL)         /* already initialized? */
374 +                return;
375 +                                /* set grid parameters */
376 +        gp->grid[0] = gp->grid[1] = gp->grid[2] = 0;
377 +        if (sscanf(vval(SECTION),
378 +                "%lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %hd %hd %hd",
379 +                        &gp->orig[0], &gp->orig[1], &gp->orig[2],
380 +                        &gp->xv[0][0], &gp->xv[0][1], &gp->xv[0][2],
381 +                        &gp->xv[1][0], &gp->xv[1][1], &gp->xv[1][2],
382 +                        &gp->xv[2][0], &gp->xv[2][1], &gp->xv[2][2],
383 +                        &gp->grid[0], &gp->grid[1], &gp->grid[2]) < 12)
384 +                badvalue(SECTION);
385 +        maxlen = 0.;
386 +        for (i = 0; i < 3; i++)
387 +                if ((len[i] = VLEN(gp->xv[i])) > maxlen)
388 +                        maxlen = len[i];
389 +        if (!vdef(GRID)) {
390 +                sprintf(buf, "%.4f", maxlen/8.);
391 +                vval(GRID) = savqstr(buf);
392 +                vdef(GRID)++;
393 +        }
394 +        if ((d = vflt(GRID)) <= FTINY)
395 +                badvalue(GRID);
396 +        for (i = 0; i < 3; i++)
397 +                if (gp->grid[i] <= 0)
398 +                        gp->grid[i] = len[i]/d + (1.-FTINY);
399   }
400  
401  
402   creatholo(gp)                   /* create a holodeck output file */
403   HDGRID  *gp;
404   {
405 +        extern char     VersionID[];
406          long    endloc = 0;
407 +        int     fd;
408          FILE    *fp;
409                                          /* open & truncate file */
410          if ((fp = fopen(hdkfile, "w+")) == NULL) {
# Line 328 | Line 413 | HDGRID *gp;
413          }
414                                          /* write information header */
415          newheader("RADIANCE", fp);
416 +        fprintf(fp, "SOFTWARE= %s\n", VersionID);
417          printvars(fp);
418          fputformat(HOLOFMT, fp);
419          fputc('\n', fp);
420          putw(HOLOMAGIC, fp);            /* put magic number & terminus */
421          fwrite(&endloc, sizeof(long), 1, fp);
422 <        fflush(fp);                     /* flush buffer */
423 <        hdinit(fileno(fp), gp);         /* allocate and initialize index */
424 <                        /* we're dropping fp here.... */
422 >        fd = dup(fileno(fp));
423 >        fclose(fp);                     /* flush and close stdio stream */
424 >        hdinit(fd, gp);                 /* allocate and initialize index */
425   }
426  
427  
# Line 365 | Line 451 | char   *s;
451  
452   loadholo()                      /* start loading a holodeck from fname */
453   {
454 +        extern long     ftell();
455          FILE    *fp;
456 <        long    endloc;
457 <                                        /* open input file */
458 <        if ((fp = fopen(hdkfile, "r+")) == NULL) {
459 <                sprintf(errmsg, "cannot open \"%s\" for appending", hdkfile);
456 >        int     fd;
457 >        long    fpos;
458 >                                        /* open holodeck file */
459 >        if ((fp = fopen(hdkfile, ncprocs>0 ? "r+" : "r")) == NULL) {
460 >                sprintf(errmsg, "cannot open \"%s\" for %s", hdkfile,
461 >                                ncprocs>0 ? "appending" : "reading");
462                  error(SYSTEM, errmsg);
463          }
464                                          /* load variables from header */
# Line 380 | Line 469 | loadholo()                     /* start loading a holodeck from fname */
469                                  hdkfile);
470                  error(USER, errmsg);
471          }
472 <        fread(&endloc, sizeof(long), 1, fp);
473 <        if (endloc != 0)
472 >        fread(&fpos, sizeof(long), 1, fp);
473 >        if (fpos != 0)
474                  error(WARNING, "ignoring multiple sections in holodeck file");
475 <        fseek(fp, 0L, 1);                       /* align system file pointer */
476 <        hdinit(fileno(fp), NULL);               /* allocate and load index */
477 <                        /* we're dropping fp here.... */
475 >        fpos = ftell(fp);                       /* get stdio position */
476 >        fd = dup(fileno(fp));
477 >        fclose(fp);                             /* done with stdio */
478 >        lseek(fd, fpos, 0);                     /* align system file pointer */
479 >        hdinit(fd, NULL);                       /* allocate and load index */
480   }
481  
482  
483   done_packets(pl)                /* handle finished packets */
484   PACKET  *pl;
485   {
486 +        static int      n2flush = 0;
487          register PACKET *p;
488  
489          while (pl != NULL) {
# Line 401 | Line 493 | PACKET *pl;
493                                  (char *)hdnewrays(hdlist[p->hd],p->bi,p->nr),
494                                  p->nr*sizeof(RAYVAL));
495                          if (outdev != NULL)     /* display it */
496 <                                disp_packet(p);
496 >                                disp_packet((PACKHEAD *)p);
497 >                        if (hdcachesize <= 0)   /* manual flushing */
498 >                                n2flush += p->nr;
499 >                        nraysdone += p->nr;
500 >                        npacksdone++;
501                  }
406 –                nraysdone += p->nr;
407 –                npacksdone++;
502                  p->nr = 0;                      /* push onto free list */
503                  p->next = freepacks;
504                  freepacks = p;
505          }
506 < }
507 <
508 <
415 < getradfile(rfargs)              /* run rad and get needed variables */
416 < char    *rfargs;
417 < {
418 <        static short    mvar[] = {VIEW,OCTREE,EXPOSURE,-1};
419 <        static char     tf1[] = TEMPLATE;
420 <        char    tf2[64];
421 <        char    combuf[256];
422 <        char    *pippt;
423 <        register int    i;
424 <        register char   *cp;
425 <                                        /* create rad command */
426 <        mktemp(tf1);
427 <        sprintf(tf2, "%s.rif", tf1);
428 <        sprintf(combuf,
429 <                "rad -v 0 -s -e -w %s OPTFILE=%s | egrep '^[ \t]*(NOMATCH",
430 <                        rfargs, tf1);
431 <        cp = combuf;
432 <        while (*cp){
433 <                if (*cp == '|') pippt = cp;
434 <                cp++;
435 <        }                               /* match unset variables */
436 <        for (i = 0; mvar[i] >= 0; i++)
437 <                if (!vdef(mvar[i])) {
438 <                        *cp++ = '|';
439 <                        strcpy(cp, vnam(mvar[i]));
440 <                        while (*cp) cp++;
441 <                        pippt = NULL;
442 <                }
443 <        if (pippt != NULL)
444 <                strcpy(pippt, "> /dev/null");   /* nothing to match */
445 <        else
446 <                sprintf(cp, ")[ \t]*=' > %s", tf2);
447 <        if (system(combuf)) {
448 <                error(SYSTEM, "cannot execute rad command");
449 <                unlink(tf2);                    /* clean up */
450 <                unlink(tf1);
451 <                quit(1);
506 >        if (n2flush > 512*RPACKSIZ*nprocs) {
507 >                hdflush(NULL);                  /* flush holodeck buffers */
508 >                n2flush = 0;
509          }
453 –        if (pippt == NULL) {
454 –                loadvars(tf2);                  /* load variables */
455 –                unlink(tf2);
456 –        }
457 –        rtargc += wordfile(rtargv+rtargc, tf1); /* get rtrace options */
458 –        unlink(tf1);                    /* clean up */
510   }
511  
512  
# Line 465 | Line 516 | register char  *rn, *fn;
516          char    *tp, *dp;
517  
518          for (tp = NULL, dp = rn; *rn = *fn++; rn++)
519 <                if (ISDIRSEP(*rn))
519 >                if (*rn == '/')
520                          dp = rn;
521                  else if (*rn == '.')
522                          tp = rn;
# Line 515 | Line 566 | int    ec;
566          int     status = 0;
567  
568          if (hdlist[0] != NULL) {        /* flush holodeck */
569 <                if (ncprocs > 0) {
570 <                        done_packets(flush_queue());
571 <                        status = end_rtrace();  /* close rtrace */
572 <                        hdflush(NULL);
573 <                        if (vdef(REPORT)) {
574 <                                long    fsiz, fuse;
575 <                                report(0);
576 <                                fsiz = lseek(hdlist[0]->fd, 0L, 2);
526 <                                fuse = hdfiluse(hdlist[0]->fd, 1);
527 <                                fprintf(stderr,
569 >                if (nprocs > 0)
570 >                        status = done_rtrace();
571 >                hdflush(NULL);
572 >                if (ncprocs > 0 && vdef(REPORT)) {
573 >                        long    fsiz, fuse;
574 >                        fsiz = hdfilen(hdlist[0]->fd);
575 >                        fuse = hdfiluse(hdlist[0]->fd, 1);
576 >                        fprintf(stderr,
577                          "%s: %.1f Mbyte holodeck file, %.1f%% fragmentation\n",
578 <                                                hdkfile, fsiz/(1024.*1024.),
579 <                                                100.*(fsiz-fuse)/fsiz);
580 <                        }
532 <                } else
533 <                        hdflush(NULL);
578 >                                        hdkfile, fsiz/(1024.*1024.),
579 >                                        100.*(fsiz-fuse)/fsiz);
580 >                }
581          }
582 +        if (orig_mode >= 0)             /* reset holodeck access mode */
583 +                fchmod(hdlist[0]->fd, orig_mode);
584 +        if (outdev != NULL)             /* close display */
585 +                disp_close();
586          exit(ec ? ec : status);         /* exit */
587   }

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines