1 (* Copyright (C) 2003-2005, HELM Team.
3 * This file is part of HELM, an Hypertextual, Electronic
4 * Library of Mathematics, developed at the Computer Science
5 * Department, University of Bologna, Italy.
7 * HELM is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation; either version 2
10 * of the License, or (at your option) any later version.
12 * HELM is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 * GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with HELM; if not, write to the Free Software
19 * Foundation, Inc., 59 Temple Place - Suite 330, Boston,
22 * For details, see the HELM World-Wide-Web page,
23 * http://cs.unibo.it/helm/.
28 module S = CicSubstitution
29 module TC = CicTypeChecker
31 module UM = UriManager
32 module Obj = LibraryObjects
35 module E = CicEnvironment
37 module PEH = ProofEngineHelpers
39 module DTI = DoubleTypeInference
40 module NU = CicNotationUtil
44 module Cl = ProceduralClassify
45 module T = ProceduralTypes
46 module Cn = ProceduralConversion
47 module H = ProceduralHelpers
50 sorts : (C.id, A.sort_kind) Hashtbl.t;
51 types : (C.id, A.anntypes) Hashtbl.t;
52 params : G.inline_param list;
53 max_depth: int option;
62 (* helpers ******************************************************************)
64 let split2_last l1 l2 =
66 let n = pred (List.length l1) in
67 let before1, after1 = HEL.split_nth n l1 in
68 let before2, after2 = HEL.split_nth n l2 in
69 before1, before2, List.hd after1, List.hd after2
70 with Invalid_argument _ -> failwith "A2P.split2_last"
72 let string_of_head = function
74 | C.AConst _ -> "const"
75 | C.AMutInd _ -> "mutind"
76 | C.AMutConstruct _ -> "mutconstruct"
80 | C.ALambda _ -> "lambda"
81 | C.ALetIn _ -> "letin"
83 | C.ACoFix _ -> "cofix"
86 | C.AMutCase _ -> "mutcase"
88 | C.AImplicit _ -> "implict"
90 let next st = {st with depth = succ st.depth}
92 let add st entry = {st with context = entry :: st.context}
94 let push st = {st with case = 1 :: st.case}
97 {st with case = match st.case with
99 | hd :: tl -> succ hd :: tl
103 let case = String.concat "." (List.rev_map string_of_int st.case) in
104 Printf.sprintf "case %s: %s" case str
108 let msg = Printf.sprintf "Depth %u: " st.depth in
109 match st.max_depth with
111 | Some d -> if st.depth < d then true, msg else false, "DEPTH EXCEDED: "
112 with Invalid_argument _ -> failwith "A2P.test_depth"
114 let is_rewrite_right st = function
115 | C.AConst (_, uri, []) -> st.defaults && Obj.is_eq_ind_r_URI uri
118 let is_rewrite_left st = function
119 | C.AConst (_, uri, []) -> st.defaults && Obj.is_eq_ind_URI uri
122 let is_fwd_rewrite_right st hd tl =
123 if is_rewrite_right st hd then match List.nth tl 3 with
128 let is_fwd_rewrite_left st hd tl =
129 if is_rewrite_left st hd then match List.nth tl 3 with
134 let get_inner_types st v =
136 let id = Ut.id_of_annterm v in
137 try match Hashtbl.find st.types id with
138 | {A.annsynthesized = st; A.annexpected = Some et} -> Some (st, et)
139 | {A.annsynthesized = st; A.annexpected = None} -> Some (st, st)
140 with Not_found -> None
141 with Invalid_argument _ -> failwith "P2.get_inner_types"
143 let get_entry st id =
144 let rec aux = function
146 | Some (C.Name name, e) :: _ when name = id -> e
151 let string_of_atomic = function
152 | C.ARel (_, _, _, s) -> s
153 | C.AVar (_, uri, _) -> H.name_of_uri uri None None
154 | C.AConst (_, uri, _) -> H.name_of_uri uri None None
155 | C.AMutInd (_, uri, i, _) -> H.name_of_uri uri (Some i) None
156 | C.AMutConstruct (_, uri, i, j, _) -> H.name_of_uri uri (Some i) (Some j)
159 let get_sub_names head l =
160 let s = string_of_atomic head in
161 if s = "" then [] else
162 let map (names, i) _ =
163 let name = Printf.sprintf "%s_%u" s i in name :: names, succ i
165 let names, _ = List.fold_left map ([], 1) l in
168 let get_type msg st t = H.get_type msg st.context (H.cic t)
170 (* proof construction *******************************************************)
172 let anonymous_premise = C.Name "UNNAMED"
174 let mk_exp_args hd tl classes synth qs =
176 let meta id = C.AImplicit (id, None) in
178 if I.overlaps synth cl then
179 let w = if H.is_atomic (H.cic v) then v else meta "" in
180 if b then v, v else meta "", w
184 let rec rev a = function
187 if snd hd <> meta "" then incr exp;
190 let rec aux = function
193 if fst hd = meta "" then aux tl else rev [] (hd :: tl)
195 let args = T.list_rev_map2 map tl classes in
196 let args = aux args in
197 let part = !exp < List.length tl in
198 if args = [] then part, hd, qs else part, C.AAppl ("", hd :: args), qs
200 let mk_convert st ?name sty ety note =
202 let a = ref "" in Ut.pp_term (fun s -> a := !a ^ s) [] st.context t; !a
204 let e = Cn.hole "" in
205 let csty, cety = H.cic sty, H.cic ety in
208 let sname = match name with None -> "" | Some (id, _) -> id in
209 Printf.sprintf "%s: %s\nSINTH: %s\nEXP: %s"
210 note sname (ppterm csty) (ppterm cety)
213 if H.alpha_equivalence ~flatten:true st.context csty cety then [T.Note note] else
214 let sty, ety = H.acic_bc st.context sty, H.acic_bc st.context ety in
216 | None -> [T.Change (sty, ety, None, e, note)]
218 begin match get_entry st id with
220 [T.Change (ety, sty, Some (id, Some id), e, note);
224 [T.Change (ety, sty, Some (id, Some id), e, note)]
227 let convert st ?name v =
228 match get_inner_types st v with
230 if !debug then [T.Note "NORMAL: NO INNER TYPES"] else []
231 | Some (sty, ety) -> mk_convert st ?name sty ety "NORMAL"
233 let get_intro = function
234 | C.Anonymous -> None
237 let mk_preamble st what script = match script with
238 | T.Exact _ :: _ -> script
239 | _ -> convert st what @ script
241 let mk_arg st = function
242 | C.ARel (_, _, i, name) as what -> convert st ~name:(name, i) what
245 let mk_fwd_rewrite st dtext name tl direction v t ity ety =
246 let compare premise = function
248 | Some s -> s = premise
250 assert (List.length tl = 6);
251 let what, where, predicate = List.nth tl 5, List.nth tl 3, List.nth tl 2 in
252 let e = Cn.mk_pattern 1 ety predicate in
253 if (Cn.does_not_occur e) then st, [] else
255 | C.ARel (_, _, i, premise) as w ->
257 let where = Some (premise, name) in
258 let script = mk_arg st what @ mk_arg st w in
259 T.Rewrite (direction, what, where, e, dtext) :: script
261 if DTI.does_not_occur (succ i) (H.cic t) || compare premise name then
262 {st with context = Cn.clear st.context premise}, script name
264 assert (Ut.is_sober st.context (H.cic ity));
265 let ity = H.acic_bc st.context ity in
266 let br1 = [T.Id ""] in
267 let br2 = List.rev (T.Exact (w, "assumption") :: script None) in
268 let text = "non-linear rewrite" in
269 st, [T.Branch ([br2; br1], ""); T.Cut (name, ity, text)]
273 let mk_rewrite st dtext where qs tl direction t ety =
274 assert (List.length tl = 5);
275 let predicate = List.nth tl 2 in
276 let e = Cn.mk_pattern 1 ety predicate in
277 let script = [T.Branch (qs, "")] in
278 if (Cn.does_not_occur e) then script else
279 T.Rewrite (direction, where, None, e, dtext) :: script
281 let rec proc_lambda st what name v t =
282 let name = match name with
283 | C.Anonymous -> H.mk_fresh_name true st.context anonymous_premise
286 let entry = Some (name, C.Decl (H.cic v)) in
287 let intro = get_intro name in
288 let script = proc_proof (add st entry) t in
289 let script = T.Intros (Some 1, [intro], "") :: script in
290 mk_preamble st what script
292 and proc_letin st what name v w t =
293 let intro = get_intro name in
294 let proceed, dtext = test_depth st in
295 let script = if proceed then
296 let st, hyp, rqv = match get_inner_types st what, get_inner_types st v with
297 | Some (C.ALetIn (_, _, iv, iw, _), _), _ when
298 H.alpha_equivalence ~flatten:true st.context (H.cic v) (H.cic iv) &&
299 H.alpha_equivalence ~flatten:true st.context (H.cic w) (H.cic iw)
301 st, C.Def (H.cic v, H.cic w), [T.Intros (Some 1, [intro], dtext)]
302 | _, Some (ity, ety) ->
303 let st, rqv = match v with
304 | C.AAppl (_, hd :: tl) when is_fwd_rewrite_right st hd tl ->
305 mk_fwd_rewrite st dtext intro tl true v t ity ety
306 | C.AAppl (_, hd :: tl) when is_fwd_rewrite_left st hd tl ->
307 mk_fwd_rewrite st dtext intro tl false v t ity ety
309 assert (Ut.is_sober st.context (H.cic ety));
310 let ety = H.acic_bc st.context ety in
311 let qs = [proc_proof (next st) v; [T.Id ""]] in
312 st, [T.Branch (qs, ""); T.Cut (intro, ety, dtext)]
314 st, C.Decl (H.cic ity), rqv
316 st, C.Def (H.cic v, H.cic w), [T.LetIn (intro, v, dtext)]
318 let entry = Some (name, hyp) in
319 let qt = proc_proof (next (add st entry)) t in
320 List.rev_append rqv qt
322 [T.Exact (what, dtext)]
324 mk_preamble st what script
326 and proc_rel st what =
327 let _, dtext = test_depth st in
328 let text = "assumption" in
329 let script = [T.Exact (what, dtext ^ text)] in
330 mk_preamble st what script
332 and proc_mutconstruct st what =
333 let _, dtext = test_depth st in
334 let script = [T.Exact (what, dtext)] in
335 mk_preamble st what script
337 and proc_const st what =
338 let _, dtext = test_depth st in
339 let script = [T.Exact (what, dtext)] in
340 mk_preamble st what script
342 and proc_appl st what hd tl =
343 let proceed, dtext = test_depth st in
344 let script = if proceed then
345 let ty = match get_inner_types st hd with
346 | Some (ity, _) -> H.cic ity
347 | None -> get_type "TC2" st hd
349 let classes, rc = Cl.classify st.context ty in
350 let goal_arity, goal = match get_inner_types st what with
353 snd (PEH.split_with_whd (st.context, H.cic ity)), Some (H.cic ety)
355 let parsno, argsno = List.length classes, List.length tl in
356 let decurry = parsno - argsno in
357 let diff = goal_arity - decurry in
359 let text = Printf.sprintf "partial application: %i" diff in
360 prerr_endline ("Procedural 2: " ^ text);
361 [T.Exact (what, dtext ^ text)]
363 let classes = Cl.adjust st.context tl ?goal classes in
364 let rec mk_synth a n =
365 if n < 0 then a else mk_synth (I.S.add n a) (pred n)
367 let synth = mk_synth I.S.empty decurry in
368 let text = "" (* Printf.sprintf "%u %s" parsno (Cl.to_string h) *) in
369 let script = List.rev (mk_arg st hd) in
370 let tactic b t n = if b then T.Apply (t, n) else T.Exact (t, n) in
372 | Some (i, j, uri, tyno) when decurry = 0 ->
373 let classes2, tl2, _, where = split2_last classes tl in
374 let script2 = List.rev (mk_arg st where) @ script in
375 let synth2 = I.S.add 1 synth in
376 let names = H.get_ind_names uri tyno in
377 let qs = proc_bkd_proofs (next st) synth2 names classes2 tl2 in
378 let ety = match get_inner_types st what with
379 | Some (_, ety) -> ety
381 Cn.fake_annotate "" st.context (get_type "TC3" st what)
383 if List.length qs <> List.length names then
384 let qs = proc_bkd_proofs (next st) synth [] classes tl in
385 let b, hd, qs = mk_exp_args hd tl classes synth qs in
386 script @ [tactic b hd (dtext ^ text); T.Branch (qs, "")]
387 else if is_rewrite_right st hd then
388 script2 @ mk_rewrite st dtext where qs tl2 false what ety
389 else if is_rewrite_left st hd then
390 script2 @ mk_rewrite st dtext where qs tl2 true what ety
392 let predicate = List.nth tl2 (parsno - i) in
393 let e = Cn.mk_pattern j ety predicate in
394 let using = Some hd in
396 [T.Elim (where, using, e, dtext ^ text); T.Branch (qs, "")]
398 let names = get_sub_names hd tl in
399 let qs = proc_bkd_proofs (next st) synth names classes tl in
400 let b, hd, qs = mk_exp_args hd tl classes synth qs in
401 script @ [tactic b hd (dtext ^ text); T.Branch (qs, "")]
403 [T.Exact (what, dtext)]
405 mk_preamble st what script
407 and proc_case st what uri tyno u v ts =
408 let proceed, dtext = test_depth st in
409 let script = if proceed then
410 let synth, classes = I.S.empty, Cl.make ts in
411 let names = H.get_ind_names uri tyno in
412 let qs = proc_bkd_proofs (next st) synth names classes ts in
413 let lpsno, _ = H.get_ind_type uri tyno in
414 let ps, _ = H.get_ind_parameters st.context (H.cic v) in
415 let _, rps = HEL.split_nth lpsno ps in
416 let rpsno = List.length rps in
417 let ety = match get_inner_types st what with
418 | Some (_, ety) -> ety
420 Cn.fake_annotate "" st.context (get_type "TC4" st what)
422 let e = Cn.mk_pattern rpsno ety u in
424 let script = List.rev (mk_arg st v) in
425 script @ [T.Cases (v, e, dtext ^ text); T.Branch (qs, "")]
427 [T.Exact (what, dtext)]
429 mk_preamble st what script
431 and proc_other st what =
432 let _, dtext = test_depth st in
433 let text = Printf.sprintf "%s: %s" "UNEXPANDED" (string_of_head what) in
434 let script = [T.Exact (what, dtext ^ text)] in
435 mk_preamble st what script
437 and proc_proof st t =
439 let xtypes, note = match get_inner_types st t with
440 | Some (it, et) -> Some (H.cic it, H.cic et),
441 (Printf.sprintf "\nInferred: %s\nExpected: %s"
442 (Pp.ppterm (H.cic it)) (Pp.ppterm (H.cic et)))
443 | None -> None, "\nNo types"
445 let context, _clears = Cn.get_clears st.context (H.cic t) xtypes in
446 {st with context = context}
449 | C.ALambda (_, name, w, t) as what -> proc_lambda (f st) what name w t
450 | C.ALetIn (_, name, v, w, t) as what -> proc_letin (f st) what name v w t
451 | C.ARel _ as what -> proc_rel (f st) what
452 | C.AMutConstruct _ as what -> proc_mutconstruct (f st) what
453 | C.AConst _ as what -> proc_const (f st) what
454 | C.AAppl (_, hd :: tl) as what -> proc_appl (f st) what hd tl
455 (* FG: we deactivate the tactic "cases" because it does not work properly
456 | C.AMutCase (_, uri, i, u, v, ts) as what -> proc_case (f st) what uri i u v ts
458 | what -> proc_other (f st) what
460 and proc_bkd_proofs st synth names classes ts =
462 let get_names b = ref (names, if b then push st else st) in
463 let get_note f b names =
466 | "" :: tl, st -> names := tl, st; f st
468 let note = case st hd in
470 if b then T.Note note :: f st else f st
472 let _, dtext = test_depth st in
474 if I.overlaps synth inv then None else
475 if I.S.is_empty inv then Some (get_note (fun st -> proc_proof st v)) else
476 Some (get_note (fun _ -> [T.Exact (v, dtext ^ "dependent")]))
478 let ps = T.list_map2_filter aux classes ts in
479 let b = List.length ps > 1 in
480 let names = get_names b in
481 List.rev_map (fun f -> f b names) ps
483 with Invalid_argument s -> failwith ("A2P.proc_bkd_proofs: " ^ s)
485 (* initialization ***********************************************************)
487 let init ~ids_to_inner_sorts ~ids_to_inner_types params context =
488 let depth_map x y = match x, y with
489 | None, G.IPDepth depth -> Some depth
493 sorts = ids_to_inner_sorts;
494 types = ids_to_inner_types;
496 max_depth = List.fold_left depth_map None params;
498 defaults = not (List.mem G.IPNoDefaults params);