plt.tight_layout()
#plt.savefig('fig7a-mu-theta.pdf', format='pdf')
plt.savefig('figs/fig7a-mu-theta.png', format='png')
#%% Particle path
flow.compute_velocity(flow.n_h-1)
k = 10
c = k + np.trapz(flow.c_u[:,0],x=flow.q)/np.pi
flow.prepare_path_computation(c)
ix = -1
n_per = 4
t_max = 2*np.pi/c*n_per
t = np.linspace(0,t_max,200*n_per)
idx = np.where(flow.p==-0.5)[0]
traj1 = flow.compute_path([flow.q[ix],flow.y[ix,flow.n_p-1]], t)
traj2 = flow.compute_path([flow.q[ix],flow.y[ix,450]], t)
traj3 = flow.compute_path([flow.q[ix],flow.y[ix,300]], t)
traj4 = flow.compute_path([flow.q[ix],flow.y[ix,150]], t)
traj5 = flow.compute_path([flow.q[ix],flow.y[ix,1]], t)
print(c+k)
#%
fig = plt.figure(figsize=(6*1.,3.6*1.))
ax = fig.gca()
plt.plot(traj1[0,0],traj1[0,1],'ro', linewidth=1.2, markersize=MARKER_SIZE)
plt.plot(traj1[-1,0],traj1[-1,1],'rs', linewidth=1.2, markersize=MARKER_SIZE)
plt.plot(traj1[:,0],traj1[:,1],'r-', linewidth=2)
plt.plot(traj2[0,0],traj2[0,1],'go', linewidth=1.2, markersize=MARKER_SIZE)
plt.plot(traj2[-1,0],traj2[-1,1],'gs', linewidth=1.2, markersize=MARKER_SIZE)
plt.plot(traj2[:,0],traj2[:,1],'g-', linewidth=1.2)
