--- ray/src/cv/bsdf2ttree.c 2013/03/22 02:15:06 2.11 +++ ray/src/cv/bsdf2ttree.c 2014/03/12 21:15:31 2.26 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.11 2013/03/22 02:15:06 greg Exp $"; +static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.26 2014/03/12 21:15:31 greg Exp $"; #endif /* * Load measured BSDF interpolant and write out as XML file with tensor tree. @@ -11,7 +11,9 @@ static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.11 #include #include #include +#include "random.h" #include "platform.h" +#include "rtprocess.h" #include "calcomp.h" #include "bsdfrep.h" /* global argv[0] */ @@ -20,7 +22,42 @@ char *progname; double pctcull = 90.; /* sampling order */ int samp_order = 6; + /* super-sampling threshold */ +const double ssamp_thresh = 0.35; + /* number of super-samples */ +const int nssamp = 100; + /* limit on number of RBF lobes */ +static int lobe_lim = 15000; + /* progress bar length */ +static int do_prog = 79; + +/* Start new progress bar */ +#define prog_start(s) if (do_prog) fprintf(stderr, "%s: %s...\n", progname, s); else + +/* Draw progress bar of the appropriate length */ +static void +prog_show(double frac) +{ + char pbar[256]; + int nchars; + + if (do_prog <= 0) return; + if (do_prog > sizeof(pbar)-2) + do_prog = sizeof(pbar)-2; + if (frac < 0) frac = 0; + else if (frac > 1) frac = 1; + nchars = do_prog*frac + .5; + pbar[0] = '\r'; + memset(pbar+1, '*', nchars); + memset(pbar+1+nchars, '-', do_prog-nchars); + pbar[do_prog+1] = '\0'; + fputs(pbar, stderr); +} + +/* Finish progress bar */ +#define prog_done() if (do_prog) fputc('\n',stderr); else + /* Output XML prologue to stdout */ static void xml_prologue(int ac, char *av[]) @@ -38,8 +75,9 @@ xml_prologue(int ac, char *av[]) puts(""); puts(""); puts("\t"); - puts("\t\tName"); - puts("\t\tManufacturer"); + printf("\t\t%s\n", bsdf_name[0] ? bsdf_name : "Unknown"); + printf("\t\t%s\n", + bsdf_manuf[0] ? bsdf_manuf : "Unknown"); puts("\t\tOther"); puts("\t"); puts("\t"); @@ -90,24 +128,33 @@ xml_epilogue(void) puts(""); } +/* Compute absolute relative difference */ +static double +abs_diff(double v1, double v0) +{ + if ((v0 < 0) | (v1 < 0)) + return(.0); + v1 = (v1-v0)*2./(v0+v1+.0001); + if (v1 < 0) + return(-v1); + return(v1); +} + /* Interpolate and output isotropic BSDF data */ static void eval_isotropic(char *funame) { const int sqres = 1<= 0) fprintf(stderr, "data with %.1f%% culling\n", pctcull); - else fputs("raw data\n", stderr); -#endif + data_prologue(); /* begin output */ if (pctcull >= 0) { - sprintf(cmd, "rttree_reduce -h -a -ff -r 3 -t %f -g %d", + sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d", pctcull, samp_order); fflush(stdout); ofp = popen(cmd, "w"); @@ -117,33 +164,75 @@ eval_isotropic(char *funame) exit(1); } SET_FILE_BINARY(ofp); +#ifdef getc_unlocked /* avoid lock/unlock overhead */ + flockfile(ofp); +#endif } else fputs("{\n", stdout); + /* need to assign Dx, Dy, Dz? */ + if (funame != NULL) + assignD = (fundefined(funame) < 6); /* run through directions */ for (ix = 0; ix < sqres/2; ix++) { RBFNODE *rbf = NULL; - iovec[0] = (ix+.5)/sqres - 1.; + iovec[0] = 2.*(ix+.5)/sqres - 1.; iovec[1] = .0; iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0]); if (funame == NULL) - rbf = advect_rbf(iovec); - for (ox = 0; ox < sqres; ox++) + rbf = advect_rbf(iovec, lobe_lim); + for (ox = 0; ox < sqres; ox++) { + float last_bsdf = -1; for (oy = 0; oy < sqres; oy++) { SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres); iovec[5] = output_orient * sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]); if (funame == NULL) - bsdf = eval_rbfrep(rbf, iovec+3) * + bsdf = eval_rbfrep(rbf, iovec+3) * output_orient/iovec[5]; - else - bsdf = funvalue(funame, 6, iovec); + else { + double ssa[3], ssvec[6], sum; + int ssi; + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } + bsdf = funvalue(funame, 6, iovec); + if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) { + sum = 0; /* super-sample voxel */ + for (ssi = nssamp; ssi--; ) { + SDmultiSamp(ssa, 3, (ssi+frandom())/nssamp); + ssvec[0] = 2.*(ix+ssa[0])/sqres - 1.; + ssvec[1] = .0; + ssvec[2] = input_orient * + sqrt(1. - ssvec[0]*ssvec[0]); + SDsquare2disk(ssvec+3, (ox+ssa[1])/sqres, + (oy+ssa[2])/sqres); + ssvec[5] = output_orient * + sqrt(1. - ssvec[3]*ssvec[3] - + ssvec[4]*ssvec[4]); + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } + sum += funvalue(funame, 6, ssvec); + } + bsdf = sum/nssamp; + } + } if (pctcull >= 0) fwrite(&bsdf, sizeof(bsdf), 1, ofp); else printf("\t%.3e\n", bsdf); + last_bsdf = bsdf; } + } if (rbf != NULL) free(rbf); + prog_show((ix+1.)*(2./sqres)); } if (pctcull >= 0) { /* finish output */ if (pclose(ofp)) { @@ -157,6 +246,7 @@ eval_isotropic(char *funame) fputs("}\n", stdout); } data_epilogue(); + prog_done(); } /* Interpolate and output anisotropic BSDF data */ @@ -165,18 +255,16 @@ eval_anisotropic(char *funame) { const int sqres = 1<= 0) fprintf(stderr, "data with %.1f%% culling\n", pctcull); - else fputs("raw data\n", stderr); -#endif + data_prologue(); /* begin output */ if (pctcull >= 0) { - sprintf(cmd, "rttree_reduce -h -a -ff -r 4 -t %f -g %d", + sprintf(cmd, "rttree_reduce%s -h -ff -r 4 -t %f -g %d", + (input_orient>0 ^ output_orient>0) ? "" : " -a", pctcull, samp_order); fflush(stdout); ofp = popen(cmd, "w"); @@ -185,35 +273,78 @@ eval_anisotropic(char *funame) progname); exit(1); } + SET_FILE_BINARY(ofp); +#ifdef getc_unlocked /* avoid lock/unlock overhead */ + flockfile(ofp); +#endif } else fputs("{\n", stdout); + /* need to assign Dx, Dy, Dz? */ + if (funame != NULL) + assignD = (fundefined(funame) < 6); /* run through directions */ for (ix = 0; ix < sqres; ix++) for (iy = 0; iy < sqres; iy++) { RBFNODE *rbf = NULL; /* Klems reversal */ - SDsquare2disk(iovec, (ix+.5)/sqres, (iy+.5)/sqres); - iovec[0] = -iovec[0]; iovec[1] = -iovec[1]; + SDsquare2disk(iovec, 1.-(ix+.5)/sqres, 1.-(iy+.5)/sqres); iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0] - iovec[1]*iovec[1]); if (funame == NULL) - rbf = advect_rbf(iovec); - for (ox = 0; ox < sqres; ox++) + rbf = advect_rbf(iovec, lobe_lim); + for (ox = 0; ox < sqres; ox++) { + float last_bsdf = -1; for (oy = 0; oy < sqres; oy++) { SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres); iovec[5] = output_orient * sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]); if (funame == NULL) - bsdf = eval_rbfrep(rbf, iovec+3) * + bsdf = eval_rbfrep(rbf, iovec+3) * output_orient/iovec[5]; - else - bsdf = funvalue(funame, 6, iovec); + else { + double ssa[4], ssvec[6], sum; + int ssi; + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } + bsdf = funvalue(funame, 6, iovec); + if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) { + sum = 0; /* super-sample voxel */ + for (ssi = nssamp; ssi--; ) { + SDmultiSamp(ssa, 4, (ssi+frandom())/nssamp); + SDsquare2disk(ssvec, 1.-(ix+ssa[0])/sqres, + 1.-(iy+ssa[1])/sqres); + ssvec[2] = output_orient * + sqrt(1. - ssvec[0]*ssvec[0] - + ssvec[1]*ssvec[1]); + SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres, + (oy+ssa[3])/sqres); + ssvec[5] = output_orient * + sqrt(1. - ssvec[3]*ssvec[3] - + ssvec[4]*ssvec[4]); + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } + sum += funvalue(funame, 6, ssvec); + } + bsdf = sum/nssamp; + } + } if (pctcull >= 0) fwrite(&bsdf, sizeof(bsdf), 1, ofp); else printf("\t%.3e\n", bsdf); + last_bsdf = bsdf; } + } if (rbf != NULL) free(rbf); + prog_show((ix*sqres+iy+1.)/(sqres*sqres)); } if (pctcull >= 0) { /* finish output */ if (pclose(ofp)) { @@ -224,6 +355,7 @@ eval_anisotropic(char *funame) } else fputs("}\n", stdout); data_epilogue(); + prog_done(); } /* Read in BSDF and interpolate as tensor tree representation */ @@ -255,10 +387,10 @@ main(int argc, char *argv[]) case 't': switch (argv[i][2]) { case '3': - single_plane_incident = 0; + single_plane_incident = 1; break; case '4': - single_plane_incident = 1; + single_plane_incident = 0; break; case '\0': pctcull = atof(argv[++i]); @@ -270,35 +402,52 @@ main(int argc, char *argv[]) case 'g': samp_order = atoi(argv[++i]); break; + case 'l': + lobe_lim = atoi(argv[++i]); + break; + case 'p': + do_prog = atoi(argv[i]+2); + break; default: goto userr; } if (single_plane_incident >= 0) { /* function-based BSDF? */ void (*evf)(char *s) = single_plane_incident ? &eval_isotropic : &eval_anisotropic; - if (i != argc-1 || fundefined(argv[i]) != 6) + if (i != argc-1 || fundefined(argv[i]) < 3) { + fprintf(stderr, + "%s: need single function with 6 arguments: bsdf(ix,iy,iz,ox,oy,oz)\n", + progname); + fprintf(stderr, "\tor 3 arguments using Dx,Dy,Dz: bsdf(ix,iy,iz)\n"); goto userr; + } + ++eclock; xml_prologue(argc, argv); /* start XML output */ if (dofwd) { input_orient = -1; output_orient = -1; - (*evf)(argv[i]); /* outside reflectance */ + prog_start("Evaluating outside reflectance"); + (*evf)(argv[i]); output_orient = 1; - (*evf)(argv[i]); /* outside -> inside */ + prog_start("Evaluating outside->inside transmission"); + (*evf)(argv[i]); } if (dobwd) { input_orient = 1; output_orient = 1; - (*evf)(argv[i]); /* inside reflectance */ + prog_start("Evaluating inside reflectance"); + (*evf)(argv[i]); output_orient = -1; - (*evf)(argv[i]); /* inside -> outside */ + prog_start("Evaluating inside->outside transmission"); + (*evf)(argv[i]); } xml_epilogue(); /* finish XML output & exit */ return(0); } if (i < argc) { /* open input files if given */ - xml_prologue(argc, argv); /* start XML output */ + int nbsdf = 0; for ( ; i < argc; i++) { /* interpolate each component */ + char pbuf[256]; FILE *fpin = fopen(argv[i], "rb"); if (fpin == NULL) { fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n", @@ -308,6 +457,10 @@ main(int argc, char *argv[]) if (!load_bsdf_rep(fpin)) return(1); fclose(fpin); + if (!nbsdf++) /* start XML on first dist. */ + xml_prologue(argc, argv); + sprintf(pbuf, "Interpolating component '%s'", argv[i]); + prog_start(pbuf); if (single_plane_incident) eval_isotropic(NULL); else @@ -320,6 +473,7 @@ main(int argc, char *argv[]) if (!load_bsdf_rep(stdin)) return(1); xml_prologue(argc, argv); /* start XML output */ + prog_start("Interpolating from standard input"); if (single_plane_incident) /* resample dist. */ eval_isotropic(NULL); else @@ -328,7 +482,7 @@ main(int argc, char *argv[]) return(0); userr: fprintf(stderr, - "Usage: %s [-g Nlog2][-t pctcull] [bsdf.sir ..] > bsdf.xml\n", + "Usage: %s [-g Nlog2][-t pctcull][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n", progname); fprintf(stderr, " or: %s -t{3|4} [-g Nlog2][-t pctcull][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n",