author | haftmann |
Thu, 02 Sep 2010 13:58:16 +0200 | |
changeset 39056 | fa197571676b |
parent 39055 | 81e0368812ad |
child 39057 | c6d146ed07ae |
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"; |
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
108 |
val _ = Output.add_mode code_presentationN; |
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]; |
|
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
118 |
val str = Print_Mode.setmp [] Pretty.str; |
28060 | 119 |
val concat = Pretty.block o Pretty.breaks; |
38778 | 120 |
val commas = Print_Mode.setmp [] Pretty.commas; |
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 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; |
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
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 ";;"]; |
37146
f652333bbf8e
renamed structure PrintMode to Print_Mode, keeping the old name as legacy alias for some time;
wenzelm
parents:
36960
diff
changeset
|
128 |
fun indent i = Print_Mode.setmp [] (Pretty.indent i); |
28060 | 129 |
|
39034
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
130 |
val stmt_nameN = "stmt_name"; |
39056 | 131 |
fun markup_stmt name = Pretty.mark (code_presentationN, [(stmt_nameN, name)]); |
132 |
fun filter_presentation presentation_names selected (XML.Elem ((name, attrs), xs)) = |
|
133 |
implode (map (filter_presentation presentation_names |
|
134 |
(selected orelse (name = code_presentationN |
|
135 |
andalso member (op =) presentation_names (the (Properties.get attrs stmt_nameN))))) xs) |
|
136 |
| filter_presentation presentation_names selected (XML.Text s) = |
|
39034
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
137 |
if selected then s else ""; |
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
138 |
|
39056 | 139 |
fun format presentation_names width p = |
140 |
if null presentation_names then Print_Mode.setmp [] (Pretty.string_of_margin width) p ^ "\n" |
|
141 |
else Print_Mode.setmp [code_presentationN] (Pretty.string_of_margin width) p |
|
39034
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
142 |
|> YXML.parse_body |
39056 | 143 |
|> map (filter_presentation presentation_names false) |
39034
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
144 |
|> implode |
ebeb48fd653b
formal framework for presentation of selected statements
haftmann
parents:
38923
diff
changeset
|
145 |
|> suffix "\n" |
34071 | 146 |
|
28060 | 147 |
|
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
148 |
(** names and variable name contexts **) |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
149 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
150 |
type var_ctxt = string Symtab.table * Name.context; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
151 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
152 |
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
|
153 |
Name.make_context names); |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
154 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
155 |
fun intro_vars names (namemap, namectxt) = |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
156 |
let |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
157 |
val (names', namectxt') = Name.variants names namectxt; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
158 |
val namemap' = fold2 (curry Symtab.update) names names' namemap; |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
159 |
in (namemap', namectxt') end; |
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 |
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
|
162 |
of SOME name' => name' |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
163 |
| NONE => error ("Invalid name in context: " ^ quote name); |
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
164 |
|
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
165 |
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
|
166 |
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
|
167 |
|
32908 | 168 |
fun aux_params vars lhss = |
169 |
let |
|
170 |
fun fish_param _ (w as SOME _) = w |
|
171 |
| fish_param (IVar (SOME v)) NONE = SOME v |
|
172 |
| fish_param _ NONE = NONE; |
|
173 |
fun fillup_param _ (_, SOME v) = v |
|
174 |
| fillup_param x (i, NONE) = x ^ string_of_int i; |
|
175 |
val fished1 = fold (map2 fish_param) lhss (replicate (length (hd lhss)) NONE); |
|
176 |
val x = Name.variant (map_filter I fished1) "x"; |
|
177 |
val fished2 = map_index (fillup_param x) fished1; |
|
178 |
val (fished3, _) = Name.variants fished2 Name.context; |
|
179 |
val vars' = intro_vars fished3 vars; |
|
180 |
in map (lookup_var vars') fished3 end; |
|
181 |
||
32913 | 182 |
fun intro_base_names no_syntax deresolve names = names |
183 |
|> map_filter (fn name => if no_syntax name then |
|
184 |
let val name' = deresolve name in |
|
185 |
if Long_Name.is_qualified name' then NONE else SOME name' |
|
186 |
end else NONE) |
|
187 |
|> intro_vars; |
|
188 |
||
30648
17365ef082f3
clarified relationship of modules Code_Name and Code_Printer
haftmann
parents:
30161
diff
changeset
|
189 |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
190 |
(** pretty literals **) |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
191 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
192 |
datatype literals = Literals of { |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
193 |
literal_char: string -> string, |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
194 |
literal_string: string -> string, |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
195 |
literal_numeral: int -> string, |
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
196 |
literal_positive_numeral: int -> string, |
37958
9728342bcd56
another refinement chapter in the neverending numeral story
haftmann
parents:
37899
diff
changeset
|
197 |
literal_alternative_numeral: int -> string, |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
198 |
literal_naive_numeral: int -> string, |
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
199 |
literal_list: Pretty.T list -> Pretty.T, |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
200 |
infix_cons: int * string |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
201 |
}; |
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 |
fun dest_Literals (Literals lits) = lits; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
204 |
|
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
205 |
val literal_char = #literal_char o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
206 |
val literal_string = #literal_string o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
207 |
val literal_numeral = #literal_numeral o dest_Literals; |
34944
970e1466028d
code literals: distinguish numeral classes by different entries
haftmann
parents:
34247
diff
changeset
|
208 |
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
|
209 |
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
|
210 |
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
|
211 |
val literal_list = #literal_list o dest_Literals; |
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
212 |
val infix_cons = #infix_cons o dest_Literals; |
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 |
|
28060 | 215 |
(** syntax printer **) |
216 |
||
217 |
(* binding priorities *) |
|
218 |
||
219 |
datatype lrx = L | R | X; |
|
220 |
||
221 |
datatype fixity = |
|
222 |
BR |
|
223 |
| NOBR |
|
224 |
| INFX of (int * lrx); |
|
225 |
||
226 |
val APP = INFX (~1, L); |
|
227 |
||
228 |
fun fixity_lrx L L = false |
|
229 |
| fixity_lrx R R = false |
|
230 |
| fixity_lrx _ _ = true; |
|
231 |
||
232 |
fun fixity NOBR _ = false |
|
233 |
| fixity _ NOBR = false |
|
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
234 |
| fixity (INFX (pr, lr)) (INFX (print_ctxt, lr_ctxt)) = |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
235 |
pr < print_ctxt |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
236 |
orelse pr = print_ctxt |
28060 | 237 |
andalso fixity_lrx lr lr_ctxt |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
238 |
orelse print_ctxt = ~1 |
28060 | 239 |
| fixity BR (INFX _) = false |
240 |
| fixity _ _ = true; |
|
241 |
||
242 |
fun gen_brackify _ [p] = p |
|
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
243 |
| gen_brackify true (ps as _::_) = enclose "(" ")" ps |
28060 | 244 |
| gen_brackify false (ps as _::_) = Pretty.block ps; |
245 |
||
246 |
fun brackify fxy_ctxt = |
|
247 |
gen_brackify (fixity BR fxy_ctxt) o Pretty.breaks; |
|
248 |
||
37242
97097e589715
brackify_infix etc.: no break before infix operator -- eases survival in Scala
haftmann
parents:
37146
diff
changeset
|
249 |
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
|
250 |
(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
|
251 |
([l, str " ", m, Pretty.brk 1, r]); |
28060 | 252 |
|
31665 | 253 |
fun brackify_block fxy_ctxt p1 ps p2 = |
254 |
let val p = Pretty.block_enclose (p1, p2) ps |
|
255 |
in if fixity BR fxy_ctxt |
|
34178
a78b8d5b91cb
take care for destructive print mode properly using dedicated pretty builders
haftmann
parents:
34176
diff
changeset
|
256 |
then enclose "(" ")" [p] |
31665 | 257 |
else p |
258 |
end; |
|
259 |
||
37638 | 260 |
fun applify opn cls f fxy_ctxt p [] = p |
261 |
| applify opn cls f fxy_ctxt p ps = |
|
34247 | 262 |
(if (fixity BR fxy_ctxt) then enclose "(" ")" else Pretty.block) |
37638 | 263 |
(p @@ enum "," opn cls (map f ps)); |
34247 | 264 |
|
38922 | 265 |
fun tuplify _ _ [] = NONE |
266 |
| tuplify print fxy [x] = SOME (print fxy x) |
|
267 |
| tuplify print _ xs = SOME (enum "," "(" ")" (map (print NOBR) xs)); |
|
268 |
||
28060 | 269 |
|
270 |
(* generic syntax *) |
|
271 |
||
272 |
type tyco_syntax = int * ((fixity -> itype -> Pretty.T) |
|
273 |
-> fixity -> itype list -> Pretty.T); |
|
34152 | 274 |
|
275 |
type simple_const_syntax = int * ((fixity -> iterm -> Pretty.T) |
|
276 |
-> fixity -> (iterm * itype) list -> Pretty.T); |
|
37881 | 277 |
|
278 |
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
|
279 |
-> (var_ctxt -> fixity -> iterm -> Pretty.T) |
35228 | 280 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T)); |
37881 | 281 |
|
282 |
datatype const_syntax = plain_const_syntax of string |
|
283 |
| complex_const_syntax of complex_const_syntax; |
|
284 |
||
285 |
fun requires_args (plain_const_syntax _) = 0 |
|
286 |
| requires_args (complex_const_syntax (k, _)) = k; |
|
28060 | 287 |
|
34176 | 288 |
fun simple_const_syntax syn = |
37881 | 289 |
complex_const_syntax (apsnd (fn f => ([], (fn _ => fn _ => fn print => fn _ => fn vars => f (print vars)))) syn); |
28060 | 290 |
|
37881 | 291 |
type activated_complex_const_syntax = int * ((var_ctxt -> fixity -> iterm -> Pretty.T) |
292 |
-> thm option -> var_ctxt -> fixity -> (iterm * itype) list -> Pretty.T) |
|
293 |
||
294 |
datatype activated_const_syntax = Plain_const_syntax of int * string |
|
295 |
| Complex_const_syntax of activated_complex_const_syntax; |
|
31056
01ac77eb660b
robustifed infrastructure for complex term syntax during code generation
haftmann
parents:
30648
diff
changeset
|
296 |
|
37881 | 297 |
fun activate_const_syntax thy literals c (plain_const_syntax s) naming = |
298 |
(Plain_const_syntax (Code.args_number thy c, s), naming) |
|
299 |
| activate_const_syntax thy literals c (complex_const_syntax (n, (cs, f))) naming = |
|
300 |
fold_map (Code_Thingol.ensure_declared_const thy) cs naming |
|
301 |
|-> (fn cs' => pair (Complex_const_syntax (n, f literals cs'))); |
|
302 |
||
38923 | 303 |
fun gen_print_app print_app_expr print_term const_syntax some_thm vars fxy (app as ((c, (_, function_typs)), ts)) = |
304 |
case const_syntax c |
|
37881 | 305 |
of NONE => brackify fxy (print_app_expr some_thm vars app) |
306 |
| SOME (Plain_const_syntax (_, s)) => brackify fxy (str s :: map (print_term some_thm vars BR) ts) |
|
307 |
| SOME (Complex_const_syntax (k, print)) => |
|
28060 | 308 |
let |
37881 | 309 |
fun print' fxy ts = print (print_term some_thm) some_thm vars fxy (ts ~~ take k function_typs); |
28060 | 310 |
in if k = length ts |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
311 |
then print' fxy ts |
28060 | 312 |
else if k < length ts |
313 |
then case chop k ts of (ts1, ts2) => |
|
37881 | 314 |
brackify fxy (print' APP ts1 :: map (print_term some_thm vars BR) ts2) |
315 |
else print_term some_thm vars fxy (Code_Thingol.eta_expand k app) |
|
28060 | 316 |
end; |
317 |
||
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
318 |
fun gen_print_bind print_term thm (fxy : fixity) pat vars = |
28060 | 319 |
let |
31889 | 320 |
val vs = Code_Thingol.fold_varnames (insert (op =)) pat []; |
31874 | 321 |
val vars' = intro_vars vs vars; |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
322 |
in (print_term thm vars' fxy pat, vars') end; |
28060 | 323 |
|
324 |
||
325 |
(* mixfix syntax *) |
|
326 |
||
327 |
datatype 'a mixfix = |
|
328 |
Arg of fixity |
|
37881 | 329 |
| String of string |
330 |
| Break; |
|
28060 | 331 |
|
332 |
fun mk_mixfix prep_arg (fixity_this, mfx) = |
|
333 |
let |
|
334 |
fun is_arg (Arg _) = true |
|
335 |
| is_arg _ = false; |
|
336 |
val i = (length o filter is_arg) mfx; |
|
337 |
fun fillin _ [] [] = |
|
338 |
[] |
|
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
339 |
| fillin print (Arg fxy :: mfx) (a :: args) = |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
340 |
(print fxy o prep_arg) a :: fillin print mfx args |
37881 | 341 |
| fillin print (String s :: mfx) args = |
342 |
str s :: fillin print mfx args |
|
343 |
| fillin print (Break :: mfx) args = |
|
344 |
Pretty.brk 1 :: fillin print mfx args; |
|
28060 | 345 |
in |
33989
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
346 |
(i, fn print => fn fixity_ctxt => fn args => |
cb136b5f6050
more speaking function names for Code_Printer; added doublesemicolon
haftmann
parents:
32924
diff
changeset
|
347 |
gen_brackify (fixity fixity_this fixity_ctxt) (fillin print mfx args)) |
28060 | 348 |
end; |
349 |
||
350 |
fun parse_infix prep_arg (x, i) s = |
|
351 |
let |
|
352 |
val l = case x of L => INFX (i, L) | _ => INFX (i, X); |
|
353 |
val r = case x of R => INFX (i, R) | _ => INFX (i, X); |
|
354 |
in |
|
37881 | 355 |
mk_mixfix prep_arg (INFX (i, x), [Arg l, String " ", String s, Break, Arg r]) |
28060 | 356 |
end; |
357 |
||
37881 | 358 |
fun parse_mixfix mk_plain mk_complex prep_arg s = |
28060 | 359 |
let |
360 |
val sym_any = Scan.one Symbol.is_regular; |
|
361 |
val parse = Scan.optional ($$ "!" >> K true) false -- Scan.repeat ( |
|
362 |
($$ "(" -- $$ "_" -- $$ ")" >> K (Arg NOBR)) |
|
363 |
|| ($$ "_" >> K (Arg BR)) |
|
37881 | 364 |
|| ($$ "/" |-- Scan.repeat ($$ " ") >> (K Break)) |
28060 | 365 |
|| (Scan.repeat1 |
366 |
( $$ "'" |-- sym_any |
|
367 |
|| Scan.unless ($$ "_" || $$ "/" || $$ "(" |-- $$ "_" |-- $$ ")") |
|
37881 | 368 |
sym_any) >> (String o implode))); |
28060 | 369 |
in case Scan.finite Symbol.stopper parse (Symbol.explode s) |
37881 | 370 |
of ((false, [String s]), []) => mk_plain s |
371 |
| ((_, p as [_]), []) => mk_complex (mk_mixfix prep_arg (NOBR, p)) |
|
372 |
| ((b, p as _ :: _ :: _), []) => mk_complex (mk_mixfix prep_arg (if b then NOBR else BR, p)) |
|
28060 | 373 |
| _ => Scan.!! |
374 |
(the_default ("malformed mixfix annotation: " ^ quote s) o snd) Scan.fail () |
|
375 |
end; |
|
376 |
||
377 |
val (infixK, infixlK, infixrK) = ("infix", "infixl", "infixr"); |
|
378 |
||
37881 | 379 |
fun parse_syntax mk_plain mk_complex prep_arg = |
380 |
Scan.option ( |
|
36960
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
381 |
((Parse.$$$ infixK >> K X) |
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
382 |
|| (Parse.$$$ infixlK >> K L) |
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
383 |
|| (Parse.$$$ infixrK >> K R)) |
37881 | 384 |
-- Parse.nat -- Parse.string |
385 |
>> (fn ((x, i), s) => mk_complex (parse_infix prep_arg (x, i) s)) |
|
386 |
|| Parse.string >> (fn s => (parse_mixfix mk_plain mk_complex prep_arg s))); |
|
28060 | 387 |
|
36960
01594f816e3a
prefer structure Keyword, Parse, Parse_Spec, Outer_Syntax;
wenzelm
parents:
36959
diff
changeset
|
388 |
val _ = List.app Keyword.keyword [infixK, infixlK, infixrK]; |
28060 | 389 |
|
37899 | 390 |
fun parse_tyco_syntax x = parse_syntax (fn s => (0, (K o K o K o str) s)) I I x; |
37881 | 391 |
|
392 |
val parse_const_syntax = parse_syntax plain_const_syntax simple_const_syntax fst; |
|
393 |
||
28060 | 394 |
end; (*struct*) |