ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rxcmain.cpp
Revision: 2.3
Committed: Wed Oct 30 02:22:23 2024 UTC (6 months ago) by greg
Branch: MAIN
Changes since 2.2: +4 -1 lines
Log Message:
fix(rcontrib,rxcontrib): Eliminated spurious warning messages

File Contents

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