ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/raytrace.c
Revision: 2.51
Committed: Tue May 31 18:01:09 2005 UTC (18 years, 11 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.50: +16 -1 lines
Log Message:
Added Russian roulette ray termination and fixed ambient weights & measures

File Contents

# Content
1 #ifndef lint
2 static const char RCSid[] = "$Id: raytrace.c,v 2.50 2005/05/26 06:55:22 greg Exp $";
3 #endif
4 /*
5 * raytrace.c - routines for tracing and shading rays.
6 *
7 * External symbols declared in ray.h
8 */
9
10 #include "copyright.h"
11
12 #include "ray.h"
13 #include "source.h"
14 #include "otypes.h"
15 #include "otspecial.h"
16 #include "random.h"
17
18 #define MAXCSET ((MAXSET+1)*2-1) /* maximum check set size */
19
20 unsigned long raynum = 0; /* next unique ray number */
21 unsigned long nrays = 0; /* number of calls to localhit */
22
23 static RREAL Lambfa[5] = {PI, PI, PI, 0.0, 0.0};
24 OBJREC Lamb = {
25 OVOID, MAT_PLASTIC, "Lambertian",
26 {0, 5, NULL, Lambfa}, NULL,
27 }; /* a Lambertian surface */
28
29 OBJREC Aftplane; /* aft clipping plane object */
30
31 #define RAYHIT (-1) /* return value for intercepted ray */
32
33 static int raymove(FVECT pos, OBJECT *cxs, int dirf, RAY *r, CUBE *cu);
34 static int checkhit(RAY *r, CUBE *cu, OBJECT *cxs);
35 static void checkset(OBJECT *os, OBJECT *cs);
36
37
38 extern int
39 rayorigin( /* start new ray from old one */
40 RAY *r,
41 int rt,
42 const RAY *ro,
43 const COLOR rc
44 )
45 {
46 double rw, re;
47 /* assign coefficient/weight */
48 if (rc == NULL) {
49 rw = 1.0;
50 setcolor(r->rcoef, 1., 1., 1.);
51 } else {
52 rw = intens(rc);
53 if (rc != r->rcoef)
54 copycolor(r->rcoef, rc);
55 }
56 if ((r->parent = ro) == NULL) { /* primary ray */
57 r->rlvl = 0;
58 r->rweight = rw;
59 r->crtype = r->rtype = rt;
60 r->rsrc = -1;
61 r->clipset = NULL;
62 r->revf = raytrace;
63 copycolor(r->cext, cextinction);
64 copycolor(r->albedo, salbedo);
65 r->gecc = seccg;
66 r->slights = NULL;
67 } else { /* spawned ray */
68 if (ro->rot >= FHUGE) {
69 memset(r, 0, sizeof(RAY));
70 return(-1); /* illegal continuation */
71 }
72 r->rlvl = ro->rlvl;
73 if (rt & RAYREFL) {
74 r->rlvl++;
75 r->rsrc = -1;
76 r->clipset = ro->clipset;
77 r->rmax = 0.0;
78 } else {
79 r->rsrc = ro->rsrc;
80 r->clipset = ro->newcset;
81 r->rmax = ro->rmax <= FTINY ? 0.0 : ro->rmax - ro->rot;
82 }
83 r->revf = ro->revf;
84 copycolor(r->cext, ro->cext);
85 copycolor(r->albedo, ro->albedo);
86 r->gecc = ro->gecc;
87 r->slights = ro->slights;
88 r->crtype = ro->crtype | (r->rtype = rt);
89 VCOPY(r->rorg, ro->rop);
90 r->rweight = ro->rweight * rw;
91 /* estimate extinction */
92 re = colval(ro->cext,RED) < colval(ro->cext,GRN) ?
93 colval(ro->cext,RED) : colval(ro->cext,GRN);
94 if (colval(ro->cext,BLU) < re) re = colval(ro->cext,BLU);
95 re *= ro->rot;
96 if (re > .1)
97 r->rweight *= exp(-re);
98 }
99 rayclear(r);
100 if (maxdepth <= 0 && rc != NULL) { /* Russian roulette */
101 if (minweight <= 0.0)
102 error(USER, "zero ray weight in Russian roulette");
103 if (maxdepth < 0 && r->rlvl > -maxdepth)
104 return(-1); /* upper reflection limit */
105 if (r->rweight >= minweight)
106 return(0);
107 if (frandom() < r->rweight/minweight)
108 return(-1);
109 rw = minweight/r->rweight; /* promote survivor */
110 scalecolor(r->rcoef, rw);
111 r->rweight = minweight;
112 return(0);
113 }
114 return(r->rlvl <= maxdepth && r->rweight >= minweight ? 0 : -1);
115 }
116
117
118 extern void
119 rayclear( /* clear a ray for (re)evaluation */
120 register RAY *r
121 )
122 {
123 r->rno = raynum++;
124 r->newcset = r->clipset;
125 r->hitf = rayhit;
126 r->robj = OVOID;
127 r->ro = NULL;
128 r->rox = NULL;
129 r->rt = r->rot = FHUGE;
130 r->pert[0] = r->pert[1] = r->pert[2] = 0.0;
131 r->uv[0] = r->uv[1] = 0.0;
132 setcolor(r->pcol, 1.0, 1.0, 1.0);
133 setcolor(r->rcol, 0.0, 0.0, 0.0);
134 }
135
136
137 extern void
138 raytrace( /* trace a ray and compute its value */
139 RAY *r
140 )
141 {
142 if (localhit(r, &thescene))
143 raycont(r); /* hit local surface, evaluate */
144 else if (r->ro == &Aftplane) {
145 r->ro = NULL; /* hit aft clipping plane */
146 r->rot = FHUGE;
147 } else if (sourcehit(r))
148 rayshade(r, r->ro->omod); /* distant source */
149
150 if (trace != NULL)
151 (*trace)(r); /* trace execution */
152
153 rayparticipate(r); /* for participating medium */
154 }
155
156
157 extern void
158 raycont( /* check for clipped object and continue */
159 register RAY *r
160 )
161 {
162 if ((r->clipset != NULL && inset(r->clipset, r->ro->omod)) ||
163 !rayshade(r, r->ro->omod))
164 raytrans(r);
165 }
166
167
168 extern void
169 raytrans( /* transmit ray as is */
170 register RAY *r
171 )
172 {
173 RAY tr;
174
175 if (rayorigin(&tr, TRANS, r, NULL) == 0) {
176 VCOPY(tr.rdir, r->rdir);
177 rayvalue(&tr);
178 copycolor(r->rcol, tr.rcol);
179 r->rt = r->rot + tr.rt;
180 }
181 }
182
183
184 extern int
185 rayshade( /* shade ray r with material mod */
186 register RAY *r,
187 int mod
188 )
189 {
190 register OBJREC *m;
191
192 r->rt = r->rot; /* set effective ray length */
193 for ( ; mod != OVOID; mod = m->omod) {
194 m = objptr(mod);
195 /****** unnecessary test since modifier() is always called
196 if (!ismodifier(m->otype)) {
197 sprintf(errmsg, "illegal modifier \"%s\"", m->oname);
198 error(USER, errmsg);
199 }
200 ******/
201 /* hack for irradiance calculation */
202 if (do_irrad && !(r->crtype & ~(PRIMARY|TRANS)) &&
203 m->otype != MAT_CLIP &&
204 (ofun[m->otype].flags & (T_M|T_X))) {
205 if (irr_ignore(m->otype)) {
206 raytrans(r);
207 return(1);
208 }
209 if (!islight(m->otype))
210 m = &Lamb;
211 }
212 if ((*ofun[m->otype].funp)(m, r))
213 return(1); /* materials call raytexture() */
214 }
215 return(0); /* no material! */
216 }
217
218
219 extern void
220 rayparticipate( /* compute ray medium participation */
221 register RAY *r
222 )
223 {
224 COLOR ce, ca;
225 double re, ge, be;
226
227 if (intens(r->cext) <= 1./FHUGE)
228 return; /* no medium */
229 re = r->rot*colval(r->cext,RED);
230 ge = r->rot*colval(r->cext,GRN);
231 be = r->rot*colval(r->cext,BLU);
232 if (r->crtype & SHADOW) { /* no scattering for sources */
233 re *= 1. - colval(r->albedo,RED);
234 ge *= 1. - colval(r->albedo,GRN);
235 be *= 1. - colval(r->albedo,BLU);
236 }
237 setcolor(ce, re<=FTINY ? 1. : re>92. ? 0. : exp(-re),
238 ge<=FTINY ? 1. : ge>92. ? 0. : exp(-ge),
239 be<=FTINY ? 1. : be>92. ? 0. : exp(-be));
240 multcolor(r->rcol, ce); /* path extinction */
241 if (r->crtype & SHADOW || intens(r->albedo) <= FTINY)
242 return; /* no scattering */
243 setcolor(ca,
244 colval(r->albedo,RED)*colval(ambval,RED)*(1.-colval(ce,RED)),
245 colval(r->albedo,GRN)*colval(ambval,GRN)*(1.-colval(ce,GRN)),
246 colval(r->albedo,BLU)*colval(ambval,BLU)*(1.-colval(ce,BLU)));
247 addcolor(r->rcol, ca); /* ambient in scattering */
248 srcscatter(r); /* source in scattering */
249 }
250
251
252 extern void
253 raytexture( /* get material modifiers */
254 RAY *r,
255 OBJECT mod
256 )
257 {
258 register OBJREC *m;
259 /* execute textures and patterns */
260 for ( ; mod != OVOID; mod = m->omod) {
261 m = objptr(mod);
262 /****** unnecessary test since modifier() is always called
263 if (!ismodifier(m->otype)) {
264 sprintf(errmsg, "illegal modifier \"%s\"", m->oname);
265 error(USER, errmsg);
266 }
267 ******/
268 if ((*ofun[m->otype].funp)(m, r)) {
269 sprintf(errmsg, "conflicting material \"%s\"",
270 m->oname);
271 objerror(r->ro, USER, errmsg);
272 }
273 }
274 }
275
276
277 extern int
278 raymixture( /* mix modifiers */
279 register RAY *r,
280 OBJECT fore,
281 OBJECT back,
282 double coef
283 )
284 {
285 RAY fr, br;
286 int foremat, backmat;
287 register int i;
288 /* bound coefficient */
289 if (coef > 1.0)
290 coef = 1.0;
291 else if (coef < 0.0)
292 coef = 0.0;
293 /* compute foreground and background */
294 foremat = backmat = 0;
295 /* foreground */
296 fr = *r;
297 if (coef > FTINY)
298 foremat = rayshade(&fr, fore);
299 /* background */
300 br = *r;
301 if (coef < 1.0-FTINY)
302 backmat = rayshade(&br, back);
303 /* check for transparency */
304 if (backmat ^ foremat) {
305 if (backmat && coef > FTINY)
306 raytrans(&fr);
307 else if (foremat && coef < 1.0-FTINY)
308 raytrans(&br);
309 }
310 /* mix perturbations */
311 for (i = 0; i < 3; i++)
312 r->pert[i] = coef*fr.pert[i] + (1.0-coef)*br.pert[i];
313 /* mix pattern colors */
314 scalecolor(fr.pcol, coef);
315 scalecolor(br.pcol, 1.0-coef);
316 copycolor(r->pcol, fr.pcol);
317 addcolor(r->pcol, br.pcol);
318 /* return value tells if material */
319 if (!foremat & !backmat)
320 return(0);
321 /* mix returned ray values */
322 scalecolor(fr.rcol, coef);
323 scalecolor(br.rcol, 1.0-coef);
324 copycolor(r->rcol, fr.rcol);
325 addcolor(r->rcol, br.rcol);
326 r->rt = bright(fr.rcol) > bright(br.rcol) ? fr.rt : br.rt;
327 return(1);
328 }
329
330
331 extern double
332 raydist( /* compute (cumulative) ray distance */
333 register const RAY *r,
334 register int flags
335 )
336 {
337 double sum = 0.0;
338
339 while (r != NULL && r->crtype&flags) {
340 sum += r->rot;
341 r = r->parent;
342 }
343 return(sum);
344 }
345
346
347 extern void
348 raycontrib( /* compute (cumulative) ray contribution */
349 COLOR rc,
350 const RAY *r,
351 int flags
352 )
353 {
354 COLOR eext, ext1;
355
356 setcolor(eext, 0., 0., 0.);
357 setcolor(rc, 1., 1., 1.);
358
359 while (r != NULL && r->crtype&flags) {
360 multcolor(rc, r->rcoef);
361 copycolor(ext1, r->cext);
362 scalecolor(ext1, r->rot);
363 addcolor(eext, ext1);
364 r = r->parent;
365 }
366 if (intens(eext) > FTINY) {
367 setcolor(ext1, exp(-colval(eext,RED)),
368 exp(-colval(eext,GRN)),
369 exp(-colval(eext,BLU)));
370 multcolor(rc, ext1);
371 }
372 }
373
374
375 extern double
376 raynormal( /* compute perturbed normal for ray */
377 FVECT norm,
378 register RAY *r
379 )
380 {
381 double newdot;
382 register int i;
383
384 /* The perturbation is added to the surface normal to obtain
385 * the new normal. If the new normal would affect the surface
386 * orientation wrt. the ray, a correction is made. The method is
387 * still fraught with problems since reflected rays and similar
388 * directions calculated from the surface normal may spawn rays behind
389 * the surface. The only solution is to curb textures at high
390 * incidence (namely, keep DOT(rdir,pert) < Rdot).
391 */
392
393 for (i = 0; i < 3; i++)
394 norm[i] = r->ron[i] + r->pert[i];
395
396 if (normalize(norm) == 0.0) {
397 objerror(r->ro, WARNING, "illegal normal perturbation");
398 VCOPY(norm, r->ron);
399 return(r->rod);
400 }
401 newdot = -DOT(norm, r->rdir);
402 if ((newdot > 0.0) != (r->rod > 0.0)) { /* fix orientation */
403 for (i = 0; i < 3; i++)
404 norm[i] += 2.0*newdot*r->rdir[i];
405 newdot = -newdot;
406 }
407 return(newdot);
408 }
409
410
411 extern void
412 newrayxf( /* get new tranformation matrix for ray */
413 RAY *r
414 )
415 {
416 static struct xfn {
417 struct xfn *next;
418 FULLXF xf;
419 } xfseed = { &xfseed }, *xflast = &xfseed;
420 register struct xfn *xp;
421 register const RAY *rp;
422
423 /*
424 * Search for transform in circular list that
425 * has no associated ray in the tree.
426 */
427 xp = xflast;
428 for (rp = r->parent; rp != NULL; rp = rp->parent)
429 if (rp->rox == &xp->xf) { /* xp in use */
430 xp = xp->next; /* move to next */
431 if (xp == xflast) { /* need new one */
432 xp = (struct xfn *)malloc(sizeof(struct xfn));
433 if (xp == NULL)
434 error(SYSTEM,
435 "out of memory in newrayxf");
436 /* insert in list */
437 xp->next = xflast->next;
438 xflast->next = xp;
439 break; /* we're done */
440 }
441 rp = r; /* start check over */
442 }
443 /* got it */
444 r->rox = &xp->xf;
445 xflast = xp;
446 }
447
448
449 extern void
450 flipsurface( /* reverse surface orientation */
451 register RAY *r
452 )
453 {
454 r->rod = -r->rod;
455 r->ron[0] = -r->ron[0];
456 r->ron[1] = -r->ron[1];
457 r->ron[2] = -r->ron[2];
458 r->pert[0] = -r->pert[0];
459 r->pert[1] = -r->pert[1];
460 r->pert[2] = -r->pert[2];
461 }
462
463
464 extern void
465 rayhit( /* standard ray hit test */
466 OBJECT *oset,
467 RAY *r
468 )
469 {
470 OBJREC *o;
471 int i;
472
473 for (i = oset[0]; i > 0; i--) {
474 o = objptr(oset[i]);
475 if ((*ofun[o->otype].funp)(o, r))
476 r->robj = oset[i];
477 }
478 }
479
480
481 extern int
482 localhit( /* check for hit in the octree */
483 register RAY *r,
484 register CUBE *scene
485 )
486 {
487 OBJECT cxset[MAXCSET+1]; /* set of checked objects */
488 FVECT curpos; /* current cube position */
489 int sflags; /* sign flags */
490 double t, dt;
491 register int i;
492
493 nrays++; /* increment trace counter */
494 sflags = 0;
495 for (i = 0; i < 3; i++) {
496 curpos[i] = r->rorg[i];
497 if (r->rdir[i] > 1e-7)
498 sflags |= 1 << i;
499 else if (r->rdir[i] < -1e-7)
500 sflags |= 0x10 << i;
501 }
502 if (sflags == 0)
503 error(CONSISTENCY, "zero ray direction in localhit");
504 /* start off assuming nothing hit */
505 if (r->rmax > FTINY) { /* except aft plane if one */
506 r->ro = &Aftplane;
507 r->rot = r->rmax;
508 for (i = 0; i < 3; i++)
509 r->rop[i] = r->rorg[i] + r->rot*r->rdir[i];
510 }
511 /* find global cube entrance point */
512 t = 0.0;
513 if (!incube(scene, curpos)) {
514 /* find distance to entry */
515 for (i = 0; i < 3; i++) {
516 /* plane in our direction */
517 if (sflags & 1<<i)
518 dt = scene->cuorg[i];
519 else if (sflags & 0x10<<i)
520 dt = scene->cuorg[i] + scene->cusize;
521 else
522 continue;
523 /* distance to the plane */
524 dt = (dt - r->rorg[i])/r->rdir[i];
525 if (dt > t)
526 t = dt; /* farthest face is the one */
527 }
528 t += FTINY; /* fudge to get inside cube */
529 if (t >= r->rot) /* clipped already */
530 return(0);
531 /* advance position */
532 for (i = 0; i < 3; i++)
533 curpos[i] += r->rdir[i]*t;
534
535 if (!incube(scene, curpos)) /* non-intersecting ray */
536 return(0);
537 }
538 cxset[0] = 0;
539 raymove(curpos, cxset, sflags, r, scene);
540 return((r->ro != NULL) & (r->ro != &Aftplane));
541 }
542
543
544 static int
545 raymove( /* check for hit as we move */
546 FVECT pos, /* current position, modified herein */
547 OBJECT *cxs, /* checked objects, modified by checkhit */
548 int dirf, /* direction indicators to speed tests */
549 register RAY *r,
550 register CUBE *cu
551 )
552 {
553 int ax;
554 double dt, t;
555
556 if (istree(cu->cutree)) { /* recurse on subcubes */
557 CUBE cukid;
558 register int br, sgn;
559
560 cukid.cusize = cu->cusize * 0.5; /* find subcube */
561 VCOPY(cukid.cuorg, cu->cuorg);
562 br = 0;
563 if (pos[0] >= cukid.cuorg[0]+cukid.cusize) {
564 cukid.cuorg[0] += cukid.cusize;
565 br |= 1;
566 }
567 if (pos[1] >= cukid.cuorg[1]+cukid.cusize) {
568 cukid.cuorg[1] += cukid.cusize;
569 br |= 2;
570 }
571 if (pos[2] >= cukid.cuorg[2]+cukid.cusize) {
572 cukid.cuorg[2] += cukid.cusize;
573 br |= 4;
574 }
575 for ( ; ; ) {
576 cukid.cutree = octkid(cu->cutree, br);
577 if ((ax = raymove(pos,cxs,dirf,r,&cukid)) == RAYHIT)
578 return(RAYHIT);
579 sgn = 1 << ax;
580 if (sgn & dirf) /* positive axis? */
581 if (sgn & br)
582 return(ax); /* overflow */
583 else {
584 cukid.cuorg[ax] += cukid.cusize;
585 br |= sgn;
586 }
587 else
588 if (sgn & br) {
589 cukid.cuorg[ax] -= cukid.cusize;
590 br &= ~sgn;
591 } else
592 return(ax); /* underflow */
593 }
594 /*NOTREACHED*/
595 }
596 if (isfull(cu->cutree)) {
597 if (checkhit(r, cu, cxs))
598 return(RAYHIT);
599 } else if (r->ro == &Aftplane && incube(cu, r->rop))
600 return(RAYHIT);
601 /* advance to next cube */
602 if (dirf&0x11) {
603 dt = dirf&1 ? cu->cuorg[0] + cu->cusize : cu->cuorg[0];
604 t = (dt - pos[0])/r->rdir[0];
605 ax = 0;
606 } else
607 t = FHUGE;
608 if (dirf&0x22) {
609 dt = dirf&2 ? cu->cuorg[1] + cu->cusize : cu->cuorg[1];
610 dt = (dt - pos[1])/r->rdir[1];
611 if (dt < t) {
612 t = dt;
613 ax = 1;
614 }
615 }
616 if (dirf&0x44) {
617 dt = dirf&4 ? cu->cuorg[2] + cu->cusize : cu->cuorg[2];
618 dt = (dt - pos[2])/r->rdir[2];
619 if (dt < t) {
620 t = dt;
621 ax = 2;
622 }
623 }
624 pos[0] += r->rdir[0]*t;
625 pos[1] += r->rdir[1]*t;
626 pos[2] += r->rdir[2]*t;
627 return(ax);
628 }
629
630
631 static int
632 checkhit( /* check for hit in full cube */
633 register RAY *r,
634 CUBE *cu,
635 OBJECT *cxs
636 )
637 {
638 OBJECT oset[MAXSET+1];
639
640 objset(oset, cu->cutree);
641 checkset(oset, cxs); /* avoid double-checking */
642
643 (*r->hitf)(oset, r); /* test for hit in set */
644
645 if (r->robj == OVOID)
646 return(0); /* no scores yet */
647
648 return(incube(cu, r->rop)); /* hit OK if in current cube */
649 }
650
651
652 static void
653 checkset( /* modify checked set and set to check */
654 register OBJECT *os, /* os' = os - cs */
655 register OBJECT *cs /* cs' = cs + os */
656 )
657 {
658 OBJECT cset[MAXCSET+MAXSET+1];
659 register int i, j;
660 int k;
661 /* copy os in place, cset <- cs */
662 cset[0] = 0;
663 k = 0;
664 for (i = j = 1; i <= os[0]; i++) {
665 while (j <= cs[0] && cs[j] < os[i])
666 cset[++cset[0]] = cs[j++];
667 if (j > cs[0] || os[i] != cs[j]) { /* object to check */
668 os[++k] = os[i];
669 cset[++cset[0]] = os[i];
670 }
671 }
672 if (!(os[0] = k)) /* new "to check" set size */
673 return; /* special case */
674 while (j <= cs[0]) /* get the rest of cs */
675 cset[++cset[0]] = cs[j++];
676 if (cset[0] > MAXCSET) /* truncate "checked" set if nec. */
677 cset[0] = MAXCSET;
678 /* setcopy(cs, cset); */ /* copy cset back to cs */
679 os = cset;
680 for (i = os[0]; i-- >= 0; )
681 *cs++ = *os++;
682 }