def analytic_model_fig(ax, var_md, time=1):
lamb = eq.d1_lambda_evol(time)
h_a = eq.d1_height(lamb, var_md["x_range"])
v_a = eq.d1_velocity(lamb, var_md["x_range"])
ax.plot(
var_md["x_range"],
eq.d1_initial(var_md["x_range"]),
"k",
var_md["x_range"],
h_a,
"b",
var_md["x_range"],
var_md["h"],
"r",
)
ax.plot(
var_md["x_range"],
0 * var_md["x_range"],
"k-",
var_md["x_range"],
v_a,
"b--",
var_md["x_range"],
var_md["vx"],
"r--",
)
ps.ticks_changes(ax)
def analytic_model_fig(ax, var_md, time=1):
lamb = eq.d1_lambda_evol(time)
h_a = eq.d1_height(lamb, var_md["x_range"])
v_a = eq.d1_velocity(lamb, var_md["x_range"])
ax.plot(var_md["x_range"], eq.d1_initial(var_md["x_range"]), 'k',
var_md["x_range"], h_a, 'b', var_md["x_range"], var_md["h"], "r")
ax.plot(var_md["x_range"], 0 * var_md["x_range"], "k-", var_md["x_range"],
v_a, 'b--', var_md["x_range"], var_md["vx"], "r--")
ps.ticks_changes(ax)
def analytic_fig(ax, x_lim, time_l=[0,1,2,3], nx=320):
x_range = np.linspace(-x_lim, x_lim, nx)
oznacz = ['k', 'b', 'c', 'y', 'g', 'm', 'r']
for it, time in enumerate(time_l):
lamb = eq.d1_lambda_evol(time)
h = eq.d1_height(lamb, x_range)
v = eq.d1_velocity(lamb, x_range)
ax.plot(x_range, h, oznacz[it])
ax.plot(x_range, v, oznacz[it]+ "--")
ps.ticks_changes(ax)
def analytic_fig(ax, x_lim, time_l=[0, 1, 2, 3], nx=320):
x_range = np.linspace(-x_lim, x_lim, nx)
oznacz = ['k', 'b', 'c', 'y', 'g', 'm', 'r']
for it, time in enumerate(time_l):
lamb = eq.d1_lambda_evol(time)
h = eq.d1_height(lamb, x_range)
v = eq.d1_velocity(lamb, x_range)
ax.plot(x_range, h, oznacz[it])
ax.plot(x_range, v, oznacz[it] + "--")
ps.ticks_changes(ax)
def plotting_3D(X, Y, Z):
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, Z, rstride=10, cstride=10, alpha=0.2)
cset = ax.contourf(X, Y, Z, zdir='z', offset=-0.1, cmap=cm.Blues)
fig.colorbar(cset)
ps.ticks_changes(ax)
ax.set_xlabel('x')
ax.set_ylabel('y')
ax.set_zlabel('h')
ax.set_zlim(-0.1, 0.1)
plt.savefig("2d_fct_iga_view.pdf")
def analytic_model_fig(ax, var_md, time=1):
x_range = var_md["x_range"]
y_range = 0 * var_md["x_range"]
ind_cs = int(var_md["h"].shape[1]/2)
lamb = eq.d2_lambda_evol(time)
h_a = eq.d2_height(lamb, x_range, y_range)
v_a = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, eq.d2_initial(x_range, y_range), 'k', x_range, h_a, 'b',
x_range, var_md["h"][:,ind_cs], "r")
ax.plot(x_range, 0*x_range, "k-", x_range, v_a, 'b--',
x_range, var_md["vx"][:,ind_cs], "r--")
ps.ticks_changes(ax)
def analytic_model_fig(ax, x_range, y_range, h_m, v_m, time=1):
lamb = eq.d2_lambda_evol(time)
h_a = eq.d2_height(lamb, x_range, y_range)
v_a = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, eq.d2_initial(x_range, y_range), 'k', x_range, h_a, 'b',
x_range, h_m, "r")
ax.plot(x_range, 0*x_range, "k-", x_range, v_a, 'b--',
x_range, v_m, "r--")
#ax.set_ylim(-2,2)
ps.ticks_changes(ax)
def analytic_model_fig(ax, var_md, time=1):
x_range = var_md["x_range"]
y_range = 0 * var_md["x_range"]
ind_cs = int(var_md["h"].shape[1] / 2)
lamb = eq.d2_lambda_evol(time)
h_a = eq.d2_height(lamb, x_range, y_range)
v_a = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, eq.d2_initial(x_range, y_range), 'k', x_range, h_a, 'b',
x_range, var_md["h"][:, ind_cs], "r")
ax.plot(x_range, 0 * x_range, "k-", x_range, v_a, 'b--', x_range,
var_md["vx"][:, ind_cs], "r--")
ps.ticks_changes(ax)
def plotting_3D(X, Y, Z):
fig = plt.figure()
ax = fig.gca(projection="3d")
ax.plot_surface(X, Y, Z, rstride=10, cstride=10, alpha=0.2)
cset = ax.contourf(X, Y, Z, zdir="z", offset=-0.1, cmap=cm.Blues)
fig.colorbar(cset)
ps.ticks_changes(ax)
ax.set_xlabel("x")
ax.set_ylabel("y")
ax.set_zlabel("h")
ax.set_zlim(-0.1, 0.1)
plt.savefig("2d_fct_iga_view.pdf")
def analytic_model_fig(ax, x_range, y_range, h_m, v_m, time=1):
lamb = eq.d2_lambda_evol(time)
h_a = eq.d2_height(lamb, x_range, y_range)
v_a = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, eq.d2_initial(x_range, y_range), 'k', x_range, h_a, 'b',
x_range, h_m, "r")
ax.plot(x_range, 0 * x_range, "k-", x_range, v_a, 'b--', x_range, v_m,
"r--")
#ax.set_ylim(-2,2)
ps.ticks_changes(ax)
def analytic_model_fig(ax, x_range, h_m, v_m, t_m, it):
lamb = eq.d1_lambda_evol(t_m[it,0])
h_a = eq.d1_height(lamb, x_range)
v_a = eq.d1_velocity(lamb, x_range)
ax.plot(x_range, eq.d1_initial(x_range), 'k', x_range, h_a, 'b',
x_range, h_m[it], "r")
ax.plot(x_range, 0*x_range, "k-", x_range, v_a, 'b--',
x_range, v_m[it], "r--")
ax.set_ylim(-2,2)
ps.ticks_changes(ax)
def analytic_model_fig(ax, x_range, h_m, v_m, t_m, it):
lamb = eq.d1_lambda_evol(t_m[it,0])
h_a = eq.d1_height(lamb, x_range)
v_a = eq.d1_velocity(lamb, x_range)
ax.plot(x_range, eq.d1_initial(x_range), 'k', x_range, h_a, 'b',
x_range, h_m[it], "r")
ax.plot(x_range, 0*x_range, "k--", x_range, v_a, 'b--',
x_range, v_m[it], "r--")
ax.set_ylim(-2,2)
ps.ticks_changes(ax)
def analytic_fig(ax, time_l = [0,1,2,3], x_range = np.linspace(-8,8,320),
y_range = np.zeros(320)):
oznacz = ['k', 'b', 'c', 'y', 'g', 'm', 'r']
y0 = eq.d2_initial(x_range, y_range)
for it, time in enumerate(time_l):
lamb = eq.d2_lambda_evol(time)
h = eq.d2_height(lamb, x_range, y_range)
v = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, h, oznacz[it])
ax.plot(x_range, v, oznacz[it]+ "--")
ps.ticks_changes(ax)
def analytic_fig(ax,
time_l=[0, 1, 2, 3],
x_range=np.linspace(-8, 8, 320),
y_range=np.zeros(320)):
oznacz = ['k', 'b', 'c', 'y', 'g', 'm', 'r']
y0 = eq.d2_initial(x_range, y_range)
for it, time in enumerate(time_l):
lamb = eq.d2_lambda_evol(time)
h = eq.d2_height(lamb, x_range, y_range)
v = eq.d2_velocity(lamb, x_range, y_range)
ax.plot(x_range, h, oznacz[it])
ax.plot(x_range, v, oznacz[it] + "--")
ps.ticks_changes(ax)
def plotting_2D(X, Y, Z):
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.plot_surface(X, Y, Z, rstride=10, cstride=10, alpha=0.2)
cset = ax.contourf(X, Y, Z, zdir='z', offset=-0.1, cmap=cm.Blues)
fig.colorbar(cset) #, fraction=0.05)
ps.ticks_changes(ax)
ax.set_xlabel('x')
#ax.set_xlim(-40, 40)
ax.set_ylabel('y')
#ax.set_ylim(-40, 40)
ax.set_zlabel('h')
ax.set_zlim(-0.1, 0.1)
plt.savefig("plot2D_it=3.pdf")
plt.show()
