matita/components/ng_disambiguation/grafiteDisambiguate.ml

   1 (*
   2  *
   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.
   6  *
   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.
  11  *
  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.
  16  *
  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,
  20  * MA  02111-1307, USA.
  21  *
  22  * For details, see the HELM World-Wide-Web page,
  23  * http://helm.cs.unibo.it/
  24  *)
  25
  26 (* $Id$ *)
  27
  28 type db = {
  29   aliases: GrafiteAst.alias_spec DisambiguateTypes.Environment.t;
  30   multi_aliases: GrafiteAst.alias_spec list DisambiguateTypes.Environment.t;
  31   new_aliases: (DisambiguateTypes.domain_item * GrafiteAst.alias_spec) list
  32 }
  33
  34 let initial_status = {
  35   aliases = DisambiguateTypes.Environment.empty;
  36   multi_aliases = DisambiguateTypes.Environment.empty;
  37   new_aliases = []
  38 }
  39
  40 class type g_status =
  41   object
  42    inherit Interpretations.g_status
  43    method disambiguate_db: db
  44   end
  45
  46 class status =
  47  object (self)
  48   inherit Interpretations.status
  49   val disambiguate_db = initial_status
  50   method disambiguate_db = disambiguate_db
  51   method set_disambiguate_db v = {< disambiguate_db = v >}
  52   method set_disambiguate_status
  53    : 'status. #g_status as 'status -> 'self
  54       = fun o -> ((self#set_interp_status o)#set_disambiguate_db o#disambiguate_db)
  55  end
  56
  57 let eval_with_new_aliases status f =
  58  let status =
  59   status#set_disambiguate_db { status#disambiguate_db with new_aliases = [] } in
  60  let res = f status in
  61  let new_aliases = status#disambiguate_db.new_aliases in
  62   new_aliases,res
  63 ;;
  64
  65 let dump_aliases out msg status =
  66    out (if msg = "" then "aliases dump:" else msg ^ ": aliases dump:");
  67    DisambiguateTypes.Environment.iter (fun _ x -> out (GrafiteAstPp.pp_alias x))
  68     status#disambiguate_db.aliases
  69
  70 let set_proof_aliases status ~implicit_aliases mode new_aliases =
  71  if mode = GrafiteAst.WithoutPreferences then
  72    status
  73  else
  74    let aliases =
  75     List.fold_left (fun acc (d,c) -> DisambiguateTypes.Environment.add d c acc)
  76      status#disambiguate_db.aliases new_aliases in
  77    let multi_aliases =
  78     List.fold_left (fun acc (d,c) ->
  79       DisambiguateTypes.Environment.cons GrafiteAst.description_of_alias
  80          d c acc)
  81      status#disambiguate_db.multi_aliases new_aliases
  82    in
  83    let new_status =
  84     {multi_aliases = multi_aliases ;
  85      aliases = aliases;
  86      new_aliases =
  87       (if implicit_aliases then new_aliases else []) @
  88         status#disambiguate_db.new_aliases}
  89    in
  90     status#set_disambiguate_db new_status
  91
  92 exception BaseUriNotSetYet
  93
  94 let singleton msg = function
  95   | [x], _ -> x
  96   | l, _   ->
  97       let debug =
  98          Printf.sprintf "GrafiteDisambiguate.singleton (%s): %u interpretations"
  99          msg (List.length l)
 100       in
 101       prerr_endline debug; assert false
 102
 103 let __Implicit = "__Implicit__"
 104 let __Closed_Implicit = "__Closed_Implicit__"
 105
 106 let ncic_mk_choice status = function
 107   | GrafiteAst.Symbol_alias (name, _, dsc) ->
 108      if name = __Implicit then
 109        dsc, `Sym_interp (fun _ -> NCic.Implicit `Term)
 110      else if name = __Closed_Implicit then
 111        dsc, `Sym_interp (fun _ -> NCic.Implicit `Closed)
 112      else
 113        DisambiguateChoices.lookup_symbol_by_dsc status
 114         ~mk_implicit:(function
 115            | true -> NCic.Implicit `Closed
 116            | false -> NCic.Implicit `Term)
 117         ~mk_appl:(function
 118            (NCic.Appl l)::tl -> NCic.Appl (l@tl) | l -> NCic.Appl l)
 119         ~term_of_nref:(fun nref -> NCic.Const nref)
 120        name dsc
 121   | GrafiteAst.Number_alias (_, dsc) ->
 122      let desc,f = DisambiguateChoices.nlookup_num_by_dsc dsc in
 123       desc, `Num_interp
 124        (fun num -> match f with `Num_interp f -> f num | _ -> assert false)
 125   | GrafiteAst.Ident_alias (name, uri) ->
 126      uri, `Sym_interp
 127       (fun l->assert(l = []);
 128         let nref = NReference.reference_of_string uri in
 129          NCic.Const nref)
 130 ;;
 131
 132
 133 let mk_implicit b =
 134   match b with
 135   | false ->
 136       GrafiteAst.Symbol_alias (__Implicit,-1,"Fake Implicit")
 137   | true ->
 138       GrafiteAst.Symbol_alias (__Closed_Implicit,-1,"Fake Closed Implicit")
 139 ;;
 140
 141 let nlookup_in_library
 142   interactive_user_uri_choice input_or_locate_uri item
 143 =
 144   match item with
 145   | DisambiguateTypes.Id id ->
 146      (try
 147        let references = NCicLibrary.resolve id in
 148         List.map
 149          (fun u -> GrafiteAst.Ident_alias (id,NReference.string_of_reference u)
 150          ) references
 151       with
 152        NCicEnvironment.ObjectNotFound _ -> [])
 153   | _ -> []
 154 ;;
 155
 156 let fix_instance item l =
 157  match item with
 158     DisambiguateTypes.Symbol (_,n) ->
 159      List.map
 160       (function
 161           GrafiteAst.Symbol_alias (s,_,d) -> GrafiteAst.Symbol_alias (s,n,d)
 162         | _ -> assert false
 163       ) l
 164   | DisambiguateTypes.Num n ->
 165      List.map
 166       (function
 167           GrafiteAst.Number_alias (_,d) -> GrafiteAst.Number_alias (n,d)
 168         | _ -> assert false
 169       ) l
 170   | DisambiguateTypes.Id _ -> l
 171 ;;
 172
 173
 174 let disambiguate_nterm estatus expty context metasenv subst thing
 175 =
 176   let diff, metasenv, subst, cic =
 177     singleton "first"
 178       (NCicDisambiguate.disambiguate_term
 179         ~rdb:estatus
 180         ~aliases:estatus#disambiguate_db.aliases
 181         ~expty
 182         ~universe:(Some estatus#disambiguate_db.multi_aliases)
 183         ~lookup_in_library:nlookup_in_library
 184         ~mk_choice:(ncic_mk_choice estatus)
 185         ~mk_implicit ~fix_instance
 186         ~description_of_alias:GrafiteAst.description_of_alias
 187         ~context ~metasenv ~subst thing)
 188   in
 189   let estatus =
 190    set_proof_aliases estatus ~implicit_aliases:true GrafiteAst.WithPreferences
 191     diff
 192   in
 193    metasenv, subst, estatus, cic
 194 ;;
 195
 196
 197 type pattern =
 198   NotationPt.term Disambiguate.disambiguator_input option *
 199   (string * NCic.term) list * NCic.term option
 200
 201 let disambiguate_npattern (text, prefix_len, (wanted, hyp_paths, goal_path)) =
 202   let interp path = NCicDisambiguate.disambiguate_path path in
 203   let goal_path = HExtlib.map_option interp goal_path in
 204   let hyp_paths = List.map (fun (name, path) -> name, interp path) hyp_paths in
 205   let wanted =
 206     match wanted with None -> None | Some x -> Some (text,prefix_len,x)
 207   in
 208    (wanted, hyp_paths, goal_path)
 209 ;;
 210
 211 let disambiguate_reduction_kind text prefix_len lexicon_status_ref = function
 212   | `Unfold (Some t) -> assert false (* MATITA 1.0 *)
 213   | `Normalize
 214   | `Simpl
 215   | `Unfold None
 216   | `Whd as kind -> kind
 217 ;;
 218
 219 let disambiguate_auto_params
 220   disambiguate_term metasenv context (oterms, params)
 221 =
 222   match oterms with
 223     | None -> metasenv, (None, params)
 224     | Some terms ->
 225         let metasenv, terms =
 226           List.fold_right
 227             (fun t (metasenv, terms) ->
 228                let metasenv,t = disambiguate_term context metasenv t in
 229                  metasenv,t::terms) terms (metasenv, [])
 230         in
 231           metasenv, (Some terms, params)
 232 ;;
 233
 234 let disambiguate_just disambiguate_term context metasenv =
 235  function
 236     `Term t ->
 237       let metasenv,t = disambiguate_term context metasenv t in
 238        metasenv, `Term t
 239   | `Auto params ->
 240       let metasenv,params = disambiguate_auto_params disambiguate_term metasenv
 241        context params
 242       in
 243        metasenv, `Auto params
 244 ;;
 245
 246 let disambiguate_nobj estatus ?baseuri (text,prefix_len,obj) =
 247   let uri =
 248    let baseuri =
 249      match baseuri with Some x -> x | None -> raise BaseUriNotSetYet
 250    in
 251    let name =
 252      match obj with
 253      | NotationPt.Inductive (_,(name,_,_,_)::_)
 254      | NotationPt.Record (_,name,_,_) -> name ^ ".ind"
 255      | NotationPt.Theorem (_,name,_,_,_) -> name ^ ".con"
 256      | NotationPt.Inductive _ -> assert false
 257    in
 258      NUri.uri_of_string (baseuri ^ "/" ^ name)
 259   in
 260   let diff, _, _, cic =
 261    singleton "third"
 262     (NCicDisambiguate.disambiguate_obj
 263       ~lookup_in_library:nlookup_in_library
 264       ~description_of_alias:GrafiteAst.description_of_alias
 265       ~mk_choice:(ncic_mk_choice estatus)
 266       ~mk_implicit ~fix_instance
 267       ~uri
 268       ~rdb:estatus
 269       ~aliases:estatus#disambiguate_db.aliases
 270       ~universe:(Some estatus#disambiguate_db.multi_aliases)
 271       (text,prefix_len,obj)) in
 272   let estatus =
 273    set_proof_aliases estatus ~implicit_aliases:true GrafiteAst.WithPreferences
 274     diff
 275   in
 276    estatus, cic
 277 ;;
 278
 279 let disambiguate_cic_appl_pattern status args =
 280  let rec disambiguate =
 281   function
 282     NotationPt.ApplPattern l ->
 283      NotationPt.ApplPattern (List.map disambiguate l)
 284   | NotationPt.VarPattern id
 285      when not
 286       (List.exists
 287        (function (NotationPt.IdentArg (_,id')) -> id'=id) args)
 288      ->
 289       let item = DisambiguateTypes.Id id in
 290        begin
 291         try
 292          match
 293           DisambiguateTypes.Environment.find item
 294            status#disambiguate_db.aliases
 295          with
 296             GrafiteAst.Ident_alias (_, uri) ->
 297              NotationPt.NRefPattern (NReference.reference_of_string uri)
 298           | _ -> assert false
 299         with Not_found ->
 300          prerr_endline
 301           ("LexiconEngine.eval_command: domain item not found: " ^
 302           (DisambiguateTypes.string_of_domain_item item));
 303          dump_aliases prerr_endline "" status;
 304          raise
 305           (Failure
 306            ((DisambiguateTypes.string_of_domain_item item) ^ " not found"))
 307              end
 308   | p -> p
 309  in
 310   disambiguate
 311 ;;
 312
 313 let aliases_for_objs refs =
 314  List.concat
 315   (List.map
 316     (fun nref ->
 317       let references = NCicLibrary.aliases_of nref in
 318        List.map
 319         (fun u ->
 320           let name = NCicPp.r2s true u in
 321            DisambiguateTypes.Id name,
 322             GrafiteAst.Ident_alias (name,NReference.string_of_reference u)
 323         ) references) refs)