ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.62
Committed: Mon Dec 14 07:31:37 2009 UTC (14 years, 4 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.61: +5 -2 lines
Log Message:
Precautionary change to rayirrad()

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rtrace.c,v 2.61 2009/12/14 04:49:16 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 (nproc > 1 && ray_fifo_flush() < 0)
164 error(USER, "lost children");
165 bogusray();
166 if (--nextflush <= 0 || !vcount) {
167 fflush(stdout);
168 nextflush = hresolu;
169 }
170 } else { /* compute and print */
171 rtcompute(orig, direc, lim_dist ? d : 0.0);
172 /* flush if time */
173 if (!--nextflush) {
174 if (nproc > 1 && ray_fifo_flush() < 0)
175 error(USER, "lost children");
176 fflush(stdout);
177 nextflush = hresolu;
178 }
179 }
180 if (ferror(stdout))
181 error(SYSTEM, "write error");
182 if (vcount && !--vcount) /* check for end */
183 break;
184 }
185 if (nproc > 1) { /* clean up children */
186 if (ray_fifo_flush() < 0)
187 error(USER, "unable to complete processing");
188 ray_pclose(0);
189 }
190 if (fflush(stdout) < 0)
191 error(SYSTEM, "write error");
192 if (vcount)
193 error(USER, "unexpected EOF on input");
194 if (fname != NULL)
195 fclose(fp);
196 }
197
198
199 static void
200 trace_sources(void) /* trace rays to light sources, also */
201 {
202 int sn;
203
204 for (sn = 0; sn < nsources; sn++)
205 source[sn].sflags |= SFOLLOW;
206 }
207
208
209 static void
210 setoutput( /* set up output tables */
211 char *vs
212 )
213 {
214 oputf_t **table = ray_out;
215
216 castonly = 1;
217 while (*vs)
218 switch (*vs++) {
219 case 'T': /* trace sources */
220 if (!*vs) break;
221 trace_sources();
222 /* fall through */
223 case 't': /* trace */
224 if (!*vs) break;
225 *table = NULL;
226 table = every_out;
227 trace = ourtrace;
228 castonly = 0;
229 break;
230 case 'o': /* origin */
231 *table++ = oputo;
232 break;
233 case 'd': /* direction */
234 *table++ = oputd;
235 break;
236 case 'v': /* value */
237 *table++ = oputv;
238 castonly = 0;
239 break;
240 case 'V': /* contribution */
241 *table++ = oputV;
242 if (ambounce > 0 && (ambacc > FTINY || ambssamp > 0))
243 error(WARNING,
244 "-otV accuracy depends on -aa 0 -as 0");
245 break;
246 case 'l': /* effective distance */
247 *table++ = oputl;
248 castonly = 0;
249 break;
250 case 'c': /* local coordinates */
251 *table++ = oputc;
252 break;
253 case 'L': /* single ray length */
254 *table++ = oputL;
255 break;
256 case 'p': /* point */
257 *table++ = oputp;
258 break;
259 case 'n': /* perturbed normal */
260 *table++ = oputn;
261 castonly = 0;
262 break;
263 case 'N': /* unperturbed normal */
264 *table++ = oputN;
265 break;
266 case 's': /* surface */
267 *table++ = oputs;
268 break;
269 case 'w': /* weight */
270 *table++ = oputw;
271 break;
272 case 'W': /* coefficient */
273 *table++ = oputW;
274 if (ambounce > 0 && (ambacc > FTINY || ambssamp > 0))
275 error(WARNING,
276 "-otW accuracy depends on -aa 0 -as 0");
277 break;
278 case 'm': /* modifier */
279 *table++ = oputm;
280 break;
281 case 'M': /* material */
282 *table++ = oputM;
283 break;
284 case '~': /* tilde */
285 *table++ = oputtilde;
286 break;
287 }
288 *table = NULL;
289 }
290
291
292 static void
293 bogusray(void) /* print out empty record */
294 {
295 thisray.rorg[0] = thisray.rorg[1] = thisray.rorg[2] =
296 thisray.rdir[0] = thisray.rdir[1] = thisray.rdir[2] = 0.0;
297 thisray.rmax = 0.0;
298 rayorigin(&thisray, PRIMARY, NULL, NULL);
299 printvals(&thisray);
300 }
301
302
303 static void
304 raycast( /* compute first ray intersection only */
305 RAY *r
306 )
307 {
308 if (!localhit(r, &thescene)) {
309 if (r->ro == &Aftplane) { /* clipped */
310 r->ro = NULL;
311 r->rot = FHUGE;
312 } else
313 sourcehit(r);
314 }
315 }
316
317
318 static void
319 rayirrad( /* compute irradiance rather than radiance */
320 RAY *r
321 )
322 {
323 void (*old_revf)(RAY *) = r->revf;
324
325 r->rot = 1e-5; /* pretend we hit surface */
326 VSUM(r->rop, r->rorg, r->rdir, r->rot);
327 r->ron[0] = -r->rdir[0];
328 r->ron[1] = -r->rdir[1];
329 r->ron[2] = -r->rdir[2];
330 r->rod = 1.0;
331 /* compute result */
332 r->revf = raytrace;
333 (*ofun[Lamb.otype].funp)(&Lamb, r);
334 r->revf = old_revf;
335 }
336
337
338 static void
339 rtcompute( /* compute and print ray value(s) */
340 FVECT org,
341 FVECT dir,
342 double dmax
343 )
344 {
345 /* set up ray */
346 rayorigin(&thisray, PRIMARY, NULL, NULL);
347 if (imm_irrad) {
348 VSUM(thisray.rorg, org, dir, 1.1e-4);
349 thisray.rdir[0] = -dir[0];
350 thisray.rdir[1] = -dir[1];
351 thisray.rdir[2] = -dir[2];
352 thisray.rmax = 0.0;
353 thisray.revf = rayirrad;
354 } else {
355 VCOPY(thisray.rorg, org);
356 VCOPY(thisray.rdir, dir);
357 thisray.rmax = dmax;
358 if (castonly)
359 thisray.revf = raycast;
360 }
361 if (ray_pnprocs > 1) { /* multiprocessing FIFO? */
362 if (ray_fifo_in(&thisray) < 0)
363 error(USER, "lost children");
364 return;
365 }
366 samplendx++; /* else do it ourselves */
367 rayvalue(&thisray);
368 printvals(&thisray);
369 }
370
371
372 static int
373 printvals( /* print requested ray values */
374 RAY *r
375 )
376 {
377 oputf_t **tp;
378
379 if (ray_out[0] == NULL)
380 return(0);
381 for (tp = ray_out; *tp != NULL; tp++)
382 (**tp)(r);
383 if (outform == 'a')
384 putchar('\n');
385 return(1);
386 }
387
388
389 static int
390 getvec( /* get a vector from fp */
391 FVECT vec,
392 int fmt,
393 FILE *fp
394 )
395 {
396 static float vf[3];
397 static double vd[3];
398 char buf[32];
399 int i;
400
401 switch (fmt) {
402 case 'a': /* ascii */
403 for (i = 0; i < 3; i++) {
404 if (fgetword(buf, sizeof(buf), fp) == NULL ||
405 !isflt(buf))
406 return(-1);
407 vec[i] = atof(buf);
408 }
409 break;
410 case 'f': /* binary float */
411 if (fread((char *)vf, sizeof(float), 3, fp) != 3)
412 return(-1);
413 VCOPY(vec, vf);
414 break;
415 case 'd': /* binary double */
416 if (fread((char *)vd, sizeof(double), 3, fp) != 3)
417 return(-1);
418 VCOPY(vec, vd);
419 break;
420 default:
421 error(CONSISTENCY, "botched input format");
422 }
423 return(0);
424 }
425
426
427 void
428 tranotify( /* record new modifier */
429 OBJECT obj
430 )
431 {
432 static int hitlimit = 0;
433 OBJREC *o = objptr(obj);
434 char **tralp;
435
436 if (obj == OVOID) { /* starting over */
437 traset[0] = 0;
438 hitlimit = 0;
439 return;
440 }
441 if (hitlimit || !ismodifier(o->otype))
442 return;
443 for (tralp = tralist; *tralp != NULL; tralp++)
444 if (!strcmp(o->oname, *tralp)) {
445 if (traset[0] >= MAXTSET) {
446 error(WARNING, "too many modifiers in trace list");
447 hitlimit++;
448 return; /* should this be fatal? */
449 }
450 insertelem(traset, obj);
451 return;
452 }
453 }
454
455
456 static void
457 ourtrace( /* print ray values */
458 RAY *r
459 )
460 {
461 oputf_t **tp;
462
463 if (every_out[0] == NULL)
464 return;
465 if (r->ro == NULL) {
466 if (traincl == 1)
467 return;
468 } else if (traincl != -1 && traincl != inset(traset, r->ro->omod))
469 return;
470 tabin(r);
471 for (tp = every_out; *tp != NULL; tp++)
472 (**tp)(r);
473 if (outform == 'a')
474 putchar('\n');
475 }
476
477
478 static void
479 tabin( /* tab in appropriate amount */
480 RAY *r
481 )
482 {
483 const RAY *rp;
484
485 for (rp = r->parent; rp != NULL; rp = rp->parent)
486 putchar('\t');
487 }
488
489
490 static void
491 oputo( /* print origin */
492 RAY *r
493 )
494 {
495 (*putreal)(r->rorg[0]);
496 (*putreal)(r->rorg[1]);
497 (*putreal)(r->rorg[2]);
498 }
499
500
501 static void
502 oputd( /* print direction */
503 RAY *r
504 )
505 {
506 (*putreal)(r->rdir[0]);
507 (*putreal)(r->rdir[1]);
508 (*putreal)(r->rdir[2]);
509 }
510
511
512 static void
513 oputv( /* print value */
514 RAY *r
515 )
516 {
517 if (outform == 'c') {
518 COLR cout;
519 setcolr(cout, colval(r->rcol,RED),
520 colval(r->rcol,GRN),
521 colval(r->rcol,BLU));
522 fwrite((char *)cout, sizeof(cout), 1, stdout);
523 return;
524 }
525 (*putreal)(colval(r->rcol,RED));
526 (*putreal)(colval(r->rcol,GRN));
527 (*putreal)(colval(r->rcol,BLU));
528 }
529
530
531 static void
532 oputV( /* print value contribution */
533 RAY *r
534 )
535 {
536 double contr[3];
537
538 raycontrib(contr, r, PRIMARY);
539 multcolor(contr, r->rcol);
540 (*putreal)(contr[RED]);
541 (*putreal)(contr[GRN]);
542 (*putreal)(contr[BLU]);
543 }
544
545
546 static void
547 oputl( /* print effective distance */
548 RAY *r
549 )
550 {
551 (*putreal)(r->rt);
552 }
553
554
555 static void
556 oputL( /* print single ray length */
557 RAY *r
558 )
559 {
560 (*putreal)(r->rot);
561 }
562
563
564 static void
565 oputc( /* print local coordinates */
566 RAY *r
567 )
568 {
569 (*putreal)(r->uv[0]);
570 (*putreal)(r->uv[1]);
571 }
572
573
574 static void
575 oputp( /* print point */
576 RAY *r
577 )
578 {
579 if (r->rot < FHUGE) {
580 (*putreal)(r->rop[0]);
581 (*putreal)(r->rop[1]);
582 (*putreal)(r->rop[2]);
583 } else {
584 (*putreal)(0.0);
585 (*putreal)(0.0);
586 (*putreal)(0.0);
587 }
588 }
589
590
591 static void
592 oputN( /* print unperturbed normal */
593 RAY *r
594 )
595 {
596 if (r->rot < FHUGE) {
597 (*putreal)(r->ron[0]);
598 (*putreal)(r->ron[1]);
599 (*putreal)(r->ron[2]);
600 } else {
601 (*putreal)(0.0);
602 (*putreal)(0.0);
603 (*putreal)(0.0);
604 }
605 }
606
607
608 static void
609 oputn( /* print perturbed normal */
610 RAY *r
611 )
612 {
613 FVECT pnorm;
614
615 if (r->rot >= FHUGE) {
616 (*putreal)(0.0);
617 (*putreal)(0.0);
618 (*putreal)(0.0);
619 return;
620 }
621 raynormal(pnorm, r);
622 (*putreal)(pnorm[0]);
623 (*putreal)(pnorm[1]);
624 (*putreal)(pnorm[2]);
625 }
626
627
628 static void
629 oputs( /* print name */
630 RAY *r
631 )
632 {
633 if (r->ro != NULL)
634 fputs(r->ro->oname, stdout);
635 else
636 putchar('*');
637 putchar('\t');
638 }
639
640
641 static void
642 oputw( /* print weight */
643 RAY *r
644 )
645 {
646 (*putreal)(r->rweight);
647 }
648
649
650 static void
651 oputW( /* print coefficient */
652 RAY *r
653 )
654 {
655 double contr[3];
656
657 raycontrib(contr, r, PRIMARY);
658 (*putreal)(contr[RED]);
659 (*putreal)(contr[GRN]);
660 (*putreal)(contr[BLU]);
661 }
662
663
664 static void
665 oputm( /* print modifier */
666 RAY *r
667 )
668 {
669 if (r->ro != NULL)
670 if (r->ro->omod != OVOID)
671 fputs(objptr(r->ro->omod)->oname, stdout);
672 else
673 fputs(VOIDID, stdout);
674 else
675 putchar('*');
676 putchar('\t');
677 }
678
679
680 static void
681 oputM( /* print material */
682 RAY *r
683 )
684 {
685 OBJREC *mat;
686
687 if (r->ro != NULL) {
688 if ((mat = findmaterial(r->ro)) != NULL)
689 fputs(mat->oname, stdout);
690 else
691 fputs(VOIDID, stdout);
692 } else
693 putchar('*');
694 putchar('\t');
695 }
696
697
698 static void
699 oputtilde( /* output tilde (spacer) */
700 RAY *r
701 )
702 {
703 fputs("~\t", stdout);
704 }
705
706
707 static void
708 puta( /* print ascii value */
709 double v
710 )
711 {
712 printf("%e\t", v);
713 }
714
715
716 static void
717 putd(v) /* print binary double */
718 double v;
719 {
720 fwrite((char *)&v, sizeof(v), 1, stdout);
721 }
722
723
724 static void
725 putf(v) /* print binary float */
726 double v;
727 {
728 float f = v;
729
730 fwrite((char *)&f, sizeof(f), 1, stdout);
731 }