Skip to content

Commit

Permalink
ci: added local function for telemac 2D extractions
Browse files Browse the repository at this point in the history
  • Loading branch information
tomsail committed Jul 15, 2024
1 parent af108f6 commit d4672df
Showing 1 changed file with 12 additions and 4 deletions.
16 changes: 12 additions & 4 deletions pyposeidon/utils/data.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,12 +15,9 @@
from pyposeidon.mesh import r2d
import pyposeidon.model as pm
from pyposeidon.tools import flat_list
from pyposeidon.utils.get_value import get_value
from pyposeidon.telemac import extract_t_elev_2D
import datetime
from pyposeidon.utils.cpoint import closest_n_points
import xarray as xr
import glob
import sys
import logging
import json

Expand All @@ -42,6 +39,17 @@ def get_output(solver_name: str, **kwargs):
return instance


def extract_t_elev_2D(
ds: xr.Dataset, x: float, y: float, var: str = "elev", xstr: str = "longitude", ystr: str = "latitude"
):
lons, lats = ds[xstr].values, ds[ystr].values
indx, _ = closest_n_points(np.array([x, y]).T, 1, np.array([lons, lats]).T)
ds_ = ds.isel(node=indx[0])
elev_ = ds_[var].values
t_ = [pd.Timestamp(ti) for ti in ds_.time.values]
return pd.Series(elev_, index=t_), float(ds_[xstr]), float(ds_[ystr])


class D3DResults:
def __init__(self, **kwargs):
rpath = kwargs.get("rpath", "./d3d/")
Expand Down

0 comments on commit d4672df

Please sign in to comment.