INNER CODE UNIT · Python
l_max
ssebastianmag/hydrogen-wavefunctions · interactive_plot/interactive_plot_util.py:193
l_max = n - 1
if l > l_max:
l = l_max
m_max = l
m = max(min(m, m_max), -m_max)
# Compute Wave Function
extent_a_mu = 2.5 * n ** 2
wf = WaveFunction(n=n, l=l, m=m, extent_a_mu=extent_a_mu, grid_points=resolution)
try:
P, extent = compute_P_and_extent(wf)
except MemoryError as e:
# Specify error
raise MemoryError(
f'Encountered MemoryError. Likely caused by too much resolution. Try reducing resolution. Error: {e}')
x_min, x_max, z_min, z_max = extent
# Plot