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

SCM Repository

[diderot] Diff of /branches/vis12/src/compiler/simplify/simplify.sml
ViewVC logotype

Diff of /branches/vis12/src/compiler/simplify/simplify.sml

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

revision 2140, Thu Feb 14 23:58:15 2013 UTC revision 2512, Thu Nov 21 16:22:48 2013 UTC
# Line 12  Line 12 
12    
13    end = struct    end = struct
14    
15      structure Ty = Types      structure TU = TypeUtil
16      structure S = Simple      structure S = Simple
17        structure VMap = Var.Map
18      structure InP = Inputs      structure InP = Inputs
19    
20      local      val cvtTy = SimpleTypes.simplify
21        val tempName = Atom.atom "_t"  
22        fun newTemp ty = SimpleVar.new ("_t", SimpleVar.LocalVar, ty)
23    
24      (* convert an AST variable to a Simple variable *)
25        fun cvtVar (env, x as Var.V{name, kind, ty=([], ty), ...}) = let
26              val x' = SimpleVar.new (name, kind, cvtTy ty)
27      in      in
28      fun newTemp ty = Var.new (tempName, AST.LocalVar, ty)              (x', VMap.insert(env, x, x'))
29      end      end
30    
31        fun cvtVars (env, xs) = List.foldr
32              (fn (x, (xs, env)) => let
33                val (x', env) = cvtVar(env, x)
34                in
35                  (x'::xs, env)
36                end) ([], env) xs
37    
38        fun lookupVar (env, x) = (case VMap.find (env, x)
39               of SOME x' => x'
40                | NONE => raise Fail(concat["lookupVar(", Var.uniqueNameOf x, ")"])
41              (* end case *))
42    
43    (* make a block out of a list of statements that are in reverse order *)    (* make a block out of a list of statements that are in reverse order *)
44      fun mkBlock stms = S.Block(List.rev stms)      fun mkBlock stms = S.Block(List.rev stms)
45    
# Line 36  Line 54 
54              | AST.E_Tuple es => raise Fail "E_Tuple not yet implemented"              | AST.E_Tuple es => raise Fail "E_Tuple not yet implemented"
55              | AST.E_Cons es => let              | AST.E_Cons es => let
56                  val shp = (case ty                  val shp = (case ty
57                         of Ty.T_Tensor(Ty.Shape shp) => List.map (fn (Ty.DimConst d) => d) shp                         of SimpleTypes.T_Tensor shp => shp
58                          | _ => raise Fail "not tensor type"                          | _ => raise Fail "not tensor type"
59                        (* end case *))                        (* end case *))
60                  fun flatten (AST.E_Lit(Literal.Int n), l) = FloatLit.fromInt n :: l                  fun flatten (AST.E_Lit(Literal.Int n), l) = FloatLit.fromInt n :: l
# Line 56  Line 74 
74              | _ => raise Fail "impossible initialization expression"              | _ => raise Fail "impossible initialization expression"
75            (* end case *))            (* end case *))
76    
77      fun inputImage (nrrd, dim, shape) = let      fun inputImage (nrrd, dim, shape) = (
78            val dim = TypeUtil.monoDim dim            case ImageInfo.fromNrrd(NrrdInfo.getInfo nrrd, dim, shape)
           val shp = TypeUtil.monoShape shape  
           in  
             case ImageInfo.fromNrrd(NrrdInfo.getInfo nrrd, dim, shp)  
79               of NONE => raise Fail(concat["nrrd file \"", nrrd, "\" does not have expected type"])               of NONE => raise Fail(concat["nrrd file \"", nrrd, "\" does not have expected type"])
80                | SOME info => InP.Proxy(nrrd, info)                | SOME info => InP.Proxy(nrrd, info)
81              (* end case *)            (* end case *))
           end  
82    
83    (* is the given statement's continuation the syntactically following statement? *)    (* is the given statement's continuation the syntactically following statement? *)
84      fun contIsNext (AST.S_Block stms) = List.all contIsNext stms      fun contIsNext (AST.S_Block stms) = List.all contIsNext stms
85          | contIsNext (AST.S_IfThenElse(_, s1, s2)) = contIsNext s1 orelse contIsNext s2
86        | contIsNext AST.S_Die = false        | contIsNext AST.S_Die = false
87        | contIsNext AST.S_Stabilize = false        | contIsNext AST.S_Stabilize = false
88        | contIsNext (AST.S_Return _) = false        | contIsNext (AST.S_Return _) = false
89        | contIsNext _ = true        | contIsNext _ = true
90    
91      fun simplifyProgram (AST.Program dcls) = let      fun simplifyProgram (AST.Program{props, decls}) = let
92            val inputs = ref []            val inputs = ref []
93            val globals = ref []            val globals = ref []
94            val globalInit = ref []            val globalInit = ref []
# Line 85  Line 100 
100  (* FIXME: the check for multiple initially decls should happen in type checking *)  (* FIXME: the check for multiple initially decls should happen in type checking *)
101                    | SOME _ => raise Fail "multiple initially declarations"                    | SOME _ => raise Fail "multiple initially declarations"
102                  (* end case *))                  (* end case *))
103            fun simplifyDecl dcl = (case dcl            fun simplifyDecl (dcl, env) = (case dcl
104                   of AST.D_Input(x, desc, NONE) => let                   of AST.D_Input(x, desc, NONE) => let
105                        val (ty, init) = (case Var.monoTypeOf x                        val (x', env) = cvtVar(env, x)
106                               of ty as Ty.T_Image{dim, shape} => let                        val (ty, init) = (case SimpleVar.typeOf x'
107                                    val info = ImageInfo.mkInfo(TypeUtil.monoDim dim, TypeUtil.monoShape shape)                               of ty as SimpleTypes.T_Image{dim, shape} => let
108                                      val info = ImageInfo.mkInfo(dim, shape)
109                                    in                                    in
110                                      (ty, SOME(InP.Image info))                                      (ty, SOME(InP.Image info))
111                                    end                                    end
# Line 97  Line 113 
113                              (* end case *))                              (* end case *))
114                        val inp = InP.INP{                        val inp = InP.INP{
115                                ty = ty,                                ty = ty,
116                                name = Var.nameOf x,                                name = SimpleVar.nameOf x',
117                                desc = desc,                                desc = desc,
118                                init = init                                init = init
119                              }                              }
120                        in                        in
121                          inputs := (x, inp) :: !inputs                          inputs := (x', inp) :: !inputs;
122                            env
123                        end                        end
124                    | AST.D_Input(x, desc, SOME(AST.E_LoadNrrd(tvs, nrrd, ty))) => let                    | AST.D_Input(x, desc, SOME(AST.E_LoadNrrd(tvs, nrrd, ty))) => let
125                          val (x', env) = cvtVar(env, x)
126                      (* load the nrrd proxy here *)                      (* load the nrrd proxy here *)
127                        val info = NrrdInfo.getInfo nrrd                        val info = NrrdInfo.getInfo nrrd
128                        val (ty, init) = (case Var.monoTypeOf x                        val (ty, init) = (case SimpleVar.typeOf x'
129                               of ty as Ty.T_DynSequence _ => (ty, InP.DynSeq nrrd)                               of ty as SimpleTypes.T_DynSequence _ => (ty, InP.DynSeq nrrd)
130                                | ty as Ty.T_Image{dim, shape} => (ty, inputImage(nrrd, dim, shape))                                | ty as SimpleTypes.T_Image{dim, shape} => (ty, inputImage(nrrd, dim, shape))
131                                | _ => raise Fail "impossible"                                | _ => raise Fail "impossible"
132                              (* end case *))                              (* end case *))
133                        val inp = InP.INP{                        val inp = InP.INP{
134                                ty = ty,                                ty = ty,
135                                name = Var.nameOf x,                                name = SimpleVar.nameOf x',
136                                desc = desc,                                desc = desc,
137                                init = SOME init                                init = SOME init
138                              }                              }
139                        in                        in
140                          inputs := (x, inp) :: !inputs                          inputs := (x', inp) :: !inputs;
141                            env
142                        end                        end
143                    | AST.D_Input(x, desc, SOME e) => let                    | AST.D_Input(x, desc, SOME e) => let
144                        val ty = Var.monoTypeOf x                        val (x', env) = cvtVar(env, x)
145                        val inp = InP.INP{                        val inp = InP.INP{
146                                ty = ty,                                ty = SimpleVar.typeOf x',
147                                name = Var.nameOf x,                                name = SimpleVar.nameOf x',
148                                desc = desc,                                desc = desc,
149                                init = SOME(expToInit(ty, e))                                init = SOME(expToInit(SimpleVar.typeOf x', e))
150                              }                              }
151                        in                        in
152                          inputs := (x, inp) :: !inputs                          inputs := (x', inp) :: !inputs;
153                            env
154                        end                        end
155                    | AST.D_Var(AST.VD_Decl(x, e)) => let                    | AST.D_Var(AST.VD_Decl(x, e)) => let
156                        val (stms, e') = simplifyExp (e, [])                        val (x', env) = cvtVar(env, x)
157                          val (stms, e') = simplifyExp (env, e, [])
158                        in                        in
159                          globals := x :: !globals;                          globals := x' :: !globals;
160                          globalInit := S.S_Assign(x, e') :: (stms @ !globalInit)                          globalInit := S.S_Assign(x', e') :: (stms @ !globalInit);
161                        end                          env
162                    | AST.D_Func(f, params, body) =>                        end
163                        funcs := S.Func{f=f, params=params, body=simplifyBlock body} :: !funcs                    | AST.D_Func(f, params, body) => let
164                    | AST.D_Strand info => strands := simplifyStrand info :: !strands                        val (f', env) = cvtVar(env, f)
165                    | AST.D_InitialArray(creat, iters) =>                        val (params', env) = cvtVars (env, params)
166                        setInitially (simplifyInit(true, creat, iters))                        val body' = simplifyBlock(env, body)
167                    | AST.D_InitialCollection(creat, iters) =>                        in
168                        setInitially (simplifyInit(false, creat, iters))                          funcs := S.Func{f=f', params=params', body=body'} :: !funcs;
169                            env
170                          end
171                      | AST.D_Strand info => (
172                          strands := simplifyStrand(env, info) :: !strands;
173                          env)
174                      | AST.D_InitialArray(creat, iters) => (
175                          setInitially (simplifyInit(env, true, creat, iters));
176                          env)
177                      | AST.D_InitialCollection(creat, iters) => (
178                          setInitially (simplifyInit(env, false, creat, iters));
179                          env)
180                  (* end case *))                  (* end case *))
181              val env = List.foldl simplifyDecl VMap.empty decls
182            in            in
             List.app simplifyDecl dcls;  
183              S.Program{              S.Program{
184                    props = props,
185                  inputs = List.rev(!inputs),                  inputs = List.rev(!inputs),
186                  globals = List.rev(!globals),                  globals = List.rev(!globals),
187                  globalInit = mkBlock (!globalInit),                  globalInit = mkBlock (!globalInit),
# Line 162  Line 195 
195                }                }
196            end            end
197    
198      and simplifyInit (isArray, AST.C_Create(strand, exps), iters) = let      and simplifyInit (env, isArray, AST.C_Create(strand, exps), iters) = let
199            val (stms, xs) = simplifyExpsToVars (exps, [])            fun simplifyIter (AST.I_Range(x, e1, e2), (env, iters, stms)) = let
200                    val (stms, lo) = simplifyExpToVar (env, e1, stms)
201                    val (stms, hi) = simplifyExpToVar (env, e2, stms)
202                    val (x', env) = cvtVar (env, x)
203                    in
204                      (env, {param=x', lo=lo, hi=hi}::iters, stms)
205                    end
206              val (env, iters, iterStms) = List.foldl simplifyIter (env, [], []) iters
207              val (stms, xs) = simplifyExpsToVars (env, exps, [])
208            val creat = S.C_Create{            val creat = S.C_Create{
209                    argInit = mkBlock stms,                    argInit = mkBlock stms,
210                    name = strand,                    name = strand,
211                    args = xs                    args = xs
212                  }                  }
           fun simplifyIter (AST.I_Range(x, e1, e2), (iters, stms)) = let  
                 val (stms, lo) = simplifyExpToVar (e1, stms)  
                 val (stms, hi) = simplifyExpToVar (e2, stms)  
                 in  
                   ({param=x, lo=lo, hi=hi}::iters, stms)  
                 end  
           val (iters, stms) = List.foldl simplifyIter ([], []) iters  
213            in            in
214              S.Initially{              S.Initially{
215                  isArray = isArray,                  isArray = isArray,
216                  rangeInit = mkBlock stms,                  rangeInit = mkBlock iterStms,
217                  iters = List.rev iters,                  iters = List.rev iters,
218                  create = creat                  create = creat
219                }                }
220            end            end
221    
222      and simplifyStrand {name, params, state, methods} = let      and simplifyStrand (env, AST.Strand{name, params, state, methods}) = let
223            fun simplifyState ([], xs, stms) = (List.rev xs, mkBlock stms)            val (params', env) = cvtVars (env, params)
224              | simplifyState (AST.VD_Decl(x, e) :: r, xs, stms) = let            fun simplifyState (env, [], xs, stms) = (List.rev xs, mkBlock stms, env)
225                  val (stms, e') = simplifyExp (e, stms)              | simplifyState (env, AST.VD_Decl(x, e) :: r, xs, stms) = let
226                    val (stms, e') = simplifyExp (env, e, stms)
227                    val (x', env) = cvtVar(env, x)
228                  in                  in
229                    simplifyState (r, x::xs, S.S_Assign(x, e') :: stms)                    simplifyState (env, r, x'::xs, S.S_Assign(x', e') :: stms)
230                  end                  end
231            val (xs, stm) = simplifyState (state, [], [])            val (xs, stm, env) = simplifyState (env, state, [], [])
232            in            in
233              S.Strand{              S.Strand{
234                  name = name,                  name = name,
235                  params = params,                  params = params',
236                  state = xs, stateInit = stm,                  state = xs, stateInit = stm,
237                  methods = List.map simplifyMethod methods                  methods = List.map (simplifyMethod env) methods
238                }                }
239            end            end
240    
241      and simplifyMethod (AST.M_Method(name, body)) =      and simplifyMethod env (AST.M_Method(name, body)) =
242            S.Method(name, simplifyBlock body)            S.Method(name, simplifyBlock(env, body))
243    
244    (* simplify a statement into a single statement (i.e., a block if it expands    (* simplify a statement into a single statement (i.e., a block if it expands
245     * into more than one new statement).     * into more than one new statement).
246     *)     *)
247      and simplifyBlock stm = mkBlock (simplifyStmt (stm, []))      and simplifyBlock (env, stm) = mkBlock (#1 (simplifyStmt (env, stm, [])))
248    
249      and simplifyStmt (stm, stms) = (case stm    (* simplify the statement stm where stms is a reverse-order list of preceeding simplified
250       * statements.  This function returns a reverse-order list of simplified statements.
251       * Note that error reporting is done in the typechecker, but it does not prune unreachable
252       * code.
253       *)
254        and simplifyStmt (env, stm, stms) = (case stm
255             of AST.S_Block body => let             of AST.S_Block body => let
256  (* FIXME: we should probably prune unreachable code in the typechecker and issue a warning! *)                  fun simplify (_, [], stms) = stms
257                  fun simplify ([], stms) = stms                    | simplify (env', stm::r, stms) = let
258                    | simplify (stm::r, stms) = if contIsNext stm                        val (stms, env') = simplifyStmt (env', stm, stms)
                       then simplify (r, simplifyStmt (stm, stms))  
                       else simplify (r, [])  (* prune unreachable statements *)  
259                  in                  in
260                    simplify (body, stms)                          if contIsNext stm
261                              then simplify (env', r, stms)
262                              else stms  (* prune the unreachable statements "r" *)
263                          end
264                    in
265                      (simplify (env, body, stms), env)
266                  end                  end
267              | AST.S_Decl(AST.VD_Decl(x, e)) => let              | AST.S_Decl(AST.VD_Decl(x, e)) => let
268                  val (stms, e') = simplifyExp (e, stms)                  val (stms, e') = simplifyExp (env, e, stms)
269                    val (x', env) = cvtVar(env, x)
270                  in                  in
271                    S.S_Assign(x, e') :: stms                    (S.S_Assign(x', e') :: stms, env)
272                  end                  end
273              | AST.S_IfThenElse(e, s1, s2) => let              | AST.S_IfThenElse(e, s1, s2) => let
274                  val (stms, x) = simplifyExpToVar (e, stms)                  val (stms, x) = simplifyExpToVar (env, e, stms)
275                  val s1 = simplifyBlock s1                  val s1 = simplifyBlock (env, s1)
276                  val s2 = simplifyBlock s2                  val s2 = simplifyBlock (env, s2)
277                  in                  in
278                    S.S_IfThenElse(x, s1, s2) :: stms                    (S.S_IfThenElse(x, s1, s2) :: stms, env)
279                  end                  end
280              | AST.S_Assign(x, e) => let              | AST.S_Assign(x, e) => let
281                  val (stms, e') = simplifyExp (e, stms)                  val (stms, e') = simplifyExp (env, e, stms)
282                  in                  in
283                    S.S_Assign(x, e') :: stms                    (S.S_Assign(lookupVar(env, x), e') :: stms, env)
284                  end                  end
285              | AST.S_New(name, args) => let              | AST.S_New(name, args) => let
286                  val (stms, xs) = simplifyExpsToVars (args, stms)                  val (stms, xs) = simplifyExpsToVars (env, args, stms)
287                  in                  in
288                    S.S_New(name, xs) :: stms                    (S.S_New(name, xs) :: stms, env)
289                  end                  end
290              | AST.S_Die => S.S_Die :: stms              | AST.S_Die => (S.S_Die :: stms, env)
291              | AST.S_Stabilize => S.S_Stabilize :: stms              | AST.S_Stabilize => (S.S_Stabilize :: stms, env)
292              | AST.S_Return e => let              | AST.S_Return e => let
293                  val (stms, x) = simplifyExpToVar (e, stms)                  val (stms, x) = simplifyExpToVar (env, e, stms)
294                  in                  in
295                    S.S_Return x :: stms                    (S.S_Return x :: stms, env)
296                  end                  end
297              | AST.S_Print args => let              | AST.S_Print args => let
298                  val (stms, xs) = simplifyExpsToVars (args, stms)                  val (stms, xs) = simplifyExpsToVars (env, args, stms)
299                  in                  in
300                    S.S_Print xs :: stms                    (S.S_Print xs :: stms, env)
301                  end                  end
302            (* end case *))            (* end case *))
303    
304      and simplifyExp (exp, stms) = (      and simplifyExp (env, exp, stms) = (
305            case exp            case exp
306             of AST.E_Var x => (case Var.kindOf x             of AST.E_Var x => (case Var.kindOf x
307                   of Var.BasisVar => let                   of Var.BasisVar => let
308                        val ty = Var.monoTypeOf x                        val ty = cvtTy(Var.monoTypeOf x)
309                        val x' = newTemp ty                        val x' = newTemp ty
310                        val stm = S.S_Assign(x', S.E_Apply(x, [], [], ty))                        val stm = S.S_Assign(x', S.E_Prim(x, [], [], ty))
311                        in                        in
312                          (stm::stms, S.E_Var x')                          (stm::stms, S.E_Var x')
313                        end                        end
314                    | _ => (stms, S.E_Var x)                    | _ => (stms, S.E_Var(lookupVar(env, x)))
315                  (* end case *))                  (* end case *))
316              | AST.E_Lit lit => (stms, S.E_Lit lit)              | AST.E_Lit lit => (stms, S.E_Lit lit)
317              | AST.E_Tuple es => raise Fail "E_Tuple not yet implemented"              | AST.E_Tuple es => raise Fail "E_Tuple not yet implemented"
318              | AST.E_Apply(f, tyArgs, args, ty) => let              | AST.E_Apply(f, tyArgs, args, ty) => let
319                  val (stms, xs) = simplifyExpsToVars (args, stms)                  val (stms, xs) = simplifyExpsToVars (env, args, stms)
320                    in
321                      case Var.kindOf f
322                       of S.FunVar => (stms, S.E_Apply(lookupVar(env, f), xs, cvtTy ty))
323                        | S.BasisVar => let
324                            fun cvtTyArg (Types.TYPE tv) = S.TY(cvtTy(TU.resolve tv))
325                              | cvtTyArg (Types.DIFF dv) = S.DIFF(TU.monoDiff(TU.resolveDiff dv))
326                              | cvtTyArg (Types.SHAPE sv) = S.SHAPE(TU.monoShape(TU.resolveShape sv))
327                              | cvtTyArg (Types.DIM dv) = S.DIM(TU.monoDim(TU.resolveDim dv))
328                            val tyArgs = List.map cvtTyArg tyArgs
329                  in                  in
330                    (stms, S.E_Apply(f, tyArgs, xs, ty))                            (stms, S.E_Prim(f, tyArgs, xs, cvtTy ty))
331                            end
332                        | _ => raise Fail "bogus application"
333                      (* end case *)
334                  end                  end
335              | AST.E_Cons es => let              | AST.E_Cons es => let
336                  val (stms, xs) = simplifyExpsToVars (es, stms)                  val (stms, xs) = simplifyExpsToVars (env, es, stms)
337                  in                  in
338                    (stms, S.E_Cons xs)                    (stms, S.E_Cons xs)
339                  end                  end
340              | AST.E_Seq es => let              | AST.E_Seq es => let
341                  val (stms, xs) = simplifyExpsToVars (es, stms)                  val (stms, xs) = simplifyExpsToVars (env, es, stms)
342                  in                  in
343                    (stms, S.E_Seq xs)                    (stms, S.E_Seq xs)
344                  end                  end
345              | AST.E_Slice(e, indices, ty) => let (* tensor slicing *)              | AST.E_Slice(e, indices, ty) => let (* tensor slicing *)
346                  val (stms, x) = simplifyExpToVar (e, stms)                  val (stms, x) = simplifyExpToVar (env, e, stms)
347                  fun f ([], ys, stms) = (stms, List.rev ys)                  fun f ([], ys, stms) = (stms, List.rev ys)
348                    | f (NONE::es, ys, stms) = f (es, NONE::ys, stms)                    | f (NONE::es, ys, stms) = f (es, NONE::ys, stms)
349                    | f (SOME e::es, ys, stms) = let                    | f (SOME e::es, ys, stms) = let
350                        val (stms, y) = simplifyExpToVar (e, stms)                        val (stms, y) = simplifyExpToVar (env, e, stms)
351                        in                        in
352                          f (es, SOME y::ys, stms)                          f (es, SOME y::ys, stms)
353                        end                        end
354                  val (stms, indices) = f (indices, [], stms)                  val (stms, indices) = f (indices, [], stms)
355                  in                  in
356                    (stms, S.E_Slice(x, indices, ty))                    (stms, S.E_Slice(x, indices, cvtTy ty))
357                  end                  end
358              | AST.E_Cond(e1, e2, e3, ty) => let              | AST.E_Cond(e1, e2, e3, ty) => let
359                (* a conditional expression gets turned into an if-then-else statememt *)                (* a conditional expression gets turned into an if-then-else statememt *)
360                  val result = newTemp ty                  val result = newTemp(cvtTy ty)
361                  val (stms, x) = simplifyExpToVar (e1, S.S_Var result :: stms)                  val (stms, x) = simplifyExpToVar (env, e1, S.S_Var result :: stms)
362                  fun simplifyBranch e = let                  fun simplifyBranch e = let
363                        val (stms, e) = simplifyExp (e, [])                        val (stms, e) = simplifyExp (env, e, [])
364                        in                        in
365                          mkBlock (S.S_Assign(result, e)::stms)                          mkBlock (S.S_Assign(result, e)::stms)
366                        end                        end
# Line 312  Line 369 
369                  in                  in
370                    (S.S_IfThenElse(x, s1, s2) :: stms, S.E_Var result)                    (S.S_IfThenElse(x, s1, s2) :: stms, S.E_Var result)
371                  end                  end
372              | AST.E_LoadNrrd(_, nrrd, ty) => (case TypeUtil.prune ty              | AST.E_LoadNrrd(_, nrrd, ty) => (case cvtTy ty
373                   of ty as Ty.T_DynSequence _ => (stms, S.E_LoadSeq(ty, nrrd))                   of ty as SimpleTypes.T_DynSequence _ => (stms, S.E_LoadSeq(ty, nrrd))
374                    | ty as Ty.T_Image{dim, shape} => let                    | ty as SimpleTypes.T_Image{dim, shape} => (
375                        val dim = TypeUtil.monoDim dim                        case ImageInfo.fromNrrd(NrrdInfo.getInfo nrrd, dim, shape)
                       val shp = TypeUtil.monoShape shape  
                       in  
                         case ImageInfo.fromNrrd(NrrdInfo.getInfo nrrd, dim, shp)  
376                           of NONE => raise Fail(concat[                           of NONE => raise Fail(concat[
377                                  "nrrd file \"", nrrd, "\" does not have expected type"                                  "nrrd file \"", nrrd, "\" does not have expected type"
378                                ])                                ])
379                            | SOME info => (stms, S.E_LoadImage(ty, nrrd, info))                            | SOME info => (stms, S.E_LoadImage(ty, nrrd, info))
380                          (* end case *)                        (* end case *))
                       end  
381                    | _ => raise Fail "bogus type for E_LoadNrrd"                    | _ => raise Fail "bogus type for E_LoadNrrd"
382                  (* end case *))                  (* end case *))
383              | AST.E_Coerce{srcTy, dstTy, e} => let              | AST.E_Coerce{srcTy, dstTy, e} => let
384                  val (stms, x) = simplifyExpToVar (e, stms)                  val (stms, x) = simplifyExpToVar (env, e, stms)
385                    val dstTy = cvtTy dstTy
386                  val result = newTemp dstTy                  val result = newTemp dstTy
387                    val rhs = S.E_Coerce{srcTy = cvtTy srcTy, dstTy = dstTy, x = x}
388                  in                  in
389                    (S.S_Assign(result, S.E_Coerce{srcTy=srcTy, dstTy=dstTy, x=x})::stms, S.E_Var result)                    (S.S_Assign(result, rhs)::stms, S.E_Var result)
390                  end                  end
391            (* end case *))            (* end case *))
392    
393      and simplifyExpToVar (exp, stms) = let      and simplifyExpToVar (env, exp, stms) = let
394            val (stms, e) = simplifyExp (exp, stms)            val (stms, e) = simplifyExp (env, exp, stms)
395            in            in
396              case e              case e
397               of S.E_Var x => (stms, x)               of S.E_Var x => (stms, x)
# Line 348  Line 403 
403              (* end case *)              (* end case *)
404            end            end
405    
406      and simplifyExpsToVars (exps, stms) = let      and simplifyExpsToVars (env, exps, stms) = let
407            fun f ([], xs, stms) = (stms, List.rev xs)            fun f ([], xs, stms) = (stms, List.rev xs)
408              | f (e::es, xs, stms) = let              | f (e::es, xs, stms) = let
409                  val (stms, x) = simplifyExpToVar (e, stms)                  val (stms, x) = simplifyExpToVar (env, e, stms)
410                  in                  in
411                    f (es, x::xs, stms)                    f (es, x::xs, stms)
412                  end                  end
# Line 361  Line 416 
416    
417      fun transform (errStrm, ast) = let      fun transform (errStrm, ast) = let
418            val simple = simplifyProgram ast            val simple = simplifyProgram ast
419            val _ = SimplePP.output (Log.logFile(), simple)       (* DEBUG *)            val _ = SimplePP.output (Log.logFile(), "simplify", simple)   (* DEBUG *)
420              val simple = Inliner.transform simple
421              val _ = SimplePP.output (Log.logFile(), "inlining", simple)   (* DEBUG *)
422  (*  (*
423            val simple = Lift.transform simple            val simple = Lift.transform simple
424                  handle Eval.Error msg => (Error.error(errStrm, msg); simple)                  handle Eval.Error msg => (Error.error(errStrm, msg); simple)
425              val _ = SimplePP.output (Log.logFile(), "lifting", simple)   (* DEBUG *)
426  *)  *)
427            in            in
428              simple              simple

Legend:
Removed from v.2140  
changed lines
  Added in v.2512

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