/* GNU fmt -- simple text formatter.
Copyright (C) 1994-2006, 2008 Free Software Foundation, Inc.
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see . */
/* Written by Ross Paterson . */
#include
#include
#include
#include
/* Redefine. Otherwise, systems (Unicos for one) with headers that define
it to be a type get syntax errors for the variable declaration below. */
#define word unused_word_type
#include "system.h"
#include "error.h"
#include "quote.h"
#include "xstrtol.h"
/* The official name of this program (e.g., no `g' prefix). */
#define PROGRAM_NAME "fmt"
#define AUTHORS proper_name ("Ross Paterson")
/* The following parameters represent the program's idea of what is
"best". Adjust to taste, subject to the caveats given. */
/* Default longest permitted line length (max_width). */
#define WIDTH 75
/* Prefer lines to be LEEWAY % shorter than the maximum width, giving
room for optimization. */
#define LEEWAY 7
/* The default secondary indent of tagged paragraph used for unindented
one-line paragraphs not preceded by any multi-line paragraphs. */
#define DEF_INDENT 3
/* Costs and bonuses are expressed as the equivalent departure from the
optimal line length, multiplied by 10. e.g. assigning something a
cost of 50 means that it is as bad as a line 5 characters too short
or too long. The definition of SHORT_COST(n) should not be changed.
However, EQUIV(n) may need tuning. */
/* FIXME: "fmt" misbehaves given large inputs or options. One
possible workaround for part of the problem is to change COST to be
a floating-point type. There are other problems besides COST,
though; see MAXWORDS below. */
typedef long int COST;
#define MAXCOST TYPE_MAXIMUM (COST)
#define SQR(n) ((n) * (n))
#define EQUIV(n) SQR ((COST) (n))
/* Cost of a filled line n chars longer or shorter than best_width. */
#define SHORT_COST(n) EQUIV ((n) * 10)
/* Cost of the difference between adjacent filled lines. */
#define RAGGED_COST(n) (SHORT_COST (n) / 2)
/* Basic cost per line. */
#define LINE_COST EQUIV (70)
/* Cost of breaking a line after the first word of a sentence, where
the length of the word is N. */
#define WIDOW_COST(n) (EQUIV (200) / ((n) + 2))
/* Cost of breaking a line before the last word of a sentence, where
the length of the word is N. */
#define ORPHAN_COST(n) (EQUIV (150) / ((n) + 2))
/* Bonus for breaking a line at the end of a sentence. */
#define SENTENCE_BONUS EQUIV (50)
/* Cost of breaking a line after a period not marking end of a sentence.
With the definition of sentence we are using (borrowed from emacs, see
get_line()) such a break would then look like a sentence break. Hence
we assign a very high cost -- it should be avoided unless things are
really bad. */
#define NOBREAK_COST EQUIV (600)
/* Bonus for breaking a line before open parenthesis. */
#define PAREN_BONUS EQUIV (40)
/* Bonus for breaking a line after other punctuation. */
#define PUNCT_BONUS EQUIV(40)
/* Credit for breaking a long paragraph one line later. */
#define LINE_CREDIT EQUIV(3)
/* Size of paragraph buffer, in words and characters. Longer paragraphs
are handled neatly (cf. flush_paragraph()), so long as these values
are considerably greater than required by the width. These values
cannot be extended indefinitely: doing so would run into size limits
and/or cause more overflows in cost calculations. FIXME: Remove these
arbitrary limits. */
#define MAXWORDS 1000
#define MAXCHARS 5000
/* Extra ctype(3)-style macros. */
#define isopen(c) (strchr ("([`'\"", c) != NULL)
#define isclose(c) (strchr (")]'\"", c) != NULL)
#define isperiod(c) (strchr (".?!", c) != NULL)
/* Size of a tab stop, for expansion on input and re-introduction on
output. */
#define TABWIDTH 8
/* Word descriptor structure. */
typedef struct Word WORD;
struct Word
{
/* Static attributes determined during input. */
const char *text; /* the text of the word */
int length; /* length of this word */
int space; /* the size of the following space */
unsigned int paren:1; /* starts with open paren */
unsigned int period:1; /* ends in [.?!])* */
unsigned int punct:1; /* ends in punctuation */
unsigned int final:1; /* end of sentence */
/* The remaining fields are computed during the optimization. */
int line_length; /* length of the best line starting here */
COST best_cost; /* cost of best paragraph starting here */
WORD *next_break; /* break which achieves best_cost */
};
/* Forward declarations. */
static void set_prefix (char *p);
static void fmt (FILE *f);
static bool get_paragraph (FILE *f);
static int get_line (FILE *f, int c);
static int get_prefix (FILE *f);
static int get_space (FILE *f, int c);
static int copy_rest (FILE *f, int c);
static bool same_para (int c);
static void flush_paragraph (void);
static void fmt_paragraph (void);
static void check_punctuation (WORD *w);
static COST base_cost (WORD *this);
static COST line_cost (WORD *next, int len);
static void put_paragraph (WORD *finish);
static void put_line (WORD *w, int indent);
static void put_word (WORD *w);
static void put_space (int space);
/* Option values. */
/* If true, first 2 lines may have different indent (default false). */
static bool crown;
/* If true, first 2 lines _must_ have different indent (default false). */
static bool tagged;
/* If true, each line is a paragraph on its own (default false). */
static bool split;
/* If true, don't preserve inter-word spacing (default false). */
static bool uniform;
/* Prefix minus leading and trailing spaces (default ""). */
static const char *prefix;
/* User-supplied maximum line width (default WIDTH). The only output
lines longer than this will each comprise a single word. */
static int max_width;
/* Values derived from the option values. */
/* The length of prefix minus leading space. */
static int prefix_full_length;
/* The length of the leading space trimmed from the prefix. */
static int prefix_lead_space;
/* The length of prefix minus leading and trailing space. */
static int prefix_length;
/* The preferred width of text lines, set to LEEWAY % less than max_width. */
static int best_width;
/* Dynamic variables. */
/* Start column of the character most recently read from the input file. */
static int in_column;
/* Start column of the next character to be written to stdout. */
static int out_column;
/* Space for the paragraph text -- longer paragraphs are handled neatly
(cf. flush_paragraph()). */
static char parabuf[MAXCHARS];
/* A pointer into parabuf, indicating the first unused character position. */
static char *wptr;
/* The words of a paragraph -- longer paragraphs are handled neatly
(cf. flush_paragraph()). */
static WORD word[MAXWORDS];
/* A pointer into the above word array, indicating the first position
after the last complete word. Sometimes it will point at an incomplete
word. */
static WORD *word_limit;
/* If true, current input file contains tab characters, and so tabs can be
used for white space on output. */
static bool tabs;
/* Space before trimmed prefix on each line of the current paragraph. */
static int prefix_indent;
/* Indentation of the first line of the current paragraph. */
static int first_indent;
/* Indentation of other lines of the current paragraph */
static int other_indent;
/* To detect the end of a paragraph, we need to look ahead to the first
non-blank character after the prefix on the next line, or the first
character on the following line that failed to match the prefix.
We can reconstruct the lookahead from that character (next_char), its
position on the line (in_column) and the amount of space before the
prefix (next_prefix_indent). See get_paragraph() and copy_rest(). */
/* The last character read from the input file. */
static int next_char;
/* The space before the trimmed prefix (or part of it) on the next line
after the current paragraph. */
static int next_prefix_indent;
/* If nonzero, the length of the last line output in the current
paragraph, used to charge for raggedness at the split point for long
paragraphs chosen by fmt_paragraph(). */
static int last_line_length;
void
usage (int status)
{
if (status != EXIT_SUCCESS)
fprintf (stderr, _("Try `%s --help' for more information.\n"),
program_name);
else
{
printf (_("Usage: %s [-DIGITS] [OPTION]... [FILE]...\n"), program_name);
fputs (_("\
Reformat each paragraph in the FILE(s), writing to standard output.\n\
If no FILE or if FILE is `-', read standard input.\n\
\n\
"), stdout);
fputs (_("\
Mandatory arguments to long options are mandatory for short options too.\n\
"), stdout);
fputs (_("\
-c, --crown-margin preserve indentation of first two lines\n\
-p, --prefix=STRING reformat only lines beginning with STRING,\n\
reattaching the prefix to reformatted lines\n\
-s, --split-only split long lines, but do not refill\n\
"),
stdout);
fputs (_("\
-t, --tagged-paragraph indentation of first line different from second\n\
-u, --uniform-spacing one space between words, two after sentences\n\
-w, --width=WIDTH maximum line width (default of 75 columns)\n\
"), stdout);
fputs (HELP_OPTION_DESCRIPTION, stdout);
fputs (VERSION_OPTION_DESCRIPTION, stdout);
fputs (_("\
\n\
With no FILE, or when FILE is -, read standard input.\n"),
stdout);
emit_bug_reporting_address ();
}
exit (status);
}
/* Decode options and launch execution. */
static const struct option long_options[] =
{
{"crown-margin", no_argument, NULL, 'c'},
{"prefix", required_argument, NULL, 'p'},
{"split-only", no_argument, NULL, 's'},
{"tagged-paragraph", no_argument, NULL, 't'},
{"uniform-spacing", no_argument, NULL, 'u'},
{"width", required_argument, NULL, 'w'},
{GETOPT_HELP_OPTION_DECL},
{GETOPT_VERSION_OPTION_DECL},
{NULL, 0, NULL, 0},
};
int
main (int argc, char **argv)
{
int optchar;
bool ok = true;
char const *max_width_option = NULL;
initialize_main (&argc, &argv);
set_program_name (argv[0]);
setlocale (LC_ALL, "");
bindtextdomain (PACKAGE, LOCALEDIR);
textdomain (PACKAGE);
atexit (close_stdout);
crown = tagged = split = uniform = false;
max_width = WIDTH;
prefix = "";
prefix_length = prefix_lead_space = prefix_full_length = 0;
if (argc > 1 && argv[1][0] == '-' && ISDIGIT (argv[1][1]))
{
/* Old option syntax; a dash followed by one or more digits. */
max_width_option = argv[1] + 1;
/* Make the option we just parsed invisible to getopt. */
argv[1] = argv[0];
argv++;
argc--;
}
while ((optchar = getopt_long (argc, argv, "0123456789cstuw:p:",
long_options, NULL))
!= -1)
switch (optchar)
{
default:
if (ISDIGIT (optchar))
error (0, 0, _("invalid option -- %c; -WIDTH is recognized\
only when it is the first\noption; use -w N instead"),
optchar);
usage (EXIT_FAILURE);
case 'c':
crown = true;
break;
case 's':
split = true;
break;
case 't':
tagged = true;
break;
case 'u':
uniform = true;
break;
case 'w':
max_width_option = optarg;
break;
case 'p':
set_prefix (optarg);
break;
case_GETOPT_HELP_CHAR;
case_GETOPT_VERSION_CHAR (PROGRAM_NAME, AUTHORS);
}
if (max_width_option)
{
/* Limit max_width to MAXCHARS / 2; otherwise, the resulting
output can be quite ugly. */
unsigned long int tmp;
if (! (xstrtoul (max_width_option, NULL, 10, &tmp, "") == LONGINT_OK
&& tmp <= MAXCHARS / 2))
error (EXIT_FAILURE, 0, _("invalid width: %s"),
quote (max_width_option));
max_width = tmp;
}
best_width = max_width * (2 * (100 - LEEWAY) + 1) / 200;
if (optind == argc)
fmt (stdin);
else
{
for (; optind < argc; optind++)
{
char *file = argv[optind];
if (STREQ (file, "-"))
fmt (stdin);
else
{
FILE *in_stream;
in_stream = fopen (file, "r");
if (in_stream != NULL)
{
fmt (in_stream);
if (fclose (in_stream) == EOF)
{
error (0, errno, "%s", file);
ok = false;
}
}
else
{
error (0, errno, _("cannot open %s for reading"),
quote (file));
ok = false;
}
}
}
}
exit (ok ? EXIT_SUCCESS : EXIT_FAILURE);
}
/* Trim space from the front and back of the string P, yielding the prefix,
and record the lengths of the prefix and the space trimmed. */
static void
set_prefix (char *p)
{
char *s;
prefix_lead_space = 0;
while (*p == ' ')
{
prefix_lead_space++;
p++;
}
prefix = p;
prefix_full_length = strlen (p);
s = p + prefix_full_length;
while (s > p && s[-1] == ' ')
s--;
*s = '\0';
prefix_length = s - p;
}
/* read file F and send formatted output to stdout. */
static void
fmt (FILE *f)
{
tabs = false;
other_indent = 0;
next_char = get_prefix (f);
while (get_paragraph (f))
{
fmt_paragraph ();
put_paragraph (word_limit);
}
}
/* Set the global variable `other_indent' according to SAME_PARAGRAPH
and other global variables. */
static void
set_other_indent (bool same_paragraph)
{
if (split)
other_indent = first_indent;
else if (crown)
{
other_indent = (same_paragraph ? in_column : first_indent);
}
else if (tagged)
{
if (same_paragraph && in_column != first_indent)
{
other_indent = in_column;
}
/* Only one line: use the secondary indent from last time if it
splits, or 0 if there have been no multi-line paragraphs in the
input so far. But if these rules make the two indents the same,
pick a new secondary indent. */
else if (other_indent == first_indent)
other_indent = first_indent == 0 ? DEF_INDENT : 0;
}
else
{
other_indent = first_indent;
}
}
/* Read a paragraph from input file F. A paragraph consists of a
maximal number of non-blank (excluding any prefix) lines subject to:
* In split mode, a paragraph is a single non-blank line.
* In crown mode, the second and subsequent lines must have the
same indentation, but possibly different from the indent of the
first line.
* Tagged mode is similar, but the first and second lines must have
different indentations.
* Otherwise, all lines of a paragraph must have the same indent.
If a prefix is in effect, it must be present at the same indent for
each line in the paragraph.
Return false if end-of-file was encountered before the start of a
paragraph, else true. */
static bool
get_paragraph (FILE *f)
{
int c;
last_line_length = 0;
c = next_char;
/* Scan (and copy) blank lines, and lines not introduced by the prefix. */
while (c == '\n' || c == EOF
|| next_prefix_indent < prefix_lead_space
|| in_column < next_prefix_indent + prefix_full_length)
{
c = copy_rest (f, c);
if (c == EOF)
{
next_char = EOF;
return false;
}
putchar ('\n');
c = get_prefix (f);
}
/* Got a suitable first line for a paragraph. */
prefix_indent = next_prefix_indent;
first_indent = in_column;
wptr = parabuf;
word_limit = word;
c = get_line (f, c);
set_other_indent (same_para (c));
/* Read rest of paragraph (unless split is specified). */
if (split)
{
/* empty */
}
else if (crown)
{
if (same_para (c))
{
do
{ /* for each line till the end of the para */
c = get_line (f, c);
}
while (same_para (c) && in_column == other_indent);
}
}
else if (tagged)
{
if (same_para (c) && in_column != first_indent)
{
do
{ /* for each line till the end of the para */
c = get_line (f, c);
}
while (same_para (c) && in_column == other_indent);
}
}
else
{
while (same_para (c) && in_column == other_indent)
c = get_line (f, c);
}
(word_limit - 1)->period = (word_limit - 1)->final = true;
next_char = c;
return true;
}
/* Copy to the output a line that failed to match the prefix, or that
was blank after the prefix. In the former case, C is the character
that failed to match the prefix. In the latter, C is \n or EOF.
Return the character (\n or EOF) ending the line. */
static int
copy_rest (FILE *f, int c)
{
const char *s;
out_column = 0;
if (in_column > next_prefix_indent || (c != '\n' && c != EOF))
{
put_space (next_prefix_indent);
for (s = prefix; out_column != in_column && *s; out_column++)
putchar (*s++);
if (c != EOF && c != '\n')
put_space (in_column - out_column);
if (c == EOF && in_column >= next_prefix_indent + prefix_length)
putchar ('\n');
}
while (c != '\n' && c != EOF)
{
putchar (c);
c = getc (f);
}
return c;
}
/* Return true if a line whose first non-blank character after the
prefix (if any) is C could belong to the current paragraph,
otherwise false. */
static bool
same_para (int c)
{
return (next_prefix_indent == prefix_indent
&& in_column >= next_prefix_indent + prefix_full_length
&& c != '\n' && c != EOF);
}
/* Read a line from input file F, given first non-blank character C
after the prefix, and the following indent, and break it into words.
A word is a maximal non-empty string of non-white characters. A word
ending in [.?!]["')\]]* and followed by end-of-line or at least two
spaces ends a sentence, as in emacs.
Return the first non-blank character of the next line. */
static int
get_line (FILE *f, int c)
{
int start;
char *end_of_parabuf;
WORD *end_of_word;
end_of_parabuf = ¶buf[MAXCHARS];
end_of_word = &word[MAXWORDS - 2];
do
{ /* for each word in a line */
/* Scan word. */
word_limit->text = wptr;
do
{
if (wptr == end_of_parabuf)
{
set_other_indent (true);
flush_paragraph ();
}
*wptr++ = c;
c = getc (f);
}
while (c != EOF && !isspace (c));
in_column += word_limit->length = wptr - word_limit->text;
check_punctuation (word_limit);
/* Scan inter-word space. */
start = in_column;
c = get_space (f, c);
word_limit->space = in_column - start;
word_limit->final = (c == EOF
|| (word_limit->period
&& (c == '\n' || word_limit->space > 1)));
if (c == '\n' || c == EOF || uniform)
word_limit->space = word_limit->final ? 2 : 1;
if (word_limit == end_of_word)
{
set_other_indent (true);
flush_paragraph ();
}
word_limit++;
}
while (c != '\n' && c != EOF);
return get_prefix (f);
}
/* Read a prefix from input file F. Return either first non-matching
character, or first non-blank character after the prefix. */
static int
get_prefix (FILE *f)
{
int c;
in_column = 0;
c = get_space (f, getc (f));
if (prefix_length == 0)
next_prefix_indent = prefix_lead_space < in_column ?
prefix_lead_space : in_column;
else
{
const char *p;
next_prefix_indent = in_column;
for (p = prefix; *p != '\0'; p++)
{
unsigned char pc = *p;
if (c != pc)
return c;
in_column++;
c = getc (f);
}
c = get_space (f, c);
}
return c;
}
/* Read blank characters from input file F, starting with C, and keeping
in_column up-to-date. Return first non-blank character. */
static int
get_space (FILE *f, int c)
{
for (;;)
{
if (c == ' ')
in_column++;
else if (c == '\t')
{
tabs = true;
in_column = (in_column / TABWIDTH + 1) * TABWIDTH;
}
else
return c;
c = getc (f);
}
}
/* Set extra fields in word W describing any attached punctuation. */
static void
check_punctuation (WORD *w)
{
char const *start = w->text;
char const *finish = start + (w->length - 1);
unsigned char fin = *finish;
w->paren = isopen (*start);
w->punct = !! ispunct (fin);
while (start < finish && isclose (*finish))
finish--;
w->period = isperiod (*finish);
}
/* Flush part of the paragraph to make room. This function is called on
hitting the limit on the number of words or characters. */
static void
flush_paragraph (void)
{
WORD *split_point;
WORD *w;
int shift;
COST best_break;
/* In the special case where it's all one word, just flush it. */
if (word_limit == word)
{
fwrite (parabuf, sizeof *parabuf, wptr - parabuf, stdout);
wptr = parabuf;
return;
}
/* Otherwise:
- format what you have so far as a paragraph,
- find a low-cost line break near the end,
- output to there,
- make that the start of the paragraph. */
fmt_paragraph ();
/* Choose a good split point. */
split_point = word_limit;
best_break = MAXCOST;
for (w = word->next_break; w != word_limit; w = w->next_break)
{
if (w->best_cost - w->next_break->best_cost < best_break)
{
split_point = w;
best_break = w->best_cost - w->next_break->best_cost;
}
if (best_break <= MAXCOST - LINE_CREDIT)
best_break += LINE_CREDIT;
}
put_paragraph (split_point);
/* Copy text of words down to start of parabuf -- we use memmove because
the source and target may overlap. */
memmove (parabuf, split_point->text, wptr - split_point->text);
shift = split_point->text - parabuf;
wptr -= shift;
/* Adjust text pointers. */
for (w = split_point; w <= word_limit; w++)
w->text -= shift;
/* Copy words from split_point down to word -- we use memmove because
the source and target may overlap. */
memmove (word, split_point, (word_limit - split_point + 1) * sizeof *word);
word_limit -= split_point - word;
}
/* Compute the optimal formatting for the whole paragraph by computing
and remembering the optimal formatting for each suffix from the empty
one to the whole paragraph. */
static void
fmt_paragraph (void)
{
WORD *start, *w;
int len;
COST wcost, best;
int saved_length;
word_limit->best_cost = 0;
saved_length = word_limit->length;
word_limit->length = max_width; /* sentinel */
for (start = word_limit - 1; start >= word; start--)
{
best = MAXCOST;
len = start == word ? first_indent : other_indent;
/* At least one word, however long, in the line. */
w = start;
len += w->length;
do
{
w++;
/* Consider breaking before w. */
wcost = line_cost (w, len) + w->best_cost;
if (start == word && last_line_length > 0)
wcost += RAGGED_COST (len - last_line_length);
if (wcost < best)
{
best = wcost;
start->next_break = w;
start->line_length = len;
}
/* This is a kludge to keep us from computing `len' as the
sum of the sentinel length and some non-zero number.
Since the sentinel w->length may be INT_MAX, adding
to that would give a negative result. */
if (w == word_limit)
break;
len += (w - 1)->space + w->length; /* w > start >= word */
}
while (len < max_width);
start->best_cost = best + base_cost (start);
}
word_limit->length = saved_length;
}
/* Return the constant component of the cost of breaking before the
word THIS. */
static COST
base_cost (WORD *this)
{
COST cost;
cost = LINE_COST;
if (this > word)
{
if ((this - 1)->period)
{
if ((this - 1)->final)
cost -= SENTENCE_BONUS;
else
cost += NOBREAK_COST;
}
else if ((this - 1)->punct)
cost -= PUNCT_BONUS;
else if (this > word + 1 && (this - 2)->final)
cost += WIDOW_COST ((this - 1)->length);
}
if (this->paren)
cost -= PAREN_BONUS;
else if (this->final)
cost += ORPHAN_COST (this->length);
return cost;
}
/* Return the component of the cost of breaking before word NEXT that
depends on LEN, the length of the line beginning there. */
static COST
line_cost (WORD *next, int len)
{
int n;
COST cost;
if (next == word_limit)
return 0;
n = best_width - len;
cost = SHORT_COST (n);
if (next->next_break != word_limit)
{
n = len - next->line_length;
cost += RAGGED_COST (n);
}
return cost;
}
/* Output to stdout a paragraph from word up to (but not including)
FINISH, which must be in the next_break chain from word. */
static void
put_paragraph (WORD *finish)
{
WORD *w;
put_line (word, first_indent);
for (w = word->next_break; w != finish; w = w->next_break)
put_line (w, other_indent);
}
/* Output to stdout the line beginning with word W, beginning in column
INDENT, including the prefix (if any). */
static void
put_line (WORD *w, int indent)
{
WORD *endline;
out_column = 0;
put_space (prefix_indent);
fputs (prefix, stdout);
out_column += prefix_length;
put_space (indent - out_column);
endline = w->next_break - 1;
for (; w != endline; w++)
{
put_word (w);
put_space (w->space);
}
put_word (w);
last_line_length = out_column;
putchar ('\n');
}
/* Output to stdout the word W. */
static void
put_word (WORD *w)
{
const char *s;
int n;
s = w->text;
for (n = w->length; n != 0; n--)
putchar (*s++);
out_column += w->length;
}
/* Output to stdout SPACE spaces, or equivalent tabs. */
static void
put_space (int space)
{
int space_target, tab_target;
space_target = out_column + space;
if (tabs)
{
tab_target = space_target / TABWIDTH * TABWIDTH;
if (out_column + 1 < tab_target)
while (out_column < tab_target)
{
putchar ('\t');
out_column = (out_column / TABWIDTH + 1) * TABWIDTH;
}
}
while (out_column < space_target)
{
putchar (' ');
out_column++;
}
}