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

Changed styling

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-03-07 12:30:50 +01:00
parent fa2e30deec
commit 371acdf9d1
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
swash/processing/post.py
View file

@ -55,13 +55,13 @@ R = np.sqrt(
log.info("Plotting results") log.info("Plotting results")
fig, (ax_watl, ax_vel) = plt.subplots(2) fig, (ax_watl, ax_vel) = plt.subplots(2)
ax_watl.plot(t, data["watl"][:, arg_x0], label="watl") ax_watl.plot(t, data["watl"][:, arg_x0], lw=1, label="watl")
ax_watl.set(xlabel="t (s)", ylabel="z (m)") ax_watl.set(xlabel="t (s)", ylabel="z (m)")
ax_watl.autoscale(axis="x", tight=True) ax_watl.autoscale(axis="x", tight=True)
ax_watl.grid() ax_watl.grid()
ax_watl.axvline(t0, c="k", alpha=0.2) ax_watl.axvline(t0, c="k", alpha=0.2)
ax_vel.plot(t, data["vel"][:, 0, arg_x0], label="vel") ax_vel.plot(t, data["vel"][:, 0, arg_x0], lw=1, label="vel")
ax_vel.set(xlabel="t (s)", ylabel="U (m/s)") ax_vel.set(xlabel="t (s)", ylabel="U (m/s)")
ax_vel.autoscale(axis="x", tight=True) ax_vel.autoscale(axis="x", tight=True)
ax_vel.grid() ax_vel.grid()
@ -77,7 +77,7 @@ ax_fft.plot(
lw=1, lw=1,
c="k", c="k",
alpha=0.2, alpha=0.2,
label="PSD", label="PSD ($\\eta$)",
) )
ax_r.plot(phi_eta[0], R, marker="+", label="R") ax_r.plot(phi_eta[0], R, marker="+", label="R")
ax_r.set(xlim=(0, 0.3), ylim=(0, 1), xlabel="f (Hz)", ylabel="R") ax_r.set(xlim=(0, 0.3), ylim=(0, 1), xlabel="f (Hz)", ylabel="R")
@ -87,7 +87,7 @@ ax_r.legend(loc="upper left")
ax_fft.legend(loc="upper right") ax_fft.legend(loc="upper right")
fig_r.tight_layout() fig_r.tight_layout()
fig_x, ax_x = plt.subplots() fig_x, ax_x = plt.subplots(figsize=(10, 1))
ax_x.plot(data["x"], -data["botl"], color="k") ax_x.plot(data["x"], -data["botl"], color="k")
ax_x.fill_between( ax_x.fill_between(
data["x"], data["x"],
@ -97,6 +97,8 @@ ax_x.fill_between(
ax_x.axvline(x0, c="k", alpha=0.2) ax_x.axvline(x0, c="k", alpha=0.2)
ax_x.set(xlabel="x (m)", ylabel="z (m)") ax_x.set(xlabel="x (m)", ylabel="z (m)")
ax_x.autoscale(axis="x", tight=True) ax_x.autoscale(axis="x", tight=True)
ax_x.set(aspect="equal")
fig_x.tight_layout()
out = pathlib.Path(config.get("post", "out")).joinpath(f"t{t0}x{x0}") out = pathlib.Path(config.get("post", "out")).joinpath(f"t{t0}x{x0}")
log.info(f"Saving plots in '{out}'") log.info(f"Saving plots in '{out}'")