ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rpmain.c
Revision: 2.29
Committed: Fri Nov 17 20:02:07 2023 UTC (17 months, 2 weeks ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.28: +3 -35 lines
Log Message:
fix: multiple bug fixes in hyperspectral code, added rvu, mkillum, rsensor, and ranimove to "working except photon map" status

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rpmain.c,v 2.28 2023/11/15 18:02:53 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 err = setview(&ourview); /* set viewing parameters */
274 if (err != NULL)
275 error(USER, err);
276 /* initialize object types */
277 initotypes();
278 /* initialize urand */
279 if (rand_samp) {
280 srandom((long)time(0));
281 initurand(0);
282 } else {
283 srandom(0L);
284 initurand(2048);
285 }
286 /* set up signal handling */
287 sigdie(SIGINT, "Interrupt");
288 #ifdef SIGHUP
289 sigdie(SIGHUP, "Hangup");
290 #endif
291 sigdie(SIGTERM, "Terminate");
292 #ifdef SIGPIPE
293 sigdie(SIGPIPE, "Broken pipe");
294 #endif
295 #ifdef SIGALRM
296 sigdie(SIGALRM, "Alarm clock");
297 #endif
298 #ifdef SIGXCPU
299 sigdie(SIGXCPU, "CPU limit exceeded");
300 sigdie(SIGXFSZ, "File size exceeded");
301 #endif
302 /* open error file */
303 if (errfile != NULL) {
304 if (freopen(errfile, "a", stderr) == NULL)
305 quit(2);
306 fprintf(stderr, "**************\n*** PID %5d: ",
307 getpid());
308 printargs(argc, argv, stderr);
309 putc('\n', stderr);
310 fflush(stderr);
311 }
312 #ifdef NICE
313 nice(NICE); /* lower priority */
314 #endif
315 /* get octree */
316 if (i == argc)
317 octname = NULL;
318 else if (i == argc-1)
319 octname = argv[i];
320 else
321 goto badopt;
322 if (seqstart > 0 && octname == NULL)
323 error(USER, "missing octree argument");
324 /* set up output */
325 #ifdef PERSIST
326 if (persist) {
327 if (recover != NULL)
328 error(USER, "persist option used with recover file");
329 if (seqstart <= 0)
330 error(USER, "persist option only for sequences");
331 if (outfile == NULL)
332 duped1 = dup(fileno(stdout)); /* don't lose our output */
333 openheader();
334 } else
335 #endif
336 if (outfile != NULL)
337 openheader();
338 SET_FILE_BINARY(stdout);
339 if (octname == NULL)
340 SET_FILE_BINARY(stdin);
341 readoct(octname, loadflags, &thescene, NULL);
342 nsceneobjs = nobjects;
343
344 if (loadflags & IO_INFO) { /* print header */
345 printargs(i, argv, stdout);
346 printf("SOFTWARE= %s\n", VersionID);
347 }
348
349 ray_init_pmap(); /* PMAP: set up & load photon maps */
350
351 marksources(); /* find and mark sources */
352
353 setambient(); /* initialize ambient calculation */
354
355 fflush(stdout); /* in case we're duplicating header */
356
357 #ifdef PERSIST
358 if (persist) {
359 if (outfile == NULL) { /* reconnect stdout */
360 dup2(duped1, fileno(stdout));
361 close(duped1);
362 }
363 if (persist == PARALLEL) { /* multiprocessing */
364 preload_objs(); /* preload scene */
365 shm_boundary = (char *)malloc(16);
366 strcpy(shm_boundary, "SHM_BOUNDARY");
367 while ((rval=fork()) == 0) { /* keep on forkin' */
368 pflock(1);
369 pfhold();
370 tstart = time((time_t *)NULL);
371 ambsync(); /* load new values */
372 }
373 if (rval < 0)
374 error(SYSTEM, "cannot fork child for persist function");
375 pfdetach(); /* parent will run then exit */
376 }
377 }
378 runagain:
379 if (persist) {
380 if (outfile == NULL) /* if out to stdout */
381 dupheader(); /* send header */
382 else /* if out to file */
383 duped1 = dup(fileno(stdout)); /* hang onto pipe */
384 }
385 #endif
386 /* batch render picture(s) */
387 rpict(seqstart, outfile, zfile, recover);
388 /* flush ambient file */
389 ambsync();
390 #ifdef PERSIST
391 if (persist == PERSIST) { /* first run-through */
392 if ((rval=fork()) == 0) { /* child loops until killed */
393 pflock(1);
394 persist = PCHILD;
395 } else { /* original process exits */
396 if (rval < 0)
397 error(SYSTEM, "cannot fork child for persist function");
398 pfdetach(); /* parent exits */
399 }
400 }
401 if (persist == PCHILD) { /* wait for a signal then go again */
402 if (outfile != NULL)
403 close(duped1); /* release output handle */
404 pfhold();
405 tstart = time((time_t *)NULL); /* reinitialize */
406 raynum = nrays = 0;
407 goto runagain;
408 }
409 #endif
410
411
412 ray_done_pmap(); /* PMAP: free photon maps */
413
414 quit(0);
415
416 badopt:
417 sprintf(errmsg, "command line error at '%s'", argv[i]);
418 error(USER, errmsg);
419 return 1; /* pro forma return */
420
421 #undef check
422 #undef check_bool
423 }
424
425
426 void
427 wputs( /* warning output function */
428 const char *s
429 )
430 {
431 int lasterrno = errno;
432 eputs(s);
433 errno = lasterrno;
434 }
435
436
437 void
438 eputs( /* put string to stderr */
439 const char *s
440 )
441 {
442 static int midline = 0;
443
444 if (!*s)
445 return;
446 if (!midline++) {
447 fputs(progname, stderr);
448 fputs(": ", stderr);
449 }
450 fputs(s, stderr);
451 if (s[strlen(s)-1] == '\n') {
452 fflush(stderr);
453 midline = 0;
454 }
455 }
456
457
458 static void
459 onsig( /* fatal signal */
460 int signo
461 )
462 {
463 static int gotsig = 0;
464
465 if (gotsig++) /* two signals and we're gone! */
466 _exit(signo);
467
468 #ifdef SIGALRM /* XXX how critical is this? */
469 alarm(15); /* allow 15 seconds to clean up */
470 signal(SIGALRM, SIG_DFL); /* make certain we do die */
471 #endif
472 eputs("signal - ");
473 eputs(sigerr[signo]);
474 eputs("\n");
475 quit(3);
476 }
477
478
479 static void
480 sigdie( /* set fatal signal */
481 int signo,
482 char *msg
483 )
484 {
485 if (signal(signo, onsig) == SIG_IGN)
486 signal(signo, SIG_IGN);
487 sigerr[signo] = msg;
488 }
489
490
491 static void
492 printdefaults(void) /* print default values to stdout */
493 {
494 printf("-vt%c\t\t\t\t# view type %s\n", ourview.type,
495 ourview.type==VT_PER ? "perspective" :
496 ourview.type==VT_PAR ? "parallel" :
497 ourview.type==VT_HEM ? "hemispherical" :
498 ourview.type==VT_ANG ? "angular" :
499 ourview.type==VT_CYL ? "cylindrical" :
500 ourview.type==VT_PLS ? "planisphere" :
501 "unknown");
502 printf("-vp %f %f %f\t# view point\n",
503 ourview.vp[0], ourview.vp[1], ourview.vp[2]);
504 printf("-vd %f %f %f\t# view direction\n",
505 ourview.vdir[0], ourview.vdir[1], ourview.vdir[2]);
506 printf("-vu %f %f %f\t# view up\n",
507 ourview.vup[0], ourview.vup[1], ourview.vup[2]);
508 printf("-vh %f\t\t\t# view horizontal size\n", ourview.horiz);
509 printf("-vv %f\t\t\t# view vertical size\n", ourview.vert);
510 printf("-vo %f\t\t\t# view fore clipping plane\n", ourview.vfore);
511 printf("-va %f\t\t\t# view aft clipping plane\n", ourview.vaft);
512 printf("-vs %f\t\t\t# view shift\n", ourview.hoff);
513 printf("-vl %f\t\t\t# view lift\n", ourview.voff);
514 printf("-x %-9d\t\t\t# x resolution\n", hresolu);
515 printf("-y %-9d\t\t\t# y resolution\n", vresolu);
516 if (out_prims == stdprims)
517 printf("-pRGB\t\t\t\t# standard RGB color output\n");
518 else if (out_prims == xyzprims)
519 printf("-pXYZ\t\t\t\t# CIE XYZ color output\n");
520 else if (out_prims != NULL)
521 printf("-pc %.4f %.4f %.4f %.4f %.4f %.4f %.4f %.4f\t# output color primaries and white point\n",
522 out_prims[RED][0], out_prims[RED][1],
523 out_prims[GRN][0], out_prims[GRN][1],
524 out_prims[BLU][0], out_prims[BLU][1],
525 out_prims[WHT][0], out_prims[WHT][1]);
526 printf("-pa %f\t\t\t# pixel aspect ratio\n", pixaspect);
527 printf("-pj %f\t\t\t# pixel jitter\n", dstrpix);
528 printf("-pm %f\t\t\t# pixel motion\n", mblur);
529 printf("-pd %f\t\t\t# pixel depth-of-field\n", dblur);
530 printf("-ps %-9d\t\t\t# pixel sample\n", psample);
531 printf("-pt %f\t\t\t# pixel threshold\n", maxdiff);
532 printf("-t %-9d\t\t\t# time between reports\n", ralrm);
533 printf(erract[WARNING].pf != NULL ?
534 "-w+\t\t\t\t# warning messages on\n" :
535 "-w-\t\t\t\t# warning messages off\n");
536 print_rdefaults();
537 }