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