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

SCM Repository

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

Annotation of /branches/vis12/src/compiler/simplify/eval.sml

Parent Directory Parent Directory | Revision Log Revision Log


Revision 234 - (view) (download)
Original Path: trunk/src/compiler/simplify/eval.sml

1 : jhr 231 (* eval.sml
2 :     *
3 :     * COPYRIGHT (c) 2010 The Diderot Project (http://diderot.cs.uchicago.edu)
4 :     * All rights reserved.
5 :     *
6 :     * Evaluation of "static" expressions.
7 :     *)
8 :    
9 : jhr 234 structure Eval =
10 : jhr 231 struct
11 :    
12 : jhr 234 structure Ty = Types
13 :     structure BV = BasisVars
14 : jhr 231 structure S = Simple
15 :     structure VMap = Var.Map
16 :     structure VTbl = Var.Tbl
17 :    
18 :     datatype value
19 :     = BV of bool
20 : jhr 234 | SV of string
21 : jhr 231 | IV of IntInf.int
22 :     | TV of (int list * real list) (* tensors *)
23 :     | FV of FieldDef.field_def
24 : jhr 234 | Img of ImageInfo.info
25 : jhr 231 | KV of Kernel.kernel
26 :    
27 :     fun RV r = TV([], [r])
28 :    
29 : jhr 234 fun toString (BV b) = Bool.toString b
30 :     | toString (IV i) = IntInf.toString i
31 :     | toString (SV s) = concat["\"", String.toString s, "\""]
32 :     | toString (TV(s, v)) = "tensor"
33 :     | toString (FV fld) = FieldDef.toString fld
34 :     | toString (Img info) = ImageInfo.toString info
35 :     | toString (KV h) = Kernel.toString h
36 : jhr 231
37 :     val tbl : (value list -> value) VTbl.hash_table = let
38 :     val tbl = VTbl.mkTable (128, Fail "Eval table")
39 :     fun intBinOp rator [IV a, IV b] = IV(rator(a, b))
40 :     fun tensorBinOp rator [TV(s1, v1), TV(s2, v2)] =
41 :     TV(s1, ListPair.mapEq rator (v1, v2))
42 : jhr 234 fun realBinOp rator [TV([], [a]), TV([], [b])] = RV(rator(a, b))
43 : jhr 231 fun intCmp rator [IV a, IV b] = BV(rator(a, b))
44 :     fun realCmp rator [TV([], [a]), TV([], [b])] = BV(rator(a, b))
45 : jhr 234 fun boolCmp rator [BV a, BV b] = BV(rator(a, b))
46 :     fun stringCmp rator [SV a, SV b] = BV(rator(a, b))
47 : jhr 231 fun kernel h [] = KV h
48 :     in
49 :     List.app (VTbl.insert tbl) [
50 :     (BV.add_ii, intBinOp (op +)),
51 :     (BV.add_tt, tensorBinOp (op +)),
52 :     (BV.sub_ii, intBinOp (op -)),
53 :     (BV.sub_tt, tensorBinOp (op -)),
54 : jhr 234 (BV.mul_ii, intBinOp (op * )),
55 :     (BV.mul_rr, realBinOp (op * )),
56 :     (*
57 : jhr 231 (BV.mul_rt, tensorOp Op.Scale),
58 :     (BV.mul_tr, fn (y, sv, [t, r]) => tensorOp Op.Scale (y, sv, [r, t])),
59 : jhr 234 *)
60 : jhr 231 (BV.div_ii, intBinOp IntInf.quot),
61 : jhr 234 (BV.div_rr, realBinOp (op /)),
62 :     (*
63 : jhr 231 (BV.div_tr, tensorOp Op.InvScale),
64 : jhr 234 *)
65 : jhr 231 (BV.lt_ii, intCmp (op <)),
66 :     (BV.lt_rr, realCmp (op <)),
67 :     (BV.lte_ii, intCmp (op <=)),
68 :     (BV.lte_rr, realCmp (op <=)),
69 :     (BV.gte_ii, intCmp (op >=)),
70 :     (BV.gte_rr, realCmp (op >=)),
71 :     (BV.gt_ii, intCmp (op >)),
72 :     (BV.gt_rr, realCmp (op >)),
73 : jhr 234 (BV.equ_bb, boolCmp (op =)),
74 : jhr 231 (BV.equ_ii, intCmp (op =)),
75 : jhr 234 (BV.equ_ss, stringCmp (op =)),
76 : jhr 231 (BV.equ_rr, realCmp Real.==),
77 : jhr 234 (BV.neq_bb, boolCmp (op <>)),
78 : jhr 231 (BV.neq_ii, intCmp (op <>)),
79 : jhr 234 (BV.neq_ss, stringCmp (op <>)),
80 : jhr 231 (BV.neq_rr, realCmp Real.!=),
81 : jhr 234 (*
82 : jhr 231 (BV.neg_i, simpleOp(Op.Neg Op.IntTy)),
83 :     (BV.neg_t, tensorOp Op.Neg),
84 :     (BV.neg_f, fn (y, _, xs) => assign(y, Op.NegField, xs)),
85 :     (BV.op_at, fn (y, _, xs) => assign(y, Op.Probe, xs)),
86 :     (BV.op_D, fn (y, _, xs) => assign(y, Op.DiffField, xs)),
87 :     (BV.op_norm, tensorOp Op.Norm),
88 :     (BV.op_not, simpleOp Op.Not),
89 :     (*
90 :     (BV.op_subscript, fn (y, [SK, NK], xs) => ??), (*FIXME*)
91 :     *)
92 :     (BV.fn_CL, fn (y, _, xs) => assign(y, Op.CL, xs)),
93 :     (BV.fn_convolve, fn (y, _, xs) => assign(y, Op.Convolve, xs)),
94 :     (BV.fn_cos, simpleOp Op.Cos),
95 :     (BV.fn_dot, vectorOp Op.Dot),
96 :     (BV.fn_inside, fn (y, _, xs) => assign(y, Op.Inside, xs)),
97 : jhr 234 *)
98 :     (BV.fn_max, realBinOp Real.min),
99 :     (BV.fn_min, realBinOp Real.max),
100 :     (BV.fn_modulate, tensorBinOp (op * )),
101 :     (BV.fn_pow, realBinOp Real.Math.pow),
102 : jhr 231 (*
103 :     (BV.fn_principleEvec, vectorOp Op.PrincipleEvec),
104 :     (BV.fn_sin, simpleOp Op.Sin),
105 : jhr 234 *)
106 : jhr 231 (BV.kn_bspln3, kernel Kernel.bspln3),
107 :     (BV.kn_bspln5, kernel Kernel.bspln5),
108 :     (BV.kn_ctmr, kernel Kernel.ctmr),
109 :     (BV.kn_tent, kernel Kernel.tent),
110 : jhr 234 (BV.i2r, fn [IV i] => RV(real(IntInf.toInt i)))(*,
111 : jhr 231 (BV.input, fn (y, [TK], xs) => ??), (*FIXME*)
112 :     (BV.optInput, fn (y, [TK], xs) => ??) (*FIXME*)
113 :     *)
114 :     ];
115 :     tbl
116 :     end
117 :    
118 : jhr 234 fun loadImage (mvs, [SV filename]) = let
119 :     val info = ImageInfo.getInfo filename
120 :     in
121 :     (* FIXME: check image info details against mvs *)
122 :     Img info
123 :     end
124 :    
125 : jhr 231 fun evalVar env x = (case VMap.find (env, x)
126 :     of SOME v => v
127 :     | NONE => raise Fail("undefined variable " ^ Var.uniqueNameOf x)
128 :     (* end case *))
129 :    
130 :     fun evalExp (env, e) = (case e
131 :     of S.E_Var x => evalVar env x
132 :     | S.E_Lit(Literal.Int i) => IV i
133 :     | S.E_Lit(Literal.Float f) => RV(FloatLit.toReal f)
134 :     | S.E_Lit(Literal.String s) => SV s
135 :     | S.E_Lit(Literal.Bool b) => BV b
136 :     | S.E_Tuple _ => raise Fail "E_Tuple"
137 :     | S.E_Apply(f, mvs, xs, _) => (
138 :     (* try *)(
139 :     if Var.same(f, BV.fn_load)
140 : jhr 234 then loadImage(mvs, List.map (evalVar env) xs)
141 : jhr 231 else (case VTbl.find tbl f
142 : jhr 234 of SOME evalFn => evalFn (List.map (evalVar env) xs)
143 : jhr 231 | NONE => raise Fail("Eval.exvalExp: unknown function " ^ Var.nameOf f)
144 :     (* end case *))
145 :     ) handle ex => raise Fail (concat [
146 : jhr 234 Var.uniqueNameOf f, "(",
147 :     String.concatWith "," (List.map Var.uniqueNameOf xs),
148 : jhr 231 ") fails with exception ", exnName ex
149 :     ]))
150 : jhr 232 | S.E_Cons xs => raise Fail "TODO: E_Cons"
151 : jhr 231 | S.E_Input(ty, name, optDefault) => let
152 :     val optDefault = Option.map (evalVar env) optDefault
153 : jhr 234 val SOME value = (case ty
154 :     of Ty.T_Bool =>
155 :     Inputs.getInput(name, (Option.map BV) o Bool.fromString, optDefault)
156 :     | Ty.T_Int =>
157 :     Inputs.getInput(name, (Option.map IV) o IntInf.fromString, optDefault)
158 :     | Ty.T_String => Inputs.getInput(name, fn s => SOME(SV s), optDefault)
159 :     | Ty.T_Tensor(Ty.Shape[]) =>
160 :     Inputs.getInput(name, (Option.map RV) o Real.fromString, optDefault)
161 :     | Ty.T_Tensor shp => raise Fail "TODO: general tensor inputs"
162 :     | _ => raise Fail(concat[
163 :     "input ", name, " has invalid type ", TypeUtil.toString ty
164 :     ])
165 :     (* end case *))
166 : jhr 231 in
167 : jhr 234 value
168 : jhr 231 end
169 :     | S.E_Field fld => FV fld
170 : jhr 234 | S.E_LoadImage info => Img info
171 : jhr 231 (* end case *))
172 :    
173 :     end

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