ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/ambient.c
(Generate patch)

Comparing ray/src/rt/ambient.c (file contents):
Revision 1.13 by greg, Thu Jun 6 10:48:50 1991 UTC vs.
Revision 2.6 by greg, Thu Jul 16 12:09:08 1992 UTC

# Line 1 | Line 1
1 < /* Copyright (c) 1986 Regents of the University of California */
1 > /* Copyright (c) 1991 Regents of the University of California */
2  
3   #ifndef lint
4   static char SCCSid[] = "$SunId$ LBL";
# Line 19 | Line 19 | static char SCCSid[] = "$SunId$ LBL";
19  
20   #include  "otypes.h"
21  
22 + #include  "ambient.h"
23 +
24   #include  "random.h"
25  
26   #define  OCTSCALE       0.5     /* ceil((valid rad.)/(cube size)) */
27  
28 + typedef struct ambtree {
29 +        AMBVAL  *alist;         /* ambient value list */
30 +        struct ambtree  *kid;   /* 8 child nodes */
31 + }  AMBTREE;                     /* ambient octree */
32 +
33   extern CUBE  thescene;          /* contains space boundaries */
34  
28 – extern COLOR  ambval;           /* global ambient component */
29 – extern double  ambacc;          /* ambient accuracy */
30 – extern int  ambres;             /* ambient resolution */
31 – extern int  ambdiv;             /* number of divisions for calculation */
32 – extern int  ambssamp;           /* number of super-samples */
33 – extern int  ambounce;           /* number of ambient bounces */
34 – extern char  *amblist[];        /* ambient include/exclude list */
35 – extern int  ambincl;            /* include == 1, exclude == 0 */
36 –
35   #define  MAXASET        511     /* maximum number of elements in ambient set */
36   OBJECT  ambset[MAXASET+1]={0};  /* ambient include/exclude set */
37  
38   double  maxarad;                /* maximum ambient radius */
39   double  minarad;                /* minimum ambient radius */
40  
43 – /*
44 – * Since we've defined our vectors as float below to save space,
45 – * watch out for changes in the definitions of VCOPY() and DOT().
46 – */
47 – typedef struct ambval {
48 –        float  pos[3];          /* position in space */
49 –        float  dir[3];          /* normal direction */
50 –        int  lvl;               /* recursion level of parent ray */
51 –        float  weight;          /* weight of parent ray */
52 –        COLOR  val;             /* computed ambient value */
53 –        float  rad;             /* validity radius */
54 –        struct ambval  *next;   /* next in list */
55 – }  AMBVAL;                      /* ambient value */
56 –
57 – typedef struct ambtree {
58 –        AMBVAL  *alist;         /* ambient value list */
59 –        struct ambtree  *kid;   /* 8 child nodes */
60 – }  AMBTREE;                     /* ambient octree */
61 –
62 – typedef struct {
63 –        float  k;               /* error contribution per sample */
64 –        COLOR  v;               /* ray sum */
65 –        int  n;                 /* number of samples */
66 –        short  t, p;            /* theta, phi indices */
67 – }  AMBSAMP;                     /* ambient sample */
68 –
41   static AMBTREE  atrunk;         /* our ambient trunk node */
42  
43   static FILE  *ambfp = NULL;     /* ambient file pointer */
44 + static int  ambheadlen;         /* length of ambient file header */
45  
46 + #define  AMBFLUSH       (BUFSIZ/AMBVALSIZ)
47 +
48   #define  newambval()    (AMBVAL *)bmalloc(sizeof(AMBVAL))
49  
50   #define  newambtree()   (AMBTREE *)calloc(8, sizeof(AMBTREE))
51  
77 – double  sumambient(), doambient(), makeambient();
52  
53 + setambres(ar)                           /* set ambient resolution */
54 + int  ar;
55 + {
56 +                                                /* set min & max radii */
57 +        if (ar <= 0) {
58 +                minarad = 0.0;
59 +                maxarad = thescene.cusize / 2.0;
60 +        } else {
61 +                minarad = thescene.cusize / ar;
62 +                maxarad = 16.0 * minarad;               /* heuristic */
63 +                if (maxarad > thescene.cusize / 2.0)
64 +                        maxarad = thescene.cusize / 2.0;
65 +        }
66 +        if (maxarad <= FTINY)
67 +                maxarad = .001;
68 + }
69  
70 +
71   setambient(afile)                       /* initialize calculation */
72   char  *afile;
73   {
74 <        long  ftell();
74 >        extern long  ftell();
75          AMBVAL  amb;
76 <
77 <        maxarad = thescene.cusize / 2.0;                /* maximum radius */
78 <                                                        /* minimum radius */
88 <        minarad = ambres > 0 ? thescene.cusize/ambres : 0.0;
89 <
90 <                                        /* open ambient file */
76 >                                                /* init ambient limits */
77 >        setambres(ambres);
78 >                                                /* open ambient file */
79          if (afile != NULL)
80                  if ((ambfp = fopen(afile, "r+")) != NULL) {
81 <                        while (fread((char *)&amb,sizeof(AMBVAL),1,ambfp) == 1)
81 >                        initambfile(0);
82 >                        while (readambval(&amb, ambfp))
83                                  avinsert(&amb, &atrunk, thescene.cuorg,
84                                                  thescene.cusize);
85                                                          /* align */
86 <                        fseek(ambfp, -(ftell(ambfp)%sizeof(AMBVAL)), 1);
87 <                } else if ((ambfp = fopen(afile, "w")) == NULL) {
86 >                        fseek(ambfp, -((ftell(ambfp)-ambheadlen)%AMBVALSIZ), 1);
87 >                } else if ((ambfp = fopen(afile, "w")) != NULL)
88 >                        initambfile(1);
89 >                else {
90                          sprintf(errmsg, "cannot open ambient file \"%s\"",
91                                          afile);
92                          error(SYSTEM, errmsg);
# Line 103 | Line 94 | char  *afile;
94   }
95  
96  
97 + initambfile(creat)              /* initialize ambient file */
98 + int  creat;
99 + {
100 +        extern char  *progname, *octname;
101 +
102 +        setbuf(ambfp, bmalloc(BUFSIZ));
103 +        if (creat) {                    /* new file */
104 +                fprintf(ambfp, "%s -av %g %g %g -ab %d -aa %g ",
105 +                                progname, colval(ambval,RED),
106 +                                colval(ambval,GRN), colval(ambval,BLU),
107 +                                ambounce, ambacc);
108 +                fprintf(ambfp, "-ad %d -as %d -ar %d %s\n",
109 +                                ambdiv, ambssamp, ambres,
110 +                                octname==NULL ? "" : octname);
111 +                fputformat(AMBFMT, ambfp);
112 +                putc('\n', ambfp);
113 +                putambmagic(ambfp);
114 +                fflush(ambfp);
115 +        } else if (checkheader(ambfp, AMBFMT, NULL) < 0 || !hasambmagic(ambfp))
116 +                error(USER, "ambient file format error");
117 +        ambheadlen = ftell(ambfp);
118 + }
119 +
120 +
121   ambnotify(obj)                  /* record new modifier */
122   OBJECT  obj;
123   {
# Line 130 | Line 145 | COLOR  acol;
145   register RAY  *r;
146   {
147          static int  rdepth = 0;                 /* ambient recursion */
148 <        double  wsum;
148 >        double  d;
149  
135 –        rdepth++;                               /* increment level */
136 –
150          if (ambdiv <= 0)                        /* no ambient calculation */
151                  goto dumbamb;
152                                                  /* check number of bounces */
153 <        if (rdepth > ambounce)
153 >        if (rdepth >= ambounce)
154                  goto dumbamb;
155                                                  /* check ambient list */
156          if (ambincl != -1 && r->ro != NULL &&
# Line 145 | Line 158 | register RAY  *r;
158                  goto dumbamb;
159  
160          if (ambacc <= FTINY) {                  /* no ambient storage */
161 <                if (doambient(acol, r) == 0.0)
161 >                rdepth++;
162 >                d = doambient(acol, r, r->rweight, NULL, NULL);
163 >                rdepth--;
164 >                if (d == 0.0)
165                          goto dumbamb;
166 <                goto done;
166 >                return;
167          }
168                                                  /* get ambient value */
169          setcolor(acol, 0.0, 0.0, 0.0);
170 <        wsum = sumambient(acol, r, &atrunk, thescene.cuorg, thescene.cusize);
171 <        if (wsum > FTINY)
172 <                scalecolor(acol, 1.0/wsum);
173 <        else if (makeambient(acol, r) == 0.0)
174 <                goto dumbamb;
175 <        goto done;
176 <
170 >        d = sumambient(acol, r, rdepth,
171 >                        &atrunk, thescene.cuorg, thescene.cusize);
172 >        if (d > FTINY)
173 >                scalecolor(acol, 1.0/d);
174 >        else {
175 >                d = makeambient(acol, r, rdepth++);
176 >                rdepth--;
177 >        }
178 >        if (d > FTINY)
179 >                return;
180   dumbamb:                                        /* return global value */
181          copycolor(acol, ambval);
163 – done:                                           /* must finish here! */
164 –        rdepth--;
182   }
183  
184  
185   double
186 < sumambient(acol, r, at, c0, s)          /* get interpolated ambient value */
186 > sumambient(acol, r, al, at, c0, s)      /* get interpolated ambient value */
187   COLOR  acol;
188   register RAY  *r;
189 + int  al;
190   AMBTREE  *at;
191   FVECT  c0;
192   double  s;
# Line 184 | Line 202 | double  s;
202          wsum = 0.0;
203          for (av = at->alist; av != NULL; av = av->next) {
204                  /*
205 <                 *  Ray strength test.
205 >                 *  Ambient level test.
206                   */
207 <                if (av->lvl > r->rlvl || av->weight < r->rweight-FTINY)
207 >                if (av->lvl > al || av->weight < r->rweight-FTINY)
208                          continue;
209                  /*
210                   *  Ambient radius test.
# Line 213 | Line 231 | double  s;
231                  for (j = 0; j < 3; j++)
232                          d += (r->rop[j] - av->pos[j]) *
233                                          (av->dir[j] + r->ron[j]);
234 <                if (d*0.5 < -minarad*ambacc)
234 >                if (d*0.5 < -minarad*ambacc-.001)
235                          continue;
236                  /*
237                   *  Jittering final test reduces image artifacts.
238                   */
239                  wt = sqrt(e1) + sqrt(e2);
240 <                wt *= .9 + .2*frandom();
240 >                wt *= .9 + .2*urand(9015+samplendx);
241                  if (wt > ambacc)
242                          continue;
243                  if (wt <= 1e-3)
# Line 227 | Line 245 | double  s;
245                  else
246                          wt = 1.0 / wt;
247                  wsum += wt;
248 <                copycolor(ct, av->val);
248 >                extambient(ct, av, r->rop, r->ron);
249                  scalecolor(ct, wt);
250                  addcolor(acol, ct);
251          }
# Line 246 | Line 264 | double  s;
264                                  break;
265                  }
266                  if (j == 3)
267 <                        wsum += sumambient(acol, r, at->kid+i, ck0, s);
267 >                        wsum += sumambient(acol, r, al, at->kid+i, ck0, s);
268          }
269          return(wsum);
270   }
271  
272  
273   double
274 < makeambient(acol, r)            /* make a new ambient value */
274 > makeambient(acol, r, al)        /* make a new ambient value */
275   COLOR  acol;
276   register RAY  *r;
277 + int  al;
278   {
279          AMBVAL  amb;
280 <
281 <        amb.rad = doambient(acol, r);           /* compute ambient */
280 >        FVECT   gp, gd;
281 >                                                /* compute weight */
282 >        amb.weight = pow(AVGREFL, (double)al);
283 >        if (r->rweight < 0.2*amb.weight)        /* heuristic */
284 >                amb.weight = r->rweight;
285 >                                                /* compute ambient */
286 >        amb.rad = doambient(acol, r, amb.weight, gp, gd);
287          if (amb.rad == 0.0)
288                  return(0.0);
289                                                  /* store it */
290          VCOPY(amb.pos, r->rop);
291          VCOPY(amb.dir, r->ron);
292 <        amb.lvl = r->rlvl;
269 <        amb.weight = r->rweight;
292 >        amb.lvl = al;
293          copycolor(amb.val, acol);
294 +        VCOPY(amb.gpos, gp);
295 +        VCOPY(amb.gdir, gd);
296                                                  /* insert into tree */
297          avinsert(&amb, &atrunk, thescene.cuorg, thescene.cusize);
298          avsave(&amb);                           /* write to file */
# Line 275 | Line 300 | register RAY  *r;
300   }
301  
302  
303 < double
304 < doambient(acol, r)                      /* compute ambient component */
305 < COLOR  acol;
306 < register RAY  *r;
303 > extambient(cr, ap, pv, nv)              /* extrapolate value at pv, nv */
304 > COLOR  cr;
305 > register AMBVAL  *ap;
306 > FVECT  pv, nv;
307   {
308 <        extern int  ambcmp();
309 <        extern double  sin(), cos(), sqrt();
310 <        int  hlist[4];
286 <        double  phi, xd, yd, zd;
287 <        double  b, b2;
288 <        register AMBSAMP  *div;
289 <        AMBSAMP  dnew;
290 <        RAY  ar;
291 <        FVECT  ux, uy;
292 <        double  arad;
293 <        int  ndivs, nt, np, ns, ne, i, j;
294 <        register int  k;
308 >        FVECT  v1, v2;
309 >        register int  i;
310 >        double  d;
311  
312 <        setcolor(acol, 0.0, 0.0, 0.0);
313 <                                        /* set number of divisions */
314 <        nt = sqrt(ambdiv * r->rweight * 0.5) + 0.5;
315 <        np = 2 * nt;
316 <        ndivs = nt * np;
317 <                                        /* check first */
318 <        if (ndivs == 0 || rayorigin(&ar, r, AMBIENT, 0.5) < 0)
319 <                return(0.0);
320 <                                        /* set number of super-samples */
321 <        ns = ambssamp * r->rweight + 0.5;
322 <        if (ns > 0) {
307 <                div = (AMBSAMP *)malloc(ndivs*sizeof(AMBSAMP));
308 <                if (div == NULL)
309 <                        error(SYSTEM, "out of memory in doambient");
310 <        } else
311 <                div = NULL;
312 <                                        /* make axes */
313 <        uy[0] = uy[1] = uy[2] = 0.0;
314 <        for (k = 0; k < 3; k++)
315 <                if (r->ron[k] < 0.6 && r->ron[k] > -0.6)
316 <                        break;
317 <        uy[k] = 1.0;
318 <        fcross(ux, r->ron, uy);
319 <        normalize(ux);
320 <        fcross(uy, ux, r->ron);
321 <                                        /* set up urand */
322 <        hlist[0] = r->rno;
323 <                                                /* sample divisions */
324 <        arad = 0.0;
325 <        ne = 0;
326 <        for (i = 0; i < nt; i++) {
327 <                hlist[1] = i;
328 <                for (j = 0; j < np; j++) {
329 <                        rayorigin(&ar, r, AMBIENT, 0.5);        /* pretested */
330 <                        hlist[2] = j;
331 <                        hlist[3] = 0;
332 <                        zd = sqrt((i+urand(ilhash(hlist,4)))/nt);
333 <                        hlist[3] = 1;
334 <                        phi = 2.0*PI * (j+urand(ilhash(hlist,4)))/np;
335 <                        xd = cos(phi) * zd;
336 <                        yd = sin(phi) * zd;
337 <                        zd = sqrt(1.0 - zd*zd);
338 <                        for (k = 0; k < 3; k++)
339 <                                ar.rdir[k] = xd*ux[k]+yd*uy[k]+zd*r->ron[k];
340 <                        dimlist[ndims++] = i*np + j + 38813;
341 <                        rayvalue(&ar);
342 <                        ndims--;
343 <                        if (ar.rot < FHUGE)
344 <                                arad += 1.0 / ar.rot;
345 <                        if (div != NULL) {              /* save division */
346 <                                div[ne].k = 0.0;
347 <                                copycolor(div[ne].v, ar.rcol);
348 <                                div[ne].n = 0;
349 <                                div[ne].t = i; div[ne].p = j;
350 <                                                        /* sum errors */
351 <                                b = bright(ar.rcol);
352 <                                if (i > 0) {            /* from above */
353 <                                        b2 = bright(div[ne-np].v) - b;
354 <                                        b2 *= b2 * 0.25;
355 <                                        div[ne].k += b2;
356 <                                        div[ne].n++;
357 <                                        div[ne-np].k += b2;
358 <                                        div[ne-np].n++;
359 <                                }
360 <                                if (j > 0) {            /* from behind */
361 <                                        b2 = bright(div[ne-1].v) - b;
362 <                                        b2 *= b2 * 0.25;
363 <                                        div[ne].k += b2;
364 <                                        div[ne].n++;
365 <                                        div[ne-1].k += b2;
366 <                                        div[ne-1].n++;
367 <                                }
368 <                                if (j == np-1) {        /* around */
369 <                                        b2 = bright(div[ne-(np-1)].v) - b;
370 <                                        b2 *= b2 * 0.25;
371 <                                        div[ne].k += b2;
372 <                                        div[ne].n++;
373 <                                        div[ne-(np-1)].k += b2;
374 <                                        div[ne-(np-1)].n++;
375 <                                }
376 <                                ne++;
377 <                        } else
378 <                                addcolor(acol, ar.rcol);
379 <                }
312 >        d = 1.0;                        /* zeroeth order */
313 >                                        /* gradient due to translation */
314 >        for (i = 0; i < 3; i++)
315 >                d += ap->gpos[i]*(pv[i]-ap->pos[i]);
316 >                                        /* gradient due to rotation */
317 >        VCOPY(v1, ap->dir);
318 >        fcross(v2, v1, nv);
319 >        d += DOT(ap->gdir, v2);
320 >        if (d <= 0.0) {
321 >                setcolor(cr, 0.0, 0.0, 0.0);
322 >                return;
323          }
324 <        for (k = 0; k < ne; k++) {              /* compute errors */
325 <                if (div[k].n > 1)
383 <                        div[k].k /= div[k].n;
384 <                div[k].n = 1;
385 <        }
386 <                                                /* sort the divisions */
387 <        qsort(div, ne, sizeof(AMBSAMP), ambcmp);
388 <                                                /* skim excess */
389 <        while (ne > ns) {
390 <                ne--;
391 <                addcolor(acol, div[ne].v);
392 <        }
393 <                                                /* super-sample */
394 <        for (i = ns; i > 0; i--) {
395 <                rayorigin(&ar, r, AMBIENT, 0.5);        /* pretested */
396 <                hlist[1] = div[0].t;
397 <                hlist[2] = div[0].p;
398 <                hlist[3] = 0;
399 <                zd = sqrt((div[0].t+urand(ilhash(hlist,4)+div[0].n))/nt);
400 <                hlist[3] = 1;
401 <                phi = 2.0*PI * (div[0].p+urand(ilhash(hlist,4)+div[0].n))/np;
402 <                xd = cos(phi) * zd;
403 <                yd = sin(phi) * zd;
404 <                zd = sqrt(1.0 - zd*zd);
405 <                for (k = 0; k < 3; k++)
406 <                        ar.rdir[k] = xd*ux[k]+yd*uy[k]+zd*r->ron[k];
407 <                dimlist[ndims++] = div[0].t*np + div[0].p + 38813;
408 <                rayvalue(&ar);
409 <                ndims--;
410 <                rayvalue(&ar);
411 <                if (ar.rot < FHUGE)
412 <                        arad += 1.0 / ar.rot;
413 <                                                /* recompute error */
414 <                copycolor(dnew.v, div[0].v);
415 <                addcolor(dnew.v, ar.rcol);
416 <                dnew.n = div[0].n + 1;
417 <                dnew.t = div[0].t; dnew.p = div[0].p;
418 <                b2 = bright(dnew.v)/dnew.n - bright(ar.rcol);
419 <                b2 = b2*b2 + div[0].k*(div[0].n*div[0].n);
420 <                dnew.k = b2/(dnew.n*dnew.n);
421 <                                                /* reinsert */
422 <                for (k = 0; k < ne-1 && dnew.k < div[k+1].k; k++)
423 <                        copystruct(&div[k], &div[k+1]);
424 <                copystruct(&div[k], &dnew);
425 <
426 <                if (ne >= i) {          /* extract darkest division */
427 <                        ne--;
428 <                        if (div[ne].n > 1) {
429 <                                b = 1.0/div[ne].n;
430 <                                scalecolor(div[ne].v, b);
431 <                                div[ne].n = 1;
432 <                        }
433 <                        addcolor(acol, div[ne].v);
434 <                }
435 <        }
436 <        scalecolor(acol, 1.0/ndivs);
437 <        if (arad <= FTINY)
438 <                arad = FHUGE;
439 <        else
440 <                arad = (ndivs+ns) / arad / sqrt(r->rweight);
441 <        if (arad > maxarad)
442 <                arad = maxarad;
443 <        else if (arad < minarad)
444 <                arad = minarad;
445 <        if (div != NULL)
446 <                free((char *)div);
447 <        return(arad);
324 >        copycolor(cr, ap->val);
325 >        scalecolor(cr, d);
326   }
327  
328  
451 – static int
452 – ambcmp(d1, d2)                          /* decreasing order */
453 – AMBSAMP  *d1, *d2;
454 – {
455 –        if (d1->k < d2->k)
456 –                return(1);
457 –        if (d1->k > d2->k)
458 –                return(-1);
459 –        return(0);
460 – }
461 –
462 –
329   static
330   avsave(av)                              /* save an ambient value */
331   AMBVAL  *av;
332   {
467 – #ifdef  AMBFLUSH
333          static int  nunflshed = 0;
334 < #endif
334 >
335          if (ambfp == NULL)
336                  return;
337 <        if (fwrite((char *)av, sizeof(AMBVAL), 1, ambfp) != 1)
337 >        if (writambval(av, ambfp) < 0)
338                  goto writerr;
474 – #ifdef  AMBFLUSH
339          if (++nunflshed >= AMBFLUSH) {
340                  if (fflush(ambfp) == EOF)
341                          goto writerr;
342                  nunflshed = 0;
343          }
480 – #endif
344          return;
345   writerr:
346          error(SYSTEM, "error writing ambient file");

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines