author | wenzelm |
Thu, 22 Nov 2007 14:51:34 +0100 | |
changeset 25456 | 6f79698f294d |
parent 25042 | a33b78d63114 |
child 25485 | 33840a854e63 |
permissions | -rw-r--r-- |
24219 | 1 |
(* Title: Tools/code/code_thingol.ML |
2 |
ID: $Id$ |
|
3 |
Author: Florian Haftmann, TU Muenchen |
|
4 |
||
5 |
Intermediate language ("Thin-gol") representing executable code. |
|
24918 | 6 |
Representation and translation. |
24219 | 7 |
*) |
8 |
||
9 |
infix 8 `%%; |
|
10 |
infixr 6 `->; |
|
11 |
infixr 6 `-->; |
|
12 |
infix 4 `$; |
|
13 |
infix 4 `$$; |
|
14 |
infixr 3 `|->; |
|
15 |
infixr 3 `|-->; |
|
16 |
||
17 |
signature BASIC_CODE_THINGOL = |
|
18 |
sig |
|
19 |
type vname = string; |
|
20 |
datatype dict = |
|
21 |
DictConst of string * dict list list |
|
22 |
| DictVar of string list * (vname * (int * int)); |
|
23 |
datatype itype = |
|
24 |
`%% of string * itype list |
|
25 |
| ITyVar of vname; |
|
24591 | 26 |
type const = string * (dict list list * itype list (*types of arguments*)) |
24219 | 27 |
datatype iterm = |
24591 | 28 |
IConst of const |
24219 | 29 |
| IVar of vname |
30 |
| `$ of iterm * iterm |
|
31 |
| `|-> of (vname * itype) * iterm |
|
32 |
| ICase of ((iterm * itype) * (iterm * iterm) list) * iterm; |
|
33 |
(*((term, type), [(selector pattern, body term )]), primitive term)*) |
|
34 |
val `-> : itype * itype -> itype; |
|
35 |
val `--> : itype list * itype -> itype; |
|
36 |
val `$$ : iterm * iterm list -> iterm; |
|
37 |
val `|--> : (vname * itype) list * iterm -> iterm; |
|
38 |
type typscheme = (vname * sort) list * itype; |
|
39 |
end; |
|
40 |
||
41 |
signature CODE_THINGOL = |
|
42 |
sig |
|
43 |
include BASIC_CODE_THINGOL; |
|
44 |
val unfoldl: ('a -> ('a * 'b) option) -> 'a -> 'a * 'b list; |
|
45 |
val unfoldr: ('a -> ('b * 'a) option) -> 'a -> 'b list * 'a; |
|
46 |
val unfold_fun: itype -> itype list * itype; |
|
47 |
val unfold_app: iterm -> iterm * iterm list; |
|
48 |
val split_abs: iterm -> (((vname * iterm option) * itype) * iterm) option; |
|
49 |
val unfold_abs: iterm -> ((vname * iterm option) * itype) list * iterm; |
|
50 |
val split_let: iterm -> (((iterm * itype) * iterm) * iterm) option; |
|
51 |
val unfold_let: iterm -> ((iterm * itype) * iterm) list * iterm; |
|
52 |
val unfold_const_app: iterm -> |
|
53 |
((string * (dict list list * itype list)) * iterm list) option; |
|
54 |
val collapse_let: ((vname * itype) * iterm) * iterm |
|
55 |
-> (iterm * itype) * (iterm * iterm) list; |
|
56 |
val eta_expand: (string * (dict list list * itype list)) * iterm list -> int -> iterm; |
|
24662
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
57 |
val contains_dictvar: iterm -> bool; |
24219 | 58 |
val fold_constnames: (string -> 'a -> 'a) -> iterm -> 'a -> 'a; |
59 |
val fold_varnames: (string -> 'a -> 'a) -> iterm -> 'a -> 'a; |
|
60 |
val fold_unbound_varnames: (string -> 'a -> 'a) -> iterm -> 'a -> 'a; |
|
61 |
||
24918 | 62 |
datatype stmt = |
24219 | 63 |
Bot |
24591 | 64 |
| Fun of typscheme * ((iterm list * iterm) * thm) list |
24219 | 65 |
| Datatype of (vname * sort) list * (string * itype list) list |
66 |
| Datatypecons of string |
|
24811 | 67 |
| Class of vname * ((class * string) list * (string * itype) list) |
24219 | 68 |
| Classrel of class * class |
24837 | 69 |
| Classparam of class |
24219 | 70 |
| Classinst of (class * (string * (vname * sort) list)) |
71 |
* ((class * (string * (string * dict list list))) list |
|
24591 | 72 |
* ((string * const) * thm) list); |
24918 | 73 |
type code = stmt Graph.T; |
24219 | 74 |
val empty_code: code; |
75 |
val merge_code: code * code -> code; |
|
76 |
val project_code: bool (*delete empty funs*) |
|
77 |
-> string list (*hidden*) -> string list option (*selected*) |
|
78 |
-> code -> code; |
|
79 |
val empty_funs: code -> string list; |
|
80 |
val is_cons: code -> string -> bool; |
|
81 |
||
24918 | 82 |
type transact; |
83 |
val ensure_const: theory -> ((sort -> sort) * Sorts.algebra) * Consts.T |
|
84 |
-> CodeFuncgr.T -> string -> transact -> string * transact; |
|
85 |
val ensure_value: theory -> ((sort -> sort) * Sorts.algebra) * Consts.T |
|
86 |
-> CodeFuncgr.T -> term -> transact -> string * transact; |
|
87 |
val add_value_stmt: iterm * itype -> code -> code; |
|
88 |
val transact: (transact -> 'a * transact) -> code -> 'a * code; |
|
24219 | 89 |
end; |
90 |
||
91 |
structure CodeThingol: CODE_THINGOL = |
|
92 |
struct |
|
93 |
||
94 |
(** auxiliary **) |
|
95 |
||
96 |
fun unfoldl dest x = |
|
97 |
case dest x |
|
98 |
of NONE => (x, []) |
|
99 |
| SOME (x1, x2) => |
|
100 |
let val (x', xs') = unfoldl dest x1 in (x', xs' @ [x2]) end; |
|
101 |
||
102 |
fun unfoldr dest x = |
|
103 |
case dest x |
|
104 |
of NONE => ([], x) |
|
105 |
| SOME (x1, x2) => |
|
106 |
let val (xs', x') = unfoldr dest x2 in (x1::xs', x') end; |
|
107 |
||
108 |
||
109 |
(** language core - types, pattern, expressions **) |
|
110 |
||
111 |
(* language representation *) |
|
112 |
||
113 |
type vname = string; |
|
114 |
||
115 |
datatype dict = |
|
116 |
DictConst of string * dict list list |
|
117 |
| DictVar of string list * (vname * (int * int)); |
|
118 |
||
119 |
datatype itype = |
|
120 |
`%% of string * itype list |
|
121 |
| ITyVar of vname; |
|
122 |
||
24591 | 123 |
type const = string * (dict list list * itype list (*types of arguments*)) |
124 |
||
24219 | 125 |
datatype iterm = |
24591 | 126 |
IConst of const |
24219 | 127 |
| IVar of vname |
128 |
| `$ of iterm * iterm |
|
129 |
| `|-> of (vname * itype) * iterm |
|
130 |
| ICase of ((iterm * itype) * (iterm * iterm) list) * iterm; |
|
131 |
(*see also signature*) |
|
132 |
||
133 |
(* |
|
134 |
variable naming conventions |
|
135 |
||
136 |
bare names: |
|
137 |
variable names v |
|
138 |
class names class |
|
139 |
type constructor names tyco |
|
140 |
datatype names dtco |
|
24837 | 141 |
const names (general) c (const) |
24219 | 142 |
constructor names co |
24837 | 143 |
class parameter names classparam |
24219 | 144 |
arbitrary name s |
145 |
||
24837 | 146 |
v, c, co, classparam also annotated with types etc. |
24219 | 147 |
|
148 |
constructs: |
|
149 |
sort sort |
|
150 |
type parameters vs |
|
151 |
type ty |
|
152 |
type schemes tysm |
|
153 |
term t |
|
154 |
(term as pattern) p |
|
155 |
instance (class, tyco) inst |
|
156 |
*) |
|
157 |
||
158 |
fun ty1 `-> ty2 = "fun" `%% [ty1, ty2]; |
|
159 |
val op `--> = Library.foldr (op `->); |
|
160 |
val op `$$ = Library.foldl (op `$); |
|
161 |
val op `|--> = Library.foldr (op `|->); |
|
162 |
||
163 |
val unfold_fun = unfoldr |
|
164 |
(fn "fun" `%% [ty1, ty2] => SOME (ty1, ty2) |
|
165 |
| _ => NONE); |
|
166 |
||
167 |
val unfold_app = unfoldl |
|
168 |
(fn op `$ t => SOME t |
|
169 |
| _ => NONE); |
|
170 |
||
171 |
val split_abs = |
|
172 |
(fn (v, ty) `|-> (t as ICase (((IVar w, _), [(p, t')]), _)) => |
|
173 |
if v = w then SOME (((v, SOME p), ty), t') else SOME (((v, NONE), ty), t) |
|
174 |
| (v, ty) `|-> t => SOME (((v, NONE), ty), t) |
|
175 |
| _ => NONE); |
|
176 |
||
177 |
val unfold_abs = unfoldr split_abs; |
|
178 |
||
179 |
val split_let = |
|
180 |
(fn ICase (((td, ty), [(p, t)]), _) => SOME (((p, ty), td), t) |
|
181 |
| _ => NONE); |
|
182 |
||
183 |
val unfold_let = unfoldr split_let; |
|
184 |
||
185 |
fun unfold_const_app t = |
|
186 |
case unfold_app t |
|
187 |
of (IConst c, ts) => SOME (c, ts) |
|
188 |
| _ => NONE; |
|
189 |
||
190 |
fun fold_aiterms f (t as IConst _) = f t |
|
191 |
| fold_aiterms f (t as IVar _) = f t |
|
192 |
| fold_aiterms f (t1 `$ t2) = fold_aiterms f t1 #> fold_aiterms f t2 |
|
193 |
| fold_aiterms f (t as _ `|-> t') = f t #> fold_aiterms f t' |
|
194 |
| fold_aiterms f (ICase (_, t)) = fold_aiterms f t; |
|
195 |
||
196 |
fun fold_constnames f = |
|
197 |
let |
|
198 |
fun add (IConst (c, _)) = f c |
|
199 |
| add _ = I; |
|
200 |
in fold_aiterms add end; |
|
201 |
||
202 |
fun fold_varnames f = |
|
203 |
let |
|
204 |
fun add (IVar v) = f v |
|
205 |
| add ((v, _) `|-> _) = f v |
|
206 |
| add _ = I; |
|
207 |
in fold_aiterms add end; |
|
208 |
||
209 |
fun fold_unbound_varnames f = |
|
210 |
let |
|
211 |
fun add _ (IConst _) = I |
|
212 |
| add vs (IVar v) = if not (member (op =) vs v) then f v else I |
|
213 |
| add vs (t1 `$ t2) = add vs t1 #> add vs t2 |
|
214 |
| add vs ((v, _) `|-> t) = add (insert (op =) v vs) t |
|
215 |
| add vs (ICase (_, t)) = add vs t; |
|
216 |
in add [] end; |
|
217 |
||
218 |
fun collapse_let (((v, ty), se), be as ICase (((IVar w, _), ds), _)) = |
|
219 |
let |
|
220 |
fun exists_v t = fold_unbound_varnames (fn w => fn b => |
|
221 |
b orelse v = w) t false; |
|
222 |
in if v = w andalso forall (fn (t1, t2) => |
|
223 |
exists_v t1 orelse not (exists_v t2)) ds |
|
224 |
then ((se, ty), ds) |
|
225 |
else ((se, ty), [(IVar v, be)]) |
|
226 |
end |
|
227 |
| collapse_let (((v, ty), se), be) = |
|
228 |
((se, ty), [(IVar v, be)]) |
|
229 |
||
230 |
fun eta_expand (c as (_, (_, tys)), ts) k = |
|
231 |
let |
|
232 |
val j = length ts; |
|
233 |
val l = k - j; |
|
234 |
val ctxt = (fold o fold_varnames) Name.declare ts Name.context; |
|
235 |
val vs_tys = Name.names ctxt "a" ((curry Library.take l o curry Library.drop j) tys); |
|
236 |
in vs_tys `|--> IConst c `$$ ts @ map (fn (v, _) => IVar v) vs_tys end; |
|
237 |
||
24662
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
238 |
fun contains_dictvar t = |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
239 |
let |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
240 |
fun contains (DictConst (_, dss)) = (fold o fold) contains dss |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
241 |
| contains (DictVar _) = K true; |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
242 |
in |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
243 |
fold_aiterms |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
244 |
(fn IConst (_, (dss, _)) => (fold o fold) contains dss | _ => I) t false |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
245 |
end; |
f79f6061525c
more precise treatment of free dictionary parameters for evaluation
haftmann
parents:
24591
diff
changeset
|
246 |
|
24219 | 247 |
|
248 |
(** definitions, transactions **) |
|
249 |
||
250 |
type typscheme = (vname * sort) list * itype; |
|
24918 | 251 |
datatype stmt = |
24219 | 252 |
Bot |
24591 | 253 |
| Fun of typscheme * ((iterm list * iterm) * thm) list |
24219 | 254 |
| Datatype of (vname * sort) list * (string * itype list) list |
255 |
| Datatypecons of string |
|
24811 | 256 |
| Class of vname * ((class * string) list * (string * itype) list) |
24219 | 257 |
| Classrel of class * class |
24837 | 258 |
| Classparam of class |
24219 | 259 |
| Classinst of (class * (string * (vname * sort) list)) |
260 |
* ((class * (string * (string * dict list list))) list |
|
24591 | 261 |
* ((string * const) * thm) list); |
24219 | 262 |
|
24918 | 263 |
type code = stmt Graph.T; |
24219 | 264 |
|
265 |
||
266 |
(* abstract code *) |
|
267 |
||
268 |
val empty_code = Graph.empty : code; (*read: "depends on"*) |
|
269 |
||
270 |
fun ensure_bot name = Graph.default_node (name, Bot); |
|
271 |
||
272 |
fun add_def_incr (name, Bot) code = |
|
273 |
(case the_default Bot (try (Graph.get_node code) name) |
|
274 |
of Bot => error "Attempted to add Bot to code" |
|
275 |
| _ => code) |
|
276 |
| add_def_incr (name, def) code = |
|
277 |
(case try (Graph.get_node code) name |
|
278 |
of NONE => Graph.new_node (name, def) code |
|
279 |
| SOME Bot => Graph.map_node name (K def) code |
|
280 |
| SOME _ => error ("Tried to overwrite definition " ^ quote name)); |
|
281 |
||
24811 | 282 |
fun add_dep (NONE, _) = I |
283 |
| add_dep (SOME name1, name2) = |
|
284 |
if name1 = name2 then I else Graph.add_edge (name1, name2); |
|
24219 | 285 |
|
286 |
val merge_code : code * code -> code = Graph.merge (K true); |
|
287 |
||
288 |
fun project_code delete_empty_funs hidden raw_selected code = |
|
289 |
let |
|
290 |
fun is_empty_fun name = case Graph.get_node code name |
|
24381 | 291 |
of Fun (_, []) => true |
24219 | 292 |
| _ => false; |
293 |
val names = subtract (op =) hidden (Graph.keys code); |
|
294 |
val deleted = Graph.all_preds code (filter is_empty_fun names); |
|
295 |
val selected = case raw_selected |
|
296 |
of NONE => names |> subtract (op =) deleted |
|
297 |
| SOME sel => sel |
|
298 |
|> delete_empty_funs ? subtract (op =) deleted |
|
299 |
|> subtract (op =) hidden |
|
300 |
|> Graph.all_succs code |
|
301 |
|> delete_empty_funs ? subtract (op =) deleted |
|
302 |
|> subtract (op =) hidden; |
|
303 |
in |
|
304 |
code |
|
305 |
|> Graph.subgraph (member (op =) selected) |
|
306 |
end; |
|
307 |
||
308 |
fun empty_funs code = |
|
24381 | 309 |
Graph.fold (fn (name, (Fun (_, []), _)) => cons name |
24219 | 310 |
| _ => I) code []; |
311 |
||
312 |
fun is_cons code name = case Graph.get_node code name |
|
313 |
of Datatypecons _ => true |
|
314 |
| _ => false; |
|
315 |
||
316 |
||
317 |
(* transaction protocol *) |
|
318 |
||
24283 | 319 |
type transact = Graph.key option * code; |
320 |
||
24918 | 321 |
fun ensure_stmt stmtgen name (dep, code) = |
24219 | 322 |
let |
323 |
fun add_def false = |
|
324 |
ensure_bot name |
|
24811 | 325 |
#> add_dep (dep, name) |
24918 | 326 |
#> curry stmtgen (SOME name) |
24252 | 327 |
##> snd |
24811 | 328 |
#-> (fn def => add_def_incr (name, def)) |
24219 | 329 |
| add_def true = |
24811 | 330 |
add_dep (dep, name); |
24219 | 331 |
in |
332 |
code |
|
333 |
|> add_def (can (Graph.get_node code) name) |
|
334 |
|> pair dep |
|
24918 | 335 |
|> pair name |
24219 | 336 |
end; |
337 |
||
24918 | 338 |
fun transact f code = |
24252 | 339 |
(NONE, code) |
340 |
|> f |
|
341 |
|-> (fn x => fn (_, code) => (x, code)); |
|
24219 | 342 |
|
24918 | 343 |
|
344 |
(* translation kernel *) |
|
345 |
||
346 |
fun ensure_class thy (algbr as ((_, algebra), _)) funcgr class = |
|
347 |
let |
|
348 |
val superclasses = (Sorts.certify_sort algebra o Sorts.super_classes algebra) class; |
|
24969 | 349 |
val cs = #params (AxClass.get_info thy class); |
24918 | 350 |
val class' = CodeName.class thy class; |
351 |
val stmt_class = |
|
352 |
fold_map (fn superclass => ensure_class thy algbr funcgr superclass |
|
353 |
##>> ensure_classrel thy algbr funcgr (class, superclass)) superclasses |
|
354 |
##>> fold_map (fn (c, ty) => ensure_const thy algbr funcgr c |
|
355 |
##>> exprgen_typ thy algbr funcgr ty) cs |
|
24969 | 356 |
#>> (fn info => Class (unprefix "'" Name.aT, info)) |
24918 | 357 |
in |
358 |
ensure_stmt stmt_class class' |
|
359 |
end |
|
360 |
and ensure_classrel thy algbr funcgr (subclass, superclass) = |
|
361 |
let |
|
362 |
val classrel' = CodeName.classrel thy (subclass, superclass); |
|
363 |
val stmt_classrel = |
|
364 |
ensure_class thy algbr funcgr subclass |
|
365 |
##>> ensure_class thy algbr funcgr superclass |
|
366 |
#>> Classrel; |
|
367 |
in |
|
368 |
ensure_stmt stmt_classrel classrel' |
|
369 |
end |
|
370 |
and ensure_tyco thy algbr funcgr "fun" = |
|
371 |
pair "fun" |
|
372 |
| ensure_tyco thy algbr funcgr tyco = |
|
373 |
let |
|
374 |
val stmt_datatype = |
|
375 |
let |
|
376 |
val (vs, cos) = Code.get_datatype thy tyco; |
|
377 |
in |
|
378 |
fold_map (exprgen_tyvar_sort thy algbr funcgr) vs |
|
379 |
##>> fold_map (fn (c, tys) => |
|
380 |
ensure_const thy algbr funcgr c |
|
381 |
##>> fold_map (exprgen_typ thy algbr funcgr) tys) cos |
|
382 |
#>> Datatype |
|
383 |
end; |
|
384 |
val tyco' = CodeName.tyco thy tyco; |
|
385 |
in |
|
386 |
ensure_stmt stmt_datatype tyco' |
|
387 |
end |
|
388 |
and exprgen_tyvar_sort thy (algbr as ((proj_sort, _), _)) funcgr (v, sort) = |
|
389 |
fold_map (ensure_class thy algbr funcgr) (proj_sort sort) |
|
390 |
#>> (fn sort => (unprefix "'" v, sort)) |
|
391 |
and exprgen_typ thy algbr funcgr (TFree vs) = |
|
392 |
exprgen_tyvar_sort thy algbr funcgr vs |
|
393 |
#>> (fn (v, sort) => ITyVar v) |
|
394 |
| exprgen_typ thy algbr funcgr (Type (tyco, tys)) = |
|
395 |
ensure_tyco thy algbr funcgr tyco |
|
396 |
##>> fold_map (exprgen_typ thy algbr funcgr) tys |
|
397 |
#>> (fn (tyco, tys) => tyco `%% tys) |
|
398 |
and exprgen_dicts thy (algbr as ((proj_sort, algebra), consts)) funcgr (ty_ctxt, sort_decl) = |
|
399 |
let |
|
400 |
val pp = Sign.pp thy; |
|
401 |
datatype typarg = |
|
402 |
Global of (class * string) * typarg list list |
|
403 |
| Local of (class * class) list * (string * (int * sort)); |
|
404 |
fun class_relation (Global ((_, tyco), yss), _) class = |
|
405 |
Global ((class, tyco), yss) |
|
406 |
| class_relation (Local (classrels, v), subclass) superclass = |
|
407 |
Local ((subclass, superclass) :: classrels, v); |
|
408 |
fun type_constructor tyco yss class = |
|
409 |
Global ((class, tyco), (map o map) fst yss); |
|
410 |
fun type_variable (TFree (v, sort)) = |
|
411 |
let |
|
412 |
val sort' = proj_sort sort; |
|
413 |
in map_index (fn (n, class) => (Local ([], (v, (n, sort'))), class)) sort' end; |
|
414 |
val typargs = Sorts.of_sort_derivation pp algebra |
|
415 |
{class_relation = class_relation, type_constructor = type_constructor, |
|
416 |
type_variable = type_variable} |
|
417 |
(ty_ctxt, proj_sort sort_decl); |
|
418 |
fun mk_dict (Global (inst, yss)) = |
|
419 |
ensure_inst thy algbr funcgr inst |
|
420 |
##>> (fold_map o fold_map) mk_dict yss |
|
421 |
#>> (fn (inst, dss) => DictConst (inst, dss)) |
|
422 |
| mk_dict (Local (classrels, (v, (k, sort)))) = |
|
423 |
fold_map (ensure_classrel thy algbr funcgr) classrels |
|
424 |
#>> (fn classrels => DictVar (classrels, (unprefix "'" v, (k, length sort)))) |
|
425 |
in |
|
426 |
fold_map mk_dict typargs |
|
427 |
end |
|
428 |
and exprgen_dict_parms thy (algbr as (_, consts)) funcgr (c, ty_ctxt) = |
|
429 |
let |
|
25042
a33b78d63114
renamed Consts.the_declaration to Consts.the_type;
wenzelm
parents:
24969
diff
changeset
|
430 |
val ty_decl = Consts.the_type consts c; |
24918 | 431 |
val (tys, tys_decl) = pairself (curry (Consts.typargs consts) c) (ty_ctxt, ty_decl); |
432 |
val sorts = map (snd o dest_TVar) tys_decl; |
|
433 |
in |
|
434 |
fold_map (exprgen_dicts thy algbr funcgr) (tys ~~ sorts) |
|
435 |
end |
|
436 |
and exprgen_eq thy algbr funcgr thm = |
|
437 |
let |
|
438 |
val (args, rhs) = (apfst (snd o strip_comb) o Logic.dest_equals |
|
439 |
o Logic.unvarify o prop_of) thm; |
|
440 |
in |
|
441 |
fold_map (exprgen_term thy algbr funcgr) args |
|
442 |
##>> exprgen_term thy algbr funcgr rhs |
|
443 |
#>> rpair thm |
|
444 |
end |
|
445 |
and ensure_inst thy (algbr as ((_, algebra), _)) funcgr (class, tyco) = |
|
446 |
let |
|
447 |
val superclasses = (Sorts.certify_sort algebra o Sorts.super_classes algebra) class; |
|
24969 | 448 |
val classparams = these (try (#params o AxClass.get_info thy) class); |
24918 | 449 |
val vs = Name.names Name.context "'a" (Sorts.mg_domain algebra tyco [class]); |
450 |
val sorts' = Sorts.mg_domain (Sign.classes_of thy) tyco [class]; |
|
451 |
val vs' = map2 (fn (v, sort1) => fn sort2 => (v, |
|
452 |
Sorts.inter_sort (Sign.classes_of thy) (sort1, sort2))) vs sorts'; |
|
453 |
val arity_typ = Type (tyco, map TFree vs); |
|
454 |
val arity_typ' = Type (tyco, map (fn (v, sort) => TVar ((v, 0), sort)) vs'); |
|
455 |
fun exprgen_superarity superclass = |
|
456 |
ensure_class thy algbr funcgr superclass |
|
457 |
##>> ensure_classrel thy algbr funcgr (class, superclass) |
|
458 |
##>> exprgen_dicts thy algbr funcgr (arity_typ, [superclass]) |
|
459 |
#>> (fn ((superclass, classrel), [DictConst (inst, dss)]) => |
|
460 |
(superclass, (classrel, (inst, dss)))); |
|
461 |
fun exprgen_classparam_inst (c, ty) = |
|
462 |
let |
|
463 |
val c_inst = Const (c, map_type_tfree (K arity_typ') ty); |
|
464 |
val thm = Class.unoverload thy (Thm.cterm_of thy c_inst); |
|
465 |
val c_ty = (apsnd Logic.unvarifyT o dest_Const o snd |
|
466 |
o Logic.dest_equals o Thm.prop_of) thm; |
|
467 |
in |
|
468 |
ensure_const thy algbr funcgr c |
|
469 |
##>> exprgen_const thy algbr funcgr c_ty |
|
470 |
#>> (fn (c, IConst c_inst) => ((c, c_inst), thm)) |
|
471 |
end; |
|
472 |
val stmt_inst = |
|
473 |
ensure_class thy algbr funcgr class |
|
474 |
##>> ensure_tyco thy algbr funcgr tyco |
|
475 |
##>> fold_map (exprgen_tyvar_sort thy algbr funcgr) vs |
|
476 |
##>> fold_map exprgen_superarity superclasses |
|
477 |
##>> fold_map exprgen_classparam_inst classparams |
|
478 |
#>> (fn ((((class, tyco), arity), superarities), classparams) => |
|
479 |
Classinst ((class, (tyco, arity)), (superarities, classparams))); |
|
480 |
val inst = CodeName.instance thy (class, tyco); |
|
481 |
in |
|
482 |
ensure_stmt stmt_inst inst |
|
483 |
end |
|
484 |
and ensure_const thy (algbr as (_, consts)) funcgr c = |
|
485 |
let |
|
486 |
val c' = CodeName.const thy c; |
|
487 |
fun stmt_datatypecons tyco = |
|
488 |
ensure_tyco thy algbr funcgr tyco |
|
489 |
#>> K (Datatypecons c'); |
|
490 |
fun stmt_classparam class = |
|
491 |
ensure_class thy algbr funcgr class |
|
492 |
#>> K (Classparam c'); |
|
493 |
fun stmt_fun trns = |
|
494 |
let |
|
495 |
val raw_thms = CodeFuncgr.funcs funcgr c; |
|
496 |
val ty = (Logic.unvarifyT o CodeFuncgr.typ funcgr) c; |
|
497 |
val vs = (map dest_TFree o Consts.typargs consts) (c, ty); |
|
498 |
val thms = if (null o Term.typ_tfrees) ty orelse (null o fst o strip_type) ty |
|
499 |
then raw_thms |
|
500 |
else map (CodeUnit.expand_eta 1) raw_thms; |
|
501 |
in |
|
502 |
trns |
|
503 |
|> fold_map (exprgen_tyvar_sort thy algbr funcgr) vs |
|
504 |
||>> exprgen_typ thy algbr funcgr ty |
|
505 |
||>> fold_map (exprgen_eq thy algbr funcgr) thms |
|
506 |
|>> (fn ((vs, ty), eqs) => Fun ((vs, ty), eqs)) |
|
507 |
end; |
|
508 |
val stmtgen = case Code.get_datatype_of_constr thy c |
|
509 |
of SOME tyco => stmt_datatypecons tyco |
|
510 |
| NONE => (case AxClass.class_of_param thy c |
|
511 |
of SOME class => stmt_classparam class |
|
512 |
| NONE => stmt_fun) |
|
513 |
in |
|
514 |
ensure_stmt stmtgen c' |
|
515 |
end |
|
516 |
and exprgen_term thy algbr funcgr (Const (c, ty)) = |
|
517 |
exprgen_app thy algbr funcgr ((c, ty), []) |
|
518 |
| exprgen_term thy algbr funcgr (Free (v, _)) = |
|
519 |
pair (IVar v) |
|
520 |
| exprgen_term thy algbr funcgr (Abs (abs as (_, ty, _))) = |
|
521 |
let |
|
522 |
val (v, t) = Syntax.variant_abs abs; |
|
523 |
in |
|
524 |
exprgen_typ thy algbr funcgr ty |
|
525 |
##>> exprgen_term thy algbr funcgr t |
|
526 |
#>> (fn (ty, t) => (v, ty) `|-> t) |
|
527 |
end |
|
528 |
| exprgen_term thy algbr funcgr (t as _ $ _) = |
|
529 |
case strip_comb t |
|
530 |
of (Const (c, ty), ts) => |
|
531 |
exprgen_app thy algbr funcgr ((c, ty), ts) |
|
532 |
| (t', ts) => |
|
533 |
exprgen_term thy algbr funcgr t' |
|
534 |
##>> fold_map (exprgen_term thy algbr funcgr) ts |
|
535 |
#>> (fn (t, ts) => t `$$ ts) |
|
536 |
and exprgen_const thy algbr funcgr (c, ty) = |
|
537 |
ensure_const thy algbr funcgr c |
|
538 |
##>> exprgen_dict_parms thy algbr funcgr (c, ty) |
|
539 |
##>> fold_map (exprgen_typ thy algbr funcgr) ((fst o Term.strip_type) ty) |
|
540 |
#>> (fn ((c, iss), tys) => IConst (c, (iss, tys))) |
|
541 |
and exprgen_app_default thy algbr funcgr (c_ty, ts) = |
|
542 |
exprgen_const thy algbr funcgr c_ty |
|
543 |
##>> fold_map (exprgen_term thy algbr funcgr) ts |
|
544 |
#>> (fn (t, ts) => t `$$ ts) |
|
545 |
and exprgen_case thy algbr funcgr n cases (app as ((c, ty), ts)) = |
|
546 |
let |
|
547 |
val (tys, _) = |
|
548 |
(chop (1 + (if null cases then 1 else length cases)) o fst o strip_type) ty; |
|
549 |
val dt = nth ts n; |
|
550 |
val dty = nth tys n; |
|
551 |
fun is_undefined (Const (c, _)) = Code.is_undefined thy c |
|
552 |
| is_undefined _ = false; |
|
553 |
fun mk_case (co, n) t = |
|
554 |
let |
|
555 |
val (vs, body) = Term.strip_abs_eta n t; |
|
556 |
val selector = list_comb (Const (co, map snd vs ---> dty), map Free vs); |
|
557 |
in if is_undefined body then NONE else SOME (selector, body) end; |
|
558 |
fun mk_ds [] = |
|
559 |
let |
|
560 |
val ([v_ty], body) = Term.strip_abs_eta 1 (the_single (nth_drop n ts)) |
|
561 |
in [(Free v_ty, body)] end |
|
562 |
| mk_ds cases = map_filter (uncurry mk_case) |
|
563 |
(AList.make (CodeUnit.no_args thy) cases ~~ nth_drop n ts); |
|
564 |
in |
|
565 |
exprgen_term thy algbr funcgr dt |
|
566 |
##>> exprgen_typ thy algbr funcgr dty |
|
567 |
##>> fold_map (fn (pat, body) => exprgen_term thy algbr funcgr pat |
|
568 |
##>> exprgen_term thy algbr funcgr body) (mk_ds cases) |
|
569 |
##>> exprgen_app_default thy algbr funcgr app |
|
570 |
#>> (fn (((dt, dty), ds), t0) => ICase (((dt, dty), ds), t0)) |
|
571 |
end |
|
572 |
and exprgen_app thy algbr funcgr ((c, ty), ts) = case Code.get_case_data thy c |
|
573 |
of SOME (n, cases) => let val i = 1 + (if null cases then 1 else length cases) in |
|
574 |
if length ts < i then |
|
575 |
let |
|
576 |
val k = length ts; |
|
577 |
val tys = (curry Library.take (i - k) o curry Library.drop k o fst o strip_type) ty; |
|
578 |
val ctxt = (fold o fold_aterms) |
|
579 |
(fn Free (v, _) => Name.declare v | _ => I) ts Name.context; |
|
580 |
val vs = Name.names ctxt "a" tys; |
|
581 |
in |
|
582 |
fold_map (exprgen_typ thy algbr funcgr) tys |
|
583 |
##>> exprgen_case thy algbr funcgr n cases ((c, ty), ts @ map Free vs) |
|
584 |
#>> (fn (tys, t) => map2 (fn (v, _) => pair v) vs tys `|--> t) |
|
585 |
end |
|
586 |
else if length ts > i then |
|
587 |
exprgen_case thy algbr funcgr n cases ((c, ty), Library.take (i, ts)) |
|
588 |
##>> fold_map (exprgen_term thy algbr funcgr) (Library.drop (i, ts)) |
|
589 |
#>> (fn (t, ts) => t `$$ ts) |
|
590 |
else |
|
591 |
exprgen_case thy algbr funcgr n cases ((c, ty), ts) |
|
592 |
end |
|
593 |
| NONE => exprgen_app_default thy algbr funcgr ((c, ty), ts); |
|
594 |
||
595 |
fun ensure_value thy algbr funcgr t = |
|
596 |
let |
|
597 |
val ty = fastype_of t; |
|
598 |
val vs = fold_term_types (K (fold_atyps (insert (eq_fst op =) |
|
599 |
o dest_TFree))) t []; |
|
600 |
val stmt_value = |
|
601 |
fold_map (exprgen_tyvar_sort thy algbr funcgr) vs |
|
602 |
##>> exprgen_typ thy algbr funcgr ty |
|
603 |
##>> exprgen_term thy algbr funcgr t |
|
604 |
#>> (fn ((vs, ty), t) => Fun ((vs, ty), [(([], t), Drule.dummy_thm)])); |
|
605 |
in |
|
606 |
ensure_stmt stmt_value CodeName.value_name |
|
607 |
end; |
|
608 |
||
609 |
fun add_value_stmt (t, ty) code = |
|
24219 | 610 |
code |
24918 | 611 |
|> Graph.new_node (CodeName.value_name, Fun (([], ty), [(([], t), Drule.dummy_thm)])) |
612 |
|> fold (curry Graph.add_edge CodeName.value_name) (Graph.keys code); |
|
24219 | 613 |
|
614 |
end; (*struct*) |
|
615 |
||
616 |
||
617 |
structure BasicCodeThingol: BASIC_CODE_THINGOL = CodeThingol; |