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

revision 2428, Wed Sep 18 18:04:04 2013 UTC revision 2518, Tue Dec 17 00:04:37 2013 UTC
# Line 3  Line 3 
3  structure Printer = struct  structure Printer = struct
4    
5      local      local
6        structure G = GenericEin  
7        structure E = Ein        structure E = Ein
8        structure S = Specialize  
      (* structure R = Rewrite*)  
9      in      in
10    
11      fun intList l = String.concat[String.concatWith "," (List.map Int.toString l)]      fun intList l = String.concat[String.concatWith "," (List.map Int.toString l)]
12    
13      fun printIndex e= (case e  fun handleIndex e= (case e
14          of E.C(cx)=> String.concat["c{",intList(cx),"}"]      of E.C(cx)=> String.concat["'",Int.toString(cx),"'"]
15          |  E.V([ix])=> Int.toString(ix)      |  E.V(ix)=> Int.toString(ix)
         |  E.V(mx)=> intList(mx)  
16      )      )
17      fun imp([])=""  
18      | imp([(i,j)])=  String.concat [" Delta_",printIndex(i),printIndex(j)]  fun handleAlpha alpha= String.concat[String.concatWith "," (List.map handleIndex alpha)]
19      | imp((i,j)::es)= String.concat[" Delta_",printIndex(i),printIndex(j)," ",imp(es)]  
20    
21    fun handleBeta(a,b)=String.concat["δ_(",handleIndex a, ",", handleIndex b,")"]
22    
23    fun handleSumRange (mu,lb,ub)= String.concat[(handleIndex mu),"[",Int.toString(lb),"-",Int.toString(ub),"]"]
24    
25    fun printbody term = (case term
26        of E.Const r => Real.toString(r)
27        | E.Tensor(id, []) =>String.concat[" S",Int.toString(id)]
28        | E.Tensor(id, alpha) =>String.concat[" T",Int.toString(id),"_",handleAlpha alpha]
29        | E.Field(id, []) =>String.concat[" F",Int.toString(id)]
30        | E.Field(id, alpha) =>String.concat[" F",Int.toString(id),"_",handleAlpha alpha]
31        | E.Krn(tid,[],pos) =>String.concat[" h ",Int.toString(tid),"[",  (printbody pos),"]"]
32        | E.Krn(tid,betas,pos) => String.concat[" h",Int.toString(tid),"^{", String.concat (List.map handleBeta betas),"} [",(printbody pos),"]"]
33        | E.Delta(ix) =>  handleBeta ix
34        | E.Value(ix)=> String.concat["_",Int.toString(ix),"_"]
35        | E.Epsilon(ix, jx, kx) => String.concat["E_",Int.toString(ix), Int.toString(jx), Int.toString(kx)]
36        | E.Sum (sumindex,e1)=>String.concat["\n Σ_",String.concatWith ","(List.map handleSumRange sumindex)," (", printbody e1,")"]
37        | E.Neg e1 => String.concat["-", printbody e1]
38        | E.Add el => String.concat["(",String.concatWith "+" (List.map printbody el),")"]
39        | E.Sub(e1, e2) => String.concat ["(",printbody(e1),"- ",printbody(e2), ")"]
40        | E.Prod el => String.concat["{{",String.concatWith "* " (List.map printbody el),"}}"]
41        | E.Div(e1,e2) => String.concat ["(",printbody(e1),"/ ",printbody (e2), ")"]
42        | E.Partial alpha => String.concat["d/dx_",handleAlpha alpha]
43        | E.Apply(e1, e2)=> String.concat [ printbody(e1),"@{", printbody(e2),"} "]
44        | E.Conv(fid,alpha, tid ,[])=> String.concat["|V",Int.toString(fid),"_{",handleAlpha alpha, "}★h",Int.toString(tid),"|"]
45        | E.Conv(fid,alpha, tid, beta)=> String.concat["|V",Int.toString(fid),"_{",handleAlpha alpha, "}★d/dx_{",handleAlpha beta,"}h",Int.toString(tid),"|"]
46        | E.Probe (e1,e2)=> String.concat ["",printbody(e1), "","[", printbody(e2),"]"]
47        | E.Img (fid,alpha,pos)=> String.concat ["\n V", Int.toString(fid),"_",handleAlpha alpha,"[",
48                    (String.concatWith "," (List.map printbody pos)),"]\n"]
49    
50    
51    
     fun printerE (Ein.EIN{params, index, body}) = let  
           fun printer2 term = (case term  
                  of E.Value(id)=> String.concat[" CI-",Int.toString(id)]  
                   | E.Tensor(id, E.V([])) =>String.concat[" S",Int.toString(id)]  
                   | E.Tensor(id, ix) =>String.concat[" T",Int.toString(id),"_",printIndex ix]  
                   | E.Pair e => String.concat["(",String.concatWith "," (List.map printer2 e),")"]  
                   | E.Const e => Real.toString(e)  
                   | E.Field(id, E.V [],E.V []) => String.concat ["F",Int.toString(id)]  
                   | E.Field(id, mx,dx) => String.concat ["F",Int.toString(id),"_",printIndex(mx),",",printIndex(dx)]  
   
                   | E.Add e        => String.concat["(",String.concatWith "+" (List.map printer2 e),")"]  
                   | E.Sub(a,b)      => String.concat ["(",printer2(a),"- ",printer2(b), ")"]  
                   | E.Div(a,b)      => String.concat ["(",printer2(a),"/ ",printer2(b), ")"]  
                   | E.Prod e       => String.concatWith "*" (List.map printer2 e)  
                   | E.Kernel [] =>  "h"  
                   | E.Kernel dels=> String.concat["h^",imp(dels)]  
                   | E.Delta dels  => imp(dels)  
                   | E.Epsilon(i,j,k)=> String.concat ["E_",Int.toString(i),Int.toString(j),Int.toString(k) ]  
                   | E.Conv(k,[]) => String.concat ["V (*)",printer2(k)]  
                   | E.Conv(k,[E.V(mx)]) =>let  
                         val n= length(mx)  
                         val ind= intList(mx)  
                         in  
                             String.concat ["V (*)",printer2(k),"^",Int.toString(n),"{",ind,"}"]  
                         end  
                   | E.Conv(k,_) => "Incorrect convolution argument"  
                   | E.Partial(E.C(ix)) => String.concat ["d^1","/dx_",intList(ix)]  
                   | E.Partial(E.V(mx))  => String.concat ["d^",Int.toString(length(mx)),"/dx_",intList(mx)]  
                   | E.Probe(a,b) => String.concat ["",printer2(a), "","@", printer2(b)]  
                   | E.Apply(a,b) => String.concat [ printer2(a)," ", printer2(b)," "]  
                   | E.Sum (id,a) => String.concat [" Sigma:", Int.toString(id), " ",printer2(a)]  
                   | _ =>("newbie")  
52                  (* end case *))                  (* end case *))
53            fun Index ((E.IX(dim))::i, count) = String.concat["(", Int.toString count,          fun printerE (Ein.EIN{params, index, body}) = let
54            fun Index (dim::i, count) = String.concat["(", Int.toString count,
55                  " DIM:", Int.toString dim, ")", Index(i,count+1)]                  " DIM:", Int.toString 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)]  
56              | Index (_, count) = ""              | Index (_, count) = ""
57          fun Param(E.TEN::param, i)=String.concat[Int.toString(i), ":TEN ", Param(param, i+1)]          fun Param(E.TEN::param, i)=String.concat[Int.toString(i), ":TEN ", Param(param, i+1)]
58              | Param(E.FLD(d)::param, i)= String.concat[Int.toString(i), ":FLD-", intList d," ",Param(param, i+1)]              | Param(E.FLD(d)::param, i)= String.concat[Int.toString(i), ":FLD-", Int.toString d," ",Param(param, i+1)]
59                   | Param(E.IMG(d)::param, i)= String.concat[Int.toString(i), ":IMG-", Int.toString d," ",Param(param, i+1)]
60                   | Param(E.KRN::param, i)= String.concat[Int.toString(i), ":KRN ",Param(param, i+1)]
61              | Param(_,_)=""              | Param(_,_)=""
62    
63          val str=String.concat [Param(params, 0)," \n<",printer2(body),">","\n _{",Index(index,0),"} \n \n \n "]          val str=String.concat ["" ,Param(params, 0)," <",printbody(body),">"," _{",Index(index,0),"}  "]
64          in          in
65              print str; str              str
66          end          end
67    
68         (* fun printerApp(R.App(e,l))= printerE(e)         (* fun printerApp(R.App(e,l))= printerE(e)

Legend:
Removed from v.2428  
changed lines
  Added in v.2518

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