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

SCM Repository

[diderot] Annotation of /branches/charisee_dev/src/compiler/ein/printer.sml
ViewVC logotype

Annotation of /branches/charisee_dev/src/compiler/ein/printer.sml

Parent Directory Parent Directory | Revision Log Revision Log


Revision 2843 - (view) (download)
Original Path: branches/charisee/src/compiler/ein/printer.sml

1 : cchiw 2383
2 : cchiw 2387 (*printer Function returns String*)
3 : cchiw 2383 structure Printer = struct
4 :    
5 :     local
6 : cchiw 2499
7 : jhr 2391 structure E = Ein
8 : cchiw 2499
9 : jhr 2391 in
10 : cchiw 2383
11 : cchiw 2605 fun intList l = String.concat[String.concatWith "," (List.map Int.toString l)]
12 : cchiw 2426
13 : cchiw 2605 fun handleIndex e= (case e
14 :     of E.C(cx)=> String.concat["'",Int.toString(cx),"'"]
15 :     | E.V(ix)=> Int.toString(ix))
16 : cchiw 2426
17 : cchiw 2605 fun handleAlpha alpha= String.concat[String.concatWith "," (List.map handleIndex alpha)]
18 :     fun handleBeta(a,b)=String.concat["δ_(",handleIndex a, ",", handleIndex b,")"]
19 :     fun handleSumRange (mu,lb,ub)= String.concat[(handleIndex mu),"[",Int.toString(lb),"-",Int.toString(ub),"]"]
20 : cchiw 2485
21 : cchiw 2605 fun printbody term = (case term
22 :     of E.Const r => Int.toString(r)
23 :     | E.Tensor(id, []) =>String.concat[" S",Int.toString(id)]
24 :     | E.Tensor(id, alpha) =>String.concat[" T",Int.toString(id),"_",handleAlpha alpha]
25 :     | E.Field(id, []) =>String.concat[" F",Int.toString(id)]
26 :     | E.Field(id, alpha) =>String.concat[" F",Int.toString(id),"_",handleAlpha alpha]
27 :     | E.Krn(tid,[],pos) =>String.concat[" H ",Int.toString(tid),"[", (printbody pos),"]"]
28 :     | E.Krn(tid,betas,pos) => String.concat[" H",Int.toString(tid),"^{", String.concat (List.map handleBeta betas),"} [",(printbody pos),"]"]
29 :     | E.Delta(ix) => handleBeta ix
30 :     | E.Value(ix)=> String.concat["_",Int.toString(ix),"_"]
31 : cchiw 2843 | E.Epsilon(ix, jx, kx) => String.concat["ϵ_",Int.toString(ix), Int.toString(jx), Int.toString(kx)]
32 :     | E.Eps2(ix, jx) => String.concat["ϵ_",Int.toString(ix), Int.toString(jx)]
33 : cchiw 2605 | E.Sum (sumindex,e1)=>String.concat[" Σ_",String.concatWith ","(List.map handleSumRange sumindex),"⋐" ,printbody e1,"⋑"]
34 :     | E.Neg e1 => String.concat["-NEG-", printbody e1]
35 :     | E.Lift e1 => String.concat["LIFT ⊢", printbody e1,"⊣"]
36 :     | E.Add el => String.concat["(",String.concatWith "+" (List.map printbody el),")"]
37 :     | E.Sub(e1, e2) => String.concat ["(",printbody(e1),"- ",printbody(e2), ")"]
38 :     | E.Prod el => String.concat["{{",String.concatWith "* " (List.map printbody el),"}}"]
39 :     | E.Div(e1,e2) => String.concat ["(",printbody(e1),"/ ",printbody (e2), ")"]
40 :     | E.Partial alpha => String.concat["d/dx_",handleAlpha alpha]
41 :     | E.Apply(e1, e2)=> String.concat [ printbody(e1),"@{", printbody(e2),"} "]
42 :     | E.Conv(fid,alpha, tid ,[])=> String.concat["|v",Int.toString(fid),"_{",handleAlpha alpha, "}★h",Int.toString(tid),"|"]
43 :     | E.Conv(fid,alpha, tid, beta)=> String.concat["|v",Int.toString(fid),"_{",handleAlpha alpha, "}★d/dx_{",handleAlpha beta,"}h",Int.toString(tid),"|"]
44 :     | E.Probe (e1,e2)=> String.concat ["[",printbody(e1), "","[", printbody(e2),"]]"]
45 :     | E.Img (fid,alpha,pos)=> String.concat [" V", Int.toString(fid),"_",handleAlpha alpha,"[",
46 : cchiw 2522 (String.concatWith "," (List.map printbody pos)),"]"]
47 : cchiw 2605 (* end case *))
48 : cchiw 2485
49 : cchiw 2605 fun printerE (Ein.EIN{params, index, body}) = let
50 : cchiw 2499 fun Index (dim::i, count) = String.concat["(", Int.toString count,
51 : cchiw 2394 " DIM:", Int.toString dim, ")", Index(i,count+1)]
52 : jhr 2391 | Index (_, count) = ""
53 : cchiw 2525 fun Param(E.TEN(t,shape)::param, i)=let
54 :     val shape'=List.map(fn e1=>Int.toString(e1)) shape
55 :     in String.concat[Int.toString(i), ":TEN", Int.toString t," ", String.concatWith"," shape'," --",Param(param, i+1)]
56 :     end
57 : cchiw 2437 | Param(E.FLD(d)::param, i)= String.concat[Int.toString(i), ":FLD-", Int.toString d," ",Param(param, i+1)]
58 : cchiw 2510 | Param(E.IMG(d)::param, i)= String.concat[Int.toString(i), ":IMG-", Int.toString d," ",Param(param, i+1)]
59 :     | Param(E.KRN::param, i)= String.concat[Int.toString(i), ":KRN ",Param(param, i+1)]
60 : cchiw 2412 | Param(_,_)=""
61 : cchiw 2424
62 : cchiw 2838 val str=String.concat ["-EIN-" ,Param(params, 0)," <",printbody(body),">"," _{",Index(index,0),"} "]
63 : cchiw 2424 in
64 : cchiw 2496 str
65 : cchiw 2424 end
66 : cchiw 2394
67 : jhr 2391 end (* local *)
68 :     end

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