--- ray/src/cv/bsdf2ttree.c 2013/05/15 03:22:31 2.15 +++ ray/src/cv/bsdf2ttree.c 2014/03/24 03:50:28 2.28 @@ -1,5 +1,5 @@ #ifndef lint -static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.15 2013/05/15 03:22:31 greg Exp $"; +static const char RCSid[] = "$Id: bsdf2ttree.c,v 2.28 2014/03/24 03:50:28 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.15 #include #include #include +#include "random.h" #include "platform.h" +#include "rtprocess.h" #include "calcomp.h" #include "bsdfrep.h" /* global argv[0] */ @@ -23,8 +25,51 @@ int samp_order = 6; /* super-sampling threshold */ const double ssamp_thresh = 0.35; /* number of super-samples */ -const int nssamp = 100; +#ifndef NSSAMP +#define NSSAMP 100 +#endif + /* 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 */ +static void +prog_done(void) +{ + int n = do_prog; + + if (n <= 1) return; + fputc('\r', stderr); + while (n--) + fputc(' ', stderr); + fputc('\r', stderr); +} + /* Output XML prologue to stdout */ static void xml_prologue(int ac, char *av[]) @@ -42,8 +87,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"); @@ -112,6 +158,7 @@ eval_isotropic(char *funame) { const int sqres = 1<= 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"); @@ -129,8 +176,14 @@ 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; @@ -138,7 +191,7 @@ eval_isotropic(char *funame) iovec[1] = .0; iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0]); if (funame == NULL) - rbf = advect_rbf(iovec); + rbf = advect_rbf(iovec, lobe_lim); for (ox = 0; ox < sqres; ox++) { float last_bsdf = -1; for (oy = 0; oy < sqres; oy++) { @@ -149,13 +202,21 @@ eval_isotropic(char *funame) bsdf = eval_rbfrep(rbf, iovec+3) * output_orient/iovec[5]; else { - double ssa[3], ssvec[6]; - int ssi; + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } bsdf = funvalue(funame, 6, iovec); +#if (NSSAMP > 0) if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) { - bsdf = 0; /* super-sample voxel */ - for (ssi = nssamp; ssi--; ) { - SDmultiSamp(ssa, 3, (ssi+drand48())/nssamp); + int ssi; + double ssa[3], ssvec[6], sum = 0; + /* super-sample voxel */ + for (ssi = NSSAMP; ssi--; ) { + SDmultiSamp(ssa, 3, (ssi+frandom()) * + (1./NSSAMP)); ssvec[0] = 2.*(ix+ssa[0])/sqres - 1.; ssvec[1] = .0; ssvec[2] = input_orient * @@ -165,10 +226,17 @@ eval_isotropic(char *funame) ssvec[5] = output_orient * sqrt(1. - ssvec[3]*ssvec[3] - ssvec[4]*ssvec[4]); - bsdf += funvalue(funame, 6, ssvec); + if (assignD) { + varset("Dx", '=', -ssvec[3]); + varset("Dy", '=', -ssvec[4]); + varset("Dz", '=', -ssvec[5]); + ++eclock; + } + sum += funvalue(funame, 6, ssvec); } - bsdf /= (float)nssamp; + bsdf = sum/NSSAMP; } +#endif } if (pctcull >= 0) fwrite(&bsdf, sizeof(bsdf), 1, ofp); @@ -179,6 +247,7 @@ eval_isotropic(char *funame) } if (rbf != NULL) free(rbf); + prog_show((ix+1.)*(2./sqres)); } if (pctcull >= 0) { /* finish output */ if (pclose(ofp)) { @@ -192,6 +261,7 @@ eval_isotropic(char *funame) fputs("}\n", stdout); } data_epilogue(); + prog_done(); } /* Interpolate and output anisotropic BSDF data */ @@ -200,6 +270,7 @@ eval_anisotropic(char *funame) { const int sqres = 1<= 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"); @@ -216,8 +288,15 @@ 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++) { @@ -226,7 +305,7 @@ eval_anisotropic(char *funame) iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0] - iovec[1]*iovec[1]); if (funame == NULL) - rbf = advect_rbf(iovec); + rbf = advect_rbf(iovec, lobe_lim); for (ox = 0; ox < sqres; ox++) { float last_bsdf = -1; for (oy = 0; oy < sqres; oy++) { @@ -237,16 +316,24 @@ eval_anisotropic(char *funame) bsdf = eval_rbfrep(rbf, iovec+3) * output_orient/iovec[5]; else { - double ssa[4], ssvec[6]; - int ssi; + if (assignD) { + varset("Dx", '=', -iovec[3]); + varset("Dy", '=', -iovec[4]); + varset("Dz", '=', -iovec[5]); + ++eclock; + } bsdf = funvalue(funame, 6, iovec); +#if (NSSAMP > 0) if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) { - bsdf = 0; /* super-sample voxel */ - for (ssi = nssamp; ssi--; ) { - SDmultiSamp(ssa, 4, (ssi+drand48())/nssamp); + 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); - ssvec[2] = output_orient * + ssvec[2] = input_orient * sqrt(1. - ssvec[0]*ssvec[0] - ssvec[1]*ssvec[1]); SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres, @@ -254,10 +341,17 @@ eval_anisotropic(char *funame) ssvec[5] = output_orient * sqrt(1. - ssvec[3]*ssvec[3] - ssvec[4]*ssvec[4]); - bsdf += funvalue(funame, 6, ssvec); + if (assignD) { + varset("Dx", '=', -ssvec[3]); + varset("Dy", '=', -ssvec[4]); + varset("Dz", '=', -ssvec[5]); + ++eclock; + } + sum += funvalue(funame, 6, ssvec); } - bsdf /= (float)nssamp; + bsdf = sum/NSSAMP; } +#endif } if (pctcull >= 0) fwrite(&bsdf, sizeof(bsdf), 1, ofp); @@ -268,6 +362,7 @@ eval_anisotropic(char *funame) } if (rbf != NULL) free(rbf); + prog_show((ix*sqres+iy+1.)/(sqres*sqres)); } if (pctcull >= 0) { /* finish output */ if (pclose(ofp)) { @@ -278,6 +373,7 @@ eval_anisotropic(char *funame) } else fputs("}\n", stdout); data_epilogue(); + prog_done(); } /* Read in BSDF and interpolate as tensor tree representation */ @@ -324,32 +420,44 @@ 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); @@ -357,6 +465,7 @@ main(int argc, char *argv[]) if (i < argc) { /* open input files if given */ 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", @@ -368,6 +477,8 @@ main(int argc, char *argv[]) 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 @@ -380,6 +491,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 @@ -388,7 +500,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",