+ | A.VarDec {name; typ=typ_opt; init; pos=pos_outter; escape=_} ->
+ let ty =
+ (match (typ_opt, transExp ~env init) with
+ | None, {ty; exp=()} ->
+ ty
+ | Some (sym, pos_inner), expty_init ->
+ let ty = env_get_typ_actual ~sym ~env ~pos:pos_inner in
+ check_same (return ty) expty_init ~pos:pos_outter;
+ ty
+ )
+ in
+ let access =
+ Translate.alloc_local
+ ~level:(Env.level_get env)
+ ~escapes:true (* Assuming all escape, for now... *)
+ in
+ Env.set_val env name (Value.Var {ty; access})
+ | A.TypeDecs typedecs ->
+ check_cycles typedecs;
+ let env =
+ List.fold_left typedecs ~init:env ~f:(
+ fun env (A.TypeDec {name; ty=_; pos=_}) ->
+ Env.set_typ env name (Type.Name (name, ref None))
+ )
+ in
+ List.iter typedecs ~f:(fun (A.TypeDec {name=ty_name; ty=ty_exp; pos}) ->
+ let ty = transTy ~env ~ty_name ~ty_exp in
+ (match env_get_typ ~sym:ty_name ~env ~pos with
+ | Type.Name (_, ty_opt_ref) ->
+ ty_opt_ref := Some ty
+ | Type.Unit
+ | Type.Nil
+ | Type.Int
+ | Type.String
+ | Type.Record _
+ | Type.Array _ ->
+ ()
+ )
+ );
+ env
+ | A.FunDecs fundecs ->
+ let env_with_fun_heads_only =
+ List.fold_left fundecs ~init:env ~f:(
+ fun env (A.FunDec {name; params; result; body=_; pos=_}) ->
+ let formals =
+ List.map params ~f:(
+ fun (A.Field {name=_; typ; pos; escape=_}) ->
+ env_get_typ_actual ~env ~sym:typ ~pos
+ )
+ in
+ let result =
+ match result with
+ | Some (s, p) -> env_get_typ_actual ~sym:s ~env ~pos:p
+ | None -> Type.Unit
+ in
+ let label = Temp.Label.gen () in
+ let level =
+ Translate.Level.next
+ (Env.level_get env)
+ ~name:label
+ (* Assuming all escape (for now) *)
+ ~formals:(List.map formals ~f:(fun _ -> true))
+ in
+ let env = Env.level_set env level in
+ Env.set_val env name (Value.Fun {formals; result; level; label})
+ )
+ in
+ List.iter fundecs ~f:(
+ fun (A.FunDec {name=_; params; result=_; body; pos=_}) ->
+ let env_with_fun_heads_and_local_vars =
+ List.fold_left params ~init:env_with_fun_heads_only ~f:(
+ fun env (A.Field {name=var_name; escape=_; typ; pos}) ->
+ let var_ty = env_get_typ_actual ~env ~sym:typ ~pos in
+ let level = Env.level_get env in
+ (* Assuming all escape, for now *)
+ let access = Translate.alloc_local ~level ~escapes:true in
+ Env.set_val
+ env
+ var_name
+ (Value.Var {ty = var_ty; access})
+ )
+ in
+ (* we only care if an exception is raised *)
+ ignore (transExp ~env:env_with_fun_heads_and_local_vars body);
+ );
+ env_with_fun_heads_only