ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.68
Committed: Fri Apr 11 20:27:23 2014 UTC (10 years ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R2P2, rad5R0, rad4R2, rad4R2P1
Changes since 2.67: +15 -2 lines
Log Message:
Made reading of vectors more consistent with FVFORMAT macro

File Contents

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