--- ray/src/cal/rcalc.c 2014/05/10 01:48:14 1.25 +++ ray/src/cal/rcalc.c 2022/03/12 15:50:13 1.33 @@ -1,61 +1,61 @@ #ifndef lint -static const char RCSid[] = "$Id: rcalc.c,v 1.25 2014/05/10 01:48:14 greg Exp $"; +static const char RCSid[] = "$Id: rcalc.c,v 1.33 2022/03/12 15:50:13 greg Exp $"; #endif /* - * rcalc.c - record calculator program. + * rcalc.c - record calculator program. * - * 9/11/87 + * 9/11/87 */ -#include -#include -#include +#include +#include +#include -#include "platform.h" -#include "rterror.h" -#include "rtmisc.h" -#include "rtio.h" -#include "calcomp.h" +#include "platform.h" +#include "rterror.h" +#include "rtmisc.h" +#include "rtio.h" +#include "calcomp.h" -#define isnum(c) (isdigit(c) || (c)=='-' || (c)=='.' \ - || (c)=='+' || (c)=='e' || (c)=='E') +#define isnum(c) (isdigit(c) || ((c)=='-') | ((c)=='.') \ + | ((c)=='+') | ((c)=='e') | ((c)=='E')) -#define isblnk(c) (igneol ? isspace(c) : (c)==' '||(c)=='\t') +#define isblnk(c) (igneol ? isspace(c) : ((c)==' ')|((c)=='\t')) -#define INBSIZ 16384 /* longest record */ -#define MAXCOL 32 /* number of columns recorded */ +#define INBSIZ 16384 /* longest record */ +#define MAXCOL 32 /* number of columns recorded */ /* field type specifications */ -#define F_NUL 0 /* empty */ -#define F_TYP 0x7000 /* mask for type */ -#define F_WID 0x0fff /* mask for width */ -#define T_LIT 0x1000 /* string literal */ -#define T_STR 0x2000 /* string variable */ -#define T_NUM 0x3000 /* numeric value */ +#define F_NUL 0 /* empty */ +#define F_TYP 0x7000 /* mask for type */ +#define F_WID 0x0fff /* mask for width */ +#define T_LIT 0x1000 /* string literal */ +#define T_STR 0x2000 /* string variable */ +#define T_NUM 0x3000 /* numeric value */ -struct strvar { /* string variable */ - char *name; - char *val; - char *preset; - struct strvar *next; +struct strvar { /* string variable */ + char *name; + char *val; + char *preset; + struct strvar *next; }; -struct field { /* record format structure */ - int type; /* type of field (& width) */ +struct field { /* record format structure */ + int type; /* type of field (& width) */ union { - char *sl; /* string literal */ - struct strvar *sv; /* string variable */ - char *nv; /* numeric variable */ - EPNODE *ne; /* numeric expression */ - } f; /* field contents */ - struct field *next; /* next field in record */ + char *sl; /* string literal */ + struct strvar *sv; /* string variable */ + char *nv; /* numeric variable */ + EPNODE *ne; /* numeric expression */ + } f; /* field contents */ + struct field *next; /* next field in record */ }; -#define savqstr(s) strcpy(emalloc(strlen(s)+1),s) -#define freqstr(s) efree(s) +#define savqstr(s) strcpy(emalloc(strlen(s)+1),s) +#define freqstr(s) efree(s) static int getinputrec(FILE *fp); -static void scaninp(void), advinp(void), resetinp(void); +static void scaninp(void), advinp(void), passinp(void), skipinp(void); static void putrec(void), putout(void), nbsynch(void); static int getrec(void); static void execute(char *file); @@ -69,48 +69,51 @@ static void bchanset(int n, double v); static struct strvar* getsvar(char *svname); static double l_in(char *); -struct field *inpfmt = NULL; /* input record format */ -struct field *outfmt = NULL; /* output record structure */ -struct strvar *svhead = NULL; /* string variables */ +struct field *inpfmt = NULL; /* input record format */ +struct field *outfmt = NULL; /* output record structure */ +struct strvar *svhead = NULL; /* string variables */ -int blnkeq = 1; /* blanks compare equal? */ -int igneol = 0; /* ignore end of line? */ -int passive = 0; /* passive mode (transmit unmatched input) */ -char sepchar = '\t'; /* input/output separator */ -int noinput = 0; /* no input records? */ -int itype = 'a'; /* input type (a/f/F/d/D) */ -int nbicols = 0; /* number of binary input columns */ -int otype = 'a'; /* output format (a/f/F/d/D) */ -char inpbuf[INBSIZ]; /* input buffer */ -double colval[MAXCOL]; /* input column values */ -unsigned long colflg = 0; /* column retrieved flags */ -int colpos; /* output column position */ +long incnt = 0; /* limit number of input records? */ +long outcnt = 0; /* limit number of output records? */ -int nowarn = 0; /* non-fatal diagnostic output */ -int unbuff = 0; /* unbuffered output (flush each record) */ +int blnkeq = 1; /* blanks compare equal? */ +int igneol = 0; /* ignore end of line? */ +int passive = 0; /* passive mode (transmit unmatched input) */ +char sepchar = '\t'; /* input/output separator */ +int noinput = 0; /* no input records? */ +int itype = 'a'; /* input type (a/f/F/d/D) */ +int nbicols = 0; /* number of binary input columns */ +int otype = 'a'; /* output format (a/f/F/d/D) */ +char inpbuf[INBSIZ]; /* input buffer */ +double colval[MAXCOL]; /* input column values */ +unsigned long colflg = 0; /* column retrieved flags */ +int colpos; /* output column position */ +int nowarn = 0; /* non-fatal diagnostic output */ +int unbuff = 0; /* unbuffered output (flush each record) */ + struct { - FILE *fin; /* input file */ - int chr; /* next character */ - char *beg; /* home position */ - char *pos; /* scan position */ - char *end; /* read position */ -} ipb; /* circular lookahead buffer */ + FILE *fin; /* input file */ + int chr; /* next character */ + char *beg; /* home position */ + char *pos; /* scan position */ + char *end; /* read position */ +} ipb; /* circular lookahead buffer */ int main( -int argc, -char *argv[] +int argc, +char *argv[] ) { - char *fpath; - int i; + char *fpath; + int i; esupport |= E_VARIABLE|E_FUNCTION|E_INCHAN|E_OUTCHAN|E_RCONST; esupport &= ~(E_REDEFW); -#ifdef BIGGERLIB +#ifdef BIGGERLIB biggerlib(); #endif varset("PI", ':', 3.14159265358979323846); @@ -125,8 +128,11 @@ char *argv[] igneol = !igneol; break; case 'p': - passive = !passive; + passive = 1; break; + case 'P': + passive = -1; + break; case 't': sepchar = argv[i][2]; break; @@ -157,6 +163,9 @@ char *argv[] nbicols = 0; readfmt(argv[++i], 0); break; + case 'n': + incnt = atol(argv[++i]); + break; case 'a': itype = 'a'; nbicols = 0; @@ -197,6 +206,9 @@ char *argv[] otype = 'a'; readfmt(argv[++i], 1); break; + case 'n': + outcnt = atol(argv[++i]); + break; case 'a': otype = 'a'; break; @@ -220,7 +232,7 @@ char *argv[] userr: eputs("Usage: "); eputs(argv[0]); -eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n"); +eputs(" [-b][-l][-n][-p|-P][-w][-u][-tS][-s svar=sval][-e expr][-f source][-i infmt][-o outfmt] [file]\n"); quit(1); } if (otype != 'a') @@ -228,7 +240,7 @@ eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e #ifdef getc_unlocked /* avoid lock/unlock overhead */ flockfile(stdout); #endif - if (noinput) { /* produce a single output record */ + if (noinput) { /* produce a single output record */ if (i < argc) { eputs(argv[0]); eputs(": file argument(s) incompatible with -n\n"); @@ -238,12 +250,17 @@ eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e putout(); quit(0); } - if (blnkeq) /* for efficiency */ + if (passive && (inpfmt == NULL) | (outfmt == NULL)) { + eputs(argv[0]); + eputs(": options -p and -P require -i and -o formats\n"); + quit(1); + } + if (blnkeq) /* for efficiency */ nbsynch(); - if (i == argc) /* from stdin */ + if (i == argc) /* from stdin */ execute(NULL); - else /* from one or more files */ + else /* from one or more files */ for ( ; i < argc; i++) execute(argv[i]); @@ -253,7 +270,7 @@ eputs(" [-b][-l][-n][-p][-w][-u][-tS][-s svar=sval][-e static void -nbsynch(void) /* non-blank starting synch character */ +nbsynch(void) /* non-blank starting synch character */ { if (inpfmt == NULL || (inpfmt->type & F_TYP) != T_LIT) return; @@ -266,20 +283,18 @@ nbsynch(void) /* non-blank starting sync static int getinputrec( /* get next input record */ -FILE *fp +FILE *fp ) { - if (inpfmt != NULL) - return(getrec()); - if (tolower(itype) == 'd') { - if (fread(inpbuf, sizeof(double), nbicols, fp) != nbicols) + if ((itype == 'd') | (itype == 'D')) { + if (getbinary(inpbuf, sizeof(double), nbicols, fp) != nbicols) return(0); if (itype == 'D') swap64(inpbuf, nbicols); return(1); } - if (tolower(itype) == 'f') { - if (fread(inpbuf, sizeof(float), nbicols, fp) != nbicols) + if ((itype == 'f') | (itype == 'F')) { + if (getbinary(inpbuf, sizeof(float), nbicols, fp) != nbicols) return(0); if (itype == 'F') swap32(inpbuf, nbicols); @@ -290,14 +305,19 @@ FILE *fp static void -execute( /* process a file */ -char *file +execute( /* process a file */ +char *file ) { - int conditional = vardefined("cond"); - long nrecs = 0; - long nout = 0; - FILE *fp; + static char condVN[] = "cond"; + static char recnoVN[] = "recno"; + static char outnoVN[] = "outno"; + const int conditional = vardefined(condVN); + const int set_recno = (varlookup(recnoVN) != NULL); + const int set_outno = (varlookup(outnoVN) != NULL); + long nrecs = 0; + long nout = 0; + FILE *fp; if (file == NULL) fp = stdin; @@ -311,25 +331,43 @@ char *file #ifdef getc_unlocked /* avoid lock/unlock overhead */ flockfile(fp); #endif + if (conditional == ':') { + eputs(condVN); + eputs(": defined as constant\n"); + quit(1); + } if (inpfmt != NULL) initinp(fp); - while (getinputrec(fp)) { - varset("recno", '=', (double)++nrecs); - varset("outno", '=', (double)(nout+1)); + while (inpfmt != NULL ? getrec() : getinputrec(fp)) { + ++nrecs; + if (set_recno) + varset(recnoVN, '=', (double)nrecs); + if (set_outno) + varset(outnoVN, '=', (double)(nout+1)); colflg = 0; eclock++; - if (!conditional || varvalue("cond") > 0.0) { + if (!conditional || varvalue(condVN) > 0.0) { putout(); ++nout; + advinp(); + } else if (inpfmt != NULL) { + if (passive < 0) + passinp(); + else + advinp(); } + if (incnt && nrecs >= incnt) + break; + if (outcnt && nout >= outcnt) + break; } fclose(fp); } static void -putout(void) /* produce an output record */ +putout(void) /* produce an output record */ { colpos = 0; @@ -349,8 +387,8 @@ putout(void) /* produce an output recor static double l_in(char *funame) /* function call for $channel */ { - int n; - char *cp; + int n; + char *cp; /* get argument as integer */ n = (int)(argument(1) + .5); if (n != 0) /* return channel value */ @@ -377,12 +415,12 @@ l_in(char *funame) /* function call for $channel */ } double -chanvalue( /* return value for column n */ -int n +chanvalue( /* return value for column n */ +int n ) { - int i; - char *cp; + int i; + char *cp; if (noinput || inpfmt != NULL) { eputs("no column input\n"); @@ -395,7 +433,7 @@ int n if (nbicols) { if (n > nbicols) return(0.0); - if (tolower(itype) == 'd') { + if ((itype == 'd') | (itype == 'D')) { cp = inpbuf + (n-1)*sizeof(double); return(*(double *)cp); } @@ -416,7 +454,7 @@ int n while (*cp && *cp++ != sepchar) ; - while (isspace(*cp)) /* some atof()'s don't like tabs */ + while (isspace(*cp)) /* some atof()'s don't like tabs */ cp++; if (n <= MAXCOL) { @@ -428,12 +466,12 @@ int n void -chanset( /* output column n */ -int n, -double v +chanset( /* output column n */ +int n, +double v ) { - if (colpos == 0) /* no leading separator */ + if (colpos == 0) /* no leading separator */ colpos = 1; while (colpos < n) { putchar(sepchar); @@ -444,53 +482,53 @@ double v void -bchanset( /* output binary channel n */ -int n, -double v +bchanset( /* output binary channel n */ +int n, +double v ) { static char zerobuf[sizeof(double)]; + const int otlen = ((otype == 'd') | (otype == 'D')) ? + sizeof(double) : sizeof(float); float fval = v; while (++colpos < n) - fwrite(zerobuf, - tolower(otype)=='d' ? sizeof(double) : sizeof(float), - 1, stdout); + putbinary(zerobuf, otlen, 1, stdout); switch (otype) { case 'D': swap64((char *)&v, 1); /* fall through */ case 'd': - fwrite(&v, sizeof(double), 1, stdout); + putbinary(&v, sizeof(double), 1, stdout); break; case 'F': swap32((char *)&fval, 1); /* fall through */ case 'f': - fwrite(&fval, sizeof(float), 1, stdout); + putbinary(&fval, sizeof(float), 1, stdout); break; } } static void -readfmt( /* read record format */ -char *spec, -int output +readfmt( /* read record format */ +char *spec, +int output ) { - int fd; - char *inptr; - struct field fmt; - int res; - struct field *f; + int fd; + char *inptr; + struct field fmt; + int res; + struct field *f; /* check for inline format */ for (inptr = spec; *inptr; inptr++) if (*inptr == '$') break; - if (*inptr) /* inline */ + if (*inptr) /* inline */ inptr = spec; - else { /* from file */ + else { /* from file */ if ((fd = open(spec, 0)) == -1) { eputs(spec); eputs(": cannot open\n"); @@ -510,7 +548,7 @@ int output close(fd); (inptr=inpbuf+2)[res] = '\0'; } - f = &fmt; /* get fields */ + f = &fmt; /* get fields */ while ((res = readfield(&inptr)) != F_NUL) { f->next = (struct field *)emalloc(sizeof(struct field)); f = f->next; @@ -542,13 +580,13 @@ int output static int -readfield( /* get next field in format */ -char **pp +readfield( /* get next field in format */ +char **pp ) { - int type = F_NUL; - int width = 0; - char *cp; + int type = F_NUL; + int width = 0; + char *cp; cp = inpbuf; while (cp < &inpbuf[INBSIZ-1] && **pp != '\0') { @@ -609,11 +647,11 @@ char **pp struct strvar * -getsvar( /* get string variable */ -char *svname +getsvar( /* get string variable */ +char *svname ) { - struct strvar *sv; + struct strvar *sv; for (sv = svhead; sv != NULL; sv = sv->next) if (!strcmp(sv->name, svname)) @@ -628,12 +666,12 @@ char *svname static void -svpreset( /* preset a string variable */ -char *eqn +svpreset( /* preset a string variable */ +char *eqn ) { - struct strvar *sv; - char *val; + struct strvar *sv; + char *val; for (val = eqn; *val != '='; val++) if (!*val) @@ -652,7 +690,7 @@ char *eqn static void clearrec(void) /* clear input record variables */ { - struct field *f; + struct field *f; for (f = inpfmt; f != NULL; f = f->next) switch (f->type & F_TYP) { @@ -670,35 +708,33 @@ clearrec(void) /* clear input record variables */ static int -getrec(void) /* get next record from file */ +getrec(void) /* get next record from file */ { - int eatline; - struct field *f; + int eatline; + struct field *f; while (ipb.chr != EOF) { if (blnkeq) { /* beware of nbsynch() */ while (isblnk(ipb.chr)) - resetinp(); + skipinp(); if (ipb.chr == EOF) return(0); } - eatline = (!igneol && ipb.chr != '\n'); + eatline = !igneol & (ipb.chr != '\n'); clearrec(); /* start with fresh record */ for (f = inpfmt; f != NULL; f = f->next) - if (getfield(f) == -1) + if (!getfield(f)) break; - if (f == NULL) { - advinp(); /* got one! */ + if (f == NULL) /* got one? */ return(1); - } - resetinp(); /* eat false start */ - if (eatline) { /* eat rest of line */ + skipinp(); /* else eat false start */ + if (eatline) { /* eat rest of line */ while (ipb.chr != '\n') { if (ipb.chr == EOF) return(0); - resetinp(); + skipinp(); } - resetinp(); + skipinp(); } } return(0); @@ -706,15 +742,15 @@ getrec(void) /* get next record from file */ static int -getfield( /* get next field */ -struct field *f +getfield( /* get next field */ +struct field *f ) { - static char buf[RMAXWORD+1]; /* no recursion! */ - int delim, inword; - double d; - char *np; - char *cp; + static char buf[RMAXWORD+1]; /* no recursion! */ + int delim, inword; + double d; + char *np; + char *cp; switch (f->type & F_TYP) { case T_LIT: @@ -722,7 +758,7 @@ struct field *f do { if (blnkeq && isblnk(*cp)) { if (!isblnk(ipb.chr)) - return(-1); + return(0); do cp++; while (isblnk(*cp)); @@ -733,9 +769,9 @@ struct field *f cp++; scaninp(); } else - return(-1); + return(0); } while (*cp); - return(0); + break; case T_STR: if (f->next == NULL || (f->next->type & F_TYP) != T_LIT) delim = EOF; @@ -743,7 +779,7 @@ struct field *f delim = f->next->f.sl[0]; cp = buf; do { - if (ipb.chr == EOF || ipb.chr == '\n') + if ((ipb.chr == EOF) | (ipb.chr == '\n')) inword = 0; else if (blnkeq && delim != EOF) inword = isblnk(delim) ? @@ -760,8 +796,8 @@ struct field *f if (f->f.sv->val == NULL) f->f.sv->val = savqstr(buf); /* first setting */ else if (strcmp(f->f.sv->val, buf)) - return(-1); /* doesn't match! */ - return(0); + return(0); /* doesn't match! */ + break; case T_NUM: if (f->next == NULL || (f->next->type & F_TYP) != T_LIT) delim = EOF; @@ -791,25 +827,25 @@ struct field *f varset(f->f.nv, '=', d); /* first setting */ else if ((d = (varvalue(f->f.nv)-d)/(d==0.?1.:d)) > .001 || d < -.001) - return(-1); /* doesn't match! */ - return(0); + return(0); /* doesn't match! */ + break; } - return -1; /* pro forma return */ + return(1); /* success! */ } static void -putrec(void) /* output a record */ +putrec(void) /* output a record */ { - char fmt[32], typ[16]; - int n; - struct field *f; - int adlast, adnext; - double dv, av; + char fmt[32], typ[16]; + int n; + struct field *f; + int adlast, adnext; + double dv, av; adlast = 0; for (f = outfmt; f != NULL; f = f->next) { - adnext = blnkeq && + adnext = blnkeq && f->next != NULL && !( (f->next->type&F_TYP) == T_LIT && f->next->f.sl[0] == ' ' ); @@ -862,25 +898,25 @@ putrec(void) /* output static void -initinp(FILE *fp) /* prepare lookahead buffer */ +initinp(FILE *fp) /* prepare lookahead buffer */ { ipb.fin = fp; ipb.beg = ipb.end = inpbuf; - ipb.pos = inpbuf-1; /* position before beginning */ + ipb.pos = inpbuf-1; /* position before beginning */ ipb.chr = '\0'; scaninp(); } static void -scaninp(void) /* scan next character */ +scaninp(void) /* scan next character */ { if (ipb.chr == EOF) return; if (++ipb.pos >= &inpbuf[INBSIZ]) ipb.pos = inpbuf; - if (ipb.pos == ipb.end) { /* new character */ + if (ipb.pos == ipb.end) { /* new character */ if ((ipb.chr = getc(ipb.fin)) != EOF) { *ipb.end = ipb.chr; if (++ipb.end >= &inpbuf[INBSIZ]) @@ -894,21 +930,36 @@ scaninp(void) /* scan next chara static void -advinp(void) /* move home to current position */ +advinp(void) /* move home to current position */ { ipb.beg = ipb.pos; } static void -resetinp(void) /* rewind position and advance 1 */ +passinp(void) /* pass beginning to current position */ { - if (ipb.beg == NULL) /* full */ + if (!passive | (ipb.beg == NULL)) { + advinp(); + return; + } + while (ipb.beg != ipb.pos) { /* transfer buffer unaltered */ + putchar(*ipb.beg); + if (++ipb.beg >= &inpbuf[INBSIZ]) + ipb.beg = inpbuf; + } +} + + +static void +skipinp(void) /* rewind position and advance 1 */ +{ + if (ipb.beg == NULL) /* full */ ipb.beg = ipb.end; ipb.pos = ipb.beg; ipb.chr = *ipb.pos; if (passive) /* transmit unmatched character? */ - fputc(ipb.chr, stdout); + putchar(ipb.chr); if (++ipb.beg >= &inpbuf[INBSIZ]) ipb.beg = inpbuf; scaninp(); @@ -916,14 +967,14 @@ resetinp(void) /* rewind position void -eputs(char *msg) +eputs(char *msg) { fputs(msg, stderr); } void -wputs(char *msg) +wputs(char *msg) { if (!nowarn) eputs(msg); @@ -931,7 +982,7 @@ wputs(char *msg) void -quit(int code) +quit(int code) { exit(code); }