Skip to content

Commit ef84edf

Browse files
committed
update to new fftdims interface
1 parent 5d5c06c commit ef84edf

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/definitions.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -592,8 +592,8 @@ function irfft_dim end
592592

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

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

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

620620
function _mul(p::AdjointPlan{T}, x::AbstractArray, ::RealProjectionStyle) where {T}
621-
dims = region(p.p)
621+
dims = fftdims(p.p)
622622
N = normalization(T, size(p.p), dims)
623623
halfdim = first(dims)
624624
d = size(p.p, halfdim)
@@ -631,7 +631,7 @@ function _mul(p::AdjointPlan{T}, x::AbstractArray, ::RealProjectionStyle) where
631631
end
632632

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

0 commit comments

Comments
 (0)