Add precompile directives (#17)

* Add precompile directives

* version bump to v0.1.16

* don't speclialize deepcopy to reduce invalidations

Author: jishnub

Project.toml

@@ -1,6 +1,6 @@
 name = "SphericalHarmonics"
 uuid = "c489a379-e885-57ff-9236-bd896d33c250"
-version = "0.1.15"
+version = "0.1.16"
 
 [deps]
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"

src/SphericalHarmonics.jl

@@ -52,21 +52,22 @@ function Base.summary(io::IO, P::AssociatedLegendrePolynomials)
     end
 end
 
-function Base.deepcopy_internal(S::SphericalHarmonicsCache, stackdict::IdDict)
-    _C = Base.deepcopy_internal(S.C, stackdict)
-    _P = Base.deepcopy_internal(S.P, stackdict)
-    _Y = Base.deepcopy_internal(S.Y, stackdict)
-    _SHT = Base.deepcopy_internal(S.SHType, stackdict)
-    typeof(S)(_C, _P, _Y, _SHT)
-end
-
-function Base.deepcopy_internal(P::AssociatedLegendrePolynomials, stackdict::IdDict)
-    _cosθ = Base.deepcopy_internal(P.cosθ, stackdict)
-    _lmax = Base.deepcopy_internal(P.lmax, stackdict)
-    _P = Base.deepcopy_internal(P.P, stackdict)
-    _initialized = Base.deepcopy_internal(P.initialized, stackdict)
-    typeof(P)(_cosθ, _lmax, _P, _initialized)
-end
+# Optimized deepcopy, but this leads to a lot of invalidations
+#function Base.deepcopy_internal(S::SphericalHarmonicsCache, stackdict::IdDict)
+#    _C = Base.deepcopy_internal(S.C, stackdict)
+#    _P = Base.deepcopy_internal(S.P, stackdict)
+#    _Y = Base.deepcopy_internal(S.Y, stackdict)
+#    _SHT = Base.deepcopy_internal(S.SHType, stackdict)
+#    typeof(S)(_C, _P, _Y, _SHT)
+#end
+
+#function Base.deepcopy_internal(P::AssociatedLegendrePolynomials, stackdict::IdDict)
+#    _cosθ = Base.deepcopy_internal(P.cosθ, stackdict)
+#    _lmax = Base.deepcopy_internal(P.lmax, stackdict)
+#    _P = Base.deepcopy_internal(P.P, stackdict)
+#    _initialized = Base.deepcopy_internal(P.initialized, stackdict)
+#    typeof(P)(_cosθ, _lmax, _P, _initialized)
+#end
 
 function SphericalHarmonicsCache(T::Type, lmax::Int, ::Type{m_range}, SHType) where {m_range}
     C = compute_coefficients(T, lmax)
@@ -1130,4 +1131,6 @@ function sphericalharmonic(θ, ϕ, l::Integer, m::Integer, SHType::HarmonicType
     norm * P
 end
 
+include("precompile.jl")
+
 end

src/precompile.jl

@@ -0,0 +1,10 @@
+precompile(SphericalHarmonicsCache, (Type{Float64}, Int, Type{FullRange}, ComplexHarmonics))
+precompile(SphericalHarmonicsCache, (Type{Float64}, Int, Type{FullRange}, RealHarmonics))
+precompile(SphericalHarmonicsCache, (Type{Float64}, Int, Type{ZeroTo}, ComplexHarmonics))
+precompile(SphericalHarmonicsCache, (Type{Float64}, Int, Type{ZeroTo}, RealHarmonics))
+_S = SphericalHarmonicsCache(Float64, 0, FullRange, ComplexHarmonics())
+precompile(computePlmx!, (typeof(_S), Float64, Int))
+precompile(computePlmx!, (typeof(_S.P), Float64, Int))
+precompile(computePlmcostheta!, (typeof(_S), Float64, Int))
+precompile(computePlmcostheta!, (typeof(_S.P), Float64, Int))
+precompile(computeYlm!, (typeof(_S), Float64, Float64, Int))