--- ray/src/cal/lam.c	2014/02/11 17:05:00	1.13
+++ ray/src/cal/lam.c	2019/07/04 17:36:04	1.19
@@ -1,5 +1,5 @@
 #ifndef lint
-static const char	RCSid[] = "$Id: lam.c,v 1.13 2014/02/11 17:05:00 greg Exp $";
+static const char	RCSid[] = "$Id: lam.c,v 1.19 2019/07/04 17:36:04 greg Exp $";
 #endif
 /*
  *  lam.c - simple program to laminate files.
@@ -14,6 +14,7 @@ static const char	RCSid[] = "$Id: lam.c,v 1.13 2014/02
 
 #include "platform.h"
 #include "rtprocess.h"
+#include "rtio.h"
 
 #define MAXFILE		512		/* maximum number of files */
 
@@ -31,14 +32,14 @@ main(argc, argv)
 int	argc;
 char	*argv[];
 {
+	long	incnt = 0;
 	int	unbuff = 0;
-	int	i;
-	char	*curtab;
-	int	curbytes;
+	int	binout = 0;
+	char	*curtab = "\t";
+	int	curbytes = 0;
 	int	puteol;
+	int	i;
 
-	curtab = "\t";
-	curbytes = 0;
 	nfiles = 0;
 	for (i = 1; i < argc; i++) {
 		if (argv[i][0] == '-') {
@@ -51,6 +52,9 @@ char	*argv[];
 				break;
 			case 'i':
 				switch (argv[i][2]) {
+				case 'n':
+					incnt = atol(argv[++i]);
+					break;
 				case 'f':
 					curbytes = sizeof(float);
 					break;
@@ -74,75 +78,93 @@ char	*argv[];
 				}
 				if (isdigit(argv[i][3]))
 					curbytes *= atoi(argv[i]+3);
-				if (abs(curbytes) > MAXLINE) {
+				curbytes += (curbytes == -1);
+				if (curbytes > MAXLINE) {
 					fputs(argv[0], stderr);
 					fputs(": input size too big\n", stderr);
-					exit(1);
+					return(1);
 				}
-				if (curbytes)
+				if (curbytes > 0) {
 					curtab = "";
+					++binout;
+				}
 				break;
 			case '\0':
 				tabc[nfiles] = curtab;
 				input[nfiles] = stdin;
 				if (curbytes > 0)
 					SET_FILE_BINARY(input[nfiles]);
-				else
-					curbytes = -curbytes;
 				bytsiz[nfiles++] = curbytes;
 				break;
 			badopt:;
 			default:
 				fputs(argv[0], stderr);
 				fputs(": bad option\n", stderr);
-				exit(1);
+				return(1);
 			}
 		} else if (argv[i][0] == '!') {
 			tabc[nfiles] = curtab;
 			if ((input[nfiles] = popen(argv[i]+1, "r")) == NULL) {
 				fputs(argv[i], stderr);
 				fputs(": cannot start command\n", stderr);
-				exit(1);
+				return(1);
 			}
 			if (curbytes > 0)
 				SET_FILE_BINARY(input[nfiles]);
-			else
-				curbytes = -curbytes;
 			bytsiz[nfiles++] = curbytes;
 		} else {
 			tabc[nfiles] = curtab;
 			if ((input[nfiles] = fopen(argv[i], "r")) == NULL) {
 				fputs(argv[i], stderr);
 				fputs(": cannot open file\n", stderr);
-				exit(1);
+				return(1);
 			}
 			if (curbytes > 0)
 				SET_FILE_BINARY(input[nfiles]);
-			else
-				curbytes = -curbytes;
 			bytsiz[nfiles++] = curbytes;
 		}
 		if (nfiles >= MAXFILE) {
 			fputs(argv[0], stderr);
 			fputs(": too many input streams\n", stderr);
-			exit(1);
+			return(1);
 		}
 	}
-	puteol = 0;				/* check for ASCII output */
+	if (!nfiles) {
+		fputs(argv[0], stderr);
+		fputs(": no input streams\n", stderr);
+		return(1);
+	}
+	if (binout)				/* binary output? */
+		SET_FILE_BINARY(stdout);
+#ifdef getc_unlocked				/* avoid lock/unlock overhead */
 	for (i = nfiles; i--; )
-		if (!bytsiz[i] || isprint(tabc[i][0]) || tabc[i][0] == '\t') {
-			puteol++;
-			break;
-		}
-	for ( ; ; ) {				/* main loop */
+		flockfile(input[i]);
+	flockfile(stdout);
+#endif
+	puteol = 0;				/* any ASCII output at all? */
+	for (i = nfiles; i--; )
+		puteol += (bytsiz[i] <= 0);
+	do {					/* main loop */
 		for (i = 0; i < nfiles; i++) {
-			if (bytsiz[i]) {		/* binary file */
-				if (fread(buf, bytsiz[i], 1, input[i]) < 1)
+			if (bytsiz[i] > 0) {		/* binary input */
+				if (getbinary(buf, bytsiz[i], 1, input[i]) < 1)
 					break;
 				if (i)
 					fputs(tabc[i], stdout);
-				fwrite(buf, bytsiz[i], 1, stdout);
-			} else {
+				putbinary(buf, bytsiz[i], 1, stdout);
+			} else if (bytsiz[i] < 0) {	/* multi-line input */
+				int	n = -bytsiz[i];
+				while (n--) {
+					if (fgets(buf, MAXLINE, input[i]) == NULL)
+						break;
+					if ((i > 0) | (n < -bytsiz[i]-1))
+						fputs(tabc[i], stdout);
+					buf[strlen(buf)-1] = '\0';
+					fputs(buf, stdout);
+				}
+				if (n >= 0)		/* fell short? */
+					break;
+			} else {			/* single-line input */
 				if (fgets(buf, MAXLINE, input[i]) == NULL)
 					break;
 				if (i)
@@ -157,6 +179,6 @@ char	*argv[];
 			putchar('\n');
 		if (unbuff)
 			fflush(stdout);
-	}
+	} while (--incnt);
 	return(0);
 }