ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.63
Committed: Wed Dec 16 03:30:50 2009 UTC (14 years, 4 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R0
Changes since 2.62: +6 -5 lines
Log Message:
Bug fix for zero-direction rays with multiprocessing

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rtrace.c,v 2.62 2009/12/14 07:31:37 greg Exp $";
3 #endif
4 /*
5 * rtrace.c - program and variables for individual ray tracing.
6 */
7
8 #include "copyright.h"
9
10 /*
11 * Input is in the form:
12 *
13 * xorg yorg zorg xdir ydir zdir
14 *
15 * The direction need not be normalized. Output is flexible.
16 * If the direction vector is (0,0,0), then the output is flushed.
17 * All values default to ascii representation of real
18 * numbers. Binary representations can be selected
19 * with '-ff' for float or '-fd' for double. By default,
20 * radiance is computed. The '-i' or '-I' options indicate that
21 * irradiance values are desired.
22 */
23
24 #include <time.h>
25
26 #include "platform.h"
27 #include "ray.h"
28 #include "ambient.h"
29 #include "source.h"
30 #include "otypes.h"
31 #include "resolu.h"
32 #include "random.h"
33
34 extern int inform; /* input format */
35 extern int outform; /* output format */
36 extern char *outvals; /* output values */
37
38 extern int imm_irrad; /* compute immediate irradiance? */
39 extern int lim_dist; /* limit distance? */
40
41 extern char *tralist[]; /* list of modifers to trace (or no) */
42 extern int traincl; /* include == 1, exclude == 0 */
43
44 extern int hresolu; /* horizontal resolution */
45 extern int vresolu; /* vertical resolution */
46
47 static int castonly = 0;
48
49 #ifndef MAXTSET
50 #define MAXTSET 8191 /* maximum number in trace set */
51 #endif
52 OBJECT traset[MAXTSET+1]={0}; /* trace include/exclude set */
53
54 static RAY thisray; /* for our convenience */
55
56 typedef void putf_t(double v);
57 static putf_t puta, putd, putf;
58
59 typedef void oputf_t(RAY *r);
60 static oputf_t oputo, oputd, oputv, oputV, oputl, oputL, oputc, oputp,
61 oputn, oputN, oputs, oputw, oputW, oputm, oputM, oputtilde;
62
63 static void setoutput(char *vs);
64 extern void tranotify(OBJECT obj);
65 static void bogusray(void);
66 static void raycast(RAY *r);
67 static void rayirrad(RAY *r);
68 static void rtcompute(FVECT org, FVECT dir, double dmax);
69 static int printvals(RAY *r);
70 static int getvec(FVECT vec, int fmt, FILE *fp);
71 static void tabin(RAY *r);
72 static void ourtrace(RAY *r);
73
74 static oputf_t *ray_out[16], *every_out[16];
75 static putf_t *putreal;
76
77
78 void
79 quit( /* quit program */
80 int code
81 )
82 {
83 if (ray_pnprocs > 0) /* close children if any */
84 ray_pclose(0);
85 #ifndef NON_POSIX
86 else if (!ray_pnprocs) {
87 headclean(); /* delete header file */
88 pfclean(); /* clean up persist files */
89 }
90 #endif
91 exit(code);
92 }
93
94
95 char *
96 formstr( /* return format identifier */
97 int f
98 )
99 {
100 switch (f) {
101 case 'a': return("ascii");
102 case 'f': return("float");
103 case 'd': return("double");
104 case 'c': return(COLRFMT);
105 }
106 return("unknown");
107 }
108
109
110 extern void
111 rtrace( /* trace rays from file */
112 char *fname,
113 int nproc
114 )
115 {
116 unsigned long vcount = (hresolu > 1) ? (unsigned long)hresolu*vresolu
117 : vresolu;
118 long nextflush = hresolu;
119 FILE *fp;
120 double d;
121 FVECT orig, direc;
122 /* set up input */
123 if (fname == NULL)
124 fp = stdin;
125 else if ((fp = fopen(fname, "r")) == NULL) {
126 sprintf(errmsg, "cannot open input file \"%s\"", fname);
127 error(SYSTEM, errmsg);
128 }
129 if (inform != 'a')
130 SET_FILE_BINARY(fp);
131 /* set up output */
132 setoutput(outvals);
133 if (imm_irrad)
134 castonly = 0;
135 else if (castonly)
136 nproc = 1; /* don't bother multiprocessing */
137 switch (outform) {
138 case 'a': putreal = puta; break;
139 case 'f': putreal = putf; break;
140 case 'd': putreal = putd; break;
141 case 'c':
142 if (strcmp(outvals, "v"))
143 error(USER, "color format with value output only");
144 break;
145 default:
146 error(CONSISTENCY, "botched output format");
147 }
148 if (nproc > 1) { /* start multiprocessing */
149 ray_popen(nproc);
150 ray_fifo_out = printvals;
151 }
152 if (hresolu > 0) {
153 if (vresolu > 0)
154 fprtresolu(hresolu, vresolu, stdout);
155 fflush(stdout);
156 }
157 /* process file */
158 while (getvec(orig, inform, fp) == 0 &&
159 getvec(direc, inform, fp) == 0) {
160
161 d = normalize(direc);
162 if (d == 0.0) { /* zero ==> flush */
163 if (--nextflush <= 0 || !vcount) {
164 if (nproc > 1 && ray_fifo_flush() < 0)
165 error(USER, "lost children");
166 bogusray();
167 fflush(stdout);
168 nextflush = hresolu;
169 } else
170 bogusray();
171 } else { /* compute and print */
172 rtcompute(orig, direc, lim_dist ? d : 0.0);
173 /* flush if time */
174 if (!--nextflush) {
175 if (nproc > 1 && ray_fifo_flush() < 0)
176 error(USER, "lost children");
177 fflush(stdout);
178 nextflush = hresolu;
179 }
180 }
181 if (ferror(stdout))
182 error(SYSTEM, "write error");
183 if (vcount && !--vcount) /* check for end */
184 break;
185 }
186 if (nproc > 1) { /* clean up children */
187 if (ray_fifo_flush() < 0)
188 error(USER, "unable to complete processing");
189 ray_pclose(0);
190 }
191 if (fflush(stdout) < 0)
192 error(SYSTEM, "write error");
193 if (vcount)
194 error(USER, "unexpected EOF on input");
195 if (fname != NULL)
196 fclose(fp);
197 }
198
199
200 static void
201 trace_sources(void) /* trace rays to light sources, also */
202 {
203 int sn;
204
205 for (sn = 0; sn < nsources; sn++)
206 source[sn].sflags |= SFOLLOW;
207 }
208
209
210 static void
211 setoutput( /* set up output tables */
212 char *vs
213 )
214 {
215 oputf_t **table = ray_out;
216
217 castonly = 1;
218 while (*vs)
219 switch (*vs++) {
220 case 'T': /* trace sources */
221 if (!*vs) break;
222 trace_sources();
223 /* fall through */
224 case 't': /* trace */
225 if (!*vs) break;
226 *table = NULL;
227 table = every_out;
228 trace = ourtrace;
229 castonly = 0;
230 break;
231 case 'o': /* origin */
232 *table++ = oputo;
233 break;
234 case 'd': /* direction */
235 *table++ = oputd;
236 break;
237 case 'v': /* value */
238 *table++ = oputv;
239 castonly = 0;
240 break;
241 case 'V': /* contribution */
242 *table++ = oputV;
243 if (ambounce > 0 && (ambacc > FTINY || ambssamp > 0))
244 error(WARNING,
245 "-otV accuracy depends on -aa 0 -as 0");
246 break;
247 case 'l': /* effective distance */
248 *table++ = oputl;
249 castonly = 0;
250 break;
251 case 'c': /* local coordinates */
252 *table++ = oputc;
253 break;
254 case 'L': /* single ray length */
255 *table++ = oputL;
256 break;
257 case 'p': /* point */
258 *table++ = oputp;
259 break;
260 case 'n': /* perturbed normal */
261 *table++ = oputn;
262 castonly = 0;
263 break;
264 case 'N': /* unperturbed normal */
265 *table++ = oputN;
266 break;
267 case 's': /* surface */
268 *table++ = oputs;
269 break;
270 case 'w': /* weight */
271 *table++ = oputw;
272 break;
273 case 'W': /* coefficient */
274 *table++ = oputW;
275 if (ambounce > 0 && (ambacc > FTINY || ambssamp > 0))
276 error(WARNING,
277 "-otW accuracy depends on -aa 0 -as 0");
278 break;
279 case 'm': /* modifier */
280 *table++ = oputm;
281 break;
282 case 'M': /* material */
283 *table++ = oputM;
284 break;
285 case '~': /* tilde */
286 *table++ = oputtilde;
287 break;
288 }
289 *table = NULL;
290 }
291
292
293 static void
294 bogusray(void) /* print out empty record */
295 {
296 thisray.rorg[0] = thisray.rorg[1] = thisray.rorg[2] =
297 thisray.rdir[0] = thisray.rdir[1] = thisray.rdir[2] = 0.0;
298 thisray.rmax = 0.0;
299 rayorigin(&thisray, PRIMARY, NULL, NULL);
300 printvals(&thisray);
301 }
302
303
304 static void
305 raycast( /* compute first ray intersection only */
306 RAY *r
307 )
308 {
309 if (!localhit(r, &thescene)) {
310 if (r->ro == &Aftplane) { /* clipped */
311 r->ro = NULL;
312 r->rot = FHUGE;
313 } else
314 sourcehit(r);
315 }
316 }
317
318
319 static void
320 rayirrad( /* compute irradiance rather than radiance */
321 RAY *r
322 )
323 {
324 void (*old_revf)(RAY *) = r->revf;
325
326 r->rot = 1e-5; /* pretend we hit surface */
327 VSUM(r->rop, r->rorg, r->rdir, r->rot);
328 r->ron[0] = -r->rdir[0];
329 r->ron[1] = -r->rdir[1];
330 r->ron[2] = -r->rdir[2];
331 r->rod = 1.0;
332 /* compute result */
333 r->revf = raytrace;
334 (*ofun[Lamb.otype].funp)(&Lamb, r);
335 r->revf = old_revf;
336 }
337
338
339 static void
340 rtcompute( /* compute and print ray value(s) */
341 FVECT org,
342 FVECT dir,
343 double dmax
344 )
345 {
346 /* set up ray */
347 rayorigin(&thisray, PRIMARY, NULL, NULL);
348 if (imm_irrad) {
349 VSUM(thisray.rorg, org, dir, 1.1e-4);
350 thisray.rdir[0] = -dir[0];
351 thisray.rdir[1] = -dir[1];
352 thisray.rdir[2] = -dir[2];
353 thisray.rmax = 0.0;
354 thisray.revf = rayirrad;
355 } else {
356 VCOPY(thisray.rorg, org);
357 VCOPY(thisray.rdir, dir);
358 thisray.rmax = dmax;
359 if (castonly)
360 thisray.revf = raycast;
361 }
362 if (ray_pnprocs > 1) { /* multiprocessing FIFO? */
363 if (ray_fifo_in(&thisray) < 0)
364 error(USER, "lost children");
365 return;
366 }
367 samplendx++; /* else do it ourselves */
368 rayvalue(&thisray);
369 printvals(&thisray);
370 }
371
372
373 static int
374 printvals( /* print requested ray values */
375 RAY *r
376 )
377 {
378 oputf_t **tp;
379
380 if (ray_out[0] == NULL)
381 return(0);
382 for (tp = ray_out; *tp != NULL; tp++)
383 (**tp)(r);
384 if (outform == 'a')
385 putchar('\n');
386 return(1);
387 }
388
389
390 static int
391 getvec( /* get a vector from fp */
392 FVECT vec,
393 int fmt,
394 FILE *fp
395 )
396 {
397 static float vf[3];
398 static double vd[3];
399 char buf[32];
400 int i;
401
402 switch (fmt) {
403 case 'a': /* ascii */
404 for (i = 0; i < 3; i++) {
405 if (fgetword(buf, sizeof(buf), fp) == NULL ||
406 !isflt(buf))
407 return(-1);
408 vec[i] = atof(buf);
409 }
410 break;
411 case 'f': /* binary float */
412 if (fread((char *)vf, sizeof(float), 3, fp) != 3)
413 return(-1);
414 VCOPY(vec, vf);
415 break;
416 case 'd': /* binary double */
417 if (fread((char *)vd, sizeof(double), 3, fp) != 3)
418 return(-1);
419 VCOPY(vec, vd);
420 break;
421 default:
422 error(CONSISTENCY, "botched input format");
423 }
424 return(0);
425 }
426
427
428 void
429 tranotify( /* record new modifier */
430 OBJECT obj
431 )
432 {
433 static int hitlimit = 0;
434 OBJREC *o = objptr(obj);
435 char **tralp;
436
437 if (obj == OVOID) { /* starting over */
438 traset[0] = 0;
439 hitlimit = 0;
440 return;
441 }
442 if (hitlimit || !ismodifier(o->otype))
443 return;
444 for (tralp = tralist; *tralp != NULL; tralp++)
445 if (!strcmp(o->oname, *tralp)) {
446 if (traset[0] >= MAXTSET) {
447 error(WARNING, "too many modifiers in trace list");
448 hitlimit++;
449 return; /* should this be fatal? */
450 }
451 insertelem(traset, obj);
452 return;
453 }
454 }
455
456
457 static void
458 ourtrace( /* print ray values */
459 RAY *r
460 )
461 {
462 oputf_t **tp;
463
464 if (every_out[0] == NULL)
465 return;
466 if (r->ro == NULL) {
467 if (traincl == 1)
468 return;
469 } else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
470 return;
471 tabin(r);
472 for (tp = every_out; *tp != NULL; tp++)
473 (**tp)(r);
474 if (outform == 'a')
475 putchar('\n');
476 }
477
478
479 static void
480 tabin( /* tab in appropriate amount */
481 RAY *r
482 )
483 {
484 const RAY *rp;
485
486 for (rp = r->parent; rp != NULL; rp = rp->parent)
487 putchar('\t');
488 }
489
490
491 static void
492 oputo( /* print origin */
493 RAY *r
494 )
495 {
496 (*putreal)(r->rorg[0]);
497 (*putreal)(r->rorg[1]);
498 (*putreal)(r->rorg[2]);
499 }
500
501
502 static void
503 oputd( /* print direction */
504 RAY *r
505 )
506 {
507 (*putreal)(r->rdir[0]);
508 (*putreal)(r->rdir[1]);
509 (*putreal)(r->rdir[2]);
510 }
511
512
513 static void
514 oputv( /* print value */
515 RAY *r
516 )
517 {
518 if (outform == 'c') {
519 COLR cout;
520 setcolr(cout, colval(r->rcol,RED),
521 colval(r->rcol,GRN),
522 colval(r->rcol,BLU));
523 fwrite((char *)cout, sizeof(cout), 1, stdout);
524 return;
525 }
526 (*putreal)(colval(r->rcol,RED));
527 (*putreal)(colval(r->rcol,GRN));
528 (*putreal)(colval(r->rcol,BLU));
529 }
530
531
532 static void
533 oputV( /* print value contribution */
534 RAY *r
535 )
536 {
537 double contr[3];
538
539 raycontrib(contr, r, PRIMARY);
540 multcolor(contr, r->rcol);
541 (*putreal)(contr[RED]);
542 (*putreal)(contr[GRN]);
543 (*putreal)(contr[BLU]);
544 }
545
546
547 static void
548 oputl( /* print effective distance */
549 RAY *r
550 )
551 {
552 (*putreal)(r->rt);
553 }
554
555
556 static void
557 oputL( /* print single ray length */
558 RAY *r
559 )
560 {
561 (*putreal)(r->rot);
562 }
563
564
565 static void
566 oputc( /* print local coordinates */
567 RAY *r
568 )
569 {
570 (*putreal)(r->uv[0]);
571 (*putreal)(r->uv[1]);
572 }
573
574
575 static void
576 oputp( /* print point */
577 RAY *r
578 )
579 {
580 if (r->rot < FHUGE) {
581 (*putreal)(r->rop[0]);
582 (*putreal)(r->rop[1]);
583 (*putreal)(r->rop[2]);
584 } else {
585 (*putreal)(0.0);
586 (*putreal)(0.0);
587 (*putreal)(0.0);
588 }
589 }
590
591
592 static void
593 oputN( /* print unperturbed normal */
594 RAY *r
595 )
596 {
597 if (r->rot < FHUGE) {
598 (*putreal)(r->ron[0]);
599 (*putreal)(r->ron[1]);
600 (*putreal)(r->ron[2]);
601 } else {
602 (*putreal)(0.0);
603 (*putreal)(0.0);
604 (*putreal)(0.0);
605 }
606 }
607
608
609 static void
610 oputn( /* print perturbed normal */
611 RAY *r
612 )
613 {
614 FVECT pnorm;
615
616 if (r->rot >= FHUGE) {
617 (*putreal)(0.0);
618 (*putreal)(0.0);
619 (*putreal)(0.0);
620 return;
621 }
622 raynormal(pnorm, r);
623 (*putreal)(pnorm[0]);
624 (*putreal)(pnorm[1]);
625 (*putreal)(pnorm[2]);
626 }
627
628
629 static void
630 oputs( /* print name */
631 RAY *r
632 )
633 {
634 if (r->ro != NULL)
635 fputs(r->ro->oname, stdout);
636 else
637 putchar('*');
638 putchar('\t');
639 }
640
641
642 static void
643 oputw( /* print weight */
644 RAY *r
645 )
646 {
647 (*putreal)(r->rweight);
648 }
649
650
651 static void
652 oputW( /* print coefficient */
653 RAY *r
654 )
655 {
656 double contr[3];
657
658 raycontrib(contr, r, PRIMARY);
659 (*putreal)(contr[RED]);
660 (*putreal)(contr[GRN]);
661 (*putreal)(contr[BLU]);
662 }
663
664
665 static void
666 oputm( /* print modifier */
667 RAY *r
668 )
669 {
670 if (r->ro != NULL)
671 if (r->ro->omod != OVOID)
672 fputs(objptr(r->ro->omod)->oname, stdout);
673 else
674 fputs(VOIDID, stdout);
675 else
676 putchar('*');
677 putchar('\t');
678 }
679
680
681 static void
682 oputM( /* print material */
683 RAY *r
684 )
685 {
686 OBJREC *mat;
687
688 if (r->ro != NULL) {
689 if ((mat = findmaterial(r->ro)) != NULL)
690 fputs(mat->oname, stdout);
691 else
692 fputs(VOIDID, stdout);
693 } else
694 putchar('*');
695 putchar('\t');
696 }
697
698
699 static void
700 oputtilde( /* output tilde (spacer) */
701 RAY *r
702 )
703 {
704 fputs("~\t", stdout);
705 }
706
707
708 static void
709 puta( /* print ascii value */
710 double v
711 )
712 {
713 printf("%e\t", v);
714 }
715
716
717 static void
718 putd(v) /* print binary double */
719 double v;
720 {
721 fwrite((char *)&v, sizeof(v), 1, stdout);
722 }
723
724
725 static void
726 putf(v) /* print binary float */
727 double v;
728 {
729 float f = v;
730
731 fwrite((char *)&f, sizeof(f), 1, stdout);
732 }