column.c revision 162453 
1/*
2 * Copyright (c) 1989, 1993, 1994
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *	This product includes software developed by the University of
16 *	California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#ifndef lint
35static const char copyright[] =
36"@(#) Copyright (c) 1989, 1993, 1994\n\
37	The Regents of the University of California.  All rights reserved.\n";
38#endif
39
40#if 0
41#ifndef lint
42static char sccsid[] = "@(#)column.c	8.4 (Berkeley) 5/4/95";
43#endif
44#endif
45
46#include <sys/cdefs.h>
47__FBSDID("$FreeBSD: head/usr.bin/column/column.c 162453 2006-09-19 22:11:43Z iedowse $");
48
49#include <sys/types.h>
50#include <sys/ioctl.h>
51#include <sys/param.h>
52
53#include <err.h>
54#include <limits.h>
55#include <locale.h>
56#include <stdio.h>
57#include <stdlib.h>
58#include <string.h>
59#include <unistd.h>
60#include <wchar.h>
61#include <wctype.h>
62
63#define	TAB	8
64
65void  c_columnate(void);
66void  input(FILE *);
67void  maketbl(void);
68void  print(void);
69void  r_columnate(void);
70void  usage(void);
71int   width(const wchar_t *);
72
73int termwidth = 80;		/* default terminal width */
74
75int entries;			/* number of records */
76int eval;			/* exit value */
77int maxlength;			/* longest record */
78wchar_t **list;			/* array of pointers to records */
79const wchar_t *separator = L"\t ";	/* field separator for table option */
80
81int
82main(int argc, char **argv)
83{
84	struct winsize win;
85	FILE *fp;
86	int ch, tflag, xflag;
87	char *p;
88	const char *src;
89	wchar_t *newsep;
90	size_t seplen;
91
92	setlocale(LC_ALL, "");
93
94	if (ioctl(1, TIOCGWINSZ, &win) == -1 || !win.ws_col) {
95		if ((p = getenv("COLUMNS")))
96			termwidth = atoi(p);
97	} else
98		termwidth = win.ws_col;
99
100	tflag = xflag = 0;
101	while ((ch = getopt(argc, argv, "c:s:tx")) != -1)
102		switch(ch) {
103		case 'c':
104			termwidth = atoi(optarg);
105			break;
106		case 's':
107			src = optarg;
108			seplen = mbsrtowcs(NULL, &src, 0, NULL);
109			if (seplen == (size_t)-1)
110				err(1, "bad separator");
111			newsep = malloc((seplen + 1) * sizeof(wchar_t));
112			if (newsep == NULL)
113				err(1, NULL);
114			mbsrtowcs(newsep, &src, seplen + 1, NULL);
115			separator = newsep;
116			break;
117		case 't':
118			tflag = 1;
119			break;
120		case 'x':
121			xflag = 1;
122			break;
123		case '?':
124		default:
125			usage();
126		}
127	argc -= optind;
128	argv += optind;
129
130	if (!*argv)
131		input(stdin);
132	else for (; *argv; ++argv)
133		if ((fp = fopen(*argv, "r"))) {
134			input(fp);
135			(void)fclose(fp);
136		} else {
137			warn("%s", *argv);
138			eval = 1;
139		}
140
141	if (!entries)
142		exit(eval);
143
144	maxlength = roundup(maxlength + TAB, TAB);
145	if (tflag)
146		maketbl();
147	else if (maxlength >= termwidth)
148		print();
149	else if (xflag)
150		c_columnate();
151	else
152		r_columnate();
153	exit(eval);
154}
155
156void
157c_columnate(void)
158{
159	int chcnt, col, cnt, endcol, numcols;
160	wchar_t **lp;
161
162	numcols = termwidth / maxlength;
163	endcol = maxlength;
164	for (chcnt = col = 0, lp = list;; ++lp) {
165		wprintf(L"%ls", *lp);
166		chcnt += width(*lp);
167		if (!--entries)
168			break;
169		if (++col == numcols) {
170			chcnt = col = 0;
171			endcol = maxlength;
172			putwchar('\n');
173		} else {
174			while ((cnt = roundup(chcnt + TAB, TAB)) <= endcol) {
175				(void)putwchar('\t');
176				chcnt = cnt;
177			}
178			endcol += maxlength;
179		}
180	}
181	if (chcnt)
182		putwchar('\n');
183}
184
185void
186r_columnate(void)
187{
188	int base, chcnt, cnt, col, endcol, numcols, numrows, row;
189
190	numcols = termwidth / maxlength;
191	numrows = entries / numcols;
192	if (entries % numcols)
193		++numrows;
194
195	for (row = 0; row < numrows; ++row) {
196		endcol = maxlength;
197		for (base = row, chcnt = col = 0; col < numcols; ++col) {
198			wprintf(L"%ls", list[base]);
199			chcnt += width(list[base]);
200			if ((base += numrows) >= entries)
201				break;
202			while ((cnt = roundup(chcnt + TAB, TAB)) <= endcol) {
203				(void)putwchar('\t');
204				chcnt = cnt;
205			}
206			endcol += maxlength;
207		}
208		putwchar('\n');
209	}
210}
211
212void
213print(void)
214{
215	int cnt;
216	wchar_t **lp;
217
218	for (cnt = entries, lp = list; cnt--; ++lp)
219		(void)wprintf(L"%ls\n", *lp);
220}
221
222typedef struct _tbl {
223	wchar_t **list;
224	int cols, *len;
225} TBL;
226#define	DEFCOLS	25
227
228void
229maketbl(void)
230{
231	TBL *t;
232	int coloff, cnt;
233	wchar_t *p, **lp;
234	int *lens, maxcols;
235	TBL *tbl;
236	wchar_t **cols;
237	wchar_t *last;
238
239	if ((t = tbl = calloc(entries, sizeof(TBL))) == NULL)
240		err(1, (char *)NULL);
241	if ((cols = calloc((maxcols = DEFCOLS), sizeof(*cols))) == NULL)
242		err(1, (char *)NULL);
243	if ((lens = calloc(maxcols, sizeof(int))) == NULL)
244		err(1, (char *)NULL);
245	for (cnt = 0, lp = list; cnt < entries; ++cnt, ++lp, ++t) {
246		for (coloff = 0, p = *lp;
247		    (cols[coloff] = wcstok(p, separator, &last));
248		    p = NULL)
249			if (++coloff == maxcols) {
250				if (!(cols = realloc(cols, ((u_int)maxcols +
251				    DEFCOLS) * sizeof(char *))) ||
252				    !(lens = realloc(lens,
253				    ((u_int)maxcols + DEFCOLS) * sizeof(int))))
254					err(1, NULL);
255				memset((char *)lens + maxcols * sizeof(int),
256				    0, DEFCOLS * sizeof(int));
257				maxcols += DEFCOLS;
258			}
259		if ((t->list = calloc(coloff, sizeof(*t->list))) == NULL)
260			err(1, (char *)NULL);
261		if ((t->len = calloc(coloff, sizeof(int))) == NULL)
262			err(1, (char *)NULL);
263		for (t->cols = coloff; --coloff >= 0;) {
264			t->list[coloff] = cols[coloff];
265			t->len[coloff] = width(cols[coloff]);
266			if (t->len[coloff] > lens[coloff])
267				lens[coloff] = t->len[coloff];
268		}
269	}
270	for (cnt = 0, t = tbl; cnt < entries; ++cnt, ++t) {
271		for (coloff = 0; coloff < t->cols  - 1; ++coloff)
272			(void)wprintf(L"%ls%*ls", t->list[coloff],
273			    lens[coloff] - t->len[coloff] + 2, L" ");
274		(void)wprintf(L"%ls\n", t->list[coloff]);
275	}
276}
277
278#define	DEFNUM		1000
279#define	MAXLINELEN	(LINE_MAX + 1)
280
281void
282input(FILE *fp)
283{
284	static int maxentry;
285	int len;
286	wchar_t *p, buf[MAXLINELEN];
287
288	if (!list)
289		if ((list = calloc((maxentry = DEFNUM), sizeof(*list))) ==
290		    NULL)
291			err(1, (char *)NULL);
292	while (fgetws(buf, MAXLINELEN, fp)) {
293		for (p = buf; *p && iswspace(*p); ++p);
294		if (!*p)
295			continue;
296		if (!(p = wcschr(p, L'\n'))) {
297			warnx("line too long");
298			eval = 1;
299			continue;
300		}
301		*p = L'\0';
302		len = width(buf);
303		if (maxlength < len)
304			maxlength = len;
305		if (entries == maxentry) {
306			maxentry += DEFNUM;
307			if (!(list = realloc(list,
308			    (u_int)maxentry * sizeof(*list))))
309				err(1, NULL);
310		}
311		list[entries] = malloc((wcslen(buf) + 1) * sizeof(wchar_t));
312		if (list[entries] == NULL)
313			err(1, NULL);
314		wcscpy(list[entries], buf);
315		entries++;
316	}
317}
318
319/* Like wcswidth(), but ignores non-printing characters. */
320int
321width(const wchar_t *wcs)
322{
323	int w, cw;
324
325	for (w = 0; *wcs != L'\0'; wcs++)
326		if ((cw = wcwidth(*wcs)) > 0)
327			w += cw;
328	return (w);
329}
330
331void
332usage(void)
333{
334
335	(void)fprintf(stderr,
336	    "usage: column [-tx] [-c columns] [-s sep] [file ...]\n");
337	exit(1);
338}
339