ligo/typecheck.ml

33 lines
939 B
OCaml
Raw Normal View History

2019-03-01 21:54:41 +04:00
open AST
open Region
module SMap = Map.Make(String)
(* open Sanity: *)
let (|>) v f = f v
let (@@) f v = f v
let map f l = List.rev (List.rev_map f l)
let type_decl_to_tenv td =
td
|> List.map (fun (_, name, _, type_expr) -> (name, type_expr))
|> List.to_seq |> SMap.of_seq
let shadow name typ env =
SMap.update name (function None -> Some [typ] | Some tl -> Some (typ :: tl)) env
let shadow_list name_typ_list env =
List.fold_left (fun acc (name, typ) -> shadow name typ acc) env name_typ_list
let tc ast =
(* te is the type environment, ve is the variable environment *)
let te = type_decl_to_tenv ast#types in
let ve =
SMap.empty
|> (match ast#parameter.value with (_,name,_,ty) -> shadow name ty)
|> shadow "storage" (snd ast#storage.value)
|> shadow "operations" (snd ast#operations.value)
(* |> shadow_list @@ map (function FunDecl {value} -> value.var.value, ) lambdas *)
in
te, ve