X-Git-Url: https://git.xandkar.net/?a=blobdiff_plain;f=compiler%2Fsrc%2Flib%2Ftiger%2Ftiger_semant.ml;h=f809509ca7e66be3a4fe2a2d6aa6cc6bc4a60e67;hb=cc540a7e2dfcee4411953075210a64de874b91e5;hp=8d0ca11a4d8c6afadcc92a7bcf743b46a51e10f1;hpb=cbb4ffb6428f52eaea42f6039332af0c392cdd0f;p=tiger.ml.git diff --git a/compiler/src/lib/tiger/tiger_semant.ml b/compiler/src/lib/tiger/tiger_semant.ml index 8d0ca11..f809509 100644 --- a/compiler/src/lib/tiger/tiger_semant.ml +++ b/compiler/src/lib/tiger/tiger_semant.ml @@ -1,9 +1,13 @@ module List = ListLabels module A = Tiger_absyn +module Dag = Tiger_dag module Env = Tiger_env module E = Tiger_error -module Symbol = Tiger_symbol +module Escape = Tiger_semant_escape +module Pos = Tiger_position +module Sym = Tiger_symbol +module Temp = Tiger_temp module Translate = Tiger_translate module Type = Tiger_env_type module Value = Tiger_env_value @@ -78,6 +82,37 @@ end = struct let check_int expty ~pos : unit = check_same return_int expty ~pos + let paths_of_typedecs typedecs : (Sym.t * Sym.t * Pos.t) list list = + let (path, paths) = + List.fold_left typedecs ~init:([], []) ~f:( + fun (path, paths) (A.TypeDec {name=child; ty; pos}) -> + match ty with + | A.NameTy {symbol=parent; _} -> + (((parent, child, pos) :: path), paths) + | A.RecordTy _ + | A.ArrayTy _ -> + ([], path :: paths) + ) + in + List.map (path :: paths) ~f:List.rev + + let check_cycles (typedecs : A.typedec list) : unit = + let non_empty_paths = + List.filter + (paths_of_typedecs typedecs) + ~f:(function [] -> false | _ -> true) + in + List.iter non_empty_paths ~f:( + fun path -> + match Dag.of_list (List.map path ~f:(fun (p, c, _) -> (p, c))) with + | Ok _ -> + () + | Error `Cycle -> + let (_, from_id, from_pos) = List.hd path in + let (_, to_id, to_pos) = List.hd (List.rev path) in + E.raise (E.Cycle_in_type_decs {from_id; from_pos; to_id; to_pos}) + ) + let rec transExp ~env exp = let rec trexp exp = (match exp with @@ -89,11 +124,21 @@ end = struct return_string | A.CallExp {func; args; pos} -> (match env_get_val ~sym:func ~env ~pos with - | Value.Fun {formals; result} -> - List.iter2 formals args ~f:(fun ty_expected exp_given -> - check_same (return (actual_ty ~pos ty_expected)) (trexp exp_given) ~pos; - ); - return (actual_ty ~pos result) + | Value.Fun {formals; result; level=_; label=_} -> + let expected = List.length formals in + let given = List.length args in + if given = expected then + begin + List.iter2 formals args ~f:(fun ty_expected exp_given -> + check_same + (return (actual_ty ~pos ty_expected)) + (trexp exp_given) + ~pos; + ); + return (actual_ty ~pos result) + end + else + E.raise (E.Wrong_number_of_args {func; expected; given; pos}) | Value.Var _ -> E.raise (E.Id_not_a_function {id=func; pos}) ) @@ -149,16 +194,28 @@ end = struct | A.WhileExp {test; body; pos} -> (* test : must be int, because we have no bool *) check_int (trexp test) ~pos; - ignore (trexp body); (* Only care if a type-error is raised *) + let (loop, env) = Env.loop_begin env in + (* Only care if an error is raised *) + ignore (transExp ~env body); + ignore (Env.loop_end env loop); return_unit | A.ForExp {var; lo; hi; body; pos; escape=_} -> check_int (trexp lo) ~pos; check_int (trexp hi) ~pos; - (* Only care if a type-error is raised *) - let env = Env.set_val env var (Value.Var {ty = Type.Int}) in + let (loop, env) = Env.loop_begin env in + let level = Env.level_get env in + (* Assuming all escape, for now *) + let access = Translate.alloc_local ~level ~escapes:true in + let env = Env.set_val env var (Value.Var {ty = Type.Int; access}) in + (* Only care if an error is raised *) ignore (transExp ~env body); + ignore (Env.loop_end env loop); return_unit - | A.BreakExp _ -> + | A.BreakExp pos -> + (match Env.loop_current env with + | Some _ -> () + | None -> E.raise (E.Break_outside_loop pos) + ); return_unit | A.LetExp {decs; body; pos=_} -> (* (1) decs augment env *) @@ -186,8 +243,10 @@ end = struct (function | A.SimpleVar {symbol=sym; pos} -> (match env_get_val ~sym ~env ~pos with - | Value.Fun _ -> E.raise (E.Id_is_a_function {id=sym; pos}) - | Value.Var {ty} -> return (actual_ty ~pos ty) + | Value.Fun _ -> + E.raise (E.Id_is_a_function {id=sym; pos}) + | Value.Var {ty; access=_} -> + return (actual_ty ~pos ty) ) | A.FieldVar {var; symbol; pos} -> let {exp=_; ty} = trvar var in @@ -274,18 +333,24 @@ end = struct ty ) in - Env.set_val env name (Value.Var {ty}) + 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=ty_exp; pos}) -> - let ty = transTy ~env ty_exp in - (match env_get_typ ~sym:name ~env ~pos with - | Type.Name (name, ty_opt_ref) -> + 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 @@ -298,30 +363,53 @@ end = struct ); env | A.FunDecs fundecs -> - List.fold_left fundecs ~init:env ~f:( - fun env (A.FunDec {name; params; result; body; pos=_}) -> - let (env_for_body, formals_in_reverse_order) = - List.fold_left params ~init:(env, []) ~f:( - fun (env, formals) (A.Field {name; escape=_; typ; pos}) -> - let ty = env_get_typ_actual ~env ~sym:typ ~pos in - let env = Env.set_val env name (Value.Var {ty}) in - (env, ty :: formals) + 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 - (* ignore because we only care if an exception is raised *) - ignore (transExp ~env:env_for_body body); - let formals = List.rev formals_in_reverse_order in - let result = - match result with - | None -> - Type.Unit - | Some (sym, pos) -> - env_get_typ_actual ~sym ~env ~pos - in - Env.set_val env name (Value.Fun {formals; result}) - ) + (* we only care if an exception is raised *) + ignore (transExp ~env:env_with_fun_heads_and_local_vars body); + ); + env_with_fun_heads_only ) - and transTy ~(env : Env.t) (ty_exp : A.ty) : Type.t = + and transTy ~(env : Env.t) ~ty_name ~(ty_exp : A.ty) : Type.t = (match ty_exp with | A.NameTy {symbol=sym; pos} -> env_get_typ ~sym ~env ~pos @@ -332,15 +420,16 @@ end = struct (name, ty) ) in - Type.new_record fields + Type.new_record ~name:ty_name ~fields | A.ArrayTy {symbol=sym; pos} -> let element_ty = env_get_typ ~sym ~env ~pos in - Type.new_array element_ty + Type.new_array ~name:ty_name ~ty:element_ty ) end open Semant let transProg absyn = + Escape.find ~prog:absyn; let {exp = _; ty = _} = transExp absyn ~env:Env.base in ()