r"$x\left(1+x^2\right)^{-\frac{1}{2}}$",
r"$\frac{2}{\pi}\mathrm{arctan}\left( \frac{\pi}{2}x \right)$",
r"$x\left(1+|x|\right)^{-1}$"
]
for i in range(1, 7):
s = "ax.plot(x,y%s(x),color=colors[i-1])" % (str(i))
eval(s)
ax.legend(leg_list, loc="best", ncol=2,
fancybox=True) # title="Legend", fontsize=12
# ax.grid(True, which='both')
ax.set_aspect('equal')
ax.set_xlim([-3.1, 3.1])
ax.set_ylim([-1.1, 1.1])
ax.annotate('1', xy=(0.08, 1 - 0.02))
ax.annotate('0', xy=(0.08, -0.2))
ax.annotate('-1', xy=(0.08, -1 - 0.03))
for i in [-3, -2, -1, 1, 2, 3]:
ax.annotate('%s' % str(i), xy=(i - 0.03, -0.2))
maybe = raw_input(
"\nUpdate figure directly in master thesis?\nEnter 'YES' (anything else = ONLY show to screen) "
)
if maybe == "YES": # Only save to disc if need to be updated
filenameWithPath = "/Users/haakonvt/Dropbox/uio/master/latex-master/Illustrations/six_sigmoids.pdf"
plt.savefig(filenameWithPath, bbox_inches='tight') #, pad_inches=0.2)
print 'Saved over previous file in location:\n "%s"' % filenameWithPath
else:
print 'Figure was only shown on screen.'
tunneling_wave(t, x1, x2, omega, amplitude1, amplitude2) for t in from_x2
]
colors = initialize_graphics()
fig = plt.figure(1)
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
#fig, ax = plt.subplots()
fig.set_size_inches(cm2inch([10, 5]))
ax.plot(full_range, potential, lw=2, color=colors[2])
ax.plot(until_x1, wave_until_x1, lw=2, color=colors[0])
ax.plot(from_x2, wave_from_x2, lw=2, color=colors[0])
ax.annotate('$U(x)$', xy=(x2, V0), xytext=(x2 + 0.3, V0 - 0.2))
ax.set_frame_on(False)
#ax.axes.get_yaxis().set_visible(False)
ax.axes.get_xaxis().set_ticks([x1, x2, 5 * np.pi + 0.5])
ax.axes.get_xaxis().set_ticklabels(['$x_1$', '$x_2$', '$x$'])
ax.axis["xzero"].set_axisline_style("-|>")
ax.axis["xzero"].set_visible(True)
ax.axis["yzero"].set_visible(False)
for direction in ["left", "right", "bottom", "top"]:
ax.axis[direction].set_visible(False)
ax.set_xlim([-0.5, 5 * np.pi + 0.5])
plt.savefig('./quantum_tunneling.pdf')
wave_from_x2 = [tunneling_wave(t, x1, x2, omega, amplitude1, amplitude2) for t in from_x2]
colors=initialize_graphics()
fig = plt.figure(1)
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
#fig, ax = plt.subplots()
fig.set_size_inches(cm2inch([10,5]))
ax.plot(full_range, potential, lw=2, color=colors[2])
ax.plot(until_x1, wave_until_x1, lw=2, color=colors[0])
ax.plot(from_x2, wave_from_x2, lw=2, color=colors[0])
ax.annotate('$U(x)$', xy=(x2, V0), xytext=(x2+0.3, V0-0.2))
ax.set_frame_on(False)
#ax.axes.get_yaxis().set_visible(False)
ax.axes.get_xaxis().set_ticks([x1, x2, 5*np.pi+0.5])
ax.axes.get_xaxis().set_ticklabels(['$x_1$','$x_2$', '$x$'])
ax.axis["xzero"].set_axisline_style("-|>")
ax.axis["xzero"].set_visible(True)
ax.axis["yzero"].set_visible(False)
for direction in ["left", "right", "bottom", "top"]:
ax.axis[direction].set_visible(False)
ax.set_xlim([-0.5, 5*np.pi+0.5])
plt.savefig('./quantum_tunneling.pdf')
