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

sws_ola adjust layered zk to water level

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-04-14 09:59:09 +02:00
parent 7239284410
commit e664259e61
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
olaflow/processing/sws_ola.py
View file

@ -44,10 +44,11 @@ olaflow_root = args.output
model = OFModel(olaflow_root) model = OFModel(olaflow_root)
model.read_mesh() model.read_mesh()
watl_t = interpolate.interp1d(x, watl[arg_t0] + config.getfloat("bathy", "level", fallback=0.)) level = config.getfloat("bathy", "level", fallback=0.)
watl_t = interpolate.interp1d(x, watl[arg_t0] + level)
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[arg_t0]) zk_t = interpolate.interp1d(x, zk[arg_t0] + level)
velk_t = interpolate.interp1d(x, velk[arg_t0, :, :])(model.x) velk_t = interpolate.interp1d(x, velk[arg_t0, :, :])(model.x)
vz_t = interpolate.interp1d(x, vz[arg_t0])(model.x) vz_t = interpolate.interp1d(x, vz[arg_t0])(model.x)
zk_tl = zk_t(model.x) zk_tl = zk_t(model.x)