m2cce/internship
m2cce/internship
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Display layered results; fix read_swash for layered vectors

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-03-15 13:34:15 +01:00
parent cb05a03a28
commit 12ad3b5c59
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
3 changed files with 86 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

25
swash/processing/animate.py
View file

@ -23,22 +23,29 @@ config.read("config.ini")
inp = pathlib.Path(config.get("post", "inp"))
root = pathlib.Path(config.get("swash", "out"))
data = np.load(inp.joinpath(config.get("post", "case")))
bathy = pd.read_hdf(
pathlib.Path(config.get("data", "out")).joinpath("bathy.h5"), "bathy"
)
x = data["x"]
t = data["t"]
wl = np.maximum(data["watl"], -data["botl"])
print(x.size, -np.arange(0, 1 * bathy.hstru.size, 1)[::-1].size)
def data(var):
return np.load(inp.joinpath(f"{var}.npy"))
x = data("xp")
t = data("tsec")
watl = data("watl")
botl = data("botl")
wl = np.maximum(watl, -botl)
# print(x.size, -np.arange(0, 1 * bathy.hstru.size, 1)[::-1].size)
fig, ax = plt.subplots()
ax.plot(x, -data["botl"], c="k")
ax.fill_between(
x, -data["botl"], -data["botl"] + bathy.hstru, color="k", alpha=0.2
)
ax.plot(x, -botl, c="k")
# ax.fill_between(
# x, -botl, -data["botl"] + bathy.hstru, color="k", alpha=0.2
# )
(line,) = ax.plot(x, wl[0])