1
Fork 0

Black reformat on processing

This commit is contained in:
Edgar P. Burkhart 2022-04-07 11:51:03 +02:00
parent eff2b09c8f
commit 358b57f23d
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
7 changed files with 62 additions and 41 deletions

View file

@ -30,7 +30,7 @@ def data(var):
x = data("x")
#t = data("t")
# t = data("t")
watl = data("watl")
botl = data("botl")

View file

@ -55,7 +55,7 @@ vk = np.sqrt((velk[n] ** 2).sum(axis=1))
lines = ax.plot(x, zk[n].T, c="#0066cc")
quiv = []
for i in range(len(lines)-1):
for i in range(len(lines) - 1):
quiv.append(
ax.quiver(
x[::50],

View file

@ -70,28 +70,46 @@ def PUV_dam(u, p, h, fs, zmesP, zmesU):
ccs.append(cs)
cccp.append(ccp)
# calcul des amplitudes des ondes incidentes a partir de uhoriz et p
aincident13.append(0.5 * np.sqrt(
a1 * a1 / (cc * cc)
+ a3
* a3
* g
* g
* xf[ii]
* xf[ii]
/ (omega[ii] * omega[ii] * ccp * ccp)
+ 2 * a1 * a3 * g * k_xb[ii] * np.cos(phi3 - phi1) / (cc * ccp * omega[ii])
))
areflechi13.append(0.5 * np.sqrt(
a1 * a1 / (cc * cc)
+ a3
* a3
* g
* g
* k_xb[ii]
* k_xb[ii]
/ (omega[ii] * omega[ii] * ccp * ccp)
- 2 * a1 * a3 * g * xf[ii] * np.cos(phi3 - phi1) / (cc * ccp * omega[ii])
))
aincident13.append(
0.5
* np.sqrt(
a1 * a1 / (cc * cc)
+ a3
* a3
* g
* g
* xf[ii]
* xf[ii]
/ (omega[ii] * omega[ii] * ccp * ccp)
+ 2
* a1
* a3
* g
* k_xb[ii]
* np.cos(phi3 - phi1)
/ (cc * ccp * omega[ii])
)
)
areflechi13.append(
0.5
* np.sqrt(
a1 * a1 / (cc * cc)
+ a3
* a3
* g
* g
* k_xb[ii]
* k_xb[ii]
/ (omega[ii] * omega[ii] * ccp * ccp)
- 2
* a1
* a3
* g
* xf[ii]
* np.cos(phi3 - phi1)
/ (cc * ccp * omega[ii])
)
)
cv = g * xf[ii] / (omega[ii] * ccp)
cp = 1 / cc
aprog.append(a3 / (g * xf[ii] / (omega[ii] * ccp)))

View file

@ -35,9 +35,9 @@ vel = np.load(inp.joinpath("vel_x.npy"))
# Cospectral calculations
x0 = config.getint("post", "x0")
arg_x0 = np.abs(x - x0).argmin()
t0 = int(config.getfloat("post", "t0")*1e3)
t0 = int(config.getfloat("post", "t0") * 1e3)
arg_t0 = np.abs(t - t0).argmin()
dt = np.diff(t).mean()*1e-3
dt = np.diff(t).mean() * 1e-3
f = 1 / dt
nperseg = config.getint("post", "nperseg", fallback=None)
log.info(f"Computing reflection coefficient at x={x0}")
@ -58,10 +58,10 @@ R = np.sqrt(
(np.abs(phi_eta[1]) + np.abs(phi_u[1]) - 2 * phi_eta_u[1].real)
/ (np.abs(phi_eta[1]) + np.abs(phi_u[1]) + 2 * phi_eta_u[1].real)
)
#R = np.sqrt(
# R = np.sqrt(
# (1 + G**2 - 2 * G * np.cos(th_eta_u))
# / (1 + G**2 + 2 * G * np.cos(th_eta_u))
#)
# )
if config.has_option("post", "compare"):
inp_comp = pathlib.Path(config.get("post", "compare"))
@ -91,27 +91,27 @@ if config.has_option("post", "compare"):
(np.abs(phi_eta_[1]) + np.abs(phi_u_[1]) - 2 * phi_eta_u_[1].real)
/ (np.abs(phi_eta_[1]) + np.abs(phi_u_[1]) + 2 * phi_eta_u_[1].real)
)
#R_ = np.sqrt(
# R_ = np.sqrt(
# (1 + G_**2 - 2 * G_ * np.cos(th_eta_u_))
# / (1 + G_**2 + 2 * G_ * np.cos(th_eta_u_))
#)
# )
# Plotting
log.info("Plotting results")
fig, (ax_watl, ax_vel) = plt.subplots(2)
ax_watl.plot(t*1e-3, watl[:, arg_x0], lw=1, label="watl")
ax_watl.plot(t * 1e-3, watl[:, arg_x0], lw=1, label="watl")
ax_watl.set(xlabel="t (s)", ylabel="z (m)")
ax_watl.autoscale(axis="x", tight=True)
ax_watl.grid()
ax_watl.axvline(t0*1e-3, c="k", alpha=0.2)
ax_watl.axvline(t0 * 1e-3, c="k", alpha=0.2)
ax_vel.plot(t*1e-3, vel[:, arg_x0], lw=1, label="vel")
ax_vel.plot(t * 1e-3, vel[:, arg_x0], lw=1, label="vel")
ax_vel.set(xlabel="t (s)", ylabel="U (m/s)")
ax_vel.autoscale(axis="x", tight=True)
ax_vel.grid()
ax_vel.axvline(t0*1e-3, c="k", alpha=0.2)
ax_vel.axvline(t0 * 1e-3, c="k", alpha=0.2)
fig.tight_layout()

View file

@ -40,7 +40,9 @@ for r in np.arange(0, 1.1, 0.1):
*Rn,
c="#ff6600",
)
axr.annotate(f"{r=:.1f}", (Rn[0][0], Rn[1][0]), bbox={"boxstyle": "square", "facecolor": "w"})
axr.annotate(
f"{r=:.1f}", (Rn[0][0], Rn[1][0]), bbox={"boxstyle": "square", "facecolor": "w"}
)
axr.grid()
axr.autoscale(True, "x", tight=True)
axr.set(ylim=(0, 1), ylabel="R", xlabel="f")

View file

@ -7,6 +7,7 @@ import numpy as np
log = logging.getLogger("read_swash")
class ReadSwash:
def __init__(self):
self._n_x = None

View file

@ -30,14 +30,14 @@ np.save(inp.joinpath("tsec"), rsws.read_time(sws_out.joinpath("tsec.dat")))
np.save(inp.joinpath("xp"), rsws.read_x(sws_out.joinpath("xp.dat")))
var = {
#"dep": rsws.read_scalar,
# "dep": rsws.read_scalar,
"botl": rsws.read_const,
"watl": rsws.read_scalar,
#"pressk": rsws.read_scalar_lay,
#"nhprsk": rsws.read_scalar_lay,
#"zk": rsws.read_scalar_lay,
#"velk": rsws.read_vector_lay,
#"vz": rsws.read_scalar_lay,
# "pressk": rsws.read_scalar_lay,
# "nhprsk": rsws.read_scalar_lay,
# "zk": rsws.read_scalar_lay,
# "velk": rsws.read_vector_lay,
# "vz": rsws.read_scalar_lay,
"vel": rsws.read_vector,
}