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

File Contents

# User Rev Content
1 greg 2.12 /* Copyright (c) 1992 Regents of the University of California */
2 greg 1.1
3     #ifndef lint
4     static char SCCSid[] = "$SunId$ LBL";
5     #endif
6    
7     /*
8     * ambient.c - routines dealing with ambient (inter-reflected) component.
9     *
10     * 5/9/86
11     */
12    
13     #include "ray.h"
14    
15     #include "octree.h"
16    
17 greg 1.8 #include "otypes.h"
18    
19 greg 1.14 #include "ambient.h"
20    
21 greg 1.1 #include "random.h"
22    
23 greg 2.12 #define OCTSCALE 0.5 /* ceil((valid rad.)/(cube size)) */
24 greg 1.1
25 greg 1.14 typedef struct ambtree {
26 greg 2.12 AMBVAL *alist; /* ambient value list */
27     struct ambtree *kid; /* 8 child nodes */
28 greg 1.14 } AMBTREE; /* ambient octree */
29    
30 greg 1.1 extern CUBE thescene; /* contains space boundaries */
31    
32 greg 2.12 #define MAXASET 511 /* maximum number of elements in ambient set */
33     OBJECT ambset[MAXASET+1]={0}; /* ambient include/exclude set */
34 greg 1.1
35 greg 2.12 double maxarad; /* maximum ambient radius */
36     double minarad; /* minimum ambient radius */
37 greg 1.1
38 greg 2.12 static AMBTREE atrunk; /* our ambient trunk node */
39 greg 1.1
40     static FILE *ambfp = NULL; /* ambient file pointer */
41    
42 greg 2.12 #define AMBFLUSH (BUFSIZ/AMBVALSIZ)
43 greg 2.6
44 greg 2.12 #define newambval() (AMBVAL *)bmalloc(sizeof(AMBVAL))
45 greg 1.1
46 greg 2.12 #define newambtree() (AMBTREE *)calloc(8, sizeof(AMBTREE))
47 greg 1.1
48 greg 2.7 extern long ftell(), lseek();
49 greg 2.10 static int initambfile(), avsave(), avinsert(), ambsync();
50 greg 1.1
51 greg 2.7
52 greg 2.3 setambres(ar) /* set ambient resolution */
53     int ar;
54     {
55     /* set min & max radii */
56     if (ar <= 0) {
57     minarad = 0.0;
58     maxarad = thescene.cusize / 2.0;
59     } else {
60     minarad = thescene.cusize / ar;
61     maxarad = 16.0 * minarad; /* heuristic */
62     if (maxarad > thescene.cusize / 2.0)
63     maxarad = thescene.cusize / 2.0;
64     }
65 greg 2.4 if (maxarad <= FTINY)
66     maxarad = .001;
67 greg 2.3 }
68    
69    
70 greg 1.1 setambient(afile) /* initialize calculation */
71     char *afile;
72     {
73 greg 2.9 long headlen;
74 greg 2.12 AMBVAL amb;
75 greg 2.3 /* init ambient limits */
76     setambres(ambres);
77     /* open ambient file */
78 greg 2.10 if (afile != NULL)
79 greg 1.1 if ((ambfp = fopen(afile, "r+")) != NULL) {
80 greg 2.6 initambfile(0);
81 greg 2.9 headlen = ftell(ambfp);
82 greg 2.5 while (readambval(&amb, ambfp))
83 greg 1.1 avinsert(&amb, &atrunk, thescene.cuorg,
84     thescene.cusize);
85     /* align */
86 greg 2.9 fseek(ambfp, -((ftell(ambfp)-headlen)%AMBVALSIZ), 1);
87 greg 2.10 } else if ((ambfp = fopen(afile, "w+")) != NULL)
88 greg 2.6 initambfile(1);
89     else {
90 greg 1.1 sprintf(errmsg, "cannot open ambient file \"%s\"",
91     afile);
92     error(SYSTEM, errmsg);
93 greg 1.8 }
94     }
95    
96    
97     ambnotify(obj) /* record new modifier */
98 greg 2.12 OBJECT obj;
99 greg 1.8 {
100 greg 1.11 static int hitlimit = 0;
101 greg 2.12 register OBJREC *o = objptr(obj);
102 greg 1.8 register char **amblp;
103    
104 greg 1.11 if (hitlimit || !ismodifier(o->otype))
105 greg 1.8 return;
106     for (amblp = amblist; *amblp != NULL; amblp++)
107     if (!strcmp(o->oname, *amblp)) {
108 greg 1.11 if (ambset[0] >= MAXASET) {
109     error(WARNING, "too many modifiers in ambient list");
110     hitlimit++;
111     return; /* should this be fatal? */
112     }
113 greg 1.8 insertelem(ambset, obj);
114     return;
115 greg 1.1 }
116     }
117    
118    
119     ambient(acol, r) /* compute ambient component for ray */
120     COLOR acol;
121     register RAY *r;
122     {
123     static int rdepth = 0; /* ambient recursion */
124 greg 2.12 double d;
125 greg 1.1
126     if (ambdiv <= 0) /* no ambient calculation */
127     goto dumbamb;
128     /* check number of bounces */
129 greg 1.16 if (rdepth >= ambounce)
130 greg 1.1 goto dumbamb;
131     /* check ambient list */
132     if (ambincl != -1 && r->ro != NULL &&
133     ambincl != inset(ambset, r->ro->omod))
134     goto dumbamb;
135    
136     if (ambacc <= FTINY) { /* no ambient storage */
137 greg 1.16 rdepth++;
138     d = doambient(acol, r, r->rweight, NULL, NULL);
139     rdepth--;
140     if (d == 0.0)
141 greg 1.1 goto dumbamb;
142 greg 1.16 return;
143 greg 1.1 }
144     /* get ambient value */
145     setcolor(acol, 0.0, 0.0, 0.0);
146 greg 1.16 d = sumambient(acol, r, rdepth,
147     &atrunk, thescene.cuorg, thescene.cusize);
148     if (d > FTINY)
149     scalecolor(acol, 1.0/d);
150     else {
151     d = makeambient(acol, r, rdepth++);
152     rdepth--;
153     }
154     if (d > FTINY)
155     return;
156 greg 1.1 dumbamb: /* return global value */
157     copycolor(acol, ambval);
158     }
159    
160    
161     double
162 greg 1.16 sumambient(acol, r, al, at, c0, s) /* get interpolated ambient value */
163 greg 1.1 COLOR acol;
164     register RAY *r;
165 greg 1.16 int al;
166 greg 2.12 AMBTREE *at;
167 greg 1.1 FVECT c0;
168 greg 2.12 double s;
169 greg 1.1 {
170 greg 2.12 double d, e1, e2, wt, wsum;
171 greg 1.1 COLOR ct;
172     FVECT ck0;
173     int i;
174     register int j;
175 greg 2.12 register AMBVAL *av;
176 greg 1.7 /* do this node */
177 greg 1.1 wsum = 0.0;
178     for (av = at->alist; av != NULL; av = av->next) {
179     /*
180 greg 1.16 * Ambient level test.
181 greg 1.1 */
182 greg 1.16 if (av->lvl > al || av->weight < r->rweight-FTINY)
183 greg 1.1 continue;
184     /*
185     * Ambient radius test.
186     */
187     e1 = 0.0;
188     for (j = 0; j < 3; j++) {
189     d = av->pos[j] - r->rop[j];
190     e1 += d * d;
191     }
192     e1 /= av->rad * av->rad;
193     if (e1 > ambacc*ambacc*1.21)
194     continue;
195     /*
196     * Normal direction test.
197     */
198     e2 = (1.0 - DOT(av->dir, r->ron)) * r->rweight;
199     if (e2 < 0.0) e2 = 0.0;
200     if (e1 + e2 > ambacc*ambacc*1.21)
201     continue;
202     /*
203     * Ray behind test.
204     */
205     d = 0.0;
206     for (j = 0; j < 3; j++)
207     d += (r->rop[j] - av->pos[j]) *
208     (av->dir[j] + r->ron[j]);
209 greg 1.18 if (d*0.5 < -minarad*ambacc-.001)
210 greg 1.1 continue;
211     /*
212     * Jittering final test reduces image artifacts.
213     */
214     wt = sqrt(e1) + sqrt(e2);
215 greg 2.2 wt *= .9 + .2*urand(9015+samplendx);
216 greg 1.6 if (wt > ambacc)
217 greg 1.1 continue;
218     if (wt <= 1e-3)
219     wt = 1e3;
220     else
221     wt = 1.0 / wt;
222     wsum += wt;
223 greg 1.15 extambient(ct, av, r->rop, r->ron);
224 greg 1.1 scalecolor(ct, wt);
225     addcolor(acol, ct);
226 greg 1.7 }
227     if (at->kid == NULL)
228     return(wsum);
229     /* do children */
230     s *= 0.5;
231     for (i = 0; i < 8; i++) {
232     for (j = 0; j < 3; j++) {
233     ck0[j] = c0[j];
234     if (1<<j & i)
235     ck0[j] += s;
236     if (r->rop[j] < ck0[j] - OCTSCALE*s)
237     break;
238     if (r->rop[j] > ck0[j] + (1.0+OCTSCALE)*s)
239     break;
240     }
241     if (j == 3)
242 greg 1.16 wsum += sumambient(acol, r, al, at->kid+i, ck0, s);
243 greg 1.1 }
244     return(wsum);
245     }
246    
247    
248     double
249 greg 1.16 makeambient(acol, r, al) /* make a new ambient value */
250 greg 1.1 COLOR acol;
251     register RAY *r;
252 greg 1.16 int al;
253 greg 1.1 {
254 greg 2.12 AMBVAL amb;
255 greg 1.14 FVECT gp, gd;
256 greg 1.16 /* compute weight */
257     amb.weight = pow(AVGREFL, (double)al);
258 greg 1.17 if (r->rweight < 0.2*amb.weight) /* heuristic */
259 greg 1.16 amb.weight = r->rweight;
260     /* compute ambient */
261     amb.rad = doambient(acol, r, amb.weight, gp, gd);
262 greg 1.1 if (amb.rad == 0.0)
263     return(0.0);
264     /* store it */
265     VCOPY(amb.pos, r->rop);
266     VCOPY(amb.dir, r->ron);
267 greg 1.16 amb.lvl = al;
268 greg 1.1 copycolor(amb.val, acol);
269 greg 1.14 VCOPY(amb.gpos, gp);
270     VCOPY(amb.gdir, gd);
271 greg 1.1 /* insert into tree */
272 greg 2.7 avsave(&amb); /* and save to file */
273 greg 1.1 return(amb.rad);
274 greg 1.15 }
275    
276    
277     extambient(cr, ap, pv, nv) /* extrapolate value at pv, nv */
278     COLOR cr;
279 greg 2.12 register AMBVAL *ap;
280 greg 1.15 FVECT pv, nv;
281     {
282     FVECT v1, v2;
283     register int i;
284 greg 2.12 double d;
285 greg 1.15
286     d = 1.0; /* zeroeth order */
287     /* gradient due to translation */
288     for (i = 0; i < 3; i++)
289     d += ap->gpos[i]*(pv[i]-ap->pos[i]);
290     /* gradient due to rotation */
291     VCOPY(v1, ap->dir);
292     fcross(v2, v1, nv);
293     d += DOT(ap->gdir, v2);
294     if (d <= 0.0) {
295     setcolor(cr, 0.0, 0.0, 0.0);
296     return;
297     }
298     copycolor(cr, ap->val);
299     scalecolor(cr, d);
300 greg 1.1 }
301    
302    
303     static
304 greg 2.9 initambfile(creat) /* initialize ambient file */
305     int creat;
306     {
307     extern char *progname, *octname, VersionID[];
308    
309 greg 2.12 #ifdef MSDOS
310     setmode(fileno(ambfp), O_BINARY);
311     #endif
312 greg 2.9 setbuf(ambfp, bmalloc(BUFSIZ));
313     if (creat) { /* new file */
314     fprintf(ambfp, "%s -av %g %g %g -ab %d -aa %g ",
315     progname, colval(ambval,RED),
316     colval(ambval,GRN), colval(ambval,BLU),
317     ambounce, ambacc);
318     fprintf(ambfp, "-ad %d -as %d -ar %d %s\n",
319     ambdiv, ambssamp, ambres,
320     octname==NULL ? "" : octname);
321     fprintf(ambfp, "SOFTWARE= %s\n", VersionID);
322     fputformat(AMBFMT, ambfp);
323     putc('\n', ambfp);
324     putambmagic(ambfp);
325     fflush(ambfp);
326 greg 2.12 #ifndef NIX
327 greg 2.11 sync(); /* protect against NFS buffering */
328     #endif
329 greg 2.10 } else if (checkheader(ambfp, AMBFMT, NULL) < 0 || !hasambmagic(ambfp))
330     error(USER, "bad ambient file");
331 greg 2.9 }
332    
333    
334     static
335 greg 2.7 avsave(av) /* insert and save an ambient value */
336 greg 2.12 AMBVAL *av;
337 greg 1.1 {
338     static int nunflshed = 0;
339 greg 2.6
340 greg 2.7 avinsert(av, &atrunk, thescene.cuorg, thescene.cusize);
341 greg 1.1 if (ambfp == NULL)
342     return;
343 greg 2.5 if (writambval(av, ambfp) < 0)
344 greg 1.1 goto writerr;
345     if (++nunflshed >= AMBFLUSH) {
346 greg 2.7 if (ambsync() == EOF)
347 greg 1.1 goto writerr;
348     nunflshed = 0;
349     }
350     return;
351     writerr:
352     error(SYSTEM, "error writing ambient file");
353     }
354    
355    
356     static
357     avinsert(aval, at, c0, s) /* insert ambient value in a tree */
358 greg 2.12 AMBVAL *aval;
359 greg 1.1 register AMBTREE *at;
360     FVECT c0;
361 greg 2.12 double s;
362 greg 1.1 {
363     FVECT ck0;
364     int branch;
365 greg 2.12 register AMBVAL *av;
366 greg 1.1 register int i;
367    
368     if ((av = newambval()) == NULL)
369     goto memerr;
370 greg 1.9 copystruct(av, aval);
371 greg 1.1 VCOPY(ck0, c0);
372     while (s*(OCTSCALE/2) > av->rad*ambacc) {
373     if (at->kid == NULL)
374     if ((at->kid = newambtree()) == NULL)
375     goto memerr;
376     s *= 0.5;
377     branch = 0;
378     for (i = 0; i < 3; i++)
379     if (av->pos[i] > ck0[i] + s) {
380     ck0[i] += s;
381     branch |= 1 << i;
382     }
383     at = at->kid + branch;
384     }
385     av->next = at->alist;
386     at->alist = av;
387     return;
388     memerr:
389     error(SYSTEM, "out of memory in avinsert");
390 greg 2.7 }
391    
392    
393 greg 2.12 #ifdef NIX
394 greg 2.10
395     static
396     ambsync() /* flush ambient file */
397     {
398     return(fflush(ambfp));
399     }
400    
401     #else
402    
403 greg 2.7 #include <fcntl.h>
404 greg 2.10 #include <sys/types.h>
405     #include <sys/stat.h>
406 greg 2.7
407     static
408     ambsync() /* synchronize ambient file */
409     {
410     static FILE *ambinp = NULL;
411     struct flock fls;
412 greg 2.10 struct stat sts;
413 greg 2.12 AMBVAL avs;
414 greg 2.7 long lastpos, flen;
415     register int n;
416     /* gain exclusive access */
417     fls.l_type = F_WRLCK;
418     fls.l_whence = 0;
419     fls.l_start = 0L;
420     fls.l_len = 0L;
421     if (fcntl(fileno(ambfp), F_SETLKW, &fls) < 0)
422     error(SYSTEM, "cannot lock ambient file");
423     /* see if file has grown */
424 greg 2.9 lastpos = lseek(fileno(ambfp), 0L, 1); /* get previous position */
425 greg 2.10 if (fstat(fileno(ambfp), &sts) < 0) /* get current length */
426     error(SYSTEM, "cannot stat ambient file");
427     flen = sts.st_size;
428 greg 2.7 if (n = (flen - lastpos)/AMBVALSIZ) { /* file has grown */
429 greg 2.10 if (ambinp == NULL) /* use duplicate file */
430     ambinp = fdopen(dup(fileno(ambfp)), "r");
431 greg 2.7 while (n--) { /* load contributed values */
432     readambval(&avs, ambinp);
433     avinsert(&avs,&atrunk,thescene.cuorg,thescene.cusize);
434 greg 2.10 } /* moves shared file pointer */
435     if (n = (flen - lastpos)%AMBVALSIZ) /* alignment */
436     lseek(fileno(ambfp), flen-n, 0);
437 greg 2.7 }
438     n = fflush(ambfp); /* calls write() at last */
439     fls.l_type = F_UNLCK; /* release file */
440     fcntl(fileno(ambfp), F_SETLKW, &fls);
441     return(n);
442 greg 1.1 }
443 greg 2.10
444     #endif