--- ray/src/cv/bsdf2ttree.c 2017/05/31 17:25:21 2.41 +++ ray/src/cv/bsdf2ttree.c 2020/10/26 21:12:20 2.53 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.41 2017/05/31 17:25:21 greg Exp $"; +static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.53 2020/10/26 21:12:20 greg Exp $"; #endif /* * Load measured BSDF interpolant and write out as XML file with tensor tree. @@ -8,9 +8,9 @@ static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.41 */ #define _USE_MATH_DEFINES -#include #include #include +#include #include "random.h" #include "platform.h" #include "paths.h" @@ -26,11 +26,9 @@ static double pctcull = 90.; /* sampling order */ static int samp_order = 6; /* super-sampling threshold */ -const double ssamp_thresh = 0.35; +static double ssamp_thresh = 0.35; /* number of super-samples */ -#ifndef NSSAMP -#define NSSAMP 64 -#endif +static int nssamp = 256; /* limit on number of RBF lobes */ static int lobe_lim = 15000; /* progress bar length */ @@ -129,6 +127,10 @@ static void eval_isotropic(char *funame) { const int sqres = 1< ssamp_thresh || + (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) || + (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) || + (oy < sqres-1 && + abs_diff(bsdf, sdv_next[oy+1].cieY) > ssamp_thresh)) { + int ssi; + double ssa[2], sum = 0, usum = 0, vsum = 0; + /* super-sample voxel */ + for (ssi = nssamp; ssi--; ) { + SDmultiSamp(ssa, 2, (ssi+frandom()) / + (double)nssamp); + SDsquare2disk(iovec+3, (ox+ssa[0])*sqfact, + (oy+ssa[1])*sqfact); + iovec[5] = output_orient * + sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]); + eval_rbfcol(&sdv, rbf, iovec+3); + sum += sdv.cieY; + if (rbf_colorimetry == RBCtristimulus) { + sdv.cieY /= + -2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.; + usum += 4.*sdv.spec.cx * sdv.cieY; + vsum += 9.*sdv.spec.cy * sdv.cieY; + } + } + bsdf = sum / (double)nssamp; + if (rbf_colorimetry == RBCtristimulus) { + uv[0] = usum / (sum+FTINY); + uv[1] = vsum / (sum+FTINY); + } + } else if (rbf_colorimetry == RBCtristimulus) { uv[0] = uv[1] = 1. / (-2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.); @@ -405,28 +545,34 @@ eval_anisotropic(char *funame) uv[1] *= 9.*sdv.spec.cy; } } else { - if (assignD) { - varset("Dx", '=', -iovec[3]); - varset("Dy", '=', -iovec[4]); - varset("Dz", '=', -iovec[5]); - ++eclock; + bsdf = val_next[oy]; + if (ox < sqres-1) { + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } + val_next[oy] = funvalue(funame, 6, iovec); } - bsdf = funvalue(funame, 6, iovec); -#if (NSSAMP > 0) - if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) { + if (abs_diff(bsdf, val_next[oy]) > ssamp_thresh || + (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) || + (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) || + (oy < sqres-1 && + abs_diff(bsdf, val_next[oy+1]) > ssamp_thresh)) { int ssi; double ssa[4], ssvec[6], sum = 0; /* super-sample voxel */ - for (ssi = NSSAMP; ssi--; ) { - SDmultiSamp(ssa, 4, (ssi+frandom()) * - (1./NSSAMP)); - SDsquare2disk(ssvec, 1.-(ix+ssa[0])/sqres, - 1.-(iy+ssa[1])/sqres); + for (ssi = nssamp; ssi--; ) { + SDmultiSamp(ssa, 4, (ssi+frandom()) / + (double)nssamp); + SDsquare2disk(ssvec, 1.-(ix+ssa[0])*sqfact, + 1.-(iy+ssa[1])*sqfact); ssvec[2] = input_orient * sqrt(1. - ssvec[0]*ssvec[0] - ssvec[1]*ssvec[1]); - SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres, - (oy+ssa[3])/sqres); + SDsquare2disk(ssvec+3, (ox+ssa[2])*sqfact, + (oy+ssa[3])*sqfact); ssvec[5] = output_orient * sqrt(1. - ssvec[3]*ssvec[3] - ssvec[4]*ssvec[4]); @@ -438,9 +584,8 @@ eval_anisotropic(char *funame) } sum += funvalue(funame, 6, ssvec); } - bsdf = sum/NSSAMP; + bsdf = sum / (double)nssamp; } -#endif } if (pctcull >= 0) putbinary(&bsdf, sizeof(bsdf), 1, ofp); @@ -456,7 +601,8 @@ eval_anisotropic(char *funame) fprintf(uvfp[1], "\t%.3e\n", uv[1]); } } - last_bsdf = bsdf; + if (val_last != NULL) + val_last[oy] = bsdf; } } if (rbf != NULL) @@ -464,6 +610,11 @@ eval_anisotropic(char *funame) prog_show((ix*sqres+iy+1.)/(sqres*sqres)); } prog_done(); + if (val_last != NULL) { + free(val_last); + if (val_next != NULL) free(val_next); + if (sdv_next != NULL) free(sdv_next); + } if (pctcull >= 0) { /* finish output */ if (pclose(ofp)) { fprintf(stderr, "%s: error running rttree_reduce on Y\n", @@ -551,6 +702,39 @@ wrap_up(void) } #endif +#define HEAD_BUFLEN 8192 +static char head_buf[HEAD_BUFLEN]; +static int cur_headlen = 0; + +/* Record header line as comment associated with this SIR input */ +static int +record2header(char *s) +{ + int len = strlen(s); + + if (cur_headlen+len >= HEAD_BUFLEN-6) + return(0); + /* includes EOL */ + strcpy(head_buf+cur_headlen, s); + cur_headlen += len; + + return(1); +} + +/* Finish off header for this file */ +static void +done_header(void) +{ + while (cur_headlen > 0 && isspace(head_buf[cur_headlen-1])) + --cur_headlen; + head_buf[cur_headlen] = '\0'; + if (!cur_headlen) + return; + add_wbsdf("-C", 1); + add_wbsdf(head_buf, 0); + head_buf[cur_headlen=0] = '\0'; +} + /* Read in BSDF and interpolate as tensor tree representation */ int main(int argc, char *argv[]) @@ -569,14 +753,27 @@ main(int argc, char *argv[]) switch (argv[i][1]) { /* get options */ case 'e': scompile(argv[++i], NULL, 0); + if (single_plane_incident < 0) + single_plane_incident = 0; break; case 'f': if (!argv[i][2]) { if (strchr(argv[++i], '=') != NULL) { add_wbsdf("-f", 1); add_wbsdf(argv[i], 1); - } else - fcompile(argv[i]); + } else { + char *fpath = getpath(argv[i], + getrlibpath(), 0); + if (fpath == NULL) { + fprintf(stderr, + "%s: cannot find file '%s'\n", + argv[0], argv[i]); + return(1); + } + fcompile(fpath); + if (single_plane_incident < 0) + single_plane_incident = 0; + } } else dofwd = (argv[i][0] == '+'); break; @@ -586,6 +783,16 @@ main(int argc, char *argv[]) case 'b': dobwd = (argv[i][0] == '+'); break; + case 'n': + nssamp = atoi(argv[++i]); + if (nssamp <= 0) + goto userr; + break; + case 's': + ssamp_thresh = atof(argv[++i]); + if (ssamp_thresh <= FTINY) + goto userr; + break; case 't': switch (argv[i][2]) { case '3': @@ -668,9 +875,13 @@ main(int argc, char *argv[]) progname, argv[i]); return(1); } + sprintf(pbuf, "%s:\n", argv[i]); + record2header(pbuf); + sir_headshare = &record2header; if (!load_bsdf_rep(fpin)) return(1); fclose(fpin); + done_header(); sprintf(pbuf, "Interpolating component '%s'", argv[i]); prog_start(pbuf); if (!nbsdf++) { @@ -698,10 +909,10 @@ main(int argc, char *argv[]) return(wrap_up()); userr: fprintf(stderr, - "Usage: %s [{+|-}a][-g Nlog2][-t pctcull][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n", + "Usage: %s [{+|-}a][-g Nlog2][-t pctcull][-n nss][-s thresh][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n", progname); fprintf(stderr, - " or: %s -t{3|4} [{+|-}a][-g Nlog2][-t pctcull][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n", + " or: %s -t{3|4} [{+|-}a][-g Nlog2][-t pctcull][-n nss][-s thresh][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n", progname); return(1); }