ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/common/bsdf_t.c
Revision: 3.54
Committed: Tue Jan 25 01:34:20 2022 UTC (2 years, 3 months ago) by greg
Content type: text/plain
Branch: MAIN
CVS Tags: rad5R4, HEAD
Changes since 3.53: +16 -2 lines
Log Message:
fix: added position-based BSDF value perturbation for reliable peak extraction

File Contents

# User Rev Content
1 greg 3.2 #ifndef lint
2 greg 3.54 static const char RCSid[] = "$Id: bsdf_t.c,v 3.53 2021/12/15 01:38:50 greg Exp $";
3 greg 3.2 #endif
4 greg 3.1 /*
5     * bsdf_t.c
6     *
7     * Definitions for variable-resolution BSDF trees
8     *
9     * Created by Greg Ward on 2/2/11.
10     *
11     */
12    
13 greg 3.17 #define _USE_MATH_DEFINES
14 greg 3.3 #include "rtio.h"
15 greg 3.1 #include <stdlib.h>
16 greg 3.3 #include <math.h>
17     #include <ctype.h>
18 greg 3.1 #include "ezxml.h"
19     #include "bsdf.h"
20     #include "bsdf_t.h"
21 greg 3.6 #include "hilbert.h"
22    
23     /* Callback function type for SDtraverseTre() */
24 greg 3.37 typedef int SDtreCallback(float val, const double *cmin, double csiz,
25     void *cptr);
26 greg 3.6 /* reference width maximum (1.0) */
27 greg 3.7 static const unsigned iwbits = sizeof(unsigned)*4;
28 greg 3.23 static const unsigned iwmax = 1<<(sizeof(unsigned)*4);
29 greg 3.7 /* maximum cumulative value */
30     static const unsigned cumlmax = ~0;
31 greg 3.15 /* constant z-vector */
32     static const FVECT zvec = {.0, .0, 1.};
33 greg 3.22 /* quantization value */
34     static double quantum = 1./256.;
35 greg 3.37 /* our RGB primaries */
36     static C_COLOR tt_RGB_prim[3];
37     static float tt_RGB_coef[3];
38    
39     static const double czero[SD_MAXDIM];
40    
41     enum {tt_Y, tt_u, tt_v}; /* tree components (tt_Y==0) */
42 greg 3.6
43     /* Struct used for our distribution-building callback */
44     typedef struct {
45 greg 3.24 short nic; /* number of input coordinates */
46     short rev; /* reversing query */
47 greg 3.7 unsigned alen; /* current array length */
48     unsigned nall; /* number of allocated entries */
49     unsigned wmin; /* minimum square size so far */
50     unsigned wmax; /* maximum square size */
51 greg 3.6 struct outdir_s {
52     unsigned hent; /* entering Hilbert index */
53     int wid; /* this square size */
54     float bsdf; /* BSDF for this square */
55     } *darr; /* output direction array */
56     } SDdistScaffold;
57 greg 3.1
58     /* Allocate a new scattering distribution node */
59     static SDNode *
60     SDnewNode(int nd, int lg)
61     {
62     SDNode *st;
63    
64     if (nd <= 0) {
65     strcpy(SDerrorDetail, "Zero dimension BSDF node request");
66     return NULL;
67     }
68     if (nd > SD_MAXDIM) {
69     sprintf(SDerrorDetail, "Illegal BSDF dimension (%d > %d)",
70     nd, SD_MAXDIM);
71     return NULL;
72     }
73     if (lg < 0) {
74     st = (SDNode *)malloc(sizeof(SDNode) +
75 greg 3.7 sizeof(st->u.t[0])*((1<<nd) - 1));
76 greg 3.12 if (st == NULL) {
77 greg 3.1 sprintf(SDerrorDetail,
78 greg 3.6 "Cannot allocate %d branch BSDF tree", 1<<nd);
79 greg 3.12 return NULL;
80     }
81     memset(st->u.t, 0, sizeof(st->u.t[0])<<nd);
82     } else {
83     st = (SDNode *)malloc(sizeof(SDNode) +
84     sizeof(st->u.v[0])*((1 << nd*lg) - 1));
85     if (st == NULL) {
86 greg 3.1 sprintf(SDerrorDetail,
87     "Cannot allocate %d BSDF leaves", 1 << nd*lg);
88 greg 3.12 return NULL;
89     }
90 greg 3.1 }
91     st->ndim = nd;
92     st->log2GR = lg;
93     return st;
94     }
95    
96     /* Free an SD tree */
97     static void
98 greg 3.6 SDfreeTre(SDNode *st)
99 greg 3.1 {
100 greg 3.12 int n;
101 greg 3.1
102     if (st == NULL)
103     return;
104 greg 3.12 for (n = (st->log2GR < 0) << st->ndim; n--; )
105     SDfreeTre(st->u.t[n]);
106 greg 3.14 free(st);
107 greg 3.1 }
108    
109 greg 3.6 /* Free a variable-resolution BSDF */
110     static void
111     SDFreeBTre(void *p)
112     {
113     SDTre *sdt = (SDTre *)p;
114    
115     if (sdt == NULL)
116     return;
117 greg 3.37 SDfreeTre(sdt->stc[tt_Y]);
118     SDfreeTre(sdt->stc[tt_u]);
119     SDfreeTre(sdt->stc[tt_v]);
120 greg 3.6 free(sdt);
121     }
122 greg 3.5
123 greg 3.7 /* Fill branch's worth of grid values from subtree */
124     static void
125     fill_grid_branch(float *dptr, const float *sptr, int nd, int shft)
126     {
127     unsigned n = 1 << (shft-1);
128    
129     if (!--nd) { /* end on the line */
130     memcpy(dptr, sptr, sizeof(*dptr)*n);
131     return;
132     }
133     while (n--) /* recurse on each slice */
134     fill_grid_branch(dptr + (n << shft*nd),
135     sptr + (n << (shft-1)*nd), nd, shft);
136     }
137    
138     /* Get pointer at appropriate offset for the given branch */
139     static float *
140     grid_branch_start(SDNode *st, int n)
141     {
142 greg 3.14 unsigned skipsiz = 1 << (st->log2GR - 1);
143 greg 3.7 float *vptr = st->u.v;
144     int i;
145    
146 greg 3.14 for (i = st->ndim; i--; skipsiz <<= st->log2GR)
147     if (1<<i & n)
148     vptr += skipsiz;
149 greg 3.7 return vptr;
150     }
151    
152     /* Simplify (consolidate) a tree by flattening uniform depth regions */
153     static SDNode *
154     SDsimplifyTre(SDNode *st)
155     {
156     int match, n;
157    
158     if (st == NULL) /* check for invalid tree */
159     return NULL;
160     if (st->log2GR >= 0) /* grid just returns unaltered */
161     return st;
162     match = 1; /* check if grids below match */
163     for (n = 0; n < 1<<st->ndim; n++) {
164     if ((st->u.t[n] = SDsimplifyTre(st->u.t[n])) == NULL)
165     return NULL; /* propogate error up call stack */
166     match &= (st->u.t[n]->log2GR == st->u.t[0]->log2GR);
167     }
168 greg 3.9 if (match && (match = st->u.t[0]->log2GR) >= 0) {
169     SDNode *stn = SDnewNode(st->ndim, match + 1);
170 greg 3.7 if (stn == NULL) /* out of memory? */
171     return st;
172     /* transfer values to new grid */
173     for (n = 1 << st->ndim; n--; )
174     fill_grid_branch(grid_branch_start(stn, n),
175 greg 3.9 st->u.t[n]->u.v, stn->ndim, stn->log2GR);
176 greg 3.7 SDfreeTre(st); /* free old tree */
177     st = stn; /* return new one */
178     }
179     return st;
180     }
181    
182 greg 3.37 /* Assign the given voxel in tree (produces no grid nodes) */
183     static SDNode *
184     SDsetVoxel(SDNode *sroot, int nd, const double *tmin, const double tsiz, float val)
185     {
186     double ctrk[SD_MAXDIM];
187     double csiz = 1.;
188     SDNode *st;
189     int i, n;
190     /* check arguments */
191     for (i = nd; i-- > 0; )
192     if ((tmin[i] < .0) | (tmin[i] >= 1.-FTINY))
193     break;
194     if ((i >= 0) | (nd <= 0) | (tsiz <= FTINY) | (tsiz > 1.+FTINY) |
195     (sroot != NULL && sroot->ndim != nd)) {
196     SDfreeTre(sroot);
197     return NULL;
198     }
199     if (tsiz >= 1.-FTINY) { /* special case when tree is a leaf */
200     SDfreeTre(sroot);
201     if ((sroot = SDnewNode(nd, 0)) != NULL)
202     sroot->u.v[0] = val;
203     return sroot;
204     }
205     /* make sure we have branching root */
206     if (sroot != NULL && sroot->log2GR >= 0) {
207     SDfreeTre(sroot); sroot = NULL;
208     }
209     if (sroot == NULL && (sroot = SDnewNode(nd, -1)) == NULL)
210     return NULL;
211     st = sroot; /* climb/grow tree */
212     memset(ctrk, 0, sizeof(ctrk));
213     for ( ; ; ) {
214     csiz *= .5; /* find appropriate branch */
215     n = 0;
216     for (i = nd; i--; )
217     if (ctrk[i]+csiz <= tmin[i]+FTINY) {
218     ctrk[i] += csiz;
219     n |= 1 << i;
220     }
221     /* reached desired voxel? */
222     if (csiz <= tsiz+FTINY) {
223     SDfreeTre(st->u.t[n]);
224     st = st->u.t[n] = SDnewNode(nd, 0);
225     break;
226     }
227     /* else grow tree as needed */
228     if (st->u.t[n] != NULL && st->u.t[n]->log2GR >= 0) {
229     SDfreeTre(st->u.t[n]); st->u.t[n] = NULL;
230     }
231     if (st->u.t[n] == NULL)
232     st->u.t[n] = SDnewNode(nd, -1);
233     if ((st = st->u.t[n]) == NULL)
234     break;
235     }
236     if (st == NULL) {
237     SDfreeTre(sroot);
238     return NULL;
239     }
240     st->u.v[0] = val; /* assign leaf and return root */
241     return sroot;
242     }
243    
244 greg 3.7 /* Find smallest leaf in tree */
245     static double
246     SDsmallestLeaf(const SDNode *st)
247     {
248     if (st->log2GR < 0) { /* tree branches */
249     double lmin = 1.;
250     int n;
251     for (n = 1<<st->ndim; n--; ) {
252     double lsiz = SDsmallestLeaf(st->u.t[n]);
253     if (lsiz < lmin)
254     lmin = lsiz;
255     }
256     return .5*lmin;
257     }
258     /* leaf grid width */
259     return 1. / (double)(1 << st->log2GR);
260     }
261    
262 greg 3.1 /* Add up N-dimensional hypercube array values over the given box */
263     static double
264 greg 3.7 SDiterSum(const float *va, int nd, int shft, const int *imin, const int *imax)
265 greg 3.1 {
266 greg 3.10 const unsigned skipsiz = 1 << --nd*shft;
267 greg 3.1 double sum = .0;
268     int i;
269 greg 3.15
270     va += *imin * skipsiz;
271    
272 greg 3.1 if (skipsiz == 1)
273     for (i = *imin; i < *imax; i++)
274 greg 3.15 sum += *va++;
275 greg 3.1 else
276 greg 3.15 for (i = *imin; i < *imax; i++, va += skipsiz)
277     sum += SDiterSum(va, nd, shft, imin+1, imax+1);
278 greg 3.1 return sum;
279     }
280    
281     /* Average BSDF leaves over an orthotope defined by the unit hypercube */
282     static double
283 greg 3.6 SDavgTreBox(const SDNode *st, const double *bmin, const double *bmax)
284 greg 3.1 {
285     unsigned n;
286     int i;
287    
288     if (!st)
289     return .0;
290     /* check box limits */
291     for (i = st->ndim; i--; ) {
292     if (bmin[i] >= 1.)
293     return .0;
294 greg 3.13 if (bmax[i] <= 0)
295 greg 3.1 return .0;
296     if (bmin[i] >= bmax[i])
297     return .0;
298     }
299     if (st->log2GR < 0) { /* iterate on subtree */
300     double sum = .0, wsum = 1e-20;
301     double sbmin[SD_MAXDIM], sbmax[SD_MAXDIM], w;
302     for (n = 1 << st->ndim; n--; ) {
303     w = 1.;
304     for (i = st->ndim; i--; ) {
305     sbmin[i] = 2.*bmin[i];
306     sbmax[i] = 2.*bmax[i];
307     if (n & 1<<i) {
308     sbmin[i] -= 1.;
309     sbmax[i] -= 1.;
310     }
311     if (sbmin[i] < .0) sbmin[i] = .0;
312     if (sbmax[i] > 1.) sbmax[i] = 1.;
313 greg 3.13 if (sbmin[i] >= sbmax[i]) {
314     w = .0;
315     break;
316     }
317 greg 3.1 w *= sbmax[i] - sbmin[i];
318     }
319     if (w > 1e-10) {
320 greg 3.6 sum += w * SDavgTreBox(st->u.t[n], sbmin, sbmax);
321 greg 3.1 wsum += w;
322     }
323     }
324     return sum / wsum;
325 greg 3.15 } else { /* iterate over leaves */
326     int imin[SD_MAXDIM], imax[SD_MAXDIM];
327    
328     n = 1;
329     for (i = st->ndim; i--; ) {
330     imin[i] = (bmin[i] <= 0) ? 0 :
331     (int)((1 << st->log2GR)*bmin[i]);
332     imax[i] = (bmax[i] >= 1.) ? (1 << st->log2GR) :
333     (int)((1 << st->log2GR)*bmax[i] + .999999);
334     n *= imax[i] - imin[i];
335     }
336     if (n)
337     return SDiterSum(st->u.v, st->ndim,
338     st->log2GR, imin, imax) / (double)n;
339 greg 3.1 }
340 greg 3.15 return .0;
341 greg 3.1 }
342    
343 greg 3.6 /* Recursive call for SDtraverseTre() */
344     static int
345     SDdotravTre(const SDNode *st, const double *pos, int cmask,
346     SDtreCallback *cf, void *cptr,
347     const double *cmin, double csiz)
348     {
349     int rv, rval = 0;
350     double bmin[SD_MAXDIM];
351     int i, n;
352 greg 3.37 /* paranoia */
353     if (st == NULL)
354     return 0;
355 greg 3.6 /* in branches? */
356     if (st->log2GR < 0) {
357     unsigned skipmask = 0;
358     csiz *= .5;
359     for (i = st->ndim; i--; )
360 greg 3.32 if (1<<i & cmask) {
361 greg 3.6 if (pos[i] < cmin[i] + csiz)
362 greg 3.13 for (n = 1 << st->ndim; n--; ) {
363 greg 3.6 if (n & 1<<i)
364     skipmask |= 1<<n;
365 greg 3.13 }
366 greg 3.6 else
367 greg 3.13 for (n = 1 << st->ndim; n--; ) {
368 greg 3.6 if (!(n & 1<<i))
369     skipmask |= 1<<n;
370 greg 3.13 }
371 greg 3.32 }
372 greg 3.6 for (n = 1 << st->ndim; n--; ) {
373     if (1<<n & skipmask)
374     continue;
375     for (i = st->ndim; i--; )
376     if (1<<i & n)
377     bmin[i] = cmin[i] + csiz;
378     else
379     bmin[i] = cmin[i];
380    
381     rval += rv = SDdotravTre(st->u.t[n], pos, cmask,
382     cf, cptr, bmin, csiz);
383     if (rv < 0)
384     return rv;
385     }
386     } else { /* else traverse leaves */
387     int clim[SD_MAXDIM][2];
388     int cpos[SD_MAXDIM];
389    
390     if (st->log2GR == 0) /* short cut */
391     return (*cf)(st->u.v[0], cmin, csiz, cptr);
392    
393     csiz /= (double)(1 << st->log2GR);
394     /* assign coord. ranges */
395     for (i = st->ndim; i--; )
396     if (1<<i & cmask) {
397     clim[i][0] = (pos[i] - cmin[i])/csiz;
398     /* check overflow from f.p. error */
399     clim[i][0] -= clim[i][0] >> st->log2GR;
400     clim[i][1] = clim[i][0] + 1;
401     } else {
402     clim[i][0] = 0;
403     clim[i][1] = 1 << st->log2GR;
404     }
405     #if (SD_MAXDIM == 4)
406     bmin[0] = cmin[0] + csiz*clim[0][0];
407     for (cpos[0] = clim[0][0]; cpos[0] < clim[0][1]; cpos[0]++) {
408     bmin[1] = cmin[1] + csiz*clim[1][0];
409     for (cpos[1] = clim[1][0]; cpos[1] < clim[1][1]; cpos[1]++) {
410     bmin[2] = cmin[2] + csiz*clim[2][0];
411 greg 3.16 if (st->ndim == 3) {
412     cpos[2] = clim[2][0];
413 greg 3.6 n = cpos[0];
414 greg 3.16 for (i = 1; i < 3; i++)
415 greg 3.6 n = (n << st->log2GR) + cpos[i];
416 greg 3.16 for ( ; cpos[2] < clim[2][1]; cpos[2]++) {
417 greg 3.6 rval += rv = (*cf)(st->u.v[n++], bmin, csiz, cptr);
418     if (rv < 0)
419     return rv;
420 greg 3.16 bmin[2] += csiz;
421     }
422     } else {
423     for (cpos[2] = clim[2][0]; cpos[2] < clim[2][1]; cpos[2]++) {
424     bmin[3] = cmin[3] + csiz*(cpos[3] = clim[3][0]);
425     n = cpos[0];
426     for (i = 1; i < 4; i++)
427     n = (n << st->log2GR) + cpos[i];
428     for ( ; cpos[3] < clim[3][1]; cpos[3]++) {
429     rval += rv = (*cf)(st->u.v[n++], bmin, csiz, cptr);
430     if (rv < 0)
431     return rv;
432     bmin[3] += csiz;
433     }
434     bmin[2] += csiz;
435 greg 3.6 }
436     }
437     bmin[1] += csiz;
438     }
439     bmin[0] += csiz;
440     }
441     #else
442     _!_ "broken code segment!"
443     #endif
444     }
445     return rval;
446     }
447    
448     /* Traverse a tree, visiting nodes in a slice that fits partial position */
449     static int
450     SDtraverseTre(const SDNode *st, const double *pos, int cmask,
451     SDtreCallback *cf, void *cptr)
452     {
453     int i;
454     /* check arguments */
455     if ((st == NULL) | (cf == NULL))
456     return -1;
457     for (i = st->ndim; i--; )
458     if (1<<i & cmask && (pos[i] < 0) | (pos[i] >= 1.))
459     return -1;
460    
461     return SDdotravTre(st, pos, cmask, cf, cptr, czero, 1.);
462     }
463 greg 3.5
464     /* Look up tree value at the given grid position */
465     static float
466 greg 3.6 SDlookupTre(const SDNode *st, const double *pos, double *hcube)
467 greg 3.5 {
468     double spos[SD_MAXDIM];
469     int i, n, t;
470 greg 3.6 /* initialize voxel return */
471     if (hcube) {
472     hcube[i = st->ndim] = 1.;
473     while (i--)
474     hcube[i] = .0;
475     }
476 greg 3.5 /* climb the tree */
477 greg 3.37 while (st != NULL && st->log2GR < 0) {
478 greg 3.5 n = 0; /* move to appropriate branch */
479 greg 3.6 if (hcube) hcube[st->ndim] *= .5;
480 greg 3.5 for (i = st->ndim; i--; ) {
481     spos[i] = 2.*pos[i];
482     t = (spos[i] >= 1.);
483     n |= t<<i;
484     spos[i] -= (double)t;
485 greg 3.6 if (hcube) hcube[i] += (double)t * hcube[st->ndim];
486 greg 3.5 }
487     st = st->u.t[n]; /* avoids tail recursion */
488     pos = spos;
489     }
490 greg 3.37 if (st == NULL) /* should never happen? */
491     return .0;
492 greg 3.6 if (st->log2GR == 0) /* short cut */
493     return st->u.v[0];
494 greg 3.5 n = t = 0; /* find grid array index */
495     for (i = st->ndim; i--; ) {
496     n += (int)((1<<st->log2GR)*pos[i]) << t;
497     t += st->log2GR;
498     }
499 greg 3.6 if (hcube) { /* compute final hypercube */
500     hcube[st->ndim] /= (double)(1<<st->log2GR);
501     for (i = st->ndim; i--; )
502     hcube[i] += floor((1<<st->log2GR)*pos[i])*hcube[st->ndim];
503     }
504     return st->u.v[n]; /* no interpolation */
505     }
506    
507 greg 3.37 /* Convert CIE (Y,u',v') color to our RGB */
508     static void
509     SDyuv2rgb(double yval, double uprime, double vprime, float rgb[3])
510     {
511     const double dfact = 1./(6.*uprime - 16.*vprime + 12.);
512     C_COLOR cxy;
513    
514     c_cset(&cxy, 9.*uprime*dfact, 4.*vprime*dfact);
515     c_toSharpRGB(&cxy, yval, rgb);
516     }
517    
518 greg 3.54 static double
519     pfrac(double x)
520     {
521     return( x - (int)x );
522     }
523    
524 greg 3.6 /* Query BSDF value and sample hypercube for the given vectors */
525 greg 3.37 static int
526     SDqueryTre(const SDTre *sdt, float *coef,
527 greg 3.52 const FVECT inVec, const FVECT outVec, double *hc)
528 greg 3.6 {
529 greg 3.24 const RREAL *vtmp;
530 greg 3.54 double hcube[SD_MAXDIM+1];
531 greg 3.37 float yval;
532 greg 3.24 FVECT rOutVec;
533 greg 3.53 RREAL gridPos[4];
534 greg 3.54 double d;
535     int i;
536 greg 3.7
537 greg 3.37 if (sdt->stc[tt_Y] == NULL) /* paranoia, I hope */
538     return 0;
539    
540 greg 3.7 switch (sdt->sidef) { /* whose side are you on? */
541 greg 3.24 case SD_FREFL:
542 greg 3.7 if ((outVec[2] < 0) | (inVec[2] < 0))
543 greg 3.37 return 0;
544 greg 3.7 break;
545 greg 3.24 case SD_BREFL:
546 greg 3.7 if ((outVec[2] > 0) | (inVec[2] > 0))
547 greg 3.37 return 0;
548 greg 3.7 break;
549 greg 3.24 case SD_FXMIT:
550     if (outVec[2] > 0) {
551     if (inVec[2] > 0)
552 greg 3.37 return 0;
553 greg 3.24 vtmp = outVec; outVec = inVec; inVec = vtmp;
554     } else if (inVec[2] < 0)
555 greg 3.37 return 0;
556 greg 3.24 break;
557     case SD_BXMIT:
558     if (inVec[2] > 0) {
559     if (outVec[2] > 0)
560 greg 3.37 return 0;
561 greg 3.24 vtmp = outVec; outVec = inVec; inVec = vtmp;
562     } else if (outVec[2] < 0)
563 greg 3.37 return 0;
564 greg 3.7 break;
565     default:
566 greg 3.37 return 0;
567 greg 3.7 }
568 greg 3.6 /* convert vector coordinates */
569 greg 3.37 if (sdt->stc[tt_Y]->ndim == 3) {
570 greg 3.15 spinvector(rOutVec, outVec, zvec, -atan2(-inVec[1],-inVec[0]));
571 greg 3.31 gridPos[0] = (.5-FTINY) -
572     .5*sqrt(inVec[0]*inVec[0] + inVec[1]*inVec[1]);
573 greg 3.53 disk2square(gridPos+1, rOutVec[0], rOutVec[1]);
574 greg 3.37 } else if (sdt->stc[tt_Y]->ndim == 4) {
575 greg 3.53 disk2square(gridPos, -inVec[0], -inVec[1]);
576     disk2square(gridPos+2, outVec[0], outVec[1]);
577 greg 3.6 } else
578 greg 3.37 return 0; /* should be internal error */
579 greg 3.54
580     if (hc == NULL) hc = hcube; /* get BSDF value */
581 greg 3.37 yval = SDlookupTre(sdt->stc[tt_Y], gridPos, hc);
582 greg 3.41 if (coef == NULL) /* just getting hypercube? */
583     return 1;
584 greg 3.54 d = 0; /* position-specific perturbation */
585     for (i = sdt->stc[tt_Y]->ndim; i--; )
586     d += pfrac((2<<i)/(hc[i]+.01687)) - .5;
587     yval *= 1. + 1e-4*d; /* assumes tolerance is > 0.04% */
588 greg 3.42 if (sdt->stc[tt_u] == NULL || sdt->stc[tt_v] == NULL) {
589 greg 3.41 *coef = yval;
590 greg 3.37 return 1; /* no color */
591     }
592     /* else decode color */
593     SDyuv2rgb(yval, SDlookupTre(sdt->stc[tt_u], gridPos, NULL),
594     SDlookupTre(sdt->stc[tt_v], gridPos, NULL), coef);
595     coef[0] *= tt_RGB_coef[0];
596     coef[1] *= tt_RGB_coef[1];
597     coef[2] *= tt_RGB_coef[2];
598     return 3;
599 greg 3.5 }
600    
601     /* Compute non-diffuse component for variable-resolution BSDF */
602     static int
603     SDgetTreBSDF(float coef[SDmaxCh], const FVECT outVec,
604 greg 3.6 const FVECT inVec, SDComponent *sdc)
605 greg 3.5 {
606 greg 3.6 /* check arguments */
607     if ((coef == NULL) | (outVec == NULL) | (inVec == NULL) | (sdc == NULL)
608     || sdc->dist == NULL)
609     return 0;
610 greg 3.5 /* get nearest BSDF value */
611 greg 3.37 return SDqueryTre((SDTre *)sdc->dist, coef, outVec, inVec, NULL);
612 greg 3.6 }
613    
614     /* Callback to build cumulative distribution using SDtraverseTre() */
615     static int
616     build_scaffold(float val, const double *cmin, double csiz, void *cptr)
617     {
618     SDdistScaffold *sp = (SDdistScaffold *)cptr;
619     int wid = csiz*(double)iwmax + .5;
620 greg 3.25 double revcmin[2];
621 greg 3.6 bitmask_t bmin[2], bmax[2];
622    
623 greg 3.25 if (sp->rev) { /* need to reverse sense? */
624     revcmin[0] = 1. - cmin[0] - csiz;
625     revcmin[1] = 1. - cmin[1] - csiz;
626     cmin = revcmin;
627     } else {
628     cmin += sp->nic; /* else skip to output coords */
629     }
630 greg 3.6 if (wid < sp->wmin) /* new minimum width? */
631     sp->wmin = wid;
632     if (wid > sp->wmax) /* new maximum? */
633     sp->wmax = wid;
634     if (sp->alen >= sp->nall) { /* need more space? */
635     struct outdir_s *ndarr;
636 greg 3.35 sp->nall = (int)(1.5*sp->nall) + 256;
637 greg 3.6 ndarr = (struct outdir_s *)realloc(sp->darr,
638     sizeof(struct outdir_s)*sp->nall);
639 greg 3.12 if (ndarr == NULL) {
640     sprintf(SDerrorDetail,
641     "Cannot grow scaffold to %u entries", sp->nall);
642 greg 3.6 return -1; /* abort build */
643 greg 3.12 }
644 greg 3.6 sp->darr = ndarr;
645     }
646     /* find Hilbert entry index */
647     bmin[0] = cmin[0]*(double)iwmax + .5;
648     bmin[1] = cmin[1]*(double)iwmax + .5;
649 greg 3.10 bmax[0] = bmin[0] + wid-1;
650     bmax[1] = bmin[1] + wid-1;
651 greg 3.7 hilbert_box_vtx(2, sizeof(bitmask_t), iwbits, 1, bmin, bmax);
652     sp->darr[sp->alen].hent = hilbert_c2i(2, iwbits, bmin);
653 greg 3.6 sp->darr[sp->alen].wid = wid;
654     sp->darr[sp->alen].bsdf = val;
655     sp->alen++; /* on to the next entry */
656     return 0;
657     }
658    
659     /* Scaffold comparison function for qsort -- ascending Hilbert index */
660     static int
661     sscmp(const void *p1, const void *p2)
662     {
663 greg 3.10 unsigned h1 = (*(const struct outdir_s *)p1).hent;
664     unsigned h2 = (*(const struct outdir_s *)p2).hent;
665    
666     if (h1 > h2)
667     return 1;
668     if (h1 < h2)
669     return -1;
670     return 0;
671 greg 3.6 }
672    
673     /* Create a new cumulative distribution for the given input direction */
674     static SDTreCDst *
675 greg 3.24 make_cdist(const SDTre *sdt, const double *invec, int rev)
676 greg 3.6 {
677     SDdistScaffold myScaffold;
678 greg 3.24 double pos[4];
679     int cmask;
680 greg 3.6 SDTreCDst *cd;
681     struct outdir_s *sp;
682     double scale, cursum;
683     int i;
684     /* initialize scaffold */
685     myScaffold.wmin = iwmax;
686     myScaffold.wmax = 0;
687 greg 3.37 myScaffold.nic = sdt->stc[tt_Y]->ndim - 2;
688 greg 3.24 myScaffold.rev = rev;
689 greg 3.6 myScaffold.alen = 0;
690 greg 3.12 myScaffold.nall = 512;
691 greg 3.6 myScaffold.darr = (struct outdir_s *)malloc(sizeof(struct outdir_s) *
692     myScaffold.nall);
693     if (myScaffold.darr == NULL)
694     return NULL;
695 greg 3.24 /* set up traversal */
696     cmask = (1<<myScaffold.nic) - 1;
697     for (i = myScaffold.nic; i--; )
698     pos[i+2*rev] = invec[i];
699     cmask <<= 2*rev;
700 greg 3.6 /* grow the distribution */
701 greg 3.37 if (SDtraverseTre(sdt->stc[tt_Y], pos, cmask,
702     build_scaffold, &myScaffold) < 0) {
703 greg 3.6 free(myScaffold.darr);
704     return NULL;
705     }
706     /* allocate result holder */
707     cd = (SDTreCDst *)malloc(sizeof(SDTreCDst) +
708     sizeof(cd->carr[0])*myScaffold.alen);
709     if (cd == NULL) {
710 greg 3.12 sprintf(SDerrorDetail,
711     "Cannot allocate %u entry cumulative distribution",
712     myScaffold.alen);
713 greg 3.6 free(myScaffold.darr);
714     return NULL;
715     }
716 greg 3.15 cd->isodist = (myScaffold.nic == 1);
717 greg 3.6 /* sort the distribution */
718     qsort(myScaffold.darr, cd->calen = myScaffold.alen,
719 greg 3.37 sizeof(struct outdir_s), sscmp);
720 greg 3.6
721     /* record input range */
722 greg 3.7 scale = myScaffold.wmin / (double)iwmax;
723 greg 3.6 for (i = myScaffold.nic; i--; ) {
724 greg 3.26 cd->clim[i][0] = floor(pos[i+2*rev]/scale) * scale;
725 greg 3.6 cd->clim[i][1] = cd->clim[i][0] + scale;
726     }
727 greg 3.15 if (cd->isodist) { /* avoid issue in SDqueryTreProjSA() */
728     cd->clim[1][0] = cd->clim[0][0];
729     cd->clim[1][1] = cd->clim[0][1];
730     }
731 greg 3.6 cd->max_psa = myScaffold.wmax / (double)iwmax;
732     cd->max_psa *= cd->max_psa * M_PI;
733 greg 3.24 if (rev)
734     cd->sidef = (sdt->sidef==SD_BXMIT) ? SD_FXMIT : SD_BXMIT;
735     else
736     cd->sidef = sdt->sidef;
737 greg 3.6 cd->cTotal = 1e-20; /* compute directional total */
738     sp = myScaffold.darr;
739     for (i = myScaffold.alen; i--; sp++)
740     cd->cTotal += sp->bsdf * (double)sp->wid * sp->wid;
741     cursum = .0; /* go back and get cumulative values */
742     scale = (double)cumlmax / cd->cTotal;
743     sp = myScaffold.darr;
744     for (i = 0; i < cd->calen; i++, sp++) {
745 greg 3.7 cd->carr[i].hndx = sp->hent;
746 greg 3.6 cd->carr[i].cuml = scale*cursum + .5;
747     cursum += sp->bsdf * (double)sp->wid * sp->wid;
748     }
749     cd->carr[i].hndx = ~0; /* make final entry */
750     cd->carr[i].cuml = cumlmax;
751     cd->cTotal *= M_PI/(double)iwmax/iwmax;
752     /* all done, clean up and return */
753     free(myScaffold.darr);
754     return cd;
755     }
756    
757     /* Find or allocate a cumulative distribution for the given incoming vector */
758     const SDCDst *
759     SDgetTreCDist(const FVECT inVec, SDComponent *sdc)
760     {
761 greg 3.48 unsigned long cacheLeft = SDmaxCache;
762 greg 3.6 const SDTre *sdt;
763 greg 3.53 RREAL inCoord[2];
764 greg 3.6 int i;
765 greg 3.24 int mode;
766 greg 3.48 SDTreCDst *cd, *cdlast, *cdlimit;
767 greg 3.6 /* check arguments */
768     if ((inVec == NULL) | (sdc == NULL) ||
769     (sdt = (SDTre *)sdc->dist) == NULL)
770     return NULL;
771 greg 3.24 switch (mode = sdt->sidef) { /* check direction */
772     case SD_FREFL:
773     if (inVec[2] < 0)
774     return NULL;
775     break;
776     case SD_BREFL:
777     if (inVec[2] > 0)
778     return NULL;
779     break;
780     case SD_FXMIT:
781     if (inVec[2] < 0)
782     mode = SD_BXMIT;
783     break;
784     case SD_BXMIT:
785     if (inVec[2] > 0)
786     mode = SD_FXMIT;
787     break;
788     default:
789     return NULL;
790     }
791 greg 3.37 if (sdt->stc[tt_Y]->ndim == 3) { /* isotropic BSDF? */
792 greg 3.24 if (mode != sdt->sidef) /* XXX unhandled reciprocity */
793     return &SDemptyCD;
794 greg 3.31 inCoord[0] = (.5-FTINY) -
795     .5*sqrt(inVec[0]*inVec[0] + inVec[1]*inVec[1]);
796 greg 3.37 } else if (sdt->stc[tt_Y]->ndim == 4) {
797 greg 3.25 if (mode != sdt->sidef) /* use reciprocity? */
798 greg 3.53 disk2square(inCoord, inVec[0], inVec[1]);
799 greg 3.25 else
800 greg 3.53 disk2square(inCoord, -inVec[0], -inVec[1]);
801 greg 3.21 } else
802 greg 3.6 return NULL; /* should be internal error */
803 greg 3.21 /* quantize to avoid f.p. errors */
804 greg 3.37 for (i = sdt->stc[tt_Y]->ndim - 2; i--; )
805 greg 3.21 inCoord[i] = floor(inCoord[i]/quantum)*quantum + .5*quantum;
806 greg 3.48 cdlast = cdlimit = NULL; /* check for direction in cache list */
807 greg 3.44 /* PLACE MUTEX LOCK HERE FOR THREAD-SAFE */
808 greg 3.6 for (cd = (SDTreCDst *)sdc->cdList; cd != NULL;
809 greg 3.20 cdlast = cd, cd = cd->next) {
810 greg 3.48 if (cacheLeft) { /* check cache size limit */
811     long csiz = sizeof(SDTreCDst) +
812     sizeof(cd->carr[0])*cd->calen;
813     if (cacheLeft > csiz)
814     cacheLeft -= csiz;
815     else {
816     cdlimit = cdlast;
817     cacheLeft = 0;
818     }
819     }
820 greg 3.24 if (cd->sidef != mode)
821     continue;
822 greg 3.37 for (i = sdt->stc[tt_Y]->ndim - 2; i--; )
823 greg 3.6 if ((cd->clim[i][0] > inCoord[i]) |
824     (inCoord[i] >= cd->clim[i][1]))
825     break;
826     if (i < 0)
827     break; /* means we have a match */
828     }
829 greg 3.48 if (cd == NULL) { /* need to create new entry? */
830     if (cdlimit != NULL) /* exceeded cache size limit? */
831     while ((cd = cdlimit->next) != NULL) {
832     cdlimit->next = cd->next;
833     free(cd);
834     }
835 greg 3.24 cdlast = cd = make_cdist(sdt, inCoord, mode != sdt->sidef);
836 greg 3.48 }
837 greg 3.6 if (cdlast != NULL) { /* move entry to head of cache list */
838     cdlast->next = cd->next;
839 greg 3.20 cd->next = (SDTreCDst *)sdc->cdList;
840 greg 3.6 sdc->cdList = (SDCDst *)cd;
841     }
842 greg 3.44 /* END MUTEX LOCK */
843 greg 3.6 return (SDCDst *)cd; /* ready to go */
844     }
845    
846     /* Query solid angle for vector(s) */
847     static SDError
848     SDqueryTreProjSA(double *psa, const FVECT v1, const RREAL *v2,
849     int qflags, SDComponent *sdc)
850     {
851     double myPSA[2];
852     /* check arguments */
853     if ((psa == NULL) | (v1 == NULL) | (sdc == NULL) ||
854     sdc->dist == NULL)
855     return SDEargument;
856     /* get projected solid angle(s) */
857     if (v2 != NULL) {
858     const SDTre *sdt = (SDTre *)sdc->dist;
859 greg 3.36 double hcube[SD_MAXDIM+1];
860 greg 3.37 if (!SDqueryTre(sdt, NULL, v1, v2, hcube)) {
861 greg 3.7 strcpy(SDerrorDetail, "Bad call to SDqueryTreProjSA");
862     return SDEinternal;
863 greg 3.6 }
864 greg 3.37 myPSA[0] = hcube[sdt->stc[tt_Y]->ndim];
865 greg 3.6 myPSA[1] = myPSA[0] *= myPSA[0] * M_PI;
866     } else {
867     const SDTreCDst *cd = (const SDTreCDst *)SDgetTreCDist(v1, sdc);
868     if (cd == NULL)
869 greg 3.29 myPSA[0] = myPSA[1] = 0;
870     else {
871     myPSA[0] = M_PI * (cd->clim[0][1] - cd->clim[0][0]) *
872     (cd->clim[1][1] - cd->clim[1][0]);
873     myPSA[1] = cd->max_psa;
874     }
875 greg 3.6 }
876     switch (qflags) { /* record based on flag settings */
877     case SDqueryVal:
878     *psa = myPSA[0];
879     break;
880     case SDqueryMax:
881     if (myPSA[1] > *psa)
882     *psa = myPSA[1];
883     break;
884     case SDqueryMin+SDqueryMax:
885     if (myPSA[1] > psa[1])
886     psa[1] = myPSA[1];
887     /* fall through */
888     case SDqueryMin:
889 greg 3.30 if ((myPSA[0] > 0) & (myPSA[0] < psa[0]))
890 greg 3.6 psa[0] = myPSA[0];
891     break;
892     }
893     return SDEnone;
894     }
895    
896     /* Sample cumulative distribution */
897     static SDError
898     SDsampTreCDist(FVECT ioVec, double randX, const SDCDst *cdp)
899     {
900     const unsigned nBitsC = 4*sizeof(bitmask_t);
901     const unsigned nExtraBits = 8*(sizeof(bitmask_t)-sizeof(unsigned));
902     const SDTreCDst *cd = (const SDTreCDst *)cdp;
903 greg 3.7 const unsigned target = randX*cumlmax;
904 greg 3.6 bitmask_t hndx, hcoord[2];
905 greg 3.53 FVECT gpos;
906     double rotangle;
907 greg 3.6 int i, iupper, ilower;
908     /* check arguments */
909     if ((ioVec == NULL) | (cd == NULL))
910     return SDEargument;
911 greg 3.24 if (!cd->sidef)
912     return SDEnone; /* XXX should never happen */
913 greg 3.7 if (ioVec[2] > 0) {
914 greg 3.24 if ((cd->sidef != SD_FREFL) & (cd->sidef != SD_FXMIT))
915 greg 3.7 return SDEargument;
916 greg 3.24 } else if ((cd->sidef != SD_BREFL) & (cd->sidef != SD_BXMIT))
917 greg 3.7 return SDEargument;
918 greg 3.6 /* binary search to find position */
919     ilower = 0; iupper = cd->calen;
920     while ((i = (iupper + ilower) >> 1) != ilower)
921 greg 3.19 if (target >= cd->carr[i].cuml)
922 greg 3.6 ilower = i;
923     else
924     iupper = i;
925     /* localize random position */
926 greg 3.7 randX = (randX*cumlmax - cd->carr[ilower].cuml) /
927 greg 3.6 (double)(cd->carr[iupper].cuml - cd->carr[ilower].cuml);
928     /* index in longer Hilbert curve */
929     hndx = (randX*cd->carr[iupper].hndx + (1.-randX)*cd->carr[ilower].hndx)
930     * (double)((bitmask_t)1 << nExtraBits);
931     /* convert Hilbert index to vector */
932     hilbert_i2c(2, nBitsC, hndx, hcoord);
933     for (i = 2; i--; )
934     gpos[i] = ((double)hcoord[i] + rand()*(1./(RAND_MAX+.5))) /
935     (double)((bitmask_t)1 << nBitsC);
936 greg 3.53 square2disk(gpos, gpos[0], gpos[1]);
937 greg 3.7 /* compute Z-coordinate */
938 greg 3.6 gpos[2] = 1. - gpos[0]*gpos[0] - gpos[1]*gpos[1];
939 greg 3.32 gpos[2] = sqrt(gpos[2]*(gpos[2]>0));
940 greg 3.7 /* emit from back? */
941 greg 3.24 if ((cd->sidef == SD_BREFL) | (cd->sidef == SD_FXMIT))
942 greg 3.6 gpos[2] = -gpos[2];
943 greg 3.34 if (cd->isodist) { /* rotate isotropic sample */
944 greg 3.15 rotangle = atan2(-ioVec[1],-ioVec[0]);
945 greg 3.34 spinvector(ioVec, gpos, zvec, rotangle);
946 greg 3.15 } else
947     VCOPY(ioVec, gpos);
948 greg 3.6 return SDEnone;
949 greg 3.5 }
950    
951 greg 3.7 /* Advance pointer to the next non-white character in the string (or nul) */
952     static int
953     next_token(char **spp)
954     {
955     while (isspace(**spp))
956     ++*spp;
957     return **spp;
958     }
959    
960 greg 3.12 /* Advance pointer past matching token (or any token if c==0) */
961 greg 3.43 #define eat_token(spp,c) ((next_token(spp)==(c)) ^ !(c) ? *(*(spp))++ : 0)
962 greg 3.9
963 greg 3.7 /* Count words from this point in string to '}' */
964     static int
965     count_values(char *cp)
966     {
967     int n = 0;
968    
969 greg 3.9 while (next_token(&cp) != '}' && *cp) {
970 greg 3.11 while (!isspace(*cp) & (*cp != ',') & (*cp != '}'))
971     if (!*++cp)
972     break;
973 greg 3.7 ++n;
974 greg 3.9 eat_token(&cp, ',');
975 greg 3.7 }
976     return n;
977     }
978    
979     /* Load an array of real numbers, returning total */
980     static int
981     load_values(char **spp, float *va, int n)
982     {
983     float *v = va;
984     char *svnext;
985    
986     while (n-- > 0 && (svnext = fskip(*spp)) != NULL) {
987 greg 3.33 if ((*v++ = atof(*spp)) < 0)
988     v[-1] = 0;
989 greg 3.7 *spp = svnext;
990 greg 3.9 eat_token(spp, ',');
991 greg 3.7 }
992     return v - va;
993     }
994    
995     /* Load BSDF tree data */
996     static SDNode *
997     load_tree_data(char **spp, int nd)
998     {
999     SDNode *st;
1000     int n;
1001    
1002 greg 3.9 if (!eat_token(spp, '{')) {
1003 greg 3.7 strcpy(SDerrorDetail, "Missing '{' in tensor tree");
1004     return NULL;
1005     }
1006     if (next_token(spp) == '{') { /* tree branches */
1007     st = SDnewNode(nd, -1);
1008     if (st == NULL)
1009     return NULL;
1010     for (n = 0; n < 1<<nd; n++)
1011     if ((st->u.t[n] = load_tree_data(spp, nd)) == NULL) {
1012     SDfreeTre(st);
1013     return NULL;
1014     }
1015     } else { /* else load value grid */
1016     int bsiz;
1017     n = count_values(*spp); /* see how big the grid is */
1018 greg 3.15 for (bsiz = 0; bsiz < 8*sizeof(size_t); bsiz += nd)
1019 greg 3.7 if (1<<bsiz == n)
1020     break;
1021     if (bsiz >= 8*sizeof(size_t)) {
1022     strcpy(SDerrorDetail, "Illegal value count in tensor tree");
1023     return NULL;
1024     }
1025     st = SDnewNode(nd, bsiz/nd);
1026     if (st == NULL)
1027     return NULL;
1028     if (load_values(spp, st->u.v, n) != n) {
1029     strcpy(SDerrorDetail, "Real format error in tensor tree");
1030     SDfreeTre(st);
1031     return NULL;
1032     }
1033     }
1034 greg 3.9 if (!eat_token(spp, '}')) {
1035 greg 3.7 strcpy(SDerrorDetail, "Missing '}' in tensor tree");
1036     SDfreeTre(st);
1037     return NULL;
1038     }
1039 greg 3.9 eat_token(spp, ',');
1040 greg 3.7 return st;
1041     }
1042    
1043     /* Compute min. proj. solid angle and max. direct hemispherical scattering */
1044     static SDError
1045     get_extrema(SDSpectralDF *df)
1046     {
1047 greg 3.37 SDNode *st = (*(SDTre *)df->comp[0].dist).stc[tt_Y];
1048 greg 3.7 double stepWidth, dhemi, bmin[4], bmax[4];
1049    
1050     stepWidth = SDsmallestLeaf(st);
1051 greg 3.22 if (quantum > stepWidth) /* adjust quantization factor */
1052     quantum = stepWidth;
1053 greg 3.7 df->minProjSA = M_PI*stepWidth*stepWidth;
1054     if (stepWidth < .03125)
1055     stepWidth = .03125; /* 1/32 resolution good enough */
1056     df->maxHemi = .0;
1057     if (st->ndim == 3) { /* isotropic BSDF */
1058     bmin[1] = bmin[2] = .0;
1059     bmax[1] = bmax[2] = 1.;
1060     for (bmin[0] = .0; bmin[0] < .5-FTINY; bmin[0] += stepWidth) {
1061     bmax[0] = bmin[0] + stepWidth;
1062     dhemi = SDavgTreBox(st, bmin, bmax);
1063     if (dhemi > df->maxHemi)
1064     df->maxHemi = dhemi;
1065     }
1066     } else if (st->ndim == 4) { /* anisotropic BSDF */
1067     bmin[2] = bmin[3] = .0;
1068     bmax[2] = bmax[3] = 1.;
1069     for (bmin[0] = .0; bmin[0] < 1.-FTINY; bmin[0] += stepWidth) {
1070     bmax[0] = bmin[0] + stepWidth;
1071     for (bmin[1] = .0; bmin[1] < 1.-FTINY; bmin[1] += stepWidth) {
1072     bmax[1] = bmin[1] + stepWidth;
1073     dhemi = SDavgTreBox(st, bmin, bmax);
1074     if (dhemi > df->maxHemi)
1075     df->maxHemi = dhemi;
1076     }
1077     }
1078     } else
1079     return SDEinternal;
1080     /* correct hemispherical value */
1081     df->maxHemi *= M_PI;
1082     return SDEnone;
1083     }
1084    
1085     /* Load BSDF distribution for this wavelength */
1086     static SDError
1087 greg 3.37 load_bsdf_data(SDData *sd, ezxml_t wdb, int ct, int ndim)
1088 greg 3.7 {
1089     SDSpectralDF *df;
1090     SDTre *sdt;
1091     char *sdata;
1092     /* allocate BSDF component */
1093     sdata = ezxml_txt(ezxml_child(wdb, "WavelengthDataDirection"));
1094     if (!sdata)
1095     return SDEnone;
1096     /*
1097     * Remember that front and back are reversed from WINDOW 6 orientations
1098     */
1099 greg 3.24 if (!strcasecmp(sdata, "Transmission Front")) {
1100 greg 3.37 if (sd->tb == NULL && (sd->tb = SDnewSpectralDF(1)) == NULL)
1101 greg 3.25 return SDEmemory;
1102     df = sd->tb;
1103     } else if (!strcasecmp(sdata, "Transmission Back")) {
1104 greg 3.37 if (sd->tf == NULL && (sd->tf = SDnewSpectralDF(1)) == NULL)
1105 greg 3.7 return SDEmemory;
1106     df = sd->tf;
1107     } else if (!strcasecmp(sdata, "Reflection Front")) {
1108 greg 3.37 if (sd->rb == NULL && (sd->rb = SDnewSpectralDF(1)) == NULL)
1109 greg 3.7 return SDEmemory;
1110     df = sd->rb;
1111     } else if (!strcasecmp(sdata, "Reflection Back")) {
1112 greg 3.37 if (sd->rf == NULL && (sd->rf = SDnewSpectralDF(1)) == NULL)
1113 greg 3.7 return SDEmemory;
1114     df = sd->rf;
1115     } else
1116     return SDEnone;
1117     /* get angle bases */
1118     sdata = ezxml_txt(ezxml_child(wdb,"AngleBasis"));
1119     if (!sdata || strcasecmp(sdata, "LBNL/Shirley-Chiu")) {
1120     sprintf(SDerrorDetail, "%s angle basis for BSDF '%s'",
1121     !sdata ? "Missing" : "Unsupported", sd->name);
1122     return !sdata ? SDEformat : SDEsupport;
1123     }
1124 greg 3.37 if (df->comp[0].dist == NULL) { /* need to allocate BSDF tree? */
1125     sdt = (SDTre *)malloc(sizeof(SDTre));
1126     if (sdt == NULL)
1127     return SDEmemory;
1128     if (df == sd->rf)
1129     sdt->sidef = SD_FREFL;
1130     else if (df == sd->rb)
1131     sdt->sidef = SD_BREFL;
1132     else if (df == sd->tf)
1133     sdt->sidef = SD_FXMIT;
1134     else /* df == sd->tb */
1135     sdt->sidef = SD_BXMIT;
1136     sdt->stc[tt_Y] = sdt->stc[tt_u] = sdt->stc[tt_v] = NULL;
1137     df->comp[0].dist = sdt;
1138     df->comp[0].func = &SDhandleTre;
1139     } else {
1140     sdt = (SDTre *)df->comp[0].dist;
1141     if (sdt->stc[ct] != NULL) {
1142     SDfreeTre(sdt->stc[ct]);
1143     sdt->stc[ct] = NULL;
1144     }
1145     }
1146 greg 3.7 /* read BSDF data */
1147     sdata = ezxml_txt(ezxml_child(wdb, "ScatteringData"));
1148     if (!sdata || !next_token(&sdata)) {
1149     sprintf(SDerrorDetail, "Missing BSDF ScatteringData in '%s'",
1150     sd->name);
1151     return SDEformat;
1152     }
1153 greg 3.37 sdt->stc[ct] = load_tree_data(&sdata, ndim);
1154     if (sdt->stc[ct] == NULL)
1155 greg 3.7 return SDEformat;
1156     if (next_token(&sdata)) { /* check for unconsumed characters */
1157     sprintf(SDerrorDetail,
1158     "Extra characters at end of ScatteringData in '%s'",
1159     sd->name);
1160     return SDEformat;
1161     }
1162     /* flatten branches where possible */
1163 greg 3.37 sdt->stc[ct] = SDsimplifyTre(sdt->stc[ct]);
1164     if (sdt->stc[ct] == NULL)
1165 greg 3.7 return SDEinternal;
1166 greg 3.37 /* compute global quantities for Y */
1167     return (ct == tt_Y) ? get_extrema(df) : SDEnone;
1168 greg 3.7 }
1169    
1170     /* Find minimum value in tree */
1171     static float
1172     SDgetTreMin(const SDNode *st)
1173     {
1174 greg 3.10 float vmin = FHUGE;
1175 greg 3.7 int n;
1176    
1177     if (st->log2GR < 0) {
1178     for (n = 1<<st->ndim; n--; ) {
1179     float v = SDgetTreMin(st->u.t[n]);
1180     if (v < vmin)
1181     vmin = v;
1182     }
1183     } else {
1184     for (n = 1<<(st->ndim*st->log2GR); n--; )
1185     if (st->u.v[n] < vmin)
1186     vmin = st->u.v[n];
1187     }
1188     return vmin;
1189     }
1190    
1191     /* Subtract the given value from all tree nodes */
1192     static void
1193     SDsubtractTreVal(SDNode *st, float val)
1194     {
1195     int n;
1196    
1197     if (st->log2GR < 0) {
1198     for (n = 1<<st->ndim; n--; )
1199     SDsubtractTreVal(st->u.t[n], val);
1200     } else {
1201     for (n = 1<<(st->ndim*st->log2GR); n--; )
1202 greg 3.15 if ((st->u.v[n] -= val) < 0)
1203     st->u.v[n] = .0f;
1204 greg 3.7 }
1205     }
1206    
1207 greg 3.37 /* Subtract minimum Y value from BSDF */
1208 greg 3.7 static double
1209 greg 3.37 subtract_min_Y(SDNode *st)
1210 greg 3.7 {
1211 greg 3.46 const float vmaxmin = 1.5/M_PI;
1212     float vmin;
1213 greg 3.7 /* be sure to skip unused portion */
1214 greg 3.10 if (st->ndim == 3) {
1215     int n;
1216 greg 3.46 vmin = vmaxmin;
1217 greg 3.10 if (st->log2GR < 0) {
1218 greg 3.15 for (n = 0; n < 8; n += 2) {
1219 greg 3.10 float v = SDgetTreMin(st->u.t[n]);
1220     if (v < vmin)
1221     vmin = v;
1222     }
1223     } else if (st->log2GR) {
1224     for (n = 1 << (3*st->log2GR - 1); n--; )
1225     if (st->u.v[n] < vmin)
1226     vmin = st->u.v[n];
1227     } else
1228     vmin = st->u.v[0];
1229 greg 3.7 } else /* anisotropic covers entire tree */
1230     vmin = SDgetTreMin(st);
1231    
1232 greg 3.46 if ((vmin >= vmaxmin) | (vmin <= .01/M_PI))
1233 greg 3.39 return .0; /* not worth bothering about */
1234 greg 3.7
1235 greg 3.8 SDsubtractTreVal(st, vmin);
1236 greg 3.7
1237     return M_PI * vmin; /* return hemispherical value */
1238     }
1239    
1240 greg 3.37 /* Struct used in callback to find RGB extrema */
1241     typedef struct {
1242     SDNode **stc; /* original Y, u' & v' trees */
1243     float rgb[3]; /* RGB value */
1244     SDNode *new_stu, *new_stv; /* replacement u' & v' trees */
1245     } SDextRGBs;
1246    
1247     /* Callback to find minimum RGB from Y value plus CIE (u',v') trees */
1248     static int
1249     get_min_RGB(float yval, const double *cmin, double csiz, void *cptr)
1250     {
1251     SDextRGBs *mp = (SDextRGBs *)cptr;
1252     double cmax[SD_MAXDIM];
1253     float rgb[3];
1254    
1255     if (mp->stc[tt_Y]->ndim == 3) {
1256     if (cmin[0] + .5*csiz >= .5)
1257     return 0; /* ignore dead half of isotropic */
1258     } else
1259     cmax[3] = cmin[3] + csiz;
1260     cmax[0] = cmin[0] + csiz;
1261     cmax[1] = cmin[1] + csiz;
1262     cmax[2] = cmin[2] + csiz;
1263     /* average RGB color over voxel */
1264     SDyuv2rgb(yval, SDavgTreBox(mp->stc[tt_u], cmin, cmax),
1265     SDavgTreBox(mp->stc[tt_v], cmin, cmax), rgb);
1266     /* track smallest components */
1267     if (rgb[0] < mp->rgb[0]) mp->rgb[0] = rgb[0];
1268     if (rgb[1] < mp->rgb[1]) mp->rgb[1] = rgb[1];
1269     if (rgb[2] < mp->rgb[2]) mp->rgb[2] = rgb[2];
1270     return 0;
1271     }
1272    
1273     /* Callback to build adjusted u' tree */
1274     static int
1275     adjust_utree(float uprime, const double *cmin, double csiz, void *cptr)
1276     {
1277     SDextRGBs *mp = (SDextRGBs *)cptr;
1278     double cmax[SD_MAXDIM];
1279 greg 3.38 double yval;
1280 greg 3.37 float rgb[3];
1281 greg 3.38 C_COLOR clr;
1282 greg 3.37
1283     if (mp->stc[tt_Y]->ndim == 3) {
1284     if (cmin[0] + .5*csiz >= .5)
1285     return 0; /* ignore dead half of isotropic */
1286     } else
1287     cmax[3] = cmin[3] + csiz;
1288     cmax[0] = cmin[0] + csiz;
1289     cmax[1] = cmin[1] + csiz;
1290     cmax[2] = cmin[2] + csiz;
1291     /* average RGB color over voxel */
1292 greg 3.38 SDyuv2rgb(yval=SDavgTreBox(mp->stc[tt_Y], cmin, cmax), uprime,
1293 greg 3.37 SDavgTreBox(mp->stc[tt_v], cmin, cmax), rgb);
1294 greg 3.38 /* subtract minimum (& clamp) */
1295     if ((rgb[0] -= mp->rgb[0]) < 1e-5*yval) rgb[0] = 1e-5*yval;
1296     if ((rgb[1] -= mp->rgb[1]) < 1e-5*yval) rgb[1] = 1e-5*yval;
1297     if ((rgb[2] -= mp->rgb[2]) < 1e-5*yval) rgb[2] = 1e-5*yval;
1298     c_fromSharpRGB(rgb, &clr); /* compute new u' for adj. RGB */
1299     uprime = 4.*clr.cx/(-2.*clr.cx + 12.*clr.cy + 3.);
1300 greg 3.37 /* assign in new u' tree */
1301     mp->new_stu = SDsetVoxel(mp->new_stu, mp->stc[tt_Y]->ndim,
1302     cmin, csiz, uprime);
1303     return -(mp->new_stu == NULL);
1304     }
1305    
1306     /* Callback to build adjusted v' tree */
1307     static int
1308     adjust_vtree(float vprime, const double *cmin, double csiz, void *cptr)
1309     {
1310     SDextRGBs *mp = (SDextRGBs *)cptr;
1311     double cmax[SD_MAXDIM];
1312 greg 3.38 double yval;
1313 greg 3.37 float rgb[3];
1314 greg 3.38 C_COLOR clr;
1315 greg 3.37
1316     if (mp->stc[tt_Y]->ndim == 3) {
1317     if (cmin[0] + .5*csiz >= .5)
1318     return 0; /* ignore dead half of isotropic */
1319     } else
1320     cmax[3] = cmin[3] + csiz;
1321     cmax[0] = cmin[0] + csiz;
1322     cmax[1] = cmin[1] + csiz;
1323     cmax[2] = cmin[2] + csiz;
1324     /* average RGB color over voxel */
1325 greg 3.38 SDyuv2rgb(yval=SDavgTreBox(mp->stc[tt_Y], cmin, cmax),
1326 greg 3.37 SDavgTreBox(mp->stc[tt_u], cmin, cmax),
1327     vprime, rgb);
1328 greg 3.38 /* subtract minimum (& clamp) */
1329     if ((rgb[0] -= mp->rgb[0]) < 1e-5*yval) rgb[0] = 1e-5*yval;
1330     if ((rgb[1] -= mp->rgb[1]) < 1e-5*yval) rgb[1] = 1e-5*yval;
1331     if ((rgb[2] -= mp->rgb[2]) < 1e-5*yval) rgb[2] = 1e-5*yval;
1332     c_fromSharpRGB(rgb, &clr); /* compute new v' for adj. RGB */
1333     vprime = 9.*clr.cy/(-2.*clr.cx + 12.*clr.cy + 3.);
1334 greg 3.37 /* assign in new v' tree */
1335     mp->new_stv = SDsetVoxel(mp->new_stv, mp->stc[tt_Y]->ndim,
1336     cmin, csiz, vprime);
1337     return -(mp->new_stv == NULL);
1338     }
1339    
1340     /* Subtract minimum (diffuse) color and return luminance & CIE (x,y) */
1341     static double
1342     subtract_min_RGB(C_COLOR *cs, SDNode *stc[])
1343     {
1344     SDextRGBs my_min;
1345     double ymin;
1346    
1347     my_min.stc = stc;
1348     my_min.rgb[0] = my_min.rgb[1] = my_min.rgb[2] = FHUGE;
1349     my_min.new_stu = my_min.new_stv = NULL;
1350     /* get minimum RGB value */
1351     SDtraverseTre(stc[tt_Y], NULL, 0, get_min_RGB, &my_min);
1352 greg 3.40 /* convert to C_COLOR */
1353     ymin = c_fromSharpRGB(my_min.rgb, cs);
1354 greg 3.46 if ((ymin >= .5*FHUGE) | (ymin <= .01/M_PI))
1355     return .0; /* close to zero or no tree */
1356 greg 3.40 /* adjust u' & v' trees */
1357 greg 3.37 SDtraverseTre(stc[tt_u], NULL, 0, adjust_utree, &my_min);
1358     SDtraverseTre(stc[tt_v], NULL, 0, adjust_vtree, &my_min);
1359     SDfreeTre(stc[tt_u]); SDfreeTre(stc[tt_v]);
1360     stc[tt_u] = SDsimplifyTre(my_min.new_stu);
1361     stc[tt_v] = SDsimplifyTre(my_min.new_stv);
1362 greg 3.40 /* subtract Y & return hemispherical */
1363 greg 3.37 SDsubtractTreVal(stc[tt_Y], ymin);
1364 greg 3.40
1365     return M_PI * ymin;
1366 greg 3.37 }
1367    
1368 greg 3.7 /* Extract and separate diffuse portion of BSDF */
1369     static void
1370     extract_diffuse(SDValue *dv, SDSpectralDF *df)
1371     {
1372     int n;
1373 greg 3.37 SDTre *sdt;
1374 greg 3.7
1375     if (df == NULL || df->ncomp <= 0) {
1376     dv->spec = c_dfcolor;
1377     dv->cieY = .0;
1378     return;
1379     }
1380 greg 3.37 sdt = (SDTre *)df->comp[0].dist;
1381     /* subtract minimum color/grayscale */
1382     if (sdt->stc[tt_u] != NULL && sdt->stc[tt_v] != NULL) {
1383     int i = 3*(tt_RGB_coef[1] < .001);
1384     while (i--) { /* initialize on first call */
1385     float rgb[3];
1386     rgb[0] = rgb[1] = rgb[2] = .0f; rgb[i] = 1.f;
1387     tt_RGB_coef[i] = c_fromSharpRGB(rgb, &tt_RGB_prim[i]);
1388     }
1389     memcpy(df->comp[0].cspec, tt_RGB_prim, sizeof(tt_RGB_prim));
1390     dv->cieY = subtract_min_RGB(&dv->spec, sdt->stc);
1391     } else {
1392 greg 3.42 df->comp[0].cspec[0] = dv->spec = c_dfcolor;
1393 greg 3.37 dv->cieY = subtract_min_Y(sdt->stc[tt_Y]);
1394 greg 3.7 }
1395     df->maxHemi -= dv->cieY; /* adjust maximum hemispherical */
1396 greg 3.47
1397     c_ccvt(&dv->spec, C_CSXY); /* make sure (x,y) is set */
1398 greg 3.7 }
1399    
1400 greg 3.1 /* Load a variable-resolution BSDF tree from an open XML file */
1401     SDError
1402 greg 3.4 SDloadTre(SDData *sd, ezxml_t wtl)
1403 greg 3.1 {
1404 greg 3.7 SDError ec;
1405     ezxml_t wld, wdb;
1406     int rank;
1407     char *txt;
1408     /* basic checks and tensor rank */
1409     txt = ezxml_txt(ezxml_child(ezxml_child(wtl,
1410     "DataDefinition"), "IncidentDataStructure"));
1411     if (txt == NULL || !*txt) {
1412     sprintf(SDerrorDetail,
1413     "BSDF \"%s\": missing IncidentDataStructure",
1414     sd->name);
1415     return SDEformat;
1416     }
1417     if (!strcasecmp(txt, "TensorTree3"))
1418     rank = 3;
1419     else if (!strcasecmp(txt, "TensorTree4"))
1420     rank = 4;
1421     else {
1422     sprintf(SDerrorDetail,
1423     "BSDF \"%s\": unsupported IncidentDataStructure",
1424     sd->name);
1425     return SDEsupport;
1426     }
1427     /* load BSDF components */
1428     for (wld = ezxml_child(wtl, "WavelengthData");
1429     wld != NULL; wld = wld->next) {
1430 greg 3.37 const char *cnm = ezxml_txt(ezxml_child(wld,"Wavelength"));
1431     int ct = -1;
1432     if (!strcasecmp(cnm, "Visible"))
1433     ct = tt_Y;
1434     else if (!strcasecmp(cnm, "CIE-u"))
1435     ct = tt_u;
1436     else if (!strcasecmp(cnm, "CIE-v"))
1437     ct = tt_v;
1438     else
1439     continue;
1440 greg 3.7 for (wdb = ezxml_child(wld, "WavelengthDataBlock");
1441     wdb != NULL; wdb = wdb->next)
1442 greg 3.37 if ((ec = load_bsdf_data(sd, wdb, ct, rank)) != SDEnone)
1443 greg 3.7 return ec;
1444     }
1445     /* separate diffuse components */
1446     extract_diffuse(&sd->rLambFront, sd->rf);
1447     extract_diffuse(&sd->rLambBack, sd->rb);
1448 greg 3.50 extract_diffuse(&sd->tLambFront, sd->tf);
1449 greg 3.51 if (sd->tb != NULL) {
1450     extract_diffuse(&sd->tLambBack, sd->tb);
1451     if (sd->tf == NULL)
1452     sd->tLambFront = sd->tLambBack;
1453     } else if (sd->tf != NULL)
1454     sd->tLambBack = sd->tLambFront;
1455 greg 3.7 /* return success */
1456     return SDEnone;
1457 greg 3.1 }
1458    
1459     /* Variable resolution BSDF methods */
1460 greg 3.45 const SDFunc SDhandleTre = {
1461 greg 3.5 &SDgetTreBSDF,
1462 greg 3.6 &SDqueryTreProjSA,
1463     &SDgetTreCDist,
1464     &SDsampTreCDist,
1465     &SDFreeBTre,
1466 greg 3.1 };