src/cal/lam.c

#ifndef lint
static const char       RCSid[] = "$Id: lam.c,v 1.25 2019/08/13 16:31:35 greg Exp $";
#endif
/*
 *  lam.c - simple program to laminate files.
 *
 *      7/14/88         Greg Ward
 */

#include <ctype.h>

#include "rtio.h"
#include "platform.h"
#include "paths.h"

#define MAXLINE         262144          /* maximum input line */

struct instream {               /* structure to hold input stream info */
        FILE            *input;         /* input stream */
        int             bytsiz;         /* bytes/component if binary */
        char            *tabc;          /* data separation string */
}       *rifile = NULL;

int     nfiles = 0;

char    buf[MAXLINE];

int
main(int argc, char *argv[])
{
        long    incnt = 0;
        int     unbuff = 0;
        int     binout = 0;
        char    *curtab = "\t";
        int     curbytes = 0;
        int     puteol;
        int     i;

        rifile = (struct instream *)calloc(argc-1, sizeof(struct instream));
        if (!rifile) {
                fputs(argv[0], stderr);
                fputs(": not enough memory\n", stderr);
                return(1);
        }
        for (i = 1; i < argc; i++) {
                if (argv[i][0] == '-') {
                        switch (argv[i][1]) {
                        case 't':
                                curtab = argv[i]+2;
                                if (!*curtab) curtab = "\n";
                                break;
                        case 'u':
                                unbuff = !unbuff;
                                break;
                        case 'i':
                                switch (argv[i][2]) {
                                case 'n':
                                        incnt = atol(argv[++i]);
                                        break;
                                case 'f':
                                case 'F':
                                        curbytes = sizeof(float);
                                        break;
                                case 'd':
                                case 'D':
                                        curbytes = sizeof(double);
                                        break;
                                case 'i':
                                case 'I':
                                        curbytes = sizeof(int);
                                        break;
                                case 'w':
                                case 'W':
                                        curbytes = 2;
                                        break;
                                case 'b':
                                        curbytes = 1;
                                        break;
                                case 'a':
                                        curbytes = argv[i][3] ? -1 : 0;
                                        break;
                                default:
                                        goto badopt;
                                }
                                if (isdigit(argv[i][3]))
                                        curbytes *= atoi(argv[i]+3);
                                curbytes += (curbytes == -1);
                                if (curbytes > MAXLINE) {
                                        fputs(argv[0], stderr);
                                        fputs(": input size too big\n", stderr);
                                        return(1);
                                }
                                if (curbytes > 0) {
                                        curtab = "";
                                        ++binout;
                                }
                                break;
                        case '\0':
                                rifile[nfiles].tabc = curtab;
                                rifile[nfiles].input = stdin;
                                if (curbytes > 0)
                                        SET_FILE_BINARY(rifile[nfiles].input);
                                rifile[nfiles++].bytsiz = curbytes;
                                break;
                        badopt:;
                        default:
                                fputs(argv[0], stderr);
                                fputs(": bad option\n", stderr);
                                return(1);
                        }
                } else if (argv[i][0] == '!') {
                        rifile[nfiles].tabc = curtab;
                        if ((rifile[nfiles].input = popen(argv[i]+1, "r")) == NULL) {
                                fputs(argv[i], stderr);
                                fputs(": cannot start command\n", stderr);
                                return(1);
                        }
                        if (curbytes > 0)
                                SET_FILE_BINARY(rifile[nfiles].input);
                        rifile[nfiles++].bytsiz = curbytes;
                } else {
                        rifile[nfiles].tabc = curtab;
                        if ((rifile[nfiles].input = fopen(argv[i], "r")) == NULL) {
                                fputs(argv[i], stderr);
                                fputs(": cannot open file\n", stderr);
                                return(1);
                        }
                        if (curbytes > 0)
                                SET_FILE_BINARY(rifile[nfiles].input);
                        rifile[nfiles++].bytsiz = curbytes;
                }
        }
        if (!nfiles) {
                fputs(argv[0], stderr);
                fputs(": no input streams\n", stderr);
                return(1);
        }                                       /* reduce array to size we need */
        rifile = (struct instream *)realloc(rifile, nfiles*sizeof(struct instream));
        if (!rifile) {
                fputs(argv[0], stderr);
                fputs(": realloc() failed!\n", stderr);
                return(1);
        }
        if (binout)                             /* binary output? */
                SET_FILE_BINARY(stdout);
#ifdef getc_unlocked                            /* avoid lock/unlock overhead */
        for (i = nfiles; i--; )
                flockfile(rifile[i].input);
        flockfile(stdout);
#endif
        puteol = 0;                             /* any ASCII output at all? */
        for (i = nfiles; i--; )
                puteol += (rifile[i].bytsiz <= 0);
        do {                                    /* main loop */
                for (i = 0; i < nfiles; i++) {
                        if (rifile[i].bytsiz > 0) {             /* binary input */
                                if (getbinary(buf, rifile[i].bytsiz, 1, rifile[i].input) < 1)
                                        break;
                                if (putbinary(buf, rifile[i].bytsiz, 1, stdout) != 1)
                                        break;
                        } else if (rifile[i].bytsiz < 0) {      /* multi-line input */
                                int     n = -rifile[i].bytsiz;
                                while (n--) {
                                        if (fgets(buf, MAXLINE, rifile[i].input) == NULL)
                                                break;
                                        if ((i > 0) | (n < -rifile[i].bytsiz-1))
                                                fputs(rifile[i].tabc, stdout);
                                        buf[strlen(buf)-1] = '\0';
                                        if (fputs(buf, stdout) == EOF)
                                                break;
                                }
                                if (n >= 0)             /* fell short? */
                                        break;
                        } else {                        /* single-line input */
                                if (fgets(buf, MAXLINE, rifile[i].input) == NULL)
                                        break;
                                if (i)
                                        fputs(rifile[i].tabc, stdout);
                                buf[strlen(buf)-1] = '\0';
                                if (fputs(buf, stdout) == EOF)
                                        break;
                        }
                }
                if (i < nfiles)
                        break;
                if (puteol)
                        putchar('\n');
                if (unbuff)
                        fflush(stdout);
        } while (--incnt);
                                                        /* check ending */
        if (fflush(stdout) == EOF) {
                fputs(argv[0], stderr);
                fputs(": write error on standard output\n", stderr);
                return(1);
        }
        if (incnt > 0) {
                fputs(argv[0], stderr);
                fputs(": warning: premature EOD\n", stderr);
        }
        return(0);
}
Revision:	1.26
Committed:	Fri Jun 7 18:19:22 2024 UTC (12 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Changes since 1.25:	+43 -35 lines
Log Message:	feature(rsplit,rlam): Modified to allow any number of files up to system limit
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	1.26	static const char RCSid[] = "$Id: lam.c,v 1.25 2019/08/13 16:31:35 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* lam.c - simple program to laminate files.
6			*
7			* 7/14/88 Greg Ward
8			*/
9
10	greg	1.7	#include <ctype.h>
11	schorsch	1.2
12	greg	1.25	#include "rtio.h"
13	schorsch	1.2	#include "platform.h"
14	greg	1.25	#include "paths.h"
15	greg	1.1
16	greg	1.26	#define MAXLINE 262144 /* maximum input line */
17	greg	1.1
18	greg	1.26	struct instream { /* structure to hold input stream info */
19			FILE input; / input stream */
20			int bytsiz; /* bytes/component if binary */
21			char tabc; / data separation string */
22			} *rifile = NULL;
23	greg	1.1
24	greg	1.20	int nfiles = 0;
25	greg	1.1
26			char buf[MAXLINE];
27
28	schorsch	1.4	int
29	greg	1.20	main(int argc, char *argv[])
30	greg	1.1	{
31	greg	1.18	long incnt = 0;
32	greg	1.11	int unbuff = 0;
33	greg	1.14	int binout = 0;
34	greg	1.18	char *curtab = "\t";
35			int curbytes = 0;
36			int puteol;
37	greg	1.11	int i;
38	greg	1.1
39	greg	1.26	rifile = (struct instream *)calloc(argc-1, sizeof(struct instream));
40			if (!rifile) {
41			fputs(argv[0], stderr);
42			fputs(": not enough memory\n", stderr);
43			return(1);
44			}
45	greg	1.1	for (i = 1; i < argc; i++) {
46			if (argv[i][0] == '-') {
47			switch (argv[i][1]) {
48			case 't':
49	greg	1.6	curtab = argv[i]+2;
50	greg	1.23	if (!*curtab) curtab = "\n";
51	greg	1.1	break;
52	greg	1.11	case 'u':
53			unbuff = !unbuff;
54			break;
55	greg	1.7	case 'i':
56			switch (argv[i][2]) {
57	greg	1.15	case 'n':
58			incnt = atol(argv[++i]);
59			break;
60	greg	1.7	case 'f':
61	greg	1.24	case 'F':
62	greg	1.7	curbytes = sizeof(float);
63			break;
64			case 'd':
65	greg	1.24	case 'D':
66	greg	1.7	curbytes = sizeof(double);
67			break;
68	greg	1.10	case 'i':
69	greg	1.24	case 'I':
70	greg	1.10	curbytes = sizeof(int);
71			break;
72	greg	1.7	case 'w':
73	greg	1.24	case 'W':
74	greg	1.10	curbytes = 2;
75			break;
76			case 'b':
77			curbytes = 1;
78	greg	1.7	break;
79			case 'a':
80	greg	1.12	curbytes = argv[i][3] ? -1 : 0;
81	greg	1.7	break;
82			default:
83			goto badopt;
84			}
85			if (isdigit(argv[i][3]))
86			curbytes *= atoi(argv[i]+3);
87	greg	1.19	curbytes += (curbytes == -1);
88			if (curbytes > MAXLINE) {
89	greg	1.7	fputs(argv[0], stderr);
90	greg	1.12	fputs(": input size too big\n", stderr);
91	greg	1.19	return(1);
92	greg	1.7	}
93	greg	1.19	if (curbytes > 0) {
94	greg	1.8	curtab = "";
95	greg	1.19	++binout;
96	greg	1.14	}
97	greg	1.7	break;
98	greg	1.1	case '\0':
99	greg	1.26	rifile[nfiles].tabc = curtab;
100			rifile[nfiles].input = stdin;
101	greg	1.12	if (curbytes > 0)
102	greg	1.26	SET_FILE_BINARY(rifile[nfiles].input);
103			rifile[nfiles++].bytsiz = curbytes;
104	greg	1.1	break;
105	greg	1.7	badopt:;
106	greg	1.1	default:
107			fputs(argv[0], stderr);
108			fputs(": bad option\n", stderr);
109	greg	1.19	return(1);
110	greg	1.1	}
111			} else if (argv[i][0] == '!') {
112	greg	1.26	rifile[nfiles].tabc = curtab;
113			if ((rifile[nfiles].input = popen(argv[i]+1, "r")) == NULL) {
114	greg	1.1	fputs(argv[i], stderr);
115			fputs(": cannot start command\n", stderr);
116	greg	1.19	return(1);
117	greg	1.1	}
118	greg	1.12	if (curbytes > 0)
119	greg	1.26	SET_FILE_BINARY(rifile[nfiles].input);
120			rifile[nfiles++].bytsiz = curbytes;
121	greg	1.1	} else {
122	greg	1.26	rifile[nfiles].tabc = curtab;
123			if ((rifile[nfiles].input = fopen(argv[i], "r")) == NULL) {
124	greg	1.1	fputs(argv[i], stderr);
125			fputs(": cannot open file\n", stderr);
126	greg	1.19	return(1);
127	greg	1.1	}
128	greg	1.12	if (curbytes > 0)
129	greg	1.26	SET_FILE_BINARY(rifile[nfiles].input);
130			rifile[nfiles++].bytsiz = curbytes;
131	greg	1.1	}
132			}
133	greg	1.19	if (!nfiles) {
134			fputs(argv[0], stderr);
135			fputs(": no input streams\n", stderr);
136			return(1);
137	greg	1.26	} /* reduce array to size we need */
138			rifile = (struct instream )realloc(rifile, nfilessizeof(struct instream));
139			if (!rifile) {
140			fputs(argv[0], stderr);
141			fputs(": realloc() failed!\n", stderr);
142			return(1);
143	greg	1.19	}
144	greg	1.14	if (binout) /* binary output? */
145			SET_FILE_BINARY(stdout);
146			#ifdef getc_unlocked /* avoid lock/unlock overhead */
147			for (i = nfiles; i--; )
148	greg	1.26	flockfile(rifile[i].input);
149	greg	1.14	flockfile(stdout);
150			#endif
151	greg	1.15	puteol = 0; /* any ASCII output at all? */
152	greg	1.7	for (i = nfiles; i--; )
153	greg	1.26	puteol += (rifile[i].bytsiz <= 0);
154	greg	1.16	do { /* main loop */
155	greg	1.1	for (i = 0; i < nfiles; i++) {
156	greg	1.26	if (rifile[i].bytsiz > 0) { /* binary input */
157			if (getbinary(buf, rifile[i].bytsiz, 1, rifile[i].input) < 1)
158	greg	1.11	break;
159	greg	1.26	if (putbinary(buf, rifile[i].bytsiz, 1, stdout) != 1)
160	greg	1.22	break;
161	greg	1.26	} else if (rifile[i].bytsiz < 0) { /* multi-line input */
162			int n = -rifile[i].bytsiz;
163	greg	1.19	while (n--) {
164	greg	1.26	if (fgets(buf, MAXLINE, rifile[i].input) == NULL)
165	greg	1.19	break;
166	greg	1.26	if ((i > 0) \| (n < -rifile[i].bytsiz-1))
167			fputs(rifile[i].tabc, stdout);
168	greg	1.19	buf[strlen(buf)-1] = '\0';
169	greg	1.22	if (fputs(buf, stdout) == EOF)
170			break;
171	greg	1.19	}
172			if (n >= 0) /* fell short? */
173			break;
174			} else { /* single-line input */
175	greg	1.26	if (fgets(buf, MAXLINE, rifile[i].input) == NULL)
176	greg	1.11	break;
177	greg	1.1	if (i)
178	greg	1.26	fputs(rifile[i].tabc, stdout);
179	greg	1.1	buf[strlen(buf)-1] = '\0';
180	greg	1.22	if (fputs(buf, stdout) == EOF)
181			break;
182	greg	1.1	}
183			}
184	greg	1.11	if (i < nfiles)
185			break;
186			if (puteol)
187	greg	1.1	putchar('\n');
188	greg	1.11	if (unbuff)
189			fflush(stdout);
190	greg	1.16	} while (--incnt);
191	greg	1.21	/* check ending */
192	greg	1.22	if (fflush(stdout) == EOF) {
193			fputs(argv[0], stderr);
194			fputs(": write error on standard output\n", stderr);
195			return(1);
196			}
197	greg	1.21	if (incnt > 0) {
198			fputs(argv[0], stderr);
199			fputs(": warning: premature EOD\n", stderr);
200			}
201	greg	1.11	return(0);
202	greg	1.1	}