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

View File

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