module SDR.Demod (
fmDemodStr,
fmDemodVec,
fmDemod
) where
import Data.Complex
import Data.Vector.Generic as VG
import Data.Vector.Fusion.Stream
import SDR.VectorUtils
import Pipes
fmDemodStr :: (RealFloat a)
=> Complex a
-> Stream (Complex a)
-> Stream a
fmDemodStr = mapAccumMV func
where
func last sample = return (sample, phase (sample * conjugate last))
fmDemodVec :: (RealFloat a, Vector v (Complex a), Vector v a)
=> Complex a
-> v (Complex a)
-> v a
fmDemodVec init = unstream . fmDemodStr init . stream
fmDemod :: (RealFloat a, Vector v (Complex a), Vector v a) => Pipe (v (Complex a)) (v a) IO ()
fmDemod = func 0
where
func lastSample = do
dat <- await
yield $ fmDemodVec lastSample dat
func $ VG.unsafeIndex dat (VG.length dat 1)