ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/src/rt/rxcmain.cpp
Revision: 2.20
Committed: Fri Oct 17 01:15:53 2025 UTC (3 days, 17 hours ago) by greg
Branch: MAIN
CVS Tags: HEAD
Changes since 2.19: +9 -3 lines
Log Message:
perf(rxcontrib): Added handshake to ensure each row is complete in children

File Contents

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