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.50 by greg, Mon May 11 20:26:58 2020 UTC vs.
Revision 2.56 by greg, Fri Nov 13 19:21:11 2020 UTC

# Line 10 | Line 10 | static const char RCSid[] = "$Id$";
10   #define _USE_MATH_DEFINES
11   #include <stdlib.h>
12   #include <math.h>
13 + #include <ctype.h>
14   #include "random.h"
15   #include "platform.h"
16   #include "paths.h"
# Line 27 | Line 28 | static int             samp_order = 6;
28                                  /* super-sampling threshold */
29   static double           ssamp_thresh = 0.35;
30                                  /* number of super-samples */
31 < #ifndef NSSAMP
31 < #define NSSAMP          256
32 < #endif
31 > static int              nssamp = 256;
32                                  /* limit on number of RBF lobes */
33   static int              lobe_lim = 15000;
34                                  /* progress bar length */
# Line 192 | Line 191 | eval_isotropic(char *funame)
191          }
192          if (funame != NULL)                     /* need to assign Dx, Dy, Dz? */
193                  assignD = (fundefined(funame) < 6);
195 – #if (NSSAMP > 0)
194          val_last = (float *)calloc(sqres, sizeof(float));
195          if (funame == NULL)
196                  sdv_next = (SDValue *)malloc(sizeof(SDValue)*sqres);
197          else
198                  val_next = (float *)malloc(sizeof(float)*sqres);
201 – #endif
199                                                  /* run through directions */
200          for (ix = 0; ix < sqres/2; ix++) {
201                  const int       zipsgn = (ix & 1)*2 - 1;
# Line 242 | Line 239 | eval_isotropic(char *funame)
239                              bsdf = sdv.cieY;
240                              if (ox < sqres-1)
241                                  eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
245 – #if (NSSAMP > 0)
242                              if (abs_diff(bsdf, sdv_next[oy].cieY) > ssamp_thresh ||
243                                  (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
244                                  (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
# Line 251 | Line 247 | eval_isotropic(char *funame)
247                                  int     ssi;
248                                  double  ssa[2], sum = 0, usum = 0, vsum = 0;
249                                                  /* super-sample voxel */
250 <                                for (ssi = NSSAMP; ssi--; ) {
251 <                                    SDmultiSamp(ssa, 2, (ssi+frandom()) *
252 <                                                        (1./NSSAMP));
250 >                                for (ssi = nssamp; ssi--; ) {
251 >                                    SDmultiSamp(ssa, 2, (ssi+frandom()) /
252 >                                                        (double)nssamp);
253                                      SDsquare2disk(iovec+3, (ox+ssa[0])*sqfact,
254                                                          (oy+ssa[1])*sqfact);
255                                      iovec[5] = output_orient *
# Line 267 | Line 263 | eval_isotropic(char *funame)
263                                          vsum += 9.*sdv.spec.cy * sdv.cieY;
264                                      }
265                                  }
266 <                                bsdf = sum * (1./NSSAMP);
266 >                                bsdf = sum / (double)nssamp;
267                                  if (rbf_colorimetry == RBCtristimulus) {
268                                      uv[0] = usum / (sum+FTINY);
269                                      uv[1] = vsum / (sum+FTINY);
270                                  }
271                              } else
276 – #endif
272                              if (rbf_colorimetry == RBCtristimulus) {
273                                  uv[0] = uv[1] = 1. /
274                                      (-2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.);
# Line 291 | Line 286 | eval_isotropic(char *funame)
286                                  }
287                                  val_next[oy] = funvalue(funame, 6, iovec);
288                              }
294 – #if (NSSAMP > 0)
289                              if (abs_diff(bsdf, val_next[oy]) > ssamp_thresh ||
290                                  (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
291                                  (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
# Line 300 | Line 294 | eval_isotropic(char *funame)
294                                  int     ssi;
295                                  double  ssa[4], ssvec[6], sum = 0;
296                                                  /* super-sample voxel */
297 <                                for (ssi = NSSAMP; ssi--; ) {
298 <                                    SDmultiSamp(ssa, 4, (ssi+frandom()) *
299 <                                                        (1./NSSAMP));
297 >                                for (ssi = nssamp; ssi--; ) {
298 >                                    SDmultiSamp(ssa, 4, (ssi+frandom()) /
299 >                                                        (double)nssamp);
300                                      ssvec[0] = 2.*sqfact*(ix+ssa[0]) - 1.;
301                                      ssvec[1] = zipsgn*sqfact*ssa[1];
302                                      ssvec[2] = 1. - ssvec[0]*ssvec[0]
# Line 325 | Line 319 | eval_isotropic(char *funame)
319                                      }
320                                      sum += funvalue(funame, 6, ssvec);
321                                  }
322 <                                bsdf = sum * (1./NSSAMP);
322 >                                bsdf = sum / (double)nssamp;
323                              }
330 – #endif
324                          }
325                          if (pctcull >= 0)
326                                  putbinary(&bsdf, sizeof(bsdf), 1, ofp);
# Line 468 | Line 461 | eval_anisotropic(char *funame)
461          }
462          if (funame != NULL)                     /* need to assign Dx, Dy, Dz? */
463                  assignD = (fundefined(funame) < 6);
471 – #if (NSSAMP > 0)
464          val_last = (float *)calloc(sqres, sizeof(float));
465          if (funame == NULL)
466                  sdv_next = (SDValue *)malloc(sizeof(SDValue)*sqres);
467          else
468                  val_next = (float *)malloc(sizeof(float)*sqres);
477 – #endif
469                                                  /* run through directions */
470          for (ix = 0; ix < sqres; ix++)
471              for (iy = 0; iy < sqres; iy++) {
# Line 517 | Line 508 | eval_anisotropic(char *funame)
508                              bsdf = sdv.cieY;
509                              if (ox < sqres-1)
510                                  eval_rbfcol(&sdv_next[oy], rbf, iovec+3);
520 – #if (NSSAMP > 0)
511                              if (abs_diff(bsdf, sdv_next[oy].cieY) > ssamp_thresh ||
512                                  (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
513                                  (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
# Line 526 | Line 516 | eval_anisotropic(char *funame)
516                                  int     ssi;
517                                  double  ssa[2], sum = 0, usum = 0, vsum = 0;
518                                                  /* super-sample voxel */
519 <                                for (ssi = NSSAMP; ssi--; ) {
520 <                                    SDmultiSamp(ssa, 2, (ssi+frandom()) *
521 <                                                        (1./NSSAMP));
519 >                                for (ssi = nssamp; ssi--; ) {
520 >                                    SDmultiSamp(ssa, 2, (ssi+frandom()) /
521 >                                                        (double)nssamp);
522                                      SDsquare2disk(iovec+3, (ox+ssa[0])*sqfact,
523                                                          (oy+ssa[1])*sqfact);
524                                      iovec[5] = output_orient *
# Line 542 | Line 532 | eval_anisotropic(char *funame)
532                                          vsum += 9.*sdv.spec.cy * sdv.cieY;
533                                      }
534                                  }
535 <                                bsdf = sum * (1./NSSAMP);
535 >                                bsdf = sum / (double)nssamp;
536                                  if (rbf_colorimetry == RBCtristimulus) {
537                                      uv[0] = usum / (sum+FTINY);
538                                      uv[1] = vsum / (sum+FTINY);
539                                  }
540                              } else
551 – #endif
541                              if (rbf_colorimetry == RBCtristimulus) {
542                                  uv[0] = uv[1] = 1. /
543                                      (-2.*sdv.spec.cx + 12.*sdv.spec.cy + 3.);
# Line 566 | Line 555 | eval_anisotropic(char *funame)
555                                  }
556                                  val_next[oy] = funvalue(funame, 6, iovec);
557                              }
569 – #if (NSSAMP > 0)
558                              if (abs_diff(bsdf, val_next[oy]) > ssamp_thresh ||
559                                  (ox && abs_diff(bsdf, val_last[oy]) > ssamp_thresh) ||
560                                  (oy && abs_diff(bsdf, val_last[oy-1]) > ssamp_thresh) ||
# Line 575 | Line 563 | eval_anisotropic(char *funame)
563                                  int     ssi;
564                                  double  ssa[4], ssvec[6], sum = 0;
565                                                  /* super-sample voxel */
566 <                                for (ssi = NSSAMP; ssi--; ) {
567 <                                    SDmultiSamp(ssa, 4, (ssi+frandom()) *
568 <                                                        (1./NSSAMP));
566 >                                for (ssi = nssamp; ssi--; ) {
567 >                                    SDmultiSamp(ssa, 4, (ssi+frandom()) /
568 >                                                        (double)nssamp);
569                                      SDsquare2disk(ssvec, 1.-(ix+ssa[0])*sqfact,
570                                                  1.-(iy+ssa[1])*sqfact);
571                                      ssvec[2] = input_orient *
# Line 596 | Line 584 | eval_anisotropic(char *funame)
584                                      }
585                                      sum += funvalue(funame, 6, ssvec);
586                                  }
587 <                                bsdf = sum * (1./NSSAMP);
587 >                                bsdf = sum / (double)nssamp;
588                              }
601 – #endif
589                          }
590                          if (pctcull >= 0)
591                                  putbinary(&bsdf, sizeof(bsdf), 1, ofp);
# Line 617 | Line 604 | eval_anisotropic(char *funame)
604                          if (val_last != NULL)
605                                  val_last[oy] = bsdf;
606                      }
607 +                }
608                  if (rbf != NULL)
609                          free(rbf);
610                  prog_show((ix*sqres+iy+1.)/(sqres*sqres));
611              }
624 –        }
612          prog_done();
613          if (val_last != NULL) {
614                  free(val_last);
# Line 664 | Line 651 | eval_anisotropic(char *funame)
651   static int
652   wrap_up(void)
653   {
654 <        char    cmd[8192];
654 >        char    cmd[32700];
655  
656          if (bsdf_manuf[0]) {
657                  add_wbsdf("-f", 1);
# Line 715 | Line 702 | wrap_up(void)
702   }
703   #endif
704  
705 + #define HEAD_BUFLEN     10240
706 + static char     head_buf[HEAD_BUFLEN];
707 + static int      cur_headlen = 0;
708 +
709 + /* Record header line as comment associated with this SIR input */
710 + static int
711 + record2header(char *s)
712 + {
713 +        int     len = strlen(s);
714 +
715 +        if (cur_headlen+len >= HEAD_BUFLEN-6)
716 +                return(0);
717 +                                        /* includes EOL */
718 +        strcpy(head_buf+cur_headlen, s);
719 +        cur_headlen += len;
720 +
721 + #if defined(_WIN32) || defined(_WIN64)
722 +        if (head_buf[cur_headlen-1] == '\n')
723 +                head_buf[cur_headlen-1] = '\t';
724 + #endif
725 +        return(1);
726 + }
727 +
728 + /* Finish off header for this file */
729 + static void
730 + done_header(void)
731 + {
732 +        while (cur_headlen > 0 && isspace(head_buf[cur_headlen-1]))
733 +                --cur_headlen;
734 +        head_buf[cur_headlen] = '\0';
735 +        if (!cur_headlen)
736 +                return;
737 +        add_wbsdf("-C", 1);
738 +        add_wbsdf(head_buf, 0);
739 +        head_buf[cur_headlen=0] = '\0';
740 + }
741 +
742   /* Read in BSDF and interpolate as tensor tree representation */
743   int
744   main(int argc, char *argv[])
745   {
746 <        static char     tfmt[2][4] = {"t4", "t3"};
747 <        int     dofwd = 0, dobwd = 1;
748 <        char    buf[2048];
749 <        int     i, na;
746 >        static const char       tfmt[2][4] = {"t4", "t3"};
747 >        int                     dofwd = 0, dobwd = 1;
748 >        int                     nsirs = 0;
749 >        char                    buf[1024];
750 >        int                     i;
751  
752          progname = argv[0];
753          esupport |= E_VARIABLE|E_FUNCTION|E_RCONST;
754          esupport &= ~(E_INCHAN|E_OUTCHAN);
755          scompile("PI:3.14159265358979323846", NULL, 0);
756          biggerlib();
757 <        for (i = 1; i < argc && (argv[i][0] == '-') | (argv[i][0] == '+'); i++)
758 <                switch (argv[i][1]) {           /* get options */
757 >        strcpy(buf, "File produced by: ");
758 >        if (convert_commandline(buf+18, sizeof(buf)-18, argv) != NULL) {
759 >                add_wbsdf("-C", 1); add_wbsdf(buf, 0);
760 >        }
761 >        for (i = 1; i < argc; i++)
762 >            if ((argv[i][0] == '-') | (argv[i][0] == '+')) {
763 >                switch (argv[i][1]) {           /* get option */
764                  case 'e':
765                          scompile(argv[++i], NULL, 0);
766                          if (single_plane_incident < 0)
# Line 763 | Line 793 | main(int argc, char *argv[])
793                  case 'b':
794                          dobwd = (argv[i][0] == '+');
795                          break;
796 +                case 'n':
797 +                        nssamp = atoi(argv[++i]);
798 +                        if (nssamp <= 0)
799 +                                goto userr;
800 +                        break;
801 +                case 's':
802 +                        ssamp_thresh = atof(argv[++i]);
803 +                        if (ssamp_thresh <= FTINY)
804 +                                goto userr;
805 +                        break;
806                  case 't':
807                          switch (argv[i][2]) {
808                          case '3':
# Line 798 | Line 838 | main(int argc, char *argv[])
838                  default:
839                          goto userr;
840                  }
841 <        strcpy(buf, "File produced by: ");
842 <        if (convert_commandline(buf+18, sizeof(buf)-18, argv) != NULL) {
843 <                add_wbsdf("-C", 1); add_wbsdf(buf, 0);
844 <        }
845 <        if (single_plane_incident >= 0) {       /* function-based BSDF? */
841 >            } else {                            /* input SIR or function */
842 >                FILE    *fpin;
843 >                if (!nsirs & (single_plane_incident >= 0))
844 >                        break;                  /* must be a function */
845 >                if (nsirs >= 4) {
846 >                        fprintf(stderr, "At most 4 SIR inputs supported\n");
847 >                        goto userr;
848 >                }
849 >                fpin = fopen(argv[i], "rb");    /* open SIR file */
850 >                if (fpin == NULL) {
851 >                        fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
852 >                                        progname, argv[i]);
853 >                        return(1);
854 >                }
855 >                sprintf(buf, "%s:\n", argv[i]);
856 >                record2header(buf);
857 >                sir_headshare = &record2header;
858 >                if (!load_bsdf_rep(fpin))
859 >                        return(1);
860 >                fclose(fpin);
861 >                done_header();
862 >                sprintf(buf, "Interpolating component '%s'", argv[i]);
863 >                prog_start(buf);
864 >                if (!nsirs++) {
865 >                        add_wbsdf("-a", 1);
866 >                        add_wbsdf(tfmt[single_plane_incident], 1);
867 >                }
868 >                if (single_plane_incident)
869 >                        eval_isotropic(NULL);
870 >                else
871 >                        eval_anisotropic(NULL);
872 >            }
873 >        if (i < argc) {                         /* function-based BSDF? */
874                  void    (*evf)(char *s) = single_plane_incident ?
875                                  eval_isotropic : eval_anisotropic;
876                  if (i != argc-1 || fundefined(argv[i]) < 3) {
# Line 813 | Line 881 | main(int argc, char *argv[])
881                          goto userr;
882                  }
883                  ++eclock;
816 –                ssamp_thresh *= 0.5;            /* lower sampling threshold */
884                  add_wbsdf("-a", 1);
885                  add_wbsdf(tfmt[single_plane_incident], 1);
886                  if (dofwd) {
# Line 834 | Line 901 | main(int argc, char *argv[])
901                          prog_start("Evaluating inside->outside transmission");
902                          (*evf)(argv[i]);
903                  }
904 <                return(wrap_up());
904 >        } else if (!nsirs) {                    /* else load SIR from stdin? */
905 >                record2header("<stdin>:\n");
906 >                sir_headshare = &record2header;
907 >                if (!load_bsdf_rep(stdin))
908 >                        return(1);
909 >                done_header();
910 >                prog_start("Interpolating from standard input");
911 >                add_wbsdf("-a", 1);
912 >                add_wbsdf(tfmt[single_plane_incident], 1);
913 >                if (single_plane_incident)      /* resample dist. */
914 >                        eval_isotropic(NULL);
915 >                else
916 >                        eval_anisotropic(NULL);
917          }
918 <        if (i < argc) {                         /* open input files if given */
840 <                int     nbsdf = 0;
841 <                for ( ; i < argc; i++) {        /* interpolate each component */
842 <                        char    pbuf[256];
843 <                        FILE    *fpin = fopen(argv[i], "rb");
844 <                        if (fpin == NULL) {
845 <                                fprintf(stderr, "%s: cannot open BSDF interpolant '%s'\n",
846 <                                                progname, argv[i]);
847 <                                return(1);
848 <                        }
849 <                        if (!load_bsdf_rep(fpin))
850 <                                return(1);
851 <                        fclose(fpin);
852 <                        sprintf(pbuf, "Interpolating component '%s'", argv[i]);
853 <                        prog_start(pbuf);
854 <                        if (!nbsdf++) {
855 <                                add_wbsdf("-a", 1);
856 <                                add_wbsdf(tfmt[single_plane_incident], 1);
857 <                        }
858 <                        if (single_plane_incident)
859 <                                eval_isotropic(NULL);
860 <                        else
861 <                                eval_anisotropic(NULL);
862 <                }
863 <                return(wrap_up());
864 <        }
865 <        SET_FILE_BINARY(stdin);                 /* load from stdin */
866 <        if (!load_bsdf_rep(stdin))
867 <                return(1);
868 <        prog_start("Interpolating from standard input");
869 <        add_wbsdf("-a", 1);
870 <        add_wbsdf(tfmt[single_plane_incident], 1);
871 <        if (single_plane_incident)              /* resample dist. */
872 <                eval_isotropic(NULL);
873 <        else
874 <                eval_anisotropic(NULL);
875 <
876 <        return(wrap_up());
918 >        return(wrap_up());                      /* call wrapBSDF */
919   userr:
920          fprintf(stderr,
921 <        "Usage: %s [{+|-}a][-g Nlog2][-t pctcull][-l maxlobes] [bsdf.sir ..] > bsdf.xml\n",
921 >        "Usage: %s [{+|-}a][-g Nlog2][-t pctcull][-n nss][-s thresh][-l maxlobes][bsdf.sir ..] > bsdf.xml\n",
922                                  progname);
923          fprintf(stderr,
924 <        "   or: %s -t{3|4} [{+|-}a][-g Nlog2][-t pctcull][{+|-}for[ward]][{+|-}b[ackward]][-e expr][-f file] bsdf_func > bsdf.xml\n",
924 >        "   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",
925                                  progname);
926          return(1);
927   }

Diff Legend

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