sharex = ax1, sharey = ax1,
subplot = 224)
ax1b.get_yaxis().set_visible(False)
ax2.get_yaxis().set_visible(False)
ax3b.get_yaxis().set_visible(False)
ax4.get_yaxis().set_visible(False)
for ax in [ax1, ax1b, ax2, ax2b]:
ax.get_xaxis().set_visible(False)
# ********************** Before ************************
ax1.plot(r, U)
ax1.plot([-r0, r0], [-Xe_Z0_Ip, -Xe_Z0_Ip], '-m', label = '5p')
for ei in xrange(len(Xe_Z0_es_E)):
r0e = min(abs(cs/Xe_Z0_es_E[ei]), r[-1])
ax1.plot([-r0e, r0e], [Xe_Z0_es_E[ei], Xe_Z0_es_E[ei]],
linestyle = colors_and_symbols.symbol(ei+5),
color = colors_and_symbols.color(ei),
label = Xe_Z0_es_n[ei])
# Impacting electron
ar1Ke = fleches.arrow('K_e', [0.75*r[0], 0.0], [0.75*r[0], Ke]) # Vertical (Ke)
ar2Ke = fleches.arrow('K_e', [0.75*r[0], Ke], [0.75*r[0]+10, Ke]) # Horizontal (v)
ar1Ke.Plot(ax1, color = 'b', label = '$K_e$', horizontalalignment = 'right', bidirectional = True)
ar2Ke.Plot(ax1, color = 'b', label = '$v_e$', verticalalignment = 'top')
ax1.plot([0.75*r[0]], [Ke], 'ob', ms = 14)
# Bound electron
ax1.plot([0.0], [-Xe_Z0_Ip], 'og', ms = 14)
# ********************** During ************************
#try:
# if (potential_shape == "HermanSkillman"):
# #hs_U = hs.HS_Fitting_Function_Xe_Potential(rU, cs)
# #hs_E = -hs.HS_Fitting_Function_Xe_Potential(rU, cs)
# #ax1.plot(rU, hs_U, ':' + colors[cs%len(colors)])
# #ax2.plot(rE, hs_E, ':' + colors[cs%len(colors)])
#
# new_r, new_E, new_IntE, pts_r, pts_E = cubic_spline(p1_cs, rU, U, rE, E)
# ax1.plot(new_r, new_IntE, colors[cs%len(colors)])
# ax2.plot(new_r, new_E, colors[cs%len(colors)])
# ax2.plot(pts_r, pts_E, colors[cs%len(colors)] + 'x', ms=10, markeredgewidth=3)
#except:
# pass
ax1.plot(rU, U, colsym.symbol(fi)+colsym.color(cs), label = str(p1_cs) + "+ " + potential_shape, lw=line_width)
ax1.plot(rU, Coulomb_U, ':'+colsym.color(cs), lw=line_width)
ax2.plot(rE, E, colsym.symbol(fi)+colsym.color(cs), label = str(p1_cs) + "+ " + potential_shape, lw=line_width)
ax2.plot(rE, Coulomb_E, ':'+colsym.color(cs), lw=line_width)
# Keep track of min and max
if (U.min() < Umin):
Umin = U.min()
if (U.max() > Umax):
Umax = U.max()
if (E.min() < Emin):
Emin = E.min()
if (E.max() > Emax):
Emax = E.max()
fi += 1