ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/ambcomp.c
Revision: 2.4
Committed: Fri Oct 2 16:14:38 1992 UTC (31 years, 7 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.3: +0 -2 lines
Log Message:
Removed problematic math function declarations

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