def get_oddw_smooth():
n = 2
l = 10
grid_length = 5
num_grid_points = 1001
a = 5
oddw_smooth = GeneralOrbitalSystem(
n,
ODQD(
l,
grid_length,
num_grid_points,
potential=ODQD.DWPotentialSmooth(a=a),
),
)
return oddw_smooth
n = 2
l = 20
grid_length = 5
num_grid_points = 1001
omega = 1
odho = ODQD(n, l, grid_length, num_grid_points)
odho.setup_system(potential=ODQD.HOPotential(omega))
save_data(odho, "odho")
length_of_dw = 5
oddw = ODQD(n, l, 6, num_grid_points)
oddw.setup_system(potential=ODQD.DWPotential(omega, length_of_dw))
save_data(oddw, "oddw")
weight = 1
center = 0
deviation = 2.5
odgauss = ODQD(n, l, 20, num_grid_points)
odgauss.setup_system(
potential=ODQD.GaussianPotential(weight, center, deviation, np=np))
save_data(odgauss, "odgauss")
oddw_smooth = ODQD(n, l, grid_length, num_grid_points)
oddw_smooth.setup_system(potential=ODQD.DWPotentialSmooth(a=5))
save_data(oddw_smooth, "oddw_smooth")
import numpy as np
import matplotlib.pyplot as plt
from quantum_systems import ODQD
n = 2
l = 10
odqd = ODQD(n, l, 4, 201)
odqd.setup_system(potential=ODQD.DWPotentialSmooth(a=4.5))
plt.plot(odqd.grid, odqd.potential(odqd.grid))
for i in range(l // 2):
plt.plot(odqd.grid, odqd.eigen_energies[i] + np.abs(odqd.spf[i * 2])**2)
plt.show()