ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/ambcomp.c
Revision: 1.7
Committed: Thu Jun 13 12:00:29 1991 UTC (32 years, 10 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 1.6: +8 -0 lines
Log Message:
placed upper limit on position gradient

File Contents

# User Rev Content
1 greg 1.1 /* Copyright (c) 1991 Regents of the University of California */
2    
3     #ifndef lint
4     static char SCCSid[] = "$SunId$ LBL";
5     #endif
6    
7     /*
8     * Routines to compute "ambient" values using Monte Carlo
9     */
10    
11     #include "ray.h"
12    
13     #include "ambient.h"
14    
15     #include "random.h"
16    
17     typedef struct {
18     short t, p; /* theta, phi indices */
19     COLOR v; /* value sum */
20 greg 1.2 float r; /* 1/distance sum */
21     float k; /* variance for this division */
22 greg 1.1 int n; /* number of subsamples */
23 greg 1.2 } AMBSAMP; /* ambient sample division */
24 greg 1.1
25     typedef struct {
26     FVECT ux, uy, uz; /* x, y and z axis directions */
27     short nt, np; /* number of theta and phi directions */
28     } AMBHEMI; /* ambient sample hemisphere */
29    
30     extern double sin(), cos(), sqrt();
31    
32    
33     static int
34     ambcmp(d1, d2) /* decreasing order */
35     AMBSAMP *d1, *d2;
36     {
37     if (d1->k < d2->k)
38     return(1);
39     if (d1->k > d2->k)
40     return(-1);
41     return(0);
42     }
43    
44    
45     static int
46     ambnorm(d1, d2) /* standard order */
47     AMBSAMP *d1, *d2;
48     {
49     register int c;
50    
51     if (c = d1->t - d2->t)
52     return(c);
53     return(d1->p - d2->p);
54     }
55    
56    
57     divsample(dp, h, r) /* sample a division */
58     register AMBSAMP *dp;
59     AMBHEMI *h;
60     RAY *r;
61     {
62     RAY ar;
63     int hlist[4];
64     double xd, yd, zd;
65     double b2;
66     double phi;
67 greg 1.2 register int i;
68 greg 1.1
69     if (rayorigin(&ar, r, AMBIENT, 0.5) < 0)
70 greg 1.4 return(-1);
71 greg 1.1 hlist[0] = r->rno;
72     hlist[1] = dp->t;
73     hlist[2] = dp->p;
74     hlist[3] = 0;
75     zd = sqrt((dp->t+urand(ilhash(hlist,4)+dp->n))/h->nt);
76     hlist[3] = 1;
77     phi = 2.0*PI * (dp->p+urand(ilhash(hlist,4)+dp->n))/h->np;
78     xd = cos(phi) * zd;
79     yd = sin(phi) * zd;
80     zd = sqrt(1.0 - zd*zd);
81 greg 1.2 for (i = 0; i < 3; i++)
82     ar.rdir[i] = xd*h->ux[i] +
83     yd*h->uy[i] +
84     zd*h->uz[i];
85     dimlist[ndims++] = dp->t*h->np + dp->p + 90171;
86 greg 1.1 rayvalue(&ar);
87     ndims--;
88     addcolor(dp->v, ar.rcol);
89 greg 1.4 if (ar.rt < FHUGE)
90     dp->r += 1.0/ar.rt;
91 greg 1.1 /* (re)initialize error */
92     if (dp->n++) {
93     b2 = bright(dp->v)/dp->n - bright(ar.rcol);
94     b2 = b2*b2 + dp->k*((dp->n-1)*(dp->n-1));
95     dp->k = b2/(dp->n*dp->n);
96     } else
97     dp->k = 0.0;
98 greg 1.4 return(0);
99 greg 1.1 }
100    
101    
102     double
103     doambient(acol, r, pg, dg) /* compute ambient component */
104     COLOR acol;
105     RAY *r;
106     FVECT pg, dg;
107     {
108     double b, d;
109     AMBHEMI hemi;
110     AMBSAMP *div;
111     AMBSAMP dnew;
112     register AMBSAMP *dp;
113     double arad;
114     int ndivs, ns;
115     register int i, j;
116     /* initialize color */
117     setcolor(acol, 0.0, 0.0, 0.0);
118     /* initialize hemisphere */
119     inithemi(&hemi, r);
120     ndivs = hemi.nt * hemi.np;
121     if (ndivs == 0)
122     return(0.0);
123     /* set number of super-samples */
124     ns = ambssamp * r->rweight + 0.5;
125     if (ns > 0 || pg != NULL || dg != NULL) {
126     div = (AMBSAMP *)malloc(ndivs*sizeof(AMBSAMP));
127     if (div == NULL)
128     error(SYSTEM, "out of memory in doambient");
129     } else
130     div = NULL;
131     /* sample the divisions */
132     arad = 0.0;
133     if ((dp = div) == NULL)
134     dp = &dnew;
135     for (i = 0; i < hemi.nt; i++)
136     for (j = 0; j < hemi.np; j++) {
137     dp->t = i; dp->p = j;
138     setcolor(dp->v, 0.0, 0.0, 0.0);
139 greg 1.2 dp->r = 0.0;
140 greg 1.1 dp->n = 0;
141 greg 1.4 if (divsample(dp, &hemi, r) < 0)
142 greg 1.1 goto oopsy;
143     if (div != NULL)
144     dp++;
145 greg 1.2 else {
146 greg 1.1 addcolor(acol, dp->v);
147 greg 1.2 arad += dp->r;
148     }
149 greg 1.1 }
150     if (ns > 0) { /* perform super-sampling */
151 greg 1.4 comperrs(div, &hemi); /* compute errors */
152 greg 1.1 qsort(div, ndivs, sizeof(AMBSAMP), ambcmp); /* sort divs */
153     /* super-sample */
154     for (i = ns; i > 0; i--) {
155     copystruct(&dnew, div);
156 greg 1.4 if (divsample(&dnew, &hemi, r) < 0)
157 greg 1.1 goto oopsy;
158     /* reinsert */
159     dp = div;
160     j = ndivs < i ? ndivs : i;
161     while (--j > 0 && dnew.k < dp[1].k) {
162     copystruct(dp, dp+1);
163     dp++;
164     }
165     copystruct(dp, &dnew);
166     }
167 greg 1.2 if (pg != NULL || dg != NULL) /* restore order */
168 greg 1.1 qsort(div, ndivs, sizeof(AMBSAMP), ambnorm);
169     }
170     /* compute returned values */
171 greg 1.3 if (div != NULL) {
172     for (i = ndivs, dp = div; i-- > 0; dp++) {
173     arad += dp->r;
174     if (dp->n > 1) {
175     b = 1.0/dp->n;
176     scalecolor(dp->v, b);
177     dp->r *= b;
178     dp->n = 1;
179     }
180     addcolor(acol, dp->v);
181     }
182 greg 1.5 b = bright(acol);
183 greg 1.6 if (b > FTINY) {
184 greg 1.5 b = ndivs/b;
185 greg 1.6 if (pg != NULL) {
186     posgradient(pg, div, &hemi);
187     for (i = 0; i < 3; i++)
188     pg[i] *= b;
189     }
190     if (dg != NULL) {
191     dirgradient(dg, div, &hemi);
192     for (i = 0; i < 3; i++)
193     dg[i] *= b;
194     }
195     } else {
196     if (pg != NULL)
197     for (i = 0; i < 3; i++)
198     pg[i] = 0.0;
199     if (dg != NULL)
200     for (i = 0; i < 3; i++)
201     dg[i] = 0.0;
202 greg 1.5 }
203 greg 1.1 free((char *)div);
204 greg 1.3 }
205 greg 1.1 b = 1.0/ndivs;
206     scalecolor(acol, b);
207     if (arad <= FTINY)
208     arad = FHUGE;
209     else
210     arad = (ndivs+ns)/arad;
211     if (arad > maxarad)
212     arad = maxarad;
213     else if (arad < minarad)
214     arad = minarad;
215     arad /= sqrt(r->rweight);
216 greg 1.7 if (pg != NULL) { /* clip pos. gradient if too large */
217     d = 4.0*DOT(pg,pg)*arad*arad;
218     if (d > 1.0) {
219     d = 1.0/sqrt(d);
220     for (i = 0; i < 3; i++)
221     pg[i] *= d;
222     }
223     }
224 greg 1.1 return(arad);
225     oopsy:
226     if (div != NULL)
227     free((char *)div);
228     return(0.0);
229     }
230    
231    
232     inithemi(hp, r) /* initialize sampling hemisphere */
233     register AMBHEMI *hp;
234     RAY *r;
235     {
236 greg 1.2 register int i;
237 greg 1.1 /* set number of divisions */
238     hp->nt = sqrt(ambdiv * r->rweight * 0.5) + 0.5;
239     hp->np = 2 * hp->nt;
240     /* make axes */
241     VCOPY(hp->uz, r->ron);
242     hp->uy[0] = hp->uy[1] = hp->uy[2] = 0.0;
243 greg 1.2 for (i = 0; i < 3; i++)
244     if (hp->uz[i] < 0.6 && hp->uz[i] > -0.6)
245 greg 1.1 break;
246 greg 1.2 if (i >= 3)
247 greg 1.1 error(CONSISTENCY, "bad ray direction in inithemi");
248 greg 1.2 hp->uy[i] = 1.0;
249 greg 1.3 fcross(hp->ux, hp->uy, hp->uz);
250     normalize(hp->ux);
251     fcross(hp->uy, hp->uz, hp->ux);
252 greg 1.1 }
253    
254    
255     comperrs(da, hp) /* compute initial error estimates */
256 greg 1.2 AMBSAMP *da; /* assumes standard ordering */
257 greg 1.1 register AMBHEMI *hp;
258     {
259     double b, b2;
260     int i, j;
261     register AMBSAMP *dp;
262     /* sum differences from neighbors */
263     dp = da;
264     for (i = 0; i < hp->nt; i++)
265     for (j = 0; j < hp->np; j++) {
266 greg 1.6 #ifdef DEBUG
267     if (dp->t != i || dp->p != j)
268     error(CONSISTENCY,
269     "division order in comperrs");
270     #endif
271 greg 1.1 b = bright(dp[0].v);
272     if (i > 0) { /* from above */
273     b2 = bright(dp[-hp->np].v) - b;
274     b2 *= b2 * 0.25;
275     dp[0].k += b2;
276     dp[-hp->np].k += b2;
277     }
278     if (j > 0) { /* from behind */
279     b2 = bright(dp[-1].v) - b;
280     b2 *= b2 * 0.25;
281     dp[0].k += b2;
282     dp[-1].k += b2;
283 greg 1.4 } else { /* around */
284     b2 = bright(dp[hp->np-1].v) - b;
285 greg 1.1 b2 *= b2 * 0.25;
286     dp[0].k += b2;
287 greg 1.4 dp[hp->np-1].k += b2;
288 greg 1.1 }
289     dp++;
290     }
291     /* divide by number of neighbors */
292     dp = da;
293     for (j = 0; j < hp->np; j++) /* top row */
294     (dp++)->k *= 1.0/3.0;
295     if (hp->nt < 2)
296     return;
297     for (i = 1; i < hp->nt-1; i++) /* central region */
298     for (j = 0; j < hp->np; j++)
299     (dp++)->k *= 0.25;
300     for (j = 0; j < hp->np; j++) /* bottom row */
301     (dp++)->k *= 1.0/3.0;
302     }
303    
304    
305     posgradient(gv, da, hp) /* compute position gradient */
306     FVECT gv;
307 greg 1.2 AMBSAMP *da; /* assumes standard ordering */
308 greg 1.1 AMBHEMI *hp;
309     {
310 greg 1.2 register int i, j;
311     double b, d;
312     double mag0, mag1;
313     double phi, cosp, sinp, xd, yd;
314     register AMBSAMP *dp;
315    
316     xd = yd = 0.0;
317     for (j = 0; j < hp->np; j++) {
318     dp = da + j;
319     mag0 = mag1 = 0.0;
320     for (i = 0; i < hp->nt; i++) {
321     #ifdef DEBUG
322     if (dp->t != i || dp->p != j)
323     error(CONSISTENCY,
324     "division order in posgradient");
325     #endif
326     b = bright(dp->v);
327     if (i > 0) {
328     d = dp[-hp->np].r;
329     if (dp[0].r > d) d = dp[0].r;
330     d *= 1.0 - sqrt((double)i/hp->nt);
331     mag0 += d*(b - bright(dp[-hp->np].v));
332     }
333     if (j > 0) {
334     d = dp[-1].r;
335     if (dp[0].r > d) d = dp[0].r;
336     mag1 += d*(b - bright(dp[-1].v));
337     } else {
338     d = dp[hp->np-1].r;
339     if (dp[0].r > d) d = dp[0].r;
340     mag1 += d*(b - bright(dp[hp->np-1].v));
341     }
342     dp += hp->np;
343     }
344     if (hp->nt > 1) {
345 greg 1.4 mag0 /= (double)hp->np;
346 greg 1.2 mag1 /= (double)hp->nt;
347     }
348     phi = 2.0*PI * (double)j/hp->np;
349     cosp = cos(phi); sinp = sin(phi);
350     xd += mag0*cosp - mag1*sinp;
351     yd += mag0*sinp + mag1*cosp;
352     }
353     for (i = 0; i < 3; i++)
354 greg 1.5 gv[i] = (xd*hp->ux[i] + yd*hp->uy[i])/PI;
355 greg 1.1 }
356    
357    
358     dirgradient(gv, da, hp) /* compute direction gradient */
359     FVECT gv;
360 greg 1.2 AMBSAMP *da; /* assumes standard ordering */
361 greg 1.1 AMBHEMI *hp;
362     {
363 greg 1.2 register int i, j;
364     double mag;
365     double phi, xd, yd;
366     register AMBSAMP *dp;
367    
368     xd = yd = 0.0;
369     for (j = 0; j < hp->np; j++) {
370     dp = da + j;
371     mag = 0.0;
372     for (i = 0; i < hp->nt; i++) {
373     #ifdef DEBUG
374     if (dp->t != i || dp->p != j)
375     error(CONSISTENCY,
376     "division order in dirgradient");
377     #endif
378     mag += sqrt((i+.5)/hp->nt)*bright(dp->v);
379     dp += hp->np;
380     }
381     phi = 2.0*PI * (j+.5)/hp->np + PI/2.0;
382     xd += mag * cos(phi);
383     yd += mag * sin(phi);
384     }
385     for (i = 0; i < 3; i++)
386 greg 1.5 gv[i] = (xd*hp->ux[i] + yd*hp->uy[i])*PI/(hp->nt*hp->np);
387 greg 1.1 }