src/gen/mkillum.c

#ifndef lint
static const char RCSid[] = "$Id: mkillum.c,v 2.25 2005/05/26 14:46:21 greg Exp $";
#endif
/*
 * Make illum sources for optimizing rendering process
 */

#include  <signal.h>
#include  <ctype.h>

#include  "platform.h"
#include  "mkillum.h"
#include  "random.h"

                                /* default parameters */
#define  SAMPDENS       48              /* points per projected steradian */
#define  NSAMPS         32              /* samples per point */
#define  DFLMAT         "illum_mat"     /* material name */
#define  DFLDAT         "illum"         /* data file name */
                                /* selection options */
#define  S_NONE         0               /* select none */
#define  S_ELEM         1               /* select specified element */
#define  S_COMPL        2               /* select all but element */
#define  S_ALL          3               /* select all */

                                /* rtrace command and defaults */
char  *rtargv[64] = { "rtrace", "-dj", ".25", "-dr", "3", "-dv-",
                "-ab", "2", "-ad", "1024", "-as", "512", "-aa", ".1", };
int  rtargc = 14;
                                /* overriding rtrace options */
char  *myrtopts[] = { "-I-", "-i-", "-ld-", "-ov", "-h-",
                        "-fff", "-y", "0", NULL };

struct rtproc   rt0;            /* head of rtrace process list */

struct illum_args  thisillum = {        /* our illum and default values */
                0,
                DFLMAT,
                DFLDAT,
                0,
                VOIDID,
                SAMPDENS,
                NSAMPS,
                0.,
        };

char    matcheck[MAXSTR];       /* current material to include or exclude */
int     matselect = S_ALL;      /* selection criterion */

FUN     ofun[NUMOTYPE] = INIT_OTYPE;    /* object types */

char    persistfn[] = "pfXXXXXX";       /* persist file name */

int     gargc;                  /* global argc */
char    **gargv;                /* global argv */
#define  progname       gargv[0]

int     doneheader = 0;         /* printed header yet? */
#define  checkhead()    if (!doneheader++) printhead(gargc,gargv)

int     warnings = 1;           /* print warnings? */

int done_rprocs(struct rtproc *rtp);
void init(int np);
void filter(register FILE       *infp, char     *name);
void xoptions(char      *s, char        *nm);
void printopts(void);
void printhead(register int  ac, register char  **av);
void xobject(FILE  *fp, char  *nm);


int
main(           /* compute illum distributions using rtrace */
        int     argc,
        char    *argv[]
)
{
        int     nprocs = 1;
        char    *rtpath;
        FILE    *fp;
        register int    i;
                                /* set global arguments */
        gargv = argv;
                                /* check for -n option */
        if (!strcmp(argv[1], "-n")) {
                nprocs = atoi(argv[2]);
                if (nprocs <= 0)
                        error(USER, "illegal number of processes");
                i = 3;
        } else
                i = 1;
                                /* set up rtrace command */
        for ( ; i < argc; i++) {
                if (argv[i][0] == '<' && argv[i][1] == '\0')
                        break;
                rtargv[rtargc++] = argv[i];
                if (argv[i][0] == '-' && argv[i][1] == 'w')
                        switch (argv[i][2]) {
                        case '\0':
                                warnings = !warnings;
                                break;
                        case '+':
                        case 'T': case 't':
                        case 'Y': case 'y':
                        case '1':
                                warnings = 1;
                                break;
                        case '-':
                        case 'F': case 'f':
                        case 'N': case 'n':
                        case '0':
                                warnings = 0;
                                break;
                        }
        }
        gargc = i;
        if (!strcmp(rtargv[--rtargc], "-defaults"))
                nprocs = 0;
        if (nprocs > 1) {       /* add persist file if parallel invocation */
                rtargv[rtargc++] = "-PP";
                rtargv[rtargc++] = mktemp(persistfn);
        }
                                /* add "mandatory" rtrace options */
        for (i = 0; myrtopts[i] != NULL; i++)
                rtargv[rtargc++] = myrtopts[i];
                                /* finally, put back final argument */
        rtargv[rtargc++] = argv[gargc-1];
        rtargv[rtargc] = NULL;
        if (!nprocs) {          /* just asking for defaults? */
                printopts(); fflush(stdout);
                rtpath = getpath(rtargv[0], getenv("PATH"), X_OK);
                if (rtpath == NULL) {
                        eputs(rtargv[0]);
                        eputs(": command not found\n");
                        exit(1);
                }
                execv(rtpath, rtargv);
                perror(rtpath); /* execv() should not return */
                exit(1);
        }
        if (gargc < 2 || argv[gargc-1][0] == '-')
                error(USER, "missing octree argument");
                                /* else initialize and run our calculation */
        init(nprocs);
        if (gargc+1 < argc)
                for (i = gargc+1; i < argc; i++) {
                        if ((fp = fopen(argv[i], "r")) == NULL) {
                                sprintf(errmsg,
                                "cannot open scene file \"%s\"", argv[i]);
                                error(SYSTEM, errmsg);
                        }
                        filter(fp, argv[i]);
                        fclose(fp);
                }
        else
                filter(stdin, "standard input");
        quit(0);
}


#ifndef SIGALRM
#define SIGALRM SIGTERM
#endif
static void
killpersist(void)                       /* kill persistent rtrace process */
{
        FILE    *fp = fopen(persistfn, "r");
        int     pid;

        if (fp == NULL)
                return;
        if (fscanf(fp, "%*s %d", &pid) != 1 || kill(pid, SIGALRM) < 0)
                unlink(persistfn);
        fclose(fp);
}


int
done_rprocs(struct rtproc *rtp)
{
        int     st0, st1 = 0;

        if (rtp->next != NULL) {        /* close last opened first! */
                st1 = done_rprocs(rtp->next);
                free((void *)rtp->next);
                rtp->next = NULL;
        }
        st0 = close_process(&rtp->pd);
        if (st0 < 0)
                error(WARNING, "unknown return status from rtrace process");
        else if (st0 > 0)
                return(st0);
        return(st1);
}

void
quit(int status)                        /* exit with status */
{
        int     rtstat;

        if (rt0.next != NULL)           /* terminate persistent rtrace */
                killpersist();
                                        /* clean up rtrace process(es) */
        rtstat = done_rprocs(&rt0);
        if (status == 0)
                status = rtstat;
        exit(status);
}

void
init(int np)                            /* start rtrace and set up buffers */
{
        struct rtproc   *rtp;
        int     i;
        int     maxbytes;
                                        /* set up object functions */
        ofun[OBJ_FACE].funp = o_face;
        ofun[OBJ_SPHERE].funp = o_sphere;
        ofun[OBJ_RING].funp = o_ring;
                                        /* set up signal handling */
        signal(SIGINT, quit);
#ifdef SIGHUP
        signal(SIGHUP, quit);
#endif
#ifdef SIGTERM
        signal(SIGTERM, quit);
#endif
#ifdef SIGPIPE
        signal(SIGPIPE, quit);
#endif
        rtp = &rt0;                     /* start rtrace process(es) */
        for (i = 0; i++ < np; ) {
                errno = 0;
                maxbytes = open_process(&rtp->pd, rtargv);
                if (maxbytes == 0) {
                        eputs(rtargv[0]);
                        eputs(": command not found\n");
                        exit(1);
                }
                if (maxbytes < 0)
                        error(SYSTEM, "cannot start rtrace process");
                rtp->bsiz = maxbytes/(6*sizeof(float));
                rtp->buf = (float *)malloc(6*sizeof(float)*rtp->bsiz--);
                rtp->dest = (float **)calloc(rtp->bsiz, sizeof(float *));
                if (rtp->buf == NULL || rtp->dest == NULL)
                        error(SYSTEM, "out of memory in init");
                rtp->nrays = 0;
                if (i == np)            /* last process? */
                        break;
                if (np > 1)
                        sleep(2);       /* wait for persist file */
                rtp->next = (struct rtproc *)malloc(sizeof(struct rtproc));
                if (rtp->next == NULL)
                        error(SYSTEM, "out of memory in init");
                rtp = rtp->next;
        }
        rtp->next = NULL;
                                        /* set up urand */
        initurand(16384);
}


void
eputs(                          /* put string to stderr */
        register char  *s
)
{
        static int  midline = 0;

        if (!*s) return;
        if (!midline) {
                fputs(progname, stderr);
                fputs(": ", stderr);
        }
        fputs(s, stderr);
        midline = s[strlen(s)-1] != '\n';
}


void
wputs(s)                        /* print warning if enabled */
char  *s;
{
        if (warnings)
                eputs(s);
}


void
filter(         /* process stream */
        register FILE   *infp,
        char    *name
)
{
        char    buf[512];
        FILE    *pfp;
        register int    c;

        while ((c = getc(infp)) != EOF) {
                if (isspace(c))
                        continue;
                if (c == '#') {                         /* comment/options */
                        buf[0] = c;
                        fgets(buf+1, sizeof(buf)-1, infp);
                        xoptions(buf, name);
                } else if (c == '!') {                  /* command */
                        buf[0] = c;
                        fgetline(buf+1, sizeof(buf)-1, infp);
                        if ((pfp = popen(buf+1, "r")) == NULL) {
                                sprintf(errmsg, "cannot execute \"%s\"", buf);
                                error(SYSTEM, errmsg);
                        }
                        filter(pfp, buf);
                        pclose(pfp);
                } else {                                /* object */
                        ungetc(c, infp);
                        xobject(infp, name);
                }
        }
}


void
xoptions(                       /* process options in string s */
        char    *s,
        char    *nm
)
{
        extern FILE     *freopen();
        char    buf[64];
        int     nerrs = 0;
        register char   *cp;

        if (strncmp(s, "#@mkillum", 9) || !isspace(s[9])) {
                fputs(s, stdout);               /* not for us */
                return;
        }
        cp = s+10;
        while (*cp) {
                switch (*cp) {
                case ' ':
                case '\t':
                case '\n':
                case '\r':
                case '\f':
                        cp++;
                        continue;
                case 'm':                       /* material name */
                        if (*++cp != '=')
                                break;
                        if (!*++cp || isspace(*cp))
                                break;
                        atos(thisillum.matname, MAXSTR, cp);
                        cp = sskip(cp);
                        if (!(thisillum.flags & IL_DATCLB)) {
                                strcpy(thisillum.datafile, thisillum.matname);
                                thisillum.dfnum = 0;
                        }
                        continue;
                case 'f':                       /* data file name */
                        if (*++cp != '=')
                                break;
                        if (!*++cp || isspace(*cp)) {
                                strcpy(thisillum.datafile,thisillum.matname);
                                thisillum.dfnum = 0;
                                thisillum.flags &= ~IL_DATCLB;
                                continue;
                        }
                        atos(thisillum.datafile, MAXSTR, cp);
                        cp = sskip(cp);
                        thisillum.dfnum = 0;
                        thisillum.flags |= IL_DATCLB;
                        continue;
                case 'i':                       /* include material */
                case 'e':                       /* exclude material */
                        if (cp[1] != '=')
                                break;
                        matselect = (*cp == 'i') ? S_ELEM : S_COMPL;
                        cp += 2;
                        atos(matcheck, MAXSTR, cp);
                        cp = sskip(cp);
                        continue;
                case 'a':                       /* use everything */
                        cp = sskip(cp);
                        matselect = S_ALL;
                        continue;
                case 'n':                       /* use nothing (passive) */
                        cp = sskip(cp);
                        matselect = S_NONE;
                        continue;
                case 'c':                       /* color calculation */
                        if (*++cp != '=')
                                break;
                        switch (*++cp) {
                        case 'a':                       /* average */
                                thisillum.flags = (thisillum.flags|IL_COLAVG)
                                                        & ~IL_COLDST;
                                break;
                        case 'd':                       /* distribution */
                                thisillum.flags |= (IL_COLDST|IL_COLAVG);
                                break;
                        case 'n':                       /* none */
                                thisillum.flags &= ~(IL_COLAVG|IL_COLDST);
                                break;
                        default:
                                goto opterr;
                        }
                        cp = sskip(cp);
                        continue;
                case 'd':                       /* point sample density */
                        if (*++cp != '=')
                                break;
                        if (!isintd(++cp, " \t\n\r"))
                                break;
                        thisillum.sampdens = atoi(cp);
                        cp = sskip(cp);
                        continue;
                case 's':                       /* point super-samples */
                        if (*++cp != '=')
                                break;
                        if (!isintd(++cp, " \t\n\r"))
                                break;
                        thisillum.nsamps = atoi(cp);
                        cp = sskip(cp);
                        continue;
                case 'l':                       /* light sources */
                        cp++;
                        if (*cp == '+')
                                thisillum.flags |= IL_LIGHT;
                        else if (*cp == '-')
                                thisillum.flags &= ~IL_LIGHT;
                        else
                                break;
                        cp++;
                        continue;
                case 'b':                       /* brightness */
                        if (*++cp != '=')
                                break;
                        if (!isfltd(++cp, " \t\n\r"))
                                break;
                        thisillum.minbrt = atof(cp);
                        if (thisillum.minbrt < 0.)
                                thisillum.minbrt = 0.;
                        cp = sskip(cp);
                        continue;
                case 'o':                       /* output file */
                        if (*++cp != '=')
                                break;
                        if (!*++cp || isspace(*cp))
                                break;
                        atos(buf, sizeof(buf), cp);
                        cp = sskip(cp);
                        if (freopen(buf, "w", stdout) == NULL) {
                                sprintf(errmsg,
                                "cannot open output file \"%s\"", buf);
                                error(SYSTEM, errmsg);
                        }
                        doneheader = 0;
                        continue;
                case '!':                       /* processed file! */
                        sprintf(errmsg, "(%s): already processed!", nm);
                        error(WARNING, errmsg);
                        matselect = S_NONE;
                        return;
                }
        opterr:                                 /* skip faulty option */
                while (*cp && !isspace(*cp))
                        cp++;
                nerrs++;
        }
                                                /* print header? */
        checkhead();
                                                /* issue warnings? */
        if (nerrs) {
                sprintf(errmsg, "(%s): %d error(s) in option line:",
                                nm, nerrs);
                error(WARNING, errmsg);
                wputs(s);
                printf("# %s: the following option line has %d error(s):\n",
                                progname, nerrs);
        }
                                                /* print pure comment */
        printf("# %s", s+2);
}

void
printopts(void)                 /* print out option default values */
{
        printf("m=%-15s\t\t# material name\n", thisillum.matname);
        printf("f=%-15s\t\t# data file name\n", thisillum.datafile);
        if (thisillum.flags & IL_COLAVG)
                if (thisillum.flags & IL_COLDST)
                        printf("c=d\t\t\t\t# color distribution\n");
                else
                        printf("c=a\t\t\t\t# color average\n");
        else
                printf("c=n\t\t\t\t# color none\n");
        if (thisillum.flags & IL_LIGHT)
                printf("l+\t\t\t\t# light type on\n");
        else
                printf("l-\t\t\t\t# light type off\n");
        printf("d=%d\t\t\t\t# density of points\n", thisillum.sampdens);
        printf("s=%d\t\t\t\t# samples per point\n", thisillum.nsamps);
        printf("b=%f\t\t\t# minimum average brightness\n", thisillum.minbrt);
}


void
printhead(                      /* print out header */
        register int  ac,
        register char  **av
)
{
        putchar('#');
        while (ac-- > 0) {
                putchar(' ');
                fputs(*av++, stdout);
        }
        fputs("\n#@mkillum !\n", stdout);
}


void
xobject(                                /* translate an object from fp */
        FILE  *fp,
        char  *nm
)
{
        OBJREC  thisobj;
        char  str[MAXSTR];
        int  doit;
                                        /* read the object */
        if (fgetword(thisillum.altmat, MAXSTR, fp) == NULL)
                goto readerr;
        if (fgetword(str, MAXSTR, fp) == NULL)
                goto readerr;
                                        /* is it an alias? */
        if (!strcmp(str, ALIASKEY)) {
                if (fgetword(str, MAXSTR, fp) == NULL)
                        goto readerr;
                printf("\n%s %s %s", thisillum.altmat, ALIASKEY, str);
                if (fgetword(str, MAXSTR, fp) == NULL)
                        goto readerr;
                printf("\t%s\n", str);
                return;
        }
        thisobj.omod = OVOID;           /* unused field */
        if ((thisobj.otype = otype(str)) < 0) {
                sprintf(errmsg, "(%s): unknown type \"%s\"", nm, str);
                error(USER, errmsg);
        }
        if (fgetword(str, MAXSTR, fp) == NULL)
                goto readerr;
        thisobj.oname = str;
        if (readfargs(&thisobj.oargs, fp) != 1)
                goto readerr;
        thisobj.os = NULL;
                                        /* check for translation */
        switch (matselect) {
        case S_NONE:
                doit = 0;
                break;
        case S_ALL:
                doit = 1;
                break;
        case S_ELEM:
                doit = !strcmp(thisillum.altmat, matcheck);
                break;
        case S_COMPL:
                doit = strcmp(thisillum.altmat, matcheck);
                break;
        }
        doit = doit && issurface(thisobj.otype);
                                                /* print header? */
        checkhead();
                                                /* process object */
        if (doit)
                (*ofun[thisobj.otype].funp)(&thisobj, &thisillum, &rt0, nm);
        else
                printobj(thisillum.altmat, &thisobj);
                                                /* free arguments */
        freefargs(&thisobj.oargs);
        return;
readerr:
        sprintf(errmsg, "(%s): error reading scene", nm);
        error(USER, errmsg);
}
Revision:	2.26
Committed:	Mon Sep 19 11:30:10 2005 UTC (18 years, 7 months ago) by schorsch
Content type:	text/plain
Branch:	MAIN
Changes since 2.25:	+6 -1 lines
Log Message:	Removed SPEED from SCons. Changes to build with mingw, as suggested by Francesco Anselmo.
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	schorsch	2.26	static const char RCSid[] = "$Id: mkillum.c,v 2.25 2005/05/26 14:46:21 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* Make illum sources for optimizing rendering process
6			*/
7
8	greg	1.3	#include <signal.h>
9	schorsch	2.13	#include <ctype.h>
10	greg	1.3
11	greg	2.18	#include "platform.h"
12	greg	2.15	#include "mkillum.h"
13	schorsch	2.19	#include "random.h"
14	greg	1.1
15			/* default parameters */
16	greg	1.8	#define SAMPDENS 48 /* points per projected steradian */
17	greg	1.1	#define NSAMPS 32 /* samples per point */
18			#define DFLMAT "illum_mat" /* material name */
19	greg	1.10	#define DFLDAT "illum" /* data file name */
20	greg	1.1	/* selection options */
21			#define S_NONE 0 /* select none */
22			#define S_ELEM 1 /* select specified element */
23			#define S_COMPL 2 /* select all but element */
24			#define S_ALL 3 /* select all */
25
26			/* rtrace command and defaults */
27	greg	2.5	char *rtargv[64] = { "rtrace", "-dj", ".25", "-dr", "3", "-dv-",
28	greg	2.17	"-ab", "2", "-ad", "1024", "-as", "512", "-aa", ".1", };
29	greg	1.4	int rtargc = 14;
30	greg	1.1	/* overriding rtrace options */
31	gregl	2.10	char *myrtopts[] = { "-I-", "-i-", "-ld-", "-ov", "-h-",
32			"-fff", "-y", "0", NULL };
33	greg	1.1
34	greg	2.20	struct rtproc rt0; /* head of rtrace process list */
35	greg	1.1
36	greg	1.2	struct illum_args thisillum = { /* our illum and default values */
37			0,
38			DFLMAT,
39	greg	1.10	DFLDAT,
40	greg	1.2	0,
41			VOIDID,
42	greg	1.3	SAMPDENS,
43	greg	1.2	NSAMPS,
44	greg	1.12	0.,
45	greg	1.2	};
46	greg	1.1
47			char matcheck[MAXSTR]; /* current material to include or exclude */
48			int matselect = S_ALL; /* selection criterion */
49
50	greg	1.2	FUN ofun[NUMOTYPE] = INIT_OTYPE; /* object types */
51	greg	1.1
52	greg	2.20	char persistfn[] = "pfXXXXXX"; /* persist file name */
53
54	greg	1.1	int gargc; /* global argc */
55			char *gargv; / global argv */
56			#define progname gargv[0]
57
58			int doneheader = 0; /* printed header yet? */
59	greg	1.2	#define checkhead() if (!doneheader++) printhead(gargc,gargv)
60	greg	1.1
61	greg	1.2	int warnings = 1; /* print warnings? */
62
63	greg	2.23	int done_rprocs(struct rtproc *rtp);
64	greg	2.20	void init(int np);
65	schorsch	2.19	void filter(register FILE infp, char name);
66			void xoptions(char s, char nm);
67			void printopts(void);
68			void printhead(register int ac, register char **av);
69			void xobject(FILE fp, char nm);
70	greg	1.1
71	schorsch	2.19
72			int
73			main( /* compute illum distributions using rtrace */
74			int argc,
75			char *argv[]
76			)
77	greg	1.1	{
78	greg	2.20	int nprocs = 1;
79	greg	1.1	char *rtpath;
80	greg	1.3	FILE *fp;
81	greg	1.1	register int i;
82			/* set global arguments */
83	greg	1.3	gargv = argv;
84	greg	2.20	/* check for -n option */
85			if (!strcmp(argv[1], "-n")) {
86			nprocs = atoi(argv[2]);
87			if (nprocs <= 0)
88			error(USER, "illegal number of processes");
89			i = 3;
90			} else
91			i = 1;
92	greg	1.1	/* set up rtrace command */
93	greg	2.20	for ( ; i < argc; i++) {
94	greg	2.6	if (argv[i][0] == '<' && argv[i][1] == '\0')
95	greg	1.3	break;
96	greg	1.1	rtargv[rtargc++] = argv[i];
97	greg	1.2	if (argv[i][0] == '-' && argv[i][1] == 'w')
98	greg	2.6	switch (argv[i][2]) {
99			case '\0':
100			warnings = !warnings;
101			break;
102			case '+':
103			case 'T': case 't':
104			case 'Y': case 'y':
105			case '1':
106			warnings = 1;
107			break;
108			case '-':
109			case 'F': case 'f':
110			case 'N': case 'n':
111			case '0':
112			warnings = 0;
113			break;
114			}
115	greg	1.2	}
116	greg	2.3	gargc = i;
117	greg	2.20	if (!strcmp(rtargv[--rtargc], "-defaults"))
118			nprocs = 0;
119			if (nprocs > 1) { /* add persist file if parallel invocation */
120			rtargv[rtargc++] = "-PP";
121			rtargv[rtargc++] = mktemp(persistfn);
122			}
123			/* add "mandatory" rtrace options */
124	greg	1.1	for (i = 0; myrtopts[i] != NULL; i++)
125			rtargv[rtargc++] = myrtopts[i];
126	greg	2.20	/* finally, put back final argument */
127	greg	1.3	rtargv[rtargc++] = argv[gargc-1];
128	greg	1.1	rtargv[rtargc] = NULL;
129	greg	2.20	if (!nprocs) { /* just asking for defaults? */
130	greg	1.9	printopts(); fflush(stdout);
131	greg	1.1	rtpath = getpath(rtargv[0], getenv("PATH"), X_OK);
132			if (rtpath == NULL) {
133			eputs(rtargv[0]);
134			eputs(": command not found\n");
135			exit(1);
136			}
137			execv(rtpath, rtargv);
138	greg	2.20	perror(rtpath); /* execv() should not return */
139	greg	1.1	exit(1);
140			}
141	greg	2.3	if (gargc < 2 \|\| argv[gargc-1][0] == '-')
142			error(USER, "missing octree argument");
143	greg	1.1	/* else initialize and run our calculation */
144	greg	2.20	init(nprocs);
145	greg	1.3	if (gargc+1 < argc)
146			for (i = gargc+1; i < argc; i++) {
147	greg	1.5	if ((fp = fopen(argv[i], "r")) == NULL) {
148	greg	1.3	sprintf(errmsg,
149			"cannot open scene file \"%s\"", argv[i]);
150			error(SYSTEM, errmsg);
151			}
152			filter(fp, argv[i]);
153			fclose(fp);
154			}
155			else
156			filter(stdin, "standard input");
157	greg	2.20	quit(0);
158	greg	1.1	}
159
160	schorsch	2.26
161			#ifndef SIGALRM
162			#define SIGALRM SIGTERM
163			#endif
164	greg	2.20	static void
165	greg	2.21	killpersist(void) /* kill persistent rtrace process */
166	greg	2.20	{
167			FILE *fp = fopen(persistfn, "r");
168			int pid;
169
170			if (fp == NULL)
171			return;
172			if (fscanf(fp, "%*s %d", &pid) != 1 \|\| kill(pid, SIGALRM) < 0)
173			unlink(persistfn);
174			fclose(fp);
175			}
176	greg	1.1
177	schorsch	2.26
178	greg	2.23	int
179			done_rprocs(struct rtproc *rtp)
180			{
181			int st0, st1 = 0;
182
183			if (rtp->next != NULL) { /* close last opened first! */
184			st1 = done_rprocs(rtp->next);
185			free((void *)rtp->next);
186			rtp->next = NULL;
187			}
188			st0 = close_process(&rtp->pd);
189			if (st0 < 0)
190			error(WARNING, "unknown return status from rtrace process");
191			else if (st0 > 0)
192			return(st0);
193			return(st1);
194			}
195
196	greg	2.11	void
197	greg	2.20	quit(int status) /* exit with status */
198	greg	1.3	{
199			int rtstat;
200
201	greg	2.20	if (rt0.next != NULL) /* terminate persistent rtrace */
202			killpersist();
203			/* clean up rtrace process(es) */
204	greg	2.23	rtstat = done_rprocs(&rt0);
205			if (status == 0)
206			status = rtstat;
207	greg	1.3	exit(status);
208			}
209
210	schorsch	2.19	void
211	greg	2.20	init(int np) /* start rtrace and set up buffers */
212	greg	1.1	{
213	greg	2.20	struct rtproc *rtp;
214			int i;
215	greg	1.1	int maxbytes;
216	greg	1.3	/* set up object functions */
217			ofun[OBJ_FACE].funp = o_face;
218	greg	1.4	ofun[OBJ_SPHERE].funp = o_sphere;
219	greg	1.3	ofun[OBJ_RING].funp = o_ring;
220			/* set up signal handling */
221	greg	2.25	signal(SIGINT, quit);
222			#ifdef SIGHUP
223			signal(SIGHUP, quit);
224			#endif
225			#ifdef SIGTERM
226			signal(SIGTERM, quit);
227			#endif
228			#ifdef SIGPIPE
229	greg	1.3	signal(SIGPIPE, quit);
230	schorsch	2.13	#endif
231	greg	2.20	rtp = &rt0; /* start rtrace process(es) */
232			for (i = 0; i++ < np; ) {
233			errno = 0;
234			maxbytes = open_process(&rtp->pd, rtargv);
235			if (maxbytes == 0) {
236			eputs(rtargv[0]);
237			eputs(": command not found\n");
238			exit(1);
239			}
240			if (maxbytes < 0)
241			error(SYSTEM, "cannot start rtrace process");
242			rtp->bsiz = maxbytes/(6*sizeof(float));
243			rtp->buf = (float )malloc(6sizeof(float)*rtp->bsiz--);
244			rtp->dest = (float *)calloc(rtp->bsiz, sizeof(float ));
245			if (rtp->buf == NULL \|\| rtp->dest == NULL)
246			error(SYSTEM, "out of memory in init");
247			rtp->nrays = 0;
248			if (i == np) /* last process? */
249			break;
250	greg	2.24	if (np > 1)
251			sleep(2); /* wait for persist file */
252	greg	2.20	rtp->next = (struct rtproc *)malloc(sizeof(struct rtproc));
253			if (rtp->next == NULL)
254			error(SYSTEM, "out of memory in init");
255			rtp = rtp->next;
256	greg	1.1	}
257	greg	2.20	rtp->next = NULL;
258	greg	1.6	/* set up urand */
259	greg	2.7	initurand(16384);
260	greg	1.1	}
261
262
263	greg	2.11	void
264	schorsch	2.19	eputs( /* put string to stderr */
265			register char *s
266			)
267	greg	1.1	{
268			static int midline = 0;
269
270			if (!*s) return;
271			if (!midline) {
272			fputs(progname, stderr);
273			fputs(": ", stderr);
274			}
275			fputs(s, stderr);
276			midline = s[strlen(s)-1] != '\n';
277			}
278
279
280	greg	2.11	void
281	greg	1.2	wputs(s) /* print warning if enabled */
282			char *s;
283			{
284			if (warnings)
285			eputs(s);
286			}
287
288
289	schorsch	2.19	void
290			filter( /* process stream */
291			register FILE *infp,
292			char *name
293			)
294	greg	1.1	{
295			char buf[512];
296			FILE *pfp;
297			register int c;
298
299			while ((c = getc(infp)) != EOF) {
300			if (isspace(c))
301			continue;
302			if (c == '#') { /* comment/options */
303			buf[0] = c;
304			fgets(buf+1, sizeof(buf)-1, infp);
305	greg	1.2	xoptions(buf, name);
306	greg	1.1	} else if (c == '!') { /* command */
307			buf[0] = c;
308			fgetline(buf+1, sizeof(buf)-1, infp);
309			if ((pfp = popen(buf+1, "r")) == NULL) {
310			sprintf(errmsg, "cannot execute \"%s\"", buf);
311			error(SYSTEM, errmsg);
312			}
313	greg	1.2	filter(pfp, buf);
314	greg	1.1	pclose(pfp);
315			} else { /* object */
316			ungetc(c, infp);
317			xobject(infp, name);
318			}
319			}
320			}
321
322
323	schorsch	2.19	void
324			xoptions( /* process options in string s */
325			char *s,
326			char *nm
327			)
328	greg	1.1	{
329			extern FILE *freopen();
330			char buf[64];
331			int nerrs = 0;
332			register char *cp;
333
334	greg	1.2	if (strncmp(s, "#@mkillum", 9) \|\| !isspace(s[9])) {
335			fputs(s, stdout); /* not for us */
336	greg	1.1	return;
337	greg	1.2	}
338	greg	1.1	cp = s+10;
339			while (*cp) {
340			switch (*cp) {
341			case ' ':
342			case '\t':
343			case '\n':
344	greg	2.11	case '\r':
345			case '\f':
346	greg	1.1	cp++;
347			continue;
348			case 'm': /* material name */
349	greg	1.2	if (*++cp != '=')
350	greg	1.1	break;
351	greg	2.8	if (!++cp \|\| isspace(cp))
352	greg	1.2	break;
353	greg	1.1	atos(thisillum.matname, MAXSTR, cp);
354			cp = sskip(cp);
355			if (!(thisillum.flags & IL_DATCLB)) {
356			strcpy(thisillum.datafile, thisillum.matname);
357			thisillum.dfnum = 0;
358			}
359			continue;
360			case 'f': /* data file name */
361	greg	1.2	if (*++cp != '=')
362	greg	1.1	break;
363	greg	2.8	if (!++cp \|\| isspace(cp)) {
364	greg	1.6	strcpy(thisillum.datafile,thisillum.matname);
365	greg	1.10	thisillum.dfnum = 0;
366	greg	1.1	thisillum.flags &= ~IL_DATCLB;
367			continue;
368			}
369			atos(thisillum.datafile, MAXSTR, cp);
370			cp = sskip(cp);
371			thisillum.dfnum = 0;
372			thisillum.flags \|= IL_DATCLB;
373			continue;
374			case 'i': /* include material */
375			case 'e': /* exclude material */
376	greg	1.6	if (cp[1] != '=')
377	greg	1.1	break;
378	greg	1.7	matselect = (*cp == 'i') ? S_ELEM : S_COMPL;
379			cp += 2;
380			atos(matcheck, MAXSTR, cp);
381	greg	1.1	cp = sskip(cp);
382			continue;
383			case 'a': /* use everything */
384			cp = sskip(cp);
385			matselect = S_ALL;
386			continue;
387			case 'n': /* use nothing (passive) */
388			cp = sskip(cp);
389			matselect = S_NONE;
390			continue;
391			case 'c': /* color calculation */
392	greg	1.2	if (*++cp != '=')
393	greg	1.1	break;
394	greg	1.2	switch (*++cp) {
395	greg	1.1	case 'a': /* average */
396	greg	1.6	thisillum.flags = (thisillum.flags\|IL_COLAVG)
397			& ~IL_COLDST;
398	greg	1.1	break;
399			case 'd': /* distribution */
400	greg	1.5	thisillum.flags \|= (IL_COLDST\|IL_COLAVG);
401	greg	1.1	break;
402			case 'n': /* none */
403			thisillum.flags &= ~(IL_COLAVG\|IL_COLDST);
404			break;
405			default:
406			goto opterr;
407			}
408			cp = sskip(cp);
409			continue;
410			case 'd': /* point sample density */
411	greg	1.2	if (*++cp != '=')
412	greg	1.1	break;
413	greg	2.11	if (!isintd(++cp, " \t\n\r"))
414	greg	1.1	break;
415	greg	1.3	thisillum.sampdens = atoi(cp);
416	greg	1.1	cp = sskip(cp);
417			continue;
418			case 's': /* point super-samples */
419	greg	1.2	if (*++cp != '=')
420	greg	1.1	break;
421	greg	2.11	if (!isintd(++cp, " \t\n\r"))
422	greg	1.1	break;
423			thisillum.nsamps = atoi(cp);
424			cp = sskip(cp);
425			continue;
426	greg	1.3	case 'l': /* light sources */
427	greg	1.4	cp++;
428			if (*cp == '+')
429	greg	1.3	thisillum.flags \|= IL_LIGHT;
430	greg	1.4	else if (*cp == '-')
431			thisillum.flags &= ~IL_LIGHT;
432	greg	1.3	else
433	greg	1.4	break;
434			cp++;
435	greg	1.3	continue;
436	greg	1.12	case 'b': /* brightness */
437			if (*++cp != '=')
438			break;
439	greg	2.11	if (!isfltd(++cp, " \t\n\r"))
440	greg	1.12	break;
441			thisillum.minbrt = atof(cp);
442	greg	1.13	if (thisillum.minbrt < 0.)
443			thisillum.minbrt = 0.;
444	greg	1.12	cp = sskip(cp);
445			continue;
446	greg	1.1	case 'o': /* output file */
447	greg	1.2	if (*++cp != '=')
448	greg	1.1	break;
449	greg	2.8	if (!++cp \|\| isspace(cp))
450	greg	1.2	break;
451	greg	1.1	atos(buf, sizeof(buf), cp);
452			cp = sskip(cp);
453			if (freopen(buf, "w", stdout) == NULL) {
454			sprintf(errmsg,
455			"cannot open output file \"%s\"", buf);
456			error(SYSTEM, errmsg);
457			}
458			doneheader = 0;
459			continue;
460	greg	1.6	case '!': /* processed file! */
461	greg	1.7	sprintf(errmsg, "(%s): already processed!", nm);
462	greg	1.6	error(WARNING, errmsg);
463	greg	1.7	matselect = S_NONE;
464	greg	1.6	return;
465	greg	1.1	}
466			opterr: /* skip faulty option */
467	greg	2.9	while (cp && !isspace(cp))
468			cp++;
469	greg	1.1	nerrs++;
470			}
471	greg	1.2	/* print header? */
472			checkhead();
473			/* issue warnings? */
474	greg	1.1	if (nerrs) {
475			sprintf(errmsg, "(%s): %d error(s) in option line:",
476			nm, nerrs);
477			error(WARNING, errmsg);
478	greg	1.2	wputs(s);
479			printf("# %s: the following option line has %d error(s):\n",
480			progname, nerrs);
481	greg	1.1	}
482	greg	1.2	/* print pure comment */
483	greg	1.4	printf("# %s", s+2);
484	greg	1.9	}
485
486	schorsch	2.19	void
487			printopts(void) /* print out option default values */
488	greg	1.9	{
489	greg	1.11	printf("m=%-15s\t\t# material name\n", thisillum.matname);
490			printf("f=%-15s\t\t# data file name\n", thisillum.datafile);
491	greg	2.2	if (thisillum.flags & IL_COLAVG)
492			if (thisillum.flags & IL_COLDST)
493			printf("c=d\t\t\t\t# color distribution\n");
494			else
495			printf("c=a\t\t\t\t# color average\n");
496			else
497			printf("c=n\t\t\t\t# color none\n");
498			if (thisillum.flags & IL_LIGHT)
499			printf("l+\t\t\t\t# light type on\n");
500			else
501			printf("l-\t\t\t\t# light type off\n");
502	greg	1.9	printf("d=%d\t\t\t\t# density of points\n", thisillum.sampdens);
503			printf("s=%d\t\t\t\t# samples per point\n", thisillum.nsamps);
504	greg	1.12	printf("b=%f\t\t\t# minimum average brightness\n", thisillum.minbrt);
505	greg	1.1	}
506	greg	1.2
507
508	schorsch	2.19	void
509			printhead( /* print out header */
510			register int ac,
511			register char **av
512			)
513	greg	1.2	{
514			putchar('#');
515			while (ac-- > 0) {
516			putchar(' ');
517			fputs(*av++, stdout);
518			}
519	greg	1.6	fputs("\n#@mkillum !\n", stdout);
520	greg	1.2	}
521
522
523	schorsch	2.19	void
524			xobject( /* translate an object from fp */
525			FILE *fp,
526			char *nm
527			)
528	greg	1.2	{
529			OBJREC thisobj;
530			char str[MAXSTR];
531			int doit;
532			/* read the object */
533			if (fgetword(thisillum.altmat, MAXSTR, fp) == NULL)
534			goto readerr;
535			if (fgetword(str, MAXSTR, fp) == NULL)
536			goto readerr;
537			/* is it an alias? */
538	greg	2.12	if (!strcmp(str, ALIASKEY)) {
539	greg	1.2	if (fgetword(str, MAXSTR, fp) == NULL)
540			goto readerr;
541	greg	2.12	printf("\n%s %s %s", thisillum.altmat, ALIASKEY, str);
542	greg	1.2	if (fgetword(str, MAXSTR, fp) == NULL)
543			goto readerr;
544	greg	1.4	printf("\t%s\n", str);
545	greg	1.2	return;
546			}
547	greg	1.4	thisobj.omod = OVOID; /* unused field */
548	greg	1.2	if ((thisobj.otype = otype(str)) < 0) {
549			sprintf(errmsg, "(%s): unknown type \"%s\"", nm, str);
550			error(USER, errmsg);
551			}
552			if (fgetword(str, MAXSTR, fp) == NULL)
553			goto readerr;
554			thisobj.oname = str;
555			if (readfargs(&thisobj.oargs, fp) != 1)
556			goto readerr;
557			thisobj.os = NULL;
558			/* check for translation */
559			switch (matselect) {
560			case S_NONE:
561			doit = 0;
562			break;
563			case S_ALL:
564	greg	1.4	doit = 1;
565	greg	1.2	break;
566			case S_ELEM:
567			doit = !strcmp(thisillum.altmat, matcheck);
568			break;
569			case S_COMPL:
570			doit = strcmp(thisillum.altmat, matcheck);
571			break;
572			}
573	greg	1.4	doit = doit && issurface(thisobj.otype);
574	greg	1.2	/* print header? */
575			checkhead();
576			/* process object */
577			if (doit)
578	greg	2.20	(*ofun[thisobj.otype].funp)(&thisobj, &thisillum, &rt0, nm);
579	greg	1.2	else
580			printobj(thisillum.altmat, &thisobj);
581			/* free arguments */
582			freefargs(&thisobj.oargs);
583			return;
584			readerr:
585			sprintf(errmsg, "(%s): error reading scene", nm);
586			error(USER, errmsg);
587			}