tgorordo/WignerSymbols.jl
1
0
Fork 0
mirror of https://github.com/tgorordo/WignerSymbols.jl.git synced 2026-06-05 15:42:15 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix fast path in conversion

Browse source
This commit is contained in:
Jutho Haegeman 2019-07-29 17:49:20 +02:00
parent 2800b274ea
commit 80cc592dd3
1 changed files with 14 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/WignerSymbols.jl
View file

@ -10,6 +10,8 @@ include("primefactorization.jl")
const Wigner3j = Dict{Tuple{UInt,UInt,UInt,Int,Int},Tuple{Rational{BigInt},Rational{BigInt}}}() const Wigner3j = Dict{Tuple{UInt,UInt,UInt,Int,Int},Tuple{Rational{BigInt},Rational{BigInt}}}()
const Wigner6j = Dict{NTuple{6,UInt},Tuple{Rational{BigInt},Rational{BigInt}}}() const Wigner6j = Dict{NTuple{6,UInt},Tuple{Rational{BigInt},Rational{BigInt}}}()
const FASTCUTOFF = convert(BigInt, typemax(Int))
function __init__() function __init__()
global bigone, bigprimetable, Wigner3j, Wigner6j global bigone, bigprimetable, Wigner3j, Wigner6j
bigone[] = big(1) bigone[] = big(1)
@ -103,8 +105,12 @@ function wigner3j(T::Type{<:AbstractFloat}, j₁, j₂, j₃, m₁, m₂, m₃ =
Wigner3j[(β₁, β₂, β₃, α₁, α₂)] = (r,s) Wigner3j[(β₁, β₂, β₃, α₁, α₂)] = (r,s)
end end
if T != BigFloat && all((s.num, s.den, r.num, r.den) .< FASTCUTOFF)
sn, sd, rn, rd = convert.(T, (s.num, s.den, r.num, r.den)) sn, sd, rn, rd = convert.(T, (s.num, s.den, r.num, r.den))
return sgn*(sn/sd)*sqrt(rn/rd) return sgn*(sn/sd)*sqrt(rn/rd)
else
return convert(T, sgn*s*sqrt(r))
end
end end
""" """
@ -202,8 +208,12 @@ function wigner6j(T::Type{<:AbstractFloat}, j₁, j₂, j₃, j₄, j₅, j₆)
Wigner6j[(β₁, β₂, β₃, α₁, α₂, α₃)] = (r, s) Wigner6j[(β₁, β₂, β₃, α₁, α₂, α₃)] = (r, s)
end end
if T != BigFloat && all((s.num, s.den, r.num, r.den) .< FASTCUTOFF)
sn, sd, rn, rd = convert.(T, (s.num, s.den, r.num, r.den)) sn, sd, rn, rd = convert.(T, (s.num, s.den, r.num, r.den))
return (sn/sd)*sqrt(rn/rd) return (sn/sd)*sqrt(rn/rd)
else
return convert(T, s*sqrt(r))
end
end end
""" """