def plot_3d(): from numpy import array from stark import stark import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D from pylab import xlabel, ylabel, xlim, ylim, subplot, title, rc, linspace, plot, xticks, yticks, grid rc('text', usetex=True) npoints = 1000 fig = plt.figure() ax = fig.add_subplot(121, projection='3d') s = stark(.08,[1.,0.,.05],[0,1,.02]) title(r'(a)') r = linspace(0,s.T_xi*2,npoints) states = array([s.cart_state(tau) for tau in r]) ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2) ax.set_zlim(0,0.3) ax = fig.add_subplot(122, projection='3d') s = stark(.1793,[1.,0.,.05],[0,1,.02]) title(r'(b)') r = linspace(0,s.T_xi*2,npoints) states = array([s.cart_state(tau) for tau in r]) ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2)
def plot_quasi_periodic(): from numpy import array from stark import stark import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D from pylab import xlabel, ylabel, xlim, ylim, subplot, title, rc, linspace, plot, xticks, yticks, grid rc('text', usetex=True) npoints = 1000 x0,v0,eps = [1.0, 0, 6.123233995736766e-17],[-0.2833431329044651, 0.8928727330044925, 0.34957175101847515],0.0612722970737 s = stark(eps,x0,v0) T = 3.4589043219148699*5 r = linspace(0,T*3,npoints) states = array([s.cart_state(_) for _ in r]) fig = plt.figure() ax = fig.add_subplot(121, projection='3d') title(r'(a)') ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2) ax = fig.add_subplot(122, projection='3d') title(r'(b)') ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2)
def plot_periodic(): from mpmath import mpf from numpy import array from stark import stark import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D from pylab import xlabel, ylabel, xlim, ylim, subplot, title, rc, linspace, plot, xticks, yticks, grid rc('text', usetex=True) npoints = 2000 x0,v0,eps = [1.0, 0, 6.123233995736766e-17], [0.020495878242533995, 1.1180786341556543, 0.05717527804838409], 0.080969489037 s = stark(eps,x0,v0) T = 43.335885307166 r = linspace(0,T,npoints) states = array([s.cart_state(_) for _ in r]) fig = plt.figure() ax = fig.add_subplot(121, projection='3d') title(r'(a)') ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2) ax = fig.add_subplot(122, projection='3d') title(r'(b)') ax.plot([_.real for _ in states[:,0]],[_.real for _ in states[:,1]],[_.real for _ in states[:,2]],'k-',linewidth=2)
def plot_bound_vs_unbound(): from pylab import xlabel, ylabel, xlim, ylim, subplot, title, rc, linspace, plot, xticks, yticks, grid from stark import stark rc('text', usetex=True) npoints = 1000 s = stark(.07,[1.,1.,.3],[.1,.4,.6]) r = linspace(-s.T_xi*2,s.T_xi*2,npoints) subplot(241) title(r'(a)') plot(r,[s.xi(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\xi\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(242) title(r'(b)') plot(r,[s.eta(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\eta\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(243) title(r'(c)') plot(r,[s.phi(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\phi\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(244) title(r'(d)') plot(r,[s.t(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$t\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() s = stark(.1,[1.,1.,.3],[.1,.4,.6]) r = linspace(-s.T_xi*2,s.T_xi*2,npoints) subplot(245) title(r'(e)') plot(r,[s.xi(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\xi\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(246) title(r'(f)') plot(r,[s.eta(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\eta\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(247) title(r'(g)') plot(r,[s.phi(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$\phi\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid() subplot(248) title(r'(h)') plot(r,[s.t(tau).real for tau in r],'k-',linewidth=2) xlabel(r'$\tau$') ylabel(r'$t\left(\tau\right)$') xticks([0],['']) yticks([0],['']) xlim(float(r[0]),float(r[-1])) #ylim(-.1,4) grid()