Removed berggren.jl
This commit is contained in:
parent
37af76e3a8
commit
85de22e94b
18
berggren.jl
18
berggren.jl
|
|
@ -1,18 +0,0 @@
|
||||||
using SparseArrays
|
|
||||||
include("ho_basis.jl")
|
|
||||||
|
|
||||||
"Basis transformation from HO to momentum space"
|
|
||||||
function get_W_matrix(basis_p, E_max, Λ, μ1ω1, μ2ω2=μ1ω1; weights=true)
|
|
||||||
Es, n1s, l1s, n2s, l2s = get_2p_basis(E_max, Λ)
|
|
||||||
W = zeros(ComplexF64, length(basis_p), length(Es))
|
|
||||||
Threads.@threads for idx in CartesianIndices(W)
|
|
||||||
(i1, i2) = Tuple(idx)
|
|
||||||
((j1, j2), (k1, w1), (k2, w2)) = basis_p[i1]
|
|
||||||
if j1 == l1s[i2] && j2 == l2s[i2]
|
|
||||||
elem1 = 1/sqrt(sqrt(μ1ω1)) * (-1)^n1s[i2] * ho_basis(j1, n1s[i2], 1/sqrt(μ1ω1) * k1)
|
|
||||||
elem2 = 1/sqrt(sqrt(μ2ω2)) * (-1)^n2s[i2] * ho_basis(j2, n2s[i2], 1/sqrt(μ2ω2) * k2)
|
|
||||||
W[idx] = elem1 * elem2 * (weights ? w1 * w2 : 1)
|
|
||||||
end
|
|
||||||
end
|
|
||||||
return sparse(W)
|
|
||||||
end
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
using LinearAlgebra, SparseArrays, Arpack
|
using LinearAlgebra, SparseArrays, Arpack
|
||||||
include("helper.jl")
|
include("helper.jl")
|
||||||
include("p_space.jl")
|
include("p_space.jl")
|
||||||
include("berggren.jl")
|
include("ho_basis.jl")
|
||||||
|
|
||||||
println("No of threads = ", Threads.nthreads())
|
println("No of threads = ", Threads.nthreads())
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,6 @@
|
||||||
using LinearAlgebra, SparseArrays, Arpack
|
using LinearAlgebra, SparseArrays, Arpack
|
||||||
include("helper.jl")
|
include("helper.jl")
|
||||||
include("p_space.jl")
|
include("p_space.jl")
|
||||||
include("berggren.jl")
|
|
||||||
|
|
||||||
println("No of threads = ", Threads.nthreads())
|
println("No of threads = ", Threads.nthreads())
|
||||||
atol = 10^-5
|
atol = 10^-5
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,6 @@ using Arpack, SparseArrays, LRUCache
|
||||||
using DelimitedFiles, Plots
|
using DelimitedFiles, Plots
|
||||||
include("../ho_basis.jl")
|
include("../ho_basis.jl")
|
||||||
include("../p_space.jl")
|
include("../p_space.jl")
|
||||||
include("../berggren.jl")
|
|
||||||
|
|
||||||
println("No of threads = ", Threads.nthreads())
|
println("No of threads = ", Threads.nthreads())
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,6 @@ using Arpack, SparseArrays, LRUCache
|
||||||
using DelimitedFiles, Plots
|
using DelimitedFiles, Plots
|
||||||
include("../ho_basis.jl")
|
include("../ho_basis.jl")
|
||||||
include("../p_space.jl")
|
include("../p_space.jl")
|
||||||
include("../berggren.jl")
|
|
||||||
|
|
||||||
println("No of threads = ", Threads.nthreads())
|
println("No of threads = ", Threads.nthreads())
|
||||||
|
|
||||||
|
|
|
||||||
18
ho_basis.jl
18
ho_basis.jl
|
|
@ -204,4 +204,20 @@ function get_src_V12_matrix(V_of_r, E_max, Λ, μω_global; atol=10^-6, maxevals
|
||||||
V12 = U' * V_relative * U
|
V12 = U' * V_relative * U
|
||||||
|
|
||||||
return V12
|
return V12
|
||||||
end
|
end
|
||||||
|
|
||||||
|
"Basis transformation from HO to momentum space"
|
||||||
|
function get_W_matrix(basis_p, E_max, Λ, μ1ω1, μ2ω2=μ1ω1; weights=true)
|
||||||
|
Es, n1s, l1s, n2s, l2s = get_2p_basis(E_max, Λ)
|
||||||
|
W = zeros(ComplexF64, length(basis_p), length(Es))
|
||||||
|
Threads.@threads for idx in CartesianIndices(W)
|
||||||
|
(i1, i2) = Tuple(idx)
|
||||||
|
((j1, j2), (k1, w1), (k2, w2)) = basis_p[i1]
|
||||||
|
if j1 == l1s[i2] && j2 == l2s[i2]
|
||||||
|
elem1 = 1/sqrt(sqrt(μ1ω1)) * (-1)^n1s[i2] * ho_basis(j1, n1s[i2], 1/sqrt(μ1ω1) * k1)
|
||||||
|
elem2 = 1/sqrt(sqrt(μ2ω2)) * (-1)^n2s[i2] * ho_basis(j2, n2s[i2], 1/sqrt(μ2ω2) * k2)
|
||||||
|
W[idx] = elem1 * elem2 * (weights ? w1 * w2 : 1)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return sparse(W)
|
||||||
|
end
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue