ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rpmain.c
Revision: 2.9
Committed: Tue Jan 18 00:33:16 2005 UTC (19 years, 3 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.8: +8 -1 lines
Log Message:
Added -pd option for depth-of-field sampling and -vd focal distance entry

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.9 static const char RCSid[] = "$Id: rpmain.c,v 2.8 2004/03/30 16:13:01 schorsch Exp $";
3 greg 2.1 #endif
4     /*
5     * rpmain.c - main for rpict batch rendering program
6     */
7    
8 greg 2.2 #include "copyright.h"
9 greg 2.1
10 schorsch 2.3 #include <sys/types.h>
11     #include <signal.h>
12    
13     #include "platform.h"
14 schorsch 2.6 #include "rtprocess.h" /* getpid() */
15 greg 2.1 #include "ray.h"
16     #include "source.h"
17     #include "ambient.h"
18     #include "random.h"
19     #include "paths.h"
20     #include "view.h"
21    
22     /* persistent processes define */
23     #ifdef F_SETLKW
24     #define PERSIST 1 /* normal persist */
25     #define PARALLEL 2 /* parallel persist */
26     #define PCHILD 3 /* child of normal persist */
27     #endif
28    
29     char *progname; /* argv[0] */
30     char *octname; /* octree name */
31     char *sigerr[NSIG]; /* signal error messages */
32     char *shm_boundary = NULL; /* boundary of shared memory */
33     char *errfile = NULL; /* error output file */
34    
35     extern time_t time();
36     extern time_t tstart; /* start time */
37    
38     extern int ralrm; /* seconds between reports */
39    
40     extern VIEW ourview; /* viewing parameters */
41    
42     extern int hresolu; /* horizontal resolution */
43     extern int vresolu; /* vertical resolution */
44     extern double pixaspect; /* pixel aspect ratio */
45    
46     extern int psample; /* pixel sample size */
47     extern double maxdiff; /* max. sample difference */
48     extern double dstrpix; /* square pixel distribution */
49    
50     extern double mblur; /* motion blur parameter */
51    
52 greg 2.9 extern double dblur; /* depth-of-field blur parameter */
53    
54 schorsch 2.8 static void onsig(int signo);
55     static void sigdie(int signo, char *msg);
56     static void printdefaults(void);
57 greg 2.1
58    
59     int
60 schorsch 2.8 main(int argc, char *argv[])
61 greg 2.1 {
62     #define check(ol,al) if (argv[i][ol] || \
63     badarg(argc-i-1,argv+i+1,al)) \
64     goto badopt
65     #define bool(olen,var) switch (argv[i][olen]) { \
66     case '\0': var = !var; break; \
67     case 'y': case 'Y': case 't': case 'T': \
68     case '+': case '1': var = 1; break; \
69     case 'n': case 'N': case 'f': case 'F': \
70     case '-': case '0': var = 0; break; \
71     default: goto badopt; }
72     char *err;
73     char *recover = NULL;
74     char *outfile = NULL;
75     char *zfile = NULL;
76     int loadflags = ~IO_FILES;
77     int seqstart = 0;
78     int persist = 0;
79     int duped1;
80     int rval;
81     int i;
82     /* record start time */
83     tstart = time((time_t *)NULL);
84     /* global program name */
85     progname = argv[0] = fixargv0(argv[0]);
86     /* option city */
87     for (i = 1; i < argc; i++) {
88     /* expand arguments */
89     while ((rval = expandarg(&argc, &argv, i)) > 0)
90     ;
91     if (rval < 0) {
92     sprintf(errmsg, "cannot expand '%s'", argv[i]);
93     error(SYSTEM, errmsg);
94     }
95     if (argv[i] == NULL || argv[i][0] != '-')
96     break; /* break from options */
97     if (!strcmp(argv[i], "-version")) {
98     puts(VersionID);
99     quit(0);
100     }
101     if (!strcmp(argv[i], "-defaults") ||
102     !strcmp(argv[i], "-help")) {
103     printdefaults();
104     quit(0);
105     }
106     rval = getrenderopt(argc-i, argv+i);
107     if (rval >= 0) {
108     i += rval;
109     continue;
110     }
111     rval = getviewopt(&ourview, argc-i, argv+i);
112     if (rval >= 0) {
113     i += rval;
114     continue;
115     }
116     /* rpict options */
117     switch (argv[i][1]) {
118     case 'v': /* view file */
119     if (argv[i][2] != 'f')
120     goto badopt;
121     check(3,"s");
122     rval = viewfile(argv[++i], &ourview, NULL);
123     if (rval < 0) {
124     sprintf(errmsg,
125     "cannot open view file \"%s\"",
126     argv[i]);
127     error(SYSTEM, errmsg);
128     } else if (rval == 0) {
129     sprintf(errmsg,
130     "bad view file \"%s\"",
131     argv[i]);
132     error(USER, errmsg);
133     }
134     break;
135     case 'p': /* pixel */
136     switch (argv[i][2]) {
137     case 's': /* sample */
138     check(3,"i");
139     psample = atoi(argv[++i]);
140     break;
141     case 't': /* threshold */
142     check(3,"f");
143     maxdiff = atof(argv[++i]);
144     break;
145     case 'j': /* jitter */
146     check(3,"f");
147     dstrpix = atof(argv[++i]);
148     break;
149     case 'a': /* aspect */
150     check(3,"f");
151     pixaspect = atof(argv[++i]);
152     break;
153     case 'm': /* motion */
154     check(3,"f");
155     mblur = atof(argv[++i]);
156     break;
157 greg 2.9 case 'd': /* aperture */
158     check(3,"f");
159     dblur = atof(argv[++i]);
160     break;
161 greg 2.1 default:
162     goto badopt;
163     }
164     break;
165     case 'x': /* x resolution */
166     check(2,"i");
167     hresolu = atoi(argv[++i]);
168     break;
169     case 'y': /* y resolution */
170     check(2,"i");
171     vresolu = atoi(argv[++i]);
172     break;
173     case 'S': /* slave index */
174     check(2,"i");
175     seqstart = atoi(argv[++i]);
176     break;
177     case 'o': /* output file */
178     check(2,"s");
179     outfile = argv[++i];
180     break;
181     case 'z': /* z file */
182     check(2,"s");
183     zfile = argv[++i];
184     break;
185     case 'r': /* recover file */
186     if (argv[i][2] == 'o') { /* +output */
187     check(3,"s");
188     outfile = argv[i+1];
189     } else
190     check(2,"s");
191     recover = argv[++i];
192     break;
193     case 't': /* timer */
194     check(2,"i");
195     ralrm = atoi(argv[++i]);
196     break;
197     #ifdef PERSIST
198     case 'P': /* persist file */
199     if (argv[i][2] == 'P') {
200     check(3,"s");
201     persist = PARALLEL;
202     } else {
203     check(2,"s");
204     persist = PERSIST;
205     }
206     persistfile(argv[++i]);
207     break;
208     #endif
209     case 'w': /* warnings */
210     rval = erract[WARNING].pf != NULL;
211     bool(2,rval);
212     if (rval) erract[WARNING].pf = wputs;
213     else erract[WARNING].pf = NULL;
214     break;
215     case 'e': /* error file */
216     check(2,"s");
217     errfile = argv[++i];
218     break;
219     default:
220     goto badopt;
221     }
222     }
223     err = setview(&ourview); /* set viewing parameters */
224     if (err != NULL)
225     error(USER, err);
226     /* initialize object types */
227     initotypes();
228     /* initialize urand */
229     initurand(2048);
230     /* set up signal handling */
231     sigdie(SIGINT, "Interrupt");
232 schorsch 2.4 #ifdef SIGHUP
233 greg 2.1 sigdie(SIGHUP, "Hangup");
234 schorsch 2.4 #endif
235 greg 2.1 sigdie(SIGTERM, "Terminate");
236 schorsch 2.4 #ifdef SIGPIPE
237 greg 2.1 sigdie(SIGPIPE, "Broken pipe");
238 schorsch 2.4 #endif
239     #ifdef SIGALRM
240 greg 2.1 sigdie(SIGALRM, "Alarm clock");
241 schorsch 2.4 #endif
242 greg 2.1 #ifdef SIGXCPU
243     sigdie(SIGXCPU, "CPU limit exceeded");
244     sigdie(SIGXFSZ, "File size exceeded");
245     #endif
246     /* open error file */
247     if (errfile != NULL) {
248     if (freopen(errfile, "a", stderr) == NULL)
249     quit(2);
250     fprintf(stderr, "**************\n*** PID %5d: ",
251     getpid());
252     printargs(argc, argv, stderr);
253     putc('\n', stderr);
254     fflush(stderr);
255     }
256     #ifdef NICE
257     nice(NICE); /* lower priority */
258     #endif
259     /* get octree */
260     if (i == argc)
261     octname = NULL;
262     else if (i == argc-1)
263     octname = argv[i];
264     else
265     goto badopt;
266     if (seqstart > 0 && octname == NULL)
267     error(USER, "missing octree argument");
268     /* set up output */
269     #ifdef PERSIST
270     if (persist) {
271     if (recover != NULL)
272     error(USER, "persist option used with recover file");
273     if (seqstart <= 0)
274     error(USER, "persist option only for sequences");
275     if (outfile == NULL)
276     duped1 = dup(fileno(stdout)); /* don't lose our output */
277     openheader();
278     } else
279     #endif
280     if (outfile != NULL)
281     openheader();
282 schorsch 2.3 #ifdef _WIN32
283     SET_FILE_BINARY(stdout);
284 greg 2.1 if (octname == NULL)
285 schorsch 2.3 SET_FILE_BINARY(stdin);
286 greg 2.1 #endif
287     readoct(octname, loadflags, &thescene, NULL);
288     nsceneobjs = nobjects;
289    
290     if (loadflags & IO_INFO) { /* print header */
291     printargs(i, argv, stdout);
292     printf("SOFTWARE= %s\n", VersionID);
293     }
294    
295     marksources(); /* find and mark sources */
296    
297     setambient(); /* initialize ambient calculation */
298    
299     #ifdef PERSIST
300     if (persist) {
301     fflush(stdout);
302     if (outfile == NULL) { /* reconnect stdout */
303     dup2(duped1, fileno(stdout));
304     close(duped1);
305     }
306     if (persist == PARALLEL) { /* multiprocessing */
307     preload_objs(); /* preload scene */
308     shm_boundary = (char *)malloc(16);
309     strcpy(shm_boundary, "SHM_BOUNDARY");
310     while ((rval=fork()) == 0) { /* keep on forkin' */
311     pflock(1);
312     pfhold();
313     tstart = time((time_t *)NULL);
314     }
315     if (rval < 0)
316     error(SYSTEM, "cannot fork child for persist function");
317     pfdetach(); /* parent exits */
318     }
319     }
320     runagain:
321 schorsch 2.7 if (persist) {
322 greg 2.1 if (outfile == NULL) /* if out to stdout */
323     dupheader(); /* send header */
324     else /* if out to file */
325     duped1 = dup(fileno(stdout)); /* hang onto pipe */
326 schorsch 2.7 }
327 greg 2.1 #endif
328     /* batch render picture(s) */
329     rpict(seqstart, outfile, zfile, recover);
330     /* flush ambient file */
331     ambsync();
332     #ifdef PERSIST
333     if (persist == PERSIST) { /* first run-through */
334     if ((rval=fork()) == 0) { /* child loops until killed */
335     pflock(1);
336     persist = PCHILD;
337     } else { /* original process exits */
338     if (rval < 0)
339     error(SYSTEM, "cannot fork child for persist function");
340     pfdetach(); /* parent exits */
341     }
342     }
343     if (persist == PCHILD) { /* wait for a signal then go again */
344     if (outfile != NULL)
345     close(duped1); /* release output handle */
346     pfhold();
347     tstart = time((time_t *)NULL); /* reinitialize */
348     raynum = nrays = 0;
349     goto runagain;
350     }
351     #endif
352     quit(0);
353    
354     badopt:
355     sprintf(errmsg, "command line error at '%s'", argv[i]);
356     error(USER, errmsg);
357 schorsch 2.8 return 1; /* pro forma return */
358 greg 2.1
359     #undef check
360     #undef bool
361     }
362    
363    
364     void
365 schorsch 2.8 wputs( /* warning output function */
366     char *s
367     )
368 greg 2.1 {
369     int lasterrno = errno;
370     eputs(s);
371     errno = lasterrno;
372     }
373    
374    
375     void
376 schorsch 2.8 eputs( /* put string to stderr */
377     register char *s
378     )
379 greg 2.1 {
380     static int midline = 0;
381    
382     if (!*s)
383     return;
384     if (!midline++) {
385     fputs(progname, stderr);
386     fputs(": ", stderr);
387     }
388     fputs(s, stderr);
389     if (s[strlen(s)-1] == '\n') {
390     fflush(stderr);
391     midline = 0;
392     }
393     }
394    
395    
396 schorsch 2.8 static void
397     onsig( /* fatal signal */
398     int signo
399     )
400 greg 2.1 {
401     static int gotsig = 0;
402    
403     if (gotsig++) /* two signals and we're gone! */
404     _exit(signo);
405    
406 schorsch 2.4 #ifdef SIGALRM /* XXX how critical is this? */
407 greg 2.1 alarm(15); /* allow 15 seconds to clean up */
408     signal(SIGALRM, SIG_DFL); /* make certain we do die */
409 schorsch 2.4 #endif
410 greg 2.1 eputs("signal - ");
411     eputs(sigerr[signo]);
412     eputs("\n");
413     quit(3);
414     }
415    
416    
417 schorsch 2.8 static void
418     sigdie( /* set fatal signal */
419     int signo,
420     char *msg
421     )
422 greg 2.1 {
423     if (signal(signo, onsig) == SIG_IGN)
424     signal(signo, SIG_IGN);
425     sigerr[signo] = msg;
426     }
427    
428    
429 schorsch 2.8 static void
430     printdefaults(void) /* print default values to stdout */
431 greg 2.1 {
432     printf("-vt%c\t\t\t\t# view type %s\n", ourview.type,
433     ourview.type==VT_PER ? "perspective" :
434     ourview.type==VT_PAR ? "parallel" :
435     ourview.type==VT_HEM ? "hemispherical" :
436     ourview.type==VT_ANG ? "angular" :
437     ourview.type==VT_CYL ? "cylindrical" :
438     "unknown");
439     printf("-vp %f %f %f\t# view point\n",
440     ourview.vp[0], ourview.vp[1], ourview.vp[2]);
441     printf("-vd %f %f %f\t# view direction\n",
442     ourview.vdir[0], ourview.vdir[1], ourview.vdir[2]);
443     printf("-vu %f %f %f\t# view up\n",
444     ourview.vup[0], ourview.vup[1], ourview.vup[2]);
445     printf("-vh %f\t\t\t# view horizontal size\n", ourview.horiz);
446     printf("-vv %f\t\t\t# view vertical size\n", ourview.vert);
447     printf("-vo %f\t\t\t# view fore clipping plane\n", ourview.vfore);
448     printf("-va %f\t\t\t# view aft clipping plane\n", ourview.vaft);
449     printf("-vs %f\t\t\t# view shift\n", ourview.hoff);
450     printf("-vl %f\t\t\t# view lift\n", ourview.voff);
451     printf("-x %-9d\t\t\t# x resolution\n", hresolu);
452     printf("-y %-9d\t\t\t# y resolution\n", vresolu);
453     printf("-pa %f\t\t\t# pixel aspect ratio\n", pixaspect);
454     printf("-pj %f\t\t\t# pixel jitter\n", dstrpix);
455     printf("-pm %f\t\t\t# pixel motion\n", mblur);
456 greg 2.9 printf("-pd %f\t\t\t# pixel depth-of-field\n", dblur);
457 greg 2.1 printf("-ps %-9d\t\t\t# pixel sample\n", psample);
458     printf("-pt %f\t\t\t# pixel threshold\n", maxdiff);
459     printf("-t %-9d\t\t\t# time between reports\n", ralrm);
460     printf(erract[WARNING].pf != NULL ?
461     "-w+\t\t\t\t# warning messages on\n" :
462     "-w-\t\t\t\t# warning messages off\n");
463     print_rdefaults();
464     }