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