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

SCM Repository

[diderot] Diff of /branches/ein16/src/compiler/high-il/normalize-ein.sml
ViewVC logotype

Diff of /branches/ein16/src/compiler/high-il/normalize-ein.sml

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

branches/charisee/src/compiler/high-il/normalize-ein.sml revision 3033, Tue Mar 10 15:17:25 2015 UTC branches/charisee_dev/src/compiler/high-il/normalize-ein.sml revision 3441, Wed Nov 18 00:24:04 2015 UTC
# Line 12  Line 12 
12    
13      in      in
14    
     fun err str=raise Fail (String.concat["Ill-formed EIN Operator",str])  
15      val testing=0      val testing=0
16      fun flatProd e =F.rewriteProd e      fun err str=raise Fail (String.concat["Ill-formed EIN Operator",str])
17      fun mkProd e= F.mkProd e      fun mkProd e= F.mkProd e
18      fun filterSca e=F.filterSca e      fun filterSca e=F.filterSca e
     fun filterField e=F.filterField e  
19      fun mkAdd e=F.mkAdd e      fun mkAdd e=F.mkAdd e
20      fun filterGreek e=F.filterGreek e      fun filterGreek e=F.filterGreek e
21        fun mkapply e= derivativeEin.mkapply e
22      fun testp n=(case testing      fun testp n=(case testing
23          of 0=> 1          of 0=> 1
24          | _ =>(print(String.concat n);1)          | _ =>(print(String.concat n);1)
25          (*end case*))          (*end case*))
26    
27      (*prodAppPartia:ein_exp list * mu list ->ein_exp      val zero=E.B(E.Const 0)
28      * chain rule      fun setConst e = E.setConst e
29      *)      fun setNeg e  =  E.setNeg e
30      fun prodAppPartial(es,p1)=(case es      fun setExp e  =  E.setExp e
31          of []      => err "Empty App Partial"      fun setDiv e= E.setDiv e
32          | [e1]     => E.Apply(E.Partial p1,e1)      fun setSub e= E.setSub e
33          | (e1::e2) => let      fun setProd e= E.setProd e
34              val l= prodAppPartial(e2,p1)      fun setAdd e= E.setAdd e
             val (_,e2')= mkProd[e1,l]  
             val (_,e1')=mkProd(e2@ [E.Apply(E.Partial p1, e1)])  
             in  
                 E.Add[e1',e2']  
             end  
         (* end case *))  
35    
36      (*mkSum:sum_indexid list * ein_exp->int *ein_exp      (*mkSum:sum_indexid list * ein_exp->int *ein_exp
37      *distribute summation expression      *distribute summation expression
38      *)      *)
39        fun mkSum(sx1,b)=(case b
40    fun mkSum(c1,e1)=(case e1          of E.Lift e         => (1,E.Lift(E.Sum(sx1,e)))
41          of E.Lift e   => (1,E.Lift(E.Sum(c1,e)))          | E.Tensor(_,[])    => (1,b)
42          | E.Tensor(_,[]) => (1,e1)          | E.B _             => (1,b)
43          | E.Const _   => (1,e1)          | E.Opn(E.Prod, es)   => filterSca(sx1,es)
44          | E.ConstR _  => (1,e1)          | _                 => (0,E.Sum(sx1,b))
         | E.Prod p    => filterSca(c1,p)  
         | _           => (0,E.Sum(c1,e1))  
         (*end case*))  
 (*  
     fun mkSum(c1,e1)=(case e1  
         of E.Conv _   => (0,E.Sum(c1,e1))  
         | E.Field _   => (0,E.Sum(c1,e1))  
         | E.Probe _   => (0,E.Sum(c1,e1))  
         | E.Apply _   => (0,E.Sum(c1,e1))  
         | E.Delta _   => (0,E.Sum(c1,e1))  
         | E.Epsilon _ => (0,E.Sum(c1,e1))  
         | E.Eps2 _    => (0,E.Sum(c1,e1))  
         | E.Tensor(_,[]) => (1,e1)  
         | E.Tensor _  => (0,E.Sum(c1,e1))  
         | E.Neg e2    => (1,E.Neg(E.Sum(c1,e2)))  
         | E.Sub (a,b) => (1,E.Sub(E.Sum(c1,a),E.Sum(c1,b)))  
         | E.Add e     => (1,E.Add (List.map (fn(a)=>E.Sum(c1,a)) e))  
         | E.Div (a,b) => (1,E.Div(E.Sum(c1,a),E.Sum(c1,b)))  
         | E.Lift e    => (1,E.Lift(E.Sum(c1,e)))  
         | E.PowReal(e,n1)=>(1,E.PowReal(E.Sum(c1,e),n1))  
         | E.Sqrt e    => (1,E.Sqrt(E.Sum(c1,e)))  
         | E.Sum(c2,e2)=> (1,E.Sum(c1@c2,e2))  
         | E.Prod p    => filterSca(c1,p)  
         | E.Const _   => (1,e1)  
         | E.ConstR _  => (1,e1)  
         | E.Partial _ => err("Sum of Partial")  
         | E.Krn _     => err("Krn used before expand")  
         | E.Value _   => err("Value used before expand")  
         | E.Img _     => err("Probe used before expand")  
45          (*end case*))          (*end case*))
 *)  
   
     (* mkapply:mu list*ein_exp->int*ein_exp  
     * rewrite Apply  
     *)  
     fun mkapply(d1,e1)=let  
   
         val (c,g) =(case e1  
         of E.Lift e   => (1,E.Const 0)  
         | E.Sqrt e  => let  
             val half=E.Div(E.Const 1 ,E.Const 2)  
             val  E.Partial dels=d1  
             val del0=E.Partial([List.hd(dels)])  
             val deln=E.Partial( List.tl(dels))  
             val applydel0=E.Apply(del0,e)  
             (*distribute just one of the derivatives over the sqrt.*)  
             val g=(case deln  
                 of E.Partial []=>  E.Prod[half, E.Div(applydel0,e1)]  
                 | _  =>  E.Prod[half,E.Apply(deln, E.Div(applydel0,e1))]  
                 (*end case*))  
             val _ = testp["\n*****\n found sqrt \n",  
                     P.printbody(E.Apply(d1,e1)),"\n==>\n",P.printbody g,"\n ***\n\n"]  
             in  
                 (1,g)  
             end  
         | E.Prod []   => err("Apply of empty product")  
         | E.Add []    => err("Apply of empty Addition")  
         | E.Conv(v, alpha, h, d2)    =>let  
             val E.Partial d3=d1  
             in  
                 (1,E.Conv(v,alpha,h,d2@d3))  
             end  
         | E.Field _   => (0,E.Apply(d1,e1))  
         | E.Probe _   => (0,E.Apply(d1,e1))  
         | E.Apply(E.Partial d2,e2)  => let  
             val E.Partial d3=d1  
             in  
                 (1,E.Apply(E.Partial(d3@d2),e2))  
             end  
         | E.Apply _   => err" Apply of non-Partial expression"  
         | E.Sum(c2,e2)=> (1,E.Sum(c2,E.Apply(d1,e2)))  
         | E.Neg e2    => (1,E.Neg(E.Apply(d1,e2)))  
         | E.Add e     => (1,E.Add (List.map (fn(a)=>E.Apply(d1,a)) e))  
         | E.Sub (a,b) => (1,E.Sub(E.Apply(d1,a),E.Apply(d1,b)))  
         | E.Div (g,E.Const b) =>(1,E.Div(E.Apply(d1,g),E.Const b))  
         (*| E.Div(E.Const c,e)=>(1,E.Div(E.Const c,E.Apply(d1,e)))*)  
         | E.Div (g,b) => let  
             val (c,EE)=(case filterField[b]  
             of (_,[]) => (1,E.Div(E.Apply(d1,g),b)) (*Division by a real*)  
             | (pre,h) => let  
                 (*quotient rule*)  
                 val _=testp["\n** Quotient rule"]  
                 val  E.Partial dels=d1  
                 val del0=E.Partial([List.hd(dels)])  
                 val deln=E.Partial( List.tl(dels))  
                 val g'=E.Apply(del0,g)  
                 val h'=E.Apply(del0,flatProd(h))  
                 val num=(case g  
                     of E.Const 1    => E.Prod[E.Const ~1,h']  
                     | E.Const c     => E.Neg(E.Prod[g,h'])  
                     | _             => E.Sub(E.Prod([g']@h),E.Prod[g,h'])  
                     (*end case*))  
                 val denom=E.Prod(pre@h@h)  
                     val e=(case deln  
                     of E.Partial []=>E.Div(num,denom)  
                     | _=>E.Apply(deln,E.Div(num,denom))  
                 (*end case*))  
                 in (1,e)  
                 end  
             (*end case*))  
             in  
                 (c,EE)  
             end  
         | E.Prod p =>let  
             val _ =testp["\n",P.printbody(E.Apply(d1,e1))]  
             val (pre, post)= filterField p  
             in (case post  
                 of []=> (1,E.Const 0)(*no fields in expression*)  
                 | _=>let  
                     val E.Partial d3=d1  
                     val (c,g)= mkProd(pre@[prodAppPartial(post,d3)])  
                     val _ = testp["\n*****\n Product rule \n",  
                         P.printbody(E.Apply(d1,e1)),"\n==>\n",P.printbody g,"\n ***\n\n"]  
                     in (1,g)  
                     end  
                 (*end case*))  
             end  
         | E.Const _   => (1,E.Const 0)(*err("Const without Lift")*)  
         | Ein.ConstR _          =>(1,E.Const 0)  
         | E.Tensor _  => err("Tensor without Lift")  
         | E.Delta _   => err("Apply of Delta")  
         | E.Epsilon _ => err("Apply of Eps")  
         | E.Eps2 _ => err("Apply of Eps")  
         | E.Partial _ => err("Apply of Partial")  
         | E.Krn _     => err("Krn used before expand")  
         | E.Value _   => err("Value used before expand")  
         | E.Img _     => err("Probe used before expand")  
         (*end case*))  
   
   
     in  
         (c,g)  
     end  
46    
47      (*mkprobe:ein_exp* ein_exp-> int ein_exp      (*mkprobe:ein_exp* ein_exp-> int ein_exp
48      *rewritten probe      *rewritten probe
49      *)      *)
50          fun mkprobe(e1,x)=let      fun mkprobe(b,x)=let
51          val (c,rtn)=(case e1          val (c,rtn)=(case b
52              of E.Lift e   => (1,e)              of (E.B _)              => (0,b)
             | E.Sqrt a    => (1,E.Sqrt(E.Probe(a,x)))  
             | E.PowReal(a,n1)    => (1,E.PowReal(E.Probe(a,x),n1))  
             | E.Prod []   => err("Probe of empty product")  
             | E.Prod p    => (1,E.Prod (List.map (fn(a)=>E.Probe(a,x)) p))  
             | E.Apply _   => (0,E.Probe(e1,x))  
             | E.Conv _    => (0,E.Probe(e1,x))  
             | E.Field _   => (0,E.Probe(e1,x))  
             | E.Sum(c,e') =>  (1,E.Sum(c,E.Probe(e',x)))  
             | E.Add e     => (1,E.Add (List.map (fn(a)=>E.Probe(a,x)) e))  
             | E.Sub (a,b) => (1,E.Sub(E.Probe(a,x),E.Probe(b,x)))  
             | E.Neg a    => (1,E.Neg(E.Probe(a,x)))  
             | E.Div (a,b) => (1,E.Div(E.Probe(a,x),E.Probe(b,x)))  
             | E.Const _   => (1,e1)(*err("Const without Lift")*)  
             | Ein.ConstR _          =>(1,e1)  
53              | E.Tensor _  => err("Tensor without Lift")              | E.Tensor _  => err("Tensor without Lift")
54              | E.Delta _   => (0,e1)              | E.G _                 => (0,b)
55              | E.Epsilon _ => (0,e1)              | E.Field _             => (0,E.Probe(b,x))
56              | E.Eps2 _    => (0,e1)              | E.Lift e1             => (1,e1)
57                | E.Conv _              => (0,E.Probe(b,x))
58              | E.Partial _ => err("Probe Partial")              | E.Partial _ => err("Probe Partial")
59                | E.Apply _             => (0,E.Probe(b,x))
60              | E.Probe _   => err("Probe of a Probe")              | E.Probe _   => err("Probe of a Probe")
             | E.Krn _     => err("Krn used before expand")  
61              | E.Value _   => err("Value used before expand")              | E.Value _   => err("Value used before expand")
62              | E.Img _     => err("Probe used before expand")              | E.Img _     => err("Probe used before expand")
63                | E.Krn _               => err("Krn used before expand")
64                | E.Sum(sx1,e1)         => (1,E.Sum(sx1,E.Probe(e1,x)))
65                | E.Op1(op1, e1)        => (1,E.Op1(op1, E.Probe(e1,x)))
66                | E.Op2(op2, e1,e2)     => (1,E.Op2(op2, E.Probe(e1,x), E.Probe(e2,x)))
67                | E.Opn(opn, [])        => err("Probe of empty operator")
68                | E.Opn(opn, es)        => (1,E.Opn(opn, List.map(fn e1=> E.Probe(e1,x)) es))
69              (*end case*))              (*end case*))
70          in          in
71              (c,rtn)              (c,rtn)
# Line 223  Line 77 
77      *)      *)
78      fun normalize (ee as Ein.EIN{params, index, body},args) = let      fun normalize (ee as Ein.EIN{params, index, body},args) = let
79        val changed = ref false        val changed = ref false
80        fun rewriteBody body =(case body        fun rewrite body =(case body
81          of E.Const _    => body          of E.B _                                => body
         | Ein.ConstR _          =>body  
82          | E.Tensor _    => body          | E.Tensor _    => body
83            | E.G _                                 => body
84                    (************** Field Terms **************)
85          | E.Field _     => body          | E.Field _     => body
86          | E.Delta _     => body          | E.Lift e1                             => E.Lift(rewrite e1)
         | E.Epsilon _   => body  
         | E.Eps2 _      => body  
87          | E.Conv _      => body          | E.Conv _      => body
88          | E.Partial _   => body          | E.Partial _   => body
89          | E.Krn _       => raise Fail"Krn before Expand"          | E.Apply(E.Partial [],e1)              => e1
         | E.Img _       => raise Fail"Img before Expand"  
         | E.Value _     => raise Fail"Value before Expand"  
             (*************Algebraic Rewrites **************)  
         | E.Neg(E.Neg e)    => rewriteBody e  
         | E.Neg e           => E.Neg(rewriteBody e)  
         | E.Lift e          => E.Lift(rewriteBody e)  
         | E.Sqrt e          => E.Sqrt(rewriteBody e)  
         | E.PowInt(e,n1)        => E.PowInt(rewriteBody e,n1)  
         | E.PowReal(e,n1)       => E.PowReal(rewriteBody e,n1)  
         | E.Add es          => let  
             val (change,body')= mkAdd(List.map rewriteBody es)  
             in if (change=1) then ( changed:=true;body') else body' end  
         (*| E.Sub(a, E.Field f)=> (changed:=true;E.Add[a, E.Neg(E.Field(f))])  
         | E.Sub(E.Sub(a,b),E.Sub(c,d))  => rewriteBody(E.Sub(E.Add[a,d],E.Add[b,c]))  
         | E.Sub(E.Sub(a,b),e2)          => rewriteBody (E.Sub(a,E.Add[b,e2]))  
         | E.Sub(e1,E.Sub(c,d))          => rewriteBody(E.Add([E.Sub(e1,c),d]))*)  
         | E.Sub (a,b)                   => E.Sub(rewriteBody a, rewriteBody b)  
         | E.Div(e1 as E.Tensor(_,[_]),e2 as E.Tensor(_,[]))=>  
                 rewriteBody (E.Prod[E.Div(E.Const 1, e2),e1])  
   
         | E.Div(E.Div(a,b),E.Div(c,d))  => rewriteBody(E.Div(E.Prod[a,d],E.Prod[b,c]))  
         | E.Div(E.Div(a,b),c)           => rewriteBody (E.Div(a, E.Prod[b,c]))  
         | E.Div(a,E.Div(b,c))           => rewriteBody (E.Div(E.Prod[a,c],b))  
         | E.Div (a, b)                  => (E.Div(rewriteBody a, rewriteBody b))  
             (**************Apply, Sum, Probe**************)  
         | E.Apply(E.Partial [],e)   => e  
90          | E.Apply(E.Partial d1, e1) =>          | E.Apply(E.Partial d1, e1) =>
91              let              let
92              val e2 = rewriteBody e1              val e2 = rewrite e1
93              val (c,e3)=mkapply(E.Partial d1,e2)              val (c,e3)=mkapply(E.Partial d1,e2)
94                    val _= testp["\nafter apply:",P.printbody body,"-->",P.printbody e3]
95              in              in
96                  (case c of 1=>(changed:=true;e3)| _ =>e3 (*end case*))                  (case c of 1=>(changed:=true;e3)| _ =>e3 (*end case*))
97              end              end
98          | E.Apply _                 => raise Fail" Not well-formed Apply expression"          | E.Apply _                             => err " Not well-formed Apply expression"
99          | E.Sum([],e)               => (changed:=true;rewriteBody e)          | E.Probe(e1,e2)              =>
         | E.Sum(c,e)                => let  
             val (c,e')=mkSum(c,rewriteBody e)  
             in  
                 (case c of 0 => e'|_ => (changed:=true;e'))  
             end  
         | E.Probe(u,v)              =>  
100              let              let
101              val (c',b')=mkprobe(rewriteBody u,rewriteBody v)                  val (c',b')=mkprobe(rewrite e1,rewrite e2)
102              in (case c'              in (case c'
103                  of 1=> (changed:=true;b')                  of 1=> (changed:=true;b')
104                  |_=> b'                  |_=> b'
105                  (*end case*))                  (*end case*))
106              end              end
107          (*************Product**************)              (************** Field Terms **************)
108          | E.Prod [] => raise Fail"missing elements in product"          | E.Value _                             => err "Value before Expand"
109          | E.Prod [e1] => rewriteBody e1          | E.Img _                               => err "Img before Expand"
110  (*          | E.Krn _                               => err "Krn before Expand"
111          | E.Prod((E.Add(e2))::e3)=>              (************** Sum **************)
112             (changed := true; E.Add(List.map (fn e=> E.Prod([e]@e3)) e2))  
113          | E.Prod((E.Sub(e2,e3))::e4)=>          | E.Sum([],e1)                           => (changed:=true;rewrite e1)
114              (changed :=true; E.Sub(E.Prod([e2]@e4), E.Prod([e3]@e4 )))*)          | E.Sum(sx1,e1)                            => let
115          | E.Prod((E.Div(e2,e3))::e4)=> (changed :=true; E.Div(E.Prod([e2]@e4), e3))                  val e2=rewrite e1
116  (*                  val (c,e')=mkSum(sx1,e2)
117          | E.Prod(e1::E.Add(e2)::e3)=>                  val _= testp["\nafter mksum:\n\t",P.printbody body,"\n\t-->",P.printbody e2,"\n\t-->",P.printbody e']
118              (changed := true; E.Add(List.map (fn e=> E.Prod([e1,e]@e3)) e2))                  in
119          | E.Prod(e1::E.Sub(e2,e3)::e4)=>                  (case c of 0 => e'|_ => (changed:=true;e'))
120              (changed :=true; E.Sub(E.Prod([e1,e2]@e4), E.Prod([e1,e3]@e4 )))                  end
121  *)              (*************Algebraic Rewrites Op1 **************)
122    
123            | E.Op1(E.Neg,e1)                       => (case e1
124                of E.Op1(E.Neg,e2)                  => rewrite e2
125                | E.B(E.Const 0)                    =>(changed:=true;zero)
126                | _                                 => E.Op1(E.Neg,rewrite e1)
127                (*end case*))
128            | E.Op1(op1,e1)                         => E.Op1(op1,rewrite e1)
129                (*************Algebraic Rewrites Op2 **************)
130            | E.Op2(E.Sub,e1,e2)                        => (case (e1,e2)
131                of (E.B(E.Const 0),_)                   => (changed:=true;setNeg(rewrite e2))
132                | (_,E.B(E.Const 0))                     => (changed:=true;rewrite e1)
133                | _                                 => setSub(rewrite e1, rewrite e2)
134                (*end case*))
135            | E.Op2(E.Div,e1,e2)                        =>(case (e1,e2)
136                of (E.B(E.Const 0),_)                    => (changed:=true;zero)
137                |(E.Op2(E.Div,a,b), E.Op2(E.Div,c,d))   => rewrite(setDiv(setProd[a,d],setProd[b,c]))
138                |(E.Op2(E.Div,a,b), c)   =>  rewrite (setDiv(a, setProd[b,c]))
139                | (a,E.Op2(E.Div,b,c))                   => rewrite (setDiv(setProd[a,c],b))
140                |  _                        => setDiv(rewrite e1, rewrite e2)
141                (*end case*))
142                (*************Algebraic Rewrites Opn **************)
143            | E.Opn(E.Add,es)          => let
144                val (change,body')= mkAdd(List.map rewrite es)
145                in if (change=1) then ( changed:=true;body') else body' end
146    
147          | E.Prod((e1 as E.Sqrt(s1))::(e2 as E.Sqrt(s2))::es)=>          (*************Product**************)
148              if(Eq.isEqual3(s1,s2,args)=0) then (changed :=true;s1)          | E.Opn(E.Prod,[])                                 => err "missing elements in product"
149            | E.Opn(E.Prod,[e1])                               => rewrite e1
150            | E.Opn(E.Prod,[E.Op1(E.Sqrt,s1),E.Op1(E.Sqrt,s2)])=>
151                if(Eq.isBodyEq(s1,s2)) then (changed :=true;s1)
152              else let              else let
153                  val a=rewriteBody e1                  val a=rewrite (E.Op1(E.Sqrt,s1))
154                  val b=rewriteBody (E.Prod([e2]@es))                  val b=rewrite (E.Op1(E.Sqrt,s2))
155                  val  (_,d)=mkProd ([a,b])                  val  (_,d)=mkProd ([a,b])
156                  in d                  in d
157                  end                  end
   
   
158          (*************Product EPS **************)          (*************Product EPS **************)
159            | E.Opn(E.Prod,(E.G(E.Epsilon e1)::ps))=> let
160          | E.Prod(E.Epsilon(i,j,k)::E.Apply(E.Partial d,e)::es)=>let              val E.G(E.Epsilon(i,j,k))=E.G(E.Epsilon e1)
161                val eps1=E.G(E.Epsilon(i,j,k))
162                val p1=List.hd(ps)
163                in (case ps
164                    of (E.Apply(E.Partial d,e)::es)=>let
165              val change= G.matchEps(0,d,[],[i,j,k])              val change= G.matchEps(0,d,[],[i,j,k])
166              in case (change,es)              in case (change,es)
167                  of (1,_) =>(changed:=true; E.Const 0)                          of (1,_)    => (changed:=true; zero)
168                  | (_,[]) =>E.Prod[E.Epsilon(i,j,k),rewriteBody (E.Apply(E.Partial d,e))]                          | (_,[])    => setProd[eps1,rewrite p1]
                 |(_,_)=> let  
                     val a=rewriteBody(E.Prod([E.Apply(E.Partial d,e)]@ es))  
                     val (_,b)=mkProd [E.Epsilon(i,j,k),a]  
                     in b end  
             end  
 (*  
         | E.Prod(E.Epsilon(i,j,k)::E.Conv(V1,[a1], h1, d1)::E.Conv(V,alpha, h, d)::es)=>let  
             val change= G.matchEps(0,alpha@d,[],[i,j,k])  
             in case (change,es)  
                 of (1,_) =>(changed:=true; E.Lift(E.Const 0))  
                 | (_,[]) =>E.Prod[E.Epsilon(i,j,k),E.Conv(V1,[a1], h1, d1),E.Conv(V,alpha, h, d)]  
169                  | (_,_) =>let                  | (_,_) =>let
170                      val a=rewriteBody(E.Prod([E.Conv(V1,[a1], h1, d1),E.Conv(V,alpha, h, d)]@ es))                              val a=rewrite(setProd([p1]@es))
171                      val (_,b) = mkProd [E.Epsilon(i,j,k),a]                              val (_,b)=mkProd [eps1,a]
172                      in b end                      in b end
173              end              end
174  *)                  | (E.Conv(V,alpha, h, d)::es)=>let
         | E.Prod(E.Epsilon(i,j,k)::E.Conv(V,alpha, h, d)::es)=>let  
175              val change= G.matchEps(0,d,[],[i,j,k])              val change= G.matchEps(0,d,[],[i,j,k])
176              in case (change,es)              in case (change,es)
177                  of (1,_) =>(changed:=true; E.Lift(E.Const 0))                          of (1,_)    => (changed:=true; E.Lift zero )
178                  | (_,[]) =>E.Prod[E.Epsilon(i,j,k),E.Conv(V,alpha, h, d)]                          | (_,[])    => setProd[eps1,p1]
179                  | (_,_) =>let                  | (_,_) =>let
180                      val a=rewriteBody(E.Prod([E.Conv(V,alpha, h, d)]@ es))                              val a=rewrite(setProd([p1]@es))
181                      val (_,b) = mkProd [E.Epsilon(i,j,k),a]                              val (_,b) = mkProd [eps1,a]
182                      in b end                      in b end
183              end              end
184          | E.Prod[(E.Epsilon(e1,e2,e3)), E.Tensor(_,[E.V i1,E.V i2])]=>                  | [E.Tensor(_,[E.V i1,E.V i2])] =>
185              if(e2=i1 andalso e3=i2)                      if(j=i1 andalso k=i2) then (changed :=true;zero) else body
186              then (changed :=true;E.Const(0))                  | _  => (case (G.epsToDels(eps1::ps))
             else body  
 (*  
         | E.Prod(E.Epsilon e1::E.Sum(c1,E.Prod(E.Sum(c2,E.Prod(E.Epsilon e2::es3))::es2))::es1) =>  
             (case G.epsToDels([E.Epsilon e1, E.Epsilon e2]@es3@es2@es1)  
             of (1,e,sx,_,_)=> (changed:=true; E.Sum(c1@c2@sx,e))  
             | (_,_,_,_,_)=>let  
                     val eA=rewriteBody(E.Epsilon e1)  
                     val eB=rewriteBody(E.Prod(E.Sum(c1,E.Prod(E.Sum(c2,E.Prod(E.Epsilon e2::es3))::es2))::es1))  
                     val (_,e)=mkProd([eA,eB])  
                 in  
                     e  
                 end  
             (*end case*))*)  
         | E.Prod(E.Epsilon eps1::ps)=> (case (G.epsToDels(E.Epsilon eps1::ps))  
187              of (1,e,[],_,_)      =>(changed:=true;e)(* Changed to Deltas *)              of (1,e,[],_,_)      =>(changed:=true;e)(* Changed to Deltas *)
188              | (1,e,sx,_,_)      =>(changed:=true;E.Sum(sx,e))              | (1,e,sx,_,_)      =>(changed:=true;E.Sum(sx,e))
                     (* Changed to Deltas *)  
189              | (_,_,_,_,[])   =>  body              | (_,_,_,_,[])   =>  body
190              | (_,_,_,epsAll,rest) => let              | (_,_,_,epsAll,rest) => let
191                  val p'=rewriteBody(E.Prod rest)                          val p'=rewrite(setProd rest)
192                  val(_,b)= mkProd(epsAll@[p'])                  val(_,b)= mkProd(epsAll@[p'])
193                  in b end                  in b end
194              (*end case*))              (*end case*))
195          | E.Prod(E.Sum(c1,E.Prod(E.Epsilon e1::es1))::E.Sum(c2,E.Prod(E.Epsilon e2::es2))::es) =>                  (*end case*))
196              (case G.epsToDels([E.Epsilon e1, E.Epsilon e2]@es1@es2@es)              end
197            | E.Opn(E.Prod,E.Sum(c1,e1)::E.Sum(c2,e2)::es)=>(case (e1,e2,es)
198                of (E.Opn(E.Prod,E.G(E.Epsilon e1)::es1),E.Opn(E.Prod,E.G(E.Epsilon e2)::es2),_) =>
199                    (case G.epsToDels([E.G(E.Epsilon e1), E.G(E.Epsilon e2)]@es1@es2@es)
200                  of (1,e,sx,_,_)=> (changed:=true; E.Sum(c1@c2@sx,e))                  of (1,e,sx,_,_)=> (changed:=true; E.Sum(c1@c2@sx,e))
201                  | (_,_,_,_,_)=>let                  | (_,_,_,_,_)=>let
202                      val eA=rewriteBody(E.Sum(c1,E.Prod(E.Epsilon e1::es1)))                          val eA=rewrite(E.Sum(c1,setProd(E.G(E.Epsilon e1)::es1)))
203                  val eB=rewriteBody(E.Prod(E.Sum(c2,E.Prod(E.Epsilon e2::es2))::es))                          val eB=rewrite(setProd(E.Sum(c2,setProd(E.G(E.Epsilon e2)::es2))::es))
204                  val (_,e)=mkProd([eA,eB])                  val (_,e)=mkProd([eA,eB])
205                  in                          in  e end
                     e  
                 end  
206              (*end case*))              (*end case*))
207          | E.Prod[E.Delta d, E.Neg e]=> (changed:=true;E.Neg(E.Prod[E.Delta d, e]))              | (_,_,[]) =>let
208          | E.Prod(E.Delta d::es)=>let                  val (_,b)=mkProd[rewrite(E.Sum(c1,e1)), rewrite(E.Sum(c2,e2))]
209              val (pre',eps, dels,post)= filterGreek(E.Delta d::es)                  in b end
210                |  _ =>let
211                    val e'=rewrite (E.Sum(c1,e1))
212                    val e2=rewrite(E.Opn(E.Prod,E.Sum(c2,e2)::es))
213                    val(_,b)=(case e2
214                        of E.Opn(E.Prod, p')=> mkProd([e']@p')
215                        | _ =>mkProd [e',e2])
216                    in b end
217                (*end case*))
218            | E.Opn(E.Prod,E.G(E.Delta d)::es)=> (case es
219                of [E.Op1(E.Neg, e1)]=> (changed:=true;setNeg(setProd[E.G(E.Delta d), e1]))
220                | _=>   let
221                    val (pre',eps, dels,post)= filterGreek(E.G(E.Delta d)::es)
222               val _= testp["\n\n Reduce delta--",P.printbody(body)]               val _= testp["\n\n Reduce delta--",P.printbody(body)]
223              val (change,a)=G.reduceDelta(eps, dels, post)              val (change,a)=G.reduceDelta(eps, dels, post)
224                val _= testp["\n\n ---delta moved--",P.printbody(a)]                val _= testp["\n\n ---delta moved--",P.printbody(a)]
225              in (case (change,a)              in (case (change,a)
226                  of (0, _)=> E.Prod [E.Delta d,rewriteBody(E.Prod es)]                      of (0, _)=> setProd [E.G(E.Delta d),rewrite(setProd es)]
227                  | (_, E.Prod p)=>let                      | (_, E.Opn(E.Prod, p))=>let
228                      val (_, p') = mkProd p                      val (_, p') = mkProd p
229                      in (changed:=true;p') end                      in (changed:=true;p') end
230                  | _ => (changed:=true;a )                  | _ => (changed:=true;a )
231                  (*end case*))                  (*end case*))
232              end              end
233        | E.Prod[e1,e2]=> let              (*end case*))
234              val (_,b)=mkProd[rewriteBody e1, rewriteBody e2]        | E.Opn(E.Prod,[e1,e2])=> let
235                val (_,b)=mkProd[rewrite e1, rewrite e2]
236              in b end              in b end
237        | E.Prod(e::es)=>let        | E.Opn(E.Prod,e1::es)=>let
238              val e'=rewriteBody e              val e'=rewrite e1
239              val e2=rewriteBody(E.Prod es)              val e2=rewrite(setProd es)
240              val(_,b)=(case e2              val(_,b)=(case e2
241                  of E.Prod p'=> mkProd([e']@p')                  of E.Opn(Prod, p')=> mkProd([e']@p')
242                  |_=>mkProd [e',e2])                  |_=>mkProd [e',e2])
243              in              in b end
                     b  
            end  
   
244      (*end case*))      (*end case*))
245    
246      val _=testp["\n********Normalize",P.printerE ee,"\n*****\n"]  
247        val _=testp["\n******** Start Normalize: \n",P.printerE ee,"\n*****\n"]
248      fun loop(body ,count) = let      fun loop(body ,count) = let
249          val _= testp["\n\n N =>",Int.toString(count),"--",P.printbody(body)]          val _= testp["\n\n N =>",Int.toString(count),"--",P.printbody(body)]
250          val body' = rewriteBody body          val body' = rewrite body
   
251          in          in
252              if !changed              if !changed
253              then  (changed := false ;loop(body',count+1))              then  (changed := false ;loop(body',count+1))
254              else (body',count)              else (body',count)
255          end          end
256      val _ =testp["\n ******************* \n Start Normalize \n\n "]  
257      val (b,count) = loop(body,0)      val (b,count) = loop(body,0)
258     val _ =testp["\n Out of normalize \n",P.printbody(b),     val _ =testp["\n Out of normalize \n",P.printbody(b),
259          "\n    Final CounterXX:",Int.toString(count),"\n\n"]          "\n    Final CounterXX:",Int.toString(count),"\n\n"]

Legend:
Removed from v.3033  
changed lines
  Added in v.3441

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