Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Diff of /branches/vis15/src/compiler/typechecker/check-expr.sml
ViewVC logotype

Diff of /branches/vis15/src/compiler/typechecker/check-expr.sml

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 3423, Fri Nov 13 14:38:59 2015 UTC revision 3424, Fri Nov 13 15:44:42 2015 UTC
# Line 16  Line 16 
16    (* type check a list of expressions *)    (* type check a list of expressions *)
17      val checkList : Env.t * Env.context * ParseTree.expr list -> (AST.expr list * Types.ty list)      val checkList : Env.t * Env.context * ParseTree.expr list -> (AST.expr list * Types.ty list)
18    
19      (* type check an iteration expression (i.e., "x 'in' expr"), returning the iterator
20       * and the environment extended with a binding for x.
21       *)
22        val checkIter : Env.t * Env.context * ParseTree.iterator -> ((AST.var * AST.expr) * Env.t)
23    
24    (* type check a dimension that is given by a constant expression *)    (* type check a dimension that is given by a constant expression *)
25      val checkDim : Env.t * Env.context * ParseTree.expr -> IntLit.t option      val checkDim : Env.t * Env.context * ParseTree.expr -> IntLit.t option
26    
# Line 412  Line 417 
417                      | _ => checkFieldApp (check (env, cxt, e))                      | _ => checkFieldApp (check (env, cxt, e))
418                    (* end case *)                    (* end case *)
419                  end                  end
420              | PT.E_Subscript(e, indices) => (case (check(env, cxt, e), indices)              | PT.E_Subscript(e, indices) => let
421                   of ((e', Ty.T_Sequence(elemTy, _)), [SOME e2]) => raise Fail "FIXME"                  fun expectedTensor ty = err(cxt, [
422                    | ((e', Ty.T_Tensor shape), _) => raise Fail "FIXME"                          S "expected tensor type for slicing, but found ", TY ty
423                    | ((_, ty), _) => err(cxt, [                        ])
424                          S "expected sequence or tensor type for object of subscripting, but found",                  fun chkIndex e = let
425                          TY ty                        val eTy as (_, ty) = check(env, cxt, e)
426                          in
427                            if Unify.equalType(ty, Ty.T_Int)
428                              then eTy
429                              else err (cxt, [
430                                  S "expected type 'int' for index, but found ", TY ty
431                        ])                        ])
432                  (* end case *))                        end
433                    in
434                      case (check(env, cxt, e), indices)
435                       of ((e', Ty.T_Error), _) => (
436                            List.app (ignore o Option.map chkIndex) indices;
437                            bogusExpTy)
438                        | ((e1', ty1 as Ty.T_Sequence(elemTy, optDim)), [SOME e2]) => let
439                            val (e2', ty2) = chkIndex e2
440                            val rator = if isSome optDim
441                                  then BV.subscript
442                                  else BV.dynSubscript
443                            val (tyArgs, Ty.T_Fun(domTy, rngTy)) = TU.instantiate(Var.typeOf rator)
444                            in
445                              if Unify.equalTypes(domTy, [ty1, ty2])
446                                then let
447                                  val exp = AST.E_Prim(rator, tyArgs, [e1', e2'], rngTy)
448                                  in
449                                    (exp, rngTy)
450                                  end
451                                else raise Fail "unexpected unification failure"
452                            end
453                        | ((e', ty as Ty.T_Sequence _), [NONE]) => expectedTensor ty
454                        | ((e', ty as Ty.T_Sequence _), _) => expectedTensor ty
455                        | ((e', Ty.T_Tensor shape), _) => raise Fail "FIXME: tensor slicing"
456                        | ((_, ty), _) => expectedTensor ty
457                      (* end case *)
458                    end
459              | PT.E_Select(e, field) => (case check(env, cxt, e)              | PT.E_Select(e, field) => (case check(env, cxt, e)
460                   of (e', Ty.T_Named strand) => (case Env.findStrand(env, strand)                   of (e', Ty.T_Named strand) => (case Env.findStrand(env, strand)
461                         of SOME sEnv => (case StrandEnv.findStateVar(sEnv, field)                         of SOME sEnv => (case StrandEnv.findStateVar(sEnv, field)
# Line 583  Line 619 
619                  ])                  ])
620            (* end case *))            (* end case *))
621    
622        and chkComprehension (env, cxt, PT.COMP_Mark m) =
623              chkComprehension(E.withEnvAndContext(env, cxt, m))
624          | chkComprehension (env, cxt, PT.COMP_Comprehension(e, [iter])) = let
625              val (iter', env') = checkIter (E.blockScope env, cxt, iter)
626              val (e', ty) = check (env', cxt, e)
627              val resTy = Ty.T_Sequence(ty, NONE)
628              in
629                (AST.E_Comprehension(e', iter', resTy), resTy)
630              end
631          | chkComprehension _ = raise Fail "impossible"
632    
633        and checkIter (env, cxt, PT.I_Mark m) = checkIter (E.withEnvAndContext (env, cxt, m))
634          | checkIter (env, cxt, PT.I_Iterator({span, tree=x}, e)) = (
635              case check (env, cxt, e)
636               of (e', ty as Ty.T_Sequence(elemTy, _)) => let
637                    val x' = Var.new(x, Error.location(#1 cxt, span), Var.LocalVar, elemTy)
638                    in
639                      ((x', e'), E.insertLocal(env, cxt, x, x'))
640                    end
641                | (e', ty) => let
642                    val x' = Var.new(x, Error.UNKNOWN, Var.IterVar, Ty.T_Error)
643                    in
644                      TypeError.error (cxt, [
645                          S "expected sequence type in iteration, but found '", TY ty, S "'"
646                        ]);
647                      ((x', bogusExp), E.insertLocal(env, cxt, x, x'))
648                    end
649              (* end case *))
650    
651    (* typecheck a list of expressions returning a list of AST expressions and a list    (* typecheck a list of expressions returning a list of AST expressions and a list
652     * of the types of the expressions.     * of the types of the expressions.
653     *)     *)

Legend:
Removed from v.3423  
changed lines
  Added in v.3424

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0