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

Updated sws_ola with new format

Browse source
This commit is contained in:
Edgar P. Burkhart 2022-04-08 12:43:26 +02:00
parent a6ec88b87b
commit 86252c67d4
Signed by: edpibu
GPG key ID: 9833D3C5A25BD227
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
olaflow/processing/sws_ola.py
View file

@ -28,12 +28,12 @@ def data(var):
return np.load(sws_out.joinpath(f"{var}.npy")) return np.load(sws_out.joinpath(f"{var}.npy"))
x = data("xp") x = data("x")
t = data("tsec") t = data("t")
watl = data("watl") watl = data("watl")
zk = data("zk") zk = data("zk")
velk = data("velk") velk, _ = data("velk")
vz = data("vz") vz = data("vz")
olaflow_root = pathlib.Path(config.get("olaflow", "root")) olaflow_root = pathlib.Path(config.get("olaflow", "root"))
@ -44,7 +44,7 @@ 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]) zk_t = interpolate.interp1d(x, zk[680])
velk_t = interpolate.interp1d(x, velk[680, :, 0, :])(model.x) velk_t = interpolate.interp1d(x, velk[680, :, :])(model.x)
vz_t = interpolate.interp1d(x, vz[680])(model.x) vz_t = interpolate.interp1d(x, vz[680])(model.x)
zk_tl = zk_t(model.x) zk_tl = zk_t(model.x)