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()