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.33 by greg, Tue Feb 2 01:43:24 2016 UTC vs.
Revision 2.54 by greg, Wed Dec 15 01:38:50 2021 UTC

# Line 18 | Line 18 | static const char RCSid[] = "$Id$";
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
21 > #ifndef MAXRCARG
22 > #define MAXRCARG        10000
23   #endif
24  
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
35 #define MAXRCARG        512
36
25   char            *progname;              /* global argv[0] */
26  
27   int             verbose = 0;            /* verbose mode (< 0 no warnings) */
# Line 78 | Line 66 | typedef struct {
66          FVECT   uva[2];                 /* tangent axes */
67          int     ntris;                  /* number of triangles */
68          struct ptri {
69 <                float   afrac;                  /* fraction of total area */
69 >                double  afrac;                  /* fraction of total area */
70                  short   vndx[3];                /* vertex indices */
71          }       tri[1];                 /* triangle array (extends struct) */
72   } POLYTRIS;                     /* triangulated polygon */
# Line 142 | Line 130 | surf_type(const char *otype)
130          return(ST_NONE);
131   }
132  
145 /* Check if any of the characters in str2 are found in str1 */
146 static int
147 matchany(const char *str1, const char *str2)
148 {
149        while (*str1) {
150                const char      *cp = str2;
151                while (*cp)
152                        if (*cp++ == *str1)
153                                return(*str1);
154                ++str1;
155        }
156        return(0);
157 }
158
133   /* Add arguments to oconv command */
134   static char *
135   oconv_command(int ac, char *av[])
# Line 166 | Line 140 | oconv_command(int ac, char *av[])
140          
141          if (ac-- <= 0)
142                  return(NULL);
143 <        while (ac-- > 0) {
144 <                strcpy(cp, *av++);
145 <                while (*cp) cp++;
146 <                *cp++ = ' ';
147 <                if (cp >= oconvbuf+(sizeof(oconvbuf)-32))
143 >        if (verbose < 0) {      /* turn off warnings */
144 >                strcpy(cp, "-w ");
145 >                cp += 3;
146 >        }
147 >        while (ac-- > 0) {      /* copy each argument */
148 >                int     len = strlen(*av);
149 >                if (cp+len+4 >= oconvbuf+sizeof(oconvbuf))
150                          goto overrun;
151 +                if (matchany(*av, SPECIALS)) {
152 +                        *cp++ = QUOTCHAR;
153 +                        strcpy(cp, *av++);
154 +                        cp += len;
155 +                        *cp++ = QUOTCHAR;
156 +                } else {
157 +                        strcpy(cp, *av++);
158 +                        cp += len;
159 +                }
160 +                *cp++ = ' ';
161          }
162                                  /* receiver goes last */
163          if (matchany(recv, SPECIALS)) {
# Line 192 | Line 178 | overrun:
178          exit(1);
179   }
180  
181 < /* Convert a set of arguments into a command line for pipe() or system() */
196 < static char *
197 < convert_commandline(char *cmd, const int len, char *av[])
198 < {
199 <        int     match;
200 <        char    *cp;
181 > #if defined(_WIN32) || defined(_WIN64)
182  
202        for (cp = cmd; *av != NULL; av++) {
203                const int       n = strlen(*av);
204                if (cp+n >= cmd+(len-3))
205                        return(NULL);
206                if (matchany(*av, SPECIALS)) {
207                        const int       quote =
208 #ifdef ALTQUOT
209                                strchr(*av, QUOTCHAR) ? ALTQUOT :
210 #endif
211                                        QUOTCHAR;
212                        *cp++ = quote;
213                        strcpy(cp, *av);
214                        cp += n;
215                        *cp++ = quote;
216                } else {
217                        strcpy(cp, *av);
218                        cp += n;
219                }
220                *cp++ = ' ';
221        }
222        if (cp <= cmd)
223                return(NULL);
224        *--cp = '\0';
225        return(cmd);
226 }
227
183   /* Open a pipe to/from a command given as an argument list */
184   static FILE *
185   popen_arglist(char *av[], char *mode)
# Line 242 | Line 197 | popen_arglist(char *av[], char *mode)
197          return(popen(cmd, mode));
198   }
199  
200 < #ifdef _WIN32
200 > #define pclose_al       pclose
201 >
202   /* Execute system command (Windows version) */
203   static int
204   my_exec(char *av[])
# Line 258 | Line 214 | my_exec(char *av[])
214                  fprintf(stderr, "%s: running: %s\n", progname, cmd);
215          return(system(cmd));
216   }
217 < #else
217 >
218 > #else   /* UNIX */
219 >
220 > static SUBPROC  rt_proc = SP_INACTIVE;  /* we only support one of these */
221 >
222 > /* Open a pipe to a command using an argument list */
223 > static FILE *
224 > popen_arglist(char *av[], char *mode)
225 > {
226 >        int     fd;
227 >
228 >        if (rt_proc.pid > 0) {
229 >                fprintf(stderr, "%s: only one i/o pipe at a time!\n", progname);
230 >                return(NULL);
231 >        }
232 >        if (verbose > 0) {
233 >                char    cmd[4096];
234 >                if (!convert_commandline(cmd, sizeof(cmd), av))
235 >                        strcpy(cmd, "COMMAND TOO LONG TO SHOW");
236 >                fprintf(stderr, "%s: opening pipe %s: %s\n",
237 >                                progname, (*mode=='w') ? "to" : "from", cmd);
238 >        }
239 >        if (*mode == 'w') {
240 >                fd = rt_proc.w = dup(fileno(stdout));
241 >                rt_proc.flags |= PF_FILT_OUT;
242 >        } else if (*mode == 'r') {
243 >                fd = rt_proc.r = dup(fileno(stdin));
244 >                rt_proc.flags |= PF_FILT_INP;
245 >        }
246 >        if (fd < 0 || open_process(&rt_proc, av) <= 0) {
247 >                perror(av[0]);
248 >                return(NULL);
249 >        }
250 >        return(fdopen(fd, mode));
251 > }
252 >
253 > /* Close command pipe (returns -1 on error to match pclose) */
254 > static int
255 > pclose_al(FILE *fp)
256 > {
257 >        int     prob = (fclose(fp) == EOF);
258 >
259 >        if (rt_proc.pid <= 0)
260 >                return(-1);
261 >
262 >        prob |= (close_process(&rt_proc) != 0);
263 >
264 >        return(-prob);
265 > }
266 >
267   /* Execute system command in our stead (Unix version) */
268   static int
269   my_exec(char *av[])
# Line 279 | Line 284 | my_exec(char *av[])
284          perror(compath);
285          return(1);
286   }
287 +
288   #endif
289  
290   /* Get normalized direction vector from string specification */
# Line 440 | Line 446 | finish_receiver(void)
446                          curparams.vup[1] = 1;
447          }
448                                          /* determine sample type/bin */
449 <        if (tolower(curparams.hemis[0]) == 'u' | curparams.hemis[0] == '1') {
449 >        if ((tolower(curparams.hemis[0]) == 'u') | (curparams.hemis[0] == '1')) {
450                  sprintf(sbuf, "if(-Dx*%g-Dy*%g-Dz*%g,0,-1)",
451                          curparams.nrm[0], curparams.nrm[1], curparams.nrm[2]);
452                  binv = savqstr(sbuf);
# Line 663 | Line 669 | ssamp_poly(FVECT orig, SURF *sp, double x)
669                  sp->priv = (void *)ptp;
670          }
671                                          /* pick triangle by partial area */
672 <        for (i = 0; i < ptp->ntris && x > ptp->tri[i].afrac; i++)
672 >        for (i = 0; i < ptp->ntris-1 && x > ptp->tri[i].afrac; i++)
673                  x -= ptp->tri[i].afrac;
674          SDmultiSamp(samp2, 2, x/ptp->tri[i].afrac);
675          samp2[0] *= samp2[1] = sqrt(samp2[1]);
# Line 704 | Line 710 | sample_origin(PARAMS *p, FVECT orig, const FVECT rdir,
710          if (p->nsurfs > nall) {         /* (re)allocate surface area cache */
711                  if (projsa) free(projsa);
712                  projsa = (double *)malloc(sizeof(double)*p->nsurfs);
713 <                if (!projsa) return(0);
713 >                if (projsa == NULL) {
714 >                        fputs(progname, stderr);
715 >                        fputs(": out of memory in sample_origin!\n", stderr);
716 >                        exit(1);
717 >                }
718                  nall = p->nsurfs;
719          }
720                                          /* compute projected areas */
721          for (i = 0, sp = p->slist; sp != NULL; i++, sp = sp->next) {
722                  projsa[i] = -DOT(sp->snrm, rdir) * sp->area;
723 <                tarea += projsa[i] *= (double)(projsa[i] > FTINY);
723 >                tarea += projsa[i] *= (double)(projsa[i] > 0);
724          }
725 <        if (tarea <= FTINY) {           /* wrong side of sender? */
725 >        if (tarea < FTINY*FTINY) {      /* wrong side of sender? */
726                  fputs(progname, stderr);
727                  fputs(": internal - sample behind all sender elements!\n",
728                                  stderr);
# Line 730 | Line 740 | sample_uniform(PARAMS *p, int b, FILE *fp)
740   {
741          int     n = sampcnt;
742          double  samp3[3];
743 <        double  duvw[3];
734 <        FVECT   orig_dir[2];
743 >        FVECT   duvw, orig_dir[2];
744          int     i;
745  
746          if (fp == NULL)                 /* just requesting number of bins? */
# Line 739 | Line 748 | sample_uniform(PARAMS *p, int b, FILE *fp)
748  
749          while (n--) {                   /* stratified hemisphere sampling */
750                  SDmultiSamp(samp3, 3, (n+frandom())/sampcnt);
751 <                SDsquare2disk(duvw, samp3[1], samp3[2]);
751 >                square2disk(duvw, samp3[1], samp3[2]);
752                  duvw[2] = -sqrt(1. - duvw[0]*duvw[0] - duvw[1]*duvw[1]);
753                  for (i = 3; i--; )
754                          orig_dir[1][i] = duvw[0]*p->udir[i] +
# Line 747 | Line 756 | sample_uniform(PARAMS *p, int b, FILE *fp)
756                                                  duvw[2]*p->nrm[i] ;
757                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
758                          return(0);
759 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
759 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
760                          return(0);
761          }
762          return(1);
# Line 759 | Line 768 | sample_shirchiu(PARAMS *p, int b, FILE *fp)
768   {
769          int     n = sampcnt;
770          double  samp3[3];
771 <        double  duvw[3];
763 <        FVECT   orig_dir[2];
771 >        FVECT   duvw, orig_dir[2];
772          int     i;
773  
774          if (fp == NULL)                 /* just requesting number of bins? */
# Line 768 | Line 776 | sample_shirchiu(PARAMS *p, int b, FILE *fp)
776  
777          while (n--) {                   /* stratified sampling */
778                  SDmultiSamp(samp3, 3, (n+frandom())/sampcnt);
779 <                SDsquare2disk(duvw, (b/p->hsiz + samp3[1])/curparams.hsiz,
779 >                square2disk(duvw, (b/p->hsiz + samp3[1])/curparams.hsiz,
780                                  (b%p->hsiz + samp3[2])/curparams.hsiz);
781                  duvw[2] = sqrt(1. - duvw[0]*duvw[0] - duvw[1]*duvw[1]);
782                  for (i = 3; i--; )
# Line 777 | Line 785 | sample_shirchiu(PARAMS *p, int b, FILE *fp)
785                                                  duvw[2]*p->nrm[i] ;
786                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
787                          return(0);
788 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
788 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
789                          return(0);
790          }
791          return(1);
# Line 813 | Line 821 | sample_reinhart(PARAMS *p, int b, FILE *fp)
821          }
822          while (n--) {                   /* stratified sampling */
823                  SDmultiSamp(samp3, 3, (n+frandom())/sampcnt);
824 +                if (row >= RowMax-1)    /* avoid crowding at zenith */
825 +                        samp3[1] *= samp3[1];
826                  alt = (row+samp3[1])*RAH;
827                  azi = (2.*PI)*(col+samp3[2]-.5)/rnaz(row);
828                  duvw[2] = cos(alt);     /* measured from horizon */
829                  duvw[0] = tsin(azi)*duvw[2];
830 <                duvw[1] = tcos(azi)*duvw[2];
830 >                duvw[1] = -tcos(azi)*duvw[2];
831                  duvw[2] = sqrt(1. - duvw[2]*duvw[2]);
832                  for (i = 3; i--; )
833                          orig_dir[1][i] = -duvw[0]*p->udir[i] -
# Line 825 | Line 835 | sample_reinhart(PARAMS *p, int b, FILE *fp)
835                                                  duvw[2]*p->nrm[i] ;
836                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp3[0]))
837                          return(0);
838 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
838 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
839                          return(0);
840          }
841          return(1);
# Line 876 | Line 886 | sample_klems(PARAMS *p, int b, FILE *fp)
886                                                  duvw[2]*p->nrm[i] ;
887                  if (!sample_origin(p, orig_dir[0], orig_dir[1], samp2[0]))
888                          return(0);
889 <                if (fwrite(orig_dir, sizeof(FVECT), 2, fp) != 2)
889 >                if (putbinary(orig_dir, sizeof(FVECT), 2, fp) != 2)
890                          return(0);
891          }
892          return(1);
# Line 924 | Line 934 | prepare_sampler(void)
934                  curparams.udir[1] *= -1.;
935                  curparams.udir[2] *= -1.;
936          }
937 <        if (tolower(curparams.hemis[0]) == 'u' | curparams.hemis[0] == '1')
937 >        if ((tolower(curparams.hemis[0]) == 'u') | (curparams.hemis[0] == '1'))
938                  curparams.sample_basis = sample_uniform;
939          else if (tolower(curparams.hemis[0]) == 's' &&
940                                  tolower(curparams.hemis[1]) == 'c')
# Line 985 | Line 995 | finish_polygon(SURF *p)
995                  VCOPY(e1, e2);
996          }
997          p->area = normalize(p->snrm)*0.5;
998 <        return(p->area > FTINY);
998 >        return(p->area > FTINY*FTINY);
999   }
1000  
1001   /* Add a surface to our current parameters */
# Line 1003 | Line 1013 | add_surface(int st, const char *oname, FILE *fp)
1013          snew = (SURF *)malloc(sizeof(SURF) + sizeof(double)*(n-1));
1014          if (snew == NULL) {
1015                  fputs(progname, stderr);
1016 <                fputs(": out of memory!\n", stderr);
1016 >                fputs(": out of memory in add_surface!\n", stderr);
1017                  exit(1);
1018          }
1019          strncpy(snew->sname, oname, sizeof(snew->sname)-1);
# Line 1048 | Line 1058 | add_surface(int st, const char *oname, FILE *fp)
1058                  snew->area *= PI*snew->area;
1059                  break;
1060          }
1061 <        if ((snew->area <= FTINY) & (verbose >= 0)) {
1061 >        if ((snew->area <= FTINY*FTINY) & (verbose >= 0)) {
1062                  fprintf(stderr, "%s: warning - zero area for surface '%s'\n",
1063                                  progname, oname);
1064                  free(snew);
# Line 1285 | Line 1295 | main(int argc, char *argv[])
1295                          na = 0;
1296                          continue;
1297                  case 'w':               /* options without arguments */
1298 <                        if (argv[a][2] != '+') verbose = -1;
1298 >                        if (!argv[a][2] || strchr("+1tTyY", argv[a][2]) == NULL)
1299 >                                verbose = -1;
1300 >                        break;
1301                  case 'V':
1302                  case 'u':
1303                  case 'h':
# Line 1294 | Line 1306 | main(int argc, char *argv[])
1306                  case 'n':               /* options with 1 argument */
1307                  case 's':
1308                  case 'o':
1309 +                case 't':
1310                          na = 2;
1311                          break;
1312                  case 'b':               /* special case */
# Line 1362 | Line 1375 | main(int argc, char *argv[])
1375                          fputs(": -i, -I supported for pass-through only\n", stderr);
1376                          return(1);
1377                  }
1378 <                fmtopt[2] = (sizeof(RREAL)==sizeof(double)) ? 'd' : 'f';
1378 > #ifdef SMLFLT
1379 >                fmtopt[2] = 'f';
1380 > #else
1381 >                fmtopt[2] = 'd';
1382 > #endif
1383                  if (sampcnt <= 0) sampcnt = 10000;
1384          }
1385          sprintf(sampcntbuf, "%d", sampcnt);
# Line 1409 | Line 1426 | main(int argc, char *argv[])
1426          for (i = 0; i < nsbins; i++)    /* send rcontrib ray samples */
1427                  if (!(*curparams.sample_basis)(&curparams, i, rcfp))
1428                          return(1);
1429 <        return(pclose(rcfp) < 0);       /* all finished! */
1429 >        return(pclose_al(rcfp) < 0);    /* all finished! */
1430   userr:
1431          if (a < argc-2)
1432                  fprintf(stderr, "%s: unsupported option '%s'", progname, argv[a]);

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines