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 []