Optimize to prevent overflows

This commit is contained in:
Nuwan Yapa 2024-02-22 12:39:18 -05:00
parent cf4ed12b6b
commit a2272122c0
2 changed files with 4 additions and 2 deletions

View File

@ -2,7 +2,9 @@ using NuclearToolkit
using SpecialFunctions using SpecialFunctions
# Gaussian potentials in HO space # Gaussian potentials in HO space
N_nl(n, l) = (-1)^n * sqrt(1/sqrt(pi) * (1/2)^(l+1) * 2^(n+2*l+3) * factorial(n) / Iterators.prod((2*n+2*l+1):-2:1)) sqrt_factorial(n) = Iterators.prod(sqrt.(n:-1:1))
sqrt_double_factorial(n) = Iterators.prod(sqrt.(n:-2:1))
N_nl(n, l) = (-1)^n * sqrt_factorial(n) * sqrt(1/sqrt(pi) * (1/2)^(l+1) * 2^(n+2*l+3)) / sqrt_double_factorial(2*n+2*l+1)
prefactor(n, l, k) = (-1)^k * binomial(n + l + 1/2, n - k) / factorial(k) prefactor(n, l, k) = (-1)^k * binomial(n + l + 1/2, n - k) / factorial(k)
Talmi(l, R, k1, k2) = (1/2) / (1 + 1/R^2)^(3/2 + l + k1 + k2) * gamma(3/2 + l + k1 + k2) Talmi(l, R, k1, k2) = (1/2) / (1 + 1/R^2)^(3/2 + l + k1 + k2) * gamma(3/2 + l + k1 + k2)
V_Gaussian(R, l, n1, n2) = N_nl(n1, l) * N_nl(n2, l) * sum([prefactor(n1, l, k1) * prefactor(n2, l, k2) * Talmi(l, R, k1, k2) for (k1, k2) in Iterators.product(0:n1, 0:n2)]) V_Gaussian(R, l, n1, n2) = N_nl(n1, l) * N_nl(n2, l) * sum([prefactor(n1, l, k1) * prefactor(n2, l, k2) * Talmi(l, R, k1, k2) for (k1, k2) in Iterators.product(0:n1, 0:n2)])

View File

@ -4,7 +4,7 @@ include("ho_basis.jl")
l = 0 l = 0
V0 = -10 V0 = -10
R = 1 R = 1
n_max = 10 n_max = 20
ns = collect(0:n_max) ns = collect(0:n_max)
ls = fill(l, n_max + 1) ls = fill(l, n_max + 1)