source: trunk/third/rs/rs.c @ 25989

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