haftmann@37743
|
1 |
(* Title: Tools/Code/code_printer.ML
|
haftmann@28060
|
2 |
Author: Florian Haftmann, TU Muenchen
|
haftmann@28060
|
3 |
|
haftmann@28060
|
4 |
Generic operations for pretty printing of target language code.
|
haftmann@28060
|
5 |
*)
|
haftmann@28060
|
6 |
|
haftmann@28060
|
7 |
signature CODE_PRINTER =
|
haftmann@28060
|
8 |
sig
|
haftmann@32908
|
9 |
type itype = Code_Thingol.itype
|
haftmann@32908
|
10 |
type iterm = Code_Thingol.iterm
|
haftmann@32908
|
11 |
type const = Code_Thingol.const
|
haftmann@32908
|
12 |
type dict = Code_Thingol.dict
|
haftmann@32908
|
13 |
|
haftmann@35228
|
14 |
val eqn_error: thm option -> string -> 'a
|
haftmann@28060
|
15 |
|
haftmann@28060
|
16 |
val @@ : 'a * 'a -> 'a list
|
haftmann@28060
|
17 |
val @| : 'a list * 'a -> 'a list
|
haftmann@28060
|
18 |
val str: string -> Pretty.T
|
haftmann@28060
|
19 |
val concat: Pretty.T list -> Pretty.T
|
haftmann@28060
|
20 |
val brackets: Pretty.T list -> Pretty.T
|
haftmann@34178
|
21 |
val enclose: string -> string -> Pretty.T list -> Pretty.T
|
haftmann@39011
|
22 |
val commas: Pretty.T list -> Pretty.T list
|
haftmann@34178
|
23 |
val enum: string -> string -> string -> Pretty.T list -> Pretty.T
|
haftmann@34178
|
24 |
val enum_default: string -> string -> string -> string -> Pretty.T list -> Pretty.T
|
haftmann@28060
|
25 |
val semicolon: Pretty.T list -> Pretty.T
|
haftmann@33986
|
26 |
val doublesemicolon: Pretty.T list -> Pretty.T
|
haftmann@34178
|
27 |
val indent: int -> Pretty.T -> Pretty.T
|
haftmann@39290
|
28 |
val markup_stmt: string -> Pretty.T -> Pretty.T
|
haftmann@39290
|
29 |
val format: string list -> int -> Pretty.T -> string
|
haftmann@28060
|
30 |
|
haftmann@30636
|
31 |
val first_upper: string -> string
|
haftmann@30636
|
32 |
val first_lower: string -> string
|
haftmann@30636
|
33 |
type var_ctxt
|
haftmann@30636
|
34 |
val make_vars: string list -> var_ctxt
|
haftmann@30636
|
35 |
val intro_vars: string list -> var_ctxt -> var_ctxt
|
haftmann@30636
|
36 |
val lookup_var: var_ctxt -> string -> string
|
haftmann@32913
|
37 |
val intro_base_names: (string -> bool) -> (string -> string)
|
haftmann@32913
|
38 |
-> string list -> var_ctxt -> var_ctxt
|
haftmann@32908
|
39 |
val aux_params: var_ctxt -> iterm list list -> string list
|
haftmann@30636
|
40 |
|
haftmann@31056
|
41 |
type literals
|
haftmann@31056
|
42 |
val Literals: { literal_char: string -> string, literal_string: string -> string,
|
haftmann@38195
|
43 |
literal_numeral: int -> string,
|
haftmann@38195
|
44 |
literal_positive_numeral: int -> string,
|
haftmann@38195
|
45 |
literal_alternative_numeral: int -> string,
|
haftmann@34931
|
46 |
literal_naive_numeral: int -> string,
|
haftmann@31056
|
47 |
literal_list: Pretty.T list -> Pretty.T, infix_cons: int * string }
|
haftmann@31056
|
48 |
-> literals
|
haftmann@31056
|
49 |
val literal_char: literals -> string -> string
|
haftmann@31056
|
50 |
val literal_string: literals -> string -> string
|
haftmann@34931
|
51 |
val literal_numeral: literals -> int -> string
|
haftmann@34931
|
52 |
val literal_positive_numeral: literals -> int -> string
|
haftmann@38195
|
53 |
val literal_alternative_numeral: literals -> int -> string
|
haftmann@34931
|
54 |
val literal_naive_numeral: literals -> int -> string
|
haftmann@31056
|
55 |
val literal_list: literals -> Pretty.T list -> Pretty.T
|
haftmann@31056
|
56 |
val infix_cons: literals -> int * string
|
haftmann@31056
|
57 |
|
haftmann@28060
|
58 |
type lrx
|
haftmann@28060
|
59 |
val L: lrx
|
haftmann@28060
|
60 |
val R: lrx
|
haftmann@28060
|
61 |
val X: lrx
|
haftmann@28060
|
62 |
type fixity
|
haftmann@28060
|
63 |
val BR: fixity
|
haftmann@28060
|
64 |
val NOBR: fixity
|
haftmann@28060
|
65 |
val INFX: int * lrx -> fixity
|
haftmann@28060
|
66 |
val APP: fixity
|
haftmann@28060
|
67 |
val brackify: fixity -> Pretty.T list -> Pretty.T
|
haftmann@37239
|
68 |
val brackify_infix: int * lrx -> fixity -> Pretty.T * Pretty.T * Pretty.T -> Pretty.T
|
haftmann@31665
|
69 |
val brackify_block: fixity -> Pretty.T -> Pretty.T list -> Pretty.T -> Pretty.T
|
haftmann@37638
|
70 |
val applify: string -> string -> ('a -> Pretty.T) -> fixity -> Pretty.T -> 'a list -> Pretty.T
|
haftmann@39148
|
71 |
val tuplify: (fixity -> 'a -> Pretty.T) -> fixity -> 'a list -> Pretty.T option
|
haftmann@39148
|
72 |
|
haftmann@28060
|
73 |
type tyco_syntax
|
haftmann@34141
|
74 |
type simple_const_syntax
|
haftmann@37854
|
75 |
type complex_const_syntax
|
haftmann@28060
|
76 |
type const_syntax
|
haftmann@37854
|
77 |
type activated_complex_const_syntax
|
haftmann@37854
|
78 |
datatype activated_const_syntax = Plain_const_syntax of int * string
|
haftmann@37854
|
79 |
| Complex_const_syntax of activated_complex_const_syntax
|
haftmann@37854
|
80 |
val requires_args: const_syntax -> int
|
haftmann@37854
|
81 |
val parse_const_syntax: Token.T list -> const_syntax option * Token.T list
|
haftmann@37854
|
82 |
val parse_tyco_syntax: Token.T list -> tyco_syntax option * Token.T list
|
haftmann@37854
|
83 |
val plain_const_syntax: string -> const_syntax
|
haftmann@37849
|
84 |
val simple_const_syntax: simple_const_syntax -> const_syntax
|
haftmann@37854
|
85 |
val complex_const_syntax: complex_const_syntax -> const_syntax
|
haftmann@31056
|
86 |
val activate_const_syntax: theory -> literals
|
haftmann@37854
|
87 |
-> string -> const_syntax -> Code_Thingol.naming -> activated_const_syntax * Code_Thingol.naming
|
haftmann@35228
|
88 |
val gen_print_app: (thm option -> var_ctxt -> const * iterm list -> Pretty.T list)
|
haftmann@35228
|
89 |
-> (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T)
|
haftmann@37849
|
90 |
-> (string -> activated_const_syntax option)
|
haftmann@35228
|
91 |
-> thm option -> var_ctxt -> fixity -> const * iterm list -> Pretty.T
|
haftmann@35228
|
92 |
val gen_print_bind: (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T)
|
haftmann@35228
|
93 |
-> thm option -> fixity
|
haftmann@31889
|
94 |
-> iterm -> var_ctxt -> Pretty.T * var_ctxt
|
haftmann@28060
|
95 |
end;
|
haftmann@28060
|
96 |
|
haftmann@28060
|
97 |
structure Code_Printer : CODE_PRINTER =
|
haftmann@28060
|
98 |
struct
|
haftmann@28060
|
99 |
|
haftmann@28060
|
100 |
open Code_Thingol;
|
haftmann@28060
|
101 |
|
haftmann@39278
|
102 |
(** generic nonsense *)
|
haftmann@39278
|
103 |
|
haftmann@35228
|
104 |
fun eqn_error (SOME thm) s = error (s ^ ",\nin equation " ^ Display.string_of_thm_without_context thm)
|
haftmann@35228
|
105 |
| eqn_error NONE s = error s;
|
haftmann@28060
|
106 |
|
haftmann@39278
|
107 |
val code_presentationN = "code_presentation";
|
haftmann@39291
|
108 |
val stmt_nameN = "stmt_name";
|
haftmann@39278
|
109 |
val _ = Markup.add_mode code_presentationN YXML.output_markup;
|
haftmann@39278
|
110 |
|
haftmann@39278
|
111 |
|
haftmann@34071
|
112 |
(** assembling and printing text pieces **)
|
haftmann@28060
|
113 |
|
haftmann@28060
|
114 |
infixr 5 @@;
|
haftmann@28060
|
115 |
infixr 5 @|;
|
haftmann@28060
|
116 |
fun x @@ y = [x, y];
|
haftmann@28060
|
117 |
fun xs @| y = xs @ [y];
|
haftmann@39884
|
118 |
val str = Print_Mode.setmp [] Pretty.str;
|
haftmann@28060
|
119 |
val concat = Pretty.block o Pretty.breaks;
|
haftmann@39884
|
120 |
val commas = Print_Mode.setmp [] Pretty.commas;
|
haftmann@39884
|
121 |
fun enclose l r = Print_Mode.setmp [] (Pretty.enclose l r);
|
haftmann@34178
|
122 |
val brackets = enclose "(" ")" o Pretty.breaks;
|
haftmann@39884
|
123 |
fun enum sep l r = Print_Mode.setmp [] (Pretty.enum sep l r);
|
haftmann@34178
|
124 |
fun enum_default default sep l r [] = str default
|
haftmann@34178
|
125 |
| enum_default default sep l r xs = enum sep l r xs;
|
haftmann@28060
|
126 |
fun semicolon ps = Pretty.block [concat ps, str ";"];
|
haftmann@33986
|
127 |
fun doublesemicolon ps = Pretty.block [concat ps, str ";;"];
|
haftmann@39884
|
128 |
fun indent i = Print_Mode.setmp [] (Pretty.indent i);
|
haftmann@28060
|
129 |
|
haftmann@39884
|
130 |
fun markup_stmt name = Print_Mode.setmp [code_presentationN]
|
haftmann@39303
|
131 |
(Pretty.mark (code_presentationN, [(stmt_nameN, name)]));
|
haftmann@39296
|
132 |
|
haftmann@39801
|
133 |
fun filter_presentation [] tree =
|
haftmann@39801
|
134 |
Buffer.empty
|
haftmann@39801
|
135 |
|> fold XML.add_content tree
|
haftmann@39801
|
136 |
| filter_presentation presentation_names tree =
|
haftmann@39291
|
137 |
let
|
haftmann@39801
|
138 |
fun is_selected (name, attrs) =
|
haftmann@39801
|
139 |
name = code_presentationN
|
haftmann@39801
|
140 |
andalso member (op =) presentation_names (the (Properties.get attrs stmt_nameN));
|
haftmann@39801
|
141 |
fun add_content_with_space tree (is_first, buf) =
|
haftmann@39801
|
142 |
buf
|
haftmann@39801
|
143 |
|> not is_first ? Buffer.add "\n\n"
|
haftmann@39801
|
144 |
|> XML.add_content tree
|
haftmann@39801
|
145 |
|> pair false;
|
haftmann@39801
|
146 |
fun filter (XML.Elem (name_attrs, xs)) =
|
haftmann@39801
|
147 |
fold (if is_selected name_attrs then add_content_with_space else filter) xs
|
haftmann@39801
|
148 |
| filter (XML.Text s) = I;
|
haftmann@39801
|
149 |
in snd (fold filter tree (true, Buffer.empty)) end;
|
haftmann@39296
|
150 |
|
haftmann@39296
|
151 |
fun format presentation_names width =
|
haftmann@39884
|
152 |
Print_Mode.setmp [code_presentationN] (Pretty.string_of_margin width)
|
haftmann@39296
|
153 |
#> YXML.parse_body
|
haftmann@39801
|
154 |
#> filter_presentation presentation_names
|
haftmann@39902
|
155 |
#> Buffer.add "\n"
|
haftmann@39801
|
156 |
#> Buffer.content;
|
haftmann@34071
|
157 |
|
haftmann@28060
|
158 |
|
haftmann@30636
|
159 |
(** names and variable name contexts **)
|
haftmann@30636
|
160 |
|
haftmann@30636
|
161 |
type var_ctxt = string Symtab.table * Name.context;
|
haftmann@30636
|
162 |
|
haftmann@30636
|
163 |
fun make_vars names = (fold (fn name => Symtab.update_new (name, name)) names Symtab.empty,
|
haftmann@30636
|
164 |
Name.make_context names);
|
haftmann@30636
|
165 |
|
haftmann@30636
|
166 |
fun intro_vars names (namemap, namectxt) =
|
haftmann@30636
|
167 |
let
|
haftmann@30636
|
168 |
val (names', namectxt') = Name.variants names namectxt;
|
haftmann@30636
|
169 |
val namemap' = fold2 (curry Symtab.update) names names' namemap;
|
haftmann@30636
|
170 |
in (namemap', namectxt') end;
|
haftmann@30636
|
171 |
|
haftmann@30636
|
172 |
fun lookup_var (namemap, _) name = case Symtab.lookup namemap name
|
haftmann@30636
|
173 |
of SOME name' => name'
|
haftmann@30636
|
174 |
| NONE => error ("Invalid name in context: " ^ quote name);
|
haftmann@30636
|
175 |
|
wenzelm@40875
|
176 |
val first_upper = implode o nth_map 0 Symbol.to_ascii_upper o raw_explode;
|
wenzelm@40875
|
177 |
val first_lower = implode o nth_map 0 Symbol.to_ascii_lower o raw_explode;
|
haftmann@30636
|
178 |
|
haftmann@32908
|
179 |
fun aux_params vars lhss =
|
haftmann@32908
|
180 |
let
|
haftmann@32908
|
181 |
fun fish_param _ (w as SOME _) = w
|
haftmann@32908
|
182 |
| fish_param (IVar (SOME v)) NONE = SOME v
|
haftmann@32908
|
183 |
| fish_param _ NONE = NONE;
|
haftmann@32908
|
184 |
fun fillup_param _ (_, SOME v) = v
|
haftmann@32908
|
185 |
| fillup_param x (i, NONE) = x ^ string_of_int i;
|
haftmann@32908
|
186 |
val fished1 = fold (map2 fish_param) lhss (replicate (length (hd lhss)) NONE);
|
wenzelm@44206
|
187 |
val x = singleton (Name.variant_list (map_filter I fished1)) "x";
|
haftmann@32908
|
188 |
val fished2 = map_index (fillup_param x) fished1;
|
haftmann@32908
|
189 |
val (fished3, _) = Name.variants fished2 Name.context;
|
haftmann@32908
|
190 |
val vars' = intro_vars fished3 vars;
|
haftmann@32908
|
191 |
in map (lookup_var vars') fished3 end;
|
haftmann@32908
|
192 |
|
haftmann@32913
|
193 |
fun intro_base_names no_syntax deresolve names = names
|
haftmann@32913
|
194 |
|> map_filter (fn name => if no_syntax name then
|
haftmann@32913
|
195 |
let val name' = deresolve name in
|
haftmann@32913
|
196 |
if Long_Name.is_qualified name' then NONE else SOME name'
|
haftmann@32913
|
197 |
end else NONE)
|
haftmann@32913
|
198 |
|> intro_vars;
|
haftmann@32913
|
199 |
|
haftmann@30636
|
200 |
|
haftmann@31056
|
201 |
(** pretty literals **)
|
haftmann@31056
|
202 |
|
haftmann@31056
|
203 |
datatype literals = Literals of {
|
haftmann@31056
|
204 |
literal_char: string -> string,
|
haftmann@31056
|
205 |
literal_string: string -> string,
|
haftmann@34931
|
206 |
literal_numeral: int -> string,
|
haftmann@34931
|
207 |
literal_positive_numeral: int -> string,
|
haftmann@38195
|
208 |
literal_alternative_numeral: int -> string,
|
haftmann@34931
|
209 |
literal_naive_numeral: int -> string,
|
haftmann@31056
|
210 |
literal_list: Pretty.T list -> Pretty.T,
|
haftmann@31056
|
211 |
infix_cons: int * string
|
haftmann@31056
|
212 |
};
|
haftmann@31056
|
213 |
|
haftmann@31056
|
214 |
fun dest_Literals (Literals lits) = lits;
|
haftmann@31056
|
215 |
|
haftmann@31056
|
216 |
val literal_char = #literal_char o dest_Literals;
|
haftmann@31056
|
217 |
val literal_string = #literal_string o dest_Literals;
|
haftmann@31056
|
218 |
val literal_numeral = #literal_numeral o dest_Literals;
|
haftmann@34931
|
219 |
val literal_positive_numeral = #literal_positive_numeral o dest_Literals;
|
haftmann@38195
|
220 |
val literal_alternative_numeral = #literal_alternative_numeral o dest_Literals;
|
haftmann@34931
|
221 |
val literal_naive_numeral = #literal_naive_numeral o dest_Literals;
|
haftmann@31056
|
222 |
val literal_list = #literal_list o dest_Literals;
|
haftmann@31056
|
223 |
val infix_cons = #infix_cons o dest_Literals;
|
haftmann@31056
|
224 |
|
haftmann@31056
|
225 |
|
haftmann@28060
|
226 |
(** syntax printer **)
|
haftmann@28060
|
227 |
|
haftmann@28060
|
228 |
(* binding priorities *)
|
haftmann@28060
|
229 |
|
haftmann@28060
|
230 |
datatype lrx = L | R | X;
|
haftmann@28060
|
231 |
|
haftmann@28060
|
232 |
datatype fixity =
|
haftmann@28060
|
233 |
BR
|
haftmann@28060
|
234 |
| NOBR
|
haftmann@28060
|
235 |
| INFX of (int * lrx);
|
haftmann@28060
|
236 |
|
haftmann@28060
|
237 |
val APP = INFX (~1, L);
|
haftmann@28060
|
238 |
|
haftmann@28060
|
239 |
fun fixity_lrx L L = false
|
haftmann@28060
|
240 |
| fixity_lrx R R = false
|
haftmann@28060
|
241 |
| fixity_lrx _ _ = true;
|
haftmann@28060
|
242 |
|
haftmann@28060
|
243 |
fun fixity NOBR _ = false
|
haftmann@28060
|
244 |
| fixity _ NOBR = false
|
haftmann@33986
|
245 |
| fixity (INFX (pr, lr)) (INFX (print_ctxt, lr_ctxt)) =
|
haftmann@33986
|
246 |
pr < print_ctxt
|
haftmann@33986
|
247 |
orelse pr = print_ctxt
|
haftmann@28060
|
248 |
andalso fixity_lrx lr lr_ctxt
|
haftmann@33986
|
249 |
orelse print_ctxt = ~1
|
haftmann@28060
|
250 |
| fixity BR (INFX _) = false
|
haftmann@28060
|
251 |
| fixity _ _ = true;
|
haftmann@28060
|
252 |
|
haftmann@28060
|
253 |
fun gen_brackify _ [p] = p
|
haftmann@34178
|
254 |
| gen_brackify true (ps as _::_) = enclose "(" ")" ps
|
haftmann@28060
|
255 |
| gen_brackify false (ps as _::_) = Pretty.block ps;
|
haftmann@28060
|
256 |
|
haftmann@28060
|
257 |
fun brackify fxy_ctxt =
|
haftmann@28060
|
258 |
gen_brackify (fixity BR fxy_ctxt) o Pretty.breaks;
|
haftmann@28060
|
259 |
|
haftmann@37239
|
260 |
fun brackify_infix infx fxy_ctxt (l, m, r) =
|
haftmann@37239
|
261 |
(if fixity (INFX infx) fxy_ctxt then enclose "(" ")" else Pretty.block)
|
haftmann@37239
|
262 |
([l, str " ", m, Pretty.brk 1, r]);
|
haftmann@28060
|
263 |
|
haftmann@31665
|
264 |
fun brackify_block fxy_ctxt p1 ps p2 =
|
haftmann@31665
|
265 |
let val p = Pretty.block_enclose (p1, p2) ps
|
haftmann@31665
|
266 |
in if fixity BR fxy_ctxt
|
haftmann@34178
|
267 |
then enclose "(" ")" [p]
|
haftmann@31665
|
268 |
else p
|
haftmann@31665
|
269 |
end;
|
haftmann@31665
|
270 |
|
haftmann@37638
|
271 |
fun applify opn cls f fxy_ctxt p [] = p
|
haftmann@37638
|
272 |
| applify opn cls f fxy_ctxt p ps =
|
haftmann@34242
|
273 |
(if (fixity BR fxy_ctxt) then enclose "(" ")" else Pretty.block)
|
haftmann@37638
|
274 |
(p @@ enum "," opn cls (map f ps));
|
haftmann@34242
|
275 |
|
haftmann@39148
|
276 |
fun tuplify _ _ [] = NONE
|
haftmann@39148
|
277 |
| tuplify print fxy [x] = SOME (print fxy x)
|
haftmann@39148
|
278 |
| tuplify print _ xs = SOME (enum "," "(" ")" (map (print NOBR) xs));
|
haftmann@39148
|
279 |
|
haftmann@28060
|
280 |
|
haftmann@28060
|
281 |
(* generic syntax *)
|
haftmann@28060
|
282 |
|
haftmann@28060
|
283 |
type tyco_syntax = int * ((fixity -> itype -> Pretty.T)
|
haftmann@28060
|
284 |
-> fixity -> itype list -> Pretty.T);
|
haftmann@34141
|
285 |
|
haftmann@34141
|
286 |
type simple_const_syntax = int * ((fixity -> iterm -> Pretty.T)
|
haftmann@34141
|
287 |
-> fixity -> (iterm * itype) list -> Pretty.T);
|
haftmann@37854
|
288 |
|
haftmann@37854
|
289 |
type complex_const_syntax = int * (string list * (literals -> string list
|
haftmann@31056
|
290 |
-> (var_ctxt -> fixity -> iterm -> Pretty.T)
|
haftmann@35228
|
291 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T));
|
haftmann@37854
|
292 |
|
haftmann@37854
|
293 |
datatype const_syntax = plain_const_syntax of string
|
haftmann@37854
|
294 |
| complex_const_syntax of complex_const_syntax;
|
haftmann@37854
|
295 |
|
haftmann@37854
|
296 |
fun requires_args (plain_const_syntax _) = 0
|
haftmann@37854
|
297 |
| requires_args (complex_const_syntax (k, _)) = k;
|
haftmann@28060
|
298 |
|
haftmann@34163
|
299 |
fun simple_const_syntax syn =
|
haftmann@37854
|
300 |
complex_const_syntax (apsnd (fn f => ([], (fn _ => fn _ => fn print => fn _ => fn vars => f (print vars)))) syn);
|
haftmann@28060
|
301 |
|
haftmann@37854
|
302 |
type activated_complex_const_syntax = int * ((var_ctxt -> fixity -> iterm -> Pretty.T)
|
haftmann@37854
|
303 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T)
|
haftmann@31056
|
304 |
|
haftmann@37854
|
305 |
datatype activated_const_syntax = Plain_const_syntax of int * string
|
haftmann@37854
|
306 |
| Complex_const_syntax of activated_complex_const_syntax;
|
haftmann@37854
|
307 |
|
haftmann@37854
|
308 |
fun activate_const_syntax thy literals c (plain_const_syntax s) naming =
|
haftmann@37854
|
309 |
(Plain_const_syntax (Code.args_number thy c, s), naming)
|
haftmann@37854
|
310 |
| activate_const_syntax thy literals c (complex_const_syntax (n, (cs, f))) naming =
|
haftmann@37854
|
311 |
fold_map (Code_Thingol.ensure_declared_const thy) cs naming
|
haftmann@37854
|
312 |
|-> (fn cs' => pair (Complex_const_syntax (n, f literals cs')));
|
haftmann@37854
|
313 |
|
haftmann@39149
|
314 |
fun gen_print_app print_app_expr print_term const_syntax some_thm vars fxy (app as ((c, (_, function_typs)), ts)) =
|
haftmann@39149
|
315 |
case const_syntax c
|
haftmann@37854
|
316 |
of NONE => brackify fxy (print_app_expr some_thm vars app)
|
haftmann@37854
|
317 |
| SOME (Plain_const_syntax (_, s)) => brackify fxy (str s :: map (print_term some_thm vars BR) ts)
|
haftmann@37854
|
318 |
| SOME (Complex_const_syntax (k, print)) =>
|
haftmann@28060
|
319 |
let
|
haftmann@37854
|
320 |
fun print' fxy ts = print (print_term some_thm) some_thm vars fxy (ts ~~ take k function_typs);
|
haftmann@28060
|
321 |
in if k = length ts
|
haftmann@33986
|
322 |
then print' fxy ts
|
haftmann@28060
|
323 |
else if k < length ts
|
haftmann@28060
|
324 |
then case chop k ts of (ts1, ts2) =>
|
haftmann@37854
|
325 |
brackify fxy (print' APP ts1 :: map (print_term some_thm vars BR) ts2)
|
haftmann@37854
|
326 |
else print_term some_thm vars fxy (Code_Thingol.eta_expand k app)
|
haftmann@28060
|
327 |
end;
|
haftmann@28060
|
328 |
|
haftmann@33986
|
329 |
fun gen_print_bind print_term thm (fxy : fixity) pat vars =
|
haftmann@28060
|
330 |
let
|
haftmann@31889
|
331 |
val vs = Code_Thingol.fold_varnames (insert (op =)) pat [];
|
haftmann@31873
|
332 |
val vars' = intro_vars vs vars;
|
haftmann@33986
|
333 |
in (print_term thm vars' fxy pat, vars') end;
|
haftmann@28060
|
334 |
|
haftmann@28060
|
335 |
|
haftmann@28060
|
336 |
(* mixfix syntax *)
|
haftmann@28060
|
337 |
|
haftmann@28060
|
338 |
datatype 'a mixfix =
|
haftmann@28060
|
339 |
Arg of fixity
|
haftmann@37854
|
340 |
| String of string
|
haftmann@37854
|
341 |
| Break;
|
haftmann@28060
|
342 |
|
haftmann@28060
|
343 |
fun mk_mixfix prep_arg (fixity_this, mfx) =
|
haftmann@28060
|
344 |
let
|
haftmann@28060
|
345 |
fun is_arg (Arg _) = true
|
haftmann@28060
|
346 |
| is_arg _ = false;
|
haftmann@28060
|
347 |
val i = (length o filter is_arg) mfx;
|
haftmann@28060
|
348 |
fun fillin _ [] [] =
|
haftmann@28060
|
349 |
[]
|
haftmann@33986
|
350 |
| fillin print (Arg fxy :: mfx) (a :: args) =
|
haftmann@33986
|
351 |
(print fxy o prep_arg) a :: fillin print mfx args
|
haftmann@37854
|
352 |
| fillin print (String s :: mfx) args =
|
haftmann@37854
|
353 |
str s :: fillin print mfx args
|
haftmann@37854
|
354 |
| fillin print (Break :: mfx) args =
|
haftmann@37854
|
355 |
Pretty.brk 1 :: fillin print mfx args;
|
haftmann@28060
|
356 |
in
|
haftmann@33986
|
357 |
(i, fn print => fn fixity_ctxt => fn args =>
|
haftmann@33986
|
358 |
gen_brackify (fixity fixity_this fixity_ctxt) (fillin print mfx args))
|
haftmann@28060
|
359 |
end;
|
haftmann@28060
|
360 |
|
haftmann@28060
|
361 |
fun parse_infix prep_arg (x, i) s =
|
haftmann@28060
|
362 |
let
|
haftmann@28060
|
363 |
val l = case x of L => INFX (i, L) | _ => INFX (i, X);
|
haftmann@28060
|
364 |
val r = case x of R => INFX (i, R) | _ => INFX (i, X);
|
haftmann@28060
|
365 |
in
|
haftmann@37854
|
366 |
mk_mixfix prep_arg (INFX (i, x), [Arg l, String " ", String s, Break, Arg r])
|
haftmann@28060
|
367 |
end;
|
haftmann@28060
|
368 |
|
haftmann@37854
|
369 |
fun parse_mixfix mk_plain mk_complex prep_arg s =
|
haftmann@28060
|
370 |
let
|
haftmann@28060
|
371 |
val sym_any = Scan.one Symbol.is_regular;
|
haftmann@28060
|
372 |
val parse = Scan.optional ($$ "!" >> K true) false -- Scan.repeat (
|
haftmann@28060
|
373 |
($$ "(" -- $$ "_" -- $$ ")" >> K (Arg NOBR))
|
haftmann@28060
|
374 |
|| ($$ "_" >> K (Arg BR))
|
haftmann@37854
|
375 |
|| ($$ "/" |-- Scan.repeat ($$ " ") >> (K Break))
|
haftmann@28060
|
376 |
|| (Scan.repeat1
|
haftmann@28060
|
377 |
( $$ "'" |-- sym_any
|
haftmann@28060
|
378 |
|| Scan.unless ($$ "_" || $$ "/" || $$ "(" |-- $$ "_" |-- $$ ")")
|
haftmann@37854
|
379 |
sym_any) >> (String o implode)));
|
haftmann@28060
|
380 |
in case Scan.finite Symbol.stopper parse (Symbol.explode s)
|
haftmann@37854
|
381 |
of ((false, [String s]), []) => mk_plain s
|
haftmann@37854
|
382 |
| ((_, p as [_]), []) => mk_complex (mk_mixfix prep_arg (NOBR, p))
|
haftmann@37854
|
383 |
| ((b, p as _ :: _ :: _), []) => mk_complex (mk_mixfix prep_arg (if b then NOBR else BR, p))
|
haftmann@28060
|
384 |
| _ => Scan.!!
|
haftmann@28060
|
385 |
(the_default ("malformed mixfix annotation: " ^ quote s) o snd) Scan.fail ()
|
haftmann@28060
|
386 |
end;
|
haftmann@28060
|
387 |
|
haftmann@28060
|
388 |
val (infixK, infixlK, infixrK) = ("infix", "infixl", "infixr");
|
haftmann@28060
|
389 |
|
haftmann@37854
|
390 |
fun parse_syntax mk_plain mk_complex prep_arg =
|
haftmann@37854
|
391 |
Scan.option (
|
wenzelm@36970
|
392 |
((Parse.$$$ infixK >> K X)
|
wenzelm@36970
|
393 |
|| (Parse.$$$ infixlK >> K L)
|
wenzelm@36970
|
394 |
|| (Parse.$$$ infixrK >> K R))
|
haftmann@37854
|
395 |
-- Parse.nat -- Parse.string
|
haftmann@37854
|
396 |
>> (fn ((x, i), s) => mk_complex (parse_infix prep_arg (x, i) s))
|
haftmann@37854
|
397 |
|| Parse.string >> (fn s => (parse_mixfix mk_plain mk_complex prep_arg s)));
|
haftmann@28060
|
398 |
|
wenzelm@36970
|
399 |
val _ = List.app Keyword.keyword [infixK, infixlK, infixrK];
|
haftmann@28060
|
400 |
|
wenzelm@38145
|
401 |
fun parse_tyco_syntax x = parse_syntax (fn s => (0, (K o K o K o str) s)) I I x;
|
haftmann@37854
|
402 |
|
haftmann@37854
|
403 |
val parse_const_syntax = parse_syntax plain_const_syntax simple_const_syntax fst;
|
haftmann@37854
|
404 |
|
haftmann@28060
|
405 |
end; (*struct*)
|