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

SCM Repository

[diderot] Diff of /branches/vis12-cl/src/compiler/c-util/tree-to-c.sml
ViewVC logotype

Diff of /branches/vis12-cl/src/compiler/c-util/tree-to-c.sml

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

revision 1858, Fri Apr 20 20:14:41 2012 UTC revision 1958, Tue Jul 10 13:01:03 2012 UTC
# Line 27  Line 27 
27    
28      val trBlock : env * TreeIL.block -> CLang.stm      val trBlock : env * TreeIL.block -> CLang.stm
29    
30        val trFree : env * TreeIL.block -> CLang.stm
31    
32      val trFragment : env * TreeIL.block -> env * CLang.stm list      val trFragment : env * TreeIL.block -> env * CLang.stm list
33    
34      val trExp : env * TreeIL.exp -> CLang.exp      val trExp : env * TreeIL.exp -> CLang.exp
# Line 77  Line 79 
79    (* translate IL basis functions *)    (* translate IL basis functions *)
80      local      local
81        fun mkLookup suffix = let        fun mkLookup suffix = let
82              val tbl = ILBasis.Tbl.mkTable (16, Fail "basis table")              val tbl = MathFuns.Tbl.mkTable (16, Fail "basis table")
83              fun ins f = ILBasis.Tbl.insert tbl (f, ILBasis.toString f ^ suffix)              fun ins f = MathFuns.Tbl.insert tbl (f, MathFuns.toString f ^ suffix)
84              in              in
85                List.app ins ILBasis.allFuns;                List.app ins MathFuns.allFuns;
86                ILBasis.Tbl.lookup tbl                MathFuns.Tbl.lookup tbl
87              end              end
88        val fLookup = mkLookup "f"        val fLookup = mkLookup "f"
89        val dLookup = mkLookup ""        val dLookup = mkLookup ""
# Line 142  Line 144 
144                  (* end case *))                  (* end case *))
145              | (Op.Dot d, args) => CL.E_Apply(N.dot d, args)              | (Op.Dot d, args) => CL.E_Apply(N.dot d, args)
146              | (Op.MulVecMat(m, n), args) =>              | (Op.MulVecMat(m, n), args) =>
147                  if (1 < m) andalso (m < 4) andalso (m = n)                  if (1 < m) andalso (m <= 4) andalso (m = n)
148                    then CL.E_Apply(N.mulVecMat(m,n), args)                    then CL.E_Apply(N.mulVecMat(m,n), args)
149                    else raise Fail "unsupported vector-matrix multiply"                    else raise Fail "unsupported vector-matrix multiply"
150              | (Op.MulMatVec(m, n), args) =>              | (Op.MulMatVec(m, n), args) =>
151                  if (1 < m) andalso (m < 4) andalso (m = n)                  if (1 < m) andalso (m <= 4) andalso (m = n)
152                    then CL.E_Apply(N.mulMatVec(m,n), args)                    then CL.E_Apply(N.mulMatVec(m,n), args)
153                    else raise Fail "unsupported matrix-vector multiply"                    else raise Fail "unsupported matrix-vector multiply"
154              | (Op.MulMatMat(m, n, p), args) =>              | (Op.MulMatMat(m, n, p), args) =>
155                  if (1 < m) andalso (m < 4) andalso (m = n) andalso (n = p)                  if (1 < m) andalso (m <= 4) andalso (m = n) andalso (n = p)
156                    then CL.E_Apply(N.mulMatMat(m,n,p), args)                    then CL.E_Apply(N.mulMatMat(m,n,p), args)
157                    else raise Fail "unsupported matrix-matrix multiply"                    else raise Fail "unsupported matrix-matrix multiply"
158                | (Op.ColonMul(Ty.TensorTy dd1, Ty.TensorTy dd2), args) =>
159                    CL.E_Apply(N.colonMul(dd1, dd2), args)
160              | (Op.Cross, args) => CL.E_Apply(N.cross(), args)              | (Op.Cross, args) => CL.E_Apply(N.cross(), args)
161              | (Op.Norm(Ty.TensorTy[n]), args) => CL.E_Apply(N.length n, args)              | (Op.Norm(Ty.TensorTy[n]), args) => CL.E_Apply(N.length n, args)
162              | (Op.Norm(Ty.TensorTy[m,n]), args) => CL.E_Apply(N.norm(m,n), args)              | (Op.Norm(Ty.TensorTy[m,n]), args) => CL.E_Apply(N.normMat(m,n), args)
163                | (Op.Norm(Ty.TensorTy[m,n,p]), args) => CL.E_Apply(N.normTen3(m,n,p), args)
164              | (Op.Normalize d, args) => CL.E_Apply(N.normalize d, args)              | (Op.Normalize d, args) => CL.E_Apply(N.normalize d, args)
165              | (Op.Scale(Ty.TensorTy[n]), args) => CL.E_Apply(N.scale n, args)              | (Op.Scale(Ty.TensorTy[n]), args) => CL.E_Apply(N.scale n, args)
166              | (Op.PrincipleEvec ty, _) => raise Fail "PrincipleEvec unimplemented"              | (Op.PrincipleEvec ty, _) => raise Fail "PrincipleEvec unimplemented"
# Line 191  Line 196 
196              | (Op.Concat ty, [seq1, seq2]) => CL.mkApply("Diderot_DynSeqConcat", [              | (Op.Concat ty, [seq1, seq2]) => CL.mkApply("Diderot_DynSeqConcat", [
197                    CL.mkSizeof(trType ty), seq1, seq2                    CL.mkSizeof(trType ty), seq1, seq2
198                  ])                  ])
199                | (Op.Length _, [seq]) => CL.mkApply("Diderot_DynSeqLength", [seq])
200              | (Op.Ceiling d, args) => CL.mkApply(N.addTySuffix("ceil", d), args)              | (Op.Ceiling d, args) => CL.mkApply(N.addTySuffix("ceil", d), args)
201              | (Op.Floor d, args) => CL.mkApply(N.addTySuffix("floor", d), args)              | (Op.Floor d, args) => CL.mkApply(N.addTySuffix("floor", d), args)
202              | (Op.Round d, args) => CL.mkApply(N.addTySuffix("round", d), args)              | (Op.Round d, args) => CL.mkApply(N.addTySuffix("round", d), args)
# Line 343  Line 349 
349                  [CL.mkCall(N.scaleMat(m,n),  lhs :: trExps(env, args))]                  [CL.mkCall(N.scaleMat(m,n),  lhs :: trExps(env, args))]
350              | IL.E_Op(Op.MulMatMat(m,n,p), args) =>              | IL.E_Op(Op.MulMatMat(m,n,p), args) =>
351                  [CL.mkCall(N.mulMatMat(m,n,p), lhs :: trExps(env, args))]                  [CL.mkCall(N.mulMatMat(m,n,p), lhs :: trExps(env, args))]
352                | IL.E_Op(Op.MulVecTen3(m, n, p), args) =>
353                    if (1 < m) andalso (m <= 4) andalso (m = n) andalso (n = p)
354                      then [CL.mkCall(N.mulVecTen3(m,n,p), lhs :: trExps(env, args))]
355                      else raise Fail "unsupported vector-tensor multiply"
356                | IL.E_Op(Op.MulTen3Vec(m, n, p), args) =>
357                    if (1 < m) andalso (m <= 4) andalso (m = n) andalso (n = p)
358                      then [CL.mkCall(N.mulTen3Vec(m,n,p), lhs :: trExps(env, args))]
359                      else raise Fail "unsupported tensor-vector multiply"
360                | IL.E_Op(Op.ColonMul(Ty.TensorTy dd1, Ty.TensorTy dd2), args) =>
361                    if (length dd1 + length dd2 > 5)
362                      then [CL.mkCall(N.colonMul(dd1, dd2), lhs :: trExps(env, args))]
363                      else [CL.mkAssign(lhs, trExp(env, rhs))]
364              | IL.E_Op(Op.EigenVals2x2, [m]) => let              | IL.E_Op(Op.EigenVals2x2, [m]) => let
365                  val (m, stms) = expToVar (env, CL.T_Named(N.matTy(2,2)), "m", m)                  val (m, stms) = expToVar (env, CL.T_Named(N.matTy(2,2)), "m", m)
366                  in                  in
# Line 370  Line 388 
388                  [CL.mkCall(N.identityMat n, [lhs])]                  [CL.mkCall(N.identityMat n, [lhs])]
389              | IL.E_Op(Op.Zero(Ty.TensorTy[m,n]), args) =>              | IL.E_Op(Op.Zero(Ty.TensorTy[m,n]), args) =>
390                  [CL.mkCall(N.zeroMat(m,n), [lhs])]                  [CL.mkCall(N.zeroMat(m,n), [lhs])]
391              | IL.E_Op(Op.TensorToWorldSpace(info, ty as Ty.TensorTy[_,_]), args) =>              | IL.E_Op(Op.TensorToWorldSpace(info, ty as Ty.TensorTy(_::_::_)), args) =>
392                  [CL.mkCall(N.toWorldSpace ty, lhs :: trExps(env, args))]                  [CL.mkCall(N.toWorldSpace ty, lhs :: trExps(env, args))]
393              | IL.E_Op(Op.LoadVoxels(info, n), [a]) =>              | IL.E_Op(Op.LoadVoxels(info, n), [a]) =>
394                  if (n > 1)                  if (n > 1)
# Line 425  Line 443 
443                  end                  end
444              | IL.E_State x => (case IL.StateVar.ty x              | IL.E_State x => (case IL.StateVar.ty x
445                   of Ty.TensorTy[n,m] => [CL.mkCall(N.copyMat(n,m), [lhs, VarToC.rvalueStateVar x])]                   of Ty.TensorTy[n,m] => [CL.mkCall(N.copyMat(n,m), [lhs, VarToC.rvalueStateVar x])]
446                    | Ty.TensorTy[n,m,l] => [CL.mkCall(N.copyTen(n,m,l), [lhs, VarToC.rvalueStateVar x])]                    | Ty.TensorTy[n,m,l] => [CL.mkCall(N.copyTen3(n,m,l), [lhs, VarToC.rvalueStateVar x])]
447                    | _ => [CL.mkAssign(lhs, VarToC.rvalueStateVar x)]                    | _ => [CL.mkAssign(lhs, VarToC.rvalueStateVar x)]
448                  (* end case *))                  (* end case *))
449              | IL.E_Var x => (case IL.Var.ty x              | IL.E_Var x => (case IL.Var.ty x
450                   of Ty.TensorTy[n,m] => [CL.mkCall(N.copyMat(n,m), [lhs, VarToC.rvalueVar(env, x)])]                   of Ty.TensorTy[n,m] => [CL.mkCall(N.copyMat(n,m), [lhs, VarToC.rvalueVar(env, x)])]
451                    | Ty.TensorTy[n,m,l] => [CL.mkCall(N.copyTen(n,m,l), [lhs, VarToC.rvalueVar(env, x)])]                    | Ty.TensorTy[n,m,l] => [CL.mkCall(N.copyTen3(n,m,l), [lhs, VarToC.rvalueVar(env, x)])]
452                    | _ => [CL.mkAssign(lhs, VarToC.rvalueVar(env, x))]                    | _ => [CL.mkAssign(lhs, VarToC.rvalueVar(env, x))]
453                  (* end case *))                  (* end case *))
454              | _ => [CL.mkAssign(lhs, trExp(env, rhs))]              | _ => [CL.mkAssign(lhs, trExp(env, rhs))]
# Line 482  Line 500 
500                CL.mkReturn(SOME(CL.mkVar "true")))]                CL.mkReturn(SOME(CL.mkVar "true")))]
501            end            end
502    
503      (* given the global initialization code, generate code to free the storage that is heap
504       * allocated for globals.
505       *)
506        fun trFree (env, IL.Block{locals, body}) = let
507              val env = trLocals (env, locals)
508              fun trStmt (env, stm) = (case stm
509                     of IL.S_Comment text => [CL.mkComment text]
510                      | IL.S_LoadImage(lhs, dim, name) => checkSts (fn sts => let
511                          val lhs = VarToC.lvalueVar (env, lhs)
512                          val imgTy = CL.T_Named(N.imageTy dim)
513                          val freeFn = N.freeImage dim
514                          in [
515                            CL.mkDecl(
516                              CL.T_Named N.statusTy, sts,
517                              SOME(CL.I_Exp(CL.E_Apply(freeFn, [
518                                  CL.mkCast(CL.T_Ptr(CL.T_Named "WorldPrefix_t"), CL.mkVar "wrld"),
519                                  addrOf lhs
520                                ]))))
521                          ] end)
522                      | _ => []
523                    (* end case *))
524              val stms = List.foldr (fn (stm, stms) => trStmt(env, stm)@stms) [] body
525              fun mkDecl (x, stms) = (case V.Map.find (env, x)
526                     of SOME(V(ty, x')) => CL.mkDecl(ty, x', NONE) :: stms
527                      | NONE => raise Fail(concat["mkDecl(", V.name x, ", _)"])
528                    (* end case *))
529              val stms = List.foldr mkDecl stms locals
530              in
531                CL.mkBlock stms
532              end
533    
534      fun trStms (env, stms) = let      fun trStms (env, stms) = let
535            fun trStmt (env, stm) = (case stm            fun trStmt (env, stm) = (case stm
536                   of IL.S_Comment text => [CL.mkComment text]                   of IL.S_Comment text => [CL.mkComment text]

Legend:
Removed from v.1858  
changed lines
  Added in v.1958

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