| 10 |
|
* |
| 11 |
|
*/ |
| 12 |
|
|
| 13 |
+ |
#define _USE_MATH_DEFINES |
| 14 |
|
#include "rtio.h" |
| 15 |
|
#include <stdlib.h> |
| 16 |
|
#include <math.h> |
| 18 |
|
#include "ezxml.h" |
| 19 |
|
#include "bsdf.h" |
| 20 |
|
#include "bsdf_t.h" |
| 21 |
+ |
#include "hilbert.h" |
| 22 |
|
|
| 23 |
+ |
/* Callback function type for SDtraverseTre() */ |
| 24 |
+ |
typedef int SDtreCallback(float val, const double *cmin, |
| 25 |
+ |
double csiz, void *cptr); |
| 26 |
+ |
|
| 27 |
+ |
/* reference width maximum (1.0) */ |
| 28 |
+ |
static const unsigned iwbits = sizeof(unsigned)*4; |
| 29 |
+ |
static const unsigned iwmax = (1<<(sizeof(unsigned)*4))-1; |
| 30 |
+ |
/* maximum cumulative value */ |
| 31 |
+ |
static const unsigned cumlmax = ~0; |
| 32 |
+ |
/* constant z-vector */ |
| 33 |
+ |
static const FVECT zvec = {.0, .0, 1.}; |
| 34 |
+ |
|
| 35 |
+ |
/* Struct used for our distribution-building callback */ |
| 36 |
+ |
typedef struct { |
| 37 |
+ |
int nic; /* number of input coordinates */ |
| 38 |
+ |
unsigned alen; /* current array length */ |
| 39 |
+ |
unsigned nall; /* number of allocated entries */ |
| 40 |
+ |
unsigned wmin; /* minimum square size so far */ |
| 41 |
+ |
unsigned wmax; /* maximum square size */ |
| 42 |
+ |
struct outdir_s { |
| 43 |
+ |
unsigned hent; /* entering Hilbert index */ |
| 44 |
+ |
int wid; /* this square size */ |
| 45 |
+ |
float bsdf; /* BSDF for this square */ |
| 46 |
+ |
} *darr; /* output direction array */ |
| 47 |
+ |
} SDdistScaffold; |
| 48 |
+ |
|
| 49 |
|
/* Allocate a new scattering distribution node */ |
| 50 |
|
static SDNode * |
| 51 |
|
SDnewNode(int nd, int lg) |
| 63 |
|
} |
| 64 |
|
if (lg < 0) { |
| 65 |
|
st = (SDNode *)malloc(sizeof(SDNode) + |
| 66 |
< |
((1<<nd) - 1)*sizeof(st->u.t[0])); |
| 67 |
< |
if (st != NULL) |
| 68 |
< |
memset(st->u.t, 0, (1<<nd)*sizeof(st->u.t[0])); |
| 69 |
< |
} else |
| 66 |
> |
sizeof(st->u.t[0])*((1<<nd) - 1)); |
| 67 |
> |
if (st == NULL) { |
| 68 |
> |
sprintf(SDerrorDetail, |
| 69 |
> |
"Cannot allocate %d branch BSDF tree", 1<<nd); |
| 70 |
> |
return NULL; |
| 71 |
> |
} |
| 72 |
> |
memset(st->u.t, 0, sizeof(st->u.t[0])<<nd); |
| 73 |
> |
} else { |
| 74 |
|
st = (SDNode *)malloc(sizeof(SDNode) + |
| 75 |
< |
((1 << nd*lg) - 1)*sizeof(st->u.v[0])); |
| 76 |
< |
|
| 45 |
< |
if (st == NULL) { |
| 46 |
< |
if (lg < 0) |
| 75 |
> |
sizeof(st->u.v[0])*((1 << nd*lg) - 1)); |
| 76 |
> |
if (st == NULL) { |
| 77 |
|
sprintf(SDerrorDetail, |
| 48 |
– |
"Cannot allocate %d branch BSDF tree", nd); |
| 49 |
– |
else |
| 50 |
– |
sprintf(SDerrorDetail, |
| 78 |
|
"Cannot allocate %d BSDF leaves", 1 << nd*lg); |
| 79 |
< |
return NULL; |
| 79 |
> |
return NULL; |
| 80 |
> |
} |
| 81 |
|
} |
| 82 |
|
st->ndim = nd; |
| 83 |
|
st->log2GR = lg; |
| 86 |
|
|
| 87 |
|
/* Free an SD tree */ |
| 88 |
|
static void |
| 89 |
< |
SDfreeTree(void *p) |
| 89 |
> |
SDfreeTre(SDNode *st) |
| 90 |
|
{ |
| 91 |
< |
SDNode *st = (SDNode *)p; |
| 64 |
< |
int i; |
| 91 |
> |
int n; |
| 92 |
|
|
| 93 |
|
if (st == NULL) |
| 94 |
|
return; |
| 95 |
< |
for (i = (st->log2GR < 0) << st->ndim; i--; ) |
| 96 |
< |
SDfreeTree(st->u.t[i]); |
| 97 |
< |
free((void *)st); |
| 95 |
> |
for (n = (st->log2GR < 0) << st->ndim; n--; ) |
| 96 |
> |
SDfreeTre(st->u.t[n]); |
| 97 |
> |
free(st); |
| 98 |
|
} |
| 99 |
|
|
| 100 |
+ |
/* Free a variable-resolution BSDF */ |
| 101 |
+ |
static void |
| 102 |
+ |
SDFreeBTre(void *p) |
| 103 |
+ |
{ |
| 104 |
+ |
SDTre *sdt = (SDTre *)p; |
| 105 |
+ |
|
| 106 |
+ |
if (sdt == NULL) |
| 107 |
+ |
return; |
| 108 |
+ |
SDfreeTre(sdt->st); |
| 109 |
+ |
free(sdt); |
| 110 |
+ |
} |
| 111 |
+ |
|
| 112 |
+ |
/* Fill branch's worth of grid values from subtree */ |
| 113 |
+ |
static void |
| 114 |
+ |
fill_grid_branch(float *dptr, const float *sptr, int nd, int shft) |
| 115 |
+ |
{ |
| 116 |
+ |
unsigned n = 1 << (shft-1); |
| 117 |
+ |
|
| 118 |
+ |
if (!--nd) { /* end on the line */ |
| 119 |
+ |
memcpy(dptr, sptr, sizeof(*dptr)*n); |
| 120 |
+ |
return; |
| 121 |
+ |
} |
| 122 |
+ |
while (n--) /* recurse on each slice */ |
| 123 |
+ |
fill_grid_branch(dptr + (n << shft*nd), |
| 124 |
+ |
sptr + (n << (shft-1)*nd), nd, shft); |
| 125 |
+ |
} |
| 126 |
+ |
|
| 127 |
+ |
/* Get pointer at appropriate offset for the given branch */ |
| 128 |
+ |
static float * |
| 129 |
+ |
grid_branch_start(SDNode *st, int n) |
| 130 |
+ |
{ |
| 131 |
+ |
unsigned skipsiz = 1 << (st->log2GR - 1); |
| 132 |
+ |
float *vptr = st->u.v; |
| 133 |
+ |
int i; |
| 134 |
+ |
|
| 135 |
+ |
for (i = st->ndim; i--; skipsiz <<= st->log2GR) |
| 136 |
+ |
if (1<<i & n) |
| 137 |
+ |
vptr += skipsiz; |
| 138 |
+ |
return vptr; |
| 139 |
+ |
} |
| 140 |
+ |
|
| 141 |
+ |
/* Simplify (consolidate) a tree by flattening uniform depth regions */ |
| 142 |
+ |
static SDNode * |
| 143 |
+ |
SDsimplifyTre(SDNode *st) |
| 144 |
+ |
{ |
| 145 |
+ |
int match, n; |
| 146 |
+ |
|
| 147 |
+ |
if (st == NULL) /* check for invalid tree */ |
| 148 |
+ |
return NULL; |
| 149 |
+ |
if (st->log2GR >= 0) /* grid just returns unaltered */ |
| 150 |
+ |
return st; |
| 151 |
+ |
match = 1; /* check if grids below match */ |
| 152 |
+ |
for (n = 0; n < 1<<st->ndim; n++) { |
| 153 |
+ |
if ((st->u.t[n] = SDsimplifyTre(st->u.t[n])) == NULL) |
| 154 |
+ |
return NULL; /* propogate error up call stack */ |
| 155 |
+ |
match &= (st->u.t[n]->log2GR == st->u.t[0]->log2GR); |
| 156 |
+ |
} |
| 157 |
+ |
if (match && (match = st->u.t[0]->log2GR) >= 0) { |
| 158 |
+ |
SDNode *stn = SDnewNode(st->ndim, match + 1); |
| 159 |
+ |
if (stn == NULL) /* out of memory? */ |
| 160 |
+ |
return st; |
| 161 |
+ |
/* transfer values to new grid */ |
| 162 |
+ |
for (n = 1 << st->ndim; n--; ) |
| 163 |
+ |
fill_grid_branch(grid_branch_start(stn, n), |
| 164 |
+ |
st->u.t[n]->u.v, stn->ndim, stn->log2GR); |
| 165 |
+ |
SDfreeTre(st); /* free old tree */ |
| 166 |
+ |
st = stn; /* return new one */ |
| 167 |
+ |
} |
| 168 |
+ |
return st; |
| 169 |
+ |
} |
| 170 |
+ |
|
| 171 |
+ |
/* Find smallest leaf in tree */ |
| 172 |
+ |
static double |
| 173 |
+ |
SDsmallestLeaf(const SDNode *st) |
| 174 |
+ |
{ |
| 175 |
+ |
if (st->log2GR < 0) { /* tree branches */ |
| 176 |
+ |
double lmin = 1.; |
| 177 |
+ |
int n; |
| 178 |
+ |
for (n = 1<<st->ndim; n--; ) { |
| 179 |
+ |
double lsiz = SDsmallestLeaf(st->u.t[n]); |
| 180 |
+ |
if (lsiz < lmin) |
| 181 |
+ |
lmin = lsiz; |
| 182 |
+ |
} |
| 183 |
+ |
return .5*lmin; |
| 184 |
+ |
} |
| 185 |
+ |
/* leaf grid width */ |
| 186 |
+ |
return 1. / (double)(1 << st->log2GR); |
| 187 |
+ |
} |
| 188 |
+ |
|
| 189 |
|
/* Add up N-dimensional hypercube array values over the given box */ |
| 190 |
|
static double |
| 191 |
< |
SDiterSum(const float *va, int nd, int siz, const int *imin, const int *imax) |
| 191 |
> |
SDiterSum(const float *va, int nd, int shft, const int *imin, const int *imax) |
| 192 |
|
{ |
| 193 |
+ |
const unsigned skipsiz = 1 << --nd*shft; |
| 194 |
|
double sum = .0; |
| 78 |
– |
unsigned skipsiz = 1; |
| 195 |
|
int i; |
| 196 |
< |
|
| 197 |
< |
for (i = nd; --i > 0; ) |
| 198 |
< |
skipsiz *= siz; |
| 196 |
> |
|
| 197 |
> |
va += *imin * skipsiz; |
| 198 |
> |
|
| 199 |
|
if (skipsiz == 1) |
| 200 |
|
for (i = *imin; i < *imax; i++) |
| 201 |
< |
sum += va[i]; |
| 201 |
> |
sum += *va++; |
| 202 |
|
else |
| 203 |
< |
for (i = *imin; i < *imax; i++) |
| 204 |
< |
sum += SDiterSum(va + i*skipsiz, |
| 89 |
< |
nd-1, siz, imin+1, imax+1); |
| 203 |
> |
for (i = *imin; i < *imax; i++, va += skipsiz) |
| 204 |
> |
sum += SDiterSum(va, nd, shft, imin+1, imax+1); |
| 205 |
|
return sum; |
| 206 |
|
} |
| 207 |
|
|
| 208 |
|
/* Average BSDF leaves over an orthotope defined by the unit hypercube */ |
| 209 |
|
static double |
| 210 |
< |
SDavgBox(const SDNode *st, const double *bmin, const double *bmax) |
| 210 |
> |
SDavgTreBox(const SDNode *st, const double *bmin, const double *bmax) |
| 211 |
|
{ |
| 97 |
– |
int imin[SD_MAXDIM], imax[SD_MAXDIM]; |
| 212 |
|
unsigned n; |
| 213 |
|
int i; |
| 214 |
|
|
| 218 |
|
for (i = st->ndim; i--; ) { |
| 219 |
|
if (bmin[i] >= 1.) |
| 220 |
|
return .0; |
| 221 |
< |
if (bmax[i] <= .0) |
| 221 |
> |
if (bmax[i] <= 0) |
| 222 |
|
return .0; |
| 223 |
|
if (bmin[i] >= bmax[i]) |
| 224 |
|
return .0; |
| 226 |
|
if (st->log2GR < 0) { /* iterate on subtree */ |
| 227 |
|
double sum = .0, wsum = 1e-20; |
| 228 |
|
double sbmin[SD_MAXDIM], sbmax[SD_MAXDIM], w; |
| 115 |
– |
|
| 229 |
|
for (n = 1 << st->ndim; n--; ) { |
| 230 |
|
w = 1.; |
| 231 |
|
for (i = st->ndim; i--; ) { |
| 237 |
|
} |
| 238 |
|
if (sbmin[i] < .0) sbmin[i] = .0; |
| 239 |
|
if (sbmax[i] > 1.) sbmax[i] = 1.; |
| 240 |
+ |
if (sbmin[i] >= sbmax[i]) { |
| 241 |
+ |
w = .0; |
| 242 |
+ |
break; |
| 243 |
+ |
} |
| 244 |
|
w *= sbmax[i] - sbmin[i]; |
| 245 |
|
} |
| 246 |
|
if (w > 1e-10) { |
| 247 |
< |
sum += w * SDavgBox(st->u.t[n], sbmin, sbmax); |
| 247 |
> |
sum += w * SDavgTreBox(st->u.t[n], sbmin, sbmax); |
| 248 |
|
wsum += w; |
| 249 |
|
} |
| 250 |
|
} |
| 251 |
|
return sum / wsum; |
| 252 |
+ |
} else { /* iterate over leaves */ |
| 253 |
+ |
int imin[SD_MAXDIM], imax[SD_MAXDIM]; |
| 254 |
+ |
|
| 255 |
+ |
n = 1; |
| 256 |
+ |
for (i = st->ndim; i--; ) { |
| 257 |
+ |
imin[i] = (bmin[i] <= 0) ? 0 : |
| 258 |
+ |
(int)((1 << st->log2GR)*bmin[i]); |
| 259 |
+ |
imax[i] = (bmax[i] >= 1.) ? (1 << st->log2GR) : |
| 260 |
+ |
(int)((1 << st->log2GR)*bmax[i] + .999999); |
| 261 |
+ |
n *= imax[i] - imin[i]; |
| 262 |
+ |
} |
| 263 |
+ |
if (n) |
| 264 |
+ |
return SDiterSum(st->u.v, st->ndim, |
| 265 |
+ |
st->log2GR, imin, imax) / (double)n; |
| 266 |
|
} |
| 267 |
< |
n = 1; /* iterate over leaves */ |
| 267 |
> |
return .0; |
| 268 |
> |
} |
| 269 |
> |
|
| 270 |
> |
/* Recursive call for SDtraverseTre() */ |
| 271 |
> |
static int |
| 272 |
> |
SDdotravTre(const SDNode *st, const double *pos, int cmask, |
| 273 |
> |
SDtreCallback *cf, void *cptr, |
| 274 |
> |
const double *cmin, double csiz) |
| 275 |
> |
{ |
| 276 |
> |
int rv, rval = 0; |
| 277 |
> |
double bmin[SD_MAXDIM]; |
| 278 |
> |
int i, n; |
| 279 |
> |
/* in branches? */ |
| 280 |
> |
if (st->log2GR < 0) { |
| 281 |
> |
unsigned skipmask = 0; |
| 282 |
> |
csiz *= .5; |
| 283 |
> |
for (i = st->ndim; i--; ) |
| 284 |
> |
if (1<<i & cmask) |
| 285 |
> |
if (pos[i] < cmin[i] + csiz) |
| 286 |
> |
for (n = 1 << st->ndim; n--; ) { |
| 287 |
> |
if (n & 1<<i) |
| 288 |
> |
skipmask |= 1<<n; |
| 289 |
> |
} |
| 290 |
> |
else |
| 291 |
> |
for (n = 1 << st->ndim; n--; ) { |
| 292 |
> |
if (!(n & 1<<i)) |
| 293 |
> |
skipmask |= 1<<n; |
| 294 |
> |
} |
| 295 |
> |
for (n = 1 << st->ndim; n--; ) { |
| 296 |
> |
if (1<<n & skipmask) |
| 297 |
> |
continue; |
| 298 |
> |
for (i = st->ndim; i--; ) |
| 299 |
> |
if (1<<i & n) |
| 300 |
> |
bmin[i] = cmin[i] + csiz; |
| 301 |
> |
else |
| 302 |
> |
bmin[i] = cmin[i]; |
| 303 |
> |
|
| 304 |
> |
rval += rv = SDdotravTre(st->u.t[n], pos, cmask, |
| 305 |
> |
cf, cptr, bmin, csiz); |
| 306 |
> |
if (rv < 0) |
| 307 |
> |
return rv; |
| 308 |
> |
} |
| 309 |
> |
} else { /* else traverse leaves */ |
| 310 |
> |
int clim[SD_MAXDIM][2]; |
| 311 |
> |
int cpos[SD_MAXDIM]; |
| 312 |
> |
|
| 313 |
> |
if (st->log2GR == 0) /* short cut */ |
| 314 |
> |
return (*cf)(st->u.v[0], cmin, csiz, cptr); |
| 315 |
> |
|
| 316 |
> |
csiz /= (double)(1 << st->log2GR); |
| 317 |
> |
/* assign coord. ranges */ |
| 318 |
> |
for (i = st->ndim; i--; ) |
| 319 |
> |
if (1<<i & cmask) { |
| 320 |
> |
clim[i][0] = (pos[i] - cmin[i])/csiz; |
| 321 |
> |
/* check overflow from f.p. error */ |
| 322 |
> |
clim[i][0] -= clim[i][0] >> st->log2GR; |
| 323 |
> |
clim[i][1] = clim[i][0] + 1; |
| 324 |
> |
} else { |
| 325 |
> |
clim[i][0] = 0; |
| 326 |
> |
clim[i][1] = 1 << st->log2GR; |
| 327 |
> |
} |
| 328 |
> |
#if (SD_MAXDIM == 4) |
| 329 |
> |
bmin[0] = cmin[0] + csiz*clim[0][0]; |
| 330 |
> |
for (cpos[0] = clim[0][0]; cpos[0] < clim[0][1]; cpos[0]++) { |
| 331 |
> |
bmin[1] = cmin[1] + csiz*clim[1][0]; |
| 332 |
> |
for (cpos[1] = clim[1][0]; cpos[1] < clim[1][1]; cpos[1]++) { |
| 333 |
> |
bmin[2] = cmin[2] + csiz*clim[2][0]; |
| 334 |
> |
if (st->ndim == 3) { |
| 335 |
> |
cpos[2] = clim[2][0]; |
| 336 |
> |
n = cpos[0]; |
| 337 |
> |
for (i = 1; i < 3; i++) |
| 338 |
> |
n = (n << st->log2GR) + cpos[i]; |
| 339 |
> |
for ( ; cpos[2] < clim[2][1]; cpos[2]++) { |
| 340 |
> |
rval += rv = (*cf)(st->u.v[n++], bmin, csiz, cptr); |
| 341 |
> |
if (rv < 0) |
| 342 |
> |
return rv; |
| 343 |
> |
bmin[2] += csiz; |
| 344 |
> |
} |
| 345 |
> |
} else { |
| 346 |
> |
for (cpos[2] = clim[2][0]; cpos[2] < clim[2][1]; cpos[2]++) { |
| 347 |
> |
bmin[3] = cmin[3] + csiz*(cpos[3] = clim[3][0]); |
| 348 |
> |
n = cpos[0]; |
| 349 |
> |
for (i = 1; i < 4; i++) |
| 350 |
> |
n = (n << st->log2GR) + cpos[i]; |
| 351 |
> |
for ( ; cpos[3] < clim[3][1]; cpos[3]++) { |
| 352 |
> |
rval += rv = (*cf)(st->u.v[n++], bmin, csiz, cptr); |
| 353 |
> |
if (rv < 0) |
| 354 |
> |
return rv; |
| 355 |
> |
bmin[3] += csiz; |
| 356 |
> |
} |
| 357 |
> |
bmin[2] += csiz; |
| 358 |
> |
} |
| 359 |
> |
} |
| 360 |
> |
bmin[1] += csiz; |
| 361 |
> |
} |
| 362 |
> |
bmin[0] += csiz; |
| 363 |
> |
} |
| 364 |
> |
#else |
| 365 |
> |
_!_ "broken code segment!" |
| 366 |
> |
#endif |
| 367 |
> |
} |
| 368 |
> |
return rval; |
| 369 |
> |
} |
| 370 |
> |
|
| 371 |
> |
/* Traverse a tree, visiting nodes in a slice that fits partial position */ |
| 372 |
> |
static int |
| 373 |
> |
SDtraverseTre(const SDNode *st, const double *pos, int cmask, |
| 374 |
> |
SDtreCallback *cf, void *cptr) |
| 375 |
> |
{ |
| 376 |
> |
static double czero[SD_MAXDIM]; |
| 377 |
> |
int i; |
| 378 |
> |
/* check arguments */ |
| 379 |
> |
if ((st == NULL) | (cf == NULL)) |
| 380 |
> |
return -1; |
| 381 |
> |
for (i = st->ndim; i--; ) |
| 382 |
> |
if (1<<i & cmask && (pos[i] < 0) | (pos[i] >= 1.)) |
| 383 |
> |
return -1; |
| 384 |
> |
|
| 385 |
> |
return SDdotravTre(st, pos, cmask, cf, cptr, czero, 1.); |
| 386 |
> |
} |
| 387 |
> |
|
| 388 |
> |
/* Look up tree value at the given grid position */ |
| 389 |
> |
static float |
| 390 |
> |
SDlookupTre(const SDNode *st, const double *pos, double *hcube) |
| 391 |
> |
{ |
| 392 |
> |
double spos[SD_MAXDIM]; |
| 393 |
> |
int i, n, t; |
| 394 |
> |
/* initialize voxel return */ |
| 395 |
> |
if (hcube) { |
| 396 |
> |
hcube[i = st->ndim] = 1.; |
| 397 |
> |
while (i--) |
| 398 |
> |
hcube[i] = .0; |
| 399 |
> |
} |
| 400 |
> |
/* climb the tree */ |
| 401 |
> |
while (st->log2GR < 0) { |
| 402 |
> |
n = 0; /* move to appropriate branch */ |
| 403 |
> |
if (hcube) hcube[st->ndim] *= .5; |
| 404 |
> |
for (i = st->ndim; i--; ) { |
| 405 |
> |
spos[i] = 2.*pos[i]; |
| 406 |
> |
t = (spos[i] >= 1.); |
| 407 |
> |
n |= t<<i; |
| 408 |
> |
spos[i] -= (double)t; |
| 409 |
> |
if (hcube) hcube[i] += (double)t * hcube[st->ndim]; |
| 410 |
> |
} |
| 411 |
> |
st = st->u.t[n]; /* avoids tail recursion */ |
| 412 |
> |
pos = spos; |
| 413 |
> |
} |
| 414 |
> |
if (st->log2GR == 0) /* short cut */ |
| 415 |
> |
return st->u.v[0]; |
| 416 |
> |
n = t = 0; /* find grid array index */ |
| 417 |
|
for (i = st->ndim; i--; ) { |
| 418 |
< |
imin[i] = (bmin[i] <= .0) ? 0 |
| 419 |
< |
: (int)((1 << st->log2GR)*bmin[i]); |
| 140 |
< |
imax[i] = (bmax[i] >= 1.) ? (1 << st->log2GR) |
| 141 |
< |
: (int)((1 << st->log2GR)*bmax[i] + .999999); |
| 142 |
< |
n *= imax[i] - imin[i]; |
| 418 |
> |
n += (int)((1<<st->log2GR)*pos[i]) << t; |
| 419 |
> |
t += st->log2GR; |
| 420 |
|
} |
| 421 |
< |
if (!n) |
| 421 |
> |
if (hcube) { /* compute final hypercube */ |
| 422 |
> |
hcube[st->ndim] /= (double)(1<<st->log2GR); |
| 423 |
> |
for (i = st->ndim; i--; ) |
| 424 |
> |
hcube[i] += floor((1<<st->log2GR)*pos[i])*hcube[st->ndim]; |
| 425 |
> |
} |
| 426 |
> |
return st->u.v[n]; /* no interpolation */ |
| 427 |
> |
} |
| 428 |
> |
|
| 429 |
> |
/* Query BSDF value and sample hypercube for the given vectors */ |
| 430 |
> |
static float |
| 431 |
> |
SDqueryTre(const SDTre *sdt, const FVECT outVec, const FVECT inVec, double *hc) |
| 432 |
> |
{ |
| 433 |
> |
FVECT rOutVec; |
| 434 |
> |
double gridPos[4]; |
| 435 |
> |
|
| 436 |
> |
switch (sdt->sidef) { /* whose side are you on? */ |
| 437 |
> |
case SD_UFRONT: |
| 438 |
> |
if ((outVec[2] < 0) | (inVec[2] < 0)) |
| 439 |
> |
return -1.; |
| 440 |
> |
break; |
| 441 |
> |
case SD_UBACK: |
| 442 |
> |
if ((outVec[2] > 0) | (inVec[2] > 0)) |
| 443 |
> |
return -1.; |
| 444 |
> |
break; |
| 445 |
> |
case SD_XMIT: |
| 446 |
> |
if ((outVec[2] > 0) == (inVec[2] > 0)) |
| 447 |
> |
return -1.; |
| 448 |
> |
break; |
| 449 |
> |
default: |
| 450 |
> |
return -1.; |
| 451 |
> |
} |
| 452 |
> |
/* convert vector coordinates */ |
| 453 |
> |
if (sdt->st->ndim == 3) { |
| 454 |
> |
spinvector(rOutVec, outVec, zvec, -atan2(-inVec[1],-inVec[0])); |
| 455 |
> |
gridPos[0] = .5 - .5*sqrt(inVec[0]*inVec[0] + inVec[1]*inVec[1]); |
| 456 |
> |
SDdisk2square(gridPos+1, rOutVec[0], rOutVec[1]); |
| 457 |
> |
} else if (sdt->st->ndim == 4) { |
| 458 |
> |
SDdisk2square(gridPos, -inVec[0], -inVec[1]); |
| 459 |
> |
SDdisk2square(gridPos+2, outVec[0], outVec[1]); |
| 460 |
> |
} else |
| 461 |
> |
return -1.; /* should be internal error */ |
| 462 |
> |
|
| 463 |
> |
return SDlookupTre(sdt->st, gridPos, hc); |
| 464 |
> |
} |
| 465 |
> |
|
| 466 |
> |
/* Compute non-diffuse component for variable-resolution BSDF */ |
| 467 |
> |
static int |
| 468 |
> |
SDgetTreBSDF(float coef[SDmaxCh], const FVECT outVec, |
| 469 |
> |
const FVECT inVec, SDComponent *sdc) |
| 470 |
> |
{ |
| 471 |
> |
/* check arguments */ |
| 472 |
> |
if ((coef == NULL) | (outVec == NULL) | (inVec == NULL) | (sdc == NULL) |
| 473 |
> |
|| sdc->dist == NULL) |
| 474 |
> |
return 0; |
| 475 |
> |
/* get nearest BSDF value */ |
| 476 |
> |
coef[0] = SDqueryTre((SDTre *)sdc->dist, outVec, inVec, NULL); |
| 477 |
> |
return (coef[0] >= 0); /* monochromatic for now */ |
| 478 |
> |
} |
| 479 |
> |
|
| 480 |
> |
/* Callback to build cumulative distribution using SDtraverseTre() */ |
| 481 |
> |
static int |
| 482 |
> |
build_scaffold(float val, const double *cmin, double csiz, void *cptr) |
| 483 |
> |
{ |
| 484 |
> |
SDdistScaffold *sp = (SDdistScaffold *)cptr; |
| 485 |
> |
int wid = csiz*(double)iwmax + .5; |
| 486 |
> |
bitmask_t bmin[2], bmax[2]; |
| 487 |
> |
|
| 488 |
> |
cmin += sp->nic; /* skip to output coords */ |
| 489 |
> |
if (wid < sp->wmin) /* new minimum width? */ |
| 490 |
> |
sp->wmin = wid; |
| 491 |
> |
if (wid > sp->wmax) /* new maximum? */ |
| 492 |
> |
sp->wmax = wid; |
| 493 |
> |
if (sp->alen >= sp->nall) { /* need more space? */ |
| 494 |
> |
struct outdir_s *ndarr; |
| 495 |
> |
sp->nall += 1024; |
| 496 |
> |
ndarr = (struct outdir_s *)realloc(sp->darr, |
| 497 |
> |
sizeof(struct outdir_s)*sp->nall); |
| 498 |
> |
if (ndarr == NULL) { |
| 499 |
> |
sprintf(SDerrorDetail, |
| 500 |
> |
"Cannot grow scaffold to %u entries", sp->nall); |
| 501 |
> |
return -1; /* abort build */ |
| 502 |
> |
} |
| 503 |
> |
sp->darr = ndarr; |
| 504 |
> |
} |
| 505 |
> |
/* find Hilbert entry index */ |
| 506 |
> |
bmin[0] = cmin[0]*(double)iwmax + .5; |
| 507 |
> |
bmin[1] = cmin[1]*(double)iwmax + .5; |
| 508 |
> |
bmax[0] = bmin[0] + wid-1; |
| 509 |
> |
bmax[1] = bmin[1] + wid-1; |
| 510 |
> |
hilbert_box_vtx(2, sizeof(bitmask_t), iwbits, 1, bmin, bmax); |
| 511 |
> |
sp->darr[sp->alen].hent = hilbert_c2i(2, iwbits, bmin); |
| 512 |
> |
sp->darr[sp->alen].wid = wid; |
| 513 |
> |
sp->darr[sp->alen].bsdf = val; |
| 514 |
> |
sp->alen++; /* on to the next entry */ |
| 515 |
> |
return 0; |
| 516 |
> |
} |
| 517 |
> |
|
| 518 |
> |
/* Scaffold comparison function for qsort -- ascending Hilbert index */ |
| 519 |
> |
static int |
| 520 |
> |
sscmp(const void *p1, const void *p2) |
| 521 |
> |
{ |
| 522 |
> |
unsigned h1 = (*(const struct outdir_s *)p1).hent; |
| 523 |
> |
unsigned h2 = (*(const struct outdir_s *)p2).hent; |
| 524 |
> |
|
| 525 |
> |
if (h1 > h2) |
| 526 |
> |
return 1; |
| 527 |
> |
if (h1 < h2) |
| 528 |
> |
return -1; |
| 529 |
> |
return 0; |
| 530 |
> |
} |
| 531 |
> |
|
| 532 |
> |
/* Create a new cumulative distribution for the given input direction */ |
| 533 |
> |
static SDTreCDst * |
| 534 |
> |
make_cdist(const SDTre *sdt, const double *pos) |
| 535 |
> |
{ |
| 536 |
> |
SDdistScaffold myScaffold; |
| 537 |
> |
SDTreCDst *cd; |
| 538 |
> |
struct outdir_s *sp; |
| 539 |
> |
double scale, cursum; |
| 540 |
> |
int i; |
| 541 |
> |
/* initialize scaffold */ |
| 542 |
> |
myScaffold.wmin = iwmax; |
| 543 |
> |
myScaffold.wmax = 0; |
| 544 |
> |
myScaffold.nic = sdt->st->ndim - 2; |
| 545 |
> |
myScaffold.alen = 0; |
| 546 |
> |
myScaffold.nall = 512; |
| 547 |
> |
myScaffold.darr = (struct outdir_s *)malloc(sizeof(struct outdir_s) * |
| 548 |
> |
myScaffold.nall); |
| 549 |
> |
if (myScaffold.darr == NULL) |
| 550 |
> |
return NULL; |
| 551 |
> |
/* grow the distribution */ |
| 552 |
> |
if (SDtraverseTre(sdt->st, pos, (1<<myScaffold.nic)-1, |
| 553 |
> |
&build_scaffold, &myScaffold) < 0) { |
| 554 |
> |
free(myScaffold.darr); |
| 555 |
> |
return NULL; |
| 556 |
> |
} |
| 557 |
> |
/* allocate result holder */ |
| 558 |
> |
cd = (SDTreCDst *)malloc(sizeof(SDTreCDst) + |
| 559 |
> |
sizeof(cd->carr[0])*myScaffold.alen); |
| 560 |
> |
if (cd == NULL) { |
| 561 |
> |
sprintf(SDerrorDetail, |
| 562 |
> |
"Cannot allocate %u entry cumulative distribution", |
| 563 |
> |
myScaffold.alen); |
| 564 |
> |
free(myScaffold.darr); |
| 565 |
> |
return NULL; |
| 566 |
> |
} |
| 567 |
> |
cd->isodist = (myScaffold.nic == 1); |
| 568 |
> |
/* sort the distribution */ |
| 569 |
> |
qsort(myScaffold.darr, cd->calen = myScaffold.alen, |
| 570 |
> |
sizeof(struct outdir_s), &sscmp); |
| 571 |
> |
|
| 572 |
> |
/* record input range */ |
| 573 |
> |
scale = myScaffold.wmin / (double)iwmax; |
| 574 |
> |
for (i = myScaffold.nic; i--; ) { |
| 575 |
> |
cd->clim[i][0] = floor(pos[i]/scale) * scale; |
| 576 |
> |
cd->clim[i][1] = cd->clim[i][0] + scale; |
| 577 |
> |
} |
| 578 |
> |
if (cd->isodist) { /* avoid issue in SDqueryTreProjSA() */ |
| 579 |
> |
cd->clim[1][0] = cd->clim[0][0]; |
| 580 |
> |
cd->clim[1][1] = cd->clim[0][1]; |
| 581 |
> |
} |
| 582 |
> |
cd->max_psa = myScaffold.wmax / (double)iwmax; |
| 583 |
> |
cd->max_psa *= cd->max_psa * M_PI; |
| 584 |
> |
cd->sidef = sdt->sidef; |
| 585 |
> |
cd->cTotal = 1e-20; /* compute directional total */ |
| 586 |
> |
sp = myScaffold.darr; |
| 587 |
> |
for (i = myScaffold.alen; i--; sp++) |
| 588 |
> |
cd->cTotal += sp->bsdf * (double)sp->wid * sp->wid; |
| 589 |
> |
cursum = .0; /* go back and get cumulative values */ |
| 590 |
> |
scale = (double)cumlmax / cd->cTotal; |
| 591 |
> |
sp = myScaffold.darr; |
| 592 |
> |
for (i = 0; i < cd->calen; i++, sp++) { |
| 593 |
> |
cd->carr[i].hndx = sp->hent; |
| 594 |
> |
cd->carr[i].cuml = scale*cursum + .5; |
| 595 |
> |
cursum += sp->bsdf * (double)sp->wid * sp->wid; |
| 596 |
> |
} |
| 597 |
> |
cd->carr[i].hndx = ~0; /* make final entry */ |
| 598 |
> |
cd->carr[i].cuml = cumlmax; |
| 599 |
> |
cd->cTotal *= M_PI/(double)iwmax/iwmax; |
| 600 |
> |
/* all done, clean up and return */ |
| 601 |
> |
free(myScaffold.darr); |
| 602 |
> |
return cd; |
| 603 |
> |
} |
| 604 |
> |
|
| 605 |
> |
/* Find or allocate a cumulative distribution for the given incoming vector */ |
| 606 |
> |
const SDCDst * |
| 607 |
> |
SDgetTreCDist(const FVECT inVec, SDComponent *sdc) |
| 608 |
> |
{ |
| 609 |
> |
const SDTre *sdt; |
| 610 |
> |
double inCoord[2]; |
| 611 |
> |
int vflags; |
| 612 |
> |
int i; |
| 613 |
> |
SDTreCDst *cd, *cdlast; |
| 614 |
> |
/* check arguments */ |
| 615 |
> |
if ((inVec == NULL) | (sdc == NULL) || |
| 616 |
> |
(sdt = (SDTre *)sdc->dist) == NULL) |
| 617 |
> |
return NULL; |
| 618 |
> |
if (sdt->st->ndim == 3) /* isotropic BSDF? */ |
| 619 |
> |
inCoord[0] = .5 - .5*sqrt(inVec[0]*inVec[0] + inVec[1]*inVec[1]); |
| 620 |
> |
else if (sdt->st->ndim == 4) |
| 621 |
> |
SDdisk2square(inCoord, -inVec[0], -inVec[1]); |
| 622 |
> |
else |
| 623 |
> |
return NULL; /* should be internal error */ |
| 624 |
> |
cdlast = NULL; /* check for direction in cache list */ |
| 625 |
> |
for (cd = (SDTreCDst *)sdc->cdList; cd != NULL; |
| 626 |
> |
cdlast = cd, cd = (SDTreCDst *)cd->next) { |
| 627 |
> |
for (i = sdt->st->ndim - 2; i--; ) |
| 628 |
> |
if ((cd->clim[i][0] > inCoord[i]) | |
| 629 |
> |
(inCoord[i] >= cd->clim[i][1])) |
| 630 |
> |
break; |
| 631 |
> |
if (i < 0) |
| 632 |
> |
break; /* means we have a match */ |
| 633 |
> |
} |
| 634 |
> |
if (cd == NULL) /* need to create new entry? */ |
| 635 |
> |
cdlast = cd = make_cdist(sdt, inCoord); |
| 636 |
> |
if (cdlast != NULL) { /* move entry to head of cache list */ |
| 637 |
> |
cdlast->next = cd->next; |
| 638 |
> |
cd->next = sdc->cdList; |
| 639 |
> |
sdc->cdList = (SDCDst *)cd; |
| 640 |
> |
} |
| 641 |
> |
return (SDCDst *)cd; /* ready to go */ |
| 642 |
> |
} |
| 643 |
> |
|
| 644 |
> |
/* Query solid angle for vector(s) */ |
| 645 |
> |
static SDError |
| 646 |
> |
SDqueryTreProjSA(double *psa, const FVECT v1, const RREAL *v2, |
| 647 |
> |
int qflags, SDComponent *sdc) |
| 648 |
> |
{ |
| 649 |
> |
double myPSA[2]; |
| 650 |
> |
/* check arguments */ |
| 651 |
> |
if ((psa == NULL) | (v1 == NULL) | (sdc == NULL) || |
| 652 |
> |
sdc->dist == NULL) |
| 653 |
> |
return SDEargument; |
| 654 |
> |
/* get projected solid angle(s) */ |
| 655 |
> |
if (v2 != NULL) { |
| 656 |
> |
const SDTre *sdt = (SDTre *)sdc->dist; |
| 657 |
> |
double hcube[SD_MAXDIM]; |
| 658 |
> |
if (SDqueryTre(sdt, v1, v2, hcube) < 0) { |
| 659 |
> |
strcpy(SDerrorDetail, "Bad call to SDqueryTreProjSA"); |
| 660 |
> |
return SDEinternal; |
| 661 |
> |
} |
| 662 |
> |
myPSA[0] = hcube[sdt->st->ndim]; |
| 663 |
> |
myPSA[1] = myPSA[0] *= myPSA[0] * M_PI; |
| 664 |
> |
} else { |
| 665 |
> |
const SDTreCDst *cd = (const SDTreCDst *)SDgetTreCDist(v1, sdc); |
| 666 |
> |
if (cd == NULL) |
| 667 |
> |
return SDEmemory; |
| 668 |
> |
myPSA[0] = M_PI * (cd->clim[0][1] - cd->clim[0][0]) * |
| 669 |
> |
(cd->clim[1][1] - cd->clim[1][0]); |
| 670 |
> |
myPSA[1] = cd->max_psa; |
| 671 |
> |
} |
| 672 |
> |
switch (qflags) { /* record based on flag settings */ |
| 673 |
> |
case SDqueryVal: |
| 674 |
> |
*psa = myPSA[0]; |
| 675 |
> |
break; |
| 676 |
> |
case SDqueryMax: |
| 677 |
> |
if (myPSA[1] > *psa) |
| 678 |
> |
*psa = myPSA[1]; |
| 679 |
> |
break; |
| 680 |
> |
case SDqueryMin+SDqueryMax: |
| 681 |
> |
if (myPSA[1] > psa[1]) |
| 682 |
> |
psa[1] = myPSA[1]; |
| 683 |
> |
/* fall through */ |
| 684 |
> |
case SDqueryMin: |
| 685 |
> |
if (myPSA[0] < psa[0]) |
| 686 |
> |
psa[0] = myPSA[0]; |
| 687 |
> |
break; |
| 688 |
> |
} |
| 689 |
> |
return SDEnone; |
| 690 |
> |
} |
| 691 |
> |
|
| 692 |
> |
/* Sample cumulative distribution */ |
| 693 |
> |
static SDError |
| 694 |
> |
SDsampTreCDist(FVECT ioVec, double randX, const SDCDst *cdp) |
| 695 |
> |
{ |
| 696 |
> |
const unsigned nBitsC = 4*sizeof(bitmask_t); |
| 697 |
> |
const unsigned nExtraBits = 8*(sizeof(bitmask_t)-sizeof(unsigned)); |
| 698 |
> |
const SDTreCDst *cd = (const SDTreCDst *)cdp; |
| 699 |
> |
const unsigned target = randX*cumlmax; |
| 700 |
> |
bitmask_t hndx, hcoord[2]; |
| 701 |
> |
double gpos[3], rotangle; |
| 702 |
> |
int i, iupper, ilower; |
| 703 |
> |
/* check arguments */ |
| 704 |
> |
if ((ioVec == NULL) | (cd == NULL)) |
| 705 |
> |
return SDEargument; |
| 706 |
> |
if (ioVec[2] > 0) { |
| 707 |
> |
if (!(cd->sidef & SD_UFRONT)) |
| 708 |
> |
return SDEargument; |
| 709 |
> |
} else if (!(cd->sidef & SD_UBACK)) |
| 710 |
> |
return SDEargument; |
| 711 |
> |
/* binary search to find position */ |
| 712 |
> |
ilower = 0; iupper = cd->calen; |
| 713 |
> |
while ((i = (iupper + ilower) >> 1) != ilower) |
| 714 |
> |
if ((long)target >= (long)cd->carr[i].cuml) |
| 715 |
> |
ilower = i; |
| 716 |
> |
else |
| 717 |
> |
iupper = i; |
| 718 |
> |
/* localize random position */ |
| 719 |
> |
randX = (randX*cumlmax - cd->carr[ilower].cuml) / |
| 720 |
> |
(double)(cd->carr[iupper].cuml - cd->carr[ilower].cuml); |
| 721 |
> |
/* index in longer Hilbert curve */ |
| 722 |
> |
hndx = (randX*cd->carr[iupper].hndx + (1.-randX)*cd->carr[ilower].hndx) |
| 723 |
> |
* (double)((bitmask_t)1 << nExtraBits); |
| 724 |
> |
/* convert Hilbert index to vector */ |
| 725 |
> |
hilbert_i2c(2, nBitsC, hndx, hcoord); |
| 726 |
> |
for (i = 2; i--; ) |
| 727 |
> |
gpos[i] = ((double)hcoord[i] + rand()*(1./(RAND_MAX+.5))) / |
| 728 |
> |
(double)((bitmask_t)1 << nBitsC); |
| 729 |
> |
SDsquare2disk(gpos, gpos[0], gpos[1]); |
| 730 |
> |
/* compute Z-coordinate */ |
| 731 |
> |
gpos[2] = 1. - gpos[0]*gpos[0] - gpos[1]*gpos[1]; |
| 732 |
> |
if (gpos[2] > 0) /* paranoia, I hope */ |
| 733 |
> |
gpos[2] = sqrt(gpos[2]); |
| 734 |
> |
/* emit from back? */ |
| 735 |
> |
if (ioVec[2] > 0 ^ cd->sidef != SD_XMIT) |
| 736 |
> |
gpos[2] = -gpos[2]; |
| 737 |
> |
if (cd->isodist) { /* rotate isotropic result */ |
| 738 |
> |
rotangle = atan2(-ioVec[1],-ioVec[0]); |
| 739 |
> |
VCOPY(ioVec, gpos); |
| 740 |
> |
spinvector(ioVec, ioVec, zvec, rotangle); |
| 741 |
> |
} else |
| 742 |
> |
VCOPY(ioVec, gpos); |
| 743 |
> |
return SDEnone; |
| 744 |
> |
} |
| 745 |
> |
|
| 746 |
> |
/* Advance pointer to the next non-white character in the string (or nul) */ |
| 747 |
> |
static int |
| 748 |
> |
next_token(char **spp) |
| 749 |
> |
{ |
| 750 |
> |
while (isspace(**spp)) |
| 751 |
> |
++*spp; |
| 752 |
> |
return **spp; |
| 753 |
> |
} |
| 754 |
> |
|
| 755 |
> |
/* Advance pointer past matching token (or any token if c==0) */ |
| 756 |
> |
#define eat_token(spp,c) (next_token(spp)==(c) ^ !(c) ? *(*(spp))++ : 0) |
| 757 |
> |
|
| 758 |
> |
/* Count words from this point in string to '}' */ |
| 759 |
> |
static int |
| 760 |
> |
count_values(char *cp) |
| 761 |
> |
{ |
| 762 |
> |
int n = 0; |
| 763 |
> |
|
| 764 |
> |
while (next_token(&cp) != '}' && *cp) { |
| 765 |
> |
while (!isspace(*cp) & (*cp != ',') & (*cp != '}')) |
| 766 |
> |
if (!*++cp) |
| 767 |
> |
break; |
| 768 |
> |
++n; |
| 769 |
> |
eat_token(&cp, ','); |
| 770 |
> |
} |
| 771 |
> |
return n; |
| 772 |
> |
} |
| 773 |
> |
|
| 774 |
> |
/* Load an array of real numbers, returning total */ |
| 775 |
> |
static int |
| 776 |
> |
load_values(char **spp, float *va, int n) |
| 777 |
> |
{ |
| 778 |
> |
float *v = va; |
| 779 |
> |
char *svnext; |
| 780 |
> |
|
| 781 |
> |
while (n-- > 0 && (svnext = fskip(*spp)) != NULL) { |
| 782 |
> |
*v++ = atof(*spp); |
| 783 |
> |
*spp = svnext; |
| 784 |
> |
eat_token(spp, ','); |
| 785 |
> |
} |
| 786 |
> |
return v - va; |
| 787 |
> |
} |
| 788 |
> |
|
| 789 |
> |
/* Load BSDF tree data */ |
| 790 |
> |
static SDNode * |
| 791 |
> |
load_tree_data(char **spp, int nd) |
| 792 |
> |
{ |
| 793 |
> |
SDNode *st; |
| 794 |
> |
int n; |
| 795 |
> |
|
| 796 |
> |
if (!eat_token(spp, '{')) { |
| 797 |
> |
strcpy(SDerrorDetail, "Missing '{' in tensor tree"); |
| 798 |
> |
return NULL; |
| 799 |
> |
} |
| 800 |
> |
if (next_token(spp) == '{') { /* tree branches */ |
| 801 |
> |
st = SDnewNode(nd, -1); |
| 802 |
> |
if (st == NULL) |
| 803 |
> |
return NULL; |
| 804 |
> |
for (n = 0; n < 1<<nd; n++) |
| 805 |
> |
if ((st->u.t[n] = load_tree_data(spp, nd)) == NULL) { |
| 806 |
> |
SDfreeTre(st); |
| 807 |
> |
return NULL; |
| 808 |
> |
} |
| 809 |
> |
} else { /* else load value grid */ |
| 810 |
> |
int bsiz; |
| 811 |
> |
n = count_values(*spp); /* see how big the grid is */ |
| 812 |
> |
for (bsiz = 0; bsiz < 8*sizeof(size_t); bsiz += nd) |
| 813 |
> |
if (1<<bsiz == n) |
| 814 |
> |
break; |
| 815 |
> |
if (bsiz >= 8*sizeof(size_t)) { |
| 816 |
> |
strcpy(SDerrorDetail, "Illegal value count in tensor tree"); |
| 817 |
> |
return NULL; |
| 818 |
> |
} |
| 819 |
> |
st = SDnewNode(nd, bsiz/nd); |
| 820 |
> |
if (st == NULL) |
| 821 |
> |
return NULL; |
| 822 |
> |
if (load_values(spp, st->u.v, n) != n) { |
| 823 |
> |
strcpy(SDerrorDetail, "Real format error in tensor tree"); |
| 824 |
> |
SDfreeTre(st); |
| 825 |
> |
return NULL; |
| 826 |
> |
} |
| 827 |
> |
} |
| 828 |
> |
if (!eat_token(spp, '}')) { |
| 829 |
> |
strcpy(SDerrorDetail, "Missing '}' in tensor tree"); |
| 830 |
> |
SDfreeTre(st); |
| 831 |
> |
return NULL; |
| 832 |
> |
} |
| 833 |
> |
eat_token(spp, ','); |
| 834 |
> |
return st; |
| 835 |
> |
} |
| 836 |
> |
|
| 837 |
> |
/* Compute min. proj. solid angle and max. direct hemispherical scattering */ |
| 838 |
> |
static SDError |
| 839 |
> |
get_extrema(SDSpectralDF *df) |
| 840 |
> |
{ |
| 841 |
> |
SDNode *st = (*(SDTre *)df->comp[0].dist).st; |
| 842 |
> |
double stepWidth, dhemi, bmin[4], bmax[4]; |
| 843 |
> |
|
| 844 |
> |
stepWidth = SDsmallestLeaf(st); |
| 845 |
> |
df->minProjSA = M_PI*stepWidth*stepWidth; |
| 846 |
> |
if (stepWidth < .03125) |
| 847 |
> |
stepWidth = .03125; /* 1/32 resolution good enough */ |
| 848 |
> |
df->maxHemi = .0; |
| 849 |
> |
if (st->ndim == 3) { /* isotropic BSDF */ |
| 850 |
> |
bmin[1] = bmin[2] = .0; |
| 851 |
> |
bmax[1] = bmax[2] = 1.; |
| 852 |
> |
for (bmin[0] = .0; bmin[0] < .5-FTINY; bmin[0] += stepWidth) { |
| 853 |
> |
bmax[0] = bmin[0] + stepWidth; |
| 854 |
> |
dhemi = SDavgTreBox(st, bmin, bmax); |
| 855 |
> |
if (dhemi > df->maxHemi) |
| 856 |
> |
df->maxHemi = dhemi; |
| 857 |
> |
} |
| 858 |
> |
} else if (st->ndim == 4) { /* anisotropic BSDF */ |
| 859 |
> |
bmin[2] = bmin[3] = .0; |
| 860 |
> |
bmax[2] = bmax[3] = 1.; |
| 861 |
> |
for (bmin[0] = .0; bmin[0] < 1.-FTINY; bmin[0] += stepWidth) { |
| 862 |
> |
bmax[0] = bmin[0] + stepWidth; |
| 863 |
> |
for (bmin[1] = .0; bmin[1] < 1.-FTINY; bmin[1] += stepWidth) { |
| 864 |
> |
bmax[1] = bmin[1] + stepWidth; |
| 865 |
> |
dhemi = SDavgTreBox(st, bmin, bmax); |
| 866 |
> |
if (dhemi > df->maxHemi) |
| 867 |
> |
df->maxHemi = dhemi; |
| 868 |
> |
} |
| 869 |
> |
} |
| 870 |
> |
} else |
| 871 |
> |
return SDEinternal; |
| 872 |
> |
/* correct hemispherical value */ |
| 873 |
> |
df->maxHemi *= M_PI; |
| 874 |
> |
return SDEnone; |
| 875 |
> |
} |
| 876 |
> |
|
| 877 |
> |
/* Load BSDF distribution for this wavelength */ |
| 878 |
> |
static SDError |
| 879 |
> |
load_bsdf_data(SDData *sd, ezxml_t wdb, int ndim) |
| 880 |
> |
{ |
| 881 |
> |
SDSpectralDF *df; |
| 882 |
> |
SDTre *sdt; |
| 883 |
> |
char *sdata; |
| 884 |
> |
int i; |
| 885 |
> |
/* allocate BSDF component */ |
| 886 |
> |
sdata = ezxml_txt(ezxml_child(wdb, "WavelengthDataDirection")); |
| 887 |
> |
if (!sdata) |
| 888 |
> |
return SDEnone; |
| 889 |
> |
/* |
| 890 |
> |
* Remember that front and back are reversed from WINDOW 6 orientations |
| 891 |
> |
*/ |
| 892 |
> |
if (!strcasecmp(sdata, "Transmission")) { |
| 893 |
> |
if (sd->tf != NULL) |
| 894 |
> |
SDfreeSpectralDF(sd->tf); |
| 895 |
> |
if ((sd->tf = SDnewSpectralDF(1)) == NULL) |
| 896 |
> |
return SDEmemory; |
| 897 |
> |
df = sd->tf; |
| 898 |
> |
} else if (!strcasecmp(sdata, "Reflection Front")) { |
| 899 |
> |
if (sd->rb != NULL) /* note back-front reversal */ |
| 900 |
> |
SDfreeSpectralDF(sd->rb); |
| 901 |
> |
if ((sd->rb = SDnewSpectralDF(1)) == NULL) |
| 902 |
> |
return SDEmemory; |
| 903 |
> |
df = sd->rb; |
| 904 |
> |
} else if (!strcasecmp(sdata, "Reflection Back")) { |
| 905 |
> |
if (sd->rf != NULL) /* note front-back reversal */ |
| 906 |
> |
SDfreeSpectralDF(sd->rf); |
| 907 |
> |
if ((sd->rf = SDnewSpectralDF(1)) == NULL) |
| 908 |
> |
return SDEmemory; |
| 909 |
> |
df = sd->rf; |
| 910 |
> |
} else |
| 911 |
> |
return SDEnone; |
| 912 |
> |
/* XXX should also check "ScatteringDataType" for consistency? */ |
| 913 |
> |
/* get angle bases */ |
| 914 |
> |
sdata = ezxml_txt(ezxml_child(wdb,"AngleBasis")); |
| 915 |
> |
if (!sdata || strcasecmp(sdata, "LBNL/Shirley-Chiu")) { |
| 916 |
> |
sprintf(SDerrorDetail, "%s angle basis for BSDF '%s'", |
| 917 |
> |
!sdata ? "Missing" : "Unsupported", sd->name); |
| 918 |
> |
return !sdata ? SDEformat : SDEsupport; |
| 919 |
> |
} |
| 920 |
> |
/* allocate BSDF tree */ |
| 921 |
> |
sdt = (SDTre *)malloc(sizeof(SDTre)); |
| 922 |
> |
if (sdt == NULL) |
| 923 |
> |
return SDEmemory; |
| 924 |
> |
if (df == sd->rf) |
| 925 |
> |
sdt->sidef = SD_UFRONT; |
| 926 |
> |
else if (df == sd->rb) |
| 927 |
> |
sdt->sidef = SD_UBACK; |
| 928 |
> |
else |
| 929 |
> |
sdt->sidef = SD_XMIT; |
| 930 |
> |
sdt->st = NULL; |
| 931 |
> |
df->comp[0].cspec[0] = c_dfcolor; /* XXX monochrome for now */ |
| 932 |
> |
df->comp[0].dist = sdt; |
| 933 |
> |
df->comp[0].func = &SDhandleTre; |
| 934 |
> |
/* read BSDF data */ |
| 935 |
> |
sdata = ezxml_txt(ezxml_child(wdb, "ScatteringData")); |
| 936 |
> |
if (!sdata || !next_token(&sdata)) { |
| 937 |
> |
sprintf(SDerrorDetail, "Missing BSDF ScatteringData in '%s'", |
| 938 |
> |
sd->name); |
| 939 |
> |
return SDEformat; |
| 940 |
> |
} |
| 941 |
> |
sdt->st = load_tree_data(&sdata, ndim); |
| 942 |
> |
if (sdt->st == NULL) |
| 943 |
> |
return SDEformat; |
| 944 |
> |
if (next_token(&sdata)) { /* check for unconsumed characters */ |
| 945 |
> |
sprintf(SDerrorDetail, |
| 946 |
> |
"Extra characters at end of ScatteringData in '%s'", |
| 947 |
> |
sd->name); |
| 948 |
> |
return SDEformat; |
| 949 |
> |
} |
| 950 |
> |
/* flatten branches where possible */ |
| 951 |
> |
sdt->st = SDsimplifyTre(sdt->st); |
| 952 |
> |
if (sdt->st == NULL) |
| 953 |
> |
return SDEinternal; |
| 954 |
> |
return get_extrema(df); /* compute global quantities */ |
| 955 |
> |
} |
| 956 |
> |
|
| 957 |
> |
/* Find minimum value in tree */ |
| 958 |
> |
static float |
| 959 |
> |
SDgetTreMin(const SDNode *st) |
| 960 |
> |
{ |
| 961 |
> |
float vmin = FHUGE; |
| 962 |
> |
int n; |
| 963 |
> |
|
| 964 |
> |
if (st->log2GR < 0) { |
| 965 |
> |
for (n = 1<<st->ndim; n--; ) { |
| 966 |
> |
float v = SDgetTreMin(st->u.t[n]); |
| 967 |
> |
if (v < vmin) |
| 968 |
> |
vmin = v; |
| 969 |
> |
} |
| 970 |
> |
} else { |
| 971 |
> |
for (n = 1<<(st->ndim*st->log2GR); n--; ) |
| 972 |
> |
if (st->u.v[n] < vmin) |
| 973 |
> |
vmin = st->u.v[n]; |
| 974 |
> |
} |
| 975 |
> |
return vmin; |
| 976 |
> |
} |
| 977 |
> |
|
| 978 |
> |
/* Subtract the given value from all tree nodes */ |
| 979 |
> |
static void |
| 980 |
> |
SDsubtractTreVal(SDNode *st, float val) |
| 981 |
> |
{ |
| 982 |
> |
int n; |
| 983 |
> |
|
| 984 |
> |
if (st->log2GR < 0) { |
| 985 |
> |
for (n = 1<<st->ndim; n--; ) |
| 986 |
> |
SDsubtractTreVal(st->u.t[n], val); |
| 987 |
> |
} else { |
| 988 |
> |
for (n = 1<<(st->ndim*st->log2GR); n--; ) |
| 989 |
> |
if ((st->u.v[n] -= val) < 0) |
| 990 |
> |
st->u.v[n] = .0f; |
| 991 |
> |
} |
| 992 |
> |
} |
| 993 |
> |
|
| 994 |
> |
/* Subtract minimum value from BSDF */ |
| 995 |
> |
static double |
| 996 |
> |
subtract_min(SDNode *st) |
| 997 |
> |
{ |
| 998 |
> |
float vmin; |
| 999 |
> |
/* be sure to skip unused portion */ |
| 1000 |
> |
if (st->ndim == 3) { |
| 1001 |
> |
int n; |
| 1002 |
> |
vmin = 1./M_PI; |
| 1003 |
> |
if (st->log2GR < 0) { |
| 1004 |
> |
for (n = 0; n < 8; n += 2) { |
| 1005 |
> |
float v = SDgetTreMin(st->u.t[n]); |
| 1006 |
> |
if (v < vmin) |
| 1007 |
> |
vmin = v; |
| 1008 |
> |
} |
| 1009 |
> |
} else if (st->log2GR) { |
| 1010 |
> |
for (n = 1 << (3*st->log2GR - 1); n--; ) |
| 1011 |
> |
if (st->u.v[n] < vmin) |
| 1012 |
> |
vmin = st->u.v[n]; |
| 1013 |
> |
} else |
| 1014 |
> |
vmin = st->u.v[0]; |
| 1015 |
> |
} else /* anisotropic covers entire tree */ |
| 1016 |
> |
vmin = SDgetTreMin(st); |
| 1017 |
> |
|
| 1018 |
> |
if (vmin <= FTINY) |
| 1019 |
|
return .0; |
| 1020 |
< |
|
| 1021 |
< |
return SDiterSum(st->u.v, st->ndim, 1 << st->log2GR, imin, imax) / |
| 1022 |
< |
(double)n; |
| 1020 |
> |
|
| 1021 |
> |
SDsubtractTreVal(st, vmin); |
| 1022 |
> |
|
| 1023 |
> |
return M_PI * vmin; /* return hemispherical value */ |
| 1024 |
|
} |
| 1025 |
|
|
| 1026 |
+ |
/* Extract and separate diffuse portion of BSDF */ |
| 1027 |
+ |
static void |
| 1028 |
+ |
extract_diffuse(SDValue *dv, SDSpectralDF *df) |
| 1029 |
+ |
{ |
| 1030 |
+ |
int n; |
| 1031 |
+ |
|
| 1032 |
+ |
if (df == NULL || df->ncomp <= 0) { |
| 1033 |
+ |
dv->spec = c_dfcolor; |
| 1034 |
+ |
dv->cieY = .0; |
| 1035 |
+ |
return; |
| 1036 |
+ |
} |
| 1037 |
+ |
dv->spec = df->comp[0].cspec[0]; |
| 1038 |
+ |
dv->cieY = subtract_min((*(SDTre *)df->comp[0].dist).st); |
| 1039 |
+ |
/* in case of multiple components */ |
| 1040 |
+ |
for (n = df->ncomp; --n; ) { |
| 1041 |
+ |
double ymin = subtract_min((*(SDTre *)df->comp[n].dist).st); |
| 1042 |
+ |
c_cmix(&dv->spec, dv->cieY, &dv->spec, ymin, &df->comp[n].cspec[0]); |
| 1043 |
+ |
dv->cieY += ymin; |
| 1044 |
+ |
} |
| 1045 |
+ |
df->maxHemi -= dv->cieY; /* adjust maximum hemispherical */ |
| 1046 |
+ |
/* make sure everything is set */ |
| 1047 |
+ |
c_ccvt(&dv->spec, C_CSXY+C_CSSPEC); |
| 1048 |
+ |
} |
| 1049 |
+ |
|
| 1050 |
|
/* Load a variable-resolution BSDF tree from an open XML file */ |
| 1051 |
|
SDError |
| 1052 |
|
SDloadTre(SDData *sd, ezxml_t wtl) |
| 1053 |
|
{ |
| 1054 |
< |
return SDEsupport; |
| 1054 |
> |
SDError ec; |
| 1055 |
> |
ezxml_t wld, wdb; |
| 1056 |
> |
int rank; |
| 1057 |
> |
char *txt; |
| 1058 |
> |
/* basic checks and tensor rank */ |
| 1059 |
> |
txt = ezxml_txt(ezxml_child(ezxml_child(wtl, |
| 1060 |
> |
"DataDefinition"), "IncidentDataStructure")); |
| 1061 |
> |
if (txt == NULL || !*txt) { |
| 1062 |
> |
sprintf(SDerrorDetail, |
| 1063 |
> |
"BSDF \"%s\": missing IncidentDataStructure", |
| 1064 |
> |
sd->name); |
| 1065 |
> |
return SDEformat; |
| 1066 |
> |
} |
| 1067 |
> |
if (!strcasecmp(txt, "TensorTree3")) |
| 1068 |
> |
rank = 3; |
| 1069 |
> |
else if (!strcasecmp(txt, "TensorTree4")) |
| 1070 |
> |
rank = 4; |
| 1071 |
> |
else { |
| 1072 |
> |
sprintf(SDerrorDetail, |
| 1073 |
> |
"BSDF \"%s\": unsupported IncidentDataStructure", |
| 1074 |
> |
sd->name); |
| 1075 |
> |
return SDEsupport; |
| 1076 |
> |
} |
| 1077 |
> |
/* load BSDF components */ |
| 1078 |
> |
for (wld = ezxml_child(wtl, "WavelengthData"); |
| 1079 |
> |
wld != NULL; wld = wld->next) { |
| 1080 |
> |
if (strcasecmp(ezxml_txt(ezxml_child(wld,"Wavelength")), |
| 1081 |
> |
"Visible")) |
| 1082 |
> |
continue; /* just visible for now */ |
| 1083 |
> |
for (wdb = ezxml_child(wld, "WavelengthDataBlock"); |
| 1084 |
> |
wdb != NULL; wdb = wdb->next) |
| 1085 |
> |
if ((ec = load_bsdf_data(sd, wdb, rank)) != SDEnone) |
| 1086 |
> |
return ec; |
| 1087 |
> |
} |
| 1088 |
> |
/* separate diffuse components */ |
| 1089 |
> |
extract_diffuse(&sd->rLambFront, sd->rf); |
| 1090 |
> |
extract_diffuse(&sd->rLambBack, sd->rb); |
| 1091 |
> |
extract_diffuse(&sd->tLamb, sd->tf); |
| 1092 |
> |
/* return success */ |
| 1093 |
> |
return SDEnone; |
| 1094 |
|
} |
| 1095 |
|
|
| 1096 |
|
/* Variable resolution BSDF methods */ |
| 1097 |
< |
const SDFunc SDhandleTre = { |
| 1098 |
< |
NULL, |
| 1099 |
< |
NULL, |
| 1100 |
< |
NULL, |
| 1101 |
< |
NULL, |
| 1102 |
< |
&SDfreeTree, |
| 1097 |
> |
SDFunc SDhandleTre = { |
| 1098 |
> |
&SDgetTreBSDF, |
| 1099 |
> |
&SDqueryTreProjSA, |
| 1100 |
> |
&SDgetTreCDist, |
| 1101 |
> |
&SDsampTreCDist, |
| 1102 |
> |
&SDFreeBTre, |
| 1103 |
|
}; |
