From 93fa3c4dc775658fdd3dddc8342b7c57adf2e8c0 Mon Sep 17 00:00:00 2001
From: Nuwan Yapa <yapanuwan@gmail.com>
Date: Mon, 4 Mar 2024 14:17:38 -0500
Subject: [PATCH] Mass parameter (mu) incorporated

---
 XZ_technique.jl       | 2 +-
 ho_basis.jl           | 4 ++--
 ho_basis_benchmark.jl | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/XZ_technique.jl b/XZ_technique.jl
index aee4fac..d3e6634 100644
--- a/XZ_technique.jl
+++ b/XZ_technique.jl
@@ -14,7 +14,7 @@ n_max = 20
 ns = collect(0:n_max)
 ls = fill(l, n_max + 1)
 
-T = 1/μ * sp_T_matrix(ns, ls; ω=ω)
+T = sp_T_matrix(ns, ls; ω=ω, μ=μ)
 V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns); ω=ω) + V2 .* V_Gaussian.(R2, l, ns, transpose(ns); ω=ω)
 
 n_EC = 8
diff --git a/ho_basis.jl b/ho_basis.jl
index 9fc71e1..b105fa7 100644
--- a/ho_basis.jl
+++ b/ho_basis.jl
@@ -55,7 +55,7 @@ end
 
 get_V_matrix(V_l, ls, ns) = throw("unimplemented")
 
-function sp_T_matrix(ns, ls; ω=1.0)
+function sp_T_matrix(ns, ls; ω=1.0, μ=1.0)
     mat = zeros(length(ns), length(ns))
     for idx in CartesianIndices(mat)
         (i, j) = Tuple(idx)
@@ -70,7 +70,7 @@ function sp_T_matrix(ns, ls; ω=1.0)
             end
         end
     end
-    return ω .* mat
+    return (ω / μ) .* mat
 end
 
 get_H_matrix(V_l, ns, ls) = get_T_matrix(ns, ls) + get_V_matrix(V_l, ns, ls)
diff --git a/ho_basis_benchmark.jl b/ho_basis_benchmark.jl
index e43583c..c3d5c5e 100644
--- a/ho_basis_benchmark.jl
+++ b/ho_basis_benchmark.jl
@@ -14,7 +14,7 @@ n_max = 15
 ns = collect(0:n_max)
 ls = fill(l, n_max + 1)
 
-T = 1/μ * sp_T_matrix(ns, ls; ω=ω)
+T = sp_T_matrix(ns, ls; ω=ω, μ=μ)
 V = V1 .* V_Gaussian.(R1, l, ns, transpose(ns); ω=ω) + V2 .* V_Gaussian.(R2, l, ns, transpose(ns); ω=ω)
 
 cs = range(1.35, 0.9, 5)