flow.compute_velocity(flow.n_h - 1)
fig_u = plt.figure(figsize=(6 * 1., 3.6 * 1.))
ax = fig_u.gca()
ax.plot(flow.p, flow.c_u[0, :].T, 'b-')
ax.grid(linestyle="--")
plt.ylabel('$c-u$', fontsize=12)
plt.xlabel('$p$', fontsize=12)
ax.grid(linestyle="--")
plt.tight_layout()
print(min(flow.c_u[0, :]))
#%% Plot variation of depth
fig_depth = plt.figure(figsize=(6 * 1., 3.6 * 1.))
ax = fig_depth.gca()
ax.plot(flow.Q[0:], flow.d[0:], 'b-o', markersize=4)
ax.grid(linestyle="--")
plt.xlabel('$Q$', fontsize=12)
plt.ylabel('$d$', fontsize=12)
plt.xticks(fontsize=12)
plt.yticks(fontsize=12)
plt.tight_layout()
#%% Plot pressure
flow.compute_pressure(0, flow.n_h - 1)
fig = plt.figure(figsize=(6 * 1., 3.6 * 1.))
ax = fig.gca()
im = ax.contourf(flow.x, flow.y2, flow.pres, 20, cmap="RdBu_r")
ax.contour(flow.x, flow.y2, flow.pres, 20, linewidths=0.5)
fig.colorbar(im)
plt.xlabel('x')
plt.xlabel('y')
