Added velocity to sws_ola
This commit is contained in:
parent
a335f2f5ce
commit
285b9476bc
2 changed files with 42 additions and 3 deletions
|
@ -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
|
@property
|
||||||
def x(self):
|
def x(self):
|
||||||
return self._x
|
return self._x
|
||||||
|
|
|
@ -5,6 +5,7 @@ import pathlib
|
||||||
|
|
||||||
import numpy as np
|
import numpy as np
|
||||||
from scipy import interpolate
|
from scipy import interpolate
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
|
||||||
from .olaflow import OFModel
|
from .olaflow import OFModel
|
||||||
|
|
||||||
|
@ -22,9 +23,16 @@ config = configparser.ConfigParser()
|
||||||
config.read("config.ini")
|
config.read("config.ini")
|
||||||
|
|
||||||
sws_out = pathlib.Path(config.get("swash", "np_out"))
|
sws_out = pathlib.Path(config.get("swash", "np_out"))
|
||||||
x = np.load(sws_out.joinpath("xp.npy"))
|
def data(var):
|
||||||
t = np.load(sws_out.joinpath("tsec.npy"))
|
return np.load(sws_out.joinpath(f"{var}.npy"))
|
||||||
watl = np.load(sws_out.joinpath("watl.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"))
|
olaflow_root = pathlib.Path(config.get("olaflow", "root"))
|
||||||
model = OFModel(olaflow_root)
|
model = OFModel(olaflow_root)
|
||||||
|
@ -33,4 +41,20 @@ model.read_mesh()
|
||||||
watl_t = interpolate.interp1d(x, watl[680])
|
watl_t = interpolate.interp1d(x, watl[680])
|
||||||
alpha_water = np.where(model.z < watl_t(model.x), 1, 0)
|
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_field("alpha.water", alpha_water)
|
||||||
|
model.write_vector_field("U", np.stack((ux, uy, uz)).T)
|
||||||
|
|
Loading…
Reference in a new issue