author | clasohm |
Wed, 29 Jun 1994 12:01:17 +0200 | |
changeset 441 | 2b97bd6415b7 |
parent 261 | 3441647c8c90 |
child 512 | 55755ed9fab9 |
permissions | -rw-r--r-- |
18 | 1 |
(* Title: Pure/Syntax/pretty.ML |
0 | 2 |
ID: $Id$ |
3 |
Author: Lawrence C Paulson |
|
4 |
Copyright 1991 University of Cambridge |
|
5 |
||
18 | 6 |
Generic pretty printing module. |
0 | 7 |
|
8 |
Loosely based on |
|
9 |
D. C. Oppen, "Pretty Printing", |
|
10 |
ACM Transactions on Programming Languages and Systems (1980), 465-483. |
|
11 |
||
12 |
The object to be printed is given as a tree with indentation and line |
|
13 |
breaking information. A "break" inserts a newline if the text until |
|
14 |
the next break is too long to fit on the current line. After the newline, |
|
15 |
text is indented to the level of the enclosing block. Normally, if a block |
|
16 |
is broken then all enclosing blocks will also be broken. Only "inconsistent |
|
17 |
breaks" are provided. |
|
18 |
||
19 |
The stored length of a block is used in breakdist (to treat each inner block as |
|
20 |
a unit for breaking). |
|
21 |
*) |
|
22 |
||
18 | 23 |
type pprint_args = (string -> unit) * (int -> unit) * (int -> unit) * |
24 |
(unit -> unit) * (unit -> unit); |
|
0 | 25 |
|
26 |
signature PRETTY = |
|
18 | 27 |
sig |
0 | 28 |
type T |
18 | 29 |
val blk: int * T list -> T |
30 |
val brk: int -> T |
|
31 |
val fbrk: T |
|
32 |
val str: string -> T |
|
33 |
val lst: string * string -> T list -> T |
|
34 |
val quote: T -> T |
|
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
35 |
val commas: T list -> T list |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
36 |
val breaks: T list -> T list |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
37 |
val fbreaks: T list -> T list |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
38 |
val block: T list -> T |
261 | 39 |
val strs: string list -> T |
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
40 |
val parents: string -> string -> T list -> T |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
41 |
val list: string -> string -> T list -> T |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
42 |
val str_list: string -> string -> string list -> T |
261 | 43 |
val big_list: string -> T list -> T |
18 | 44 |
val string_of: T -> string |
261 | 45 |
val writeln: T -> unit |
18 | 46 |
val str_of: T -> string |
47 |
val pprint: T -> pprint_args -> unit |
|
0 | 48 |
val setdepth: int -> unit |
49 |
val setmargin: int -> unit |
|
18 | 50 |
end; |
0 | 51 |
|
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
52 |
functor PrettyFun(): PRETTY = |
0 | 53 |
struct |
54 |
||
55 |
(*Printing items: compound phrases, strings, and breaks*) |
|
56 |
datatype T = Block of T list * int * int (*indentation, length*) |
|
57 |
| String of string |
|
58 |
| Break of bool*int (*mandatory flag; width if not taken*); |
|
59 |
||
60 |
(*Add the lengths of the expressions until the next Break; if no Break then |
|
61 |
include "after", to account for text following this block. *) |
|
62 |
fun breakdist (Block(_,_,len)::es, after) = len + breakdist(es, after) |
|
63 |
| breakdist (String s :: es, after) = size s + breakdist (es, after) |
|
64 |
| breakdist (Break _ :: es, after) = 0 |
|
65 |
| breakdist ([], after) = after; |
|
66 |
||
67 |
fun repstring a 0 = "" |
|
68 |
| repstring a 1 = a |
|
69 |
| repstring a k = |
|
70 |
if k mod 2 = 0 then repstring(a^a) (k div 2) |
|
71 |
else repstring(a^a) (k div 2) ^ a; |
|
72 |
||
73 |
(*** Type for lines of text: string, # of lines, position on line ***) |
|
74 |
||
75 |
type text = {tx: string, nl: int, pos: int}; |
|
76 |
||
77 |
val emptytext = {tx="", nl=0, pos=0}; |
|
78 |
||
79 |
fun blanks wd {tx,nl,pos} = |
|
118 | 80 |
{tx = tx ^ repstring " " wd, |
0 | 81 |
nl = nl, |
82 |
pos = pos+wd}; |
|
83 |
||
84 |
fun newline {tx,nl,pos} = |
|
85 |
{tx = tx ^ "\n", |
|
86 |
nl = nl+1, |
|
87 |
pos = 0}; |
|
88 |
||
89 |
fun string s {tx,nl,pos} = |
|
90 |
{tx = tx ^ s, |
|
91 |
nl = nl, |
|
92 |
pos = pos + size(s)}; |
|
93 |
||
94 |
(*** Formatting ***) |
|
95 |
||
96 |
val margin = ref 80 (*right margin, or page width*) |
|
97 |
and breakgain = ref 4 (*minimum added space required of a break*) |
|
98 |
and emergencypos = ref 40; (*position too far to right*) |
|
99 |
||
100 |
fun setmargin m = |
|
101 |
(margin := m; |
|
102 |
breakgain := !margin div 20; |
|
103 |
emergencypos := !margin div 2); |
|
104 |
||
118 | 105 |
(*Search for the next break (at this or higher levels) and force it to occur*) |
0 | 106 |
fun forcenext [] = [] |
107 |
| forcenext (Break(_,wd) :: es) = Break(true,0) :: es |
|
108 |
| forcenext (e :: es) = e :: forcenext es; |
|
109 |
||
118 | 110 |
(*es is list of expressions to print; |
111 |
blockin is the indentation of the current block; |
|
112 |
after is the width of the following context until next break. *) |
|
0 | 113 |
fun format ([], _, _) text = text |
114 |
| format (e::es, blockin, after) (text as {pos,nl,...}) = |
|
115 |
(case e of |
|
116 |
Block(bes,indent,wd) => |
|
117 |
let val blockin' = (pos + indent) mod !emergencypos |
|
118 |
val btext = format(bes, blockin', breakdist(es,after)) text |
|
118 | 119 |
(*If this block was broken then force the next break.*) |
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
120 |
val es2 = if nl < #nl(btext) then forcenext es else es |
0 | 121 |
in format (es2,blockin,after) btext end |
122 |
| String s => format (es,blockin,after) (string s text) |
|
118 | 123 |
| Break(force,wd) => (*no break if text to next break fits on this line |
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
124 |
or if breaking would add only breakgain to space *) |
0 | 125 |
format (es,blockin,after) |
126 |
(if not force andalso |
|
127 |
pos+wd <= max[!margin - breakdist(es,after), |
|
128 |
blockin + !breakgain] |
|
118 | 129 |
then blanks wd text (*just insert wd blanks*) |
0 | 130 |
else blanks blockin (newline text))); |
131 |
||
132 |
(*** Exported functions to create formatting expressions ***) |
|
133 |
||
134 |
fun length (Block(_,_,len)) = len |
|
135 |
| length (String s) = size s |
|
136 |
| length (Break(_,wd)) = wd; |
|
137 |
||
138 |
val str = String |
|
139 |
and fbrk = Break(true,0); |
|
140 |
||
141 |
fun brk wd = Break(false,wd); |
|
142 |
||
143 |
fun blk (indent,es) = |
|
144 |
let fun sum([], k) = k |
|
145 |
| sum(e::es, k) = sum(es, length e + k) |
|
146 |
in Block(es,indent, sum(es,0)) end; |
|
147 |
||
118 | 148 |
(*Join the elements of es as a comma-separated list, bracketted by lp and rp*) |
0 | 149 |
fun lst(lp,rp) es = |
150 |
let fun add(e,es) = str"," :: brk 1 :: e :: es; |
|
151 |
fun list(e :: (es as _::_)) = str lp :: e :: foldr add (es,[str rp]) |
|
152 |
| list(e::[]) = [str lp, e, str rp] |
|
153 |
| list([]) = [] |
|
154 |
in blk(size lp, list es) end; |
|
155 |
||
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
156 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
157 |
(* utils *) |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
158 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
159 |
fun quote prt = |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
160 |
blk (1, [str "\"", prt, str "\""]); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
161 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
162 |
fun commas prts = |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
163 |
flat (separate [str ",", brk 1] (map (fn x => [x]) prts)); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
164 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
165 |
fun breaks prts = separate (brk 1) prts; |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
166 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
167 |
fun fbreaks prts = separate fbrk prts; |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
168 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
169 |
fun block prts = blk (2, prts); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
170 |
|
261 | 171 |
val strs = block o breaks o (map str); |
172 |
||
236
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
173 |
fun parents lpar rpar prts = |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
174 |
block (str lpar :: (prts @ [str rpar])); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
175 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
176 |
fun list lpar rpar prts = |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
177 |
parents lpar rpar (commas prts); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
178 |
|
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
179 |
fun str_list lpar rpar strs = |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
180 |
list lpar rpar (map str strs); |
d33cd0011e48
added some utils: commas, breaks, fbreaks, block, parents, list, str_list;
wenzelm
parents:
118
diff
changeset
|
181 |
|
261 | 182 |
fun big_list name prts = |
183 |
block (fbreaks (str name :: prts)); |
|
184 |
||
0 | 185 |
|
186 |
||
187 |
(*** Pretty printing with depth limitation ***) |
|
188 |
||
189 |
val depth = ref 0; (*maximum depth; 0 means no limit*) |
|
190 |
||
191 |
fun setdepth dp = (depth := dp); |
|
192 |
||
118 | 193 |
(*Recursively prune blocks, discarding all text exceeding depth dp*) |
0 | 194 |
fun pruning dp (Block(bes,indent,wd)) = |
195 |
if dp>0 then blk(indent, map (pruning(dp-1)) bes) |
|
196 |
else String"..." |
|
197 |
| pruning dp e = e; |
|
198 |
||
199 |
fun prune dp e = if dp>0 then pruning dp e else e; |
|
200 |
||
201 |
||
118 | 202 |
fun string_of e = #tx (format ([prune (!depth) e], 0, 0) emptytext); |
0 | 203 |
|
261 | 204 |
val writeln = writeln o string_of; |
205 |
||
0 | 206 |
|
118 | 207 |
(*Create a single flat string: no line breaking*) |
0 | 208 |
fun str_of prt = |
209 |
let |
|
210 |
fun s_of (Block (prts, _, _)) = implode (map s_of prts) |
|
211 |
| s_of (String s) = s |
|
18 | 212 |
| s_of (Break (false, wd)) = repstring " " wd |
213 |
| s_of (Break (true, _)) = " "; |
|
0 | 214 |
in |
215 |
s_of (prune (! depth) prt) |
|
216 |
end; |
|
217 |
||
118 | 218 |
(*Part of the interface to the Poly/ML and New Jersey ML pretty printers*) |
18 | 219 |
fun pprint prt (put_str, begin_blk, put_brk, put_fbrk, end_blk) = |
0 | 220 |
let |
221 |
fun pp (Block (prts, ind, _)) = (begin_blk ind; pp_lst prts; end_blk ()) |
|
222 |
| pp (String s) = put_str s |
|
18 | 223 |
| pp (Break (false, wd)) = put_brk wd |
224 |
| pp (Break (true, _)) = put_fbrk () |
|
0 | 225 |
and pp_lst [] = () |
226 |
| pp_lst (prt :: prts) = (pp prt; pp_lst prts); |
|
227 |
in |
|
228 |
pp (prune (! depth) prt) |
|
229 |
end; |
|
230 |
||
231 |
||
232 |
(*** Initialization ***) |
|
233 |
||
234 |
val () = setmargin 80; |
|
235 |
||
236 |
end; |
|
237 |