ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.84
Committed: Wed Aug 14 22:33:02 2019 UTC (4 years, 9 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.83: +1 -2 lines
Log Message:
Took out redundant setting

File Contents

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