author | haftmann |
Mon, 19 Jul 2010 11:55:44 +0200 | |
changeset 37881 | 096c8397c989 |
parent 37876 | 48116a1764c5 |
child 37899 | 527cedd71356 |
permissions | -rw-r--r-- |
37744 | 1 |
(* Title: Tools/Code/code_printer.ML |
28060 | 2 |
Author: Florian Haftmann, TU Muenchen |
3 |
||
4 |
Generic operations for pretty printing of target language code. |
|
5 |
*) |
|
6 |
||
7 |
signature CODE_PRINTER = |
|
8 |
sig |
|
32908 | 9 |
type itype = Code_Thingol.itype |
10 |
type iterm = Code_Thingol.iterm |
|
11 |
type const = Code_Thingol.const |
|
12 |
type dict = Code_Thingol.dict |
|
13 |
||
35228 | 14 |
val eqn_error: thm option -> string -> 'a |
28060 | 15 |
|
16 |
val @@ : 'a * 'a -> 'a list |
|
17 |
val @| : 'a list * 'a -> 'a list |
|
18 |
val str: string -> Pretty.T |
|
19 |
val concat: Pretty.T list -> Pretty.T |
|
20 |
val brackets: Pretty.T list -> Pretty.T |
|
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
21 |
val enclose: string -> string -> Pretty.T list -> Pretty.T |
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
22 |
val enum: string -> string -> string -> Pretty.T list -> Pretty.T |
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
23 |
val enum_default: string -> string -> string -> string -> Pretty.T list -> Pretty.T |
28060 | 24 |
val semicolon: Pretty.T list -> Pretty.T |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
25 |
val doublesemicolon: Pretty.T list -> Pretty.T |
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
26 |
val indent: int -> Pretty.T -> Pretty.T |
34071 | 27 |
val string_of_pretty: int -> Pretty.T -> string |
28 |
val writeln_pretty: int -> Pretty.T -> unit |
|
28060 | 29 |
|
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
30 |
val first_upper: string -> string |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
31 |
val first_lower: string -> string |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
32 |
type var_ctxt |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
33 |
val make_vars: string list -> var_ctxt |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
34 |
val intro_vars: string list -> var_ctxt -> var_ctxt |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
35 |
val lookup_var: var_ctxt -> string -> string |
32913 | 36 |
val intro_base_names: (string -> bool) -> (string -> string) |
37 |
-> string list -> var_ctxt -> var_ctxt |
|
32908 | 38 |
val aux_params: var_ctxt -> iterm list list -> string list |
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
39 |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
40 |
type literals |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
41 |
val Literals: { literal_char: string -> string, literal_string: string -> string, |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
42 |
literal_numeral: int -> string, literal_positive_numeral: int -> string, |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
43 |
literal_naive_numeral: int -> string, |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
44 |
literal_list: Pretty.T list -> Pretty.T, infix_cons: int * string } |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
45 |
-> literals |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
46 |
val literal_char: literals -> string -> string |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
47 |
val literal_string: literals -> string -> string |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
48 |
val literal_numeral: literals -> int -> string |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
49 |
val literal_positive_numeral: literals -> int -> string |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
50 |
val literal_naive_numeral: literals -> int -> string |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
51 |
val literal_list: literals -> Pretty.T list -> Pretty.T |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
52 |
val infix_cons: literals -> int * string |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
53 |
|
28060 | 54 |
type lrx |
55 |
val L: lrx |
|
56 |
val R: lrx |
|
57 |
val X: lrx |
|
58 |
type fixity |
|
59 |
val BR: fixity |
|
60 |
val NOBR: fixity |
|
61 |
val INFX: int * lrx -> fixity |
|
62 |
val APP: fixity |
|
63 |
val brackify: fixity -> Pretty.T list -> Pretty.T |
|
37242
97097e589715
brackify_infix etc.: no break before infix operator -- eases survival in Scala
haftmann
parents:
37146
diff
changeset
|
64 |
val brackify_infix: int * lrx -> fixity -> Pretty.T * Pretty.T * Pretty.T -> Pretty.T |
31665 | 65 |
val brackify_block: fixity -> Pretty.T -> Pretty.T list -> Pretty.T -> Pretty.T |
37638 | 66 |
val applify: string -> string -> ('a -> Pretty.T) -> fixity -> Pretty.T -> 'a list -> Pretty.T |
28060 | 67 |
|
68 |
type tyco_syntax |
|
34152 | 69 |
type simple_const_syntax |
37881 | 70 |
type complex_const_syntax |
28060 | 71 |
type const_syntax |
37881 | 72 |
type activated_complex_const_syntax |
73 |
datatype activated_const_syntax = Plain_const_syntax of int * string |
|
74 |
| Complex_const_syntax of activated_complex_const_syntax |
|
75 |
val requires_args: const_syntax -> int |
|
76 |
val parse_const_syntax: Token.T list -> const_syntax option * Token.T list |
|
77 |
val parse_tyco_syntax: Token.T list -> tyco_syntax option * Token.T list |
|
78 |
val plain_const_syntax: string -> const_syntax |
|
37876 | 79 |
val simple_const_syntax: simple_const_syntax -> const_syntax |
37881 | 80 |
val complex_const_syntax: complex_const_syntax -> const_syntax |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
81 |
val activate_const_syntax: theory -> literals |
37881 | 82 |
-> string -> const_syntax -> Code_Thingol.naming -> activated_const_syntax * Code_Thingol.naming |
35228 | 83 |
val gen_print_app: (thm option -> var_ctxt -> const * iterm list -> Pretty.T list) |
84 |
-> (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T) |
|
37876 | 85 |
-> (string -> activated_const_syntax option) |
35228 | 86 |
-> thm option -> var_ctxt -> fixity -> const * iterm list -> Pretty.T |
87 |
val gen_print_bind: (thm option -> var_ctxt -> fixity -> iterm -> Pretty.T) |
|
88 |
-> thm option -> fixity |
|
31889 | 89 |
-> iterm -> var_ctxt -> Pretty.T * var_ctxt |
28060 | 90 |
|
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
91 |
val mk_name_module: Name.context -> string option -> (string -> string option) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
92 |
-> 'a Graph.T -> string -> string |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
93 |
val dest_name: string -> string * string |
28060 | 94 |
end; |
95 |
||
96 |
structure Code_Printer : CODE_PRINTER = |
|
97 |
struct |
|
98 |
||
99 |
open Code_Thingol; |
|
100 |
||
35228 | 101 |
fun eqn_error (SOME thm) s = error (s ^ ",\nin equation " ^ Display.string_of_thm_without_context thm) |
102 |
| eqn_error NONE s = error s; |
|
28060 | 103 |
|
34071 | 104 |
(** assembling and printing text pieces **) |
28060 | 105 |
|
106 |
infixr 5 @@; |
|
107 |
infixr 5 @|; |
|
108 |
fun x @@ y = [x, y]; |
|
109 |
fun xs @| y = xs @ [y]; |
|
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
110 |
val str = Print_Mode.setmp [] Pretty.str; |
28060 | 111 |
val concat = Pretty.block o Pretty.breaks; |
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
112 |
fun enclose l r = Print_Mode.setmp [] (Pretty.enclose l r); |
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
113 |
val brackets = enclose "(" ")" o Pretty.breaks; |
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
114 |
fun enum sep l r = Print_Mode.setmp [] (Pretty.enum sep l r); |
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
115 |
fun enum_default default sep l r [] = str default |
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
116 |
| enum_default default sep l r xs = enum sep l r xs; |
28060 | 117 |
fun semicolon ps = Pretty.block [concat ps, str ";"]; |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
118 |
fun doublesemicolon ps = Pretty.block [concat ps, str ";;"]; |
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
119 |
fun indent i = Print_Mode.setmp [] (Pretty.indent i); |
28060 | 120 |
|
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
121 |
fun string_of_pretty width p = Print_Mode.setmp [] (Pretty.string_of_margin width) p ^ "\n"; |
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
122 |
fun writeln_pretty width p = writeln (Print_Mode.setmp [] (Pretty.string_of_margin width) p); |
34071 | 123 |
|
28060 | 124 |
|
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
125 |
(** names and variable name contexts **) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
126 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
127 |
type var_ctxt = string Symtab.table * Name.context; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
128 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
129 |
fun make_vars names = (fold (fn name => Symtab.update_new (name, name)) names Symtab.empty, |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
130 |
Name.make_context names); |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
131 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
132 |
fun intro_vars names (namemap, namectxt) = |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
133 |
let |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
134 |
val (names', namectxt') = Name.variants names namectxt; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
135 |
val namemap' = fold2 (curry Symtab.update) names names' namemap; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
136 |
in (namemap', namectxt') end; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
137 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
138 |
fun lookup_var (namemap, _) name = case Symtab.lookup namemap name |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
139 |
of SOME name' => name' |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
140 |
| NONE => error ("Invalid name in context: " ^ quote name); |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
141 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
142 |
val first_upper = implode o nth_map 0 Symbol.to_ascii_upper o explode; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
143 |
val first_lower = implode o nth_map 0 Symbol.to_ascii_lower o explode; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
144 |
|
32908 | 145 |
fun aux_params vars lhss = |
146 |
let |
|
147 |
fun fish_param _ (w as SOME _) = w |
|
148 |
| fish_param (IVar (SOME v)) NONE = SOME v |
|
149 |
| fish_param _ NONE = NONE; |
|
150 |
fun fillup_param _ (_, SOME v) = v |
|
151 |
| fillup_param x (i, NONE) = x ^ string_of_int i; |
|
152 |
val fished1 = fold (map2 fish_param) lhss (replicate (length (hd lhss)) NONE); |
|
153 |
val x = Name.variant (map_filter I fished1) "x"; |
|
154 |
val fished2 = map_index (fillup_param x) fished1; |
|
155 |
val (fished3, _) = Name.variants fished2 Name.context; |
|
156 |
val vars' = intro_vars fished3 vars; |
|
157 |
in map (lookup_var vars') fished3 end; |
|
158 |
||
32913 | 159 |
fun intro_base_names no_syntax deresolve names = names |
160 |
|> map_filter (fn name => if no_syntax name then |
|
161 |
let val name' = deresolve name in |
|
162 |
if Long_Name.is_qualified name' then NONE else SOME name' |
|
163 |
end else NONE) |
|
164 |
|> intro_vars; |
|
165 |
||
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
166 |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
167 |
(** pretty literals **) |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
168 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
169 |
datatype literals = Literals of { |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
170 |
literal_char: string -> string, |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
171 |
literal_string: string -> string, |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
172 |
literal_numeral: int -> string, |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
173 |
literal_positive_numeral: int -> string, |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
174 |
literal_naive_numeral: int -> string, |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
175 |
literal_list: Pretty.T list -> Pretty.T, |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
176 |
infix_cons: int * string |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
177 |
}; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
178 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
179 |
fun dest_Literals (Literals lits) = lits; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
180 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
181 |
val literal_char = #literal_char o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
182 |
val literal_string = #literal_string o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
183 |
val literal_numeral = #literal_numeral o dest_Literals; |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
184 |
val literal_positive_numeral = #literal_positive_numeral o dest_Literals; |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
185 |
val literal_naive_numeral = #literal_naive_numeral o dest_Literals; |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
186 |
val literal_list = #literal_list o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
187 |
val infix_cons = #infix_cons o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
188 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
189 |
|
28060 | 190 |
(** syntax printer **) |
191 |
||
192 |
(* binding priorities *) |
|
193 |
||
194 |
datatype lrx = L | R | X; |
|
195 |
||
196 |
datatype fixity = |
|
197 |
BR |
|
198 |
| NOBR |
|
199 |
| INFX of (int * lrx); |
|
200 |
||
201 |
val APP = INFX (~1, L); |
|
202 |
||
203 |
fun fixity_lrx L L = false |
|
204 |
| fixity_lrx R R = false |
|
205 |
| fixity_lrx _ _ = true; |
|
206 |
||
207 |
fun fixity NOBR _ = false |
|
208 |
| fixity _ NOBR = false |
|
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
209 |
| fixity (INFX (pr, lr)) (INFX (print_ctxt, lr_ctxt)) = |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
210 |
pr < print_ctxt |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
211 |
orelse pr = print_ctxt |
28060 | 212 |
andalso fixity_lrx lr lr_ctxt |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
213 |
orelse print_ctxt = ~1 |
28060 | 214 |
| fixity BR (INFX _) = false |
215 |
| fixity _ _ = true; |
|
216 |
||
217 |
fun gen_brackify _ [p] = p |
|
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
218 |
| gen_brackify true (ps as _::_) = enclose "(" ")" ps |
28060 | 219 |
| gen_brackify false (ps as _::_) = Pretty.block ps; |
220 |
||
221 |
fun brackify fxy_ctxt = |
|
222 |
gen_brackify (fixity BR fxy_ctxt) o Pretty.breaks; |
|
223 |
||
37242
97097e589715
brackify_infix etc.: no break before infix operator -- eases survival in Scala
haftmann
parents:
37146
diff
changeset
|
224 |
fun brackify_infix infx fxy_ctxt (l, m, r) = |
97097e589715
brackify_infix etc.: no break before infix operator -- eases survival in Scala
haftmann
parents:
37146
diff
changeset
|
225 |
(if fixity (INFX infx) fxy_ctxt then enclose "(" ")" else Pretty.block) |
97097e589715
brackify_infix etc.: no break before infix operator -- eases survival in Scala
haftmann
parents:
37146
diff
changeset
|
226 |
([l, str " ", m, Pretty.brk 1, r]); |
28060 | 227 |
|
31665 | 228 |
fun brackify_block fxy_ctxt p1 ps p2 = |
229 |
let val p = Pretty.block_enclose (p1, p2) ps |
|
230 |
in if fixity BR fxy_ctxt |
|
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
231 |
then enclose "(" ")" [p] |
31665 | 232 |
else p |
233 |
end; |
|
234 |
||
37638 | 235 |
fun applify opn cls f fxy_ctxt p [] = p |
236 |
| applify opn cls f fxy_ctxt p ps = |
|
34247 | 237 |
(if (fixity BR fxy_ctxt) then enclose "(" ")" else Pretty.block) |
37638 | 238 |
(p @@ enum "," opn cls (map f ps)); |
34247 | 239 |
|
28060 | 240 |
|
241 |
(* generic syntax *) |
|
242 |
||
243 |
type tyco_syntax = int * ((fixity -> itype -> Pretty.T) |
|
244 |
-> fixity -> itype list -> Pretty.T); |
|
34152 | 245 |
|
246 |
type simple_const_syntax = int * ((fixity -> iterm -> Pretty.T) |
|
247 |
-> fixity -> (iterm * itype) list -> Pretty.T); |
|
37881 | 248 |
|
249 |
type complex_const_syntax = int * (string list * (literals -> string list |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
250 |
-> (var_ctxt -> fixity -> iterm -> Pretty.T) |
35228 | 251 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T)); |
37881 | 252 |
|
253 |
datatype const_syntax = plain_const_syntax of string |
|
254 |
| complex_const_syntax of complex_const_syntax; |
|
255 |
||
256 |
fun requires_args (plain_const_syntax _) = 0 |
|
257 |
| requires_args (complex_const_syntax (k, _)) = k; |
|
28060 | 258 |
|
34176 | 259 |
fun simple_const_syntax syn = |
37881 | 260 |
complex_const_syntax (apsnd (fn f => ([], (fn _ => fn _ => fn print => fn _ => fn vars => f (print vars)))) syn); |
28060 | 261 |
|
37881 | 262 |
type activated_complex_const_syntax = int * ((var_ctxt -> fixity -> iterm -> Pretty.T) |
263 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T) |
|
264 |
||
265 |
datatype activated_const_syntax = Plain_const_syntax of int * string |
|
266 |
| Complex_const_syntax of activated_complex_const_syntax; |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
267 |
|
37881 | 268 |
fun activate_const_syntax thy literals c (plain_const_syntax s) naming = |
269 |
(Plain_const_syntax (Code.args_number thy c, s), naming) |
|
270 |
| activate_const_syntax thy literals c (complex_const_syntax (n, (cs, f))) naming = |
|
271 |
fold_map (Code_Thingol.ensure_declared_const thy) cs naming |
|
272 |
|-> (fn cs' => pair (Complex_const_syntax (n, f literals cs'))); |
|
273 |
||
274 |
fun gen_print_app print_app_expr print_term syntax_const some_thm vars fxy (app as ((c, (_, function_typs)), ts)) = |
|
28060 | 275 |
case syntax_const c |
37881 | 276 |
of NONE => brackify fxy (print_app_expr some_thm vars app) |
277 |
| SOME (Plain_const_syntax (_, s)) => brackify fxy (str s :: map (print_term some_thm vars BR) ts) |
|
278 |
| SOME (Complex_const_syntax (k, print)) => |
|
28060 | 279 |
let |
37881 | 280 |
fun print' fxy ts = print (print_term some_thm) some_thm vars fxy (ts ~~ take k function_typs); |
28060 | 281 |
in if k = length ts |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
282 |
then print' fxy ts |
28060 | 283 |
else if k < length ts |
284 |
then case chop k ts of (ts1, ts2) => |
|
37881 | 285 |
brackify fxy (print' APP ts1 :: map (print_term some_thm vars BR) ts2) |
286 |
else print_term some_thm vars fxy (Code_Thingol.eta_expand k app) |
|
28060 | 287 |
end; |
288 |
||
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
289 |
fun gen_print_bind print_term thm (fxy : fixity) pat vars = |
28060 | 290 |
let |
31889 | 291 |
val vs = Code_Thingol.fold_varnames (insert (op =)) pat []; |
31874 | 292 |
val vars' = intro_vars vs vars; |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
293 |
in (print_term thm vars' fxy pat, vars') end; |
28060 | 294 |
|
295 |
||
296 |
(* mixfix syntax *) |
|
297 |
||
298 |
datatype 'a mixfix = |
|
299 |
Arg of fixity |
|
37881 | 300 |
| String of string |
301 |
| Break; |
|
28060 | 302 |
|
303 |
fun mk_mixfix prep_arg (fixity_this, mfx) = |
|
304 |
let |
|
305 |
fun is_arg (Arg _) = true |
|
306 |
| is_arg _ = false; |
|
307 |
val i = (length o filter is_arg) mfx; |
|
308 |
fun fillin _ [] [] = |
|
309 |
[] |
|
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
310 |
| fillin print (Arg fxy :: mfx) (a :: args) = |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
311 |
(print fxy o prep_arg) a :: fillin print mfx args |
37881 | 312 |
| fillin print (String s :: mfx) args = |
313 |
str s :: fillin print mfx args |
|
314 |
| fillin print (Break :: mfx) args = |
|
315 |
Pretty.brk 1 :: fillin print mfx args; |
|
28060 | 316 |
in |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
317 |
(i, fn print => fn fixity_ctxt => fn args => |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
318 |
gen_brackify (fixity fixity_this fixity_ctxt) (fillin print mfx args)) |
28060 | 319 |
end; |
320 |
||
321 |
fun parse_infix prep_arg (x, i) s = |
|
322 |
let |
|
323 |
val l = case x of L => INFX (i, L) | _ => INFX (i, X); |
|
324 |
val r = case x of R => INFX (i, R) | _ => INFX (i, X); |
|
325 |
in |
|
37881 | 326 |
mk_mixfix prep_arg (INFX (i, x), [Arg l, String " ", String s, Break, Arg r]) |
28060 | 327 |
end; |
328 |
||
37881 | 329 |
fun parse_mixfix mk_plain mk_complex prep_arg s = |
28060 | 330 |
let |
331 |
val sym_any = Scan.one Symbol.is_regular; |
|
332 |
val parse = Scan.optional ($$ "!" >> K true) false -- Scan.repeat ( |
|
333 |
($$ "(" -- $$ "_" -- $$ ")" >> K (Arg NOBR)) |
|
334 |
|| ($$ "_" >> K (Arg BR)) |
|
37881 | 335 |
|| ($$ "/" |-- Scan.repeat ($$ " ") >> (K Break)) |
28060 | 336 |
|| (Scan.repeat1 |
337 |
( $$ "'" |-- sym_any |
|
338 |
|| Scan.unless ($$ "_" || $$ "/" || $$ "(" |-- $$ "_" |-- $$ ")") |
|
37881 | 339 |
sym_any) >> (String o implode))); |
28060 | 340 |
in case Scan.finite Symbol.stopper parse (Symbol.explode s) |
37881 | 341 |
of ((false, [String s]), []) => mk_plain s |
342 |
| ((_, p as [_]), []) => mk_complex (mk_mixfix prep_arg (NOBR, p)) |
|
343 |
| ((b, p as _ :: _ :: _), []) => mk_complex (mk_mixfix prep_arg (if b then NOBR else BR, p)) |
|
28060 | 344 |
| _ => Scan.!! |
345 |
(the_default ("malformed mixfix annotation: " ^ quote s) o snd) Scan.fail () |
|
346 |
end; |
|
347 |
||
348 |
val (infixK, infixlK, infixrK) = ("infix", "infixl", "infixr"); |
|
349 |
||
37881 | 350 |
fun parse_syntax mk_plain mk_complex prep_arg = |
351 |
Scan.option ( |
|
36960
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
352 |
((Parse.$$$ infixK >> K X) |
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
353 |
|| (Parse.$$$ infixlK >> K L) |
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
354 |
|| (Parse.$$$ infixrK >> K R)) |
37881 | 355 |
-- Parse.nat -- Parse.string |
356 |
>> (fn ((x, i), s) => mk_complex (parse_infix prep_arg (x, i) s)) |
|
357 |
|| Parse.string >> (fn s => (parse_mixfix mk_plain mk_complex prep_arg s))); |
|
28060 | 358 |
|
36960
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
359 |
val _ = List.app Keyword.keyword [infixK, infixlK, infixrK]; |
28060 | 360 |
|
37881 | 361 |
val parse_tyco_syntax = parse_syntax (fn s => (0, (K o K o K o str) s)) I I; |
362 |
||
363 |
val parse_const_syntax = parse_syntax plain_const_syntax simple_const_syntax fst; |
|
364 |
||
28064 | 365 |
|
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
366 |
(** module name spaces **) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
367 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
368 |
val dest_name = |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
369 |
apfst Long_Name.implode o split_last o fst o split_last o Long_Name.explode; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
370 |
|
32924
d2e9b2dab760
dropped Code_Printer prefix where feasible; fixed whitespace issues; more coherent terminology
haftmann
parents:
32913
diff
changeset
|
371 |
fun mk_name_module reserved module_prefix module_alias program = |
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
372 |
let |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
373 |
fun mk_alias name = case module_alias name |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
374 |
of SOME name' => name' |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
375 |
| NONE => name |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
376 |
|> Long_Name.explode |
32924
d2e9b2dab760
dropped Code_Printer prefix where feasible; fixed whitespace issues; more coherent terminology
haftmann
parents:
32913
diff
changeset
|
377 |
|> map (fn name => (the_single o fst) (Name.variants [name] reserved)) |
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
378 |
|> Long_Name.implode; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
379 |
fun mk_prefix name = case module_prefix |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
380 |
of SOME module_prefix => Long_Name.append module_prefix name |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
381 |
| NONE => name; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
382 |
val tab = |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
383 |
Symtab.empty |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
384 |
|> Graph.fold ((fn name => Symtab.default (name, (mk_alias #> mk_prefix) name)) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
385 |
o fst o dest_name o fst) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
386 |
program |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
387 |
in the o Symtab.lookup tab end; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
388 |
|
28060 | 389 |
end; (*struct*) |