helm/software/components/ng_tactics/nTacStatus.ml

   1 (*
   2     ||M||  This file is part of HELM, an Hypertextual, Electronic
   3     ||A||  Library of Mathematics, developed at the Computer Science
   4     ||T||  Department, University of Bologna, Italy.
   5     ||I||
   6     ||T||  HELM is free software; you can redistribute it and/or
   7     ||A||  modify it under the terms of the GNU General Public License
   8     \   /  version 2 or (at your option) any later version.
   9      \ /   This software is distributed as is, NO WARRANTY.
  10       V_______________________________________________________________ *)
  11
  12 (* $Id: nCic.ml 9058 2008-10-13 17:42:30Z tassi $ *)
  13
  14 exception Error of string lazy_t * exn option
  15 let fail ?exn msg = raise (Error (msg,exn))
  16
  17 let wrap f x =
  18   try f x
  19   with
  20   | MultiPassDisambiguator.DisambiguationError _
  21   | NCicRefiner.RefineFailure _
  22   | NCicUnification.UnificationFailure _
  23   | NCicTypeChecker.TypeCheckerFailure _
  24   | NCicMetaSubst.MetaSubstFailure _ as exn -> fail ~exn (lazy "")
  25 ;;
  26
  27 class pstatus =
  28  fun (o: NCic.obj) ->
  29   object
  30    inherit NEstatus.status
  31    val obj = o
  32    method obj = obj
  33    method set_obj o = {< obj = o >}
  34   end
  35
  36 type tactic_term = CicNotationPt.term Disambiguate.disambiguator_input
  37 type tactic_pattern = GrafiteAst.npattern Disambiguate.disambiguator_input
  38
  39 let pp_status status =
  40   prerr_endline (NCicPp.ppobj status#obj)
  41 ;;
  42
  43 type cic_term = NCic.conjecture (* attrs, context, term *)
  44 let ctx_of (_,c,_) = c ;;
  45
  46 let relocate status destination (name,source,t as orig) =
  47  if source == destination then status, orig else
  48   let u, d, metasenv, subst, o = status#obj in
  49   let rec lcp ctx j i = function
  50     | (n1, NCic.Decl t1 as e)::cl1, (n2, NCic.Decl t2)::cl2 ->
  51         if n1 = n2 &&
  52            NCicReduction.are_convertible ctx ~subst ~metasenv t1 t2 then
  53           NCic.Rel i :: lcp (e::ctx)(j-1) (i-1) (cl1,cl2)
  54         else
  55           HExtlib.mk_list (NCic.Appl
  56             [NCic.Sort NCic.Prop; NCic.Sort NCic.Prop]) j
  57     | (n1, NCic.Def (b1,t1) as e)::cl1, (n2, NCic.Def (b2,t2))::cl2 ->
  58         if n1 = n2 &&
  59            NCicReduction.are_convertible ctx ~subst ~metasenv t1 t2 &&
  60            NCicReduction.are_convertible ctx ~subst ~metasenv b1 b2 then
  61           NCic.Rel i :: lcp (e::ctx)(j-1) (i-1) (cl1,cl2)
  62         else
  63           HExtlib.mk_list (NCic.Appl
  64             [NCic.Sort NCic.Prop; NCic.Sort NCic.Prop]) j
  65     | (n1, NCic.Def (b1,t1) as e)::cl1, (n2, NCic.Decl t2)::cl2 ->
  66         if n1 = n2 &&
  67            NCicReduction.are_convertible ctx ~subst ~metasenv t1 t2 then
  68           NCic.Rel i :: lcp (e::ctx)(j-1) (i-1) (cl1,cl2)
  69         else
  70           HExtlib.mk_list (NCic.Appl
  71             [NCic.Sort NCic.Prop; NCic.Sort NCic.Prop]) j
  72     | (n1, NCic.Decl _)::cl1, (n2, NCic.Def _)::cl2 -> assert false
  73     | _::_, [] ->
  74           HExtlib.mk_list (NCic.Appl
  75             [NCic.Sort NCic.Prop; NCic.Sort NCic.Prop]) j
  76     | _ -> []
  77   in
  78   let lc =
  79     lcp [] (List.length destination) (List.length source)
  80       (List.rev destination, List.rev source)
  81   in
  82   let lc = (0,NCic.Ctx (List.rev lc)) in
  83   let (metasenv, subst), t =
  84     NCicMetaSubst.delift
  85        ~unify:(fun m s c t1 t2 ->
  86          try Some (NCicUnification.unify status m s c t1 t2)
  87          with
  88           | NCicUnification.UnificationFailure _
  89           | NCicUnification.Uncertain _ -> None)
  90      metasenv subst source 0 lc t
  91   in
  92   let status = status#set_obj (u, d, metasenv, subst, o) in
  93   status, (name, destination, t)
  94 ;;
  95 let relocate a b c = wrap (relocate a b) c;;
  96
  97 let term_of_cic_term s t c =
  98   let s, (_,_,t) = relocate s c t in
  99   s, t
 100 ;;
 101
 102 let ppterm status t =
 103  let uri,height,metasenv,subst,obj = status#obj in
 104  let _,context,t = t in
 105   NCicPp.ppterm ~metasenv ~subst ~context t
 106 ;;
 107
 108 let disambiguate status t ty context =
 109  let status, expty =
 110    match ty with
 111    | None -> status, None
 112    | Some ty ->
 113        let status, (_,_,x) = relocate status context ty in status, Some x
 114  in
 115  let uri,height,metasenv,subst,obj = status#obj in
 116  let metasenv, subst, status, t =
 117    GrafiteDisambiguate.disambiguate_nterm expty status context metasenv subst t
 118  in
 119  let new_pstatus = uri,height,metasenv,subst,obj in
 120   status#set_obj new_pstatus, ([], context, t)
 121 ;;
 122 let disambiguate a b c d = wrap (disambiguate a b c) d;;
 123
 124 let typeof status ctx t =
 125   let status, (_,_,t) = relocate status ctx t in
 126   let _,_,metasenv,subst,_ = status#obj in
 127   let ty = NCicTypeChecker.typeof ~subst ~metasenv ctx t in
 128   status, ([], ctx, ty)
 129 ;;
 130 let typeof a b c = wrap (typeof a b) c;;
 131
 132 let whd status ?delta ctx t =
 133   let status, (name,_,t) = relocate status ctx t in
 134   let _,_,_,subst,_ = status#obj in
 135   let t = NCicReduction.whd ~subst ?delta ctx t in
 136   status, (name, ctx, t)
 137 ;;
 138
 139 let normalize status ?delta ctx t =
 140   let status, (name,_,t) = relocate status ctx t in
 141   let _,_,_,subst,_ = status#obj in
 142   let t = NCicTacReduction.normalize ~subst ?delta ctx t in
 143   status, (name, ctx, t)
 144 ;;
 145
 146 let unify status ctx a b =
 147   let status, (_,_,a) = relocate status ctx a in
 148   let status, (_,_,b) = relocate status ctx b in
 149   let n,h,metasenv,subst,o = status#obj in
 150   let metasenv, subst = NCicUnification.unify status metasenv subst ctx a b in
 151    status#set_obj (n,h,metasenv,subst,o)
 152 ;;
 153 let unify a b c d = wrap (unify a b c) d;;
 154
 155 let fix_sorts (name,ctx,t) =
 156  let f () =
 157   let t = NCicUnification.fix_sorts t in
 158    name,ctx,t
 159  in
 160   wrap f ()
 161 ;;
 162
 163 let refine status ctx term expty =
 164   let status, (nt,_,term) = relocate status ctx term in
 165   let status, ne, expty =
 166     match expty with
 167       None -> status, [], None
 168     | Some e ->
 169         let status, (n,_, e) = relocate status ctx e in status, n, Some e
 170   in
 171   let name,height,metasenv,subst,obj = status#obj in
 172   let metasenv,subst,t,ty =
 173    NCicRefiner.typeof status metasenv subst ctx term expty
 174   in
 175    status#set_obj (name,height,metasenv,subst,obj), (nt,ctx,t), (ne,ctx,ty)
 176 ;;
 177 let refine a b c d = wrap (refine a b c) d;;
 178
 179 let get_goalty status g =
 180  let _,_,metasenv,_,_ = status#obj in
 181  List.assoc g metasenv
 182 ;;
 183
 184 let instantiate status i t =
 185  let (gname, context, _ as gty) = get_goalty status i in
 186  let status, (_,_,t), (_,_,ty) =
 187    refine status (ctx_of gty) t (Some gty)
 188  in
 189
 190  let name,height,metasenv,subst,obj = status#obj in
 191  let metasenv = List.filter (fun j,_ -> j <> i) metasenv in
 192  let subst = (i, (gname, context, t, ty)) :: subst in
 193   status#set_obj (name,height,metasenv,subst,obj)
 194 ;;
 195
 196 let mk_meta status ?(attrs=[]) ctx bo_or_ty =
 197   match bo_or_ty with
 198   | `Decl ty ->
 199       let status, (_,_,ty) = relocate status ctx ty in
 200       let n,h,metasenv,subst,o = status#obj in
 201       let metasenv, _, instance, _ =
 202         NCicMetaSubst.mk_meta ~attrs metasenv ctx (`WithType ty)
 203       in
 204       let status = status#set_obj (n,h,metasenv,subst,o) in
 205       status, ([],ctx,instance)
 206   | `Def bo ->
 207       let status, (_,_,bo_ as bo) = relocate status ctx bo in
 208       let status, (_,_,ty) = typeof status ctx bo in
 209       let n,h,metasenv,subst,o = status#obj in
 210       let metasenv, metano, instance, _ =
 211         NCicMetaSubst.mk_meta ~attrs metasenv ctx (`WithType ty) in
 212       let metasenv = List.filter (fun j,_ -> j <> metano) metasenv in
 213       let subst = (metano, (attrs, ctx, bo_, ty)) :: subst in
 214       let status = status#set_obj (n,h,metasenv,subst,o) in
 215       status, ([],ctx,instance)
 216 ;;
 217
 218 let mk_in_scope status t =
 219   mk_meta status ~attrs:[`InScope] (ctx_of t) (`Def t)
 220 ;;
 221
 222 let mk_out_scope n status t =
 223   mk_meta status ~attrs:[`OutScope n] (ctx_of t) (`Def t)
 224 ;;
 225
 226 (* the following unification problem will be driven by
 227  *   select s ~found:mk_in_scope ~postprocess:(mk_out_scope argsno) t pattern
 228  *
 229  *   ? args = t
 230  *
 231  *   where argsn = length args and the pattern matches t
 232  *
 233  *  found is called on every selected term to map them
 234  *  postprocess is called on the entire term after selection
 235  *)
 236 let select_term
 237   low_status ~found ~postprocess (name,context,term) (wanted,path)
 238 =
 239   let is_found status ctx t wanted =
 240     (* we could lift wanted step-by-step *)
 241     try true, unify status ctx ([], ctx, t) wanted
 242     with
 243     | Error (_, Some (NCicUnification.UnificationFailure _))
 244     | Error (_, Some (NCicUnification.Uncertain _)) -> false, status
 245   in
 246   let match_term status ctx (wanted : cic_term) t =
 247     let rec aux ctx (status,already_found) t =
 248       let b, status = is_found status ctx t wanted in
 249       if b then
 250          let status , (_,_,t) = found status ([], ctx, t) in
 251          (status,true),t
 252       else
 253         let _,_,_,subst,_ = status#obj in
 254         match t with
 255         | NCic.Meta (i,lc) when List.mem_assoc i subst ->
 256             let _,_,t,_ = NCicUtils.lookup_subst i subst in
 257             aux ctx (status,already_found) t
 258         | NCic.Meta _ -> (status,already_found),t
 259         | _ ->
 260           NCicUntrusted.map_term_fold_a (fun e c -> e::c) ctx aux
 261            (status,already_found) t
 262      in
 263        aux ctx (status,false) t
 264   in
 265   let _,_,_,subst,_ = low_status#obj in
 266   let rec select status ctx pat cic =
 267     match pat, cic with
 268     | _, NCic.Meta (i,lc) when List.mem_assoc i subst ->
 269         let cic =
 270           let _,_,t,_ = NCicUtils.lookup_subst i subst in
 271           NCicSubstitution.subst_meta lc t
 272         in
 273         select status ctx pat cic
 274     | NCic.LetIn (_,t1,s1,b1), NCic.LetIn (n,t2,s2,b2) ->
 275         let status, t = select status ctx t1 t2 in
 276         let status, s = select status ctx s1 s2 in
 277         let ctx = (n, NCic.Def (s2,t2)) :: ctx in
 278         let status, b = select status ctx b1 b2 in
 279         status, NCic.LetIn (n,t,s,b)
 280     | NCic.Lambda (_,s1,t1), NCic.Lambda (n,s2,t2) ->
 281         let status, s = select status ctx s1 s2 in
 282         let ctx = (n, NCic.Decl s2) :: ctx in
 283         let status, t = select status ctx t1 t2 in
 284         status, NCic.Lambda (n,s,t)
 285     | NCic.Prod (_,s1,t1), NCic.Prod (n,s2,t2) ->
 286         let status, s = select status ctx s1 s2 in
 287         let ctx = (n, NCic.Decl s2) :: ctx in
 288         let status, t = select status ctx t1 t2 in
 289         status, NCic.Prod (n,s,t)
 290     | NCic.Appl l1, NCic.Appl l2 ->
 291         let status, l =
 292            List.fold_left2
 293              (fun (status,l) x y ->
 294               let status, x = select status ctx x y in
 295               status, x::l)
 296              (status,[]) l1 l2
 297         in
 298         status, NCic.Appl (List.rev l)
 299     | NCic.Match (_,ot1,t1,pl1), NCic.Match (u,ot2,t2,pl2) ->
 300         let status, t = select status ctx t1 t2 in
 301         let status, ot = select status ctx ot1 ot2 in
 302         let status, pl =
 303            List.fold_left2
 304              (fun (status,l) x y ->
 305               let status, x = select status ctx x y in
 306               status, x::l)
 307              (status,[]) pl1 pl2
 308         in
 309         status, NCic.Match (u,ot,t,List.rev pl)
 310     | NCic.Implicit `Hole, t ->
 311         (match wanted with
 312         | Some wanted ->
 313              let status', wanted = disambiguate status wanted None ctx in
 314              let (status',found), t' = match_term status' ctx wanted t in
 315               if found then status',t' else status,t
 316         | None ->
 317            let (status,_),t = match_term status ctx ([],ctx,t) t in
 318             status,t)
 319     | NCic.Implicit _, t -> status, t
 320     | _,t ->
 321         fail (lazy ("malformed pattern: " ^ NCicPp.ppterm ~metasenv:[]
 322           ~context:[] ~subst:[] pat ^ " against " ^
 323           NCicPp.ppterm ~metasenv:[] ~subst:[] ~context:[] t))
 324   in
 325   let status, term = select low_status context path term in
 326   let term = (name, context, term) in
 327   postprocess status term
 328 ;;
 329
 330 let analyse_indty status ty =
 331  let status, reduct = whd status (ctx_of ty) ty in
 332  let ref, args =
 333   match reduct with
 334    | _,_,NCic.Const ref -> ref, []
 335    | _,_,NCic.Appl (NCic.Const ref :: args) -> ref, args
 336    | _,_,_ -> fail (lazy ("not an inductive type")) in
 337  let _,lno,tl,_,i = NCicEnvironment.get_checked_indtys ref in
 338  let _,_,_,cl = List.nth tl i in
 339  let consno = List.length cl in
 340  let left, right = HExtlib.split_nth lno args in
 341  status, (ref, consno, left, right)
 342 ;;
 343
 344 let mk_cic_term c t = [],c,t ;;
 345
 346 let apply_subst status ctx t =
 347  let status, (name,_,t) = relocate status ctx t in
 348  let _,_,_,subst,_ = status#obj in
 349   status, (name, ctx, NCicUntrusted.apply_subst subst ctx t)
 350 ;;
 351
 352 (* ============= move this elsewhere ====================*)
 353
 354 class ['stack] status =
 355  fun (o: NCic.obj) (s: 'stack) ->
 356   object
 357    inherit (pstatus o)
 358    val stack = s
 359    method stack = stack
 360    method set_stack s = {< stack = s >}
 361   end
 362
 363 class type lowtac_status = [unit] status
 364
 365 type 'status lowtactic = #lowtac_status as 'status -> int -> 'status
 366
 367 class type tac_status = [Continuationals.Stack.t] status
 368
 369 type 'status tactic = #tac_status as 'status -> 'status
 370