11 |
|
#include "face.h" |
12 |
|
#include "cone.h" |
13 |
|
#include "source.h" |
14 |
+ |
#include "paths.h" |
15 |
|
|
16 |
+ |
#ifndef NBSDFSAMPS |
17 |
+ |
#define NBSDFSAMPS 256 /* BSDF resampling count */ |
18 |
+ |
#endif |
19 |
|
|
20 |
|
COLORV * distarr = NULL; /* distribution array */ |
21 |
|
int distsiz = 0; |
22 |
+ |
COLORV * direct_discount = NULL; /* amount to take off direct */ |
23 |
|
|
24 |
|
|
25 |
|
void |
46 |
|
} |
47 |
|
|
48 |
|
|
49 |
+ |
static void |
50 |
+ |
new_discount() /* allocate space for direct contrib. record */ |
51 |
+ |
{ |
52 |
+ |
if (distsiz <= 0) |
53 |
+ |
return; |
54 |
+ |
direct_discount = (COLORV *)calloc(distsiz, sizeof(COLOR)); |
55 |
+ |
if (direct_discount == NULL) |
56 |
+ |
error(SYSTEM, "out of memory in new_discount"); |
57 |
+ |
} |
58 |
+ |
|
59 |
+ |
|
60 |
+ |
static void |
61 |
+ |
done_discount() /* clear off direct contrib. record */ |
62 |
+ |
{ |
63 |
+ |
if (direct_discount == NULL) |
64 |
+ |
return; |
65 |
+ |
free((void *)direct_discount); |
66 |
+ |
direct_discount = NULL; |
67 |
+ |
} |
68 |
+ |
|
69 |
+ |
|
70 |
|
int |
71 |
|
process_ray( /* process a ray result or report error */ |
72 |
|
RAY *r, |
84 |
|
multcolor(r->rcol, r->rcoef); /* in case it's a source ray */ |
85 |
|
colp = &distarr[r->rno * 3]; |
86 |
|
addcolor(colp, r->rcol); |
87 |
+ |
if (r->rsrc >= 0 && /* remember source contrib. */ |
88 |
+ |
direct_discount != NULL) { |
89 |
+ |
colp = &direct_discount[r->rno * 3]; |
90 |
+ |
addcolor(colp, r->rcol); |
91 |
+ |
} |
92 |
|
return(1); |
93 |
|
} |
94 |
|
|
158 |
|
d = 5.*FTINY; |
159 |
|
for (i = 3; i--; ) |
160 |
|
sr.rorg[i] += d*nrm[i]; |
161 |
+ |
samplendx++; /* increment sample counter */ |
162 |
|
if (!srcray(&sr, NULL, &si)) |
163 |
|
break; /* end of sources */ |
164 |
|
/* index direction */ |
171 |
|
if (i < 0) |
172 |
|
continue; /* must not be important */ |
173 |
|
sr.rno = i; |
174 |
< |
d = getBSDF_incrad(il->sd, i); |
143 |
< |
d = 1.0/(PI*d*d); |
174 |
> |
d = 1.0/getBSDF_incohm(il->sd, i); |
175 |
|
} else { |
176 |
|
if (v[2] >= -FTINY) |
177 |
|
continue; /* only sample transmission */ |
178 |
< |
d = 1.0 - v[2]*v[2]; |
179 |
< |
i = d*nalt; |
180 |
< |
d = atan2(-v[1], -v[0])/(2.*PI); |
150 |
< |
if (d < 0.0) d += 1.0; |
151 |
< |
j = d*nazi + 0.5; |
152 |
< |
if (j >= nazi) j = 0; |
153 |
< |
sr.rno = i*nazi + j; |
154 |
< |
d = nalt*nazi/PI; |
178 |
> |
v[0] = -v[0]; v[1] = -v[1]; v[2] = -v[2]; |
179 |
> |
sr.rno = flatindex(v, nalt, nazi); |
180 |
> |
d = nalt*nazi*(1./PI) * v[2]; |
181 |
|
} |
182 |
|
d *= si.dom; /* solid angle correction */ |
183 |
|
scalecolor(sr.rcoef, d); |
251 |
|
|
252 |
|
|
253 |
|
int |
254 |
+ |
flatindex( /* compute index for hemispherical direction */ |
255 |
+ |
FVECT dv, |
256 |
+ |
int nalt, |
257 |
+ |
int nazi |
258 |
+ |
) |
259 |
+ |
{ |
260 |
+ |
double d; |
261 |
+ |
int i, j; |
262 |
+ |
|
263 |
+ |
d = 1.0 - dv[2]*dv[2]; |
264 |
+ |
i = d*nalt; |
265 |
+ |
d = atan2(dv[1], dv[0]) * (0.5/PI); |
266 |
+ |
if (d < 0.0) d += 1.0; |
267 |
+ |
j = d*nazi + 0.5; |
268 |
+ |
if (j >= nazi) j = 0; |
269 |
+ |
return(i*nazi + j); |
270 |
+ |
} |
271 |
+ |
|
272 |
+ |
|
273 |
+ |
int |
274 |
+ |
printgeom( /* print out detailed geometry for BSDF */ |
275 |
+ |
struct BSDF_data *sd, |
276 |
+ |
char *xfrot, |
277 |
+ |
FVECT ctr, |
278 |
+ |
double s1, |
279 |
+ |
double s2 |
280 |
+ |
) |
281 |
+ |
{ |
282 |
+ |
static char mgftemp[] = TEMPLATE; |
283 |
+ |
char cmdbuf[64]; |
284 |
+ |
FILE *fp; |
285 |
+ |
double sca; |
286 |
+ |
|
287 |
+ |
if (sd == NULL || sd->mgf == NULL) |
288 |
+ |
return(0); |
289 |
+ |
if (sd->dim[0] <= FTINY || sd->dim[1] <= FTINY) |
290 |
+ |
return(0); |
291 |
+ |
if ((s1 > s2) ^ (sd->dim[0] > sd->dim[1])) { |
292 |
+ |
sca = s1; s1 = s2; s2 = sca; |
293 |
+ |
} |
294 |
+ |
s1 /= sd->dim[0]; |
295 |
+ |
s2 /= sd->dim[1]; |
296 |
+ |
sca = s1 > s2 ? s1 : s2; |
297 |
+ |
strcpy(mgftemp, TEMPLATE); |
298 |
+ |
if ((fp = fopen(mktemp(mgftemp), "w")) == NULL) |
299 |
+ |
error(SYSTEM, "cannot create temporary file for MGF"); |
300 |
+ |
/* prepend our transform */ |
301 |
+ |
fprintf(fp, "xf%s -s %.5f -t %.5g %.5g %.5g\n", |
302 |
+ |
xfrot, sca, ctr[0], ctr[1], ctr[2]); |
303 |
+ |
/* output given MGF description */ |
304 |
+ |
fputs(sd->mgf, fp); |
305 |
+ |
fputs("\nxf\n", fp); |
306 |
+ |
if (fclose(fp) == EOF) |
307 |
+ |
error(SYSTEM, "error writing MGF temporary file"); |
308 |
+ |
/* execute mgf2rad to convert MGF */ |
309 |
+ |
strcpy(cmdbuf, "mgf2rad "); |
310 |
+ |
strcpy(cmdbuf+8, mgftemp); |
311 |
+ |
fflush(stdout); |
312 |
+ |
system(cmdbuf); |
313 |
+ |
unlink(mgftemp); /* clean up */ |
314 |
+ |
return(1); |
315 |
+ |
} |
316 |
+ |
|
317 |
+ |
|
318 |
+ |
int |
319 |
|
my_default( /* default illum action */ |
320 |
|
OBJREC *ob, |
321 |
|
struct illum_args *il, |
345 |
|
FVECT u, v; |
346 |
|
double ur[2], vr[2]; |
347 |
|
MAT4 xfm; |
348 |
+ |
char xfrot[64]; |
349 |
|
int nallow; |
350 |
|
FACE *fa; |
351 |
|
int i, j; |
357 |
|
} |
358 |
|
/* set up sampling */ |
359 |
|
if (il->sd != NULL) { |
360 |
< |
if (!getBSDF_xfm(xfm, fa->norm, il->udir)) { |
360 |
> |
if (!getBSDF_xfm(xfm, fa->norm, il->udir, xfrot)) { |
361 |
|
objerror(ob, WARNING, "illegal up direction"); |
362 |
|
freeface(ob); |
363 |
|
return(my_default(ob, il, nm)); |
399 |
|
if (r2 < vr[0]) vr[0] = r2; |
400 |
|
if (r2 > vr[1]) vr[1] = r2; |
401 |
|
} |
402 |
+ |
/* output detailed geometry? */ |
403 |
+ |
if (!(il->flags & IL_LIGHT) && il->sd != NULL && il->sd->mgf != NULL && |
404 |
+ |
il->thick <= FTINY) { |
405 |
+ |
for (j = 3; j--; ) |
406 |
+ |
org[j] = .5*(ur[0]+ur[1])*u[j] + |
407 |
+ |
.5*(vr[0]+vr[1])*v[j] + |
408 |
+ |
fa->offset*fa->norm[j]; |
409 |
+ |
printgeom(il->sd, xfrot, org, ur[1]-ur[0], vr[1]-vr[0]); |
410 |
+ |
} |
411 |
|
dim[0] = random(); |
412 |
|
/* sample polygon */ |
413 |
|
nallow = 5*n*il->nsamps; |
452 |
|
raysamp(dim[1], org, dir); |
453 |
|
} |
454 |
|
/* add in direct component? */ |
455 |
< |
if (!directvis && il->flags & IL_LIGHT) { |
455 |
> |
if (!directvis && (il->flags & IL_LIGHT || il->sd != NULL)) { |
456 |
|
MAT4 ixfm; |
457 |
|
if (il->sd == NULL) { |
458 |
|
for (i = 3; i--; ) { |
463 |
|
} |
464 |
|
ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.; |
465 |
|
ixfm[3][3] = 1.; |
466 |
< |
} else if (!invmat4(ixfm, xfm)) |
467 |
< |
objerror(ob, INTERNAL, "cannot invert BSDF transform"); |
466 |
> |
} else { |
467 |
> |
if (!invmat4(ixfm, xfm)) |
468 |
> |
objerror(ob, INTERNAL, |
469 |
> |
"cannot invert BSDF transform"); |
470 |
> |
if (!(il->flags & IL_LIGHT)) |
471 |
> |
new_discount(); |
472 |
> |
} |
473 |
|
dim[0] = random(); |
474 |
|
nallow = 10*il->nsamps; |
475 |
|
for (i = 0; i < il->nsamps; i++) { |
499 |
|
nalt = sqrt(il->sd->nout/PI) + .5; |
500 |
|
nazi = PI*nalt + .5; |
501 |
|
redistribute(il->sd, nalt, nazi, u, v, fa->norm, xfm); |
502 |
+ |
done_discount(); |
503 |
+ |
if (!il->sampdens) |
504 |
+ |
il->sampdens = nalt*nazi/PI + .999; |
505 |
|
} |
506 |
|
/* write out the face and its distribution */ |
507 |
|
if (average(il, distarr, n)) { |
606 |
|
co = getcone(ob, 0); |
607 |
|
/* set up sampling */ |
608 |
|
if (il->sd != NULL) { |
609 |
< |
if (!getBSDF_xfm(xfm, co->ad, il->udir)) { |
609 |
> |
if (!getBSDF_xfm(xfm, co->ad, il->udir, NULL)) { |
610 |
|
objerror(ob, WARNING, "illegal up direction"); |
611 |
|
freecone(ob); |
612 |
|
return(my_default(ob, il, nm)); |
660 |
|
raysamp(dim[1], org, dir); |
661 |
|
} |
662 |
|
/* add in direct component? */ |
663 |
< |
if (!directvis && il->flags & IL_LIGHT) { |
663 |
> |
if (!directvis && (il->flags & IL_LIGHT || il->sd != NULL)) { |
664 |
|
MAT4 ixfm; |
665 |
|
if (il->sd == NULL) { |
666 |
|
for (i = 3; i--; ) { |
671 |
|
} |
672 |
|
ixfm[3][0] = ixfm[3][1] = ixfm[3][2] = 0.; |
673 |
|
ixfm[3][3] = 1.; |
674 |
< |
} else if (!invmat4(ixfm, xfm)) |
675 |
< |
objerror(ob, INTERNAL, "cannot invert BSDF transform"); |
674 |
> |
} else { |
675 |
> |
if (!invmat4(ixfm, xfm)) |
676 |
> |
objerror(ob, INTERNAL, |
677 |
> |
"cannot invert BSDF transform"); |
678 |
> |
if (!(il->flags & IL_LIGHT)) |
679 |
> |
new_discount(); |
680 |
> |
} |
681 |
|
dim[0] = random(); |
682 |
|
for (i = 0; i < il->nsamps; i++) { |
683 |
|
/* randomize location */ |
700 |
|
nalt = sqrt(il->sd->nout/PI) + .5; |
701 |
|
nazi = PI*nalt + .5; |
702 |
|
redistribute(il->sd, nalt, nazi, u, v, co->ad, xfm); |
703 |
+ |
done_discount(); |
704 |
+ |
if (!il->sampdens) |
705 |
+ |
il->sampdens = nalt*nazi/PI + .999; |
706 |
|
} |
707 |
|
/* write out the ring and its distribution */ |
708 |
|
if (average(il, distarr, n)) { |
714 |
|
/* clean up */ |
715 |
|
freecone(ob); |
716 |
|
return(1); |
717 |
+ |
} |
718 |
+ |
|
719 |
+ |
|
720 |
+ |
void |
721 |
+ |
redistribute( /* pass distarr ray sums through BSDF */ |
722 |
+ |
struct BSDF_data *b, |
723 |
+ |
int nalt, |
724 |
+ |
int nazi, |
725 |
+ |
FVECT u, |
726 |
+ |
FVECT v, |
727 |
+ |
FVECT w, |
728 |
+ |
MAT4 xm |
729 |
+ |
) |
730 |
+ |
{ |
731 |
+ |
int nout = 0; |
732 |
+ |
MAT4 mymat, inmat; |
733 |
+ |
COLORV *idist; |
734 |
+ |
COLORV *cp; |
735 |
+ |
FVECT dv; |
736 |
+ |
double wt; |
737 |
+ |
int i, j, k, c, o; |
738 |
+ |
COLOR col, cinc; |
739 |
+ |
/* copy incoming distribution */ |
740 |
+ |
if (b->ninc > distsiz) |
741 |
+ |
error(INTERNAL, "error 1 in redistribute"); |
742 |
+ |
idist = (COLORV *)malloc(sizeof(COLOR)*b->ninc); |
743 |
+ |
if (idist == NULL) |
744 |
+ |
error(SYSTEM, "out of memory in redistribute"); |
745 |
+ |
memcpy(idist, distarr, sizeof(COLOR)*b->ninc); |
746 |
+ |
/* compose direction transform */ |
747 |
+ |
for (i = 3; i--; ) { |
748 |
+ |
mymat[i][0] = u[i]; |
749 |
+ |
mymat[i][1] = v[i]; |
750 |
+ |
mymat[i][2] = w[i]; |
751 |
+ |
mymat[i][3] = 0.; |
752 |
+ |
} |
753 |
+ |
mymat[3][0] = mymat[3][1] = mymat[3][2] = 0.; |
754 |
+ |
mymat[3][3] = 1.; |
755 |
+ |
if (xm != NULL) |
756 |
+ |
multmat4(mymat, xm, mymat); |
757 |
+ |
for (i = 3; i--; ) { /* make sure it's normalized */ |
758 |
+ |
wt = 1./sqrt( mymat[0][i]*mymat[0][i] + |
759 |
+ |
mymat[1][i]*mymat[1][i] + |
760 |
+ |
mymat[2][i]*mymat[2][i] ); |
761 |
+ |
for (j = 3; j--; ) |
762 |
+ |
mymat[j][i] *= wt; |
763 |
+ |
} |
764 |
+ |
if (!invmat4(inmat, mymat)) /* need inverse as well */ |
765 |
+ |
error(INTERNAL, "cannot invert BSDF transform"); |
766 |
+ |
newdist(nalt*nazi); /* resample distribution */ |
767 |
+ |
for (i = b->ninc; i--; ) { |
768 |
+ |
int direct_out = -1; |
769 |
+ |
COLOR cdir; |
770 |
+ |
getBSDF_incvec(dv, b, i); /* compute incident irrad. */ |
771 |
+ |
multv3(dv, dv, mymat); |
772 |
+ |
if (dv[2] < 0.0) { |
773 |
+ |
dv[0] = -dv[0]; dv[1] = -dv[1]; dv[2] = -dv[2]; |
774 |
+ |
direct_out += (direct_discount != NULL); |
775 |
+ |
} |
776 |
+ |
wt = getBSDF_incohm(b, i); |
777 |
+ |
wt *= dv[2]; /* solid_angle*cosine(theta) */ |
778 |
+ |
cp = &idist[3*i]; |
779 |
+ |
copycolor(cinc, cp); |
780 |
+ |
scalecolor(cinc, wt); |
781 |
+ |
if (!direct_out) { /* discount direct contr. */ |
782 |
+ |
cp = &direct_discount[3*i]; |
783 |
+ |
copycolor(cdir, cp); |
784 |
+ |
scalecolor(cdir, -wt); |
785 |
+ |
if (b->nout != b->ninc) |
786 |
+ |
direct_out = flatindex(dv, nalt, nazi); |
787 |
+ |
else |
788 |
+ |
direct_out = i; /* assumes dist. mirroring */ |
789 |
+ |
} |
790 |
+ |
for (k = nalt; k--; ) /* loop over distribution */ |
791 |
+ |
for (j = nazi; j--; ) { |
792 |
+ |
int rstart = random(); |
793 |
+ |
for (c = NBSDFSAMPS; c--; ) { |
794 |
+ |
double sp[2]; |
795 |
+ |
multisamp(sp, 2, urand(rstart+c)); |
796 |
+ |
flatdir(dv, (k + sp[0])/nalt, |
797 |
+ |
(j + .5 - sp[1])/nazi); |
798 |
+ |
multv3(dv, dv, inmat); |
799 |
+ |
/* evaluate BSDF @ outgoing */ |
800 |
+ |
o = getBSDF_outndx(b, dv); |
801 |
+ |
if (o < 0) { |
802 |
+ |
nout++; |
803 |
+ |
continue; |
804 |
+ |
} |
805 |
+ |
wt = BSDF_value(b, i, o) * (1./NBSDFSAMPS); |
806 |
+ |
copycolor(col, cinc); |
807 |
+ |
if (b->nout != b->ninc) |
808 |
+ |
o = k*nazi + j; |
809 |
+ |
if (o == direct_out) |
810 |
+ |
addcolor(col, cdir); /* minus direct */ |
811 |
+ |
scalecolor(col, wt); |
812 |
+ |
cp = &distarr[3*(k*nazi + j)]; |
813 |
+ |
addcolor(cp, col); /* sum into distribution */ |
814 |
+ |
} |
815 |
+ |
} |
816 |
+ |
} |
817 |
+ |
free(idist); /* free temp space */ |
818 |
+ |
if (nout) { |
819 |
+ |
sprintf(errmsg, "missing %.1f%% of BSDF directions", |
820 |
+ |
100.*nout/(b->ninc*nalt*nazi*NBSDFSAMPS)); |
821 |
+ |
error(WARNING, errmsg); |
822 |
+ |
} |
823 |
|
} |