ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/ambient.c
Revision: 2.16
Committed: Fri Jan 22 09:51:13 1993 UTC (31 years, 3 months ago) by greg
Content type: text/plain
Branch: MAIN
Changes since 2.15: +22 -13 lines
Log Message:
bug fixes and improvements on stdio flushing

File Contents

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