1
Fork 0

Added velocity to sws_ola

This commit is contained in:
Edgar P. Burkhart 2022-03-28 10:59:35 +02:00
parent a335f2f5ce
commit 285b9476bc
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
2 changed files with 42 additions and 3 deletions

View file

@ -25,6 +25,21 @@ class OFModel:
)
)
def write_vector_field(self, field, values):
with open(self._root.joinpath("0", field), "r") as aw_file:
aw_raw = aw_file.read()
with open(self._root.joinpath("0", field), "w") as aw_file:
aw_file.write(
re.sub(
r"(?<=\(\n).*?(?=\n\))",
"\n".join(map(lambda x: f"({' '.join(x.astype('str'))})", values)),
aw_raw,
count=1,
flags=re.S,
)
)
@property
def x(self):
return self._x

View file

@ -5,6 +5,7 @@ import pathlib
import numpy as np
from scipy import interpolate
import matplotlib.pyplot as plt
from .olaflow import OFModel
@ -22,9 +23,16 @@ config = configparser.ConfigParser()
config.read("config.ini")
sws_out = pathlib.Path(config.get("swash", "np_out"))
x = np.load(sws_out.joinpath("xp.npy"))
t = np.load(sws_out.joinpath("tsec.npy"))
watl = np.load(sws_out.joinpath("watl.npy"))
def data(var):
return np.load(sws_out.joinpath(f"{var}.npy"))
x = data("xp")
t = data("tsec")
watl = data("watl")
zk = data("zk")
velk = data("velk")
vz = data("vz")
olaflow_root = pathlib.Path(config.get("olaflow", "root"))
model = OFModel(olaflow_root)
@ -33,4 +41,20 @@ model.read_mesh()
watl_t = interpolate.interp1d(x, watl[680])
alpha_water = np.where(model.z < watl_t(model.x), 1, 0)
zk_t = interpolate.interp1d(x, zk[680])
velk_t = interpolate.interp1d(x, velk[680,:,0,:])(model.x)
vz_t = interpolate.interp1d(x, vz[680])(model.x)
zk_tl = zk_t(model.x)
print(velk.shape)
ux = np.zeros(model.x.shape)
uy = np.zeros(model.x.shape)
uz = np.zeros(model.x.shape)
print(zk_tl.shape, velk.shape, vz.shape)
for zk_l, velk_l, vz_l in zip(zk_tl, velk_t, vz_t):
ux = np.where(model.z < zk_l, velk_l, ux)
uz = np.where(model.z < zk_l, vz_l, uz)
print(np.stack((ux,uy,uz)).T)
model.write_field("alpha.water", alpha_water)
model.write_vector_field("U", np.stack((ux, uy, uz)).T)