33 lines
939 B
OCaml
33 lines
939 B
OCaml
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
|