updated convert_time to take xarray

Author: browniea

mhkit/river/io/d3d.py

@@ -23,8 +23,8 @@ def get_all_time(data: (netCDF4.Dataset, xr.Dataset)) -> NDArray:
 
     Parameters
     ----------
-    data: NetCDF4 object or xarray Dataset
-       A NetCDF4 object or xarray Dataset that contains spatial data, e.g. velocity or shear
+    data: NetCDF4 object
+       A NetCDF4 object that contains spatial data, e.g. velocity or shear
        stress generated by running a Delft3D model.
 
     Returns
@@ -36,9 +36,13 @@ def get_all_time(data: (netCDF4.Dataset, xr.Dataset)) -> NDArray:
     """
 
     if not isinstance(data, (netCDF4.Dataset, xr.Dataset)):
-        raise TypeError("data must be a NetCDF4 object or xarray Dataset")
+        raise TypeError("data must be a NetCDF4 object Dataset")
 
-    seconds_run = np.ma.getdata(data.variables["time"][:], False)
+    if isinstance(data, netCDF4.Dataset):
+        seconds_run = np.ma.getdata(data.variables["time"][:], False)
+
+    elif isinstance(data, xr.Dataset):
+        seconds_run = (data.time.values - data.time.values[0]).astype('timedelta64[s]').astype(int)
 
     return seconds_run
 
@@ -88,7 +92,7 @@ def seconds_to_index(data: netCDF4.Dataset, seconds_run: Union[int, float]) -> i
 
 
 def _convert_time(
-    data: netCDF4.Dataset,
+    data: netCDF4.Dataset or xr.Dataset,
     time_index: Optional[Union[int, float]] = None,
     seconds_run: Optional[Union[int, float]] = None,
 ) -> Union[int, float]: