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

Comparing ray/src/cal/rcalc.c (file contents):
Revision 1.11 by schorsch, Fri Nov 14 17:31:24 2003 UTC vs.
Revision 1.28 by greg, Sun May 19 20:02:27 2019 UTC

# Line 8 | Line 8 | static const char RCSid[] = "$Id$";
8   */
9  
10   #include  <stdlib.h>
11 #include  <fcntl.h>
12 #include  <stdio.h>
13 #include  <string.h>
11   #include  <math.h>
12   #include  <ctype.h>
13  
14   #include  "platform.h"
18 #include  "calcomp.h"
15   #include  "rterror.h"
16 + #include  "rtmisc.h"
17 + #include  "rtio.h"
18 + #include  "calcomp.h"
19  
20   #define  isnum(c)       (isdigit(c) || (c)=='-' || (c)=='.' \
21                                  || (c)=='+' || (c)=='e' || (c)=='E')
22  
23   #define  isblnk(c)      (igneol ? isspace(c) : (c)==' '||(c)=='\t')
24  
25 < #define  INBSIZ         4096    /* longest record */
25 > #define  INBSIZ         16384   /* longest record */
26   #define  MAXCOL         32      /* number of columns recorded */
27  
28                                  /* field type specifications */
# Line 74 | Line 73 | struct field  *inpfmt = NULL;   /* input record format
73   struct field  *outfmt = NULL;   /* output record structure */
74   struct strvar  *svhead = NULL;  /* string variables */
75  
76 + long  incnt = 0;                /* limit number of input records? */
77 + long  outcnt = 0;               /* limit number of output records? */
78 +
79   int  blnkeq = 1;                /* blanks compare equal? */
80   int  igneol = 0;                /* ignore end of line? */
81 + int  passive = 0;               /* passive mode (transmit unmatched input) */
82   char  sepchar = '\t';           /* input/output separator */
83   int  noinput = 0;               /* no input records? */
84 + int  itype = 'a';               /* input type (a/f/F/d/D) */
85   int  nbicols = 0;               /* number of binary input columns */
86 < int  bocols = 0;                /* produce binary output columns */
86 > int  otype = 'a';               /* output format (a/f/F/d/D) */
87   char  inpbuf[INBSIZ];           /* input buffer */
88   double  colval[MAXCOL];         /* input column values */
89   unsigned long  colflg = 0;      /* column retrieved flags */
# Line 103 | Line 107 | int  argc,
107   char  *argv[]
108   )
109   {
110 +        char  *fpath;
111          int  i;
112  
113          esupport |= E_VARIABLE|E_FUNCTION|E_INCHAN|E_OUTCHAN|E_RCONST;
# Line 122 | Line 127 | char  *argv[]
127                  case 'l':
128                          igneol = !igneol;
129                          break;
130 +                case 'p':
131 +                        passive = !passive;
132 +                        break;
133                  case 't':
134                          sepchar = argv[i][2];
135                          break;
# Line 129 | Line 137 | char  *argv[]
137                          svpreset(argv[++i]);
138                          break;
139                  case 'f':
140 <                        fcompile(argv[++i]);
140 >                        fpath = getpath(argv[++i], getrlibpath(), 0);
141 >                        if (fpath == NULL) {
142 >                                eputs(argv[0]);
143 >                                eputs(": cannot find file '");
144 >                                eputs(argv[i]);
145 >                                eputs("'\n");
146 >                                quit(1);
147 >                        }
148 >                        fcompile(fpath);
149                          break;
150                  case 'e':
151                          scompile(argv[++i], NULL, 0);
# Line 140 | Line 156 | char  *argv[]
156                  case 'i':
157                          switch (argv[i][2]) {
158                          case '\0':
159 +                                itype = 'a';
160                                  nbicols = 0;
161                                  readfmt(argv[++i], 0);
162                                  break;
163 +                        case 'n':
164 +                                incnt = atol(argv[++i]);
165 +                                break;
166                          case 'a':
167 +                                itype = 'a';
168                                  nbicols = 0;
169                                  break;
170                          case 'd':
171 +                        case 'D':
172 +                                itype = argv[i][2];
173                                  if (isdigit(argv[i][3]))
174                                          nbicols = atoi(argv[i]+3);
175                                  else
# Line 158 | Line 181 | char  *argv[]
181                                  }
182                                  break;
183                          case 'f':
184 +                        case 'F':
185 +                                itype = argv[i][2];
186                                  if (isdigit(argv[i][3]))
187 <                                        nbicols = -atoi(argv[i]+3);
187 >                                        nbicols = atoi(argv[i]+3);
188                                  else
189 <                                        nbicols = -1;
190 <                                if (-nbicols*sizeof(float) > INBSIZ) {
189 >                                        nbicols = 1;
190 >                                if (nbicols*sizeof(float) > INBSIZ) {
191                                          eputs(argv[0]);
192                                          eputs(": too many input columns\n");
193                                          quit(1);
# Line 175 | Line 200 | char  *argv[]
200                  case 'o':
201                          switch (argv[i][2]) {
202                          case '\0':
203 <                                bocols = 0;
203 >                                otype = 'a';
204                                  readfmt(argv[++i], 1);
205                                  break;
206 +                        case 'n':
207 +                                outcnt = atol(argv[++i]);
208 +                                break;
209                          case 'a':
210 <                                bocols = 0;
210 >                                otype = 'a';
211                                  break;
212                          case 'd':
213 <                                bocols = 1;
186 <                                break;
213 >                        case 'D':
214                          case 'f':
215 <                                bocols = -1;
215 >                        case 'F':
216 >                                otype = argv[i][2];
217                                  break;
218 +                        default:
219 +                                goto userr;
220                          }
221                          break;
222                  case 'w':
# Line 199 | Line 229 | char  *argv[]
229                  userr:
230                          eputs("Usage: ");
231                          eputs(argv[0]);
232 < eputs(" [-b][-l][-n][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n");
232 > eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n");
233                          quit(1);
234                  }
235 <
235 >        if (otype != 'a')
236 >                SET_FILE_BINARY(stdout);
237 > #ifdef getc_unlocked            /* avoid lock/unlock overhead */
238 >        flockfile(stdout);
239 > #endif
240          if (noinput) {          /* produce a single output record */
241 +                if (i < argc) {
242 +                        eputs(argv[0]);
243 +                        eputs(": file argument(s) incompatible with -n\n");
244 +                        quit(1);
245 +                }
246                  eclock++;
247                  putout();
248                  quit(0);
249          }
211
250          if (blnkeq)             /* for efficiency */
251                  nbsynch();
252  
# Line 242 | Line 280 | FILE  *fp
280   {
281          if (inpfmt != NULL)
282                  return(getrec());
283 <        if (nbicols > 0)
284 <                return(fread(inpbuf, sizeof(double),
285 <                                        nbicols, fp) == nbicols);
286 <        if (nbicols < 0)
287 <                return(fread(inpbuf, sizeof(float),
288 <                                        -nbicols, fp) == -nbicols);
283 >        if (tolower(itype) == 'd') {
284 >                if (getbinary(inpbuf, sizeof(double), nbicols, fp) != nbicols)
285 >                        return(0);
286 >                if (itype == 'D')
287 >                        swap64(inpbuf, nbicols);
288 >                return(1);
289 >        }
290 >        if (tolower(itype) == 'f') {
291 >                if (getbinary(inpbuf, sizeof(float), nbicols, fp) != nbicols)
292 >                        return(0);
293 >                if (itype == 'F')
294 >                        swap32(inpbuf, nbicols);
295 >                return(1);
296 >        }
297          return(fgets(inpbuf, INBSIZ, fp) != NULL);
298   }
299  
# Line 257 | Line 303 | execute(           /* process a file */
303   char  *file
304   )
305   {
306 <        int  conditional = vardefined("cond");
306 >        const int  conditional = vardefined("cond");
307 >        const int  set_recno = (varlookup("recno") != NULL);
308 >        const int  set_outno = (varlookup("outno") != NULL);
309          long  nrecs = 0;
310          long  nout = 0;
311          FILE  *fp;
# Line 269 | Line 317 | char  *file
317                  eputs(": cannot open\n");
318                  quit(1);
319          }
320 +        if (itype != 'a')
321 +                SET_FILE_BINARY(fp);
322 + #ifdef getc_unlocked            /* avoid lock/unlock overhead */
323 +        flockfile(fp);
324 + #endif
325          if (inpfmt != NULL)
326                  initinp(fp);
327          
328          while (getinputrec(fp)) {
329 <                varset("recno", '=', (double)++nrecs);
329 >                ++nrecs;
330 >                if (set_recno)
331 >                        varset("recno", '=', (double)nrecs);
332 >                if (set_outno)
333 >                        varset("outno", '=', (double)(nout+1));
334                  colflg = 0;
335                  eclock++;
336                  if (!conditional || varvalue("cond") > 0.0) {
280                        varset("outno", '=', (double)++nout);
337                          putout();
338 +                        ++nout;
339                  }
340 +                if (incnt && nrecs >= incnt)
341 +                        break;
342 +                if (outcnt && nout >= outcnt)
343 +                        break;
344          }
345          fclose(fp);
346   }
# Line 292 | Line 353 | putout(void)                /* produce an output recor
353          colpos = 0;
354          if (outfmt != NULL)
355                  putrec();
356 <        else if (bocols)
296 <                chanout(bchanset);
297 <        else
356 >        else if (otype == 'a')
357                  chanout(chanset);
358 <        if (colpos && !bocols)
358 >        else
359 >                chanout(bchanset);
360 >        if (colpos && otype == 'a')
361                  putchar('\n');
362          if (unbuff)
363                  fflush(stdout);
# Line 307 | Line 368 | static double
368   l_in(char *funame)      /* function call for $channel */
369   {
370          int  n;
371 <        register char  *cp;
371 >        char  *cp;
372                          /* get argument as integer */
373          n = (int)(argument(1) + .5);
374          if (n != 0)     /* return channel value */
# Line 315 | Line 376 | l_in(char *funame)     /* function call for $channel */
376                          /* determine number of channels */
377          if (noinput || inpfmt != NULL)
378                  return(0);
379 <        if (nbicols > 0)
379 >        if (nbicols)
380                  return(nbicols);
320        if (nbicols < 0)
321                return(-nbicols);
381          cp = inpbuf;    /* need to count */
382          for (n = 0; *cp; )
383                  if (blnkeq && isspace(sepchar)) {
# Line 341 | Line 400 | int  n
400   )
401   {
402          int  i;
403 <        register char  *cp;
403 >        char  *cp;
404  
405          if (noinput || inpfmt != NULL) {
406                  eputs("no column input\n");
# Line 351 | Line 410 | int  n
410                  eputs("illegal channel number\n");
411                  quit(1);
412          }
413 <        if (nbicols > 0) {
413 >        if (nbicols) {
414                  if (n > nbicols)
415                          return(0.0);
416 <                cp = inpbuf + (n-1)*sizeof(double);
417 <                return(*(double *)cp);
418 <        }
419 <        if (nbicols < 0) {
361 <                if (n > -nbicols)
362 <                        return(0.0);
416 >                if (tolower(itype) == 'd') {
417 >                        cp = inpbuf + (n-1)*sizeof(double);
418 >                        return(*(double *)cp);
419 >                }
420                  cp = inpbuf + (n-1)*sizeof(float);
421                  return(*(float *)cp);
422          }
# Line 411 | Line 468 | double  v
468   )
469   {
470          static char     zerobuf[sizeof(double)];
471 +        float   fval = v;
472  
473          while (++colpos < n)
474 <                fwrite(zerobuf,
475 <                        bocols>0 ? sizeof(double) : sizeof(float),
474 >                putbinary(zerobuf,
475 >                        tolower(otype)=='d' ? sizeof(double) : sizeof(float),
476                          1, stdout);
477 <        if (bocols > 0)
478 <                fwrite(&v, sizeof(double), 1, stdout);
479 <        else {
480 <                float   fval = v;
481 <                fwrite(&fval, sizeof(float), 1, stdout);
477 >        switch (otype) {
478 >        case 'D':
479 >                swap64((char *)&v, 1);
480 >                /* fall through */
481 >        case 'd':
482 >                putbinary(&v, sizeof(double), 1, stdout);
483 >                break;
484 >        case 'F':
485 >                swap32((char *)&fval, 1);
486 >                /* fall through */
487 >        case 'f':
488 >                putbinary(&fval, sizeof(float), 1, stdout);
489 >                break;
490          }
491   }
492  
# Line 435 | Line 501 | int  output
501          char  *inptr;
502          struct field  fmt;
503          int  res;
504 <        register struct field  *f;
504 >        struct field  *f;
505                                                  /* check for inline format */
506          for (inptr = spec; *inptr; inptr++)
507                  if (*inptr == '$')
# Line 448 | Line 514 | int  output
514                          eputs(": cannot open\n");
515                          quit(1);
516                  }
517 <                res = read(fd, inpbuf+1, INBSIZ-1);
517 >                res = read(fd, inpbuf+2, INBSIZ-2);
518                  if (res <= 0 || res >= INBSIZ-1) {
519                          eputs(spec);
520                          if (res < 0)
# Line 460 | Line 526 | int  output
526                          quit(1);
527                  }
528                  close(fd);
529 <                (inptr=inpbuf+1)[res] = '\0';
529 >                (inptr=inpbuf+2)[res] = '\0';
530          }
531          f = &fmt;                               /* get fields */
532          while ((res = readfield(&inptr)) != F_NUL) {
# Line 495 | Line 561 | int  output
561  
562   static int
563   readfield(                   /* get next field in format */
564 < register char  **pp
564 > char  **pp
565   )
566   {
567          int  type = F_NUL;
568          int  width = 0;
569 <        register char  *cp;
569 >        char  *cp;
570          
571          cp = inpbuf;
572          while (cp < &inpbuf[INBSIZ-1] && **pp != '\0') {
# Line 565 | Line 631 | getsvar(                         /* get string variabl
631   char  *svname
632   )
633   {
634 <        register struct strvar  *sv;
634 >        struct strvar  *sv;
635          
636          for (sv = svhead; sv != NULL; sv = sv->next)
637                  if (!strcmp(sv->name, svname))
# Line 584 | Line 650 | svpreset(                    /* preset a string variab
650   char  *eqn
651   )
652   {
653 <        register struct strvar  *sv;
654 <        register char  *val;
653 >        struct strvar  *sv;
654 >        char  *val;
655  
656          for (val = eqn; *val != '='; val++)
657                  if (!*val)
# Line 604 | Line 670 | char  *eqn
670   static void
671   clearrec(void)                  /* clear input record variables */
672   {
673 <        register struct field  *f;
673 >        struct field  *f;
674  
675          for (f = inpfmt; f != NULL; f = f->next)
676                  switch (f->type & F_TYP) {
# Line 622 | Line 688 | clearrec(void)                 /* clear input record variables */
688  
689  
690   static int
691 < getrec(void)                                /* get next record from file */
691 > getrec(void)                            /* get next record from file */
692   {
693          int  eatline;
694 <        register struct field  *f;
695 <        
694 >        struct field  *f;
695 >
696          while (ipb.chr != EOF) {
697 <                eatline = !igneol && ipb.chr != '\n';
632 <                if (blnkeq)             /* beware of nbsynch() */
697 >                if (blnkeq) {           /* beware of nbsynch() */
698                          while (isblnk(ipb.chr))
699 <                                scaninp();
699 >                                resetinp();
700 >                        if (ipb.chr == EOF)
701 >                                return(0);
702 >                }
703 >                eatline = (!igneol && ipb.chr != '\n');
704                  clearrec();             /* start with fresh record */
705                  for (f = inpfmt; f != NULL; f = f->next)
706                          if (getfield(f) == -1)
707                                  break;
708                  if (f == NULL) {
709 <                        advinp();
709 >                        advinp();       /* got one! */
710                          return(1);
711                  }
712 <                resetinp();
712 >                resetinp();             /* eat false start */
713                  if (eatline) {          /* eat rest of line */
714                          while (ipb.chr != '\n') {
715                                  if (ipb.chr == EOF)
716                                          return(0);
717 <                                scaninp();
717 >                                resetinp();
718                          }
719 <                        scaninp();
651 <                        advinp();
719 >                        resetinp();
720                  }
721          }
722          return(0);
# Line 657 | Line 725 | getrec(void)                                /* get nex
725  
726   static int
727   getfield(                             /* get next field */
728 < register struct field  *f
728 > struct field  *f
729   )
730   {
731          static char  buf[RMAXWORD+1];            /* no recursion! */
732          int  delim, inword;
733          double  d;
734          char  *np;
735 <        register char  *cp;
735 >        char  *cp;
736  
737          switch (f->type & F_TYP) {
738          case T_LIT:
# Line 751 | Line 819 | register struct field  *f
819   static void
820   putrec(void)                                /* output a record */
821   {
822 <        char  fmt[32];
823 <        register int  n;
824 <        register struct field  *f;
822 >        char  fmt[32], typ[16];
823 >        int  n;
824 >        struct field  *f;
825          int  adlast, adnext;
826 +        double  dv, av;
827          
828          adlast = 0;
829          for (f = outfmt; f != NULL; f = f->next) {
# Line 785 | Line 854 | putrec(void)                                /* output
854                          break;
855                  case T_NUM:
856                          n = f->type & F_WID;
857 +                        dv = evalue(f->f.ne);
858 +                        av = fabs(dv);
859 +                        if (n <= 9)
860 +                                strcpy(typ, "g");
861 +                        else
862 +                                sprintf(typ, ".%de", n-5);
863 +                        if (av < 1L<<31) {
864 +                                long    iv = (int)(av + .5);
865 +                                if (iv && fabs(av-iv) <= av*1e-14)
866 +                                        strcpy(typ, ".0f");
867 +                        }
868                          if (adlast && adnext)
869 <                                strcpy(fmt, "%g");
869 >                                sprintf(fmt, "%%%s", typ);
870                          else if (adlast)
871 <                                sprintf(fmt, "%%-%dg", n);
871 >                                sprintf(fmt, "%%-%d%s", n, typ);
872                          else
873 <                                sprintf(fmt, "%%%dg", n);
874 <                        printf(fmt, evalue(f->f.ne));
873 >                                sprintf(fmt, "%%%d%s", n, typ);
874 >                        printf(fmt, dv);
875                          adlast = 1;
876                          break;
877                  }
# Line 845 | Line 925 | resetinp(void)                      /* rewind position
925                  ipb.beg = ipb.end;
926          ipb.pos = ipb.beg;
927          ipb.chr = *ipb.pos;
928 +        if (passive)                    /* transmit unmatched character? */
929 +                fputc(ipb.chr, stdout);
930          if (++ipb.beg >= &inpbuf[INBSIZ])
931                  ipb.beg = inpbuf;
932          scaninp();

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines