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

SCM Repository

[diderot] Diff of /branches/vis12/src/compiler/high-to-mid/high-to-mid.sml
ViewVC logotype

Diff of /branches/vis12/src/compiler/high-to-mid/high-to-mid.sml

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

revision 365, Wed Sep 29 20:07:58 2010 UTC revision 511, Tue Feb 8 17:01:43 2011 UTC
# Line 1  Line 1 
1  (* high-to-mid.sml  (* high-to-mid.sml
2   *   *
3   * COPYRIGHT (c) 2010 The Diderot Project (http://diderot.cs.uchicago.edu)   * COPYRIGHT (c) 2010 The Diderot Project (http://diderot-language.cs.uchicago.edu)
4   * All rights reserved.   * All rights reserved.
5   *   *
6   * Translation from HighIL to MidIL representations.   * Translation from HighIL to MidIL representations.
# Line 13  Line 13 
13    end = struct    end = struct
14    
15      structure SrcIL = HighIL      structure SrcIL = HighIL
16        structure SrcTy = HighILTypes
17      structure SrcOp = HighOps      structure SrcOp = HighOps
18      structure VTbl = SrcIL.Var.Tbl      structure VTbl = SrcIL.Var.Tbl
19      structure DstIL = MidIL      structure DstIL = MidIL
20        structure DstTy = MidILTypes
21      structure DstOp = MidOps      structure DstOp = MidOps
22    
23      type var_env = DstIL.var SrcIL.Var.Tbl.hash_table      type var_env = DstIL.var VTbl.hash_table
24    
25        fun cvtTy SrcTy.BoolTy = DstTy.BoolTy
26          | cvtTy SrcTy.StringTy = DstTy.StringTy
27          | cvtTy SrcTy.IntTy = DstTy.intTy
28          | cvtTy (SrcTy.TensorTy[]) = DstTy.realTy
29          | cvtTy (SrcTy.TensorTy[d]) = DstTy.VecTy d
30          | cvtTy SrcTy.ImageTy = DstTy.ImageTy
31          | cvtTy ty = raise Fail("unexpected type " ^ SrcTy.toString ty)
32    
33      fun rename (env : var_env, x) = (case VTbl.find env x      fun rename (env : var_env, x) = (case VTbl.find env x
34             of SOME x' => x'             of SOME x' => x'
35              | NONE => let              | NONE => let
36                  val x' = DstIL.Var.new (SrcIL.Var.name x)                  val x' = DstIL.Var.new (SrcIL.Var.name x, cvtTy(SrcIL.Var.ty x))
37                  in                  in
38                    VTbl.insert env (x, x');                    VTbl.insert env (x, x');
39                    x'                    x'
40                  end                  end
41            (* end case *))            (* end case *))
42        fun renameList (env, xs) = List.map (fn x => rename(env, x)) xs
43    
44    (* expand the field Inside operator into a image-space test *)    (* expand the field Inside operator into a image-space test *)
45      fun expandInside (env, result, pos, fld) = let      fun expandInside (env, result, pos, fld) = let
# Line 38  Line 49 
49                    | _ => raise Fail "bogus field binding"                    | _ => raise Fail "bogus field binding"
50                  (* end case *))                  (* end case *))
51            fun expand (FieldDef.CONV(_, img, _)) = let            fun expand (FieldDef.CONV(_, img, _)) = let
52                  val imgPos = DstIL.Var.new "x"                  val imgPos = DstIL.Var.new ("x", DstTy.VecTy(ImageInfo.dim img))
53                  in [                  in [
54                    (imgPos, DstIL.OP(DstOp.Transform img, [pos])),                    (imgPos, DstIL.OP(DstOp.PosToImgSpace img, [pos])),
55                    (result, DstIL.OP(DstOp.Inside img, [imgPos]))                    (result, DstIL.OP(DstOp.Inside img, [imgPos]))
56                  ] end                  ] end
57              | expand (FieldDef.NEG fld) = expand fld              | expand (FieldDef.NEG fld) = expand fld
# Line 59  Line 70 
70              Probe.expand (result, fld, pos)              Probe.expand (result, fld, pos)
71            end            end
72    
73      fun cvtTy SrcOp.BoolTy = DstOp.BoolTy      fun arity (SrcTy.TensorTy[]) = 1
74        | cvtTy SrcOp.StringTy = DstOp.StringTy        | arity (SrcTy.TensorTy[d]) = d
       | cvtTy SrcOp.IntTy = DstOp.IntTy  
       | cvtTy (SrcOp.TensorTy[]) = DstOp.realTy  
       | cvtTy (SrcOp.TensorTy[d]) = DstOp.VecTy d  
       | cvtTy _ = raise Fail "unexpected higher-order tensor type"  
   
     fun arity (SrcOp.TensorTy[]) = 1  
       | arity (SrcOp.TensorTy[d]) = d  
75        | arity _ = raise Fail "arity"        | arity _ = raise Fail "arity"
76    
77      fun expandOp (env, y, rator, args) = let      fun expandOp (env, y, rator, args) = let
78            fun assign rator' =            fun assign rator' =
79                  [(y, DstIL.OP(rator', List.map (fn x => rename(env, x)) args))]                  [(y, DstIL.OP(rator', renameList(env, args)))]
80            in            in
81              case rator              case rator
82               of SrcOp.Add ty => assign (DstOp.Add(cvtTy ty))               of SrcOp.Add ty => assign (DstOp.Add(cvtTy ty))
# Line 106  Line 110 
110                | SrcOp.CeilToInt => assign (DstOp.CeilToInt 1)                | SrcOp.CeilToInt => assign (DstOp.CeilToInt 1)
111                | SrcOp.FloorToInt => assign (DstOp.FloorToInt 1)                | SrcOp.FloorToInt => assign (DstOp.FloorToInt 1)
112                | SrcOp.LoadImage info => assign (DstOp.LoadImage info)                | SrcOp.LoadImage info => assign (DstOp.LoadImage info)
113                | SrcOp.Inside => (case args                | SrcOp.Inside _ => (case args
114                     of [pos, fld] => expandInside(env, y, pos, fld)                     of [pos, fld] => expandInside(env, y, pos, fld)
115                    (* end case *))                    (* end case *))
116  (* QUESTION: there is no MidIL equivalant, but these might be floating                | SrcOp.Field fld => raise Fail "impossible"
117   * around, which could cause invalid MidIL to be generated.                | SrcOp.Probe _ => (case args
  *)  
               | SrcOp.Field fld => []  
               | SrcOp.Probe => (case args  
118                     of [fld, pos] => expandProbe(env, y, fld, pos)                     of [fld, pos] => expandProbe(env, y, fld, pos)
119                    (* end case *))                    (* end case *))
120                | SrcOp.Input s => assign (DstOp.Input s)                | SrcOp.Input(ty, s) => assign (DstOp.Input(cvtTy ty, s))
121                | SrcOp.InputWithDefault s => assign (DstOp.InputWithDefault s)                | SrcOp.InputWithDefault(ty, s) => assign (DstOp.InputWithDefault(cvtTy ty, s))
122                | _ => raise Fail("unexpected " ^ SrcOp.toString rator)                | _ => raise Fail("unexpected " ^ SrcOp.toString rator)
123              (* end case *)              (* end case *)
124            end            end
125    
126      (* expand a SrcIL assignment to a list of DstIL assignments *)
127      fun expand (env, (y, rhs)) = let      fun expand (env, (y, rhs)) = let
128            val y' = rename (env, y)            fun assign rhs = [(rename (env, y), rhs)]
           fun assign rhs = [(y', rhs)]  
129            in            in
130              case rhs              case rhs
131               of SrcIL.VAR x => assign (DstIL.VAR(rename(env, x)))               of SrcIL.VAR x => assign (DstIL.VAR(rename(env, x)))
132                | SrcIL.LIT lit => assign (DstIL.LIT lit)                | SrcIL.LIT lit => assign (DstIL.LIT lit)
133                | SrcIL.OP(rator, args) =>                | SrcIL.OP(SrcOp.Field _, args) => []
134                    expandOp (env, y', rator, args)                | SrcIL.OP(rator, args) => expandOp (env, rename (env, y), rator, args)
135                | SrcIL.CONS args =>                | SrcIL.CONS args => assign (DstIL.CONS(renameList(env, args)))
                   assign (DstIL.CONS(List.map (fn x => rename(env, x)) args))  
136              (* end case *)              (* end case *)
137            end            end
138    
# Line 147  Line 147 
147          val expand = expand          val expand = expand
148        end)        end)
149    
150      fun translate (SrcIL.Program{globals, globalInit, actors}) =      fun translate (SrcIL.Program{globals, globalInit, strands}) = let
151            raise Fail "unimplemented"            val env = VTbl.mkTable (256, Fail "env")
152              fun transMethod (SrcIL.Method{name, stateIn, stateOut, body}) =
153                    DstIL.Method{
154                        name = name,
155                        stateIn = renameList (env, stateIn),
156                        stateOut = renameList (env, stateOut),
157                        body = Trans.translate (env, body)
158                      }
159              fun transStrand (SrcIL.Strand{name, params, state, stateInit, methods}) =
160                    DstIL.Strand{
161                        name = name,
162                        params = renameList (env, params),
163                        state = renameList (env, state),
164                        stateInit = Trans.translate (env, stateInit),
165                        methods = List.map transMethod methods
166                      }
167            (* filter out field variables, since they will be made explicit in
168             * the MidIL.
169             *)
170              val globals = let
171                    fun filter x = (case SrcIL.Var.ty x
172                           of SrcTy.FieldTy => NONE
173                            | _ => SOME x
174                          (* end case *))
175                    in
176                      List.mapPartial filter globals
177                    end
178              in
179                DstIL.Program{
180                    globals = renameList (env, globals),
181                    globalInit = Trans.translate (env, globalInit),
182                    strands = List.map transStrand strands
183                  }
184              end
185    
186    end    end

Legend:
Removed from v.365  
changed lines
  Added in v.511

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