[snipe.git] / trans / trans.sml

(* L3 Compiler
 * AST -> IR Translator
 * Author: Kaustuv Chaudhuri <kaustuv+@cs.cmu.edu>
 * Modified by: Alex Vaynberg <alv@andrew.cmu.edu>
 * Modified: Frank Pfenning <fp@cs.cmu.edu>
 * Modified: Chris Lu <czl@andrew.cmu.edu>
 * Modified: Joshua Wise <jwise@andrew.cmu.edu>
 *)

signature TRANS =
sig
  (* translate abstract syntax tree to IR tree *)
  val translate : Ast.program -> Tree.program
end

structure Trans :> TRANS = 
struct

  structure A = Ast
  structure T = Tree
  
  fun trans_oper A.PLUS = T.ADD
    | trans_oper A.MINUS = T.SUB
    | trans_oper A.TIMES = T.MUL
    | trans_oper A.DIVIDEDBY = T.DIV
    | trans_oper A.MODULO = T.MOD
    | trans_oper A.LSH = T.LSH
    | trans_oper A.RSH = T.RSH
    | trans_oper A.LOGOR = T.LOGOR
    | trans_oper A.LOGAND = T.LOGAND
    | trans_oper A.BITOR = T.BITOR
    | trans_oper A.BITXOR = T.BITXOR
    | trans_oper A.BITAND = T.BITAND
    | trans_oper A.NEQ = T.NEQ
    | trans_oper A.EQ = T.EQ
    | trans_oper A.LT = T.LT
    | trans_oper A.LE = T.LE
    | trans_oper A.GE = T.GE
    | trans_oper A.GT = T.GT
    | trans_oper _ = raise ErrorMsg.InternalError "expected AST binop, got AST unop"

  fun translate (defs, funcs) =
    let
      val funclist = Symbol.elemsi funcs
      val _ = Type.alignment_reset()
      val _ = Type.sizeof_reset()
      fun sizeof a = Type.sizeof defs a
      fun alignment a = Type.alignment defs a
      fun align t curpos = Type.align defs t curpos

      fun offset_s id (Type.Typedef(id')) =
        let
          val shit = Symbol.look' defs id'
          fun eat (Type.Struct(l)) = l
            | eat (Type.MarkedTypedef(a)) = eat (Mark.data a)
          fun offset_s' ((id1,t)::l') curofs =
            let
              val a = align t curofs
            in
              if Symbol.compare(id,id1) = EQUAL
                then a
              else offset_s' l' (a + sizeof t)
            end
            | offset_s' nil _ = raise ErrorMsg.InternalError "looking for offset of something that isn't in the structure"
        in
          offset_s' (eat shit) 0
        end
        | offset_s _ _ = raise ErrorMsg.InternalError "cannot find offset into non-typedef"
      
      fun type_s id (Type.Typedef id') =
        let
          val td = 
            case Type.defdata (Symbol.look' defs id')
            of Type.Struct d => d
             | _ => raise ErrorMsg.InternalError "data didn't return struct"
          fun type_s' ((id',t)::l) =
            if (Symbol.compare (id, id') = EQUAL)
            then t
            else type_s' l
            | type_s' nil = raise ErrorMsg.InternalError "struct member not found in type_s"
        in
          type_s' td
        end
        | type_s id _ = raise ErrorMsg.InternalError "cannot find internal type non-typedef"
      
      fun deref (Type.Pointer i) = i
        | deref (Type.Array i) = i
        | deref _ = raise ErrorMsg.InternalError "cannot deref non-pointer"

      fun trans_unop A.NEGATIVE = T.NEG
        | trans_unop A.BITNOT = T.BITNOT
        | trans_unop A.BANG = T.BANG
        | trans_unop _ = raise ErrorMsg.InternalError "expected AST unop, got AST binop"
      
      fun typeof' vartypes exp = TypeChecker.typeof (defs, funcs) vartypes NONE exp

      fun trans_exp env vartypes (A.Var(id)) =
        (* after type-checking, id must be declared; do not guard lookup *)
            T.TEMP (Symbol.look' env id)
        | trans_exp env vartypes (A.ConstExp c) = T.CONST(c)
        | trans_exp env vartypes (A.OpExp(oper, [e1, e2])) =
            T.BINOP(trans_oper oper, trans_exp env vartypes e1, trans_exp env vartypes e2)
        | trans_exp env vartypes (A.OpExp(oper, [e])) =
            T.UNOP(trans_unop oper, trans_exp env vartypes e)
        | trans_exp env vartypes (A.OpExp(oper, _)) =
            raise ErrorMsg.InternalError "expected one or two operands, got it in the oven"
        | trans_exp env vartypes (A.Marked(marked_exp)) =
            trans_exp env vartypes (Mark.data marked_exp)
        | trans_exp env vartypes (A.FuncCall(func, stms)) =
            T.CALL(func,
              List.map
                (fn exp => (trans_exp env vartypes exp))
                stms)
        | trans_exp env vartypes (A.Member (exp, id)) =
            let
              val apk = T.BINOP (T.ADD, trans_exp env vartypes exp, T.CONST (Word32.fromInt (offset_s id (typeof' vartypes exp))))
              val tipo = type_s id (typeof' vartypes exp)
            in
	      if Type.issmall tipo
	      then T.MEMORY(apk)
	      else apk
	    end
        | trans_exp env vartypes (A.DerefMember (exp, id)) =
            trans_exp env vartypes (A.Member (A.Dereference (exp), id))
        | trans_exp env vartypes (A.Dereference(exp)) =
            if (Type.issmall (deref (typeof' vartypes exp)))
            then T.MEMORY(trans_exp env vartypes exp)
            else trans_exp env vartypes exp
        | trans_exp env vartypes (A.ArrIndex(exp1, exp2)) =
            let
              val asubk = T.BINOP(T.ADD, trans_exp env vartypes exp1, 
                                  T.BINOP(T.MUL, trans_exp env vartypes exp2,
                                          T.CONST(Word32.fromInt(sizeof (deref (typeof' vartypes exp1))))))
              val tipo = deref (typeof' vartypes exp1)
              val d =
                if not (Flag.isset Flags.safe)
                then asubk
                else T.COND (T.BINOP
                              (T.BE,
                               T.MEMORY (T.BINOP (
                                 T.SUB,
                                 trans_exp env vartypes exp1, 
                                 T.CONST 0w8)),
                               trans_exp env vartypes exp2),
                             T.NULLPTR,
                             asubk)
            in
              if Type.issmall tipo
              then T.MEMORY(d)
              else d
            end
        | trans_exp env vartypes (A.New(tipo)) =
            let
              val t1 = T.TEMP (Temp.new "result")
            in
              T.STMVAR (
                [T.MOVE (t1, T.ALLOC (T.CONST (Word32.fromInt(sizeof tipo)))),
                 T.EFFECT (T.MEMORY (t1))],
                t1)
            end
        | trans_exp env vartypes (A.NewArr(tipo, exp)) =
            let
              val size = T.BINOP(T.MUL, trans_exp env vartypes exp, T.CONST(Word32.fromInt(sizeof tipo)))
              val t1 = T.TEMP (Temp.new "allocated address")
              val ts = T.TEMP (Temp.new "size")
            in
              if not (Flag.isset Flags.safe)
              then T.STMVAR ([T.MOVE (t1, T.ALLOC size),
                              T.EFFECT (T.COND (T.BINOP (T.EQ, trans_exp env vartypes exp, T.CONST 0w0), T.CONST 0w0, T.MEMORY (t1)))],
                             t1)
              else T.COND (T.BINOP(T.EQ, size, T.CONST 0w0),
                           T.NULLPTR,
                           T.STMVAR (
                             [T.MOVE(t1,
                                T.COND(
                                  T.BINOP(T.LT, size, T.CONST 0w0),
                                  T.NULLPTR,
                                  T.ALLOC (T.BINOP (T.ADD, size, T.CONST 0w8)))
                                ),
                              T.MOVE(T.MEMORY (t1), trans_exp env vartypes exp)],
                             T.BINOP(T.ADD, t1, T.CONST 0w8)))
            end
        | trans_exp env vartypes (A.Null) = T.NULLPTR
        | trans_exp env vartypes (A.Conditional(c,e1,e2)) = T.COND(trans_exp env vartypes c, trans_exp env vartypes e1, trans_exp env vartypes e2)

        (* anything else should be impossible *)

      (* trans_stms : Temp.temp Symbol.table -> (Label.label * Label.label) option -> A.stm list -> Tree.stm list
       * translates a statement to the corresponding IR
       * we pass around the environment and the current loop context, if any
       * (usually called ls, which contains a continue label and a break label)
       *)
      fun trans_stms vars vartypes ls (A.Assign(e1,e2)::stms) = T.MOVE(trans_exp vars vartypes e1, trans_exp vars vartypes e2)::(trans_stms vars vartypes ls stms)
        | trans_stms vars vartypes ls (A.AsnOp(oop,e1,e2)::stms) =
          let
            val te1 = trans_exp vars vartypes e1
            val te2 = trans_exp vars vartypes e2
            val t1 = T.TEMP (Temp.new "memory deref cache")
          in
            case te1
            of T.MEMORY(m) => T.MOVE(t1, m) :: T.MOVE (T.MEMORY(t1), T.BINOP(trans_oper oop, T.MEMORY(t1), te2)) :: (trans_stms vars vartypes ls stms)
             | _ => T.MOVE(te1, T.BINOP(trans_oper oop, te1, te2)) :: (trans_stms vars vartypes ls stms)
          end
        | trans_stms vars vartypes ls (A.Return e::stms) =
          let
            val remainder = trans_stms vars vartypes ls stms
          in 
            T.RETURN (trans_exp vars vartypes e)
            :: remainder
          end
        | trans_stms vars vartypes ls (A.If(e, s, NONE)::stms) =
          let
            val l = Label.new ()
            val strans = trans_stms vars vartypes ls s
            val remainder = trans_stms vars vartypes ls stms
          in
            (T.JUMPIFN(trans_exp vars vartypes e, l)
            :: strans
            @ [T.LABEL (l)]
            @ remainder)
          end
        | trans_stms vars vartypes ls (A.If(e, s, SOME s2)::stms) =
          let
            val l = Label.new ()
            val l2 = Label.new ()
            val s1trans = trans_stms vars vartypes ls s
            val s2trans = trans_stms vars vartypes ls s2
            val remainder = trans_stms vars vartypes ls stms
          in
            (T.JUMPIFN(trans_exp vars vartypes e, l)
            :: s1trans
            @ [T.JUMP (l2), T.LABEL (l)]
            @ s2trans
            @ [T.LABEL (l2)]
            @ remainder)
          end
        | trans_stms vars vartypes ls (A.For(s1, e, s2, s)::stms) = 
          let
            val head = Label.new ()
            val tail = Label.new ()
            val loop = Label.new ()
            val stm1 = if isSome s1 then trans_stms vars vartypes NONE [valOf s1] else nil
            val strans = trans_stms vars vartypes (SOME(loop,tail)) s
            val stm2 = if isSome s2 then trans_stms vars vartypes NONE [valOf s2] else nil
            val remainder = trans_stms vars vartypes ls stms
          in
            (stm1
            @ [T.LABEL head, T.JUMPIFN(trans_exp vars vartypes e, tail)]
            @ strans
            @ [T.LABEL loop]
            @ stm2
            @ [T.JUMP head, T.LABEL tail]
            @ remainder)
          end
        | trans_stms vars vartypes ls (A.While(e, s)::stms) =
          let
            val head = Label.new ()
            val tail = Label.new ()
            val strans = trans_stms vars vartypes (SOME(head,tail)) s
            val remainder = trans_stms vars vartypes ls stms
          in
            (T.LABEL head
            :: T.JUMPIFN(trans_exp vars vartypes e, tail)
            :: strans
            @ [T.JUMP head, T.LABEL tail]
            @ remainder)
          end
        | trans_stms vars vartypes ls (A.Effect(e)::stms) = (T.EFFECT (trans_exp vars vartypes e)) :: (trans_stms vars vartypes ls stms)
        | trans_stms vars vartypes (SOME(b,e)) (A.Break::stms) =
          let
            val remainder = trans_stms vars vartypes (SOME(b,e)) stms
          in
            ((T.JUMP e) :: remainder)
          end
        | trans_stms vars vartypes  NONE       (A.Break::_) = raise ErrorMsg.InternalError "Break from invalid location... should have been caught in typechecker"
        | trans_stms vars vartypes (SOME(b,e)) (A.Continue::stms) =
          let
            val remainder = trans_stms vars vartypes (SOME(b,e)) stms
          in
            ((T.JUMP b) :: remainder)
          end
        | trans_stms vars vartypes  NONE       (A.Continue::_) = raise ErrorMsg.InternalError "Continue from invalid location... should have been caught in typechecker"
        | trans_stms vars vartypes ls (A.Nop::stms) = trans_stms vars vartypes ls stms
        | trans_stms vars vartypes ls (A.MarkedStm m :: stms) = trans_stms vars vartypes ls ((Mark.data m) :: stms)
        | trans_stms vars vartypes _ nil = nil

      fun trans_funcs ((id, A.Extern(_, _))::l) = trans_funcs l
        | trans_funcs ((id, A.MarkedFunction a)::l) = trans_funcs ((id, Mark.data a)::l)
        | trans_funcs ((id, A.Function(t, args, vars, body))::l) =
            let
              val allvars = foldr
                              (fn ((name, t),b) =>
                                Symbol.bind b (name, Temp.new (Symbol.name(name))))
                              Symbol.empty
                              (args @ vars)
              val vartypes = foldr (fn ((i, t), b) => Symbol.bind b (i, t)) Symbol.empty (args @ vars)
              val b = trans_stms allvars vartypes NONE body
              val (argn,_) = ListPair.unzip args
              val numberedargs = ListPair.zip (List.tabulate (length argn, fn x => x), argn)
              val argmv = map
                (fn (n, argname) => T.MOVE(T.TEMP (Symbol.look' allvars argname), T.ARG (n)))
                numberedargs
            in
              (T.FUNCTION(id, argmv @ b)) :: (trans_funcs l)
            end
        | trans_funcs nil = nil

    in
      trans_funcs funclist
    end
end
Commit	Line	Data
6ade8b0a	1	(* L3 Compiler
12aa4087 JW	2	* AST -> IR Translator
	3	* Author: Kaustuv Chaudhuri <kaustuv+@cs.cmu.edu>
	4	* Modified by: Alex Vaynberg <alv@andrew.cmu.edu>
	5	* Modified: Frank Pfenning <fp@cs.cmu.edu>
0a24e44d JW	6	* Modified: Chris Lu <czl@andrew.cmu.edu>
0a24e44d JW	7	* Modified: Joshua Wise <jwise@andrew.cmu.edu>
12aa4087 JW	8	*)
	9
	10	signature TRANS =
	11	sig
	12	(* translate abstract syntax tree to IR tree *)
5c79bb68	13	val translate : Ast.program -> Tree.program
12aa4087 JW	14	end
	15
	16	structure Trans :> TRANS =
	17	struct
	18
	19	structure A = Ast
	20	structure T = Tree
0a24e44d	21
12aa4087 JW	22	fun trans_oper A.PLUS = T.ADD
	23	\| trans_oper A.MINUS = T.SUB
	24	\| trans_oper A.TIMES = T.MUL
	25	\| trans_oper A.DIVIDEDBY = T.DIV
	26	\| trans_oper A.MODULO = T.MOD
0a24e44d JW	27	\| trans_oper A.LSH = T.LSH
	28	\| trans_oper A.RSH = T.RSH
	29	\| trans_oper A.LOGOR = T.LOGOR
	30	\| trans_oper A.LOGAND = T.LOGAND
	31	\| trans_oper A.BITOR = T.BITOR
	32	\| trans_oper A.BITXOR = T.BITXOR
	33	\| trans_oper A.BITAND = T.BITAND
	34	\| trans_oper A.NEQ = T.NEQ
	35	\| trans_oper A.EQ = T.EQ
	36	\| trans_oper A.LT = T.LT
	37	\| trans_oper A.LE = T.LE
	38	\| trans_oper A.GE = T.GE
	39	\| trans_oper A.GT = T.GT
	40	\| trans_oper _ = raise ErrorMsg.InternalError "expected AST binop, got AST unop"
12aa4087	41
1144856b	42	fun translate (defs, funcs) =
6ade8b0a	43	let
1144856b	44	val funclist = Symbol.elemsi funcs
5c79bb68 JW	45	val _ = Type.alignment_reset()
	46	val _ = Type.sizeof_reset()
	47	fun sizeof a = Type.sizeof defs a
	48	fun alignment a = Type.alignment defs a
	49	fun align t curpos = Type.align defs t curpos
1144856b	50
5c79bb68	51	fun offset_s id (Type.Typedef(id')) =
1144856b JW	52	let
1144856b JW	53	val shit = Symbol.look' defs id'
5c79bb68 JW	54	fun eat (Type.Struct(l)) = l
5c79bb68 JW	55	\| eat (Type.MarkedTypedef(a)) = eat (Mark.data a)
1144856b JW	56	fun offset_s' ((id1,t)::l') curofs =
	57	let
	58	val a = align t curofs
	59	in
	60	if Symbol.compare(id,id1) = EQUAL
	61	then a
5c79bb68	62	else offset_s' l' (a + sizeof t)
1144856b JW	63	end
	64	\| offset_s' nil _ = raise ErrorMsg.InternalError "looking for offset of something that isn't in the structure"
	65	in
	66	offset_s' (eat shit) 0
	67	end
	68	\| offset_s _ _ = raise ErrorMsg.InternalError "cannot find offset into non-typedef"
	69
5c79bb68	70	fun type_s id (Type.Typedef id') =
1144856b JW	71	let
1144856b JW	72	val td =
5c79bb68 JW	73	case Type.defdata (Symbol.look' defs id')
5c79bb68 JW	74	of Type.Struct d => d
1144856b JW	75	\| _ => raise ErrorMsg.InternalError "data didn't return struct"
	76	fun type_s' ((id',t)::l) =
	77	if (Symbol.compare (id, id') = EQUAL)
	78	then t
	79	else type_s' l
	80	\| type_s' nil = raise ErrorMsg.InternalError "struct member not found in type_s"
	81	in
	82	type_s' td
	83	end
	84	\| type_s id _ = raise ErrorMsg.InternalError "cannot find internal type non-typedef"
	85
5c79bb68 JW	86	fun deref (Type.Pointer i) = i
5c79bb68 JW	87	\| deref (Type.Array i) = i
1144856b	88	\| deref _ = raise ErrorMsg.InternalError "cannot deref non-pointer"
6ade8b0a JW	89
	90	fun trans_unop A.NEGATIVE = T.NEG
	91	\| trans_unop A.BITNOT = T.BITNOT
	92	\| trans_unop A.BANG = T.BANG
	93	\| trans_unop _ = raise ErrorMsg.InternalError "expected AST unop, got AST binop"
1144856b JW	94
1144856b JW	95	fun typeof' vartypes exp = TypeChecker.typeof (defs, funcs) vartypes NONE exp
6ade8b0a	96
1144856b	97	fun trans_exp env vartypes (A.Var(id)) =
6ade8b0a JW	98	(* after type-checking, id must be declared; do not guard lookup *)
6ade8b0a JW	99	T.TEMP (Symbol.look' env id)
1144856b JW	100	\| trans_exp env vartypes (A.ConstExp c) = T.CONST(c)
	101	\| trans_exp env vartypes (A.OpExp(oper, [e1, e2])) =
	102	T.BINOP(trans_oper oper, trans_exp env vartypes e1, trans_exp env vartypes e2)
	103	\| trans_exp env vartypes (A.OpExp(oper, [e])) =
	104	T.UNOP(trans_unop oper, trans_exp env vartypes e)
	105	\| trans_exp env vartypes (A.OpExp(oper, _)) =
6ade8b0a	106	raise ErrorMsg.InternalError "expected one or two operands, got it in the oven"
1144856b JW	107	\| trans_exp env vartypes (A.Marked(marked_exp)) =
	108	trans_exp env vartypes (Mark.data marked_exp)
	109	\| trans_exp env vartypes (A.FuncCall(func, stms)) =
	110	T.CALL(func,
	111	List.map
141a7120 JW	112	(fn exp => (trans_exp env vartypes exp))
141a7120 JW	113	stms)
1144856b JW	114	\| trans_exp env vartypes (A.Member (exp, id)) =
	115	let
	116	val apk = T.BINOP (T.ADD, trans_exp env vartypes exp, T.CONST (Word32.fromInt (offset_s id (typeof' vartypes exp))))
5c79bb68	117	val tipo = type_s id (typeof' vartypes exp)
1144856b	118	in
5c79bb68	119	if Type.issmall tipo
141a7120	120	then T.MEMORY(apk)
1144856b JW	121	else apk
	122	end
	123	\| trans_exp env vartypes (A.DerefMember (exp, id)) =
	124	trans_exp env vartypes (A.Member (A.Dereference (exp), id))
	125	\| trans_exp env vartypes (A.Dereference(exp)) =
5c79bb68	126	if (Type.issmall (deref (typeof' vartypes exp)))
141a7120	127	then T.MEMORY(trans_exp env vartypes exp)
1144856b JW	128	else trans_exp env vartypes exp
	129	\| trans_exp env vartypes (A.ArrIndex(exp1, exp2)) =
	130	let
	131	val asubk = T.BINOP(T.ADD, trans_exp env vartypes exp1,
	132	T.BINOP(T.MUL, trans_exp env vartypes exp2,
5c79bb68 JW	133	T.CONST(Word32.fromInt(sizeof (deref (typeof' vartypes exp1))))))
	134	val tipo = deref (typeof' vartypes exp1)
	135	val d =
	136	if not (Flag.isset Flags.safe)
	137	then asubk
	138	else T.COND (T.BINOP
	139	(T.BE,
	140	T.MEMORY (T.BINOP (
	141	T.SUB,
	142	trans_exp env vartypes exp1,
141a7120	143	T.CONST 0w8)),
5c79bb68 JW	144	trans_exp env vartypes exp2),
	145	T.NULLPTR,
	146	asubk)
1144856b	147	in
5c79bb68	148	if Type.issmall tipo
141a7120	149	then T.MEMORY(d)
5c79bb68	150	else d
1144856b JW	151	end
1144856b JW	152	\| trans_exp env vartypes (A.New(tipo)) =
5c79bb68	153	let
141a7120	154	val t1 = T.TEMP (Temp.new "result")
5c79bb68 JW	155	in
	156	T.STMVAR (
	157	[T.MOVE (t1, T.ALLOC (T.CONST (Word32.fromInt(sizeof tipo)))),
141a7120	158	T.EFFECT (T.MEMORY (t1))],
5c79bb68 JW	159	t1)
5c79bb68 JW	160	end
1144856b	161	\| trans_exp env vartypes (A.NewArr(tipo, exp)) =
5c79bb68 JW	162	let
5c79bb68 JW	163	val size = T.BINOP(T.MUL, trans_exp env vartypes exp, T.CONST(Word32.fromInt(sizeof tipo)))
141a7120 JW	164	val t1 = T.TEMP (Temp.new "allocated address")
141a7120 JW	165	val ts = T.TEMP (Temp.new "size")
5c79bb68 JW	166	in
	167	if not (Flag.isset Flags.safe)
	168	then T.STMVAR ([T.MOVE (t1, T.ALLOC size),
141a7120	169	T.EFFECT (T.COND (T.BINOP (T.EQ, trans_exp env vartypes exp, T.CONST 0w0), T.CONST 0w0, T.MEMORY (t1)))],
5c79bb68 JW	170	t1)
	171	else T.COND (T.BINOP(T.EQ, size, T.CONST 0w0),
	172	T.NULLPTR,
	173	T.STMVAR (
	174	[T.MOVE(t1,
	175	T.COND(
	176	T.BINOP(T.LT, size, T.CONST 0w0),
	177	T.NULLPTR,
	178	T.ALLOC (T.BINOP (T.ADD, size, T.CONST 0w8)))
	179	),
141a7120	180	T.MOVE(T.MEMORY (t1), trans_exp env vartypes exp)],
5c79bb68 JW	181	T.BINOP(T.ADD, t1, T.CONST 0w8)))
	182	end
	183	\| trans_exp env vartypes (A.Null) = T.NULLPTR
	184	\| trans_exp env vartypes (A.Conditional(c,e1,e2)) = T.COND(trans_exp env vartypes c, trans_exp env vartypes e1, trans_exp env vartypes e2)
12aa4087	185
6ade8b0a JW	186	(* anything else should be impossible *)
	187
	188	(* trans_stms : Temp.temp Symbol.table -> (Label.label * Label.label) option -> A.stm list -> Tree.stm list
	189	* translates a statement to the corresponding IR
	190	* we pass around the environment and the current loop context, if any
	191	* (usually called ls, which contains a continue label and a break label)
	192	*)
5c79bb68	193	fun trans_stms vars vartypes ls (A.Assign(e1,e2)::stms) = T.MOVE(trans_exp vars vartypes e1, trans_exp vars vartypes e2)::(trans_stms vars vartypes ls stms)
1144856b	194	\| trans_stms vars vartypes ls (A.AsnOp(oop,e1,e2)::stms) =
6ade8b0a	195	let
1144856b JW	196	val te1 = trans_exp vars vartypes e1
1144856b JW	197	val te2 = trans_exp vars vartypes e2
141a7120	198	val t1 = T.TEMP (Temp.new "memory deref cache")
6ade8b0a	199	in
1144856b	200	case te1
141a7120	201	of T.MEMORY(m) => T.MOVE(t1, m) :: T.MOVE (T.MEMORY(t1), T.BINOP(trans_oper oop, T.MEMORY(t1), te2)) :: (trans_stms vars vartypes ls stms)
5c79bb68	202	\| _ => T.MOVE(te1, T.BINOP(trans_oper oop, te1, te2)) :: (trans_stms vars vartypes ls stms)
6ade8b0a	203	end
1144856b	204	\| trans_stms vars vartypes ls (A.Return e::stms) =
6ade8b0a	205	let
1144856b	206	val remainder = trans_stms vars vartypes ls stms
6ade8b0a	207	in
141a7120	208	T.RETURN (trans_exp vars vartypes e)
6ade8b0a JW	209	:: remainder
6ade8b0a JW	210	end
1144856b	211	\| trans_stms vars vartypes ls (A.If(e, s, NONE)::stms) =
6ade8b0a JW	212	let
6ade8b0a JW	213	val l = Label.new ()
1144856b JW	214	val strans = trans_stms vars vartypes ls s
1144856b JW	215	val remainder = trans_stms vars vartypes ls stms
6ade8b0a	216	in
1144856b	217	(T.JUMPIFN(trans_exp vars vartypes e, l)
0a24e44d JW	218	:: strans
0a24e44d JW	219	@ [T.LABEL (l)]
6ade8b0a JW	220	@ remainder)
6ade8b0a JW	221	end
1144856b	222	\| trans_stms vars vartypes ls (A.If(e, s, SOME s2)::stms) =
6ade8b0a JW	223	let
6ade8b0a JW	224	val l = Label.new ()
0a24e44d	225	val l2 = Label.new ()
1144856b JW	226	val s1trans = trans_stms vars vartypes ls s
	227	val s2trans = trans_stms vars vartypes ls s2
	228	val remainder = trans_stms vars vartypes ls stms
6ade8b0a	229	in
1144856b	230	(T.JUMPIFN(trans_exp vars vartypes e, l)
0a24e44d JW	231	:: s1trans
	232	@ [T.JUMP (l2), T.LABEL (l)]
	233	@ s2trans
	234	@ [T.LABEL (l2)]
6ade8b0a JW	235	@ remainder)
6ade8b0a JW	236	end
1144856b	237	\| trans_stms vars vartypes ls (A.For(s1, e, s2, s)::stms) =
6ade8b0a JW	238	let
	239	val head = Label.new ()
	240	val tail = Label.new ()
	241	val loop = Label.new ()
1144856b JW	242	val stm1 = if isSome s1 then trans_stms vars vartypes NONE [valOf s1] else nil
	243	val strans = trans_stms vars vartypes (SOME(loop,tail)) s
	244	val stm2 = if isSome s2 then trans_stms vars vartypes NONE [valOf s2] else nil
	245	val remainder = trans_stms vars vartypes ls stms
6ade8b0a JW	246	in
6ade8b0a JW	247	(stm1
1144856b	248	@ [T.LABEL head, T.JUMPIFN(trans_exp vars vartypes e, tail)]
6ade8b0a JW	249	@ strans
	250	@ [T.LABEL loop]
	251	@ stm2
	252	@ [T.JUMP head, T.LABEL tail]
	253	@ remainder)
	254	end
1144856b	255	\| trans_stms vars vartypes ls (A.While(e, s)::stms) =
6ade8b0a JW	256	let
	257	val head = Label.new ()
	258	val tail = Label.new ()
1144856b JW	259	val strans = trans_stms vars vartypes (SOME(head,tail)) s
1144856b JW	260	val remainder = trans_stms vars vartypes ls stms
6ade8b0a JW	261	in
6ade8b0a JW	262	(T.LABEL head
1144856b	263	:: T.JUMPIFN(trans_exp vars vartypes e, tail)
6ade8b0a JW	264	:: strans
	265	@ [T.JUMP head, T.LABEL tail]
	266	@ remainder)
	267	end
5c79bb68	268	\| trans_stms vars vartypes ls (A.Effect(e)::stms) = (T.EFFECT (trans_exp vars vartypes e)) :: (trans_stms vars vartypes ls stms)
1144856b	269	\| trans_stms vars vartypes (SOME(b,e)) (A.Break::stms) =
6ade8b0a	270	let
1144856b	271	val remainder = trans_stms vars vartypes (SOME(b,e)) stms
6ade8b0a JW	272	in
	273	((T.JUMP e) :: remainder)
	274	end
1144856b JW	275	\| trans_stms vars vartypes NONE (A.Break::_) = raise ErrorMsg.InternalError "Break from invalid location... should have been caught in typechecker"
1144856b JW	276	\| trans_stms vars vartypes (SOME(b,e)) (A.Continue::stms) =
6ade8b0a	277	let
1144856b	278	val remainder = trans_stms vars vartypes (SOME(b,e)) stms
6ade8b0a JW	279	in
	280	((T.JUMP b) :: remainder)
	281	end
1144856b JW	282	\| trans_stms vars vartypes NONE (A.Continue::_) = raise ErrorMsg.InternalError "Continue from invalid location... should have been caught in typechecker"
	283	\| trans_stms vars vartypes ls (A.Nop::stms) = trans_stms vars vartypes ls stms
	284	\| trans_stms vars vartypes ls (A.MarkedStm m :: stms) = trans_stms vars vartypes ls ((Mark.data m) :: stms)
	285	\| trans_stms vars vartypes _ nil = nil
12aa4087	286
1144856b JW	287	fun trans_funcs ((id, A.Extern(_, _))::l) = trans_funcs l
	288	\| trans_funcs ((id, A.MarkedFunction a)::l) = trans_funcs ((id, Mark.data a)::l)
	289	\| trans_funcs ((id, A.Function(t, args, vars, body))::l) =
6ade8b0a	290	let
1144856b JW	291	val allvars = foldr
1144856b JW	292	(fn ((name, t),b) =>
141a7120	293	Symbol.bind b (name, Temp.new (Symbol.name(name))))
1144856b JW	294	Symbol.empty
	295	(args @ vars)
	296	val vartypes = foldr (fn ((i, t), b) => Symbol.bind b (i, t)) Symbol.empty (args @ vars)
	297	val b = trans_stms allvars vartypes NONE body
6ade8b0a JW	298	val (argn,_) = ListPair.unzip args
	299	val numberedargs = ListPair.zip (List.tabulate (length argn, fn x => x), argn)
	300	val argmv = map
141a7120	301	(fn (n, argname) => T.MOVE(T.TEMP (Symbol.look' allvars argname), T.ARG (n)))
6ade8b0a JW	302	numberedargs
	303	in
	304	(T.FUNCTION(id, argmv @ b)) :: (trans_funcs l)
	305	end
	306	\| trans_funcs nil = nil
1144856b	307
6ade8b0a	308	in
1144856b	309	trans_funcs funclist
6ade8b0a	310	end
12aa4087	311	end