ax1.text(0.,
1.05,
'y',
size=20,
transform=BlendedGenericTransform(ax1.transData, ax1.transAxes))
ax1.text(1.05,
-0.15,
'x',
size=20,
transform=BlendedGenericTransform(ax1.transAxes, ax1.transData))
vec_phi_xy = ax1.quiver(0,
0,
0,
0,
width=10,
scale=1,
units='x',
label=r'$\phi_n$',
color='b')
vec_Snn_xy = ax1.quiver(0,
0,
0,
0,
width=6,
scale=1,
units='x',
label=r'$\sigma_{nn}$',
color='c')
vec_Snt_xy = ax1.quiver(0,
0,
g = orig + acceleration
print h, x, g
soa = np.array([h, x, g]) # vectors
print soa
X, Y, U, V = zip(*soa) # convert to turples of U and V components
fig = plt.figure(1)
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
colors = ('r', 'g', 'b')
qv = ax.quiver(X,
Y,
U,
V,
color=colors,
angles='xy',
scale_units='xy',
scale=1)
labels = ('heading: {} deg'.format(hdeg),
'Orientation, drift: {} deg'.format(drift),
'{} g at {} deg'.format(aforce, adeg))
pos = ('N', 'E', 'S')
for x, y, l, c, p in zip(U, V, labels, colors, pos):
plt.quiverkey(qv, x, y, 0, l, color=c, coordinates='data', labelpos=p)
ax.set_xlim([-2, 2])
ax.set_ylim([-2, 2])
# show cartisian axis
ic = [[0.2, 0.2], [0.2, -0.2], [0.2, -0.5], [0.2, 0.7]]
color = ['r', 'b', 'g', 'y']
for direction in ["xzero", "yzero"]:
ax.axis[direction].set_axisline_style("-|>")
ax.axis[direction].set_visible(True)
for k in range(len(ic)):
Y = []
T = []
S = []
r.set_initial_value(ic[k], t0).set_f_params()
while r.successful() and r.t + dt < tEnd:
r.integrate(r.t + dt)
Y.append(r.y)
S = np.array(np.real(Y))
ax.plot(S[:, 0], S[:, 1], color=color[k], lw=1.25)
X, Y = np.meshgrid(np.linspace(-5, 5, 40), np.linspace(-5, 5, 20))
U = 1
V = X / Y
N = np.sqrt(U**2 + V**2)
U2, V2 = U / N, V / N
ax.quiver(X, Y, U2, V2)
plt.xlim([-5, 5])
plt.ylim([-5, 5])
plt.ylabel(r"$y$")
plt.show()
h = orig + heading # normalized heading
x = orig + orientation # normalized orientation
g = orig + acceleration
print h,x,g
soa = np.array([h,x,g]) # vectors
print soa
X,Y,U,V = zip(*soa) # convert to turples of U and V components
fig = plt.figure(1)
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
colors = ('r','g','b')
qv = ax.quiver(X,Y,U,V,color=colors,angles='xy',scale_units='xy',scale=1)
labels = ('heading: {} deg'.format(hdeg), 'Orientation, drift: {} deg'.format(drift), '{} g at {} deg'.format(aforce,adeg))
pos = ('N','E','S')
for x,y,l,c,p in zip(U,V,labels,colors,pos):
plt.quiverkey(qv,x,y,0,l,color=c,coordinates='data',labelpos=p)
ax.set_xlim([-2,2])
ax.set_ylim([-2,2])
# show cartisian axis
# for direction in ["xzero", "yzero"]:
# ax.axis[direction].set_visible(True)
# turn off side axis
for direction in ["left", "right", "bottom", "top"]:
from mpl_toolkits.axes_grid.axislines import SubplotZero
from matplotlib import colors
roll = 15
pitch = 5
orig = [0, 0]
X, Y = (0, 0) # origin
U = roll
V = pitch
fig = plt.figure(1)
ax = SubplotZero(fig, 111)
fig.add_subplot(ax)
qv = ax.quiver(X, Y, U, V, color="y", angles="xy", scale_units="xy", scale=1)
ax.set_xlim([-45, 45])
ax.set_ylim([-45, 45])
# show cartisian axis
for direction in ["xzero", "yzero"]:
ax.axis[direction].set_visible(True)
# turn off side axis
# for direction in ["left", "right", "bottom", "top"]:
# ax.axis[direction].set_visible(False)
plt.draw()
plt.show()
