ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.88
Committed: Thu Mar 12 17:19:18 2020 UTC (4 years, 2 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.87: +4 -4 lines
Log Message:
Added precautionary testing when checking for FHUGE

File Contents

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