pr.c

Go to the documentation of this file.

/* pr - print files                     Author: Michiel Huisjes */

/* Pr - print files
 *
 * Author: Michiel Huisjes.
 * Modified: Jacob P. Bunschoten.       (30 nov 87)
 *      When "columns" is not given and numbering is on:
 *              line numbers are correlated with input lines.
 *      (try pr [-1] -n file )
 *      tabs are accounted for.
 *      When numbering is turned on, width know this.
 *      automatic line-folding. -f to get the original program.
 *      backspaces are accounted for. -b to disable this.
 *      multi-column mode changed.
 *      header can be given and used.
 *      format changed may occur between printing of several files:
 *              pr -l30 file1 -w75 file2
 *
 * Modified: Rick Thomas.               (Sept 12, 1988)
 *      added "-M" option to cover functionality of old "-n" option,
 *      and made "-n" option behavior compatible with system V.
 *
 * Usage: pr [+page] [-columns] [-h header] [-wwidth] [-llength] [-ntm] [files]
 *        -t : Do not print the 5 line header and trailer at the page.
 *        -n : Turn on line numbering.
 *        -M : Use "Minix" style line numbering -- Each page begins at
 *             a line number that is an even multiple of the page length.
 *             Like the listings in Appendix E of the book.
 *        +page    : Start printing at page n.
 *        -columns : Print files in n-columns.
 *        -l length: Take the length of the page to be n instead of 66
 *        -h header: Take next argument as page header.
 *        -w width  : Take the width of the page to be n instead of default 79
 *        -f : do not fold lines.
 *
 * Modified: Lars Fredriksen            (Jan 19, 1990)
 *      fixed the program so that
 *              pr -n *.c
 *      would work. The clobal variable 'width' was decremented
 *      by NUM_WIDTH, for each file, resulting in width finally
 *      being so small that nothing was printed. Used the local
 *      variable 'w' for the width adjustment (in print())
 *
 * Modified: Kenneth J. Hendrickson     (10 April 1991)
 *      date in header should be last modification date for files,
 *      and the current time for stdin.
 *
 * Modified: Kees J. Bot                (5 October 1992)
 *      Use localtime(3) to get the date, it knows TZ.
 */

#include <sys/types.h>
#include <sys/stat.h>
#include <string.h>
#include <time.h>
#include <stdlib.h>
#include <stdio.h>


#define DEF_LENGTH      66
#define DEF_WIDTH       79
#define NUM_WIDTH       8
#define TAB_WIDTH       8       /* fixed tab_width */

/* Used to compute next (fixed) tabstop */
#define TO_TAB(x)       (( (x) + TAB_WIDTH ) & ~07 )

typedef char BOOL;

#define FALSE           0
#define TRUE            1

/* EAT: eat rest of input line */
#define EAT(fp)         while((c=getc(fp))!='\n' && c!=EOF)

/* L_BUF: calculate address of pointer to char (string) used in format */
#define L_BUF(i,j)      * (char **) (line_buf + (i + j*length)*sizeof(char *))

char *header;
BOOL no_header;
BOOL number = FALSE;
BOOL minix_number = FALSE;
BOOL ext_header_set = FALSE;    /* external header given */
BOOL back_space = TRUE;         /* back space correction in line width */
BOOL dont_fold = FALSE;         /* original. If the line does not fit eat it. */
short columns;
short cwidth;
short start_page = 1;
short width = DEF_WIDTH;
short length = DEF_LENGTH;
short linenr;
char *line_buf;                 /* used in format for multi-column output */

char output[1024];

_PROTOTYPE(int main, (int argc, char **argv));
_PROTOTYPE(static char *myalloc, (size_t size));
_PROTOTYPE(char skip_page, (int lines, int width, FILE * filep));
_PROTOTYPE(void format, (FILE * filep));
_PROTOTYPE(void print_page, (int pagenr, int maxcol));
_PROTOTYPE(void print, (FILE * filep));
_PROTOTYPE(void out_header, (int page));
_PROTOTYPE(void print_time, (time_t t));

int main(argc, argv)
int argc;
char *argv[];
{
  FILE *file;
  char *ptr;
  int index = 1;                /* index is one ahead of argc */
  int line, col;

  setbuf(stdout, output);
  do {
        if (argc == index)      /* No arguments (left) */
                goto pr_files;

        ptr = argv[index++];
        if (*ptr == '+') {
                start_page = atoi(++ptr);
                continue;
        }
        if (*ptr != '-') {      /* no flags */
                index--;
                goto pr_files;
        }
        if (*++ptr >= '0' && *ptr <= '9') {
                columns = atoi(ptr);
                if (columns <= 0) columns = 1;
                continue;       /* Fetch next flag */
        }
        while (*ptr) switch (*ptr++) {
                    case 't':   no_header = TRUE;       break;
                    case 'n':
                        number = TRUE;
                        minix_number = FALSE;
                        break;
                    case 'M':
                        number = TRUE;
                        minix_number = TRUE;
                        break;
                    case 'h':
                        header = argv[index++];
                        ext_header_set = TRUE;
                        break;
                    case 'w':
                        if ((width = atoi(ptr)) <= 0) width = DEF_WIDTH;
                        *ptr = '\0';
                        break;
                    case 'l':
                        if ((length = atoi(ptr)) <= 0) length = DEF_LENGTH;
                        *ptr = '\0';
                        break;
                    case 'b':   /* back_space correction off */
                        back_space = FALSE;
                        break;
                    case 'f':   /* do not fold lines */
                        dont_fold = TRUE;
                        break;
                    default:
                        fprintf(stderr, "Usage: %s [+page] [-columns] [-h header] [-w<width>] [-l<length>] [-nMt] [files]\n", argv[0]);
                        exit(1);
                }
        continue;               /* Scan for next flags */


        /* ==============  flags are read. Print the file(s) ========= */

pr_files:

        if (!no_header) length -= 10;

        if (columns) {
                cwidth = width / columns + 1;
                if (columns > width) {
                        fprintf(stderr, "Too many columns for page width.\n");
                        exit(1);
                }

                /* Allocate piece of mem to hold some pointers */
                line_buf = myalloc(length * columns * sizeof(char *));
        }
        for (line = 0; line < length; line++)
                for (col = 0; col < columns; col++)
                        L_BUF(line, col) = NULL;

        if (length <= 0) {
                fprintf(stderr, "Minimal length should be %d\n", no_header ?
                        1 : 11);
                exit(1);
        }
        while (index <= argc) { /* print all files, including stdin */
                if (index < argc && (*argv[index] == '-' || *argv[index] == '+'))
                        break;  /* Format change */

                if (argc == index) {    /* no file specified, so stdin */
                        if (!ext_header_set) header = "";
                        file = stdin;
                } else {
                        if ((file = fopen(argv[index], "r")) == (FILE *) 0) {
                                fprintf(stderr, "Cannot open %s\n", argv[index++]);
                                continue;
                        }
                        if (!ext_header_set) header = argv[index];
                }
                if (columns)
                        format(file);
                else
                        print(file);
                fclose(file);
                if (++index >= argc)
                        break;  /* all files (including stdin) done */
        }
        if (index >= argc) break;
        /* When control comes here. format changes are to be done.
         * reinitialize some variables */
        if (!no_header) length += 10;

        start_page = 1;
        ext_header_set = FALSE;
        if (columns) free(line_buf);
  } while (index <= argc);      /* "pr -l60" should work too */

  (void) fflush(stdout);
  return(0);
}

char skip_page(lines, width, filep)
int lines, width;
FILE *filep;
{
  short c;
  int char_cnt;
  int w;

  do {
        w = width;
        if (number)             /* first lines are shorter */
                if (!columns || /* called from print(file)  */
                    !(lines % columns)) /* called from format(file) */
                        w -= NUM_WIDTH;

        char_cnt = 0;
        while ((c = getc(filep)) != '\n' && c != EOF && char_cnt < w) {
                /* Calculate if this line is longer than "width (w)"
                 * characters */
                if (c == '\b' && back_space) {
                        if (--char_cnt < 0) char_cnt = 0;
                } else if (c == '\t')
                        char_cnt = TO_TAB(char_cnt);
                else
                        char_cnt++;
        }
        if (dont_fold && c != '\n' && c != EOF) EAT(filep);
        lines--;
        if (c == '\n') linenr++;
  } while (lines > 0 && c != EOF);

  return c;                     /* last char read */
}

void format(filep)
FILE *filep;
{
  char buf[512];
  short c = '\0';
  short index, lines, i;
  short page_number = 0;
  short maxcol = columns;
  short wdth;
  short line, col;

  do {
        /* Check printing of page */
        page_number++;

        if (page_number < start_page && c != EOF) {
                c = (char) skip_page(columns * length, cwidth, filep);
                continue;
        }
        if (c == EOF) return;

        lines = columns * length;
        for (line = 0; line < length; line++)
                for (col = 0; col < columns; col++) {
                        if (L_BUF(line, col) != NULL)
                                free(L_BUF(line, col));
                        L_BUF(line, col) = (char *) NULL;
                }
        line = 0;
        col = 0;
        do {
                index = 0;
                wdth = cwidth - 1;
                if (number && !col)     /* need room for numbers */
                        wdth -= NUM_WIDTH;

                /* Intermidiate colums are shortened by 1 char */
                /* Last column not */
                if (col + 1 == columns) wdth++;
                for (i = 0; i < wdth - 1; i++) {
                        c = getc(filep);
                        if (c == '\n' || c == EOF) break;

                        if (c == '\b' && back_space) {
                                buf[index++] = '\b';
                                if (--i < 0) {  /* just in case ... */
                                        i = 0;
                                        index = 0;
                                }
                        } else if (c == '\t') {
                                int cnt, max;

                                max = TO_TAB(i);
                                for (cnt = i; cnt < max; cnt++)
                                        buf[index++] = ' ';
                                i = max - 1;
                        } else
                                buf[index++] = (char) c;
                }
                buf[index++] = '\0';
                /* Collected enough chars (or eoln, or EOF) */

                /* First char is EOF */
                if (i == 0 && lines == columns * length && c == EOF) return;

                /* Alloc mem to hold this (sub) string */
                L_BUF(line, col) = myalloc(index * sizeof(char));
                strcpy(L_BUF(line, col), buf);

                line++;
                line %= length;
                if (line == 0) {
                        col++;
                        col %= columns;
                }
                if (dont_fold && c != '\n' && c != EOF) EAT(filep);
                lines--;        /* line ready for output */
                if (c == EOF) {
                        maxcol = columns - lines / length;
                }
        } while (c != EOF && lines);
        print_page(page_number, maxcol);
  } while (c != EOF);
}

void print_page(pagenr, maxcol)
short pagenr, maxcol;
{
  short pad, i, j;
  short width;
  char *p;

  if (minix_number)
        linenr = (pagenr - 1) * length + 1;
  else
        linenr = 1;

  if (!no_header) out_header(pagenr);

  for (i = 0; i < length; i++) {
        for (j = 0; j < maxcol; j++) {
                width = cwidth;
                if (number && j == 0) { /* first columns */
                        printf("%7.7d ", linenr++);     /* 7 == NUM_WIDTH-1 */
                        width -= NUM_WIDTH;
                }
                pad = 0;
                if (p = (char *) L_BUF(i, j))
                        for (; pad < width - 1 && *p; pad++) putchar(*p++);
                if (j < maxcol - 1) while (pad++ < width - 1)
                                putchar(' ');
        }
        putchar('\n');
  }
  if (!no_header) printf("\n\n\n\n\n");
}

void print(filep)
FILE *filep;
{
  short c = '\0';
  short page_number = 0;
  short lines;
  short cnt;
  short w = width;
  BOOL pr_number = TRUE;        /* only real lines are numbered, not folded
                         * parts */

  linenr = 1;
  if (number) w -= NUM_WIDTH;

  do {
        /* Check printing of page */
        page_number++;

        if (page_number < start_page && c != EOF) {
                pr_number = FALSE;
                c = skip_page(length, w, filep);
                if (c == '\n') pr_number = TRUE;
                continue;
        }
        if (c == EOF) return;

        if (minix_number) linenr = (page_number - 1) * length + 1;

        if (page_number == start_page) c = getc(filep);

        /* Print the page */
        lines = length;
        while (lines && c != EOF) {
                if (lines == length && !no_header) out_header(page_number);
                if (number)
                        if (pr_number)
                                printf("%7.7d ", linenr++);     /* 7 == NUM_WIDTH-1 */
                        else
                                printf("%7c ", ' ');    /* 7 == NUM_WIDTH-1 */
                pr_number = FALSE;
                cnt = 0;
                while (c != '\n' && c != EOF && cnt < w) {
                        if (c == '\t') {
                                int i, max;
                                max = TO_TAB(cnt);
                                for (i = cnt; i < max; i++) putchar(' ');
                                cnt = max - 1;
                        } else if (c == '\b' && back_space) {
                                putchar('\b');
                                cnt--;
                        } else
                                putchar(c);
                        c = getc(filep);
                        cnt++;
                }
                putchar('\n');
                if (dont_fold && c != '\n' && c != EOF) EAT(filep);
                lines--;
                if (c == '\n') {
                        c = getc(filep);
                        pr_number = TRUE;
                }
        }
        if (lines == length)    /* We never printed anything on this
                                 * page --  */
                return;         /* even the header, so dont try to fill it up */
        if (!no_header)         /* print the trailer -- 5 blank lines */
                printf("\n\n\n\n\n");
  } while (c != EOF);

  /* Fill last page */
  if (page_number >= start_page) {
        while (lines--) putchar('\n');
  }
}

static char *myalloc(size)
size_t size;                    /* How many bytes */
{
  void *ptr;

  ptr = malloc(size);
  if (ptr == NULL) {
        fprintf(stderr, "malloc returned NULL\n");
        exit(1);
  }
  return(char *) ptr;
}

void out_header(page)
short page;
{
  time_t t;
  struct stat buf;

  if (strlen(header)) {
        stat(header, &buf);     /* use last modify time for file */
        t = buf.st_mtime;
  } else
        (void) time(&t);        /* use current time for stdin */
  print_time(t);
  printf("  %s   Page %d\n\n\n", header, page);
}

char *moname[] = {
          "Jan", "Feb", "Mar", "Apr", "May", "Jun",
          "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
};

/* Print the date. */
void print_time(t)
time_t t;
{
  struct tm *tm;

  tm = localtime(&t);

  printf("\n\n%s %2d %2d:%02d %d",
         moname[tm->tm_mon],
         tm->tm_mday,
         tm->tm_hour,
         tm->tm_min,
         1900 + tm->tm_year
        );
}

---