ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/cv/bsdf2ttree.c
(Generate patch)

Comparing ray/src/cv/bsdf2ttree.c (file contents):
Revision 2.19 by greg, Thu Oct 3 17:01:02 2013 UTC vs.
Revision 2.29 by greg, Thu Aug 21 10:33:48 2014 UTC

# Line 11 | Line 11 | static const char RCSid[] = "$Id$";
11   #include <stdio.h>
12   #include <stdlib.h>
13   #include <math.h>
14 + #include "random.h"
15   #include "platform.h"
16 + #include "rtprocess.h"
17   #include "calcomp.h"
18   #include "bsdfrep.h"
19                                  /* global argv[0] */
# Line 23 | Line 25 | int                    samp_order = 6;
25                                  /* super-sampling threshold */
26   const double            ssamp_thresh = 0.35;
27                                  /* number of super-samples */
28 < const int               nssamp = 100;
28 > #ifndef NSSAMP
29 > #define NSSAMP          100
30 > #endif
31                                  /* limit on number of RBF lobes */
32   static int              lobe_lim = 15000;
33 +                                /* progress bar length */
34 + static int              do_prog = 79;
35  
36 +
37 + /* Start new progress bar */
38 + #define prog_start(s)   if (do_prog) fprintf(stderr, "%s: %s...\n", progname, s); else
39 +
40 + /* Draw progress bar of the appropriate length */
41 + static void
42 + prog_show(double frac)
43 + {
44 +        char    pbar[256];
45 +        int     nchars;
46 +
47 +        if (do_prog <= 0) return;
48 +        if (do_prog > sizeof(pbar)-2)
49 +                do_prog = sizeof(pbar)-2;
50 +        if (frac < 0) frac = 0;
51 +        else if (frac > 1) frac = 1;
52 +        nchars = do_prog*frac + .5;
53 +        pbar[0] = '\r';
54 +        memset(pbar+1, '*', nchars);
55 +        memset(pbar+1+nchars, '-', do_prog-nchars);
56 +        pbar[do_prog+1] = '\0';
57 +        fputs(pbar, stderr);
58 + }
59 +
60 + /* Finish progress bar */
61 + static void
62 + prog_done(void)
63 + {
64 +        int     n = do_prog;
65 +
66 +        if (n <= 1) return;
67 +        fputc('\r', stderr);
68 +        while (n--)
69 +                fputc(' ', stderr);
70 +        fputc('\r', stderr);
71 + }
72 +
73   /* Output XML prologue to stdout */
74   static void
75   xml_prologue(int ac, char *av[])
# Line 44 | Line 87 | xml_prologue(int ac, char *av[])
87          puts("<Optical>");
88          puts("<Layer>");
89          puts("\t<Material>");
90 <        puts("\t\t<Name>Name</Name>");
91 <        puts("\t\t<Manufacturer>Manufacturer</Manufacturer>");
90 >        printf("\t\t<Name>%s</Name>\n", bsdf_name[0] ? bsdf_name : "Unknown");
91 >        printf("\t\t<Manufacturer>%s</Manufacturer>\n",
92 >                        bsdf_manuf[0] ? bsdf_manuf : "Unknown");
93          puts("\t\t<DeviceType>Other</DeviceType>");
94          puts("\t</Material>");
95          puts("\t<DataDefinition>");
# Line 122 | Line 166 | eval_isotropic(char *funame)
166  
167          data_prologue();                        /* begin output */
168          if (pctcull >= 0) {
169 <                sprintf(cmd, "rttree_reduce%s -h -ff -r 3 -t %f -g %d",
126 <                                (input_orient>0 ^ output_orient>0) ? "" : " -a",
169 >                sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d",
170                                  pctcull, samp_order);
171                  fflush(stdout);
172                  ofp = popen(cmd, "w");
# Line 133 | Line 176 | eval_isotropic(char *funame)
176                          exit(1);
177                  }
178                  SET_FILE_BINARY(ofp);
179 + #ifdef getc_unlocked                            /* avoid lock/unlock overhead */
180 +                flockfile(ofp);
181 + #endif
182          } else
183                  fputs("{\n", stdout);
184                                                  /* need to assign Dx, Dy, Dz? */
# Line 153 | Line 199 | eval_isotropic(char *funame)
199                          iovec[5] = output_orient *
200                                  sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
201                          if (funame == NULL)
202 <                            bsdf = eval_rbfrep(rbf, iovec+3) *
157 <                                                output_orient/iovec[5];
202 >                            bsdf = eval_rbfrep(rbf, iovec+3);
203                          else {
159 –                            double      ssa[3], ssvec[6], sum;
160 –                            int         ssi;
204                              if (assignD) {
205                                  varset("Dx", '=', -iovec[3]);
206                                  varset("Dy", '=', -iovec[4]);
# Line 165 | Line 208 | eval_isotropic(char *funame)
208                                  ++eclock;
209                              }
210                              bsdf = funvalue(funame, 6, iovec);
211 + #if (NSSAMP > 0)
212                              if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
213 <                                sum = 0;        /* super-sample voxel */
214 <                                for (ssi = nssamp; ssi--; ) {
215 <                                    SDmultiSamp(ssa, 3, (ssi+drand48())/nssamp);
213 >                                int     ssi;
214 >                                double  ssa[3], ssvec[6], sum = 0;
215 >                                                /* super-sample voxel */
216 >                                for (ssi = NSSAMP; ssi--; ) {
217 >                                    SDmultiSamp(ssa, 3, (ssi+frandom()) *
218 >                                                        (1./NSSAMP));
219                                      ssvec[0] = 2.*(ix+ssa[0])/sqres - 1.;
220                                      ssvec[1] = .0;
221                                      ssvec[2] = input_orient *
# Line 179 | Line 226 | eval_isotropic(char *funame)
226                                                  sqrt(1. - ssvec[3]*ssvec[3] -
227                                                          ssvec[4]*ssvec[4]);
228                                      if (assignD) {
229 <                                        varset("Dx", '=', -iovec[3]);
230 <                                        varset("Dy", '=', -iovec[4]);
231 <                                        varset("Dz", '=', -iovec[5]);
229 >                                        varset("Dx", '=', -ssvec[3]);
230 >                                        varset("Dy", '=', -ssvec[4]);
231 >                                        varset("Dz", '=', -ssvec[5]);
232                                          ++eclock;
233                                      }
234                                      sum += funvalue(funame, 6, ssvec);
235                                  }
236 <                                bsdf = sum/nssamp;
236 >                                bsdf = sum/NSSAMP;
237                              }
238 + #endif
239                          }
240                          if (pctcull >= 0)
241                                  fwrite(&bsdf, sizeof(bsdf), 1, ofp);
# Line 198 | Line 246 | eval_isotropic(char *funame)
246                  }
247                  if (rbf != NULL)
248                          free(rbf);
249 +                prog_show((ix+1.)*(2./sqres));
250          }
251          if (pctcull >= 0) {                     /* finish output */
252                  if (pclose(ofp)) {
# Line 211 | Line 260 | eval_isotropic(char *funame)
260                  fputs("}\n", stdout);
261          }
262          data_epilogue();
263 +        prog_done();
264   }
265  
266   /* Interpolate and output anisotropic BSDF data */
# Line 237 | Line 287 | eval_anisotropic(char *funame)
287                                          progname);
288                          exit(1);
289                  }
290 +                SET_FILE_BINARY(ofp);
291 + #ifdef getc_unlocked                            /* avoid lock/unlock overhead */
292 +                flockfile(ofp);
293 + #endif
294          } else
295                  fputs("{\n", stdout);
296                                                  /* need to assign Dx, Dy, Dz? */
# Line 258 | Line 312 | eval_anisotropic(char *funame)
312                          iovec[5] = output_orient *
313                                  sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
314                          if (funame == NULL)
315 <                            bsdf = eval_rbfrep(rbf, iovec+3) *
262 <                                                output_orient/iovec[5];
315 >                            bsdf = eval_rbfrep(rbf, iovec+3);
316                          else {
264 –                            double      ssa[4], ssvec[6], sum;
265 –                            int         ssi;
317                              if (assignD) {
318                                  varset("Dx", '=', -iovec[3]);
319                                  varset("Dy", '=', -iovec[4]);
# Line 270 | Line 321 | eval_anisotropic(char *funame)
321                                  ++eclock;
322                              }
323                              bsdf = funvalue(funame, 6, iovec);
324 + #if (NSSAMP > 0)
325                              if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
326 <                                sum = 0;        /* super-sample voxel */
327 <                                for (ssi = nssamp; ssi--; ) {
328 <                                    SDmultiSamp(ssa, 4, (ssi+drand48())/nssamp);
326 >                                int     ssi;
327 >                                double  ssa[4], ssvec[6], sum = 0;
328 >                                                /* super-sample voxel */
329 >                                for (ssi = NSSAMP; ssi--; ) {
330 >                                    SDmultiSamp(ssa, 4, (ssi+frandom()) *
331 >                                                        (1./NSSAMP));
332                                      SDsquare2disk(ssvec, 1.-(ix+ssa[0])/sqres,
333                                                  1.-(iy+ssa[1])/sqres);
334 <                                    ssvec[2] = output_orient *
334 >                                    ssvec[2] = input_orient *
335                                                  sqrt(1. - ssvec[0]*ssvec[0] -
336                                                          ssvec[1]*ssvec[1]);
337                                      SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres,
# Line 285 | Line 340 | eval_anisotropic(char *funame)
340                                                  sqrt(1. - ssvec[3]*ssvec[3] -
341                                                          ssvec[4]*ssvec[4]);
342                                      if (assignD) {
343 <                                        varset("Dx", '=', -iovec[3]);
344 <                                        varset("Dy", '=', -iovec[4]);
345 <                                        varset("Dz", '=', -iovec[5]);
343 >                                        varset("Dx", '=', -ssvec[3]);
344 >                                        varset("Dy", '=', -ssvec[4]);
345 >                                        varset("Dz", '=', -ssvec[5]);
346                                          ++eclock;
347                                      }
348                                      sum += funvalue(funame, 6, ssvec);
349                                  }
350 <                                bsdf = sum/nssamp;
350 >                                bsdf = sum/NSSAMP;
351                              }
352 + #endif
353                          }
354                          if (pctcull >= 0)
355                                  fwrite(&bsdf, sizeof(bsdf), 1, ofp);
# Line 304 | Line 360 | eval_anisotropic(char *funame)
360                  }
361                  if (rbf != NULL)
362                          free(rbf);
363 +                prog_show((ix*sqres+iy+1.)/(sqres*sqres));
364              }
365          if (pctcull >= 0) {                     /* finish output */
366                  if (pclose(ofp)) {
# Line 314 | Line 371 | eval_anisotropic(char *funame)
371          } else
372                  fputs("}\n", stdout);
373          data_epilogue();
374 +        prog_done();
375   }
376  
377   /* Read in BSDF and interpolate as tensor tree representation */
# Line 363 | Line 421 | main(int argc, char *argv[])
421                  case 'l':
422                          lobe_lim = atoi(argv[++i]);
423                          break;
424 +                case 'p':
425 +                        do_prog = atoi(argv[i]+2);
426 +                        break;
427                  default:
428                          goto userr;
429                  }
# Line 373 | Line 434 | main(int argc, char *argv[])
434                          fprintf(stderr,
435          "%s: need single function with 6 arguments: bsdf(ix,iy,iz,ox,oy,oz)\n",
436                                          progname);
437 <                        fprintf(stderr, "\tor 3 arguments using Dx,Dy,Dz: bsdf(ix,iy,iz)\n",
377 <                                        progname);
437 >                        fprintf(stderr, "\tor 3 arguments using Dx,Dy,Dz: bsdf(ix,iy,iz)\n");
438                          goto userr;
439                  }
440                  ++eclock;
# Line 382 | Line 442 | main(int argc, char *argv[])
442                  if (dofwd) {
443                          input_orient = -1;
444                          output_orient = -1;
445 <                        (*evf)(argv[i]);        /* outside reflectance */
445 >                        prog_start("Evaluating outside reflectance");
446 >                        (*evf)(argv[i]);
447                          output_orient = 1;
448 <                        (*evf)(argv[i]);        /* outside -> inside */
448 >                        prog_start("Evaluating outside->inside transmission");
449 >                        (*evf)(argv[i]);
450                  }
451                  if (dobwd) {
452                          input_orient = 1;
453                          output_orient = 1;
454 <                        (*evf)(argv[i]);        /* inside reflectance */
454 >                        prog_start("Evaluating inside reflectance");
455 >                        (*evf)(argv[i]);
456                          output_orient = -1;
457 <                        (*evf)(argv[i]);        /* inside -> outside */
457 >                        prog_start("Evaluating inside->outside transmission");
458 >                        (*evf)(argv[i]);
459                  }
460                  xml_epilogue();                 /* finish XML output & exit */
461                  return(0);
# Line 399 | Line 463 | main(int argc, char *argv[])
463          if (i < argc) {                         /* open input files if given */
464                  int     nbsdf = 0;
465                  for ( ; i < argc; i++) {        /* interpolate each component */
466 +                        char    pbuf[256];
467                          FILE    *fpin = fopen(argv[i], "rb");
468                          if (fpin == NULL) {
469                                  fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
# Line 410 | Line 475 | main(int argc, char *argv[])
475                          fclose(fpin);
476                          if (!nbsdf++)           /* start XML on first dist. */
477                                  xml_prologue(argc, argv);
478 +                        sprintf(pbuf, "Interpolating component '%s'", argv[i]);
479 +                        prog_start(pbuf);
480                          if (single_plane_incident)
481                                  eval_isotropic(NULL);
482                          else
# Line 422 | Line 489 | main(int argc, char *argv[])
489          if (!load_bsdf_rep(stdin))
490                  return(1);
491          xml_prologue(argc, argv);               /* start XML output */
492 +        prog_start("Interpolating from standard input");
493          if (single_plane_incident)              /* resample dist. */
494                  eval_isotropic(NULL);
495          else

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines