ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.82
Committed: Thu Jul 25 16:50:54 2019 UTC (4 years, 9 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.81: +4 -3 lines
Log Message:
Keep track of ray flips so rayreorient() can put it back as it was

File Contents

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