helm/ocaml/cic_notation/cicNotationEnv.ml

   1 (* Copyright (C) 2004-2005, HELM Team.
   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 open CicNotationPt
  27
  28 exception Value_not_found of string
  29
  30 type value =
  31   | TermValue of term
  32   | StringValue of string
  33   | NumValue of string
  34   | OptValue of value option
  35   | ListValue of value list
  36
  37 type value_type =
  38   | TermType
  39   | StringType
  40   | NumType
  41   | OptType of value_type
  42   | ListType of value_type
  43
  44 type declaration = string * value_type
  45 type binding = string * (value_type * value)
  46 type t = (string * (value_type * value)) list
  47
  48 let lookup_value env name =
  49   try
  50     snd (List.assoc name env)
  51   with Not_found -> raise (Value_not_found name)
  52
  53 let remove env name = List.remove_assoc name env
  54
  55 let lookup_term env name =
  56   match lookup_value env name with
  57   | TermValue x -> x
  58   | _ -> raise (Value_not_found name)
  59
  60 let lookup_num env name =
  61   match lookup_value env name with
  62   | NumValue x -> x
  63   | _ -> raise (Value_not_found name)
  64
  65 let lookup_string env name =
  66   match lookup_value env name with
  67   | StringValue x -> x
  68   | _ -> raise (Value_not_found name)
  69
  70 let opt_binding_some (n, (ty, v)) = (n, (OptType ty, OptValue (Some v)))
  71 let opt_binding_none (n, (ty, v)) = (n, (OptType ty, OptValue None))
  72 let opt_binding_of_name (n, ty) = (n, (OptType ty, OptValue None))
  73 let list_binding_of_name (n, ty) = (n, (ListType ty, ListValue []))
  74 let opt_declaration (n, ty) = (n, OptType ty)
  75 let list_declaration (n, ty) = (n, ListType ty)
  76
  77 let declaration_of_var = function
  78   | NumVar s -> s, NumType
  79   | IdentVar s -> s, StringType
  80   | TermVar s -> s, TermType
  81   | _ -> assert false
  82
  83 let value_of_term = function
  84   | Num (s, _) -> NumValue s
  85   | Ident (s, None) -> StringValue s
  86   | t -> TermValue t
  87
  88 let term_of_value = function
  89   | NumValue s -> Num (s, 0)
  90   | StringValue s -> Ident (s, None)
  91   | TermValue t -> t
  92   | _ -> assert false (* TO BE UNDERSTOOD *)
  93
  94 let rec well_typed ty value =
  95   match ty, value with
  96   | TermType, TermValue _
  97   | StringType, StringValue _
  98   | OptType _, OptValue None
  99   | NumType, NumValue _ -> true
 100   | OptType ty', OptValue (Some value') -> well_typed ty' value'
 101   | ListType ty', ListValue vl ->
 102       List.for_all (fun value' -> well_typed ty' value') vl
 103   | _ -> false
 104
 105 let declarations_of_env = List.map (fun (name, (ty, _)) -> (name, ty))
 106
 107 let coalesce_env declarations env_list =
 108   let env0 = List.map list_binding_of_name declarations in
 109   let grow_env_entry env n v =
 110     List.map
 111       (function
 112         | (n', (ty, ListValue vl)) as entry ->
 113             if n' = n then n', (ty, ListValue (v :: vl)) else entry
 114         | _ -> assert false)
 115       env
 116   in
 117   let grow_env env_i env =
 118     List.fold_left
 119       (fun env (n, (_, v)) -> grow_env_entry env n v)
 120       env env_i
 121   in
 122   List.fold_right grow_env env_list env0
 123