Precompilation works

Author: dannys4

Project.toml

@@ -5,11 +5,14 @@ version = "0.2.2"
 
 [deps]
 AbstractFFTs = "621f4979-c628-5d54-868e-fcf4e3e8185c"
+ComputedFieldTypes = "459fdd68-db75-56b8-8c15-d717a790f88e"
 DocStringExtensions = "ffbed154-4ef7-542d-bbb7-c09d3a79fcae"
 Documenter = "e30172f5-a6a5-5a46-863b-614d45cd2de4"
+LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MuladdMacro = "46d2c3a1-f734-5fdb-9937-b9b9aeba4221"
 Primes = "27ebfcd6-29c5-5fa9-bf4b-fb8fc14df3ae"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 
 [compat]
 DocStringExtensions = "0.8"

src/FFTA.jl

@@ -1,10 +1,13 @@
 module FFTA
 
-using Primes, DocStringExtensions, MuladdMacro, AbstractFFTs, ComputedFieldTypes
-export fft, bfft
+using Primes, DocStringExtensions, Reexport, MuladdMacro, ComputedFieldTypes, LinearAlgebra
+@reexport using AbstractFFTs
+
+import AbstractFFTs: Plan
 
 include("callgraph.jl")
 include("algos.jl")
+include("plan.jl")
 
 #=
 """

src/plan.jl

@@ -1,8 +1,37 @@
-@computed struct FFTAPlan{T} <: Plan{T} where {T <: Union{Real, Complex}}
-    callgraph::CallGraph{(T<:Real) ? Complex{T} : T}
-    pinv::FFTAPlan{T}
+struct FFTAInvPlan{T} <: Plan{T} end
+
+@computed struct FFTAPlan{T<:Union{Real, Complex},N} <: Plan{T}
+    callgraph::NTuple{N, CallGraph{(T<:Real) ? Complex{T} : T}}
+    region::NTuple{N, Int}
+    dir::Direction
+    pinv::FFTAInvPlan{T}
 end
 
-function AbstractFFTs.plan_fft(x::AbstractArray{T,N}, region; kwargs...)::FFTAPlan{T}
+function AbstractFFTs.plan_fft(x::AbstractArray{T}, region; kwargs...)::FFTAPlan{T} where T
+    N = length(region)
     @assert N <= 2 "Only supports vectors and matrices"
+    if N == 1
+        g = CallGraph{T}(size(x,region[]))
+        pinv = FFTAInvPlan()
+        return FFTAPlan{T,N}((g,), region, FFT_FORWARD, pinv)
+    else
+        g1 = CallGraph{T}(size(x,region[1]))
+        g2 = CallGraph{T}(size(x,region[2]))
+        pinv = FFTAInvPlan()
+        return FFTAPlan{T,N}((g1,g2), region, FFT_FORWARD, pinv)
+    end
+end
+
+function AbstractFFTs.plan_bfft(p::FFTAPlan{T,N}) where {T,N}
+    return FFTAPlan{T,N}(p.callgraph, p.region, -p.dir, p.pinv)
+end
+
+function LinearAlgebra.mul!(y, p::FFTAPlan, x)
+    fft!(y, x, 1, 1, p.dir, p.callgraph[1].type, p.callgraph, 1)
+end
+
+function *(p::FFTAPlan, x)
+    y = similar(x)
+    LinearAlgebra.mul!(y, p, x)
+    y
 end