ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/gen/mkillum.c
Revision: 2.45
Committed: Tue Apr 22 04:45:25 2025 UTC (11 days, 14 hours ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.44: +5 -2 lines
Log Message:
fix: Added check to return value of pclose() in several tools

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: mkillum.c,v 2.44 2023/02/06 22:40:21 greg Exp $";
3 #endif
4 /*
5 * Make illum sources for optimizing rendering process
6 */
7
8 #include <signal.h>
9 #include <ctype.h>
10
11 #include "paths.h" /* win_popen() */
12 #include "mkillum.h"
13
14 /* default parameters */
15 #define SAMPDENS 48 /* points per projected steradian */
16 #define NSAMPS 32 /* samples per point */
17 #define DFLMAT "illum_mat" /* material name */
18 #define DFLDAT "illum" /* data file name */
19 /* selection options */
20 #define S_NONE 0 /* select none */
21 #define S_ELEM 1 /* select specified element */
22 #define S_COMPL 2 /* select all but element */
23 #define S_ALL 3 /* select all */
24
25 struct illum_args thisillum = { /* our illum and default values */
26 0,
27 DFLMAT,
28 DFLDAT,
29 0,
30 VOIDID,
31 SAMPDENS,
32 NSAMPS,
33 0.,
34 };
35
36 char matcheck[MAXSTR]; /* current material to include or exclude */
37 int matselect = S_ALL; /* selection criterion */
38
39 int gargc; /* global argc */
40 char **gargv; /* global argv */
41
42 int doneheader = 0; /* printed header yet? */
43 #define checkhead() if (!doneheader++) printhead(gargc,gargv)
44
45 int warnings = 1; /* print warnings? */
46
47 void init(char *octnm, int np);
48 void filter(FILE *infp, char *name);
49 void xoptions(char *s, char *nm);
50 void printopts(void);
51 void printhead(int ac, char **av);
52 void xobject(FILE *fp, char *nm);
53
54
55 int
56 main( /* compute illum distributions using rtrace */
57 int argc,
58 char *argv[]
59 )
60 {
61 int nprocs = 1;
62 FILE *fp;
63 int rval;
64 int i;
65 /* set global arguments */
66 gargv = argv;
67 progname = gargv[0];
68 /* set up rendering defaults */
69 dstrsrc = 0.5;
70 directrelay = 3;
71 ambounce = 2;
72 /* get options from command line */
73 for (i = 1; i < argc; i++) {
74 while ((rval = expandarg(&argc, &argv, i)) > 0)
75 ;
76 if (rval < 0) {
77 sprintf(errmsg, "cannot expand '%s'", argv[i]);
78 error(SYSTEM, errmsg);
79 }
80 if (argv[i][0] != '-')
81 break;
82 if (!strcmp(argv[i], "-w")) {
83 warnings = 0;
84 continue;
85 }
86 if (!strcmp(argv[i], "-n")) {
87 nprocs = atoi(argv[++i]);
88 if (nprocs <= 0)
89 error(USER, "illegal number of processes");
90 continue;
91 }
92 if (!strcmp(argv[i], "-defaults")) {
93 printopts();
94 print_rdefaults();
95 quit(0);
96 }
97 rval = getrenderopt(argc-i, argv+i);
98 if (rval < 0) {
99 sprintf(errmsg, "bad render option at '%s'", argv[i]);
100 error(USER, errmsg);
101 }
102 i += rval;
103 }
104 gargc = ++i;
105 /* add "mandatory" render options */
106 do_irrad = 0;
107 if (gargc > argc || argv[gargc-1][0] == '-')
108 error(USER, "missing octree argument");
109 /* else initialize and run our calculation */
110 init(argv[gargc-1], nprocs);
111 if (gargc < argc) {
112 if (gargc == argc-1 || argv[gargc][0] != '<' || argv[gargc][1])
113 error(USER, "use '< file1 file2 ..' for multiple inputs");
114 for (i = gargc+1; i < argc; i++) {
115 if ((fp = fopen(argv[i], "r")) == NULL) {
116 sprintf(errmsg,
117 "cannot open scene file \"%s\"", argv[i]);
118 error(SYSTEM, errmsg);
119 }
120 filter(fp, argv[i]);
121 fclose(fp);
122 }
123 } else
124 filter(stdin, "standard input");
125 quit(0);
126 return 0; /* pro forma return */
127 }
128
129
130 void
131 init(char *octnm, int np) /* start rendering process(es) */
132 {
133 /* set up signal handling */
134 signal(SIGINT, quit);
135 #ifdef SIGHUP
136 signal(SIGHUP, quit);
137 #endif
138 #ifdef SIGTERM
139 signal(SIGTERM, quit);
140 #endif
141 #ifdef SIGPIPE
142 signal(SIGPIPE, quit);
143 #endif
144 /* start rendering process(es) */
145 ray_pinit(octnm, np);
146 }
147
148
149 void
150 eputs( /* put string to stderr */
151 const char *s
152 )
153 {
154 static int midline = 0;
155
156 if (!*s) return;
157 if (!midline) {
158 fputs(progname, stderr);
159 fputs(": ", stderr);
160 }
161 fputs(s, stderr);
162 midline = s[strlen(s)-1] != '\n';
163 }
164
165
166 void
167 wputs(const char *s) /* print warning if enabled */
168 {
169 if (warnings)
170 eputs(s);
171 }
172
173
174 void
175 quit(int ec) /* make sure exit is called */
176 {
177 if (ray_pnprocs > 0) /* close children if any */
178 ray_pclose(0);
179 else if (ray_pnprocs < 0)
180 _exit(ec); /* avoid flush() in child */
181 exit(ec);
182 }
183
184
185 void
186 filter( /* process stream */
187 FILE *infp,
188 char *name
189 )
190 {
191 char buf[512];
192 FILE *pfp;
193 int c;
194
195 while ((c = getc(infp)) != EOF) {
196 if (isspace(c))
197 continue;
198 if (c == '#') { /* comment/options */
199 buf[0] = c;
200 fgets(buf+1, sizeof(buf)-1, infp);
201 xoptions(buf, name);
202 } else if (c == '!') { /* command */
203 buf[0] = c;
204 fgetline(buf+1, sizeof(buf)-1, infp);
205 if ((pfp = popen(buf+1, "r")) == NULL) {
206 sprintf(errmsg, "cannot execute \"%s\"", buf);
207 error(SYSTEM, errmsg);
208 }
209 filter(pfp, buf);
210 if (pclose(pfp) != 0) {
211 sprintf(errmsg, "bad status from \"%s\"", buf);
212 error(USER, errmsg);
213 }
214 } else { /* object */
215 ungetc(c, infp);
216 xobject(infp, name);
217 }
218 }
219 }
220
221
222 void
223 xoptions( /* process options in string s */
224 char *s,
225 char *nm
226 )
227 {
228 char buf[64];
229 int negax;
230 int nerrs = 0;
231 char *cp;
232
233 if (strncmp(s, "#@mkillum", 9) || !isspace(s[9])) {
234 fputs(s, stdout); /* not for us */
235 return;
236 }
237 cp = s+10;
238 while (*cp) {
239 switch (*cp) {
240 case ' ':
241 case '\t':
242 case '\n':
243 case '\r':
244 case '\f':
245 cp++;
246 continue;
247 case 'm': /* material name */
248 if (*++cp != '=')
249 break;
250 if (!*++cp || isspace(*cp))
251 break;
252 atos(thisillum.matname, MAXSTR, cp);
253 cp = sskip(cp);
254 if (!(thisillum.flags & IL_DATCLB)) {
255 strcpy(thisillum.datafile, thisillum.matname);
256 thisillum.dfnum = 0;
257 }
258 continue;
259 case 'f': /* data file name */
260 if (*++cp != '=')
261 break;
262 if (!*++cp || isspace(*cp)) {
263 strcpy(thisillum.datafile,thisillum.matname);
264 thisillum.dfnum = 0;
265 thisillum.flags &= ~IL_DATCLB;
266 continue;
267 }
268 atos(thisillum.datafile, MAXSTR, cp);
269 cp = sskip(cp);
270 thisillum.dfnum = 0;
271 thisillum.flags |= IL_DATCLB;
272 continue;
273 case 'i': /* include material */
274 case 'e': /* exclude material */
275 if (cp[1] != '=')
276 break;
277 matselect = (*cp == 'i') ? S_ELEM : S_COMPL;
278 cp += 2;
279 atos(matcheck, MAXSTR, cp);
280 cp = sskip(cp);
281 continue;
282 case 'a': /* use everything */
283 cp = sskip(cp);
284 matselect = S_ALL;
285 continue;
286 case 'n': /* use nothing (passive) */
287 cp = sskip(cp);
288 matselect = S_NONE;
289 continue;
290 case 'c': /* color calculation */
291 if (*++cp != '=')
292 break;
293 switch (*++cp) {
294 case 'a': /* average */
295 thisillum.flags = (thisillum.flags|IL_COLAVG)
296 & ~IL_COLDST;
297 break;
298 case 'd': /* distribution */
299 thisillum.flags |= (IL_COLDST|IL_COLAVG);
300 break;
301 case 'n': /* none */
302 thisillum.flags &= ~(IL_COLAVG|IL_COLDST);
303 break;
304 default:
305 goto opterr;
306 }
307 cp = sskip(cp);
308 continue;
309 case 'd': /* sample density */
310 if (*++cp != '=')
311 break;
312 if (!*++cp || isspace(*cp))
313 continue;
314 if (isintd(cp, " \t\n\r")) {
315 thisillum.sampdens = atoi(cp);
316 } else {
317 error(WARNING, "direct BSDF input unsupported");
318 goto opterr;
319 }
320 cp = sskip(cp);
321 continue;
322 case 's': /* surface super-samples */
323 if (*++cp != '=')
324 break;
325 if (!isintd(++cp, " \t\n\r"))
326 break;
327 thisillum.nsamps = atoi(cp);
328 cp = sskip(cp);
329 continue;
330 case 'l': /* light sources */
331 cp++;
332 if (*cp == '+')
333 thisillum.flags |= IL_LIGHT;
334 else if (*cp == '-')
335 thisillum.flags &= ~IL_LIGHT;
336 else
337 break;
338 cp++;
339 continue;
340 case 'b': /* brightness */
341 if (*++cp != '=')
342 break;
343 if (!isfltd(++cp, " \t\n\r"))
344 break;
345 thisillum.minbrt = atof(cp);
346 if (thisillum.minbrt < 0.)
347 thisillum.minbrt = 0.;
348 cp = sskip(cp);
349 continue;
350 case 'o': /* output file */
351 if (*++cp != '=')
352 break;
353 if (!*++cp || isspace(*cp))
354 break;
355 atos(buf, sizeof(buf), cp);
356 cp = sskip(cp);
357 if (freopen(buf, "w", stdout) == NULL) {
358 sprintf(errmsg,
359 "cannot open output file \"%s\"", buf);
360 error(SYSTEM, errmsg);
361 }
362 doneheader = 0;
363 continue;
364 case '!': /* processed file! */
365 sprintf(errmsg, "(%s): already processed!", nm);
366 error(WARNING, errmsg);
367 matselect = S_NONE;
368 return;
369 }
370 opterr: /* skip faulty option */
371 while (*cp && !isspace(*cp))
372 cp++;
373 nerrs++;
374 }
375 /* print header? */
376 checkhead();
377 /* issue warnings? */
378 if (nerrs) {
379 sprintf(errmsg, "(%s): %d error(s) in option line:",
380 nm, nerrs);
381 error(WARNING, errmsg);
382 wputs(s);
383 printf("# %s: the following option line has %d error(s):\n",
384 progname, nerrs);
385 }
386 /* print pure comment */
387 printf("# %s", s+2);
388 }
389
390 void
391 printopts(void) /* print out option default values */
392 {
393 printf("m=%-15s\t\t# material name\n", thisillum.matname);
394 printf("f=%-15s\t\t# data file name\n", thisillum.datafile);
395 if (thisillum.flags & IL_COLAVG)
396 if (thisillum.flags & IL_COLDST)
397 printf("c=d\t\t\t\t# color distribution\n");
398 else
399 printf("c=a\t\t\t\t# color average\n");
400 else
401 printf("c=n\t\t\t\t# color none\n");
402 if (thisillum.flags & IL_LIGHT)
403 printf("l+\t\t\t\t# light type on\n");
404 else
405 printf("l-\t\t\t\t# light type off\n");
406 printf("d=%d\t\t\t\t# density of directions\n", thisillum.sampdens);
407 printf("s=%d\t\t\t\t# samples per direction\n", thisillum.nsamps);
408 printf("b=%f\t\t\t# minimum average brightness\n", thisillum.minbrt);
409 }
410
411
412 void
413 printhead( /* print out header */
414 int ac,
415 char **av
416 )
417 {
418 putchar('#');
419 while (ac-- > 0) {
420 putchar(' ');
421 fputs(*av++, stdout);
422 }
423 fputs("\n#@mkillum !\n", stdout);
424 }
425
426
427 void
428 xobject( /* translate an object from fp */
429 FILE *fp,
430 char *nm
431 )
432 {
433 OBJREC thisobj;
434 char str[MAXSTR];
435 int doit;
436 /* read the object */
437 if (fgetword(thisillum.altmat, MAXSTR, fp) == NULL)
438 goto readerr;
439 if (fgetword(str, MAXSTR, fp) == NULL)
440 goto readerr;
441 /* is it an alias? */
442 if (!strcmp(str, ALIASKEY)) {
443 if (fgetword(str, MAXSTR, fp) == NULL)
444 goto readerr;
445 printf("\n%s %s %s", thisillum.altmat, ALIASKEY, str);
446 if (fgetword(str, MAXSTR, fp) == NULL)
447 goto readerr;
448 printf("\t%s\n", str);
449 return;
450 }
451 thisobj.omod = OVOID; /* unused field */
452 if ((thisobj.otype = otype(str)) < 0) {
453 sprintf(errmsg, "(%s): unknown type \"%s\"", nm, str);
454 error(USER, errmsg);
455 }
456 if (fgetword(str, MAXSTR, fp) == NULL)
457 goto readerr;
458 thisobj.oname = str;
459 if (readfargs(&thisobj.oargs, fp) != 1)
460 goto readerr;
461 thisobj.os = NULL;
462 /* check for translation */
463 switch (matselect) {
464 case S_NONE:
465 doit = 0;
466 break;
467 case S_ALL:
468 doit = 1;
469 break;
470 case S_ELEM:
471 doit = !strcmp(thisillum.altmat, matcheck);
472 break;
473 case S_COMPL:
474 doit = strcmp(thisillum.altmat, matcheck);
475 break;
476 }
477 doit = doit && issurface(thisobj.otype);
478 /* print header? */
479 checkhead();
480 /* process object */
481 if (doit)
482 switch (thisobj.otype) {
483 case OBJ_FACE:
484 my_face(&thisobj, &thisillum, nm);
485 break;
486 case OBJ_SPHERE:
487 my_sphere(&thisobj, &thisillum, nm);
488 break;
489 case OBJ_RING:
490 my_ring(&thisobj, &thisillum, nm);
491 break;
492 default:
493 my_default(&thisobj, &thisillum, nm);
494 break;
495 }
496 else
497 printobj(thisillum.altmat, &thisobj);
498 /* free arguments */
499 freefargs(&thisobj.oargs);
500 return;
501 readerr:
502 sprintf(errmsg, "(%s): error reading input", nm);
503 error(USER, errmsg);
504 }