Skip to content

Commit

Permalink
Fix a few imports
Browse files Browse the repository at this point in the history
  • Loading branch information
moble committed Jul 27, 2024
1 parent d317f7b commit 105387e
Show file tree
Hide file tree
Showing 3 changed files with 14 additions and 8 deletions.
18 changes: 12 additions & 6 deletions src/SphericalFunctions.jl
Original file line number Diff line number Diff line change
@@ -1,11 +1,17 @@
module SphericalFunctions

using FFTW, AbstractFFTs, FastTransforms
using LinearAlgebra, ProgressMeter, Quaternionic, OffsetArrays, StaticArrays
import SpecialFunctions, DoubleFloats
import LoopVectorization: @turbo
import Hwloc: num_physical_cores
import Base.Threads: @threads, nthreads, threadid
using FastTransforms: FFTW, fft, fftshift!, ifft, ifftshift!, irfft,
plan_bfft!, plan_fft, plan_fft!
using LinearAlgebra: LinearAlgebra, Bidiagonal, Diagonal, convert, ldiv!, mul!
using OffsetArrays: OffsetArray, OffsetVector
using ProgressMeter: Progress, next!
using Quaternionic: Quaternionic, Rotor, from_spherical_coordinates,
to_euler_phases, to_spherical_coordinates
using StaticArrays: @SVector
using SpecialFunctions, DoubleFloats
using LoopVectorization: @turbo
using Hwloc: num_physical_cores
using Base.Threads: @threads, nthreads

const MachineFloat = Union{Float16, Float32, Float64}

Expand Down
2 changes: 1 addition & 1 deletion src/evaluate.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
### 2. This is probably a much more significant advantage for ALFs.

using .SphericalFunctions: complex_powers!
using Quaternionic: AbstractQuaternion, to_euler_phases!
using Quaternionic: AbstractQuaternion, to_euler_phases

@inline ϵ(m) = ifelse(m > 0 && isodd(m), -1, 1)

Expand Down
2 changes: 1 addition & 1 deletion src/map2salm.jl
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ function plan_map2salm(map_data::AbstractArray{Complex{T}}, spin::Int, ℓmax::I
H_rec_coeffs = H_recursion_coefficients(ℓmax, T)
weight = clenshaw_curtis(Nϑ, T)
expiθ = complex_powers(cis/ T(Nϑ-1)), Nϑ-1)
ϵs = SphericalFunctions.ϵ(-spin)
ϵs = ϵ(-spin)
extra_dims = Base.Iterators.product((1:e for e in Nextra)...)
fftplan = T<:MachineFloat ? plan_fft(map_data[:, 1, first(extra_dims)...]) : nothing

Expand Down

0 comments on commit 105387e

Please sign in to comment.