Update Project.toml

Signed-off-by: ErikQQY <[email protected]>

Author: ErikQQY

Project.toml

@@ -1,7 +1,7 @@
 name = "FractionalTransforms"
 uuid = "e50ca838-b4f0-4a10-ad18-4b920bf1ae5c"
 authors = ["Qingyu Qu <[email protected]>"]
-version = "0.1.1"
+version = "0.1.2"
 
 [deps]
 DSP = "717857b8-e6f2-59f4-9121-6e50c889abd2"

src/frft.jl

@@ -1,3 +1,5 @@
+using FFTW, DSP
+
 """
     frft(signal, α)
 
@@ -37,23 +39,23 @@ function frft(signal, α)::Vector{ComplexF64}
         y = reverse(signal, dims=1)
         return y
     elseif α == 3
-        y = fft([signal[M+1:N]; signal[1:M]])*sqrt(N)
-        y = [y[M+2 : N]; y[1:M+1]]
+        y = ifft([signal[M+1:N]; signal[1:M]])*sqrt(N)
+        y = [y[M+2:N]; y[1:M+1]]
         return y
     end
 
     if α > 2
         signal = reverse(signal, dims=1)
         α = α-2
     end
-    if 1.5 < α < 2
-        y = fft([signal[M+1:N]; signal[1:M]]) ./sqrt(N)
-        y = [y[M+2 : N]; y[1:M+1]]
+    if α > 1.5
+        signal = fft([signal[M+1:N]; signal[1:M]]) ./sqrt(N)
+        signal = [signal[M+2 : N]; signal[1:M+1]]
         α = α-1
     end
     if α < 0.5
-        y = fft([signal[M+1:N]; signal[1:M]]) .*sqrt(N)
-        y = [y[M+2 : N]; y[1:M+1]]
+        signal = ifft([signal[M+1:N]; signal[1:M]]) .*sqrt(N)
+        signal = [signal[M+2:N]; signal[1:M+1]]
         α = α+1
     end
 
@@ -126,4 +128,6 @@ function sinc_interp(x, rate)
     h =  sinc.(collect(-(N-1-1/rate):1/rate:(N-1-1/rate)))
     out = conv(y, h)
     out = out[(rate*N-rate):(end-rate*N+rate+1)]
-end
+end
+
+frft([2, 2, 2], 0.3)

test/frft.jl

@@ -2,8 +2,25 @@ using FractionalTransforms
 using Test
 
 @testset "Test FRFT" begin
-    @test isapprox(real(frft([1,2,3], 0.5)), [0.100464613588213, 3.07460010423315, 1.31566432887283], atol=1e-5)
-    @test isapprox(imag(frft([1,2,3], 0.5)), [-0.259409480977277, 0.199349381540632, -0.93645356561191], atol=1e-5)
+    @test isapprox(real(frft([1, 2, 3], 0.5)), [0.100464613588213, 3.07460010423315, 1.31566432887283], atol=1e-5)
+    @test isapprox(imag(frft([1, 2, 3], 0.5)), [-0.259409480977277, 0.199349381540632, -0.93645356561191], atol=1e-5)
+
+    @test isapprox(real(frft([1, 2, 3, 4, 5], 0.3)), [0.6219417958205868
+    1.9285647455299149
+    2.930147902457687
+    4.781892308544647
+    1.1657970121710457]; atol=1e-6)
+    @test isapprox(imag(frft([1, 2, 3, 4, 5], 0.3)), [-0.620033106942918
+    -0.3534712784226586
+     1.0387260647856131
+    -1.0039877236507386
+    -3.2564984970643764]; atol=1e-6)
+
+    @test isapprox(real(frft([1, 2, 3, 4, 5], 1)), [0.0
+    0.0
+    6.7082039324993685
+    0.0
+    0.0]; atol=1e-6)
 end
 
 @testset "Test FRFT auxillary functions" begin