def write_image(k):
q,p,mu,q1 = jpf.state_to_weinstein_darboux( y_data[k] )
f = gf.display_velocity(q,p,mu)
time_s = str(time_data[k])
plt.suptitle('t = '+ time_s[0:4] , fontsize=16 , x = 0.75 , y = 0.25 )
fname = './movie_frames/frame_%03i.png' % k
f.savefig( fname )
plt.close(f)
sys.stdout.write(' '+str(k))
sys.stdout.flush()
def write_image(k):
q, p, mu, q1 = jpf.state_to_weinstein_darboux(y_data[k])
f = gf.display_velocity(q, p, mu)
time_s = str(time_data[k])
plt.suptitle('t = ' + time_s[0:4], fontsize=16, x=0.75, y=0.25)
fname = './movie_frames/frame_%03i.png' % k
f.savefig(fname)
plt.close(f)
sys.stdout.write(' ' + str(k))
sys.stdout.flush()
#zero_jet_tackles_another
#dynamic_one_jet
#state = np.load('./movies/zero_jet_tackles_another.npy')
#time = np.load('./movies/zero_jet_tackles_another_time.npy')
#step = 100
spin = np.array([ [0. , -1.] , [1. , 0.]])
stretch = np.array([ [0. , 1.] , [1. , 0.]])
shear = np.array([ [0. , 0.] , [1. , 0.]])
q = np.zeros([N,DIM])
p = np.zeros([N,DIM])
mu = np.zeros([N,DIM,DIM])
mu[0] = spin
#p[0,0] = 1.
#q,p,mu = jpf.state_to_weinstein_darboux(state[step])
#print 'time = ' + str(time[step])
gf.display_velocity(q,p,mu)
plt.axis('equal')
plt.show()
gf.display_vorticity(q,p,mu,quiver="on")
plt.axis('equal')
plt.show()
#gf.display_streamlines(q,p,mu)
#plt.axis('equal')
#plt.show()
#zero_jet_tackles_another
#dynamic_one_jet
#state = np.load('./movies/zero_jet_tackles_another.npy')
#time = np.load('./movies/zero_jet_tackles_another_time.npy')
#step = 100
spin = np.array([[0., -1.], [1., 0.]])
stretch = np.array([[0., 1.], [1., 0.]])
shear = np.array([[0., 0.], [1., 0.]])
q = np.zeros([N, DIM])
p = np.zeros([N, DIM])
mu = np.zeros([N, DIM, DIM])
mu[0] = spin
#p[0,0] = 1.
#q,p,mu = jpf.state_to_weinstein_darboux(state[step])
#print 'time = ' + str(time[step])
gf.display_velocity(q, p, mu)
plt.axis('equal')
plt.show()
gf.display_vorticity(q, p, mu, quiver="on")
plt.axis('equal')
plt.show()
#gf.display_streamlines(q,p,mu)
#plt.axis('equal')
#plt.show()
