ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rpmain.c
Revision: 2.31
Committed: Thu Jan 18 00:28:54 2024 UTC (3 months, 4 weeks ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.30: +2 -3 lines
Log Message:
fix(rcontrib, rpict, rvu): Better check for spectral sampling

File Contents

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