ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rxcmain.cpp
Revision: 2.13
Committed: Tue Dec 24 16:58:13 2024 UTC (4 months, 1 week ago) by greg
Branch: MAIN
Changes since 2.12: +5 -5 lines
Log Message:
fix(rxcontrib): Correct final time reporting

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rxcmain.cpp,v 2.12 2024/12/23 18:14:20 greg Exp $";
3 #endif
4 /*
5 * rxcmain.c - main for rxcontrib ray contribution tracer
6 */
7
8 #include "copyright.h"
9
10 #include <signal.h>
11 #include <time.h>
12 #include <ctype.h>
13 #include "RcontribSimulManager.h"
14 #include "platform.h"
15 #include "func.h"
16
17 const char *sigerr[NSIG]; /* signal error messages */
18
19 int nproc = 1; /* number of processes requested */
20
21 int inpfmt = 'a'; /* input format */
22 int outfmt = 'f'; /* output format */
23
24 int report_intvl = 0; /* reporting interval (seconds) */
25
26 extern char * progname; // global argv[0]
27
28 RcontribSimulManager myRCmanager; // global rcontrib simulation manager
29
30 #define RCONTRIB_FEATURES "Multiprocessing\n" \
31 "Accumulation\nRecovery\n" \
32 "ImmediateIrradiance\n" \
33 "ProgressReporting\nDistanceLimiting\n" \
34 "InputFormats=a,f,d\nOutputFormats=f,d,c\n" \
35 "Outputs=V,W\n" \
36 "OutputCS=RGB,spec\n"
37
38 static void rxcontrib(const int rstart = 0);
39
40 static void
41 printdefaults(void) /* print default values to stdout */
42 {
43 printf("-c %-5d\t\t\t# accumulated rays per record\n", myRCmanager.accum);
44 printf(myRCmanager.HasFlag(RCcontrib) ?
45 "-V+\t\t\t\t# output contributions\n" :
46 "-V-\t\t\t\t# output coefficients\n");
47 if (myRCmanager.HasFlag(RTimmIrrad))
48 printf("-I+\t\t\t\t# immediate irradiance on\n");
49 printf("-n %-2d\t\t\t\t# number of rendering processes\n", nproc);
50 if (myRCmanager.xres > 0)
51 printf("-x %-9d\t\t\t# x resolution\n", myRCmanager.xres);
52 printf("-y %-9d\t\t\t# y resolution\n", myRCmanager.yres);
53 printf(myRCmanager.HasFlag(RTlimDist) ?
54 "-ld+\t\t\t\t# limit distance on\n" :
55 "-ld-\t\t\t\t# limit distance off\n");
56 printf("-f%c%c\t\t\t\t# format input/output = %s/%s\n",
57 inpfmt, outfmt, formstr(inpfmt), formstr(outfmt));
58 if (report_intvl > 0)
59 printf("-t %-9d\t\t\t# time between reports\n", report_intvl);
60 printf(erract[WARNING].pf != NULL ?
61 "-w+\t\t\t\t# warning messages on\n" :
62 "-w-\t\t\t\t# warning messages off\n");
63 print_rdefaults();
64 }
65
66
67 static void
68 onsig( /* fatal signal */
69 int signo
70 )
71 {
72 static int gotsig = 0;
73
74 if (gotsig++) /* two signals and we're gone! */
75 _exit(signo);
76
77 #ifdef SIGALRM
78 alarm(180); /* allow 3 minutes to clean up */
79 signal(SIGALRM, SIG_DFL); /* make certain we do die */
80 #endif
81 eputs("signal - ");
82 eputs(sigerr[signo]);
83 eputs("\n");
84 quit(3);
85 }
86
87
88 static void
89 sigdie( /* set fatal signal */
90 int signo,
91 const char *msg
92 )
93 {
94 if (signal(signo, onsig) == SIG_IGN)
95 signal(signo, SIG_IGN);
96 sigerr[signo] = msg;
97 }
98
99 /* set input/output format */
100 static void
101 setformat(const char *fmt)
102 {
103 switch (fmt[0]) {
104 case 'f':
105 case 'd':
106 SET_FILE_BINARY(stdin);
107 /* fall through */
108 case 'a':
109 inpfmt = fmt[0];
110 break;
111 default:
112 goto fmterr;
113 }
114 switch (fmt[1]) {
115 case '\0':
116 if (inpfmt == 'a')
117 goto fmterr;
118 outfmt = inpfmt;
119 return;
120 case 'f':
121 case 'd':
122 case 'c':
123 outfmt = fmt[1];
124 break;
125 default:
126 goto fmterr;
127 }
128 if (!fmt[2])
129 return;
130 fmterr:
131 sprintf(errmsg, "Unsupported i/o format: -f%s", fmt);
132 error(USER, errmsg);
133 }
134
135 /* Set default options */
136 static void
137 default_options(void)
138 {
139 rand_samp = 1;
140 dstrsrc = 0.9;
141 directrelay = 3;
142 vspretest = 512;
143 srcsizerat = .2;
144 specthresh = .02;
145 specjitter = 1.;
146 maxdepth = -10;
147 minweight = 2e-3;
148 ambres = 256;
149 ambdiv = 350;
150 ambounce = 1;
151 }
152
153 /* Set overriding options */
154 static void
155 override_options(void)
156 {
157 shadthresh = 0;
158 ambssamp = 0;
159 ambacc = 0;
160 }
161
162 int
163 main(int argc, char *argv[])
164 {
165 #define check(ol,al) if (argv[i][ol] || \
166 badarg(argc-i-1,argv+i+1,al)) \
167 goto badopt
168 #define check_bool(olen,var) switch (argv[i][olen]) { \
169 case '\0': var = !var; break; \
170 case 'y': case 'Y': case 't': case 'T': \
171 case '+': case '1': var = 1; break; \
172 case 'n': case 'N': case 'f': case 'F': \
173 case '-': case '0': var = 0; break; \
174 default: goto badopt; }
175 int force_open = 0;
176 int recover = 0;
177 char *curout = NULL;
178 char *prms = NULL;
179 char *binval = NULL;
180 int bincnt = 0;
181 int rval;
182 int i;
183 /* global program name */
184 progname = argv[0];
185 /* feature check only? */
186 strcat(RFeatureList, RCONTRIB_FEATURES);
187 if (argc > 1 && !strcmp(argv[1], "-features"))
188 return feature_status(argc-2, argv+2);
189 /* initialize calcomp routines early */
190 initfunc();
191 calcontext(RCCONTEXT);
192 /* set rcontrib defaults */
193 default_options();
194 /* option city */
195 for (i = 1; i < argc; i++) {
196 /* expand arguments */
197 while ((rval = expandarg(&argc, &argv, i)) > 0)
198 ;
199 if (rval < 0) {
200 sprintf(errmsg, "cannot expand '%s'", argv[i]);
201 error(SYSTEM, errmsg);
202 }
203 if (argv[i] == NULL || argv[i][0] != '-')
204 break; /* break from options */
205 if (!strcmp(argv[i], "-version")) {
206 puts(VersionID);
207 quit(0);
208 }
209 if (!strcmp(argv[i], "-defaults") ||
210 !strcmp(argv[i], "-help")) {
211 override_options();
212 printdefaults();
213 quit(0);
214 }
215 rval = getrenderopt(argc-i, argv+i);
216 if (rval >= 0) {
217 i += rval;
218 continue;
219 }
220 switch (argv[i][1]) {
221 case 'n': /* number of processes */
222 check(2,"i");
223 nproc = atoi(argv[++i]);
224 if (nproc < 0 && (nproc += RadSimulManager::GetNCores()) <= 0)
225 nproc = 1;
226 break;
227 case 'V': /* output contributions? */
228 rval = myRCmanager.HasFlag(RCcontrib);
229 check_bool(2,rval);
230 myRCmanager.SetFlag(RCcontrib, rval);
231 break;
232 case 'x': /* x resolution */
233 check(2,"i");
234 myRCmanager.xres = atoi(argv[++i]);
235 break;
236 case 'y': /* y resolution */
237 check(2,"i");
238 myRCmanager.yres = atoi(argv[++i]);
239 break;
240 case 'w': /* warnings on/off */
241 rval = (erract[WARNING].pf != NULL);
242 check_bool(2,rval);
243 if (rval) erract[WARNING].pf = wputs;
244 else erract[WARNING].pf = NULL;
245 break;
246 case 'e': /* .cal expression */
247 check(2,"s");
248 scompile(argv[++i], NULL, 0);
249 break;
250 case 'l': /* limit distance */
251 if (argv[i][2] != 'd')
252 goto badopt;
253 rval = myRCmanager.HasFlag(RTlimDist);
254 check_bool(3,rval);
255 myRCmanager.SetFlag(RTlimDist, rval);
256 break;
257 case 'I': /* immed. irradiance */
258 rval = myRCmanager.HasFlag(RTimmIrrad);
259 check_bool(2,rval);
260 myRCmanager.SetFlag(RTimmIrrad, rval);
261 break;
262 case 'f': /* .cal file or force or format */
263 if (!argv[i][2]) {
264 check(2,"s");
265 loadfunc(argv[++i]);
266 break;
267 }
268 if (argv[i][2] == 'o') {
269 check_bool(3,force_open);
270 break;
271 }
272 setformat(argv[i]+2);
273 myRCmanager.SetDataFormat(outfmt);
274 break;
275 case 'o': /* output file */
276 check(2,"s");
277 curout = argv[++i];
278 break;
279 case 'r': /* recover output */
280 check_bool(2,recover);
281 break;
282 case 'p': /* parameter setting(s) */
283 check(2,"s");
284 set_eparams(prms = argv[++i]);
285 break;
286 case 'c': /* sample count */
287 check(2,"i");
288 myRCmanager.accum = atoi(argv[++i]);
289 break;
290 case 'b': /* bin expression/count */
291 if (argv[i][2] == 'n') {
292 check(3,"s");
293 bincnt = (int)(eval(argv[++i]) + .5);
294 break;
295 }
296 check(2,"s");
297 binval = argv[++i];
298 break;
299 case 'm': /* modifier name */
300 check(2,"s");
301 myRCmanager.AddModifier(argv[++i], curout, prms, binval, bincnt);
302 break;
303 case 'M': /* file of modifier names */
304 check(2,"s");
305 myRCmanager.AddModFile(argv[++i], curout, prms, binval, bincnt);
306 break;
307 case 't': /* reporting interval */
308 check(2,"i");
309 report_intvl = atoi(argv[++i]);
310 break;
311 default:
312 goto badopt;
313 }
314 }
315 if (i != argc-1)
316 error(USER, "expected single octree argument");
317
318 override_options(); /* override some option settings */
319
320 if (!myRCmanager.GetOutput()) // check that we have work to do
321 error(USER, "missing required modifier argument");
322 // get ready to rock...
323 if (setspectrsamp(CNDX, WLPART) < 0)
324 error(USER, "unsupported spectral sampling");
325 /* set up signal handling */
326 sigdie(SIGINT, "Interrupt");
327 #ifdef SIGHUP
328 sigdie(SIGHUP, "Hangup");
329 #endif
330 sigdie(SIGTERM, "Terminate");
331 #ifdef SIGPIPE
332 sigdie(SIGPIPE, "Broken pipe");
333 #endif
334 #ifdef SIGALRM
335 sigdie(SIGALRM, "Alarm clock");
336 #endif
337 #ifdef SIGXCPU
338 sigdie(SIGXCPU, "CPU limit exceeded");
339 sigdie(SIGXFSZ, "File size exceeded");
340 #endif
341 #ifdef NICE
342 nice(NICE); /* lower priority */
343 #endif
344 // load octree
345 myRCmanager.LoadOctree(argv[argc-1]);
346 // add to header
347 myRCmanager.AddHeader(argc-1, argv);
348 // prepare output files
349 if (recover)
350 myRCmanager.outOp = RCOrecover;
351 else if (force_open)
352 myRCmanager.outOp = RCOforce;
353 else
354 myRCmanager.outOp = RCOnew;
355 // rval = # rows recovered
356 rval = myRCmanager.PrepOutput();
357 // check if recovered everything
358 if (rval >= myRCmanager.GetRowMax()) {
359 error(WARNING, "nothing left to compute");
360 quit(0);
361 }
362 rxcontrib(rval); /* trace ray contributions (loop) */
363
364 quit(0); /* exit clean */
365
366 badopt:
367 fprintf(stderr,
368 "Usage: %s [-V][-c count][-r][-e expr][-f source][-o ospec][-p p1=V1,p2=V2][-b binv][-bn N] {-m mod | -M file} [rtrace options] octree\n",
369 progname);
370 sprintf(errmsg, "command line error at '%s'", argv[i]);
371 error(USER, errmsg);
372 return(1); /* pro forma return */
373
374 #undef check
375 #undef check_bool
376 }
377
378
379 // skip specified number of bytes, return false if EOF
380 static bool
381 skipBytes(int n2skip)
382 {
383 while (n2skip-- > 0)
384 if (getchar() == EOF)
385 return false;
386 return true;
387 }
388
389
390 // skip specified number of whitespace-separated words, return false if EOF
391 static bool
392 skipWords(int n2skip)
393 {
394 int c;
395
396 while (n2skip-- > 0) {
397 do {
398 c = getchar();
399 } while (isspace(c));
400 do {
401 if (c == EOF) return false;
402 c = getchar();
403 } while (!isspace(c));
404 }
405 return true;
406 }
407
408
409 // read a bundle of myRCmanager.accum ray origins and directions
410 static bool
411 getRayBundle(FVECT *orig_dir = NULL)
412 {
413 int n2go = myRCmanager.accum;
414
415 switch (inpfmt) {
416 case 'a': // ASCII input
417 if (!orig_dir)
418 return skipWords(6*n2go);
419 while (n2go-- > 0) {
420 if (scanf(FVFORMAT, &orig_dir[0][0],
421 &orig_dir[0][1], &orig_dir[0][2]) != 3)
422 return false;
423 if (scanf(FVFORMAT, &orig_dir[1][0],
424 &orig_dir[1][1], &orig_dir[1][2]) != 3)
425 return false;
426 orig_dir += 2;
427 }
428 break;
429 case 'f': // float input
430 if (!orig_dir)
431 return skipBytes(6*sizeof(float)*n2go);
432 #ifdef SMLFLT
433 if (getbinary(orig_dir, sizeof(FVECT), 2*n2go, stdin) != 2*n2go)
434 return false;
435 orig_dir += 2*n2go;
436 #else
437 while (n2go-- > 0) {
438 float fvecs[6];
439 if (getbinary(fvecs, sizeof(fvecs), 1, stdin) != 1)
440 return false;
441 for (int i = 6; i--; )
442 orig_dir[0][i] = fvecs[i];
443 orig_dir += 2;
444 }
445 #endif
446 break;
447 case 'd': // double input
448 if (!orig_dir)
449 return skipBytes(6*sizeof(double)*n2go);
450 #ifndef SMLFLT
451 if (getbinary(orig_dir, sizeof(FVECT), 2*n2go, stdin) != 2*n2go)
452 return false;
453 orig_dir += 2*n2go;
454 #else
455 while (n2go-- > 0) {
456 double dvecs[6];
457 if (getbinary(dvecs, sizeof(dvecs), 1, stdin) != 1)
458 return false;
459 for (int i = 6; i--; )
460 orig_dir[0][i] = dvecs[i];
461 orig_dir += 2;
462 }
463 #endif
464 break;
465 default:
466 error(INTERNAL, "unsupported format in getRayBundle()");
467 return false;
468 }
469 n2go = myRCmanager.accum; // normalize directions
470 while (n2go-- > 0) {
471 orig_dir -= 2;
472 normalize(orig_dir[1]);
473 }
474 return true;
475 }
476
477
478 // Run loop to load data, report progress, etc.
479 void
480 rxcontrib(const int rstart)
481 {
482 const int totRows = myRCmanager.GetRowMax();
483 FVECT * odarr = (FVECT *)emalloc(sizeof(FVECT)*2*myRCmanager.accum);
484 time_t tstart, last_report;
485 int r = 0;
486
487 while (r < rstart) { // skip input rays already done
488 if (!getRayBundle())
489 goto readerr;
490 r++;
491 }
492 if (report_intvl > 0) { // set up reporting
493 if (r > 0) {
494 sprintf(errmsg, "recovered %.2f%% of total\n",
495 100.*r/totRows);
496 eputs(errmsg);
497 }
498 last_report = tstart = time(0);
499 }
500 // start children as requested
501 myRCmanager.SetThreadCount(nproc);
502
503 while (r < totRows) { // loop until done
504 time_t tnow;
505 if (!getRayBundle(odarr))
506 goto readerr;
507 if (myRCmanager.ComputeRecord(odarr) <= 0)
508 return; // error reported, hopefully...
509 r++;
510 if (report_intvl <= 0)
511 continue;
512 if (r == totRows) // need to finish up?
513 myRCmanager.SetThreadCount(1);
514 tnow = time(0);
515 if ((r < totRows) & (tnow < last_report+report_intvl))
516 continue;
517 sprintf(errmsg, "%.2f%% done after %.3f hours\n",
518 100.*myRCmanager.GetRowFinished()/totRows,
519 (1./3600.)*(tnow - tstart));
520 eputs(errmsg);
521 last_report = tnow;
522 }
523 efree(odarr);
524 return;
525 readerr:
526 sprintf(errmsg, "unexpected EOF on standard input (record %d of %d)",
527 r, totRows);
528 error(USER, errmsg);
529 }
530
531
532 void
533 wputs( /* warning output function */
534 const char *s
535 )
536 {
537 if (!erract[WARNING].pf) return;
538 int lasterrno = errno;
539 eputs(s);
540 errno = lasterrno;
541 }
542
543
544 void
545 eputs( /* put string to stderr */
546 const char *s
547 )
548 {
549 static int midline = 0;
550
551 if (!*s)
552 return;
553 if (!midline++) {
554 fputs(progname, stderr);
555 fputs(": ", stderr);
556 }
557 fputs(s, stderr);
558 if (s[strlen(s)-1] == '\n') {
559 fflush(stderr);
560 midline = 0;
561 }
562 }
563
564
565 /* Exit program */
566 void
567 quit(
568 int code
569 )
570 {
571 if (!code && myRCmanager.Ready()) // clean up on normal exit
572 code = myRCmanager.Cleanup();
573
574 exit(code);
575 }