module Data.Array.Accelerate.Math.DFT.Roots (
rootsOfUnity, inverseRootsOfUnity,
) where
import Prelude as P
import Data.Array.Accelerate as A
import Data.Array.Accelerate.Data.Complex
rootsOfUnity
:: (Elt e, IsFloating e, Shape sh, Slice sh)
=> Exp (sh :. Int)
-> Acc (Array (sh:.Int) (Complex e))
rootsOfUnity sh =
let n = A.fromIntegral (A.indexHead sh)
in
A.generate sh (\ix -> let i = A.fromIntegral (A.indexHead ix)
k = 2 * pi * i / n
in
A.lift ( cos k :+ (sin k) ))
inverseRootsOfUnity
:: (Elt e, IsFloating e, Shape sh, Slice sh)
=> Exp (sh :. Int)
-> Acc (Array (sh:.Int) (Complex e))
inverseRootsOfUnity sh =
let n = A.fromIntegral (A.indexHead sh)
in
A.generate sh (\ix -> let i = A.fromIntegral (A.indexHead ix)
k = 2 * pi * i / n
in
A.lift ( cos k :+ sin k ))