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

SCM Repository

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

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

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

branches/charisee/src/compiler/ein/printer.sml revision 2496, Tue Oct 29 05:29:46 2013 UTC branches/charisee_dev/src/compiler/ein/printer.sml revision 3369, Mon Nov 2 19:38:34 2015 UTC
# Line 1  Line 1 
   
1  (*printer Function returns String*)  (*printer Function returns String*)
2  structure Printer = struct  structure Printer = struct
3    
4      local      local
       structure G = GenericEin  
       structure E = Ein  
       structure S = Specialize  
      (* structure R = Rewrite*)  
     in  
   
     fun intList l = String.concat[String.concatWith "," (List.map Int.toString l)]  
   
     fun printIndex e= (case e  
         of E.C(cx)=> String.concat["'",Int.toString(cx),"'"]  
         |  E.V(ix)=> Int.toString(ix)  
     )  
5    
6      fun printAlpha alpha= String.concat[String.concatWith "," (List.map printIndex alpha)]        structure E = Ein
7    
8        in
9    
10      fun handleBeta(a,b)=String.concat["Delta_(",printIndex a, ",", printIndex b,")"]      fun iTos e=Int.toString e
11        fun intList l = String.concatWith "," (List.map iTos l)
12        fun handleIndex e= (case e
13            of E.C(cx)=> String.concat["'",iTos(cx),"'"]
14            |  E.V(ix)=> iTos(ix)
15        (*end case*))
16        fun handleAlpha alpha= String.concat[String.concatWith "," (List.map handleIndex alpha)]
17        fun handleBeta(a,b)=String.concat["δ_(",handleIndex a, ",", handleIndex b,")"]
18        fun handleSumRange (mu,lb,ub)= String.concat[(handleIndex mu),"[",iTos(lb),"-",iTos(ub),"]"]
19    
20    
21            fun printbody term = (case term            fun printbody term = (case term
22              of E.Const r => Real.toString(r)          of E.Const r => iTos(r)
23              | E.Tensor(id, []) =>String.concat[" S",Int.toString(id)]          | E.ConstR r =>RationalEin.toString r
24              | E.Tensor(id, alpha) =>String.concat[" T",Int.toString(id),"_",printAlpha alpha]          | E.Tensor(id, []) =>String.concat[" S",iTos(id)]
25              | E.Field(id, []) =>String.concat[" F",Int.toString(id)]          | E.Tensor(id, alpha) =>String.concat[" T",iTos(id),"_",handleAlpha alpha]
26              | E.Field(id, alpha) =>String.concat[" F",Int.toString(id),"_",printAlpha alpha]          | E.Field(id, []) =>String.concat[" F",iTos(id)]
27            | E.Field(id, alpha) =>String.concat[" F",iTos(id),"_",handleAlpha alpha]
28            | E.Krn(tid,[],pos) =>String.concat[" H ",iTos(tid),"[",  (printbody pos),"]"]
29              | E.Krn(tid,[],pos) =>String.concat["h ",Int.toString(tid),"[",  (printbody pos),"]"]          | E.Krn(tid,betas,pos) => String.concat[" H",iTos(tid),"^{", String.concat (List.map handleBeta betas),"} [",(printbody pos),"]"]
   
             | E.Krn(tid,betas,pos) =>String.concat["h",Int.toString(tid),"^{",String.concat(List.map handleBeta betas),"} [",  
                 (printbody pos),"]"]  
   
   
30              | E.Delta(ix) =>  handleBeta ix              | E.Delta(ix) =>  handleBeta ix
31              | E.Value(ix)=> String.concat["_",Int.toString(ix),"_"]          | E.Value(ix)=> String.concat["_",iTos(ix),"_"]
32              | E.Epsilon(ix, jx, kx) => String.concat["E_",Int.toString(ix), Int.toString(jx), Int.toString(kx)]          | E.Epsilon(ix, jx, kx) => String.concat["ϵ_",iTos(ix), iTos(jx), iTos(kx)]
33              | E.Sum (alpha,e1)=> String.concat["\n \t Sigma:", printAlpha alpha," (", printbody e1,")"]          | E.Eps2(ix, jx) => String.concat["ϵ_",iTos(ix), iTos(jx)]
34              | E.Neg e1 => String.concat["-", printbody e1]          | E.Sum ([(v,_,_)],e1)=>String.concat[" Σ_",handleIndex v,"⋐" ,printbody e1,"⋑",handleIndex v]
35            | E.Sum (sumindex,e1)=>String.concat[" Σ_", String.concatWith ","(List.map handleSumRange sumindex),"⋐" ,printbody e1,"⋑"]
36            | E.Neg e1 => String.concat["-NEG-", printbody e1]
37            | E.Lift e1 => String.concat["LIFT ⊢", printbody e1,"⊣"]
38              | E.Add el => String.concat["(",String.concatWith "+" (List.map printbody el),")"]              | E.Add el => String.concat["(",String.concatWith "+" (List.map printbody el),")"]
39              | E.Sub(e1, e2) => String.concat ["(",printbody(e1),"- ",printbody(e2), ")"]          | E.Sub(e1, e2) => String.concat ["(",printbody(e1),"---- ",printbody(e2)]
40              | E.Prod el => String.concat["{{",String.concatWith "* " (List.map printbody el),"}}"]   | E.Prod el => String.concat["{*{",String.concatWith "* " (List.map printbody el),"}*}"]
41              | E.Div(e1,e2) => String.concat ["(",printbody(e1),"/ ",printbody (e2), ")"]          | E.Div(e1,e2) => String.concat ["(",printbody(e1),")/( ",printbody (e2),")"]
42              | E.Partial alpha => String.concat["d/dx_",printAlpha alpha]          | E.Partial alpha => String.concat["d/dx_",handleAlpha alpha]
43              | E.Apply(e1, e2)=> String.concat [ printbody(e1),"@{", printbody(e2),"} "]              | E.Apply(e1, e2)=> String.concat [ printbody(e1),"@{", printbody(e2),"} "]
44    
45              | E.Conv(fid,alpha, tid ,[])=> String.concat["|V",Int.toString(fid),"_{",printAlpha alpha, "}★h|"]          | E.Conv(fid,alpha, tid ,[])=> String.concat["|v",iTos(fid),"_{",handleAlpha alpha, "}★h(",iTos(tid),")|"]
46              | E.Conv(fid,alpha, tid, beta)=> String.concat["|V",Int.toString(fid),"_{",printAlpha alpha, "}★d/dx_{",printAlpha beta,"}h|"]                  | E.Conv(fid,alpha, tid, [b0,b1])=> String.concat["|v",iTos(fid),"_{",handleAlpha alpha, "}★h2(",iTos(tid),")dx_",handleAlpha [b0,b1],"}|"]
47            | E.Conv(fid,alpha, tid, beta)=> String.concat["|v",iTos(fid),"_{",handleAlpha alpha, "}★h(",iTos(tid),")dx_",handleAlpha beta,"}|"]
48              | E.Probe (e1,e2)=> String.concat ["",printbody(e1), "","[", printbody(e2),"]"]   (*
49              | E.Img (fid,alpha,pos)=> String.concat ["V", Int.toString(fid),"_",printAlpha alpha,"[",           | E.Conv(fid,alpha, tid,[])=> String.concat["F_",handleAlpha alpha]
50             | E.Conv(fid,alpha, tid, beta)=> String.concat["F_",handleAlpha alpha,"∇",handleAlpha beta]
51     *)
52            | E.Probe (e1,e2)=> String.concat [printbody(e1)(*,"[", printbody(e2),"]"*)]
53            | E.Img (fid,alpha,pos)=> String.concat [" V", iTos(fid),"_",handleAlpha alpha,"[",
54                  (String.concatWith "," (List.map printbody pos)),"]"]                  (String.concatWith "," (List.map printbody pos)),"]"]
55            | E.Sqrt e => String.concat[ "√(", printbody e ,")"]
56            | E.PowInt(e,n1) => String.concat[  "(",printbody e ,")^",iTos n1]
57            | E.PowReal(e,n1) => String.concat[ "(", printbody e ,")^",RationalEin.toString n1]
58            | E.Cosine e=>  String.concat["Cosine(",printbody e ,")"]
59            | E.ArcCosine e=> String.concat["ArcCosine(",printbody e ,")"]
60            | E.Sine e => String.concat["Sine(",printbody e ,")"]
61            | E.ArcSine e => String.concat["ArcSine(",printbody e ,")"]
62    | E.PowEmb (e,sx,n)=>String.concat["POW(",printbody e,"^", iTos n,")"]
63              (* end case *))              (* end case *))
64    
65        (*printerE:EIN ->string*)
66          fun printerE (Ein.EIN{params, index, body}) = let          fun printerE (Ein.EIN{params, index, body}) = let
67          fun Index ((E.IX(dim))::i, count) = String.concat["(", Int.toString count,          fun Index (dim::i, count) = String.concat["(", iTos count,
68                  " DIM:", Int.toString dim, ")", Index(i,count+1)]                  " DIM:", iTos dim, ")", Index(i,count+1)]
             | Index ((E.SX(lb,ub))::i, count) = String.concat[ "} {Summation:(",Int.toString(count),  
                 " DIM:",Int.toString(lb),",", Int.toString(ub),")", Index(i,count+1)]  
69              | Index (_, count) = ""              | Index (_, count) = ""
70          fun Param(E.TEN::param, i)=String.concat[Int.toString(i), ":TEN ", Param(param, i+1)]          fun Param(E.TEN(t,shape)::param, i)=
71              | Param(E.FLD(d)::param, i)= String.concat[Int.toString(i), ":FLD-", Int.toString d," ",Param(param, i+1)]                  String.concat[iTos(i), ":TEN "(*, iTos t*),"<", intList shape,">, ",Param(param, i+1)]
72                | Param(E.FLD(d)::param, i)= String.concat[iTos(i), ":FLD(", iTos d,"), ",Param(param, i+1)]
73                | Param(E.IMG(d,shape)::param, i)= String.concat[iTos(i), ":IMG(", iTos d,")< ",
74                            (intList shape),  ">, ",Param(param, i+1)]
75                | Param(E.KRN::param, i)= String.concat[iTos(i), ":KRN ",Param(param, i+1)]
76              | Param(_,_)=""              | Param(_,_)=""
77    
         val str=String.concat ["\n\t" ,Param(params, 0)," \n \t <",printbody(body),">","\n \t _{",Index(index,0),"} \n "]  
78          in          in
79              str             String.concat["-EIN-" (*,Param(params, 0)*)," <",printbody(body),">"," _{",Index(index,0),"} "]
         end  
80    
81         (* fun printerApp(R.App(e,l))= printerE(e)          end
             | printerApp(_)= ""*)  
82    
83      end (* local *)      end (* local *)
84    end    end

Legend:
Removed from v.2496  
changed lines
  Added in v.3369

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