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.13 by greg, Wed Jun 9 16:00:13 2004 UTC vs.
Revision 1.36 by greg, Sun Mar 13 16:11:48 2022 UTC

# Line 2 | Line 2
2   static const char RCSid[] = "$Id$";
3   #endif
4   /*
5 < *  rcalc.c - record calculator program.
5 > * rcalc.c - record calculator program.
6   *
7 < *     9/11/87
7 > * 9/11/87
8   */
9  
10 < #include  <stdlib.h>
11 < #include  <fcntl.h>
12 < #include  <stdio.h>
13 < #include  <string.h>
14 < #include  <math.h>
15 < #include  <ctype.h>
10 > #include <stdlib.h>
11 > #include <math.h>
12 > #include <ctype.h>
13  
14 < #include  "platform.h"
15 < #include  "rterror.h"
16 < #include  "rtmisc.h"
17 < #include  "rtio.h"
18 < #include  "calcomp.h"
14 > #include "platform.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')
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')
23 > #define isblnk(c) (igneol ? isspace(c) : ((c)==' ')|((c)=='\t'))
24  
25 < #define  INBSIZ         4096    /* longest record */
26 < #define  MAXCOL         32      /* number of columns recorded */
25 > #define INBSIZ  16384   /* longest record */
26 > #define MAXCOL  32      /* number of columns recorded */
27  
28 <                                /* field type specifications */
29 < #define  F_NUL          0               /* empty */
30 < #define  F_TYP          0x7000          /* mask for type */
31 < #define  F_WID          0x0fff          /* mask for width */
32 < #define  T_LIT          0x1000          /* string literal */
33 < #define  T_STR          0x2000          /* string variable */
34 < #define  T_NUM          0x3000          /* numeric value */
28 >                        /* field type specifications */
29 > #define F_NUL   0               /* empty */
30 > #define F_TYP   0x7000          /* mask for type */
31 > #define F_WID   0x0fff          /* mask for width */
32 > #define T_LIT   0x1000          /* string literal */
33 > #define T_STR   0x2000          /* string variable */
34 > #define T_NUM   0x3000          /* numeric value */
35  
36 < struct strvar {                 /* string variable */
37 <        char  *name;
38 <        char  *val;
39 <        char  *preset;
40 <        struct strvar  *next;
36 > struct strvar {         /* string variable */
37 >        char *name;
38 >        char *val;
39 >        char *preset;
40 >        struct strvar *next;
41   };
42  
43 < struct field {                  /* record format structure */
44 <        int  type;                      /* type of field (& width) */
43 > struct field {          /* record format structure */
44 >        int type;               /* type of field (& width) */
45          union {
46 <                char  *sl;                      /* string literal */
47 <                struct strvar  *sv;             /* string variable */
48 <                char  *nv;                      /* numeric variable */
49 <                EPNODE  *ne;                    /* numeric expression */
50 <        } f;                            /* field contents */
51 <        struct field  *next;            /* next field in record */
46 >                char *sl;               /* string literal */
47 >                struct strvar *sv;      /* string variable */
48 >                char *nv;               /* numeric variable */
49 >                EPNODE *ne;             /* numeric expression */
50 >        } f;                    /* field contents */
51 >        struct field *next;     /* next field in record */
52   };
53  
54 < #define  savqstr(s)     strcpy(emalloc(strlen(s)+1),s)
55 < #define  freqstr(s)     efree(s)
54 > #define savqstr(s) strcpy(emalloc(strlen(s)+1),s)
55 > #define freqstr(s) efree(s)
56  
57   static int getinputrec(FILE *fp);
58 < static void scaninp(void), advinp(void), resetinp(void);
58 > static void scaninp(void), skipinp(void), advinp(int skip);
59   static void putrec(void), putout(void), nbsynch(void);
60   static int getrec(void);
61   static void execute(char *file);
# Line 72 | Line 69 | static void bchanset(int n, double v);
69   static struct strvar* getsvar(char *svname);
70   static double l_in(char *);
71  
72 < struct field  *inpfmt = NULL;   /* input record format */
73 < struct field  *outfmt = NULL;   /* output record structure */
74 < struct strvar  *svhead = NULL;  /* string variables */
72 > struct field *inpfmt = NULL;    /* input record format */
73 > struct field *outfmt = NULL;    /* output record structure */
74 > struct strvar *svhead = NULL;   /* string variables */
75  
76 < int  blnkeq = 1;                /* blanks compare equal? */
77 < 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  nbicols = 0;               /* number of binary input columns */
85 < int  bocols = 0;                /* produce binary output columns */
86 < char  inpbuf[INBSIZ];           /* input buffer */
87 < double  colval[MAXCOL];         /* input column values */
88 < unsigned long  colflg = 0;      /* column retrieved flags */
89 < int  colpos;                    /* output column position */
76 > long incnt = 0;                 /* limit number of input records? */
77 > long outcnt = 0;                /* limit number of output records? */
78  
79 < int  nowarn = 0;                /* non-fatal diagnostic output */
80 < int  unbuff = 0;                /* unbuffered output (flush each record) */
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 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 */
90 > int colpos;                     /* output column position */
91  
92 + int nowarn = 0;                 /* non-fatal diagnostic output */
93 + int unbuff = 0;                 /* unbuffered output (flush each record) */
94 +
95   struct {
96 <        FILE  *fin;                     /* input file */
97 <        int  chr;                       /* next character */
98 <        char  *beg;                     /* home position */
99 <        char  *pos;                     /* scan position */
100 <        char  *end;                     /* read position */
101 < } ipb;                          /* circular lookahead buffer */
96 >        FILE *fin;              /* input file */
97 >        int chr;                /* next character */
98 >        char *beg;              /* home position */
99 >        char *pos;              /* scan position */
100 >        char *end;              /* read position */
101 > } ipb;                  /* circular lookahead buffer */
102  
103  
104   int
105   main(
106 < int  argc,
107 < char  *argv[]
106 > int argc,
107 > char *argv[]
108   )
109   {
110 <        int  i;
110 >        char *fpath;
111 >        int i;
112  
113          esupport |= E_VARIABLE|E_FUNCTION|E_INCHAN|E_OUTCHAN|E_RCONST;
114          esupport &= ~(E_REDEFW);
115  
116 < #ifdef  BIGGERLIB
116 > #ifdef BIGGERLIB
117          biggerlib();
118   #endif
119          varset("PI", ':', 3.14159265358979323846);
# Line 126 | Line 128 | char  *argv[]
128                          igneol = !igneol;
129                          break;
130                  case 'p':
131 <                        passive = !passive;
131 >                        passive = 1;
132                          break;
133 +                case 'P':
134 +                        passive = -1;
135 +                        break;
136                  case 't':
137                          sepchar = argv[i][2];
138                          break;
# Line 135 | Line 140 | char  *argv[]
140                          svpreset(argv[++i]);
141                          break;
142                  case 'f':
143 <                        fcompile(argv[++i]);
143 >                        fpath = getpath(argv[++i], getrlibpath(), 0);
144 >                        if (fpath == NULL) {
145 >                                eputs(argv[0]);
146 >                                eputs(": cannot find file '");
147 >                                eputs(argv[i]);
148 >                                eputs("'\n");
149 >                                quit(1);
150 >                        }
151 >                        fcompile(fpath);
152                          break;
153                  case 'e':
154                          scompile(argv[++i], NULL, 0);
# Line 146 | Line 159 | char  *argv[]
159                  case 'i':
160                          switch (argv[i][2]) {
161                          case '\0':
162 +                                itype = 'a';
163                                  nbicols = 0;
164                                  readfmt(argv[++i], 0);
165                                  break;
166 +                        case 'n':
167 +                                incnt = atol(argv[++i]);
168 +                                break;
169                          case 'a':
170 +                                itype = 'a';
171                                  nbicols = 0;
172                                  break;
173                          case 'd':
174 +                        case 'D':
175 +                                itype = argv[i][2];
176                                  if (isdigit(argv[i][3]))
177                                          nbicols = atoi(argv[i]+3);
178                                  else
# Line 164 | Line 184 | char  *argv[]
184                                  }
185                                  break;
186                          case 'f':
187 +                        case 'F':
188 +                                itype = argv[i][2];
189                                  if (isdigit(argv[i][3]))
190 <                                        nbicols = -atoi(argv[i]+3);
190 >                                        nbicols = atoi(argv[i]+3);
191                                  else
192 <                                        nbicols = -1;
193 <                                if (-nbicols*sizeof(float) > INBSIZ) {
192 >                                        nbicols = 1;
193 >                                if (nbicols*sizeof(float) > INBSIZ) {
194                                          eputs(argv[0]);
195                                          eputs(": too many input columns\n");
196                                          quit(1);
# Line 181 | Line 203 | char  *argv[]
203                  case 'o':
204                          switch (argv[i][2]) {
205                          case '\0':
206 <                                bocols = 0;
206 >                                otype = 'a';
207                                  readfmt(argv[++i], 1);
208                                  break;
209 +                        case 'n':
210 +                                outcnt = atol(argv[++i]);
211 +                                break;
212                          case 'a':
213 <                                bocols = 0;
213 >                                otype = 'a';
214                                  break;
215                          case 'd':
216 <                                bocols = 1;
192 <                                break;
216 >                        case 'D':
217                          case 'f':
218 <                                bocols = -1;
218 >                        case 'F':
219 >                                otype = argv[i][2];
220                                  break;
221 +                        default:
222 +                                goto userr;
223                          }
224                          break;
225                  case 'w':
# Line 205 | Line 232 | char  *argv[]
232                  userr:
233                          eputs("Usage: ");
234                          eputs(argv[0]);
235 < eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n");
235 > eputs(" [-b][-l][-n][-p|-P][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n");
236                          quit(1);
237                  }
238 <
239 <        if (noinput) {          /* produce a single output record */
238 >        if (otype != 'a')
239 >                SET_FILE_BINARY(stdout);
240 > #ifdef getc_unlocked            /* avoid lock/unlock overhead */
241 >        flockfile(stdout);
242 > #endif
243 >        if (noinput) {  /* produce a single output record */
244 >                if (i < argc) {
245 >                        eputs(argv[0]);
246 >                        eputs(": file argument(s) incompatible with -n\n");
247 >                        quit(1);
248 >                }
249                  eclock++;
250                  putout();
251                  quit(0);
252          }
253 <
254 <        if (blnkeq)             /* for efficiency */
253 >        if (passive && (inpfmt == NULL) | (outfmt == NULL)) {
254 >                eputs(argv[0]);
255 >                eputs(": options -p and -P require -i and -o formats\n");
256 >                quit(1);
257 >        }
258 >        if (blnkeq)     /* for efficiency */
259                  nbsynch();
260  
261 <        if (i == argc)          /* from stdin */
261 >        if (i == argc)  /* from stdin */
262                  execute(NULL);
263 <        else                    /* from one or more files */
263 >        else            /* from one or more files */
264                  for ( ; i < argc; i++)
265                          execute(argv[i]);
266          
# Line 230 | Line 270 | eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e
270  
271  
272   static void
273 < nbsynch(void)               /* non-blank starting synch character */
273 > nbsynch(void)   /* non-blank starting synch character */
274   {
275          if (inpfmt == NULL || (inpfmt->type & F_TYP) != T_LIT)
276                  return;
# Line 243 | Line 283 | nbsynch(void)               /* non-blank starting sync
283  
284   static int
285   getinputrec(            /* get next input record */
286 < FILE  *fp
286 > FILE *fp
287   )
288   {
289 <        if (inpfmt != NULL)
290 <                return(getrec());
291 <        if (nbicols > 0)
292 <                return(fread(inpbuf, sizeof(double),
293 <                                        nbicols, fp) == nbicols);
294 <        if (nbicols < 0)
295 <                return(fread(inpbuf, sizeof(float),
296 <                                        -nbicols, fp) == -nbicols);
289 >        if ((itype == 'd') | (itype == 'D')) {
290 >                if (getbinary(inpbuf, sizeof(double), nbicols, fp) != nbicols)
291 >                        return(0);
292 >                if (itype == 'D')
293 >                        swap64(inpbuf, nbicols);
294 >                return(1);
295 >        }
296 >        if ((itype == 'f') | (itype == 'F')) {
297 >                if (getbinary(inpbuf, sizeof(float), nbicols, fp) != nbicols)
298 >                        return(0);
299 >                if (itype == 'F')
300 >                        swap32(inpbuf, nbicols);
301 >                return(1);
302 >        }
303          return(fgets(inpbuf, INBSIZ, fp) != NULL);
304   }
305  
306  
307   static void
308 < execute(           /* process a file */
309 < char  *file
308 > execute(        /* process a file */
309 > char *file
310   )
311   {
312 <        int  conditional = vardefined("cond");
313 <        long  nrecs = 0;
314 <        long  nout = 0;
315 <        FILE  *fp;
312 >        static char     condVN[] = "cond";
313 >        static char     recnoVN[] = "recno";
314 >        static char     outnoVN[] = "outno";
315 >        const int       conditional = vardefined(condVN);
316 >        const int       set_recno = (varlookup(recnoVN) != NULL);
317 >        const int       set_outno = (varlookup(outnoVN) != NULL);
318 >        long            nrecs = 0;
319 >        long            nout = 0;
320 >        FILE            *fp;
321          
322          if (file == NULL)
323                  fp = stdin;
# Line 275 | Line 326 | char  *file
326                  eputs(": cannot open\n");
327                  quit(1);
328          }
329 +        if (itype != 'a')
330 +                SET_FILE_BINARY(fp);
331 + #ifdef getc_unlocked            /* avoid lock/unlock overhead */
332 +        flockfile(fp);
333 + #endif
334 +        if (conditional == ':') {
335 +                eputs(condVN);
336 +                eputs(": defined as constant\n");
337 +                quit(1);
338 +        }
339          if (inpfmt != NULL)
340                  initinp(fp);
341          
342 <        while (getinputrec(fp)) {
343 <                varset("recno", '=', (double)++nrecs);
342 >        while (inpfmt != NULL ? getrec() : getinputrec(fp)) {
343 >                ++nrecs;
344 >                if (set_recno)
345 >                        varset(recnoVN, '=', (double)nrecs);
346 >                if (set_outno)
347 >                        varset(outnoVN, '=', (double)(nout+1));
348                  colflg = 0;
349                  eclock++;
350 <                if (!conditional || varvalue("cond") > 0.0) {
351 <                        varset("outno", '=', (double)++nout);
350 >                if (!conditional || varvalue(condVN) > 0.0) {
351 >                        if (inpfmt != NULL)
352 >                                advinp(0);
353                          putout();
354 +                        ++nout;
355 +                } else if (inpfmt != NULL) {
356 +                        advinp(1);
357                  }
358 +                if (incnt && nrecs >= incnt)
359 +                        break;
360 +                if (outcnt && nout >= outcnt)
361 +                        break;
362          }
363          fclose(fp);
364   }
365  
366  
367   static void
368 < putout(void)                /* produce an output record */
368 > putout(void)            /* produce an output record */
369   {
370  
371          colpos = 0;
372          if (outfmt != NULL)
373                  putrec();
374 <        else if (bocols)
302 <                chanout(bchanset);
303 <        else
374 >        else if (otype == 'a')
375                  chanout(chanset);
376 <        if (colpos && !bocols)
376 >        else
377 >                chanout(bchanset);
378 >        if (colpos && otype == 'a')
379                  putchar('\n');
380          if (unbuff)
381                  fflush(stdout);
# Line 312 | Line 385 | putout(void)                /* produce an output recor
385   static double
386   l_in(char *funame)      /* function call for $channel */
387   {
388 <        int  n;
389 <        register char  *cp;
388 >        int n;
389 >        char *cp;
390                          /* get argument as integer */
391          n = (int)(argument(1) + .5);
392          if (n != 0)     /* return channel value */
# Line 321 | Line 394 | l_in(char *funame)     /* function call for $channel */
394                          /* determine number of channels */
395          if (noinput || inpfmt != NULL)
396                  return(0);
397 <        if (nbicols > 0)
397 >        if (nbicols)
398                  return(nbicols);
326        if (nbicols < 0)
327                return(-nbicols);
399          cp = inpbuf;    /* need to count */
400          for (n = 0; *cp; )
401                  if (blnkeq && isspace(sepchar)) {
# Line 342 | Line 413 | l_in(char *funame)     /* function call for $channel */
413   }
414  
415   double
416 < chanvalue(            /* return value for column n */
417 < int  n
416 > chanvalue(      /* return value for column n */
417 > int n
418   )
419   {
420 <        int  i;
421 <        register char  *cp;
420 >        int i;
421 >        char *cp;
422  
423          if (noinput || inpfmt != NULL) {
424                  eputs("no column input\n");
# Line 357 | Line 428 | int  n
428                  eputs("illegal channel number\n");
429                  quit(1);
430          }
431 <        if (nbicols > 0) {
431 >        if (nbicols) {
432                  if (n > nbicols)
433                          return(0.0);
434 <                cp = inpbuf + (n-1)*sizeof(double);
435 <                return(*(double *)cp);
436 <        }
437 <        if (nbicols < 0) {
367 <                if (n > -nbicols)
368 <                        return(0.0);
434 >                if ((itype == 'd') | (itype == 'D')) {
435 >                        cp = inpbuf + (n-1)*sizeof(double);
436 >                        return(*(double *)cp);
437 >                }
438                  cp = inpbuf + (n-1)*sizeof(float);
439                  return(*(float *)cp);
440          }
# Line 383 | Line 452 | int  n
452                          while (*cp && *cp++ != sepchar)
453                                  ;
454  
455 <        while (isspace(*cp))            /* some atof()'s don't like tabs */
455 >        while (isspace(*cp))    /* some atof()'s don't like tabs */
456                  cp++;
457  
458          if (n <= MAXCOL) {
# Line 395 | Line 464 | int  n
464  
465  
466   void
467 < chanset(                   /* output column n */
468 < int  n,
469 < double  v
467 > chanset(                /* output column n */
468 > int n,
469 > double v
470   )
471   {
472 <        if (colpos == 0)                /* no leading separator */
472 >        if (colpos == 0)                /* no leading separator */
473                  colpos = 1;
474          while (colpos < n) {
475                  putchar(sepchar);
# Line 411 | Line 480 | double  v
480  
481  
482   void
483 < bchanset(                   /* output binary channel n */
484 < int  n,
485 < double  v
483 > bchanset(               /* output binary channel n */
484 > int n,
485 > double v
486   )
487   {
488          static char     zerobuf[sizeof(double)];
489 +        const int       otlen = ((otype == 'd') | (otype == 'D')) ?
490 +                                        sizeof(double) : sizeof(float);
491 +        float   fval = v;
492  
493          while (++colpos < n)
494 <                fwrite(zerobuf,
495 <                        bocols>0 ? sizeof(double) : sizeof(float),
496 <                        1, stdout);
497 <        if (bocols > 0)
498 <                fwrite(&v, sizeof(double), 1, stdout);
499 <        else {
500 <                float   fval = v;
501 <                fwrite(&fval, sizeof(float), 1, stdout);
494 >                putbinary(zerobuf, otlen, 1, stdout);
495 >        switch (otype) {
496 >        case 'D':
497 >                swap64((char *)&v, 1);
498 >                /* fall through */
499 >        case 'd':
500 >                putbinary(&v, sizeof(double), 1, stdout);
501 >                break;
502 >        case 'F':
503 >                swap32((char *)&fval, 1);
504 >                /* fall through */
505 >        case 'f':
506 >                putbinary(&fval, sizeof(float), 1, stdout);
507 >                break;
508          }
509   }
510  
511  
512   static void
513 < readfmt(                   /* read record format */
514 < char  *spec,
515 < int  output
513 > readfmt(                /* read record format */
514 > char *spec,
515 > int output
516   )
517   {
518 <        int  fd;
519 <        char  *inptr;
520 <        struct field  fmt;
521 <        int  res;
522 <        register struct field  *f;
518 >        int fd;
519 >        char *inptr;
520 >        struct field fmt;
521 >        int res;
522 >        struct field *f;
523                                                  /* check for inline format */
524          for (inptr = spec; *inptr; inptr++)
525                  if (*inptr == '$')
526                          break;
527 <        if (*inptr)                             /* inline */
527 >        if (*inptr)                     /* inline */
528                  inptr = spec;
529 <        else {                                  /* from file */
529 >        else {                          /* from file */
530                  if ((fd = open(spec, 0)) == -1) {
531                          eputs(spec);
532                          eputs(": cannot open\n");
533                          quit(1);
534                  }
535 <                res = read(fd, inpbuf+1, INBSIZ-1);
535 >                res = read(fd, inpbuf+2, INBSIZ-2);
536                  if (res <= 0 || res >= INBSIZ-1) {
537                          eputs(spec);
538                          if (res < 0)
# Line 466 | Line 544 | int  output
544                          quit(1);
545                  }
546                  close(fd);
547 <                (inptr=inpbuf+1)[res] = '\0';
547 >                (inptr=inpbuf+2)[res] = '\0';
548          }
549 <        f = &fmt;                               /* get fields */
549 >        f = &fmt;                       /* get fields */
550          while ((res = readfield(&inptr)) != F_NUL) {
551                  f->next = (struct field *)emalloc(sizeof(struct field));
552                  f = f->next;
# Line 500 | Line 578 | int  output
578  
579  
580   static int
581 < readfield(                   /* get next field in format */
582 < register char  **pp
581 > readfield(              /* get next field in format */
582 > char **pp
583   )
584   {
585 <        int  type = F_NUL;
586 <        int  width = 0;
587 <        register char  *cp;
585 >        int type = F_NUL;
586 >        int width = 0;
587 >        char *cp;
588          
589          cp = inpbuf;
590          while (cp < &inpbuf[INBSIZ-1] && **pp != '\0') {
# Line 567 | Line 645 | register char  **pp
645  
646  
647   struct strvar *
648 < getsvar(                         /* get string variable */
649 < char  *svname
648 > getsvar(                        /* get string variable */
649 > char *svname
650   )
651   {
652 <        register struct strvar  *sv;
652 >        struct strvar *sv;
653          
654          for (sv = svhead; sv != NULL; sv = sv->next)
655                  if (!strcmp(sv->name, svname))
# Line 586 | Line 664 | char  *svname
664  
665  
666   static void
667 < svpreset(                    /* preset a string variable */
668 < char  *eqn
667 > svpreset(               /* preset a string variable */
668 > char *eqn
669   )
670   {
671 <        register struct strvar  *sv;
672 <        register char  *val;
671 >        struct strvar *sv;
672 >        char *val;
673  
674          for (val = eqn; *val != '='; val++)
675                  if (!*val)
# Line 610 | Line 688 | char  *eqn
688   static void
689   clearrec(void)                  /* clear input record variables */
690   {
691 <        register struct field  *f;
691 >        struct field *f;
692  
693          for (f = inpfmt; f != NULL; f = f->next)
694                  switch (f->type & F_TYP) {
# Line 628 | Line 706 | clearrec(void)                 /* clear input record variables */
706  
707  
708   static int
709 < getrec(void)                            /* get next record from file */
709 > getrec(void)                    /* get next record from file */
710   {
711 <        int  eatline;
712 <        register struct field  *f;
713 <        
711 >        int eatline;
712 >        struct field *f;
713 >
714          while (ipb.chr != EOF) {
715 <                eatline = !igneol && ipb.chr != '\n';
638 <                if (blnkeq)             /* beware of nbsynch() */
715 >                if (blnkeq) {           /* beware of nbsynch() */
716                          while (isblnk(ipb.chr))
717 <                                resetinp();
717 >                                skipinp();
718 >                        if (ipb.chr == EOF)
719 >                                return(0);
720 >                }
721 >                eatline = !igneol & (ipb.chr != '\n');
722                  clearrec();             /* start with fresh record */
723                  for (f = inpfmt; f != NULL; f = f->next)
724 <                        if (getfield(f) == -1)
724 >                        if (!getfield(f))
725                                  break;
726 <                if (f == NULL) {
646 <                        advinp();       /* got one! */
726 >                if (f == NULL)          /* got one? */
727                          return(1);
728 <                }
729 <                resetinp();             /* eat false start */
650 <                if (eatline) {          /* eat rest of line */
728 >                skipinp();              /* else eat false start */
729 >                if (eatline) {          /* eat rest of line */
730                          while (ipb.chr != '\n') {
731                                  if (ipb.chr == EOF)
732                                          return(0);
733 <                                resetinp();
733 >                                skipinp();
734                          }
735 <                        resetinp();
735 >                        skipinp();
736                  }
737          }
738          return(0);
# Line 661 | Line 740 | getrec(void)                           /* get next record from file */
740  
741  
742   static int
743 < getfield(                             /* get next field */
744 < register struct field  *f
743 > getfield(                       /* get next field */
744 > struct field *f
745   )
746   {
747 <        static char  buf[RMAXWORD+1];            /* no recursion! */
748 <        int  delim, inword;
749 <        double  d;
750 <        char  *np;
751 <        register char  *cp;
747 >        static char buf[RMAXWORD+1];    /* no recursion! */
748 >        int delim, inword;
749 >        double d;
750 >        char *np;
751 >        char *cp;
752  
753          switch (f->type & F_TYP) {
754          case T_LIT:
# Line 677 | Line 756 | register struct field  *f
756                  do {
757                          if (blnkeq && isblnk(*cp)) {
758                                  if (!isblnk(ipb.chr))
759 <                                        return(-1);
759 >                                        return(0);
760                                  do
761                                          cp++;
762                                  while (isblnk(*cp));
# Line 688 | Line 767 | register struct field  *f
767                                  cp++;
768                                  scaninp();
769                          } else
770 <                                return(-1);
770 >                                return(0);
771                  } while (*cp);
772 <                return(0);
772 >                break;
773          case T_STR:
774                  if (f->next == NULL || (f->next->type & F_TYP) != T_LIT)
775                          delim = EOF;
# Line 698 | Line 777 | register struct field  *f
777                          delim = f->next->f.sl[0];
778                  cp = buf;
779                  do {
780 <                        if (ipb.chr == EOF || ipb.chr == '\n')
780 >                        if ((ipb.chr == EOF) | (ipb.chr == '\n'))
781                                  inword = 0;
782                          else if (blnkeq && delim != EOF)
783                                  inword = isblnk(delim) ?
# Line 715 | Line 794 | register struct field  *f
794                  if (f->f.sv->val == NULL)
795                          f->f.sv->val = savqstr(buf);    /* first setting */
796                  else if (strcmp(f->f.sv->val, buf))
797 <                        return(-1);                     /* doesn't match! */
798 <                return(0);
797 >                        return(0);                      /* doesn't match! */
798 >                break;
799          case T_NUM:
800                  if (f->next == NULL || (f->next->type & F_TYP) != T_LIT)
801                          delim = EOF;
# Line 746 | Line 825 | register struct field  *f
825                          varset(f->f.nv, '=', d);        /* first setting */
826                  else if ((d = (varvalue(f->f.nv)-d)/(d==0.?1.:d)) > .001
827                                  || d < -.001)
828 <                        return(-1);                     /* doesn't match! */
829 <                return(0);
828 >                        return(0);                      /* doesn't match! */
829 >                break;
830          }
831 <        return -1; /* pro forma return */
831 >        return(1);      /* success! */
832   }
833  
834  
835   static void
836 < putrec(void)                                /* output a record */
836 > putrec(void)                            /* output a record */
837   {
838 <        char  fmt[32];
839 <        register int  n;
840 <        register struct field  *f;
841 <        int  adlast, adnext;
838 >        char fmt[32], typ[16];
839 >        int n;
840 >        struct field *f;
841 >        int adlast, adnext;
842 >        double dv, av;
843          
844          adlast = 0;
845          for (f = outfmt; f != NULL; f = f->next) {
846 <                adnext =        blnkeq &&
846 >                adnext =        blnkeq &&
847                                  f->next != NULL &&
848                                  !( (f->next->type&F_TYP) == T_LIT &&
849                                          f->next->f.sl[0] == ' ' );
# Line 790 | Line 870 | putrec(void)                                /* output
870                          break;
871                  case T_NUM:
872                          n = f->type & F_WID;
873 +                        dv = evalue(f->f.ne);
874 +                        av = fabs(dv);
875 +                        if (n <= 9)
876 +                                strcpy(typ, "g");
877 +                        else
878 +                                sprintf(typ, ".%de", n-5);
879 +                        if (av < 1L<<31) {
880 +                                long    iv = (int)(av + .5);
881 +                                if (iv && fabs(av-iv) <= av*1e-14)
882 +                                        strcpy(typ, ".0f");
883 +                        }
884                          if (adlast && adnext)
885 <                                strcpy(fmt, "%g");
885 >                                sprintf(fmt, "%%%s", typ);
886                          else if (adlast)
887 <                                sprintf(fmt, "%%-%dg", n);
887 >                                sprintf(fmt, "%%-%d%s", n, typ);
888                          else
889 <                                sprintf(fmt, "%%%dg", n);
890 <                        printf(fmt, evalue(f->f.ne));
889 >                                sprintf(fmt, "%%%d%s", n, typ);
890 >                        printf(fmt, dv);
891                          adlast = 1;
892                          break;
893                  }
# Line 805 | Line 896 | putrec(void)                                /* output
896  
897  
898   static void
899 < initinp(FILE  *fp)                     /* prepare lookahead buffer */
899 > initinp(FILE *fp)               /* prepare lookahead buffer */
900  
901   {
902          ipb.fin = fp;
903          ipb.beg = ipb.end = inpbuf;
904 <        ipb.pos = inpbuf-1;             /* position before beginning */
904 >        ipb.pos = inpbuf-1;             /* position before beginning */
905          ipb.chr = '\0';
906          scaninp();
907   }
908  
909  
910   static void
911 < scaninp(void)                       /* scan next character */
911 > scaninp(void)                   /* scan next character */
912   {
913          if (ipb.chr == EOF)
914                  return;
915          if (++ipb.pos >= &inpbuf[INBSIZ])
916                  ipb.pos = inpbuf;
917 <        if (ipb.pos == ipb.end) {               /* new character */
917 >        if (ipb.pos == ipb.end) {       /* new character */
918                  if ((ipb.chr = getc(ipb.fin)) != EOF) {
919                          *ipb.end = ipb.chr;
920                          if (++ipb.end >= &inpbuf[INBSIZ])
# Line 837 | Line 928 | scaninp(void)                       /* scan next chara
928  
929  
930   static void
931 < advinp(void)                        /* move home to current position */
931 > skipinp(void)                   /* rewind position and advance 1 */
932   {
933 <        ipb.beg = ipb.pos;
843 < }
844 <
845 <
846 < static void
847 < resetinp(void)                      /* rewind position and advance 1 */
848 < {
849 <        if (ipb.beg == NULL)            /* full */
933 >        if (ipb.beg == NULL)            /* can't fully rewind? */
934                  ipb.beg = ipb.end;
935          ipb.pos = ipb.beg;
936          ipb.chr = *ipb.pos;
937          if (passive)                    /* transmit unmatched character? */
938 <                fputc(ipb.chr, stdout);
938 >                putchar(ipb.chr);
939          if (++ipb.beg >= &inpbuf[INBSIZ])
940                  ipb.beg = inpbuf;
941          scaninp();
942   }
943  
944  
945 + static void
946 + advinp(int skip)                /* advance home to current position */
947 + {
948 +        if (!skip | (passive >= 0)) {
949 +                ipb.beg = ipb.pos;      /* no need to copy input */
950 +                return;
951 +        }
952 +        if (ipb.beg == NULL) {          /* buffer overflowed a bit? */
953 +                wputs("buffer overflow\n");
954 +                puts("\n*** MISSING DATA ***");
955 +                ipb.beg = ipb.end;
956 +        }
957 +        while (ipb.beg != ipb.pos) {    /* copy buffer to current */
958 +                putchar(*ipb.beg);
959 +                if (++ipb.beg >= &inpbuf[INBSIZ])
960 +                        ipb.beg = inpbuf;
961 +        }
962 + }
963 +
964 +
965   void
966 < eputs(char  *msg)
966 > eputs(char *msg)
967   {
968          fputs(msg, stderr);
969   }
970  
971  
972   void
973 < wputs(char  *msg)
973 > wputs(char *msg)
974   {
975          if (!nowarn)
976                  eputs(msg);
# Line 874 | Line 978 | wputs(char  *msg)
978  
979  
980   void
981 < quit(int  code)
981 > quit(int code)
982   {
983          exit(code);
984   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines