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

Comparing ray/src/util/rfluxmtx.c (file contents):
Revision 2.20 by greg, Mon Feb 2 16:04:11 2015 UTC vs.
Revision 2.41 by greg, Sat Aug 20 03:43:24 2016 UTC

# Line 11 | Line 11 | static const char RCSid[] = "$Id$";
11   #include <stdlib.h>
12   #include "rtio.h"
13   #include "rtmath.h"
14 < #include "rtprocess.h"
14 > #include "paths.h"
15   #include "bsdf.h"
16   #include "bsdf_m.h"
17   #include "random.h"
18   #include "triangulate.h"
19   #include "platform.h"
20  
21 – #ifdef getc_unlocked            /* avoid horrendous overhead of flockfile */
22 – #undef getc
23 – #define getc    getc_unlocked
24 – #endif
25 –
26 – #ifdef _WIN32
27 – #define SPECIALS        " \t\"$*?"
28 – #define QUOTCHAR        '"'
29 – #else
30 – #define SPECIALS        " \t\n'\"()${}*?[];|&"
31 – #define QUOTCHAR        '\''
32 – #define ALTQUOT         '"'
33 – #endif
34 –
21   #define MAXRCARG        512
22  
23   char            *progname;              /* global argv[0] */
# Line 78 | Line 64 | typedef struct {
64          FVECT   uva[2];                 /* tangent axes */
65          int     ntris;                  /* number of triangles */
66          struct ptri {
67 <                float   afrac;                  /* fraction of total area */
67 >                double  afrac;                  /* fraction of total area */
68                  short   vndx[3];                /* vertex indices */
69          }       tri[1];                 /* triangle array (extends struct) */
70   } POLYTRIS;                     /* triangulated polygon */
71  
72   typedef struct param_s {
73 <        char            hemis[32];      /* hemispherical sampling spec. */
73 >        char            sign;           /* '-' for axis reversal */
74 >        char            hemis[31];      /* hemispherical sampling spec. */
75          int             hsiz;           /* hemisphere basis size */
76          int             nsurfs;         /* number of surfaces */
77          SURF            *slist;         /* list of surfaces */
78          FVECT           vup;            /* up vector (zero if unset) */
79          FVECT           nrm;            /* average normal direction */
80 <        FVECT           udir, vdir;     /* v-up tangent axes */
80 >        FVECT           udir, vdir;     /* tangent axes */
81          char            *outfn;         /* output file name (receiver) */
82          int             (*sample_basis)(struct param_s *p, int, FILE *);
83   } PARAMS;                       /* sender/receiver parameters */
# Line 155 | Line 142 | oconv_command(int ac, char *av[])
142                  strcpy(cp, *av++);
143                  while (*cp) cp++;
144                  *cp++ = ' ';
145 <                if (cp >= oconvbuf+(sizeof(oconvbuf)-32)) {
146 <                        fputs(progname, stderr);
160 <                        fputs(": too many file arguments!\n", stderr);
161 <                        exit(1);
162 <                }
145 >                if (cp >= oconvbuf+(sizeof(oconvbuf)-32))
146 >                        goto overrun;
147          }
148 <        strcpy(cp, recv);       /* receiver goes last */
148 >                                /* receiver goes last */
149 >        if (matchany(recv, SPECIALS)) {
150 >                *cp++ = QUOTCHAR;
151 >                while (*recv) {
152 >                        if (cp >= oconvbuf+(sizeof(oconvbuf)-3))
153 >                                goto overrun;
154 >                        *cp++ = *recv++;
155 >                }
156 >                *cp++ = QUOTCHAR;
157 >                *cp = '\0';
158 >        } else
159 >                strcpy(cp, recv);
160          return(oconvbuf);
161 + overrun:
162 +        fputs(progname, stderr);
163 +        fputs(": too many file arguments!\n", stderr);
164 +        exit(1);
165   }
166  
168 – /* Check if any of the characters in str2 are found in str1 */
169 – static int
170 – matchany(const char *str1, const char *str2)
171 – {
172 –        while (*str1) {
173 –                const char      *cp = str2;
174 –                while (*cp)
175 –                        if (*cp++ == *str1)
176 –                                return(*str1);
177 –                ++str1;
178 –        }
179 –        return(0);
180 – }
181 –
182 –
183 – /* Convert a set of arguments into a command line for pipe() or system() */
184 – static char *
185 – convert_commandline(char *cmd, const int len, char *av[])
186 – {
187 –        int     match;
188 –        char    *cp;
189 –
190 –        for (cp = cmd; *av != NULL; av++) {
191 –                const int       n = strlen(*av);
192 –                if (cp+n >= cmd+(len-3)) {
193 –                        fputs(progname, stderr);
194 –                        return(NULL);
195 –                }
196 –                if ((match = matchany(*av, SPECIALS))) {
197 –                        const int       quote =
198 – #ifdef ALTQUOT
199 –                                (match == QUOTCHAR) ? ALTQUOT :
200 – #endif
201 –                                        QUOTCHAR;
202 –                        *cp++ = quote;
203 –                        strcpy(cp, *av);
204 –                        cp += n;
205 –                        *cp++ = quote;
206 –                } else {
207 –                        strcpy(cp, *av);
208 –                        cp += n;
209 –                }
210 –                *cp++ = ' ';
211 –        }
212 –        if (cp <= cmd)
213 –                return(NULL);
214 –        *--cp = '\0';
215 –        return(cmd);
216 – }
217 –
167   /* Open a pipe to/from a command given as an argument list */
168   static FILE *
169   popen_arglist(char *av[], char *mode)
# Line 232 | Line 181 | popen_arglist(char *av[], char *mode)
181          return(popen(cmd, mode));
182   }
183  
184 < #ifdef _WIN32
184 > #if defined(_WIN32) || defined(_WIN64)
185   /* Execute system command (Windows version) */
186   static int
187   my_exec(char *av[])
# Line 317 | Line 266 | parse_params(PARAMS *p, char *pargs)
266   {
267          char    *cp = pargs;
268          int     nparams = 0;
269 +        int     quot;
270          int     i;
271  
272          for ( ; ; ) {
# Line 324 | Line 274 | parse_params(PARAMS *p, char *pargs)
274                  case 'h':
275                          if (*cp++ != '=')
276                                  break;
277 +                        if ((*cp == '+') | (*cp == '-'))
278 +                                p->sign = *cp++;
279 +                        else
280 +                                p->sign = '+';
281                          p->hsiz = 0;
282                          i = 0;
283                          while (*cp && !isspace(*cp)) {
# Line 349 | Line 303 | parse_params(PARAMS *p, char *pargs)
303                  case 'o':
304                          if (*cp++ != '=')
305                                  break;
306 +                        quot = 0;
307 +                        if ((*cp == '"') | (*cp == '\''))
308 +                                quot = *cp++;
309                          i = 0;
310 <                        while (*cp && !isspace(*cp++))
311 <                                i++;
310 >                        while (*cp && (quot ? (*cp != quot) : !isspace(*cp))) {
311 >                                i++; cp++;
312 >                        }
313                          if (!i)
314                                  break;
315 <                        *--cp = '\0';
315 >                        if (!*cp) {
316 >                                if (quot)
317 >                                        break;
318 >                                cp[1] = '\0';
319 >                        }
320 >                        *cp = '\0';
321                          p->outfn = savqstr(cp-i);
322 <                        *cp++ = ' ';
322 >                        *cp++ = quot ? quot : ' ';
323                          ++nparams;
324                          continue;
325                  case ' ':
326                  case '\t':
327                  case '\r':
365 –                        continue;
328                  case '\n':
329 +                        continue;
330                  case '\0':
331                          return(nparams);
332                  default:
# Line 431 | Line 394 | finish_receiver(void)
394                          exit(1);
395                  }
396                  calfn = shirchiufn; shirchiufn = NULL;
397 <                sprintf(sbuf, "SCdim=%d,rNx=%g,rNy=%g,rNz=%g,Ux=%g,Uy=%g,Uz=%g",
397 >                sprintf(sbuf, "SCdim=%d,rNx=%g,rNy=%g,rNz=%g,Ux=%g,Uy=%g,Uz=%g,RHS=%c1",
398                                  curparams.hsiz,
399                          curparams.nrm[0], curparams.nrm[1], curparams.nrm[2],
400 <                        curparams.vup[0], curparams.vup[1], curparams.vup[2]);
400 >                        curparams.vup[0], curparams.vup[1], curparams.vup[2],
401 >                        curparams.sign);
402                  params = savqstr(sbuf);
403                  binv = "scbin";
404                  nbins = "SCdim*SCdim";
405          } else if ((tolower(curparams.hemis[0]) == 'r') |
406                          (tolower(curparams.hemis[0]) == 't')) {
407                  calfn = reinhfn; reinhfn = NULL;
408 <                sprintf(sbuf, "MF=%d,rNx=%g,rNy=%g,rNz=%g,Ux=%g,Uy=%g,Uz=%g",
408 >                sprintf(sbuf, "MF=%d,rNx=%g,rNy=%g,rNz=%g,Ux=%g,Uy=%g,Uz=%g,RHS=%c1",
409                                  curparams.hsiz,
410                          curparams.nrm[0], curparams.nrm[1], curparams.nrm[2],
411 <                        curparams.vup[0], curparams.vup[1], curparams.vup[2]);
411 >                        curparams.vup[0], curparams.vup[1], curparams.vup[2],
412 >                        curparams.sign);
413                  params = savqstr(sbuf);
414                  binv = "rbin";
415                  nbins = "Nrbins";
# Line 472 | Line 437 | finish_receiver(void)
437                                  progname, curparams.hemis);
438                  exit(1);
439          }
440 +        if (tolower(curparams.hemis[0]) == 'k') {
441 +                sprintf(sbuf, "RHS=%c1", curparams.sign);
442 +                params = savqstr(sbuf);
443 +        }
444          if (!uniform & (curparams.slist->styp == ST_SOURCE)) {
445                  SURF    *sp;
446                  for (sp = curparams.slist; sp != NULL; sp = sp->next)
# Line 519 | Line 488 | make_axes(FVECT uva[2], const FVECT nrm)
488   {
489          int     i;
490  
491 <        if (!getperpendicular(uva[0], nrm)) {
491 >        if (!getperpendicular(uva[0], nrm, 1)) {
492                  fputs(progname, stderr);
493                  fputs(": bad surface normal in make_axes!\n", stderr);
494                  exit(1);
# Line 633 | Line 602 | ssamp_poly(FVECT orig, SURF *sp, double x)
602                  sp->priv = (void *)ptp;
603          }
604                                          /* pick triangle by partial area */
605 <        for (i = 0; i < ptp->ntris && x > ptp->tri[i].afrac; i++)
605 >        for (i = 0; i < ptp->ntris-1 && x > ptp->tri[i].afrac; i++)
606                  x -= ptp->tri[i].afrac;
607          SDmultiSamp(samp2, 2, x/ptp->tri[i].afrac);
608          samp2[0] *= samp2[1] = sqrt(samp2[1]);
# Line 663 | Line 632 | sample_origin(PARAMS *p, FVECT orig, const FVECT rdir,
632                                          /* special case for lone surface */
633          if (p->nsurfs == 1) {
634                  sp = p->slist;
635 <                if (DOT(sp->snrm, rdir) >= -FTINY) {
635 >                if (DOT(sp->snrm, rdir) >= FTINY) {
636                          fprintf(stderr,
637                                  "%s: internal - sample behind sender '%s'\n",
638                                          progname, sp->sname);
# Line 674 | Line 643 | sample_origin(PARAMS *p, FVECT orig, const FVECT rdir,
643          if (p->nsurfs > nall) {         /* (re)allocate surface area cache */
644                  if (projsa) free(projsa);
645                  projsa = (double *)malloc(sizeof(double)*p->nsurfs);
646 <                if (!projsa) return(0);
646 >                if (projsa == NULL) {
647 >                        fputs(progname, stderr);
648 >                        fputs(": out of memory in sample_origin!\n", stderr);
649 >                        exit(1);
650 >                }
651                  nall = p->nsurfs;
652          }
653                                          /* compute projected areas */
# Line 717 | Line 690 | sample_uniform(PARAMS *p, int b, FILE *fp)
690                                                  duvw[2]*p->nrm[i] ;
691                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
692                          return(0);
693 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
693 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
694                          return(0);
695          }
696          return(1);
# Line 747 | Line 720 | sample_shirchiu(PARAMS *p, int b, FILE *fp)
720                                                  duvw[2]*p->nrm[i] ;
721                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
722                          return(0);
723 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
723 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
724                          return(0);
725          }
726          return(1);
# Line 795 | Line 768 | sample_reinhart(PARAMS *p, int b, FILE *fp)
768                                                  duvw[2]*p->nrm[i] ;
769                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
770                          return(0);
771 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
771 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
772                          return(0);
773          }
774          return(1);
# Line 838 | Line 811 | sample_klems(PARAMS *p, int b, FILE *fp)
811  
812          while (n--) {                   /* stratified sampling */
813                  SDmultiSamp(samp2, 2, (n+frandom())/sampcnt);
814 <                if (!bi_getvec(duvw, b+samp2[1], kbasis[bi]))
814 >                if (!fo_getvec(duvw, b+samp2[1], kbasis[bi]))
815                          return(0);
816                  for (i = 3; i--; )
817 <                        orig_dir[1][i] = duvw[0]*p->udir[i] +
818 <                                                duvw[1]*p->vdir[i] +
817 >                        orig_dir[1][i] = -duvw[0]*p->udir[i] -
818 >                                                duvw[1]*p->vdir[i] -
819                                                  duvw[2]*p->nrm[i] ;
820                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp2[0]))
821                          return(0);
822 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
822 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
823                          return(0);
824          }
825          return(1);
# Line 882 | Line 855 | prepare_sampler(void)
855                  else
856                          curparams.vup[1] = 1;
857          }
858 <        VCROSS(curparams.udir, curparams.vup, curparams.nrm);
858 >        fcross(curparams.udir, curparams.vup, curparams.nrm);
859          if (normalize(curparams.udir) == 0) {
860                  fputs(progname, stderr);
861                  fputs(": up vector coincides with sender normal\n", stderr);
862                  return(-1);
863          }
864 <        VCROSS(curparams.vdir, curparams.nrm, curparams.udir);
864 >        fcross(curparams.vdir, curparams.nrm, curparams.udir);
865 >        if (curparams.sign == '-') {    /* left-handed coordinate system? */
866 >                curparams.udir[0] *= -1.;
867 >                curparams.udir[1] *= -1.;
868 >                curparams.udir[2] *= -1.;
869 >        }
870          if (tolower(curparams.hemis[0]) == 'u' | curparams.hemis[0] == '1')
871                  curparams.sample_basis = sample_uniform;
872          else if (tolower(curparams.hemis[0]) == 's' &&
# Line 968 | Line 946 | add_surface(int st, const char *oname, FILE *fp)
946          snew = (SURF *)malloc(sizeof(SURF) + sizeof(double)*(n-1));
947          if (snew == NULL) {
948                  fputs(progname, stderr);
949 <                fputs(": out of memory!\n", stderr);
949 >                fputs(": out of memory in add_surface!\n", stderr);
950                  exit(1);
951          }
952          strncpy(snew->sname, oname, sizeof(snew->sname)-1);
# Line 1083 | Line 1061 | static int
1061   add_send_object(FILE *fp)
1062   {
1063          int             st;
1064 <        char            otype[32], oname[128];
1064 >        char            thismod[128], otype[32], oname[128];
1065          int             n;
1066  
1067 <        if (fscanf(fp, "%*s %s %s", otype, oname) != 2)
1067 >        if (fscanf(fp, "%s %s %s", thismod, otype, oname) != 3)
1068                  return(0);              /* must have hit EOF! */
1069          if (!strcmp(otype, "alias")) {
1070                  fscanf(fp, "%*s");      /* skip alias */
# Line 1099 | Line 1077 | add_send_object(FILE *fp)
1077                          fputs(": cannot use source as a sender!\n", stderr);
1078                          return(-1);
1079                  }
1080 +                if (strcmp(thismod, curmod)) {
1081 +                        if (curmod[0]) {
1082 +                                fputs(progname, stderr);
1083 +                                fputs(": warning - multiple modifiers in sender\n",
1084 +                                                stderr);
1085 +                        }
1086 +                        strcpy(curmod, thismod);
1087 +                }
1088                  parse_params(&curparams, newparams);
1089                  newparams[0] = '\0';
1090                  add_surface(st, oname, fp);     /* read & store surface */
# Line 1268 | Line 1254 | main(int argc, char *argv[])
1254                          if (argv[a][2] != 'v') na = 2;
1255                          break;
1256                  case 'a':               /* special case */
1257 <                        na = (argv[a][2] == 'v') ? 4 : 2;
1257 >                        if (argv[a][2] == 'p') {
1258 >                                na = 2; /* photon map [+ bandwidth(s)] */
1259 >                                if (a < argc-3 && atoi(argv[a+1]) > 0)
1260 >                                        na += 1 + (a < argc-4 && atoi(argv[a+2]) > 0);
1261 >                        } else
1262 >                                na = (argv[a][2] == 'v') ? 4 : 2;
1263                          break;
1264                  case 'm':               /* special case */
1265                          if (!argv[a][2]) goto userr;
# Line 1333 | Line 1324 | main(int argc, char *argv[])
1324                  return(my_exec(rcarg)); /* rcontrib does everything */
1325          }
1326          clear_params(&curparams, 0);    /* else load sender surface & params */
1327 +        curmod[0] = '\0';
1328          if (load_scene(sendfn, add_send_object) < 0)
1329                  return(1);
1330          if ((nsbins = prepare_sampler()) <= 0)
# Line 1350 | Line 1342 | main(int argc, char *argv[])
1342   #ifdef getc_unlocked
1343          flockfile(rcfp);
1344   #endif
1345 <        if (verbose) {
1345 >        if (verbose > 0) {
1346                  fprintf(stderr, "%s: sampling %d directions", progname, nsbins);
1347                  if (curparams.nsurfs > 1)
1348                          fprintf(stderr, " (%d elements)\n", curparams.nsurfs);
# Line 1360 | Line 1352 | main(int argc, char *argv[])
1352          for (i = 0; i < nsbins; i++)    /* send rcontrib ray samples */
1353                  if (!(*curparams.sample_basis)(&curparams, i, rcfp))
1354                          return(1);
1355 <        return(pclose(rcfp) == 0);      /* all finished! */
1355 >        return(pclose(rcfp) < 0);       /* all finished! */
1356   userr:
1357          if (a < argc-2)
1358                  fprintf(stderr, "%s: unsupported option '%s'", progname, argv[a]);

Diff Legend

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