blob: bffab8721acaf4aff250263574b845d3b9a6ecff (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
{-# LANGUAGE TypeFamilies #-}
{- |
Module : Numeric.GSL.Fourier
Copyright : (c) Alberto Ruiz 2006
License : GPL
Maintainer : Alberto Ruiz
Stability : provisional
Fourier Transform.
<http://www.gnu.org/software/gsl/manual/html_node/Fast-Fourier-Transforms.html#Fast-Fourier-Transforms>
-}
module Numeric.GSL.Fourier (
fft,
ifft
) where
import Numeric.LinearAlgebra.HMatrix
import Numeric.GSL.Internal
import Foreign.C.Types
import System.IO.Unsafe (unsafePerformIO)
genfft code v = unsafePerformIO $ do
r <- createVector (size v)
(v `applyRaw` (r `applyRaw` id)) (c_fft code) #|"fft"
return r
foreign import ccall unsafe "gsl-aux.h fft" c_fft :: CInt -> TCV (TCV Res)
{- | Fast 1D Fourier transform of a 'Vector' @(@'Complex' 'Double'@)@ using /gsl_fft_complex_forward/. It uses the same scaling conventions as GNU Octave.
>>> fft (fromList [1,2,3,4])
fromList [10.0 :+ 0.0,(-2.0) :+ 2.0,(-2.0) :+ 0.0,(-2.0) :+ (-2.0)]
-}
fft :: Vector (Complex Double) -> Vector (Complex Double)
fft = genfft 0
-- | The inverse of 'fft', using /gsl_fft_complex_inverse/.
ifft :: Vector (Complex Double) -> Vector (Complex Double)
ifft = genfft 1
|