WIP type-checking -- check function call expression

[tiger.ml.git] / compiler / src / lib / tiger / tiger_semant.ml

diff --git a/compiler/src/lib/tiger/tiger_semant.ml b/compiler/src/lib/tiger/tiger_semant.ml
index 659bb85..40407ef 100644 (file)
--- a/compiler/src/lib/tiger/tiger_semant.ml
+++ b/compiler/src/lib/tiger/tiger_semant.ml
@@ -56,6 +56,8 @@ end = struct
   let check_int expty ~pos : unit =
     check_same {exp=(); ty=Type.Int} expty ~pos
 
+  (* TODO: actual_ty *)
+
   let rec transExp ~env exp =
     let rec trexp exp =
       (match exp with
@@ -65,8 +67,16 @@ end = struct
           return_int
       | A.StringExp {string=_; _} ->
           return_string
-      | A.CallExp {func=_; args=_; pos=_} ->
-          unimplemented ()
+      | 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 {exp=(); ty = ty_expected} (trexp exp_given) ~pos;
+              );
+              return result
+          | Value.Var _ ->
+              E.raise (E.Id_not_a_function {id=func; pos})
+          )
       | A.OpExp {oper; left; right; pos} ->
           trop oper ~left ~right ~pos
       | A.RecordExp {fields=_; typ=_; pos=_} ->
@@ -134,8 +144,13 @@ end = struct
               )
             )
             ~otherwise:(fun () -> E.raise (E.Exp_not_a_record {ty; pos}))
-      | A.SubscriptVar {var=_; exp=_; pos=_} ->
-          unimplemented ()
+      | A.SubscriptVar {var; exp; pos} ->
+          let {exp=_; ty} = trvar var in
+          check_int (trexp exp) ~pos;
+          Type.if_array
+            ty
+            ~f:(fun ty_elements -> return ty_elements)
+            ~otherwise:(fun () -> E.raise (E.Exp_not_an_array {ty; pos}))
       )
     and trop oper ~left ~right ~pos =
       let expty_left  = trexp left in