module FRP.Yampa.Integration
(
integral
, imIntegral
, impulseIntegral
, count
, derivative
, iterFrom
)
where
import Control.Arrow
import Data.VectorSpace
import FRP.Yampa.Event
import FRP.Yampa.Hybrid
import FRP.Yampa.InternalCore (SF(..), SF'(..), DTime)
{-# INLINE integral #-}
integral :: (Fractional s, VectorSpace a s) => SF a a
integral :: forall s a. (Fractional s, VectorSpace a s) => SF a a
integral = SF {sfTF :: a -> Transition a a
sfTF = forall {a}. (VectorSpace a a, Fractional a) => a -> Transition a a
tf0}
where
tf0 :: a -> Transition a a
tf0 a
a0 = (forall {a} {a}.
(VectorSpace a a, Fractional a) =>
a -> a -> SF' a a
integralAux a
igrl0 a
a0, a
igrl0)
igrl0 :: a
igrl0 = forall v a. VectorSpace v a => v
zeroVector
integralAux :: a -> a -> SF' a a
integralAux a
igrl a
a_prev = forall a b. (DTime -> a -> Transition a b) -> SF' a b
SF' forall {a} {p}.
(VectorSpace a a, Real p, Fractional a) =>
p -> a -> (SF' a a, a)
tf
where
tf :: p -> a -> (SF' a a, a)
tf p
dt a
a = (a -> a -> SF' a a
integralAux a
igrl' a
a, a
igrl')
where
igrl' :: a
igrl' = a
igrl forall v a. VectorSpace v a => v -> v -> v
^+^ forall a b. (Real a, Fractional b) => a -> b
realToFrac p
dt forall v a. VectorSpace v a => a -> v -> v
*^ a
a_prev
imIntegral :: (Fractional s, VectorSpace a s) => a -> SF a a
imIntegral :: forall s a. (Fractional s, VectorSpace a s) => a -> SF a a
imIntegral = ((\ a
_ a
a' DTime
dt a
v -> a
v forall v a. VectorSpace v a => v -> v -> v
^+^ forall a b. (Real a, Fractional b) => a -> b
realToFrac DTime
dt forall v a. VectorSpace v a => a -> v -> v
*^ a
a') forall a b. (a -> a -> DTime -> b -> b) -> b -> SF a b
`iterFrom`)
iterFrom :: (a -> a -> DTime -> b -> b) -> b -> SF a b
a -> a -> DTime -> b -> b
f iterFrom :: forall a b. (a -> a -> DTime -> b -> b) -> b -> SF a b
`iterFrom` b
b = forall a b. (a -> Transition a b) -> SF a b
SF (b -> a -> Transition a b
iterAux b
b)
where
iterAux :: b -> a -> Transition a b
iterAux b
b a
a = (forall a b. (DTime -> a -> Transition a b) -> SF' a b
SF' (\ DTime
dt a
a' -> b -> a -> Transition a b
iterAux (a -> a -> DTime -> b -> b
f a
a a
a' DTime
dt b
b) a
a'), b
b)
derivative :: (Fractional s, VectorSpace a s) => SF a a
derivative :: forall s a. (Fractional s, VectorSpace a s) => SF a a
derivative = SF {sfTF :: a -> Transition a a
sfTF = forall {a} {b} {b} {a}.
(Fractional a, VectorSpace b a, VectorSpace b a) =>
b -> (SF' b b, b)
tf0}
where
tf0 :: b -> (SF' b b, b)
tf0 b
a0 = (forall {b} {a}. (VectorSpace b a, Fractional a) => b -> SF' b b
derivativeAux b
a0, forall v a. VectorSpace v a => v
zeroVector)
derivativeAux :: b -> SF' b b
derivativeAux b
a_prev = forall a b. (DTime -> a -> Transition a b) -> SF' a b
SF' forall {a} {a}.
(VectorSpace b a, Real a, Fractional a) =>
a -> b -> (SF' b b, b)
tf
where
tf :: a -> b -> (SF' b b, b)
tf a
dt b
a = (b -> SF' b b
derivativeAux b
a, (b
a forall v a. VectorSpace v a => v -> v -> v
^-^ b
a_prev) forall v a. VectorSpace v a => v -> a -> v
^/ forall a b. (Real a, Fractional b) => a -> b
realToFrac a
dt)
impulseIntegral :: (Fractional k, VectorSpace a k) => SF (a, Event a) a
impulseIntegral :: forall k a. (Fractional k, VectorSpace a k) => SF (a, Event a) a
impulseIntegral = (forall s a. (Fractional s, VectorSpace a s) => SF a a
integral forall (a :: * -> * -> *) b c b' c'.
Arrow a =>
a b c -> a b' c' -> a (b, b') (c, c')
*** forall b a. (b -> a -> b) -> b -> SF (Event a) b
accumHoldBy forall v a. VectorSpace v a => v -> v -> v
(^+^) forall v a. VectorSpace v a => v
zeroVector) forall (a :: * -> * -> *) b c d.
Arrow a =>
a b c -> (c -> d) -> a b d
>>^ forall a b c. (a -> b -> c) -> (a, b) -> c
uncurry forall v a. VectorSpace v a => v -> v -> v
(^+^)
count :: Integral b => SF (Event a) (Event b)
count :: forall b a. Integral b => SF (Event a) (Event b)
count = forall b a. (b -> a -> b) -> b -> SF (Event a) (Event b)
accumBy (\b
n a
_ -> b
n forall a. Num a => a -> a -> a
+ b
1) b
0