yapanuwan
/
NuclearRMF
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Only use BigFloat internally

This commit is contained in:
Nuwan Yapa 2024-07-17 15:54:19 -04:00
parent e389adf885
commit 429c2708f4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dirac.jl
View File

@ -25,7 +25,7 @@ end
"Solve the Dirac equation and return the wave function u(r)=[g(r), f(r)] where "Solve the Dirac equation and return the wave function u(r)=[g(r), f(r)] where
divs is the number of mesh divisions so solution would be returned as a 2×(1+divs) matrix, divs is the number of mesh divisions so solution would be returned as a 2×(1+divs) matrix,
refine determines whether to switch to high-precision mode and optimize the energy beforehand (assuming a bound state), refine determines whether to internally enable high-precision mode and optimize the energy beforehand (assuming a bound state),
the other parameters are the same from dirac!(...)." the other parameters are the same from dirac!(...)."
function solveWf(κ, p, E, Φ0, W0, B0, A0, r_max, divs; refine=true, normalize=true) function solveWf(κ, p, E, Φ0, W0, B0, A0, r_max, divs; refine=true, normalize=true)
if refine if refine
@ -41,7 +41,7 @@ function solveWf(κ, p, E, Φ0, W0, B0, A0, r_max, divs; refine=true, normalize=
prob = ODEProblem(dirac!, convert.(dtype, [0, 1]), (0, r_max)) prob = ODEProblem(dirac!, convert.(dtype, [0, 1]), (0, r_max))
sol = solve(prob, algo, p=(κ, p, E, Φ0, W0, B0, A0), saveat=r_max/divs) sol = solve(prob, algo, p=(κ, p, E, Φ0, W0, B0, A0), saveat=r_max/divs)
wf = hcat(sol.u...) wf = Float64.(hcat(sol.u...))
if normalize if normalize
norm = sum(wf .* wf) * r_max / divs # integration by Reimann sum norm = sum(wf .* wf) * r_max / divs # integration by Reimann sum