From 105387ec8f07a26dfcb0b2c81eecae2b14d4e0b0 Mon Sep 17 00:00:00 2001 From: Mike Boyle Date: Fri, 26 Jul 2024 21:11:24 -0400 Subject: [PATCH] Fix a few imports --- src/SphericalFunctions.jl | 18 ++++++++++++------ src/evaluate.jl | 2 +- src/map2salm.jl | 2 +- 3 files changed, 14 insertions(+), 8 deletions(-) diff --git a/src/SphericalFunctions.jl b/src/SphericalFunctions.jl index db05a25..22827a9 100644 --- a/src/SphericalFunctions.jl +++ b/src/SphericalFunctions.jl @@ -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} diff --git a/src/evaluate.jl b/src/evaluate.jl index 85ca792..5a21129 100644 --- a/src/evaluate.jl +++ b/src/evaluate.jl @@ -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) diff --git a/src/map2salm.jl b/src/map2salm.jl index f191ece..50fc7cf 100644 --- a/src/map2salm.jl +++ b/src/map2salm.jl @@ -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