ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/srcsamp.c
Revision: 2.16
Committed: Sat Jun 6 02:11:44 2009 UTC (14 years, 10 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad4R0
Changes since 2.15: +16 -14 lines
Log Message:
Fixed erroneous "aiming failure" with certain mirrored virtual sources

File Contents

# User Rev Content
1 greg 1.1 #ifndef lint
2 greg 2.16 static const char RCSid[] = "$Id: srcsamp.c,v 2.15 2008/12/10 17:33:23 greg Exp $";
3 greg 1.1 #endif
4     /*
5     * Source sampling routines
6 greg 2.7 *
7     * External symbols declared in source.h
8     */
9    
10 greg 2.8 #include "copyright.h"
11 greg 1.1
12 greg 1.4 #include "ray.h"
13 greg 1.1
14     #include "source.h"
15    
16     #include "random.h"
17    
18    
19 greg 2.5 static int cyl_partit(), flt_partit();
20    
21    
22 greg 1.1 double
23 greg 1.4 nextssamp(r, si) /* compute sample for source, rtn. distance */
24     register RAY *r; /* origin is read, direction is set */
25 greg 1.1 register SRCINDEX *si; /* source index (modified to current) */
26     {
27     int cent[3], size[3], parr[2];
28 greg 2.16 SRCREC *srcp;
29 greg 1.1 FVECT vpos;
30     double d;
31     register int i;
32 greg 2.2 nextsample:
33 greg 1.1 while (++si->sp >= si->np) { /* get next sample */
34     if (++si->sn >= nsources)
35     return(0.0); /* no more */
36 greg 1.7 if (source[si->sn].sflags & SSKIP)
37     si->np = 0;
38     else if (srcsizerat <= FTINY)
39 greg 1.4 nopart(si, r);
40 greg 1.1 else {
41     for (i = si->sn; source[i].sflags & SVIRTUAL;
42     i = source[i].sa.sv.sn)
43     ; /* partition source */
44 greg 1.4 (*sfun[source[i].so->otype].of->partit)(si, r);
45 greg 1.1 }
46     si->sp = -1;
47     }
48     /* get partition */
49     cent[0] = cent[1] = cent[2] = 0;
50     size[0] = size[1] = size[2] = MAXSPART;
51     parr[0] = 0; parr[1] = si->sp;
52     if (!skipparts(cent, size, parr, si->spt))
53     error(CONSISTENCY, "bad source partition in nextssamp");
54     /* compute sample */
55 greg 2.16 srcp = source + si->sn;
56 greg 1.1 if (dstrsrc > FTINY) { /* jitter sample */
57     dimlist[ndims] = si->sn + 8831;
58     dimlist[ndims+1] = si->sp + 3109;
59     d = urand(ilhash(dimlist,ndims+2)+samplendx);
60 greg 2.16 if (srcp->sflags & SFLAT) {
61 greg 1.1 multisamp(vpos, 2, d);
62 greg 2.12 vpos[SW] = 0.5;
63 greg 1.1 } else
64     multisamp(vpos, 3, d);
65     for (i = 0; i < 3; i++)
66     vpos[i] = dstrsrc * (1. - 2.*vpos[i]) *
67     (double)size[i]/MAXSPART;
68     } else
69     vpos[0] = vpos[1] = vpos[2] = 0.0;
70    
71     for (i = 0; i < 3; i++)
72     vpos[i] += (double)cent[i]/MAXSPART;
73 greg 2.12 /* avoid circular aiming failures */
74 greg 2.16 if ((srcp->sflags & SCIR) && (si->np > 1 || dstrsrc > 0.7)) {
75 greg 2.12 FVECT trim;
76 greg 2.16 if (srcp->sflags & (SFLAT|SDISTANT)) {
77 greg 2.12 d = 1.12837917; /* correct setflatss() */
78     trim[SU] = d*sqrt(1.0 - 0.5*vpos[SV]*vpos[SV]);
79     trim[SV] = d*sqrt(1.0 - 0.5*vpos[SU]*vpos[SU]);
80     trim[SW] = 0.0;
81     } else {
82     trim[SW] = trim[SU] = vpos[SU]*vpos[SU];
83     d = vpos[SV]*vpos[SV];
84     if (d > trim[SW]) trim[SW] = d;
85     trim[SU] += d;
86     d = vpos[SW]*vpos[SW];
87     if (d > trim[SW]) trim[SW] = d;
88     trim[SU] += d;
89 greg 2.15 if (trim[SU] > FTINY*FTINY) {
90     d = 1.0/0.7236; /* correct sphsetsrc() */
91     trim[SW] = trim[SV] = trim[SU] =
92     d*sqrt(trim[SW]/trim[SU]);
93     } else
94     trim[SW] = trim[SV] = trim[SU] = 0.0;
95 greg 2.12 }
96     for (i = 0; i < 3; i++)
97     vpos[i] *= trim[i];
98     }
99 greg 1.1 /* compute direction */
100     for (i = 0; i < 3; i++)
101 greg 2.16 r->rdir[i] = srcp->sloc[i] +
102     vpos[SU]*srcp->ss[SU][i] +
103     vpos[SV]*srcp->ss[SV][i] +
104     vpos[SW]*srcp->ss[SW][i];
105 greg 1.1
106 greg 2.16 if (!(srcp->sflags & SDISTANT))
107 greg 1.1 for (i = 0; i < 3; i++)
108 greg 1.4 r->rdir[i] -= r->rorg[i];
109 greg 1.1 /* compute distance */
110 greg 1.4 if ((d = normalize(r->rdir)) == 0.0)
111 greg 2.2 goto nextsample; /* at source! */
112 greg 1.1
113     /* compute sample size */
114 greg 2.16 if (srcp->sflags & SFLAT) {
115 greg 2.6 si->dom = sflatform(si->sn, r->rdir);
116 greg 2.4 si->dom *= size[SU]*size[SV]/(MAXSPART*(double)MAXSPART);
117 greg 2.16 } else if (srcp->sflags & SCYL) {
118 greg 2.6 si->dom = scylform(si->sn, r->rdir);
119 greg 2.4 si->dom *= size[SU]/(double)MAXSPART;
120 greg 1.1 } else {
121 greg 2.6 si->dom = size[SU]*size[SV]*(double)size[SW] /
122 greg 2.4 (MAXSPART*MAXSPART*(double)MAXSPART) ;
123 greg 1.1 }
124 greg 2.16 if (srcp->sflags & SDISTANT) {
125     si->dom *= srcp->ss2;
126 greg 1.1 return(FHUGE);
127 greg 2.6 }
128 greg 2.3 if (si->dom <= 1e-4)
129 greg 2.2 goto nextsample; /* behind source? */
130 greg 2.16 si->dom *= srcp->ss2/(d*d);
131 greg 1.1 return(d); /* sample OK, return distance */
132     }
133    
134    
135 greg 2.7 int
136 greg 1.1 skipparts(ct, sz, pp, pt) /* skip to requested partition */
137     int ct[3], sz[3]; /* center and size of partition (returned) */
138     register int pp[2]; /* current index, number to skip (modified) */
139     unsigned char *pt; /* partition array */
140     {
141     register int p;
142     /* check this partition */
143     p = spart(pt, pp[0]);
144     pp[0]++;
145 schorsch 2.10 if (p == S0) { /* leaf partition */
146 greg 1.1 if (pp[1]) {
147     pp[1]--;
148     return(0); /* not there yet */
149     } else
150     return(1); /* we've arrived */
151 schorsch 2.10 }
152 greg 1.1 /* else check lower */
153     sz[p] >>= 1;
154     ct[p] -= sz[p];
155     if (skipparts(ct, sz, pp, pt))
156     return(1); /* return hit */
157     /* else check upper */
158     ct[p] += sz[p] << 1;
159     if (skipparts(ct, sz, pp, pt))
160     return(1); /* return hit */
161     /* else return to starting position */
162     ct[p] -= sz[p];
163     sz[p] <<= 1;
164     return(0); /* return miss */
165     }
166    
167    
168 greg 2.7 void
169 greg 1.4 nopart(si, r) /* single source partition */
170 greg 1.1 register SRCINDEX *si;
171 greg 1.4 RAY *r;
172 greg 1.1 {
173     clrpart(si->spt);
174     setpart(si->spt, 0, S0);
175     si->np = 1;
176     }
177    
178    
179 greg 2.7 void
180 greg 1.4 cylpart(si, r) /* partition a cylinder */
181 greg 1.1 SRCINDEX *si;
182 greg 1.4 register RAY *r;
183 greg 1.1 {
184     double dist2, safedist2, dist2cent, rad2;
185     FVECT v;
186     register SRCREC *sp;
187     int pi;
188     /* first check point location */
189     clrpart(si->spt);
190 greg 1.4 sp = source + si->sn;
191 greg 1.3 rad2 = 1.365 * DOT(sp->ss[SV],sp->ss[SV]);
192 greg 1.4 v[0] = r->rorg[0] - sp->sloc[0];
193     v[1] = r->rorg[1] - sp->sloc[1];
194     v[2] = r->rorg[2] - sp->sloc[2];
195 greg 1.1 dist2 = DOT(v,sp->ss[SU]);
196     safedist2 = DOT(sp->ss[SU],sp->ss[SU]);
197     dist2 *= dist2 / safedist2;
198     dist2cent = DOT(v,v);
199     dist2 = dist2cent - dist2;
200     if (dist2 <= rad2) { /* point inside extended cylinder */
201     si->np = 0;
202     return;
203     }
204 greg 1.4 safedist2 *= 4.*r->rweight*r->rweight/(srcsizerat*srcsizerat);
205 greg 1.5 if (dist2 <= 4.*rad2 || /* point too close to subdivide */
206     dist2cent >= safedist2) { /* or too far */
207 greg 1.1 setpart(si->spt, 0, S0);
208     si->np = 1;
209     return;
210     }
211     pi = 0;
212 greg 1.4 si->np = cyl_partit(r->rorg, si->spt, &pi, MAXSPART,
213 greg 1.1 sp->sloc, sp->ss[SU], safedist2);
214     }
215    
216    
217     static int
218     cyl_partit(ro, pt, pi, mp, cent, axis, d2) /* slice a cylinder */
219     FVECT ro;
220     unsigned char *pt;
221     register int *pi;
222     int mp;
223     FVECT cent, axis;
224     double d2;
225     {
226     FVECT newct, newax;
227     int npl, npu;
228    
229     if (mp < 2 || dist2(ro, cent) >= d2) { /* hit limit? */
230     setpart(pt, *pi, S0);
231     (*pi)++;
232     return(1);
233     }
234     /* subdivide */
235     setpart(pt, *pi, SU);
236     (*pi)++;
237     newax[0] = .5*axis[0];
238     newax[1] = .5*axis[1];
239     newax[2] = .5*axis[2];
240     d2 *= 0.25;
241     /* lower half */
242     newct[0] = cent[0] - newax[0];
243     newct[1] = cent[1] - newax[1];
244     newct[2] = cent[2] - newax[2];
245 greg 1.2 npl = cyl_partit(ro, pt, pi, mp/2, newct, newax, d2);
246 greg 1.1 /* upper half */
247     newct[0] = cent[0] + newax[0];
248     newct[1] = cent[1] + newax[1];
249     newct[2] = cent[2] + newax[2];
250 greg 1.2 npu = cyl_partit(ro, pt, pi, mp/2, newct, newax, d2);
251 greg 1.1 /* return total */
252     return(npl + npu);
253     }
254    
255    
256 greg 2.7 void
257 greg 1.4 flatpart(si, r) /* partition a flat source */
258 greg 1.1 register SRCINDEX *si;
259 greg 1.5 register RAY *r;
260 greg 1.1 {
261 schorsch 2.9 register RREAL *vp;
262 greg 1.5 FVECT v;
263 greg 1.1 double du2, dv2;
264     int pi;
265    
266 greg 1.5 clrpart(si->spt);
267     vp = source[si->sn].sloc;
268     v[0] = r->rorg[0] - vp[0];
269     v[1] = r->rorg[1] - vp[1];
270     v[2] = r->rorg[2] - vp[2];
271     vp = source[si->sn].snorm;
272 greg 2.11 if (DOT(v,vp) <= 0.) { /* behind source */
273 greg 1.5 si->np = 0;
274     return;
275     }
276 greg 1.4 dv2 = 2.*r->rweight/srcsizerat;
277     dv2 *= dv2;
278 greg 1.1 vp = source[si->sn].ss[SU];
279 greg 1.4 du2 = dv2 * DOT(vp,vp);
280 greg 1.1 vp = source[si->sn].ss[SV];
281 greg 1.4 dv2 *= DOT(vp,vp);
282 greg 1.1 pi = 0;
283 greg 1.4 si->np = flt_partit(r->rorg, si->spt, &pi, MAXSPART,
284     source[si->sn].sloc,
285 greg 1.1 source[si->sn].ss[SU], source[si->sn].ss[SV], du2, dv2);
286     }
287    
288    
289     static int
290     flt_partit(ro, pt, pi, mp, cent, u, v, du2, dv2) /* partition flatty */
291     FVECT ro;
292     unsigned char *pt;
293     register int *pi;
294     int mp;
295     FVECT cent, u, v;
296     double du2, dv2;
297     {
298     double d2;
299     FVECT newct, newax;
300     int npl, npu;
301    
302     if (mp < 2 || ((d2 = dist2(ro, cent)) >= du2
303     && d2 >= dv2)) { /* hit limit? */
304     setpart(pt, *pi, S0);
305     (*pi)++;
306     return(1);
307     }
308     if (du2 > dv2) { /* subdivide in U */
309     setpart(pt, *pi, SU);
310     (*pi)++;
311     newax[0] = .5*u[0];
312     newax[1] = .5*u[1];
313     newax[2] = .5*u[2];
314     u = newax;
315     du2 *= 0.25;
316     } else { /* subdivide in V */
317     setpart(pt, *pi, SV);
318     (*pi)++;
319     newax[0] = .5*v[0];
320     newax[1] = .5*v[1];
321     newax[2] = .5*v[2];
322     v = newax;
323     dv2 *= 0.25;
324     }
325     /* lower half */
326     newct[0] = cent[0] - newax[0];
327     newct[1] = cent[1] - newax[1];
328     newct[2] = cent[2] - newax[2];
329 greg 1.2 npl = flt_partit(ro, pt, pi, mp/2, newct, u, v, du2, dv2);
330 greg 1.1 /* upper half */
331     newct[0] = cent[0] + newax[0];
332     newct[1] = cent[1] + newax[1];
333     newct[2] = cent[2] + newax[2];
334 greg 1.2 npu = flt_partit(ro, pt, pi, mp/2, newct, u, v, du2, dv2);
335 greg 1.1 /* return total */
336     return(npl + npu);
337     }
338    
339    
340     double
341     scylform(sn, dir) /* compute cosine for cylinder's projection */
342     int sn;
343     register FVECT dir; /* assume normalized */
344     {
345 schorsch 2.9 register RREAL *dv;
346 greg 1.1 double d;
347    
348     dv = source[sn].ss[SU];
349     d = DOT(dir, dv);
350     d *= d / DOT(dv,dv);
351     return(sqrt(1. - d));
352     }