yapanuwan
/
BergEC-jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

More verbosity

This commit is contained in:
Nuwan Yapa 2024-03-08 12:27:48 -05:00
parent cfd2eddad1
commit 917583904e
1 changed files with 11 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
ho_basis_3body.jl
View File

@ -17,25 +17,24 @@ Rb = sqrt(10)
c1 = 1/sqrt(2)
c2 = sqrt(2)
println("No of threads = ", Threads.nthreads())
Es, n1s, l1s, n2s, l2s = get_2p_basis(E_max)
println("Basis size = ", length(Es))
println("Constructing KE matrices")
T1 = sp_T_matrix(n1s, l1s; ω=ω, μ=c1^2 * μ1)
T2 = sp_T_matrix(n2s, l2s; ω=ω, μ=c2^2 * μ2)
@time "T1" T1 = sp_T_matrix(n1s, l1s; ω=ω, μ=c1^2 * μ1)
@time "T2" T2 = sp_T_matrix(n2s, l2s; ω=ω, μ=c2^2 * μ2)
println("Constructing PE matrices")
V_l(l, n1, n2) = Va * V_Gaussian(Ra, l, n1, n2; ω=ω) + Vb * V_Gaussian(Rb, l, n1, n2; ω=ω)
V1 = sp_V_matrix(V_l, n1s, l1s)
V_relative = sp_V_matrix(V_l, n1s, l1s) + sp_V_matrix(V_l, n2s, l2s)
@time "V1" V1 = sp_V_matrix(V_l, n1s, l1s)
@time "V_relative" V_relative = sp_V_matrix(V_l, n1s, l1s) + sp_V_matrix(V_l, n2s, l2s)
@time "Moshinsky brackets" Moshinsky_mat = Moshinsky_transform(Es, n1s, l1s, n2s, l2s, Λ)
@time "V2 via transform" V2 = Moshinsky_mat * V_relative
println("Performing Moshinsky transform")
@time "Calculating brackets" Moshinsky_mat = Moshinsky_transform(Es, n1s, l1s, n2s, l2s, Λ)
@time "Applying transform" V2 = Moshinsky_mat * V_relative
H = T1 + T2 + V1 + V2
println("Diagonalizing H matrix")
@time "Diagonalizing" evals, _ = eigs(H, nev=5, which=:SM)
println("Calculating spectrum")
@time "H" H = T1 + T2 + V1 + V2
@time "Eigenvalues" evals, _ = eigs(H, nev=5, which=:SM)
display(evals)