ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/rtrace.c
Revision: 2.80
Committed: Thu Jul 4 01:06:35 2019 UTC (4 years, 10 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.79: +27 -16 lines
Log Message:
Made -f?c RGBE output compatible with any of 'vVWrx' options and added warning

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: rtrace.c,v 2.79 2019/05/06 16:49:38 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("vVWrx", outvals[0])))
150 error(USER, "color format only with one of 'vVWrx'");
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 if (do_irrad | imm_irrad) /* compatibile? */
320 for (table = ray_out; *table != NULL; table++)
321 if ((*table == oputr) | (*table == oputR) |
322 (*table == oputx) | (*table == oputX)) {
323 error(WARNING, "-orRxX options incompatible with -I+ and -i+");
324 break;
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)
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)
666 (*putreal)(r->ron, 3);
667 else
668 (*putreal)(vdummy, 3);
669 }
670
671
672 static void
673 oputn( /* print perturbed normal */
674 RAY *r
675 )
676 {
677 FVECT pnorm;
678
679 if (r->rot >= FHUGE) {
680 (*putreal)(vdummy, 3);
681 return;
682 }
683 raynormal(pnorm, r);
684 (*putreal)(pnorm, 3);
685 }
686
687
688 static void
689 oputs( /* print name */
690 RAY *r
691 )
692 {
693 if (r->ro != NULL)
694 fputs(r->ro->oname, stdout);
695 else
696 putchar('*');
697 putchar('\t');
698 }
699
700
701 static void
702 oputw( /* print weight */
703 RAY *r
704 )
705 {
706 RREAL rwt = r->rweight;
707
708 (*putreal)(&rwt, 1);
709 }
710
711
712 static void
713 oputW( /* print coefficient */
714 RAY *r
715 )
716 {
717 RREAL contr[3];
718 /* shadow ray not on source? */
719 if (r->rsrc >= 0 && source[r->rsrc].so != r->ro)
720 setcolor(contr, 0.0, 0.0, 0.0);
721 else
722 raycontrib(contr, r, PRIMARY);
723
724 (*putreal)(contr, 3);
725 }
726
727
728 static void
729 oputm( /* print modifier */
730 RAY *r
731 )
732 {
733 if (r->ro != NULL)
734 if (r->ro->omod != OVOID)
735 fputs(objptr(r->ro->omod)->oname, stdout);
736 else
737 fputs(VOIDID, stdout);
738 else
739 putchar('*');
740 putchar('\t');
741 }
742
743
744 static void
745 oputM( /* print material */
746 RAY *r
747 )
748 {
749 OBJREC *mat;
750
751 if (r->ro != NULL) {
752 if ((mat = findmaterial(r->ro)) != NULL)
753 fputs(mat->oname, stdout);
754 else
755 fputs(VOIDID, stdout);
756 } else
757 putchar('*');
758 putchar('\t');
759 }
760
761
762 static void
763 oputtilde( /* output tilde (spacer) */
764 RAY *r
765 )
766 {
767 fputs("~\t", stdout);
768 }
769
770
771 static void
772 puta( /* print ascii value(s) */
773 RREAL *v, int n
774 )
775 {
776 if (n == 3) {
777 printf("%e\t%e\t%e\t", v[0], v[1], v[2]);
778 return;
779 }
780 while (n--)
781 printf("%e\t", *v++);
782 }
783
784
785 static void
786 putd(RREAL *v, int n) /* output binary double(s) */
787 {
788 #ifdef SMLFLT
789 double da[3];
790 int i;
791
792 if (n > 3)
793 error(INTERNAL, "code error in putd()");
794 for (i = n; i--; )
795 da[i] = v[i];
796 putbinary(da, sizeof(double), n, stdout);
797 #else
798 putbinary(v, sizeof(RREAL), n, stdout);
799 #endif
800 }
801
802
803 static void
804 putf(RREAL *v, int n) /* output binary float(s) */
805 {
806 #ifndef SMLFLT
807 float fa[3];
808 int i;
809
810 if (n > 3)
811 error(INTERNAL, "code error in putf()");
812 for (i = n; i--; )
813 fa[i] = v[i];
814 putbinary(fa, sizeof(float), n, stdout);
815 #else
816 putbinary(v, sizeof(RREAL), n, stdout);
817 #endif
818 }
819
820
821 static void
822 putrgbe(RREAL *v, int n) /* output RGBE color */
823 {
824 COLR cout;
825
826 if (n != 3)
827 error(INTERNAL, "putrgbe() not called with 3 components");
828 setcolr(cout, v[0], v[1], v[2]);
829 putbinary(cout, sizeof(cout), 1, stdout);
830 }