src/hd/rholo.c

#ifndef lint
static const char       RCSid[] = "$Id: rholo.c,v 3.83 2023/02/09 21:54:11 greg Exp $";
#endif
/*
 * Radiance holodeck generation controller
 */

#include <stdio.h>
#include <time.h>
#include <signal.h>
#include <sys/stat.h>
#include <string.h>

#include "platform.h"
#include "rterror.h"
#include "resolu.h"
#include "rholo.h"
#include "random.h"

#ifndef FRAGWARN
#define FRAGWARN        20              /* fragmentation for warning (%) */
#endif
#ifndef MAXQTIME
#define MAXQTIME        3               /* target maximum seconds in queue */
#endif
                                        /* manual cache flushing frequency */
#ifndef RTFLUSH
#if MAXQTIME
#define RTFLUSH         (300/MAXQTIME*totqlen)  /* <= 5 minutes */
#else
#define RTFLUSH         (50*totqlen)            /* just guess */
#endif
#endif
                        /* the following must be consistent with rholo.h */
int     NVARS = NRHVARS;                /* total number of variables */

VARIABLE        vv[] = RHVINIT;         /* variable-value pairs */

char    *progname;              /* our program name */
char    *hdkfile;               /* holodeck file name */
char    froot[256];             /* root file name */

int     ncprocs = 0;            /* desired number of compute processes */

char    *outdev = NULL;         /* output device name */

int     readinp = 0;            /* read commands from stdin */

int     force = 0;              /* allow overwrite of holodeck (-1 == read-only) */

time_t  starttime;              /* time we got started */
time_t  endtime;                /* time we should end by */
time_t  reporttime;             /* time for next report */

off_t   maxdisk;                /* maximum file space (bytes) */

int     rtargc = 1;             /* rtrace command */
char    *rtargv[MAXRTARGC] = {"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 */
int     totqlen;                /* maximum queue length (number of packets) */

char  *sigerr[NSIG];            /* signal error messages */

extern int      nowarn;         /* turn warnings off? */

static void onsig(int  signo);
static void sigdie(int  signo, char  *msg);
static int resfmode(int fd, int mod);
static void initrholo(void);
static int rholo(void);
static void setdefaults(HDGRID  *gp);
static void creatholo(HDGRID    *gp);
static gethfunc headline;
static void loadholo(void);
static void rootname(char       *rn, char       *fn);
static void badvalue(int        vc);


int
main(
        int     argc,
        char    *argv[]
)
{
        int     i;

        progname = argv[0];                     /* get arguments */
        for (i = 1; i < argc && argv[i][0] == '-'; i++)
                switch (argv[i][1]) {
                case 'w':                       /* turn off warnings */
                        nowarn++;
                        break;
                case 'f':                       /* force overwrite */
                        force = 1;
                        break;
                case 'r':                       /* read-only mode */
                        force = -1;
                        break;
                case 'i':                       /* read input from stdin */
                        readinp++;
                        break;
                case 'n':                       /* compute processes */
                        if (i >= argc-2)
                                goto userr;
                        ncprocs = atoi(argv[++i]);
                        break;
                case 'o':                       /* output display */
                        if (i >= argc-2)
                                goto userr;
                        outdev = argv[++i];
                        break;
                default:
                        goto userr;
                }
                                                /* get root file name */
        if (i >= argc)
                goto userr;
        rootname(froot, hdkfile=argv[i++]);
                                                /* load variables? */
        if (i < argc)
                if (argv[i][0] != '-' && argv[i][0] != '+')
                        loadvars(argv[i]);      /* load variables from file */

        if (i >= argc || argv[i][0] == '+')
                loadholo();                     /* load existing holodeck */

        while (++i < argc)                      /* get command line settings */
                if (setvariable(argv[i], matchvar) < 0) {
                        sprintf(errmsg, "unknown variable: %s", argv[i]);
                        error(USER, errmsg);
                }
                                                /* check settings */
        checkvalues();
                                                /* load rad input file */
        getradfile();

        if (hdlist[0] == NULL) {                /* create new holodeck */
                HDGRID  hdg[HDMAX];
                                                        /* set defaults */
                setdefaults(hdg);
                                                        /* check read-only */
                if (force < 0)
                        error(USER, "cannot create read-only holodeck");
                                                        /* holodeck exists? */
                if (!force && access(hdkfile, R_OK|W_OK) == 0)
                        error(USER,
                                "holodeck file exists -- use -f to overwrite");
                                                        /* create holodeck */
                creatholo(hdg);
        } else                                  /* else just set defaults */
                setdefaults(NULL);
                                                /* initialize */
        initrholo();
                                                /* main loop */
        while (rholo())
                ;
                                                /* done */
        quit(0);
userr:
        fprintf(stderr,
"Usage: %s [-n nprocs][-o disp][-i][-w][-r|-f] output.hdk [control.hif|+|- [VAR=val ..]]\n",
                        progname);
        quit(1);
        return 1; /* pro forma return */
}


static void
onsig(                          /* fatal signal */
        int  signo
)
{
        static int  gotsig = 0;

        if (gotsig > 1)                 /* we're going as fast as we can! */
                return;
        if (gotsig++) {                 /* two signals and we split */
                hdsync(NULL, 0);        /* don't leave w/o saying goodbye */
                _exit(signo);
        }
        alarm(300);                     /* allow 10 minutes to clean up */
        eputs("signal - ");
        eputs(sigerr[signo]);
        eputs("\n");
        quit(3);
}


static void
sigdie(                 /* set fatal signal */
        int  signo,
        char  *msg
)
{
        if (signal(signo, onsig) == SIG_IGN)
                signal(signo, SIG_IGN);
        sigerr[signo] = msg;
}


static int
resfmode(               /* restrict open file access mode */
        int     fd,
        int     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 & 0777))
                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 */
}


static void
initrholo(void)                 /* get our holodeck running */
{
        int     i;
                                                /* close holodeck on exec() */
        fcntl(hdlist[0]->fd, F_SETFD, FD_CLOEXEC);

        if (outdev != NULL)                     /* open output device */
                disp_open(outdev);
        else if (ncprocs > 0)                   /* else use global ray feed */
                init_global();
                                                /* record disk space limit */
        if (!vdef(DISKSPACE))
                maxdisk = ((off_t)1<<(sizeof(off_t)*8-2)) - 1024;
        else
                maxdisk = 1024.*1024.*vflt(DISKSPACE);
                                                /* set up memory cache */
        if (outdev == NULL)
                hdcachesize = 0;                /* manual flushing */
        else if (vdef(CACHE))
                hdcachesize = 1024.*1024.*vflt(CACHE);
                                                /* open report file */
        if (vdef(REPORT)) {
                char    *s = sskip2(vval(REPORT), 1);
                if (*s && freopen(s, "a", stderr) == NULL)
                        quit(2);
        }
                                                /* mark the starting time */
        starttime = time(NULL);
                                                /* compute end time */
        if (!vdef(TIME) || vflt(TIME) <= FTINY)
                endtime = 0;
        else
                endtime = starttime + vflt(TIME)*3600. + .5;
                                                /* start rtrace */
        if (ncprocs > 0) {
                totqlen = i = start_rtrace();
                if (i < 1)
                        error(USER, "cannot start rtrace process(es)");
                if (vdef(REPORT)) {             /* make first report */
                        printargs(rtargc, rtargv, stderr);
                        report(0);
                }
                                                /* allocate packets */
                freepacks = (PACKET *)bmalloc(i*sizeof(PACKET));
                if (freepacks == NULL)
                        goto memerr;
                freepacks[--i].nr = 0;
                freepacks[i].next = NULL;
                if (!vdef(OBSTRUCTIONS) || !vbool(OBSTRUCTIONS)) {
                        freepacks[i].offset = (float *)bmalloc(
                                        RPACKSIZ*sizeof(float)*(i+1) );
                        if (freepacks[i].offset == NULL)
                                goto memerr;
                } else
                        freepacks[i].offset = NULL;
                while (i--) {
                        freepacks[i].nr = 0;
                        freepacks[i].offset = freepacks[i+1].offset == NULL ?
                                        NULL : freepacks[i+1].offset+RPACKSIZ ;
                        freepacks[i].next = &freepacks[i+1];
                }
        }
                                        /* set up signal handling */
        sigdie(SIGINT, "Interrupt");
        sigdie(SIGTERM, "Terminate");
#ifdef SIGHUP
        sigdie(SIGHUP, "Hangup");
#endif
#ifdef SIGPIPE
        sigdie(SIGPIPE, "Broken pipe");
#endif
#ifdef SIGALRM
        sigdie(SIGALRM, "Alarm clock");
#endif
#ifdef  SIGXCPU
        sigdie(SIGXCPU, "CPU limit exceeded");
#endif
#ifdef  SIGXFSZ
        sigdie(SIGXFSZ, "File size exceeded");
#endif
                                        /* protect holodeck file */
        orig_mode = resfmode(hdlist[0]->fd, (ncprocs>0) & (force>=0) ? 0 : 0444);
        return;
memerr:
        error(SYSTEM, "out of memory in initrholo");
}


static int
rholo(void)                             /* holodeck main loop */
{
        static off_t    nextfragwarn = 100L<<20;
        static int      idle = 0;
        PACKET  *pl = NULL, *plend;
        off_t   fsiz;
        int     pksiz;
        PACKET  *p;
        time_t  t;
                                        /* check display */
        if (nprocs <= 0)
                idle = 1;
        if (outdev != NULL) {
                if (!disp_check(idle))
                        return(0);      /* quit request */
                if (nprocs <= 0)
                        return(1);
        } else if (idle)
                return(0);              /* all done */
        fsiz = hdfilen(hdlist[0]->fd);  /* check file size */
        if (maxdisk > 0 && fsiz >= maxdisk) {
                error(USER, "file limit exceeded");
                done_rtrace();
                return(1);      /* comes back */
        }
#if FRAGWARN
        if (fsiz >= nextfragwarn) {
                double  pctfrag = 100.*(fsiz-hdfiluse(hdlist[0]->fd))/fsiz;
                if (pctfrag >= (double)FRAGWARN) {
                        sprintf(errmsg, "holodeck file fragmentation is %.0f%%",
                                        pctfrag);
                        error(WARNING, errmsg);
                        nextfragwarn = fsiz + (fsiz>>2);
                } else
                        nextfragwarn = fsiz + (10L<<20);
        }
#endif
        t = time(NULL);                 /* check time */
        if (endtime > 0 && t >= endtime) {
                error(USER, "time limit exceeded");
                done_rtrace();
                return(1);      /* comes back */
        }
        if (reporttime > 0 && t >= reporttime)
                report(t);
                                        /* figure out good packet size */
        pksiz = RPACKSIZ;
#if MAXQTIME
        if (!chunkycmp) {
                pksiz = nraysdone*MAXQTIME/(totqlen*(t - starttime + 1L));
                if (pksiz < 1) pksiz = 1;
                else if (pksiz > RPACKSIZ) pksiz = RPACKSIZ;
        }
#endif
        idle = 0;                       /* get packets to process */
        while (freepacks != NULL) {
                p = freepacks; freepacks = p->next; p->next = NULL;
                if (!next_packet(p, pksiz)) {
                        p->next = freepacks; freepacks = p;
                        idle = 1;
                        break;
                }
                if (pl == NULL) pl = p;
                else plend->next = p;
                plend = p;
        }
                                        /* process packets */
        done_packets(do_packets(pl));
        return(1);                      /* and continue */
}


static void
setdefaults(                    /* set default values */
        HDGRID  *gp
)
{
        int     i;
        int     n;
        double  len[3], d;

        if (!vdef(SECTION)) {
                sprintf(errmsg, "%s must be defined", vnam(SECTION));
                error(USER, errmsg);
        }
        if (!vdef(OCTREE)) {
                if ((vval(OCTREE) = bmalloc(strlen(froot)+5)) == NULL)
                        error(SYSTEM, "out of memory");
                sprintf(vval(OCTREE), "%s.oct", froot);
                vdef(OCTREE)++;
        }
        if (!vdef(VDIST)) {
                vval(VDIST) = "F";
                vdef(VDIST)++;
        }
                                /* append rendering options */
        if (vdef(RENDER))
                rtargc += wordstring(rtargv+rtargc, MAXRTARGC-rtargc, vval(RENDER));
        
        if (gp == NULL)         /* already initialized? */
                return;
                                /* set grid parameters */
        for (n = 0; n < vdef(SECTION); n++, gp++) {
                gp->grid[0] = gp->grid[1] = gp->grid[2] = 0;
                if (sscanf(nvalue(SECTION, n),
                "%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);
                for (i = 0; i < 3; i++)
                        len[i] = VLEN(gp->xv[i]);
                if (!vdef(GRID)) {
                        d = 2/2e5*( len[0]*len[0]*(len[1]*len[1] +
                                        len[2]*len[2] + 4*len[1]*len[2])
                                + len[1]*len[1]*len[2]*(len[2] + 4*len[0])
                                + 4*len[0]*len[1]*len[2]*len[2] );
                        d = sqrt(sqrt(d));
                } else 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);
        }
}


static void
creatholo(                      /* create a holodeck output file */
        HDGRID  *gp
)
{
        extern char     VersionID[];
        off_t   lastloc, nextloc;
        int     n;
        int     fd;
        FILE    *fp;
                                        /* open & truncate file */
        if ((fp = fopen(hdkfile, "wb+")) == NULL) {
                sprintf(errmsg, "cannot open \"%s\" for writing", hdkfile);
                error(SYSTEM, errmsg);
        }
                                        /* write information header */
        newheader("RADIANCE", fp);
        fprintf(fp, "SOFTWARE= %s\n", VersionID);
        printvars(fp);
        fputendian(fp);
        fputformat(HOLOFMT, fp);
        fputc('\n', fp);
        putw(HOLOMAGIC, fp);            /* put magic number */
        fd = dup(fileno(fp));
        fclose(fp);                     /* flush and close stdio stream */
        lastloc = lseek(fd, (off_t)0, SEEK_END);
        for (n = vdef(SECTION); n--; gp++) {    /* initialize each section */
                nextloc = 0L;
                write(fd, (char *)&nextloc, sizeof(nextloc));
                hdinit(fd, gp);                 /* writes beam index */
                if (!n)
                        break;
                nextloc = hdfilen(fd);          /* write section pointer */
                if (lseek(fd, lastloc, SEEK_SET) < 0)
                        error(SYSTEM,
                                "cannot seek on holodeck file in creatholo");
                write(fd, (char *)&nextloc, sizeof(nextloc));
                lseek(fd, (lastloc=nextloc), SEEK_SET);
        }
}


static int
headline(                       /* process information header line */
        char    *s,
        void    *p
)
{
        extern char     FMTSTR[];
        char    *cp;
        char    fmt[MAXFMTLEN];

        if (formatval(fmt, s)) {
                if (strcmp(fmt, HOLOFMT)) {
                        sprintf(errmsg, "%s file \"%s\" has %s%s",
                                        HOLOFMT, hdkfile, FMTSTR, fmt);
                        error(USER, errmsg);
                }
                return(0);
        }
        for (cp = s; *cp; cp++)         /* take off any comments */
                if (*cp == '#') {
                        *cp = '\0';
                        break;
                }
        setvariable(s, matchvar);       /* don't flag errors */
        return(0);
}


static void
loadholo(void)                  /* start loading a holodeck from fname */
{
        FILE    *fp;
        int     fd;
        int     n;
        off_t   nextloc;
        
        if ((ncprocs > 0) & (force >= 0))
                fp = fopen(hdkfile, "rb+");
        else
                fp = NULL;
        if (fp == NULL) {
                if ((fp = fopen(hdkfile, "rb")) == NULL) {
                        sprintf(errmsg, "cannot open \"%s\"", hdkfile);
                        error(SYSTEM, errmsg);
                }
                if (ncprocs > 0) {
                        sprintf(errmsg, "\"%s\" is read-only", hdkfile);
                        if (outdev == NULL)
                                error(USER, errmsg);
                        strcat(errmsg, "; new rays will be discarded");
                        error(WARNING, errmsg);
                        force = -1;
                }
        }
                                        /* load variables from header */
        getheader(fp, headline, NULL);
                                        /* check magic number */
        if (getw(fp) != HOLOMAGIC) {
                sprintf(errmsg, "bad magic number in holodeck file \"%s\"",
                                hdkfile);
                error(USER, errmsg);
        }
        nextloc = ftell(fp);                    /* get stdio position */
        fd = dup(fileno(fp));
        fclose(fp);                             /* done with stdio */
        for (n = 0; nextloc > 0L; n++) {        /* initialize each section */
                lseek(fd, nextloc, SEEK_SET);
                read(fd, (char *)&nextloc, sizeof(nextloc));
                hdinit(fd, NULL);
        }
        if (n != vdef(SECTION)) {
                sprintf(errmsg, "number of sections does not match %s setting",
                                vnam(SECTION));
                error(WARNING, errmsg);
        }
}


void
done_packets(           /* handle finished packets */
        PACKET  *pl
)
{
        static int      n2flush = 0;
        PACKET  *p;

        while (pl != NULL) {
                p = pl; pl = p->next; p->next = NULL;
                if (p->nr > 0) {                /* add to holodeck */
                        memcpy( (void *)hdnewrays(hdlist[p->hd],p->bi,p->nr),
                                (void *)p->ra,
                                p->nr*sizeof(RAYVAL));
                        if (outdev != NULL)     /* display it */
                                disp_packet((PACKHEAD *)p);
                        if (hdcachesize <= 0)
                                n2flush++;
                        nraysdone += p->nr;
                        npacksdone++;
                        p->nr = 0;
                }
                p->next = freepacks;            /* push onto free list */
                freepacks = p;
        }
        if (n2flush >= RTFLUSH) {
                if (outdev != NULL)
                        hdsync(NULL, 1);
                else
                        hdflush(NULL);
                n2flush = 0;
        }
}


static void
rootname(               /* remove tail from end of fn */
        char    *rn,
        char    *fn
)
{
        char    *tp, *dp;

        for (tp = NULL, dp = rn; (*rn = *fn++); rn++) {
                if (*rn == '/')
                        dp = rn;
                else if (*rn == '.')
                        tp = rn;
        }
        if (tp != NULL && tp > dp)
                *tp = '\0';
}


static void
badvalue(                       /* report bad variable value and exit */
        int     vc
)
{
        sprintf(errmsg, "bad value for variable '%s'", vnam(vc));
        error(USER, errmsg);
}


void
eputs(const char *s)            /* put error message to stderr */
{
        static int  midline = 0;

        if (!*s)
                return;
        if (!midline++) {       /* prepend line with program name */
                fputs(progname, stderr);
                fputs(": ", stderr);
        }
        fputs(s, stderr);
        if (s[strlen(s)-1] == '\n') {
                fflush(stderr);
                midline = 0;
        }
}


void
quit(int ec)                    /* exit program gracefully */
{
        int     status = 0;

        if (hdlist[0] != NULL) {        /* close holodeck */
                if (nprocs > 0)
                        status = done_rtrace();         /* calls hdsync() */
                if ((ncprocs > 0) & (force >= 0) && vdef(REPORT)) {
                        off_t   fsiz, fuse;
                        fsiz = hdfilen(hdlist[0]->fd);
                        fuse = hdfiluse(hdlist[0]->fd);
                        fprintf(stderr,
                        "%s: %.1f Mbyte holodeck file, %.1f%% fragmentation\n",
                                        hdkfile, fsiz/(1024.*1024.),
                                        100.*(fsiz-fuse)/fsiz);
                }
        }
        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 */
}
Revision:	3.84
Committed:	Tue Jan 21 22:30:01 2025 UTC (4 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 3.83:	+1 -3 lines
Log Message:	fix(rholo): Minimum # rays/packet was zero in some cases
#	User	Rev	Content
1	gregl	3.1	#ifndef lint
2	greg	3.84	static const char RCSid[] = "$Id: rholo.c,v 3.83 2023/02/09 21:54:11 greg Exp $";
3	gregl	3.1	#endif
4			/*
5			* Radiance holodeck generation controller
6			*/
7
8	schorsch	3.64	#include <stdio.h>
9			#include <time.h>
10	schorsch	3.59	#include <signal.h>
11			#include <sys/stat.h>
12			#include <string.h>
13
14	greg	3.63	#include "platform.h"
15	schorsch	3.64	#include "rterror.h"
16			#include "resolu.h"
17	gregl	3.1	#include "rholo.h"
18	gregl	3.14	#include "random.h"
19	gregl	3.1
20	gwlarson	3.44	#ifndef FRAGWARN
21			#define FRAGWARN 20 /* fragmentation for warning (%) */
22			#endif
23	gregl	3.33	#ifndef MAXQTIME
24	greg	3.73	#define MAXQTIME 3 /* target maximum seconds in queue */
25	gregl	3.33	#endif
26	gwlarson	3.49	/* manual cache flushing frequency */
27			#ifndef RTFLUSH
28			#if MAXQTIME
29			#define RTFLUSH (300/MAXQTIMEtotqlen) / <= 5 minutes */
30			#else
31			#define RTFLUSH (50totqlen) / just guess */
32			#endif
33			#endif
34	gregl	3.1	/* the following must be consistent with rholo.h */
35			int NVARS = NRHVARS; /* total number of variables */
36
37			VARIABLE vv[] = RHVINIT; /* variable-value pairs */
38
39			char progname; / our program name */
40			char hdkfile; / holodeck file name */
41	gregl	3.22	char froot[256]; /* root file name */
42	gregl	3.1
43			int ncprocs = 0; /* desired number of compute processes */
44
45			char outdev = NULL; / output device name */
46
47	gregl	3.17	int readinp = 0; /* read commands from stdin */
48
49	gwlarson	3.51	int force = 0; /* allow overwrite of holodeck (-1 == read-only) */
50	gregl	3.20
51	gregl	3.1	time_t starttime; /* time we got started */
52			time_t endtime; /* time we should end by */
53			time_t reporttime; /* time for next report */
54
55	greg	3.54	off_t maxdisk; /* maximum file space (bytes) */
56	gregl	3.8
57	gregl	3.1	int rtargc = 1; /* rtrace command */
58	greg	3.78	char *rtargv[MAXRTARGC] = {"rtrace", NULL};
59	gregl	3.1
60	gregl	3.12	int orig_mode = -1; /* original file mode (-1 if unchanged) */
61
62	gregl	3.1	long nraysdone = 0L; /* number of rays done */
63			long npacksdone = 0L; /* number of packets done */
64
65			PACKET freepacks; / available packets */
66	gwlarson	3.49	int totqlen; /* maximum queue length (number of packets) */
67	gregl	3.1
68	gregl	3.12	char sigerr[NSIG]; / signal error messages */
69
70	gwlarson	3.40	extern int nowarn; /* turn warnings off? */
71
72	schorsch	3.64	static void onsig(int signo);
73			static void sigdie(int signo, char *msg);
74			static int resfmode(int fd, int mod);
75			static void initrholo(void);
76			static int rholo(void);
77			static void setdefaults(HDGRID *gp);
78			static void creatholo(HDGRID *gp);
79	schorsch	3.65	static gethfunc headline;
80	schorsch	3.64	static void loadholo(void);
81			static void rootname(char rn, char fn);
82			static void badvalue(int vc);
83	gregl	3.1
84
85	schorsch	3.64	int
86			main(
87			int argc,
88			char *argv[]
89			)
90	gregl	3.1	{
91			int i;
92	gregl	3.33
93	gregl	3.1	progname = argv[0]; /* get arguments */
94			for (i = 1; i < argc && argv[i][0] == '-'; i++)
95			switch (argv[i][1]) {
96			case 'w': /* turn off warnings */
97			nowarn++;
98			break;
99			case 'f': /* force overwrite */
100	gwlarson	3.51	force = 1;
101	gregl	3.1	break;
102	gwlarson	3.51	case 'r': /* read-only mode */
103			force = -1;
104			break;
105	gregl	3.17	case 'i': /* read input from stdin */
106			readinp++;
107			break;
108	gregl	3.1	case 'n': /* compute processes */
109			if (i >= argc-2)
110			goto userr;
111			ncprocs = atoi(argv[++i]);
112			break;
113			case 'o': /* output display */
114			if (i >= argc-2)
115			goto userr;
116			outdev = argv[++i];
117			break;
118			default:
119			goto userr;
120			}
121			/* get root file name */
122	gregl	3.32	if (i >= argc)
123			goto userr;
124	gregl	3.1	rootname(froot, hdkfile=argv[i++]);
125	gregl	3.27	/* load variables? */
126			if (i < argc)
127			if (argv[i][0] != '-' && argv[i][0] != '+')
128			loadvars(argv[i]); /* load variables from file */
129
130			if (i >= argc \|\| argv[i][0] == '+')
131			loadholo(); /* load existing holodeck */
132
133			while (++i < argc) /* get command line settings */
134			if (setvariable(argv[i], matchvar) < 0) {
135			sprintf(errmsg, "unknown variable: %s", argv[i]);
136			error(USER, errmsg);
137			}
138			/* check settings */
139			checkvalues();
140	gregl	3.36	/* load rad input file */
141			getradfile();
142	gregl	3.27
143	gregl	3.28	if (hdlist[0] == NULL) { /* create new holodeck */
144	gregl	3.29	HDGRID hdg[HDMAX];
145	gregl	3.1	/* set defaults */
146	gregl	3.29	setdefaults(hdg);
147	gwlarson	3.51	/* check read-only */
148			if (force < 0)
149			error(USER, "cannot create read-only holodeck");
150	gregl	3.1	/* holodeck exists? */
151			if (!force && access(hdkfile, R_OK\|W_OK) == 0)
152			error(USER,
153			"holodeck file exists -- use -f to overwrite");
154			/* create holodeck */
155	gregl	3.29	creatholo(hdg);
156	gregl	3.27	} else /* else just set defaults */
157	gregl	3.15	setdefaults(NULL);
158	gregl	3.1	/* initialize */
159			initrholo();
160	gregl	3.15	/* main loop */
161	gregl	3.1	while (rholo())
162			;
163			/* done */
164			quit(0);
165			userr:
166			fprintf(stderr,
167	greg	3.74	"Usage: %s [-n nprocs][-o disp][-i][-w][-r\|-f] output.hdk [control.hif\|+\|- [VAR=val ..]]\n",
168	gregl	3.1	progname);
169			quit(1);
170	schorsch	3.64	return 1; /* pro forma return */
171	gregl	3.1	}
172
173
174	schorsch	3.64	static void
175			onsig( /* fatal signal */
176			int signo
177			)
178	gregl	3.12	{
179			static int gotsig = 0;
180
181	gwlarson	3.47	if (gotsig > 1) /* we're going as fast as we can! */
182			return;
183			if (gotsig++) { /* two signals and we split */
184			hdsync(NULL, 0); /* don't leave w/o saying goodbye */
185	gregl	3.12	_exit(signo);
186	gwlarson	3.47	}
187	greg	3.77	alarm(300); /* allow 10 minutes to clean up */
188	gregl	3.12	eputs("signal - ");
189			eputs(sigerr[signo]);
190			eputs("\n");
191			quit(3);
192			}
193
194
195	schorsch	3.64	static void
196			sigdie( /* set fatal signal */
197			int signo,
198			char *msg
199			)
200	gregl	3.12	{
201			if (signal(signo, onsig) == SIG_IGN)
202			signal(signo, SIG_IGN);
203			sigerr[signo] = msg;
204			}
205
206
207	schorsch	3.64	static int
208			resfmode( /* restrict open file access mode */
209			int fd,
210			int mod
211			)
212	gregl	3.12	{
213			struct stat stbuf;
214			/* get original mode */
215			if (fstat(fd, &stbuf) < 0)
216			error(SYSTEM, "cannot stat open holodeck file");
217			mod &= stbuf.st_mode; /* always more restrictive */
218	gregl	3.35	if (mod == (stbuf.st_mode & 0777))
219	gregl	3.12	return(-1); /* already set */
220			/* else change it */
221			if (fchmod(fd, mod) < 0) {
222			error(WARNING, "cannot change holodeck file access mode");
223			return(-1);
224			}
225			return(stbuf.st_mode); /* return original mode */
226			}
227
228
229	schorsch	3.64	static void
230			initrholo(void) /* get our holodeck running */
231	gregl	3.1	{
232	greg	3.80	int i;
233	gregl	3.17	/* close holodeck on exec() */
234			fcntl(hdlist[0]->fd, F_SETFD, FD_CLOEXEC);
235	gregl	3.4
236			if (outdev != NULL) /* open output device */
237			disp_open(outdev);
238	gregl	3.1	else if (ncprocs > 0) /* else use global ray feed */
239			init_global();
240	gregl	3.8	/* record disk space limit */
241			if (!vdef(DISKSPACE))
242	greg	3.58	maxdisk = ((off_t)1<<(sizeof(off_t)*8-2)) - 1024;
243	gregl	3.8	else
244	gregl	3.9	maxdisk = 1024.1024.vflt(DISKSPACE);
245	gregl	3.1	/* set up memory cache */
246	gregl	3.5	if (outdev == NULL)
247	gwlarson	3.50	hdcachesize = 0; /* manual flushing */
248	gregl	3.5	else if (vdef(CACHE))
249			hdcachesize = 1024.1024.vflt(CACHE);
250	gregl	3.1	/* open report file */
251			if (vdef(REPORT)) {
252	greg	3.80	char *s = sskip2(vval(REPORT), 1);
253	gregl	3.1	if (*s && freopen(s, "a", stderr) == NULL)
254			quit(2);
255			}
256	gregl	3.33	/* mark the starting time */
257			starttime = time(NULL);
258			/* compute end time */
259			if (!vdef(TIME) \|\| vflt(TIME) <= FTINY)
260			endtime = 0;
261			else
262			endtime = starttime + vflt(TIME)*3600. + .5;
263	gregl	3.1	/* start rtrace */
264			if (ncprocs > 0) {
265	gwlarson	3.38	totqlen = i = start_rtrace();
266	gregl	3.1	if (i < 1)
267	gregl	3.20	error(USER, "cannot start rtrace process(es)");
268	gregl	3.1	if (vdef(REPORT)) { /* make first report */
269			printargs(rtargc, rtargv, stderr);
270			report(0);
271			}
272			/* allocate packets */
273			freepacks = (PACKET )bmalloc(isizeof(PACKET));
274			if (freepacks == NULL)
275			goto memerr;
276			freepacks[--i].nr = 0;
277			freepacks[i].next = NULL;
278	gregl	3.25	if (!vdef(OBSTRUCTIONS) \|\| !vbool(OBSTRUCTIONS)) {
279	gregl	3.1	freepacks[i].offset = (float *)bmalloc(
280			RPACKSIZsizeof(float)(i+1) );
281			if (freepacks[i].offset == NULL)
282			goto memerr;
283			} else
284			freepacks[i].offset = NULL;
285			while (i--) {
286			freepacks[i].nr = 0;
287			freepacks[i].offset = freepacks[i+1].offset == NULL ?
288			NULL : freepacks[i+1].offset+RPACKSIZ ;
289			freepacks[i].next = &freepacks[i+1];
290			}
291			}
292	gregl	3.12	/* set up signal handling */
293			sigdie(SIGINT, "Interrupt");
294	schorsch	3.62	sigdie(SIGTERM, "Terminate");
295			#ifdef SIGHUP
296	gregl	3.12	sigdie(SIGHUP, "Hangup");
297	schorsch	3.62	#endif
298			#ifdef SIGPIPE
299	gregl	3.12	sigdie(SIGPIPE, "Broken pipe");
300	schorsch	3.62	#endif
301			#ifdef SIGALRM
302	gregl	3.12	sigdie(SIGALRM, "Alarm clock");
303	schorsch	3.62	#endif
304	gregl	3.12	#ifdef SIGXCPU
305			sigdie(SIGXCPU, "CPU limit exceeded");
306	schorsch	3.62	#endif
307			#ifdef SIGXFSZ
308	gregl	3.12	sigdie(SIGXFSZ, "File size exceeded");
309			#endif
310	gwlarson	3.50	/* protect holodeck file */
311	schorsch	3.60	orig_mode = resfmode(hdlist[0]->fd, (ncprocs>0) & (force>=0) ? 0 : 0444);
312	gregl	3.1	return;
313			memerr:
314			error(SYSTEM, "out of memory in initrholo");
315			}
316
317
318	schorsch	3.64	static int
319			rholo(void) /* holodeck main loop */
320	gregl	3.1	{
321	greg	3.71	static off_t nextfragwarn = 100L<<20;
322	gregl	3.30	static int idle = 0;
323	gregl	3.1	PACKET pl = NULL, plend;
324	greg	3.68	off_t fsiz;
325	gregl	3.33	int pksiz;
326	greg	3.80	PACKET *p;
327	gregl	3.1	time_t t;
328	gregl	3.30	/* check display */
329			if (nprocs <= 0)
330			idle = 1;
331			if (outdev != NULL) {
332	gregl	3.4	if (!disp_check(idle))
333	gregl	3.30	return(0); /* quit request */
334			if (nprocs <= 0)
335			return(1);
336			} else if (idle)
337			return(0); /* all done */
338	gwlarson	3.44	fsiz = hdfilen(hdlist[0]->fd); /* check file size */
339			if (maxdisk > 0 && fsiz >= maxdisk) {
340	greg	3.70	error(USER, "file limit exceeded");
341	gregl	3.21	done_rtrace();
342			return(1); /* comes back */
343	gregl	3.8	}
344	gwlarson	3.44	#if FRAGWARN
345	greg	3.68	if (fsiz >= nextfragwarn) {
346			double pctfrag = 100.*(fsiz-hdfiluse(hdlist[0]->fd))/fsiz;
347			if (pctfrag >= (double)FRAGWARN) {
348			sprintf(errmsg, "holodeck file fragmentation is %.0f%%",
349			pctfrag);
350			error(WARNING, errmsg);
351			nextfragwarn = fsiz + (fsiz>>2);
352			} else
353			nextfragwarn = fsiz + (10L<<20);
354	gwlarson	3.44	}
355			#endif
356	gregl	3.33	t = time(NULL); /* check time */
357	gregl	3.8	if (endtime > 0 && t >= endtime) {
358	greg	3.70	error(USER, "time limit exceeded");
359	gregl	3.21	done_rtrace();
360			return(1); /* comes back */
361	gregl	3.8	}
362	gregl	3.3	if (reporttime > 0 && t >= reporttime)
363	gregl	3.1	report(t);
364	gregl	3.33	/* figure out good packet size */
365	gwlarson	3.48	pksiz = RPACKSIZ;
366	gregl	3.33	#if MAXQTIME
367	gwlarson	3.48	if (!chunkycmp) {
368			pksiz = nraysdoneMAXQTIME/(totqlen(t - starttime + 1L));
369			if (pksiz < 1) pksiz = 1;
370			else if (pksiz > RPACKSIZ) pksiz = RPACKSIZ;
371			}
372	gregl	3.33	#endif
373	gregl	3.28	idle = 0; /* get packets to process */
374	gregl	3.1	while (freepacks != NULL) {
375			p = freepacks; freepacks = p->next; p->next = NULL;
376	gregl	3.33	if (!next_packet(p, pksiz)) {
377	gregl	3.1	p->next = freepacks; freepacks = p;
378	gregl	3.21	idle = 1;
379	gregl	3.1	break;
380			}
381			if (pl == NULL) pl = p;
382			else plend->next = p;
383			plend = p;
384			}
385	gregl	3.21	/* process packets */
386	gregl	3.1	done_packets(do_packets(pl));
387			return(1); /* and continue */
388			}
389
390
391	schorsch	3.64	static void
392			setdefaults( /* set default values */
393	greg	3.80	HDGRID *gp
394	schorsch	3.64	)
395	gregl	3.1	{
396	greg	3.80	int i;
397	gregl	3.29	int n;
398	gregl	3.37	double len[3], d;
399	gregl	3.1
400			if (!vdef(SECTION)) {
401			sprintf(errmsg, "%s must be defined", vnam(SECTION));
402			error(USER, errmsg);
403			}
404			if (!vdef(OCTREE)) {
405			if ((vval(OCTREE) = bmalloc(strlen(froot)+5)) == NULL)
406			error(SYSTEM, "out of memory");
407			sprintf(vval(OCTREE), "%s.oct", froot);
408			vdef(OCTREE)++;
409	gregl	3.14	}
410			if (!vdef(VDIST)) {
411			vval(VDIST) = "F";
412			vdef(VDIST)++;
413	gregl	3.1	}
414			/* append rendering options */
415			if (vdef(RENDER))
416	greg	3.78	rtargc += wordstring(rtargv+rtargc, MAXRTARGC-rtargc, vval(RENDER));
417	gregl	3.15
418			if (gp == NULL) /* already initialized? */
419			return;
420			/* set grid parameters */
421	gregl	3.29	for (n = 0; n < vdef(SECTION); n++, gp++) {
422			gp->grid[0] = gp->grid[1] = gp->grid[2] = 0;
423			if (sscanf(nvalue(SECTION, n),
424	gregl	3.16	"%lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %lf %hd %hd %hd",
425	gregl	3.29	&gp->orig[0], &gp->orig[1], &gp->orig[2],
426			&gp->xv[0][0], &gp->xv[0][1], &gp->xv[0][2],
427			&gp->xv[1][0], &gp->xv[1][1], &gp->xv[1][2],
428			&gp->xv[2][0], &gp->xv[2][1], &gp->xv[2][2],
429			&gp->grid[0], &gp->grid[1], &gp->grid[2]) < 12)
430			badvalue(SECTION);
431			for (i = 0; i < 3; i++)
432	gregl	3.37	len[i] = VLEN(gp->xv[i]);
433			if (!vdef(GRID)) {
434	gwlarson	3.43	d = 2/2e5( len[0]len[0](len[1]len[1] +
435	gregl	3.37	len[2]len[2] + 4len[1]*len[2])
436			+ len[1]len[1]len[2](len[2] + 4len[0])
437			+ 4len[0]len[1]len[2]len[2] );
438			d = sqrt(sqrt(d));
439			} else if ((d = vflt(GRID)) <= FTINY)
440	gregl	3.29	badvalue(GRID);
441			for (i = 0; i < 3; i++)
442			if (gp->grid[i] <= 0)
443			gp->grid[i] = len[i]/d + (1.-FTINY);
444	gregl	3.15	}
445	gregl	3.1	}
446
447
448	schorsch	3.64	static void
449			creatholo( /* create a holodeck output file */
450			HDGRID *gp
451			)
452	gregl	3.1	{
453	gregl	3.12	extern char VersionID[];
454	greg	3.75	off_t lastloc, nextloc;
455	gregl	3.29	int n;
456	gregl	3.10	int fd;
457	gregl	3.1	FILE *fp;
458			/* open & truncate file */
459	greg	3.81	if ((fp = fopen(hdkfile, "wb+")) == NULL) {
460	gregl	3.1	sprintf(errmsg, "cannot open \"%s\" for writing", hdkfile);
461			error(SYSTEM, errmsg);
462			}
463			/* write information header */
464			newheader("RADIANCE", fp);
465	gregl	3.12	fprintf(fp, "SOFTWARE= %s\n", VersionID);
466	gregl	3.1	printvars(fp);
467	greg	3.82	fputendian(fp);
468	gregl	3.1	fputformat(HOLOFMT, fp);
469			fputc('\n', fp);
470	gregl	3.29	putw(HOLOMAGIC, fp); /* put magic number */
471	gregl	3.10	fd = dup(fileno(fp));
472			fclose(fp); /* flush and close stdio stream */
473	greg	3.61	lastloc = lseek(fd, (off_t)0, SEEK_END);
474	gregl	3.29	for (n = vdef(SECTION); n--; gp++) { /* initialize each section */
475			nextloc = 0L;
476			write(fd, (char *)&nextloc, sizeof(nextloc));
477			hdinit(fd, gp); /* writes beam index */
478			if (!n)
479			break;
480			nextloc = hdfilen(fd); /* write section pointer */
481	greg	3.75	if (lseek(fd, lastloc, SEEK_SET) < 0)
482	gregl	3.29	error(SYSTEM,
483			"cannot seek on holodeck file in creatholo");
484			write(fd, (char *)&nextloc, sizeof(nextloc));
485	greg	3.75	lseek(fd, (lastloc=nextloc), SEEK_SET);
486	gregl	3.29	}
487	gregl	3.1	}
488
489
490	schorsch	3.64	static int
491			headline( /* process information header line */
492	schorsch	3.65	char *s,
493			void *p
494	schorsch	3.64	)
495	gregl	3.1	{
496			extern char FMTSTR[];
497	greg	3.80	char *cp;
498	greg	3.79	char fmt[MAXFMTLEN];
499	gregl	3.1
500	gregl	3.2	if (formatval(fmt, s)) {
501	gregl	3.1	if (strcmp(fmt, HOLOFMT)) {
502			sprintf(errmsg, "%s file \"%s\" has %s%s",
503			HOLOFMT, hdkfile, FMTSTR, fmt);
504			error(USER, errmsg);
505			}
506	gwlarson	3.42	return(0);
507	gregl	3.1	}
508			for (cp = s; cp; cp++) / take off any comments */
509			if (*cp == '#') {
510			*cp = '\0';
511			break;
512			}
513			setvariable(s, matchvar); /* don't flag errors */
514	gwlarson	3.42	return(0);
515	gregl	3.1	}
516
517
518	schorsch	3.64	static void
519			loadholo(void) /* start loading a holodeck from fname */
520	gregl	3.1	{
521			FILE *fp;
522	gregl	3.10	int fd;
523	gregl	3.29	int n;
524	greg	3.75	off_t nextloc;
525	gwlarson	3.51
526	schorsch	3.60	if ((ncprocs > 0) & (force >= 0))
527	greg	3.81	fp = fopen(hdkfile, "rb+");
528	gwlarson	3.51	else
529			fp = NULL;
530			if (fp == NULL) {
531	greg	3.81	if ((fp = fopen(hdkfile, "rb")) == NULL) {
532	gwlarson	3.51	sprintf(errmsg, "cannot open \"%s\"", hdkfile);
533			error(SYSTEM, errmsg);
534			}
535			if (ncprocs > 0) {
536	greg	3.72	sprintf(errmsg, "\"%s\" is read-only", hdkfile);
537			if (outdev == NULL)
538			error(USER, errmsg);
539			strcat(errmsg, "; new rays will be discarded");
540	gwlarson	3.51	error(WARNING, errmsg);
541			force = -1;
542			}
543	gregl	3.1	}
544			/* load variables from header */
545			getheader(fp, headline, NULL);
546			/* check magic number */
547			if (getw(fp) != HOLOMAGIC) {
548			sprintf(errmsg, "bad magic number in holodeck file \"%s\"",
549			hdkfile);
550			error(USER, errmsg);
551			}
552	gregl	3.29	nextloc = ftell(fp); /* get stdio position */
553	gregl	3.10	fd = dup(fileno(fp));
554			fclose(fp); /* done with stdio */
555	gregl	3.29	for (n = 0; nextloc > 0L; n++) { /* initialize each section */
556	greg	3.75	lseek(fd, nextloc, SEEK_SET);
557	gregl	3.29	read(fd, (char *)&nextloc, sizeof(nextloc));
558			hdinit(fd, NULL);
559			}
560			if (n != vdef(SECTION)) {
561			sprintf(errmsg, "number of sections does not match %s setting",
562			vnam(SECTION));
563			error(WARNING, errmsg);
564			}
565	gregl	3.1	}
566
567
568	greg	3.80	void
569	schorsch	3.64	done_packets( /* handle finished packets */
570			PACKET *pl
571			)
572	gregl	3.1	{
573	gregl	3.13	static int n2flush = 0;
574	greg	3.80	PACKET *p;
575	gregl	3.1
576			while (pl != NULL) {
577			p = pl; pl = p->next; p->next = NULL;
578			if (p->nr > 0) { /* add to holodeck */
579	schorsch	3.59	memcpy( (void *)hdnewrays(hdlist[p->hd],p->bi,p->nr),
580			(void *)p->ra,
581	gregl	3.1	p->nr*sizeof(RAYVAL));
582			if (outdev != NULL) /* display it */
583	gregl	3.7	disp_packet((PACKHEAD *)p);
584	gwlarson	3.49	if (hdcachesize <= 0)
585	gregl	3.33	n2flush++;
586	gregl	3.6	nraysdone += p->nr;
587			npacksdone++;
588	gregl	3.33	p->nr = 0;
589	gregl	3.1	}
590	gregl	3.33	p->next = freepacks; /* push onto free list */
591	gregl	3.1	freepacks = p;
592	gregl	3.5	}
593	gwlarson	3.50	if (n2flush >= RTFLUSH) {
594	gwlarson	3.46	if (outdev != NULL)
595			hdsync(NULL, 1);
596	gwlarson	3.41	else
597	gwlarson	3.46	hdflush(NULL);
598	gregl	3.13	n2flush = 0;
599	gregl	3.1	}
600			}
601
602
603	schorsch	3.64	static void
604			rootname( /* remove tail from end of fn */
605	greg	3.80	char *rn,
606			char *fn
607	schorsch	3.64	)
608	gregl	3.1	{
609			char tp, dp;
610
611	schorsch	3.60	for (tp = NULL, dp = rn; (rn = fn++); rn++) {
612	gregl	3.22	if (*rn == '/')
613	gregl	3.1	dp = rn;
614			else if (*rn == '.')
615			tp = rn;
616	schorsch	3.60	}
617	gregl	3.1	if (tp != NULL && tp > dp)
618			*tp = '\0';
619			}
620
621
622	schorsch	3.64	static void
623			badvalue( /* report bad variable value and exit */
624			int vc
625			)
626	gregl	3.1	{
627			sprintf(errmsg, "bad value for variable '%s'", vnam(vc));
628			error(USER, errmsg);
629			}
630
631
632	greg	3.52	void
633	greg	3.83	eputs(const char s) / put error message to stderr */
634	gregl	3.1	{
635			static int midline = 0;
636
637			if (!*s)
638			return;
639			if (!midline++) { /* prepend line with program name */
640			fputs(progname, stderr);
641			fputs(": ", stderr);
642			}
643			fputs(s, stderr);
644			if (s[strlen(s)-1] == '\n') {
645			fflush(stderr);
646			midline = 0;
647			}
648			}
649
650
651	greg	3.52	void
652	greg	3.83	quit(int ec) /* exit program gracefully */
653	gregl	3.1	{
654			int status = 0;
655
656	gregl	3.24	if (hdlist[0] != NULL) { /* close holodeck */
657	gregl	3.21	if (nprocs > 0)
658	gregl	3.24	status = done_rtrace(); /* calls hdsync() */
659	schorsch	3.60	if ((ncprocs > 0) & (force >= 0) && vdef(REPORT)) {
660	greg	3.54	off_t fsiz, fuse;
661	gregl	3.19	fsiz = hdfilen(hdlist[0]->fd);
662	greg	3.67	fuse = hdfiluse(hdlist[0]->fd);
663	gregl	3.19	fprintf(stderr,
664	gregl	3.3	"%s: %.1f Mbyte holodeck file, %.1f%% fragmentation\n",
665	gregl	3.19	hdkfile, fsiz/(1024.*1024.),
666			100.*(fsiz-fuse)/fsiz);
667			}
668	gregl	3.1	}
669	gregl	3.12	if (orig_mode >= 0) /* reset holodeck access mode */
670			fchmod(hdlist[0]->fd, orig_mode);
671	gregl	3.11	if (outdev != NULL) /* close display */
672			disp_close();
673	gregl	3.1	exit(ec ? ec : status); /* exit */
674			}