lift_epsilon.ml 2.69 KB
Newer Older
Andrei Paskevich's avatar
Andrei Paskevich committed
1 2 3 4 5 6 7 8 9 10
(********************************************************************)
(*                                                                  *)
(*  The Why3 Verification Platform   /   The Why3 Development Team  *)
(*  Copyright 2010-2012   --   INRIA - CNRS - Paris-Sud University  *)
(*                                                                  *)
(*  This software is distributed under the terms of the GNU Lesser  *)
(*  General Public License version 2.1, with the special exception  *)
(*  on linking described in file LICENSE.                           *)
(*                                                                  *)
(********************************************************************)
11

12
open Close_epsilon
13 14 15 16
open Term
open Theory
open Task

17 18 19 20
type lift_kind =
(*   | Goal (* prove the existence of a witness *) *)
  | Implied (* require the existence of a witness in an axiom *)
  | Implicit (* do not require a witness *)
21

22 23 24 25
let lift kind =
  let rec term acc t =
    match t.t_node with
    | Teps fb ->
26
        let fv = Mvs.keys t.t_vars in
27
        let x, f = t_open_bound fb in
28 29 30 31
        let acc, f = form acc f in
        let tys = List.map (fun x -> x.vs_ty) fv in
        let xs = Ident.id_derive "epsilon" x.vs_name in
        let xl = create_fsymbol xs tys x.vs_ty in
32
        let acc = add_decl acc (Decl.create_param_decl xl) in
33 34
        let axs =
          Decl.create_prsymbol (Ident.id_derive ("epsilon_def") x.vs_name) in
35
        let xlapp = t_app xl (List.map t_var fv) t.t_ty in
36 37 38 39
        let f =
          match kind with
          (* assume that lambdas always exist *)
          | Implied when not (is_lambda t) ->
40 41
              t_forall_close_merge fv
                (t_implies (t_exists_close [x] [] f)
42 43
                   (t_subst_single x xlapp f))
          | _ -> t_subst_single x xlapp f
44 45 46
        in
        let acc = add_decl acc (Decl.create_prop_decl Decl.Paxiom axs f) in
        acc, xlapp
47 48
    | _ -> TermTF.t_map_fold term form acc t
  and form acc f = TermTF.t_map_fold term form acc f in
49 50 51 52
  fun th acc ->
    let th = th.task_decl in
    match th.td_node with
    | Decl d ->
53
        let acc, d = Decl.DeclTF.decl_map_fold term form acc d in
54 55
        add_decl acc d
    | _ -> add_tdecl acc th
56

Francois Bobot's avatar
Francois Bobot committed
57
let lift_epsilon kind = Trans.fold (lift kind) None
58

Francois Bobot's avatar
Francois Bobot committed
59
let meta_epsilon = Theory.register_meta_excl "lift_epsilon" [MTstring]
60
  ~desc:"TODO"
Francois Bobot's avatar
Francois Bobot committed
61

62 63 64 65 66
let lift_epsilon = Trans.on_meta_excl meta_epsilon
  (fun alo ->
    let kind = match alo with
      | Some [MAstr "implicit"] -> Implicit
      | Some [MAstr "implied"] | None -> Implied
Francois Bobot's avatar
Francois Bobot committed
67 68 69
      | _ -> failwith "lift_epsilon accepts only 'implicit' and 'implied'"
    in
    lift_epsilon kind)
70

Francois Bobot's avatar
Francois Bobot committed
71
let () = Trans.register_transform "lift_epsilon" lift_epsilon
72
  ~desc:"TODO"