def sine_timeplot_fig(mu_vals = [-1e-3, -0.1, -0.3, -0.40]):
# set up the model parameters for this figure
# mu = -0.2
alpha = 0.23333
k = 1
n_phis = np.linspace(0, 2*math.pi, 4*20+1)
dx = 1e-4
dy = 0.
mag = math.sqrt(dx**2 + dy**2)
phasescale = 4 / (2 * math.pi) # convert from (0,2 \pi) to (0,4)
# create a new figure
fig = plt.figure(figsize=(6,6))
width = 1./len(mu_vals)
padding = 0.2*width
for i in range(len(mu_vals)):
mu = mu_vals[i]
axes = plt.axes((2*padding, 1-(i+1) * width+padding,
1 - width - 2*padding, width - 1.5*padding))
# draw the trajectory components vs. time
T, y0, error = iris.sine_limit_cycle(mu, alpha, k)
ts = np.linspace(0, 3*T, 1000);
vals = integrate.odeint(iris.sine_system,
[0, -y0],
ts,
args=(mu, alpha, k))
axes.plot(ts, vals[:,1], '-', color='0.8', lw=2)
axes.plot(ts, vals[:,0], 'k-', lw=2)
axes.set_xlim(0, 3*T)
# make the y-axis symmetric around zero
#ymaxabs = np.max(np.abs(axes.get_ylim()))
ymaxabs = 0.6*math.pi
axes.set_ylim(-ymaxabs, ymaxabs)
# draw the phase plot for reference
axes = plt.axes((1-width+padding, 1-(i+1) * width + padding,
width - 1.5 * padding, width - 1.5 * padding))
iris.draw_fancy_sine_system(axes, mu, alpha, k, scale=3.)
# center the plot and clean up the scale bars
border = 0.2
axes.set_xlim(-math.pi/2-border, math.pi/2+border)
axes.set_ylim(-math.pi/2-border, math.pi/2+border)
axes.set_xticks([])
axes.set_yticks([])
axes.set_frame_on(False)
return fig
def sine_prc_fig(mu_vals = [-1e-3, -0.1, -0.3, -0.40]):
# set up the model parameters for this figure
# mu = -0.2
alpha = 0.23333
k = 1
n_phis = np.linspace(0, 2*math.pi, 4*20+1)
dx = 1e-4
dy = 0.
mag = math.sqrt(dx**2 + dy**2)
phasescale = 4 / (2 * math.pi) # convert from (0,2 \pi) to (0,4)
# create a new figure
fig = plt.figure(figsize=(6,6))
width = 1./len(mu_vals)
padding = 0.2*width
for i in range(len(mu_vals)):
mu = mu_vals[i]
axes = plt.axes((2*padding, 1-(i+1) * width+padding,
1 - width - 2*padding, width - 1.5*padding))
# draw the orthogonal prc found numerically
T, y0, error = iris.sine_limit_cycle(mu, alpha, k)
n_prc_o = np.array([
iris.sine_phase_reset(phi, dx=-dy, dy=dx, mu=mu, alpha=alpha, k=k,
y0=y0, T=T,
steps_per_cycle=100000)
for phi in n_phis
])
orthogonal_color = '0.8'
axes.plot(n_phis, n_prc_o/mag*phasescale, '--', markersize=3,
color=orthogonal_color)
axes.plot(n_phis, n_prc_o/mag*phasescale, 'bo', markersize=3,
color=orthogonal_color, markeredgecolor=orthogonal_color)
axes.set_xlim(0, 2*math.pi)
# draw the prc found numerically
T, y0, error = iris.sine_limit_cycle(mu, alpha, k)
n_prc = np.array([
iris.sine_phase_reset(phi, dx=dx, dy=dy, mu=mu, alpha=alpha, k=k,
y0=y0, T=T,
steps_per_cycle=100000)
for phi in n_phis
])
axes.plot(n_phis, n_prc/mag*phasescale, '--k', markersize=3)
axes.plot(n_phis, n_prc/mag*phasescale, 'bo', markersize=3)
axes.set_xlim(0, 2*math.pi)
plt.xticks(np.arange(5.)*math.pi/2,
#['$0$', '$\\pi/2$', '$\\pi$',
# '$3\\pi/2$', '$2\\pi$'])
# phase rescaled from 0 to 4 to match analysis
['$0$', '$1$', '$2$', '$3$', '$4$'])
# make the y-axis symmetric around zero
ymaxabs = np.max(np.abs(axes.get_ylim()))
axes.set_ylim(-ymaxabs, ymaxabs)
# draw the phase plot for reference
axes = plt.axes((1-width+padding, 1-(i+1) * width + padding,
width - 1.5 * padding, width - 1.5 * padding))
iris.draw_fancy_sine_system(axes, mu, alpha, k, scale=3.)
# center the plot and clean up the scale bars
border = 0.2
axes.set_xlim(-math.pi/2-border, math.pi/2+border)
axes.set_ylim(-math.pi/2-border, math.pi/2+border)
axes.set_xticks([])
axes.set_yticks([])
axes.set_frame_on(False)
return fig
