m2cce/internship
m2cce/internship
1
Fork 0
Code Issues Activity

Fixed optional comparison

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-03-10 10:55:34 +01:00
parent fdfbd081d9
commit bf4e9e6958
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 16 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
swash/processing/post.py
View file

@ -111,15 +111,16 @@ ax_fft.plot(
alpha=0.2,
label="PSD ($\\eta$, cas 1)",
)
ax_fft.plot(
*sgl.welch(eta_, 1 / dt, nperseg=nperseg),
lw=1,
c="k",
alpha=0.2,
label="PSD ($\\eta$, cas 2)",
)
ax_r.plot(phi_eta[0], R, marker="+", label="R (cas 1)")
ax_r.plot(phi_eta[0], R_, marker="+", label="R (cas 2)")
if config.has_option("post", "compare"):
ax_fft.plot(
*sgl.welch(eta_, 1 / dt, nperseg=nperseg),
lw=1,
c="k",
alpha=0.2,
label="PSD ($\\eta$, cas 2)",
)
ax_r.plot(phi_eta[0], R_, marker="+", label="R (cas 2)")
ax_r.set(xlim=(0, 0.3), ylim=(0, 1), xlabel="f (Hz)", ylabel="R")
ax_fft.set(ylim=0, ylabel="PSD (m²/Hz)")
ax_r.grid()
@ -129,16 +130,17 @@ fig_r.tight_layout()
fig_x, ax_x = plt.subplots(figsize=(10, 1))
ax_x.plot(data["x"], -data["botl"], color="k")
ax_x.plot(data["x"], -data_comp["botl"], color="k", ls="-.")
ax_x.plot(
data["x"],
np.maximum(data["watl"][arg_t0, :], -data["botl"]),
)
ax_x.plot(
data["x"],
np.maximum(data_comp["watl"][arg_t0, :], -data_comp["botl"]),
ls="-.",
)
if config.has_option("post", "compare"):
ax_x.plot(data["x"], -data_comp["botl"], color="k", ls="-.")
ax_x.plot(
data["x"],
np.maximum(data_comp["watl"][arg_t0, :], -data_comp["botl"]),
ls="-.",
)
ax_x.axvline(x0, c="k", alpha=0.2)
ax_x.set(xlabel="x (m)", ylabel="z (m)")
ax_x.autoscale(axis="x", tight=True)