Skip to content

Commit 40cce00

Browse files
committed
update to new fftdims interface
1 parent da62b2a commit 40cce00

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

Diff for: src/definitions.jl

+5-5
Original file line numberDiff line numberDiff line change
@@ -591,8 +591,8 @@ function irfft_dim end
591591

592592
output_size(p::Plan) = _output_size(p, ProjectionStyle(p))
593593
_output_size(p::Plan, ::NoProjectionStyle) = size(p)
594-
_output_size(p::Plan, ::RealProjectionStyle) = rfft_output_size(size(p), region(p))
595-
_output_size(p::Plan, s::RealInverseProjectionStyle) = brfft_output_size(size(p), s.dim, region(p))
594+
_output_size(p::Plan, ::RealProjectionStyle) = rfft_output_size(size(p), fftdims(p))
595+
_output_size(p::Plan, s::RealInverseProjectionStyle) = brfft_output_size(size(p), s.dim, fftdims(p))
596596

597597
mutable struct AdjointPlan{T,P<:Plan} <: Plan{T}
598598
p::P
@@ -611,13 +611,13 @@ output_size(p::AdjointPlan) = size(p.p)
611611
Base.:*(p::AdjointPlan, x::AbstractArray) = _mul(p, x, ProjectionStyle(p.p))
612612

613613
function _mul(p::AdjointPlan{T}, x::AbstractArray, ::NoProjectionStyle) where {T}
614-
dims = region(p.p)
614+
dims = fftdims(p.p)
615615
N = normalization(T, size(p.p), dims)
616616
return (p.p \ x) / N
617617
end
618618

619619
function _mul(p::AdjointPlan{T}, x::AbstractArray, ::RealProjectionStyle) where {T}
620-
dims = region(p.p)
620+
dims = fftdims(p.p)
621621
N = normalization(T, size(p.p), dims)
622622
halfdim = first(dims)
623623
d = size(p.p, halfdim)
@@ -630,7 +630,7 @@ function _mul(p::AdjointPlan{T}, x::AbstractArray, ::RealProjectionStyle) where
630630
end
631631

632632
function _mul(p::AdjointPlan{T}, x::AbstractArray, ::RealInverseProjectionStyle) where {T}
633-
dims = region(p.p)
633+
dims = fftdims(p.p)
634634
N = normalization(real(T), output_size(p.p), dims)
635635
halfdim = first(dims)
636636
n = size(p.p, halfdim)

0 commit comments

Comments
 (0)