module Data.Array.Repa.Algorithms.DFT.Center
( center1d
, center2d
, center3d)
where
import Data.Array.Repa
import Data.Array.Repa.Algorithms.Complex
center1d :: Array DIM1 Complex -> Array DIM1 Complex
center1d arr
= traverse arr id
(\get ix@(_ :. x) -> ((1) ^ x) * get ix)
center2d :: Array DIM2 Complex -> Array DIM2 Complex
center2d arr
= traverse arr id
(\get ix@(_ :. y :. x) -> ((1) ^ (y + x)) * get ix)
center3d :: Array DIM3 Complex -> Array DIM3 Complex
center3d arr
= traverse arr id
(\get ix@(_ :. z :. y :. x) -> ((1) ^ (z + y + x)) * get ix)