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

Comparing ray/src/rt/ambcomp.c (file contents):
Revision 2.54 by greg, Fri May 9 04:55:19 2014 UTC vs.
Revision 2.56 by greg, Fri May 9 20:05:00 2014 UTC

# Line 25 | Line 25 | static const char      RCSid[] = "$Id$";
25  
26   extern void             SDsquare2disk(double ds[2], double seedx, double seedy);
27  
28 –                                /* vertex direction bit positions */
29 – #define VDB_xy  0
30 – #define VDB_y   01
31 – #define VDB_x   02
32 – #define VDB_Xy  03
33 – #define VDB_xY  04
34 – #define VDB_X   05
35 – #define VDB_Y   06
36 – #define VDB_XY  07
37 –                                /* get opposite vertex direction bit */
38 – #define VDB_OPP(f)      (~(f) & 07)
39 –                                /* adjacent triangle vertex flags */
40 – static const int  adjacent_trifl[8] = {
41 –                        0,                      /* forbidden diagonal */
42 –                        1<<VDB_x|1<<VDB_y|1<<VDB_Xy,
43 –                        1<<VDB_y|1<<VDB_x|1<<VDB_xY,
44 –                        1<<VDB_y|1<<VDB_Xy|1<<VDB_X,
45 –                        1<<VDB_x|1<<VDB_xY|1<<VDB_Y,
46 –                        1<<VDB_Xy|1<<VDB_X|1<<VDB_Y,
47 –                        1<<VDB_xY|1<<VDB_Y|1<<VDB_X,
48 –                        0,                      /* forbidden diagonal */
49 –                };
50 –
28   typedef struct {
29          COLOR   v;              /* hemisphere sample value */
30          float   d;              /* reciprocal distance (1/rt) */
# Line 62 | Line 39 | typedef struct {
39          AMBSAMP sa[1];          /* sample array (extends struct) */
40   }  AMBHEMI;             /* ambient sample hemisphere */
41  
42 < #define ambndx(h,i,j)   ((i)*(h)->ns + (j))
43 < #define ambsam(h,i,j)   (h)->sa[ambndx(h,i,j)]
42 > #define AI(h,i,j)       ((i)*(h)->ns + (j))
43 > #define ambsam(h,i,j)   (h)->sa[AI(h,i,j)]
44  
45   typedef struct {
46          FVECT   r_i, r_i1, e_i, rcp, rI2_eJ2;
47          double  I1, I2;
71 –        int     valid;
48   } FFTRI;                /* vectors and coefficients for Hessian calculation */
49  
50  
75 – /* Get index for adjacent vertex */
76 – static int
77 – adjacent_verti(AMBHEMI *hp, int i, int j, int dbit)
78 – {
79 –        int     i0 = i*hp->ns + j;
80 –
81 –        switch (dbit) {
82 –        case VDB_y:     return(i0 - hp->ns);
83 –        case VDB_x:     return(i0 - 1);
84 –        case VDB_Xy:    return(i0 - hp->ns + 1);
85 –        case VDB_xY:    return(i0 + hp->ns - 1);
86 –        case VDB_X:     return(i0 + 1);
87 –        case VDB_Y:     return(i0 + hp->ns);
88 –                                /* the following should never occur */
89 –        case VDB_xy:    return(i0 - hp->ns - 1);
90 –        case VDB_XY:    return(i0 + hp->ns + 1);
91 –        }
92 –        return(-1);
93 – }
94 –
95 –
96 – /* Get vertex direction bit for the opposite edge to complete triangle */
97 – static int
98 – vdb_edge(int db1, int db2)
99 – {
100 –        switch (db1) {
101 –        case VDB_x:     return(db2==VDB_y ? VDB_Xy : VDB_Y);
102 –        case VDB_y:     return(db2==VDB_x ? VDB_xY : VDB_X);
103 –        case VDB_X:     return(db2==VDB_Xy ? VDB_y : VDB_xY);
104 –        case VDB_Y:     return(db2==VDB_xY ? VDB_x : VDB_Xy);
105 –        case VDB_xY:    return(db2==VDB_x ? VDB_y : VDB_X);
106 –        case VDB_Xy:    return(db2==VDB_y ? VDB_x : VDB_Y);
107 –        }
108 –        error(CONSISTENCY, "forbidden diagonal in vdb_edge()");
109 –        return(-1);
110 – }
111 –
112 –
51   static AMBHEMI *
52   inithemi(                       /* initialize sampling hemisphere */
53          COLOR   ac,
# Line 190 | Line 128 | getambsamp(RAY *arp, AMBHEMI *hp, int i, int j, int n)
128                                  spt[1]*hp->uy[ii] +
129                                  zd*hp->rp->ron[ii];
130          checknorm(arp->rdir);
131 <        dimlist[ndims++] = ambndx(hp,i,j) + 90171;
131 >        dimlist[ndims++] = AI(hp,i,j) + 90171;
132          rayvalue(arp);                  /* evaluate ray */
133          ndims--;                        /* apply coefficient */
134          multcolor(arp->rcol, arp->rcoef);
# Line 225 | Line 163 | ambsample(                             /* initial ambient division sample */
163   static float *
164   getambdiffs(AMBHEMI *hp)
165   {
166 <        float   *earr = (float *)malloc(sizeof(float)*hp->ns*hp->ns);
166 >        float   *earr = (float *)calloc(hp->ns*hp->ns, sizeof(float));
167          float   *ep;
168          AMBSAMP *ap;
169          double  b, d2;
# Line 236 | Line 174 | getambdiffs(AMBHEMI *hp)
174                                          /* compute squared neighbor diffs */
175          for (ap = hp->sa, ep = earr, i = 0; i < hp->ns; i++)
176              for (j = 0; j < hp->ns; j++, ap++, ep++) {
239 –                ep[0] = FTINY;
177                  b = bright(ap[0].v);
178                  if (i) {                /* from above */
179                          d2 = b - bright(ap[-hp->ns].v);
# Line 244 | Line 181 | getambdiffs(AMBHEMI *hp)
181                          ep[0] += d2;
182                          ep[-hp->ns] += d2;
183                  }
184 <                if (j) {                /* from behind */
185 <                        d2 = b - bright(ap[-1].v);
186 <                        d2 *= d2;
187 <                        ep[0] += d2;
188 <                        ep[-1] += d2;
189 <                }
184 >                if (!j) continue;
185 >                                        /* from behind */
186 >                d2 = b - bright(ap[-1].v);
187 >                d2 *= d2;
188 >                ep[0] += d2;
189 >                ep[-1] += d2;
190 >                if (!i) continue;
191 >                                        /* diagonal */
192 >                d2 = b - bright(ap[-hp->ns-1].v);
193 >                d2 *= d2;
194 >                ep[0] += d2;
195 >                ep[-hp->ns-1] += d2;
196              }
197                                          /* correct for number of neighbors */
198 <        earr[0] *= 2.f;
199 <        earr[hp->ns-1] *= 2.f;
200 <        earr[(hp->ns-1)*hp->ns] *= 2.f;
201 <        earr[(hp->ns-1)*hp->ns + hp->ns-1] *= 2.f;
198 >        earr[0] *= 8./3.;
199 >        earr[hp->ns-1] *= 8./3.;
200 >        earr[(hp->ns-1)*hp->ns] *= 8./3.;
201 >        earr[(hp->ns-1)*hp->ns + hp->ns-1] *= 8./3.;
202          for (i = 1; i < hp->ns-1; i++) {
203 <                earr[i*hp->ns] *= 4./3.;
204 <                earr[i*hp->ns + hp->ns-1] *= 4./3.;
203 >                earr[i*hp->ns] *= 8./5.;
204 >                earr[i*hp->ns + hp->ns-1] *= 8./5.;
205          }
206          for (j = 1; j < hp->ns-1; j++) {
207 <                earr[j] *= 4./3.;
208 <                earr[(hp->ns-1)*hp->ns + j] *= 4./3.;
207 >                earr[j] *= 8./5.;
208 >                earr[(hp->ns-1)*hp->ns + j] *= 8./5.;
209          }
210          return(earr);
211   }
# Line 278 | Line 221 | ambsupersamp(double acol[3], AMBHEMI *hp, int cnt)
221          RAY     ar;
222          double  asum[3];
223          float   *ep;
224 <        int     i, j, n;
224 >        int     i, j, n, nss;
225  
226          if (earr == NULL)               /* just skip calc. if no memory */
227                  return;
228                                          /* accumulate estimated variances */
229 <        for (ep = earr + hp->ns*hp->ns; ep-- > earr; )
230 <                e2rem += *ep;
229 >        for (ep = earr + hp->ns*hp->ns; ep > earr; )
230 >                e2rem += *--ep;
231          ep = earr;                      /* perform super-sampling */
232          for (ap = hp->sa, i = 0; i < hp->ns; i++)
233              for (j = 0; j < hp->ns; j++, ap++) {
234 <                int     nss = *ep/e2rem*cnt + frandom();
234 >                if (e2rem <= FTINY)
235 >                        goto done;      /* nothing left to do */
236 >                nss = *ep/e2rem*cnt + frandom();
237                  asum[0] = asum[1] = asum[2] = 0.0;
238                  for (n = 1; n <= nss; n++) {
239                          if (!getambsamp(&ar, hp, i, j, n)) {
# Line 306 | Line 251 | ambsupersamp(double acol[3], AMBHEMI *hp, int cnt)
251                  e2rem -= *ep++;         /* update remainders */
252                  cnt -= nss;
253          }
254 + done:
255          free(earr);
256   }
257  
258  
313 – /* Compute vertex flags, indicating farthest in each direction */
314 – static uby8 *
315 – vertex_flags(AMBHEMI *hp)
316 – {
317 –        uby8    *vflags = (uby8 *)calloc(hp->ns*hp->ns, sizeof(uby8));
318 –        uby8    *vf;
319 –        AMBSAMP *ap;
320 –        int     i, j;
321 –
322 –        if (vflags == NULL)
323 –                error(SYSTEM, "out of memory in vertex_flags()");
324 –        vf = vflags;
325 –        ap = hp->sa;            /* compute farthest along first row */
326 –        for (j = 0; j < hp->ns-1; j++, vf++, ap++)
327 –                if (ap[0].d <= ap[1].d)
328 –                        vf[0] |= 1<<VDB_X;
329 –                else
330 –                        vf[1] |= 1<<VDB_x;
331 –        ++vf; ++ap;
332 –                                /* flag subsequent rows */
333 –        for (i = 1; i < hp->ns; i++) {
334 –            for (j = 0; j < hp->ns-1; j++, vf++, ap++) {
335 –                if (ap[0].d <= ap[-hp->ns].d)   /* row before */
336 –                        vf[0] |= 1<<VDB_y;
337 –                else
338 –                        vf[-hp->ns] |= 1<<VDB_Y;
339 –                if (ap[0].d <= ap[1-hp->ns].d)  /* diagonal we care about */
340 –                        vf[0] |= 1<<VDB_Xy;
341 –                else
342 –                        vf[1-hp->ns] |= 1<<VDB_xY;
343 –                if (ap[0].d <= ap[1].d)         /* column after */
344 –                        vf[0] |= 1<<VDB_X;
345 –                else
346 –                        vf[1] |= 1<<VDB_x;
347 –            }
348 –            if (ap[0].d <= ap[-hp->ns].d)       /* final column edge */
349 –                vf[0] |= 1<<VDB_y;
350 –            else
351 –                vf[-hp->ns] |= 1<<VDB_Y;
352 –            ++vf; ++ap;
353 –        }
354 –        return(vflags);
355 – }
356 –
357 –
259   /* Return brightness of farthest ambient sample */
260   static double
261 < back_ambval(AMBHEMI *hp, int i, int j, int dbit1, int dbit2, const uby8 *vflags)
261 > back_ambval(AMBHEMI *hp, const int n1, const int n2, const int n3)
262   {
263 <        const int       v0 = ambndx(hp,i,j);
264 <        const int       tflags = (1<<dbit1 | 1<<dbit2);
265 <        int             v1, v2;
266 <
267 <        if ((vflags[v0] & tflags) == tflags)    /* is v0 the farthest? */
268 <                return(colval(hp->sa[v0].v,CIEY));
269 <        v1 = adjacent_verti(hp, i, j, dbit1);
270 <        if (vflags[v0] & 1<<dbit2)              /* v1 farthest if v0>v2 */
370 <                return(colval(hp->sa[v1].v,CIEY));
371 <        v2 = adjacent_verti(hp, i, j, dbit2);
372 <        if (vflags[v0] & 1<<dbit1)              /* v2 farthest if v0>v1 */
373 <                return(colval(hp->sa[v2].v,CIEY));
374 <                                                /* else check if v1>v2 */
375 <        if (vflags[v1] & 1<<vdb_edge(dbit1,dbit2))
376 <                return(colval(hp->sa[v1].v,CIEY));
377 <        return(colval(hp->sa[v2].v,CIEY));
263 >        if (hp->sa[n1].d <= hp->sa[n2].d) {
264 >                if (hp->sa[n1].d <= hp->sa[n3].d)
265 >                        return(colval(hp->sa[n1].v,CIEY));
266 >                return(colval(hp->sa[n3].v,CIEY));
267 >        }
268 >        if (hp->sa[n2].d <= hp->sa[n3].d)
269 >                return(colval(hp->sa[n2].v,CIEY));
270 >        return(colval(hp->sa[n3].v,CIEY));
271   }
272  
273  
274   /* Compute vectors and coefficients for Hessian/gradient calcs */
275   static void
276 < comp_fftri(FFTRI *ftp, AMBHEMI *hp, int i, int j, int dbit, const uby8 *vflags)
276 > comp_fftri(FFTRI *ftp, AMBHEMI *hp, const int n0, const int n1)
277   {
278 <        const int       i0 = ambndx(hp,i,j);
279 <        double          rdot_cp, dot_e, dot_er, rdot_r, rdot_r1, J2;
387 <        int             i1, ii;
278 >        double  rdot_cp, dot_e, dot_er, rdot_r, rdot_r1, J2;
279 >        int     ii;
280  
281 <        ftp->valid = 0;                 /* check if we can skip this edge */
282 <        ii = adjacent_trifl[dbit];
283 <        if ((vflags[i0] & ii) == ii)    /* cancels if vertex used as value */
392 <                return;
393 <        i1 = adjacent_verti(hp, i, j, dbit);
394 <        ii = adjacent_trifl[VDB_OPP(dbit)];
395 <        if ((vflags[i1] & ii) == ii)    /* on either end (for both triangles) */
396 <                return;
397 <                                        /* else go ahead with calculation */
398 <        VSUB(ftp->r_i, hp->sa[i0].p, hp->rp->rop);
399 <        VSUB(ftp->r_i1, hp->sa[i1].p, hp->rp->rop);
400 <        VSUB(ftp->e_i, hp->sa[i1].p, hp->sa[i0].p);
281 >        VSUB(ftp->r_i, hp->sa[n0].p, hp->rp->rop);
282 >        VSUB(ftp->r_i1, hp->sa[n1].p, hp->rp->rop);
283 >        VSUB(ftp->e_i, hp->sa[n1].p, hp->sa[n0].p);
284          VCROSS(ftp->rcp, ftp->r_i, ftp->r_i1);
285          rdot_cp = 1.0/DOT(ftp->rcp,ftp->rcp);
286          dot_e = DOT(ftp->e_i,ftp->e_i);
# Line 411 | Line 294 | comp_fftri(FFTRI *ftp, AMBHEMI *hp, int i, int j, int
294          J2 =  ( 0.5*(rdot_r - rdot_r1) - dot_er*ftp->I2 ) / dot_e;
295          for (ii = 3; ii--; )
296                  ftp->rI2_eJ2[ii] = ftp->I2*ftp->r_i[ii] + J2*ftp->e_i[ii];
414 –        ftp->valid++;
297   }
298  
299  
# Line 437 | Line 319 | comp_hessian(FVECT hess[3], FFTRI *ftp, FVECT nrm)
319          double  d1, d2, d3, d4;
320          double  I3, J3, K3;
321          int     i, j;
440 –
441 –        if (!ftp->valid) {              /* preemptive test */
442 –                memset(hess, 0, sizeof(FVECT)*3);
443 –                return;
444 –        }
322                                          /* compute intermediate coefficients */
323          d1 = 1.0/DOT(ftp->r_i,ftp->r_i);
324          d2 = 1.0/DOT(ftp->r_i1,ftp->r_i1);
# Line 505 | Line 382 | comp_gradient(FVECT grad, FFTRI *ftp, FVECT nrm)
382          double  f1;
383          int     i;
384  
508 –        if (!ftp->valid) {              /* preemptive test */
509 –                memset(grad, 0, sizeof(FVECT));
510 –                return;
511 –        }
385          f1 = 2.0*DOT(nrm, ftp->rcp);
386          VCROSS(ncp, nrm, ftp->e_i);
387          for (i = 3; i--; )
# Line 598 | Line 471 | ambHessian(                            /* anisotropic radii & pos. gradient */
471          static char     memerrmsg[] = "out of memory in ambHessian()";
472          FVECT           (*hessrow)[3] = NULL;
473          FVECT           *gradrow = NULL;
601 –        uby8            *vflags;
474          FVECT           hessian[3];
475          FVECT           gradient;
476          FFTRI           fftr;
# Line 620 | Line 492 | ambHessian(                            /* anisotropic radii & pos. gradient */
492                          error(SYSTEM, memerrmsg);
493                  memset(gradient, 0, sizeof(gradient));
494          }
623 –                                        /* get vertex position flags */
624 –        vflags = vertex_flags(hp);
495                                          /* compute first row of edges */
496          for (j = 0; j < hp->ns-1; j++) {
497 <                comp_fftri(&fftr, hp, 0, j, VDB_X, vflags);
497 >                comp_fftri(&fftr, hp, AI(hp,0,j), AI(hp,0,j+1));
498                  if (hessrow != NULL)
499                          comp_hessian(hessrow[j], &fftr, hp->rp->ron);
500                  if (gradrow != NULL)
# Line 634 | Line 504 | ambHessian(                            /* anisotropic radii & pos. gradient */
504          for (i = 0; i < hp->ns-1; i++) {
505              FVECT       hesscol[3];     /* compute first vertical edge */
506              FVECT       gradcol;
507 <            comp_fftri(&fftr, hp, i, 0, VDB_Y, vflags);
507 >            comp_fftri(&fftr, hp, AI(hp,i,0), AI(hp,i+1,0));
508              if (hessrow != NULL)
509                  comp_hessian(hesscol, &fftr, hp->rp->ron);
510              if (gradrow != NULL)
# Line 643 | Line 513 | ambHessian(                            /* anisotropic radii & pos. gradient */
513                  FVECT   hessdia[3];     /* compute triangle contributions */
514                  FVECT   graddia;
515                  double  backg;
516 <                backg = back_ambval(hp, i, j, VDB_X, VDB_Y, vflags);
516 >                backg = back_ambval(hp, AI(hp,i,j),
517 >                                        AI(hp,i,j+1), AI(hp,i+1,j));
518                                          /* diagonal (inner) edge */
519 <                comp_fftri(&fftr, hp, i, j+1, VDB_xY, vflags);
519 >                comp_fftri(&fftr, hp, AI(hp,i,j+1), AI(hp,i+1,j));
520                  if (hessrow != NULL) {
521                      comp_hessian(hessdia, &fftr, hp->rp->ron);
522                      rev_hessian(hesscol);
# Line 657 | Line 528 | ambHessian(                            /* anisotropic radii & pos. gradient */
528                      add2gradient(gradient, gradrow[j], graddia, gradcol, backg);
529                  }
530                                          /* initialize edge in next row */
531 <                comp_fftri(&fftr, hp, i+1, j+1, VDB_x, vflags);
531 >                comp_fftri(&fftr, hp, AI(hp,i+1,j+1), AI(hp,i+1,j));
532                  if (hessrow != NULL)
533                      comp_hessian(hessrow[j], &fftr, hp->rp->ron);
534                  if (gradrow != NULL)
535                      comp_gradient(gradrow[j], &fftr, hp->rp->ron);
536                                          /* new column edge & paired triangle */
537 <                backg = back_ambval(hp, i+1, j+1, VDB_x, VDB_y, vflags);
538 <                comp_fftri(&fftr, hp, i, j+1, VDB_Y, vflags);
537 >                backg = back_ambval(hp, AI(hp,i+1,j+1),
538 >                                        AI(hp,i+1,j), AI(hp,i,j+1));
539 >                comp_fftri(&fftr, hp, AI(hp,i,j+1), AI(hp,i+1,j+1));
540                  if (hessrow != NULL) {
541                      comp_hessian(hesscol, &fftr, hp->rp->ron);
542                      rev_hessian(hessdia);
# Line 684 | Line 556 | ambHessian(                            /* anisotropic radii & pos. gradient */
556                                          /* release row buffers */
557          if (hessrow != NULL) free(hessrow);
558          if (gradrow != NULL) free(gradrow);
687 –        free(vflags);
559          
560          if (ra != NULL)                 /* extract eigenvectors & radii */
561                  eigenvectors(uv, ra, hessian);

Diff Legend

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