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.39 by greg, Thu Oct 13 16:59:29 2016 UTC vs.
Revision 2.64 by greg, Sat Jun 7 05:09:45 2025 UTC

# Line 8 | Line 8 | static const char RCSid[] = "$Id$";
8   */
9  
10   #define _USE_MATH_DEFINES
11 #include <stdio.h>
11   #include <stdlib.h>
12   #include <math.h>
13 + #include <ctype.h>
14   #include "random.h"
15 #include "platform.h"
16 #include "paths.h"
15   #include "rtio.h"
16   #include "calcomp.h"
17   #include "bsdfrep.h"
18 <                                /* global argv[0] */
19 < char                    *progname;
18 >                                /* reciprocity averaging option */
19 > static const char       *recip = "";
20                                  /* percentage to cull (<0 to turn off) */
21   static double           pctcull = 90.;
22                                  /* sampling order */
23   static int              samp_order = 6;
24                                  /* super-sampling threshold */
25 < const double            ssamp_thresh = 0.35;
25 > static double           ssamp_thresh = 0.35;
26                                  /* number of super-samples */
27 < #ifndef NSSAMP
30 < #define NSSAMP          64
31 < #endif
27 > static int              nssamp = 64;
28                                  /* limit on number of RBF lobes */
29   static int              lobe_lim = 15000;
30                                  /* progress bar length */
# Line 127 | Line 123 | static void
123   eval_isotropic(char *funame)
124   {
125          const int       sqres = 1<<samp_order;
126 +        const double    sqfact = 1./(double)sqres;
127 +        float           *val_last = NULL;
128 +        float           *val_next = NULL;
129 +        SDValue         *sdv_next = NULL;
130          FILE            *ofp, *uvfp[2];
131          int             assignD = 0;
132          char            cmd[128];
133          int             ix, ox, oy;
134 <        double          iovec[6];
134 >        RREAL           iovec[6];
135          float           bsdf, uv[2];
136  
137          if (pctcull >= 0) {
138 <                sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d > %s",
139 <                                pctcull, samp_order, create_component_file(0));
138 >                sprintf(cmd, "rttree_reduce%s -h -ff -r 3 -t %f -g %d > %s",
139 >                                recip, pctcull, samp_order, create_component_file(0));
140                  ofp = popen(cmd, "w");
141                  if (ofp == NULL) {
142                          fprintf(stderr, "%s: cannot create pipe to rttree_reduce\n",
# Line 149 | Line 149 | eval_isotropic(char *funame)
149   #endif
150                  if (rbf_colorimetry == RBCtristimulus) {
151                          double  uvcull = 100. - (100.-pctcull)*.25;
152 <                        sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d > %s",
153 <                                        uvcull, samp_order, create_component_file(1));
152 >                        sprintf(cmd, "rttree_reduce%s -h -ff -r 3 -t %f -g %d > %s",
153 >                                        recip, uvcull, samp_order, create_component_file(1));
154                          uvfp[0] = popen(cmd, "w");
155 <                        sprintf(cmd, "rttree_reduce -a -h -ff -r 3 -t %f -g %d > %s",
156 <                                        uvcull, samp_order, create_component_file(2));
155 >                        sprintf(cmd, "rttree_reduce%s -h -ff -r 3 -t %f -g %d > %s",
156 >                                        recip, uvcull, samp_order, create_component_file(2));
157                          uvfp[1] = popen(cmd, "w");
158                          if ((uvfp[0] == NULL) | (uvfp[1] == NULL)) {
159                                  fprintf(stderr, "%s: cannot open pipes to uv output\n",
# Line 187 | Line 187 | eval_isotropic(char *funame)
187          }
188          if (funame != NULL)                     /* need to assign Dx, Dy, Dz? */
189                  assignD = (fundefined(funame) < 6);
190 +        val_last = (float *)calloc(sqres, sizeof(float));
191 +        if (funame == NULL)
192 +                sdv_next = (SDValue *)malloc(sizeof(SDValue)*sqres);
193 +        else
194 +                val_next = (float *)malloc(sizeof(float)*sqres);
195                                                  /* run through directions */
196          for (ix = 0; ix < sqres/2; ix++) {
197 <                RBFNODE *rbf = NULL;
198 <                iovec[0] = 2.*(ix+.5)/sqres - 1.;
199 <                iovec[1] = .0;
200 <                iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0]);
197 >                const int       zipsgn = (ix & 1)*2 - 1;
198 >                RBFNODE         *rbf = NULL;
199 >                iovec[0] = 2.*sqfact*(ix+.5) - 1.;
200 >                iovec[1] = zipsgn*sqfact*.5;
201 >                iovec[2] = input_orient * sqrt(1. - iovec[0]*iovec[0]
202 >                                                - iovec[1]*iovec[1]);
203                  if (funame == NULL)
204                          rbf = advect_rbf(iovec, lobe_lim);
205 +                                                /* presample first row */
206 +                for (oy = 0; oy < sqres; oy++) {
207 +                    square2disk(iovec+3, .5*sqfact, (oy+.5)*sqfact);
208 +                    iovec[5] = output_orient *
209 +                                sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
210 +                    if (funame == NULL) {
211 +                        eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
212 +                    } else {
213 +                        if (assignD) {
214 +                            varset("Dx", '=', -iovec[3]);
215 +                            varset("Dy", '=', -iovec[4]);
216 +                            varset("Dz", '=', -iovec[5]);
217 +                            ++eclock;
218 +                        }
219 +                        val_next[oy] = funvalue(funame, 6, iovec);
220 +                    }
221 +                }
222                  for (ox = 0; ox < sqres; ox++) {
223 <                    float       last_bsdf = -1;
223 >                    /*
224 >                     * Super-sample when we detect a difference from before
225 >                     * or after in this row, above or below.
226 >                     */
227                      for (oy = 0; oy < sqres; oy++) {
228 <                        SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres);
229 <                        iovec[5] = output_orient *
228 >                        if (ox < sqres-1) {     /* keeping one row ahead... */
229 >                            square2disk(iovec+3, (ox+1.5)*sqfact, (oy+.5)*sqfact);
230 >                            iovec[5] = output_orient *
231                                  sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
232 +                        }
233                          if (funame == NULL) {
234 <                            SDValue     sdv;
206 <                            eval_rbfcol(&sdv, rbf, iovec+3);
234 >                            SDValue     sdv = sdv_next[oy];
235                              bsdf = sdv.cieY;
236 <                            if (rbf_colorimetry == RBCtristimulus) {
237 <                                c_ccvt(&sdv.spec, C_CSXY);
236 >                            if (ox < sqres-1)
237 >                                eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
238 >                            if (abs_diff(bsdf, sdv_next[oy].cieY) > ssamp_thresh ||
239 >                                (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
240 >                                (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
241 >                                (oy < sqres-1 &&
242 >                                    abs_diff(bsdf, sdv_next[oy+1].cieY) > ssamp_thresh)) {
243 >                                int     ssi;
244 >                                double  ssa[2], sum = 0, usum = 0, vsum = 0;
245 >                                                /* super-sample voxel */
246 >                                for (ssi = nssamp; ssi--; ) {
247 >                                    SDmultiSamp(ssa, 2, (ssi+frandom()) /
248 >                                                        (double)nssamp);
249 >                                    square2disk(iovec+3, (ox+ssa[0])*sqfact,
250 >                                                        (oy+ssa[1])*sqfact);
251 >                                    iovec[5] = output_orient *
252 >                                        sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
253 >                                    eval_rbfcol(&sdv, rbf, iovec+3);
254 >                                    sum += sdv.cieY;
255 >                                    if (rbf_colorimetry == RBCtristimulus) {
256 >                                        sdv.cieY /=
257 >                                            -2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.;
258 >                                        usum += 4.*sdv.spec.cx * sdv.cieY;
259 >                                        vsum += 9.*sdv.spec.cy * sdv.cieY;
260 >                                    }
261 >                                }
262 >                                bsdf = sum / (double)nssamp;
263 >                                if (rbf_colorimetry == RBCtristimulus) {
264 >                                    uv[0] = usum / (sum+FTINY);
265 >                                    uv[1] = vsum / (sum+FTINY);
266 >                                }
267 >                            } else if (rbf_colorimetry == RBCtristimulus) {
268                                  uv[0] = uv[1] = 1. /
269                                      (-2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.);
270                                  uv[0] *= 4.*sdv.spec.cx;
271                                  uv[1] *= 9.*sdv.spec.cy;
272                              }
273                          } else {
274 <                            if (assignD) {
275 <                                varset("Dx", '=', -iovec[3]);
276 <                                varset("Dy", '=', -iovec[4]);
277 <                                varset("Dz", '=', -iovec[5]);
278 <                                ++eclock;
274 >                            bsdf = val_next[oy];
275 >                            if (ox < sqres-1) {
276 >                                if (assignD) {
277 >                                    varset("Dx", '=', -iovec[3]);
278 >                                    varset("Dy", '=', -iovec[4]);
279 >                                    varset("Dz", '=', -iovec[5]);
280 >                                    ++eclock;
281 >                                }
282 >                                val_next[oy] = funvalue(funame, 6, iovec);
283                              }
284 <                            bsdf = funvalue(funame, 6, iovec);
285 < #if (NSSAMP > 0)
286 <                            if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
284 >                            if (abs_diff(bsdf, val_next[oy]) > ssamp_thresh ||
285 >                                (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
286 >                                (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
287 >                                (oy < sqres-1 &&
288 >                                    abs_diff(bsdf, val_next[oy+1]) > ssamp_thresh)) {
289                                  int     ssi;
290 <                                double  ssa[3], ssvec[6], sum = 0;
290 >                                double  ssa[4], ssvec[6], sum = 0;
291                                                  /* super-sample voxel */
292 <                                for (ssi = NSSAMP; ssi--; ) {
293 <                                    SDmultiSamp(ssa, 3, (ssi+frandom()) *
294 <                                                        (1./NSSAMP));
295 <                                    ssvec[0] = 2.*(ix+ssa[0])/sqres - 1.;
296 <                                    ssvec[1] = .0;
297 <                                    ssvec[2] = input_orient *
298 <                                                sqrt(1. - ssvec[0]*ssvec[0]);
299 <                                    SDsquare2disk(ssvec+3, (ox+ssa[1])/sqres,
300 <                                                (oy+ssa[2])/sqres);
292 >                                for (ssi = nssamp; ssi--; ) {
293 >                                    SDmultiSamp(ssa, 4, (ssi+frandom()) /
294 >                                                        (double)nssamp);
295 >                                    ssvec[0] = 2.*sqfact*(ix+ssa[0]) - 1.;
296 >                                    ssvec[1] = zipsgn*sqfact*ssa[1];
297 >                                    ssvec[2] = 1. - ssvec[0]*ssvec[0]
298 >                                                        - ssvec[1]*ssvec[1];
299 >                                    if (ssvec[2] < .0) {
300 >                                        ssvec[1] = 0;
301 >                                        ssvec[2] = 1. - ssvec[0]*ssvec[0];
302 >                                    }
303 >                                    ssvec[2] = input_orient * sqrt(ssvec[2]);
304 >                                    square2disk(ssvec+3, (ox+ssa[2])*sqfact,
305 >                                                (oy+ssa[3])*sqfact);
306                                      ssvec[5] = output_orient *
307                                                  sqrt(1. - ssvec[3]*ssvec[3] -
308                                                          ssvec[4]*ssvec[4]);
# Line 245 | Line 314 | eval_isotropic(char *funame)
314                                      }
315                                      sum += funvalue(funame, 6, ssvec);
316                                  }
317 <                                bsdf = sum/NSSAMP;
317 >                                bsdf = sum / (double)nssamp;
318                              }
250 #endif
319                          }
320                          if (pctcull >= 0)
321                                  putbinary(&bsdf, sizeof(bsdf), 1, ofp);
# Line 263 | Line 331 | eval_isotropic(char *funame)
331                                          fprintf(uvfp[1], "\t%.3e\n", uv[1]);
332                                  }
333                          }
334 <                        last_bsdf = bsdf;
334 >                        if (val_last != NULL)
335 >                                val_last[oy] = bsdf;
336                      }
337                  }
338                  if (rbf != NULL)
339                          free(rbf);
340 <                prog_show((ix+1.)*(2./sqres));
340 >                prog_show((ix+1.)*(2.*sqfact));
341          }
342          prog_done();
343 +        if (val_last != NULL) {
344 +                free(val_last);
345 +                if (val_next != NULL) free(val_next);
346 +                if (sdv_next != NULL) free(sdv_next);
347 +        }
348          if (pctcull >= 0) {                     /* finish output */
349                  if (pclose(ofp)) {
350                          fprintf(stderr, "%s: error running rttree_reduce on Y\n",
# Line 313 | Line 387 | static void
387   eval_anisotropic(char *funame)
388   {
389          const int       sqres = 1<<samp_order;
390 +        const double    sqfact = 1./(double)sqres;
391 +        float           *val_last = NULL;
392 +        float           *val_next = NULL;
393 +        SDValue         *sdv_next = NULL;
394          FILE            *ofp, *uvfp[2];
395          int             assignD = 0;
396          char            cmd[128];
397          int             ix, iy, ox, oy;
398 <        double          iovec[6];
398 >        RREAL           iovec[6];
399          float           bsdf, uv[2];
400  
401          if (pctcull >= 0) {
402                  const char      *avgopt = (input_orient>0 ^ output_orient>0)
403 <                                                ? "" : " -a";
403 >                                                ? "" : recip;
404                  sprintf(cmd, "rttree_reduce%s -h -ff -r 4 -t %f -g %d > %s",
405                                  avgopt, pctcull, samp_order,
406                                  create_component_file(0));
# Line 378 | Line 456 | eval_anisotropic(char *funame)
456          }
457          if (funame != NULL)                     /* need to assign Dx, Dy, Dz? */
458                  assignD = (fundefined(funame) < 6);
459 +        val_last = (float *)calloc(sqres, sizeof(float));
460 +        if (funame == NULL)
461 +                sdv_next = (SDValue *)malloc(sizeof(SDValue)*sqres);
462 +        else
463 +                val_next = (float *)malloc(sizeof(float)*sqres);
464                                                  /* run through directions */
465          for (ix = 0; ix < sqres; ix++)
466              for (iy = 0; iy < sqres; iy++) {
467                  RBFNODE *rbf = NULL;            /* Klems reversal */
468 <                SDsquare2disk(iovec, 1.-(ix+.5)/sqres, 1.-(iy+.5)/sqres);
468 >                square2disk(iovec, 1.-(ix+.5)*sqfact, 1.-(iy+.5)*sqfact);
469                  iovec[2] = input_orient *
470                                  sqrt(1. - iovec[0]*iovec[0] - iovec[1]*iovec[1]);
471                  if (funame == NULL)
472                          rbf = advect_rbf(iovec, lobe_lim);
473 +                                                /* presample first row */
474 +                for (oy = 0; oy < sqres; oy++) {
475 +                    square2disk(iovec+3, .5*sqfact, (oy+.5)*sqfact);
476 +                    iovec[5] = output_orient *
477 +                                sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
478 +                    if (funame == NULL) {
479 +                        eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
480 +                    } else {
481 +                        if (assignD) {
482 +                            varset("Dx", '=', -iovec[3]);
483 +                            varset("Dy", '=', -iovec[4]);
484 +                            varset("Dz", '=', -iovec[5]);
485 +                            ++eclock;
486 +                        }
487 +                        val_next[oy] = funvalue(funame, 6, iovec);
488 +                    }
489 +                }
490                  for (ox = 0; ox < sqres; ox++) {
491 <                    float       last_bsdf = -1;
491 >                    /*
492 >                     * Super-sample when we detect a difference from before
493 >                     * or after in this row, above or below.
494 >                     */
495                      for (oy = 0; oy < sqres; oy++) {
496 <                        SDsquare2disk(iovec+3, (ox+.5)/sqres, (oy+.5)/sqres);
497 <                        iovec[5] = output_orient *
496 >                        if (ox < sqres-1) {     /* keeping one row ahead... */
497 >                            square2disk(iovec+3, (ox+1.5)*sqfact, (oy+.5)*sqfact);
498 >                            iovec[5] = output_orient *
499                                  sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
500 +                        }
501                          if (funame == NULL) {
502 <                            SDValue     sdv;
398 <                            eval_rbfcol(&sdv, rbf, iovec+3);
502 >                            SDValue     sdv = sdv_next[oy];
503                              bsdf = sdv.cieY;
504 <                            if (rbf_colorimetry == RBCtristimulus) {
505 <                                c_ccvt(&sdv.spec, C_CSXY);
504 >                            if (ox < sqres-1)
505 >                                eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
506 >                            if (abs_diff(bsdf, sdv_next[oy].cieY) > ssamp_thresh ||
507 >                                (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
508 >                                (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
509 >                                (oy < sqres-1 &&
510 >                                    abs_diff(bsdf, sdv_next[oy+1].cieY) > ssamp_thresh)) {
511 >                                int     ssi;
512 >                                double  ssa[2], sum = 0, usum = 0, vsum = 0;
513 >                                                /* super-sample voxel */
514 >                                for (ssi = nssamp; ssi--; ) {
515 >                                    SDmultiSamp(ssa, 2, (ssi+frandom()) /
516 >                                                        (double)nssamp);
517 >                                    square2disk(iovec+3, (ox+ssa[0])*sqfact,
518 >                                                        (oy+ssa[1])*sqfact);
519 >                                    iovec[5] = output_orient *
520 >                                        sqrt(1. - iovec[3]*iovec[3] - iovec[4]*iovec[4]);
521 >                                    eval_rbfcol(&sdv, rbf, iovec+3);
522 >                                    sum += sdv.cieY;
523 >                                    if (rbf_colorimetry == RBCtristimulus) {
524 >                                        sdv.cieY /=
525 >                                            -2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.;
526 >                                        usum += 4.*sdv.spec.cx * sdv.cieY;
527 >                                        vsum += 9.*sdv.spec.cy * sdv.cieY;
528 >                                    }
529 >                                }
530 >                                bsdf = sum / (double)nssamp;
531 >                                if (rbf_colorimetry == RBCtristimulus) {
532 >                                    uv[0] = usum / (sum+FTINY);
533 >                                    uv[1] = vsum / (sum+FTINY);
534 >                                }
535 >                            } else if (rbf_colorimetry == RBCtristimulus) {
536                                  uv[0] = uv[1] = 1. /
537                                      (-2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.);
538                                  uv[0] *= 4.*sdv.spec.cx;
539                                  uv[1] *= 9.*sdv.spec.cy;
540                              }
541                          } else {
542 <                            if (assignD) {
543 <                                varset("Dx", '=', -iovec[3]);
544 <                                varset("Dy", '=', -iovec[4]);
545 <                                varset("Dz", '=', -iovec[5]);
546 <                                ++eclock;
542 >                            bsdf = val_next[oy];
543 >                            if (ox < sqres-1) {
544 >                                if (assignD) {
545 >                                    varset("Dx", '=', -iovec[3]);
546 >                                    varset("Dy", '=', -iovec[4]);
547 >                                    varset("Dz", '=', -iovec[5]);
548 >                                    ++eclock;
549 >                                }
550 >                                val_next[oy] = funvalue(funame, 6, iovec);
551                              }
552 <                            bsdf = funvalue(funame, 6, iovec);
553 < #if (NSSAMP > 0)
554 <                            if (abs_diff(bsdf, last_bsdf) > ssamp_thresh) {
552 >                            if (abs_diff(bsdf, val_next[oy]) > ssamp_thresh ||
553 >                                (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
554 >                                (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
555 >                                (oy < sqres-1 &&
556 >                                    abs_diff(bsdf, val_next[oy+1]) > ssamp_thresh)) {
557                                  int     ssi;
558                                  double  ssa[4], ssvec[6], sum = 0;
559                                                  /* super-sample voxel */
560 <                                for (ssi = NSSAMP; ssi--; ) {
561 <                                    SDmultiSamp(ssa, 4, (ssi+frandom()) *
562 <                                                        (1./NSSAMP));
563 <                                    SDsquare2disk(ssvec, 1.-(ix+ssa[0])/sqres,
564 <                                                1.-(iy+ssa[1])/sqres);
560 >                                for (ssi = nssamp; ssi--; ) {
561 >                                    SDmultiSamp(ssa, 4, (ssi+frandom()) /
562 >                                                        (double)nssamp);
563 >                                    square2disk(ssvec, 1.-(ix+ssa[0])*sqfact,
564 >                                                1.-(iy+ssa[1])*sqfact);
565                                      ssvec[2] = input_orient *
566                                                  sqrt(1. - ssvec[0]*ssvec[0] -
567                                                          ssvec[1]*ssvec[1]);
568 <                                    SDsquare2disk(ssvec+3, (ox+ssa[2])/sqres,
569 <                                                (oy+ssa[3])/sqres);
568 >                                    square2disk(ssvec+3, (ox+ssa[2])*sqfact,
569 >                                                (oy+ssa[3])*sqfact);
570                                      ssvec[5] = output_orient *
571                                                  sqrt(1. - ssvec[3]*ssvec[3] -
572                                                          ssvec[4]*ssvec[4]);
# Line 438 | Line 578 | eval_anisotropic(char *funame)
578                                      }
579                                      sum += funvalue(funame, 6, ssvec);
580                                  }
581 <                                bsdf = sum/NSSAMP;
581 >                                bsdf = sum / (double)nssamp;
582                              }
443 #endif
583                          }
584                          if (pctcull >= 0)
585                                  putbinary(&bsdf, sizeof(bsdf), 1, ofp);
# Line 456 | Line 595 | eval_anisotropic(char *funame)
595                                          fprintf(uvfp[1], "\t%.3e\n", uv[1]);
596                                  }
597                          }
598 <                        last_bsdf = bsdf;
598 >                        if (val_last != NULL)
599 >                                val_last[oy] = bsdf;
600                      }
601                  }
602                  if (rbf != NULL)
# Line 464 | Line 604 | eval_anisotropic(char *funame)
604                  prog_show((ix*sqres+iy+1.)/(sqres*sqres));
605              }
606          prog_done();
607 +        if (val_last != NULL) {
608 +                free(val_last);
609 +                if (val_next != NULL) free(val_next);
610 +                if (sdv_next != NULL) free(sdv_next);
611 +        }
612          if (pctcull >= 0) {                     /* finish output */
613                  if (pclose(ofp)) {
614                          fprintf(stderr, "%s: error running rttree_reduce on Y\n",
# Line 500 | Line 645 | eval_anisotropic(char *funame)
645   static int
646   wrap_up(void)
647   {
648 <        char    cmd[8192];
648 >        char    cmd[32700];
649  
650          if (bsdf_manuf[0]) {
651                  add_wbsdf("-f", 1);
# Line 551 | Line 696 | wrap_up(void)
696   }
697   #endif
698  
699 + #define HEAD_BUFLEN     10240
700 + static char     head_buf[HEAD_BUFLEN];
701 + static int      cur_headlen = 0;
702 +
703 + /* Record header line as comment associated with this SIR input */
704 + static int
705 + record2header(char *s)
706 + {
707 +        int     len = strlen(s);
708 +
709 +        if (cur_headlen+len >= HEAD_BUFLEN-6)
710 +                return(0);
711 +                                        /* includes EOL */
712 +        strcpy(head_buf+cur_headlen, s);
713 +        cur_headlen += len;
714 +
715 + #if defined(_WIN32) || defined(_WIN64)
716 +        if (head_buf[cur_headlen-1] == '\n')
717 +                head_buf[cur_headlen-1] = '\t';
718 + #endif
719 +        return(1);
720 + }
721 +
722 + /* Finish off header for this file */
723 + static void
724 + done_header(void)
725 + {
726 +        while (cur_headlen > 0 && isspace(head_buf[cur_headlen-1]))
727 +                --cur_headlen;
728 +        head_buf[cur_headlen] = '\0';
729 +        if (!cur_headlen)
730 +                return;
731 +        add_wbsdf("-C", 1);
732 +        add_wbsdf(head_buf, 0);
733 +        head_buf[cur_headlen=0] = '\0';
734 + }
735 +
736   /* Read in BSDF and interpolate as tensor tree representation */
737   int
738   main(int argc, char *argv[])
739   {
740 <        static char     tfmt[2][4] = {"t4", "t3"};
741 <        int     dofwd = 0, dobwd = 1;
742 <        char    buf[2048];
743 <        int     i, na;
744 <
745 <        progname = argv[0];
740 >        static const char       tfmt[2][4] = {"t4", "t3"};
741 >        int                     dofwd = 0, dobwd = 1;
742 >        int                     nsirs = 0;
743 >        char                    buf[1024];
744 >        int                     i;
745 >                                                /* set global progname */
746 >        fixargv0(argv[0]);
747          esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
748          esupport &= ~(E_INCHAN|E_OUTCHAN);
749          scompile("PI:3.14159265358979323846", NULL, 0);
750          biggerlib();
751 <        for (i = 1; i < argc-1 && (argv[i][0] == '-') | (argv[i][0] == '+'); i++)
752 <                switch (argv[i][1]) {           /* get options */
751 >        strcpy(buf, "File produced by: ");
752 >        if (convert_commandline(buf+18, sizeof(buf)-18, argv) != NULL) {
753 >                add_wbsdf("-C", 1); add_wbsdf(buf, 0);
754 >        }
755 >        for (i = 1; i < argc; i++)
756 >            if ((argv[i][0] == '-') | (argv[i][0] == '+')) {
757 >                switch (argv[i][1]) {           /* get option */
758                  case 'e':
759                          scompile(argv[++i], NULL, 0);
760 +                        if (single_plane_incident < 0)
761 +                                single_plane_incident = 0;
762                          break;
763                  case 'f':
764 <                        if (!argv[i][2]) {
764 >                        if ((argv[i][0] == '-') & !argv[i][2]) {
765                                  if (strchr(argv[++i], '=') != NULL) {
766                                          add_wbsdf("-f", 1);
767                                          add_wbsdf(argv[i], 1);
768 <                                } else
769 <                                        fcompile(argv[i]);
768 >                                } else {
769 >                                        char    *fpath = getpath(argv[i],
770 >                                                            getrlibpath(), 0);
771 >                                        if (fpath == NULL) {
772 >                                                fprintf(stderr,
773 >                                                "%s: cannot find file '%s'\n",
774 >                                                        argv[0], argv[i]);
775 >                                                return(1);
776 >                                        }
777 >                                        fcompile(fpath);
778 >                                        if (single_plane_incident < 0)
779 >                                                single_plane_incident = 0;
780 >                                }
781                          } else
782                                  dofwd = (argv[i][0] == '+');
783                          break;
784 +                case 'a':
785 +                        recip = (argv[i][0] == '+') ? " -a" : "";
786 +                        break;
787                  case 'b':
788                          dobwd = (argv[i][0] == '+');
789                          break;
790 +                case 'n':
791 +                        nssamp = atoi(argv[++i]);
792 +                        if (nssamp <= 0)
793 +                                goto userr;
794 +                        break;
795 +                case 's':
796 +                        ssamp_thresh = atof(argv[++i]);
797 +                        if (ssamp_thresh <= FTINY)
798 +                                goto userr;
799 +                        break;
800                  case 't':
801                          switch (argv[i][2]) {
802                          case '3':
# Line 618 | Line 832 | main(int argc, char *argv[])
832                  default:
833                          goto userr;
834                  }
835 <        strcpy(buf, "File produced by: ");
836 <        if (convert_commandline(buf+18, sizeof(buf)-18, argv) != NULL) {
837 <                add_wbsdf("-C", 1); add_wbsdf(buf, 0);
838 <        }
839 <        if (single_plane_incident >= 0) {       /* function-based BSDF? */
835 >            } else {                            /* input SIR or function */
836 >                FILE    *fpin;
837 >                if (!nsirs & (single_plane_incident >= 0))
838 >                        break;                  /* must be a function */
839 >                if (nsirs >= 4) {
840 >                        fprintf(stderr, "At most 4 SIR inputs supported\n");
841 >                        goto userr;
842 >                }
843 >                fpin = fopen(argv[i], "rb");    /* open SIR file */
844 >                if (fpin == NULL) {
845 >                        fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
846 >                                        progname, argv[i]);
847 >                        return(1);
848 >                }
849 >                sprintf(buf, "%s:\n", argv[i]);
850 >                record2header(buf);
851 >                sir_headshare = &record2header;
852 >                if (!load_bsdf_rep(fpin))
853 >                        return(1);
854 >                fclose(fpin);
855 >                done_header();
856 >                sprintf(buf, "Interpolating component '%s'", argv[i]);
857 >                prog_start(buf);
858 >                if (!nsirs++) {
859 >                        add_wbsdf("-a", 1);
860 >                        add_wbsdf(tfmt[single_plane_incident], 1);
861 >                }
862 >                if (single_plane_incident)
863 >                        eval_isotropic(NULL);
864 >                else
865 >                        eval_anisotropic(NULL);
866 >            }
867 >        if (i < argc) {                         /* function-based BSDF? */
868                  void    (*evf)(char *s) = single_plane_incident ?
869                                  eval_isotropic : eval_anisotropic;
870                  if (i != argc-1 || fundefined(argv[i]) < 3) {
# Line 632 | Line 874 | main(int argc, char *argv[])
874                          fprintf(stderr, "\tor 3 arguments using Dx,Dy,Dz: bsdf(ix,iy,iz)\n");
875                          goto userr;
876                  }
877 +                doptimize(1);                   /* optimize definitions */
878                  ++eclock;
879                  add_wbsdf("-a", 1);
880                  add_wbsdf(tfmt[single_plane_incident], 1);
# Line 653 | Line 896 | main(int argc, char *argv[])
896                          prog_start("Evaluating inside->outside transmission");
897                          (*evf)(argv[i]);
898                  }
899 <                return(wrap_up());
899 >        } else if (!nsirs) {                    /* else load SIR from stdin? */
900 >                record2header("<stdin>:\n");
901 >                sir_headshare = &record2header;
902 >                if (!load_bsdf_rep(stdin))
903 >                        return(1);
904 >                done_header();
905 >                prog_start("Interpolating from standard input");
906 >                add_wbsdf("-a", 1);
907 >                add_wbsdf(tfmt[single_plane_incident], 1);
908 >                if (single_plane_incident)      /* resample dist. */
909 >                        eval_isotropic(NULL);
910 >                else
911 >                        eval_anisotropic(NULL);
912          }
913 <        if (i < argc) {                         /* open input files if given */
659 <                int     nbsdf = 0;
660 <                for ( ; i < argc; i++) {        /* interpolate each component */
661 <                        char    pbuf[256];
662 <                        FILE    *fpin = fopen(argv[i], "rb");
663 <                        if (fpin == NULL) {
664 <                                fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
665 <                                                progname, argv[i]);
666 <                                return(1);
667 <                        }
668 <                        if (!load_bsdf_rep(fpin))
669 <                                return(1);
670 <                        fclose(fpin);
671 <                        sprintf(pbuf, "Interpolating component '%s'", argv[i]);
672 <                        prog_start(pbuf);
673 <                        if (!nbsdf++) {
674 <                                add_wbsdf("-a", 1);
675 <                                add_wbsdf(tfmt[single_plane_incident], 1);
676 <                        }
677 <                        if (single_plane_incident)
678 <                                eval_isotropic(NULL);
679 <                        else
680 <                                eval_anisotropic(NULL);
681 <                }
682 <                return(wrap_up());
683 <        }
684 <        SET_FILE_BINARY(stdin);                 /* load from stdin */
685 <        if (!load_bsdf_rep(stdin))
686 <                return(1);
687 <        prog_start("Interpolating from standard input");
688 <        add_wbsdf("-a", 1);
689 <        add_wbsdf(tfmt[single_plane_incident], 1);
690 <        if (single_plane_incident)              /* resample dist. */
691 <                eval_isotropic(NULL);
692 <        else
693 <                eval_anisotropic(NULL);
694 <
695 <        return(wrap_up());
913 >        return(wrap_up());                      /* call wrapBSDF */
914   userr:
915          fprintf(stderr,
916 <        "Usage: %s [-g Nlog2][-t pctcull][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n",
916 >        "Usage: %s [{+|-}a][-g Nlog2][-t pctcull][-n nss][-s thresh][-l maxlobes][bsdf.sir ..] > bsdf.xml\n",
917                                  progname);
918          fprintf(stderr,
919 <        "   or: %s -t{3|4} [-g Nlog2][-t pctcull][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n",
919 >        "   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",
920                                  progname);
921          return(1);
922   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines