Home | History | Annotate | Line # | Download | only in rs
rs.c revision 1.8
      1 /*	$NetBSD: rs.c,v 1.8 2002/06/11 06:06:20 itojun Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. All advertising materials mentioning features or use of this software
     16  *    must display the following acknowledgement:
     17  *	This product includes software developed by the University of
     18  *	California, Berkeley and its contributors.
     19  * 4. Neither the name of the University nor the names of its contributors
     20  *    may be used to endorse or promote products derived from this software
     21  *    without specific prior written permission.
     22  *
     23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     33  * SUCH DAMAGE.
     34  */
     35 
     36 #include <sys/cdefs.h>
     37 #ifndef lint
     38 __COPYRIGHT("@(#) Copyright (c) 1993\n\
     39 	The Regents of the University of California.  All rights reserved.\n");
     40 #endif /* not lint */
     41 
     42 #ifndef lint
     43 #if 0
     44 static char sccsid[] = "@(#)rs.c	8.1 (Berkeley) 6/6/93";
     45 #else
     46 __RCSID("$NetBSD: rs.c,v 1.8 2002/06/11 06:06:20 itojun Exp $");
     47 #endif
     48 #endif /* not lint */
     49 
     50 /*
     51  *	rs - reshape a data array
     52  *	Author:  John Kunze, Office of Comp. Affairs, UCB
     53  *		BEWARE: lots of unfinished edges
     54  */
     55 
     56 #include <ctype.h>
     57 #include <err.h>
     58 #include <stdio.h>
     59 #include <stdlib.h>
     60 #include <string.h>
     61 #include <stdarg.h>
     62 
     63 long	flags;
     64 #define	TRANSPOSE	000001
     65 #define	MTRANSPOSE	000002
     66 #define	ONEPERLINE	000004
     67 #define	ONEISEPONLY	000010
     68 #define	ONEOSEPONLY	000020
     69 #define	NOTRIMENDCOL	000040
     70 #define	SQUEEZE		000100
     71 #define	SHAPEONLY	000200
     72 #define	DETAILSHAPE	000400
     73 #define	RIGHTADJUST	001000
     74 #define	NULLPAD		002000
     75 #define	RECYCLE		004000
     76 #define	SKIPPRINT	010000
     77 #define	ICOLBOUNDS	020000
     78 #define	OCOLBOUNDS	040000
     79 #define ONEPERCHAR	0100000
     80 #define NOARGS		0200000
     81 
     82 short	*colwidths;
     83 short	*cord;
     84 short	*icbd;
     85 short	*ocbd;
     86 int	nelem;
     87 char	**elem;
     88 char	**endelem;
     89 char	*curline;
     90 int	allocsize = BUFSIZ;
     91 int	curlen;
     92 int	irows, icols;
     93 int	orows, ocols;
     94 int	maxlen;
     95 int	skip;
     96 int	propgutter;
     97 char	isep = ' ', osep = ' ';
     98 int	owidth = 80, gutter = 2;
     99 
    100 void	  usage __P((char *, ...))
    101      __attribute__((__format__(__printf__, 1, 2)));
    102 void	  getargs __P((int, char *[]));
    103 void	  getfile __P((void));
    104 int	  getline __P((void));
    105 char	 *getlist __P((short **, char *));
    106 char	 *getnum __P((int *, char *, int));
    107 char	**getptrs __P((char **));
    108 int	  main __P((int, char **));
    109 void	  prepfile __P((void));
    110 void	  prints __P((char *, int));
    111 void	  putfile __P((void));
    112 
    113 #define INCR(ep) do {			\
    114 	if (++ep >= endelem)		\
    115 		ep = getptrs(ep);	\
    116 } while(0)
    117 
    118 int
    119 main(argc, argv)
    120 	int argc;
    121 	char *argv[];
    122 {
    123 	getargs(argc, argv);
    124 	getfile();
    125 	if (flags & SHAPEONLY) {
    126 		printf("%d %d\n", irows, icols);
    127 		exit(0);
    128 	}
    129 	prepfile();
    130 	putfile();
    131 	exit(0);
    132 }
    133 
    134 void
    135 getfile()
    136 {
    137 	char *p;
    138 	char *endp;
    139 	char **ep = 0;
    140 	int multisep = (flags & ONEISEPONLY ? 0 : 1);
    141 	int nullpad = flags & NULLPAD;
    142 	char **padto;
    143 
    144 	while (skip--) {
    145 		getline();
    146 		if (flags & SKIPPRINT)
    147 			puts(curline);
    148 	}
    149 	getline();
    150 	if (flags & NOARGS && curlen < owidth)
    151 		flags |= ONEPERLINE;
    152 	if (flags & ONEPERLINE)
    153 		icols = 1;
    154 	else				/* count cols on first line */
    155 		for (p = curline, endp = curline + curlen; p < endp; p++) {
    156 			if (*p == isep && multisep)
    157 				continue;
    158 			icols++;
    159 			while (*p && *p != isep)
    160 				p++;
    161 		}
    162 	ep = getptrs(elem);
    163 	p = curline;
    164 	do {
    165 		if (flags & ONEPERLINE) {
    166 			*ep = curline;
    167 			INCR(ep);		/* prepare for next entry */
    168 			if (maxlen < curlen)
    169 				maxlen = curlen;
    170 			irows++;
    171 			continue;
    172 		}
    173 		for (p = curline, endp = curline + curlen; p < endp; p++) {
    174 			if (*p == isep && multisep)
    175 				continue;	/* eat up column separators */
    176 			if (*p == isep)		/* must be an empty column */
    177 				*ep = "";
    178 			else			/* store column entry */
    179 				*ep = p;
    180 			while (p < endp && *p != isep)
    181 				p++;		/* find end of entry */
    182 			*p = '\0';		/* mark end of entry */
    183 			if (maxlen < p - *ep)	/* update maxlen */
    184 				maxlen = p - *ep;
    185 			INCR(ep);		/* prepare for next entry */
    186 		}
    187 		irows++;			/* update row count */
    188 		if (nullpad) {			/* pad missing entries */
    189 			padto = elem + irows * icols;
    190 			while (ep < padto) {
    191 				*ep = "";
    192 				INCR(ep);
    193 			}
    194 		}
    195 	} while (getline() != EOF);
    196 	*ep = 0;				/* mark end of pointers */
    197 	nelem = ep - elem;
    198 }
    199 
    200 void
    201 putfile()
    202 {
    203 	char **ep;
    204 	int i, j, n;
    205 
    206 	ep = elem;
    207 	if (flags & TRANSPOSE) {
    208 		for (i = 0; i < orows; i++) {
    209 			for (j = i; j < nelem; j += orows)
    210 				prints(ep[j], (j - i) / orows);
    211 			putchar('\n');
    212 		}
    213 	} else {
    214 		for (n = 0, i = 0; i < orows && n < nelem; i++) {
    215 			for (j = 0; j < ocols; j++) {
    216 				if (n++ >= nelem)
    217 					break;
    218 				prints(*ep++, j);
    219 			}
    220 			putchar('\n');
    221 		}
    222 	}
    223 }
    224 
    225 void
    226 prints(s, col)
    227 	char *s;
    228 	int col;
    229 {
    230 	int n;
    231 	char *p = s;
    232 
    233 	while (*p)
    234 		p++;
    235 	n = (flags & ONEOSEPONLY ? 1 : colwidths[col] - (p - s));
    236 	if (flags & RIGHTADJUST)
    237 		while (n-- > 0)
    238 			putchar(osep);
    239 	for (p = s; *p; p++)
    240 		putchar(*p);
    241 	while (n-- > 0)
    242 		putchar(osep);
    243 }
    244 
    245 void
    246 usage(char *msg, ...)
    247 {
    248 	va_list ap;
    249 
    250 	va_start(ap, msg);
    251 	vwarnx(msg, ap);
    252 	va_end(ap);
    253 	fprintf(stderr,
    254 "Usage:  rs [ -[csCS][x][kKgGw][N]tTeEnyjhHm ] [ rows [ cols ] ]\n");
    255 	exit(1);
    256 }
    257 
    258 void
    259 prepfile()
    260 {
    261 	char **ep;
    262 	int  i;
    263 	int  j;
    264 	char **lp;
    265 	int colw;
    266 	int max = 0;
    267 	int n;
    268 
    269 	ep = NULL;
    270 	if (!nelem)
    271 		exit(0);
    272 	gutter += maxlen * propgutter / 100.0;
    273 	colw = maxlen + gutter;
    274 	if (flags & MTRANSPOSE) {
    275 		orows = icols;
    276 		ocols = irows;
    277 	}
    278 	else if (orows == 0 && ocols == 0) {	/* decide rows and cols */
    279 		ocols = owidth / colw;
    280 		if (ocols == 0) {
    281 			warnx("Display width %d is less than column width %d", owidth, colw);
    282 			ocols = 1;
    283 		}
    284 		if (ocols > nelem)
    285 			ocols = nelem;
    286 		orows = nelem / ocols + (nelem % ocols ? 1 : 0);
    287 	}
    288 	else if (orows == 0)			/* decide on rows */
    289 		orows = nelem / ocols + (nelem % ocols ? 1 : 0);
    290 	else if (ocols == 0)			/* decide on cols */
    291 		ocols = nelem / orows + (nelem % orows ? 1 : 0);
    292 	lp = elem + orows * ocols;
    293 	while (lp > endelem) {
    294 		getptrs(elem + nelem);
    295 		lp = elem + orows * ocols;
    296 	}
    297 	if (flags & RECYCLE) {
    298 		for (ep = elem + nelem; ep < lp; ep++)
    299 			*ep = *(ep - nelem);
    300 		nelem = lp - elem;
    301 	}
    302 	if (!(colwidths = (short *) malloc(ocols * sizeof(short))))
    303 		errx(1, "malloc:  No gutter space");
    304 	if (flags & SQUEEZE) {
    305 		if (flags & TRANSPOSE)
    306 			for (ep = elem, i = 0; i < ocols; i++) {
    307 				for (j = 0; j < orows; j++)
    308 					if ((n = strlen(*ep++)) > max)
    309 						max = n;
    310 				colwidths[i] = max + gutter;
    311 			}
    312 		else
    313 			for (ep = elem, i = 0; i < ocols; i++) {
    314 				for (j = i; j < nelem; j += ocols)
    315 					if ((n = strlen(ep[j])) > max)
    316 						max = n;
    317 				colwidths[i] = max + gutter;
    318 			}
    319 	}
    320 	/*	for (i = 0; i < orows; i++) {
    321 			for (j = i; j < nelem; j += orows)
    322 				prints(ep[j], (j - i) / orows);
    323 			putchar('\n');
    324 		}
    325 	else
    326 		for (i = 0; i < orows; i++) {
    327 			for (j = 0; j < ocols; j++)
    328 				prints(*ep++, j);
    329 			putchar('\n');
    330 		}*/
    331 	else
    332 		for (i = 0; i < ocols; i++)
    333 			colwidths[i] = colw;
    334 	if (!(flags & NOTRIMENDCOL)) {
    335 		if (flags & RIGHTADJUST)
    336 			colwidths[0] -= gutter;
    337 		else
    338 			colwidths[ocols - 1] = 0;
    339 	}
    340 	n = orows * ocols;
    341 	if (n > nelem && (flags & RECYCLE))
    342 		nelem = n;
    343 	/*for (i = 0; i < ocols; i++)
    344 		fprintf(stderr, "%d ",colwidths[i]);
    345 	fprintf(stderr, "is colwidths, nelem %d\n", nelem);*/
    346 }
    347 
    348 #define	BSIZE	2048
    349 char	ibuf[BSIZE];		/* two screenfuls should do */
    350 
    351 int
    352 getline()	/* get line; maintain curline, curlen; manage storage */
    353 {
    354 	static	int putlength;
    355 	static	char *endblock = ibuf + BSIZE;
    356 	char *p;
    357 	int c, i;
    358 
    359 	if (!irows) {
    360 		curline = ibuf;
    361 		putlength = flags & DETAILSHAPE;
    362 	}
    363 	else if (skip <= 0) {			/* don't waste storage */
    364 		curline += curlen + 1;
    365 		if (putlength)		/* print length, recycle storage */
    366 			printf(" %d line %d\n", curlen, irows);
    367 	}
    368 	if (!putlength && endblock - curline < BUFSIZ) {   /* need storage */
    369 		/*ww = endblock-curline; tt += ww;*/
    370 		/*printf("#wasted %d total %d\n",ww,tt);*/
    371 		if (!(curline = (char *) malloc(BSIZE)))
    372 			errx(1, "File too large");
    373 		endblock = curline + BSIZE;
    374 		/*printf("#endb %d curline %d\n",endblock,curline);*/
    375 	}
    376 	for (p = curline, i = 1; i < BUFSIZ; *p++ = c, i++)
    377 		if ((c = getchar()) == EOF || c == '\n')
    378 			break;
    379 	*p = '\0';
    380 	curlen = i - 1;
    381 	return(c);
    382 }
    383 
    384 char **
    385 getptrs(sp)
    386 	char **sp;
    387 {
    388 	char **p;
    389 
    390 	allocsize += allocsize;
    391 	p = (char **)realloc(elem, allocsize * sizeof(char *));
    392 	if (p == (char **)0)
    393 		err(1, "no memory");
    394 
    395 	sp += (p - elem);
    396 	endelem = (elem = p) + allocsize;
    397 	return(sp);
    398 }
    399 
    400 void
    401 getargs(ac, av)
    402 	int ac;
    403 	char *av[];
    404 {
    405 	char *p;
    406 
    407 	if (ac == 1) {
    408 		flags |= NOARGS | TRANSPOSE;
    409 	}
    410 	while (--ac && **++av == '-')
    411 		for (p = *av+1; *p; p++)
    412 			switch (*p) {
    413 			case 'T':
    414 				flags |= MTRANSPOSE;
    415 			case 't':
    416 				flags |= TRANSPOSE;
    417 				break;
    418 			case 'c':		/* input col. separator */
    419 				flags |= ONEISEPONLY;
    420 			case 's':		/* one or more allowed */
    421 				if (p[1])
    422 					isep = *++p;
    423 				else
    424 					isep = '\t';	/* default is ^I */
    425 				break;
    426 			case 'C':
    427 				flags |= ONEOSEPONLY;
    428 			case 'S':
    429 				if (p[1])
    430 					osep = *++p;
    431 				else
    432 					osep = '\t';	/* default is ^I */
    433 				break;
    434 			case 'w':		/* window width, default 80 */
    435 				p = getnum(&owidth, p, 0);
    436 				if (owidth <= 0)
    437 				usage("Width must be a positive integer");
    438 				break;
    439 			case 'K':			/* skip N lines */
    440 				flags |= SKIPPRINT;
    441 			case 'k':			/* skip, do not print */
    442 				p = getnum(&skip, p, 0);
    443 				if (!skip)
    444 					skip = 1;
    445 				break;
    446 			case 'm':
    447 				flags |= NOTRIMENDCOL;
    448 				break;
    449 			case 'g':		/* gutter space */
    450 				p = getnum(&gutter, p, 0);
    451 				break;
    452 			case 'G':
    453 				p = getnum(&propgutter, p, 0);
    454 				break;
    455 			case 'e':		/* each line is an entry */
    456 				flags |= ONEPERLINE;
    457 				break;
    458 			case 'E':
    459 				flags |= ONEPERCHAR;
    460 				break;
    461 			case 'j':			/* right adjust */
    462 				flags |= RIGHTADJUST;
    463 				break;
    464 			case 'n':	/* null padding for missing values */
    465 				flags |= NULLPAD;
    466 				break;
    467 			case 'y':
    468 				flags |= RECYCLE;
    469 				break;
    470 			case 'H':			/* print shape only */
    471 				flags |= DETAILSHAPE;
    472 			case 'h':
    473 				flags |= SHAPEONLY;
    474 				break;
    475 			case 'z':			/* squeeze col width */
    476 				flags |= SQUEEZE;
    477 				break;
    478 			/*case 'p':
    479 				ipagespace = atoi(++p);	(default is 1)
    480 				break;*/
    481 			case 'o':			/* col order */
    482 				p = getlist(&cord, p);
    483 				break;
    484 			case 'b':
    485 				flags |= ICOLBOUNDS;
    486 				p = getlist(&icbd, p);
    487 				break;
    488 			case 'B':
    489 				flags |= OCOLBOUNDS;
    490 				p = getlist(&ocbd, p);
    491 				break;
    492 			default:
    493 				usage("Bad flag:  %.1s", p);
    494 			}
    495 	/*if (!osep)
    496 		osep = isep;*/
    497 	switch (ac) {
    498 	/*case 3:
    499 		opages = atoi(av[2]);*/
    500 	case 2:
    501 		ocols = atoi(av[1]);
    502 	case 1:
    503 		orows = atoi(av[0]);
    504 	case 0:
    505 		break;
    506 	default:
    507 		usage("Too many arguments.");
    508 	}
    509 }
    510 
    511 char *
    512 getlist(list, p)
    513 	short **list;
    514 	char *p;
    515 {
    516 	int count = 1;
    517 	char *t;
    518 
    519 	for (t = p + 1; *t; t++) {
    520 		if (!isdigit(*t))
    521 			usage("Option %.1s requires a list of unsigned numbers separated by commas", t);
    522 		count++;
    523 		while (*t && isdigit(*t))
    524 			t++;
    525 		if (*t != ',')
    526 			break;
    527 	}
    528 	if (!(*list = (short *) malloc(count * sizeof(short))))
    529 		errx(1, "No list space");
    530 	count = 0;
    531 	for (t = p + 1; *t; t++) {
    532 		(*list)[count++] = atoi(t);
    533 		printf("++ %d ", (*list)[count-1]);
    534 		fflush(stdout);
    535 		while (*t && isdigit(*t))
    536 			t++;
    537 		if (*t != ',')
    538 			break;
    539 	}
    540 	(*list)[count] = 0;
    541 	return(t - 1);
    542 }
    543 
    544 char *
    545 getnum(num, p, strict)	/* num = number p points to; if (strict) complain */
    546 	int *num, strict;	/* returns pointer to end of num */
    547 	char *p;
    548 {
    549 	char *t = p;
    550 
    551 	if (!isdigit(*++t)) {
    552 		if (strict || *t == '-' || *t == '+')
    553 			usage("Option %.1s requires an unsigned integer", p);
    554 		*num = 0;
    555 		return(p);
    556 	}
    557 	*num = atoi(t);
    558 	while (*++t)
    559 		if (!isdigit(*t))
    560 			break;
    561 	return(--t);
    562 }
    563