]> gerrit.simantics Code Review - simantics/platform.git/blob - tests/org.simantics.scl.compiler.tests/src/org/simantics/scl/compiler/tests/scl/Polynomials.scl
Supply SVG text editor with element measurement context
[simantics/platform.git] / tests / org.simantics.scl.compiler.tests / src / org / simantics / scl / compiler / tests / scl / Polynomials.scl
1 import "Prelude"
2
3 data Poly a = Poly [a]
4
5 normalize l = go (length l)
6   where
7     go i = if i > 0 && l!(i-1)==zero
8            then go (i-1)
9            else take i l 
10
11 instance (Additive a) => Additive (Poly a) where
12     zero = Poly []
13     Poly a + Poly b = 
14         Poly ( 
15             normalize (
16                zipWith (+) a b + 
17                  if la > lb
18                  then drop lb a
19                  else drop la b
20             )
21         )
22           where
23             la = length a
24             lb = length b        
25
26 instance (Ring a) => Ring (Poly a) where
27     one = Poly [one]
28     neg (Poly l) = Poly (map neg l)
29     a - b = a + (neg b)
30     Poly a * Poly b = 
31         Poly ( if aDeg < bDeg
32                then [ segSum n 0        n    | n <- [0     ..aDeg]   ]
33                   + [ segSum n 0        aDeg | n <- [aDeg+1..bDeg]   ]
34                   + [ segSum n (n-bDeg) aDeg | n <- [bDeg+1..sumDeg] ]
35                else [ segSum n 0        n    | n <- [0     ..bDeg]   ]
36                   + [ segSum n (n-bDeg) n    | n <- [bDeg+1..aDeg]   ]
37                   + [ segSum n (n-bDeg) aDeg | n <- [aDeg+1..sumDeg] ]
38              )
39           where 
40             aDeg = length a - 1
41             bDeg = length b - 1
42             sumDeg = aDeg + bDeg
43             segSum n low high = sum [ a!i * b!(n-i) | i <- [low..high] ]
44     fromInteger x = Poly [fromInteger x]
45             
46 a = Poly [4.0,5.0,8.0,3.0,2.0,1.0]
47 b = Poly [1.0,0.0,2.0,1.0]
48 main = a * a + a * b + b * a + b * b - (a+b)*(a+b)
49 --
50 []