begin to add xarray capability and add new variables in coords

Author: browniea

mhkit/river/io/d3d.py

@@ -35,8 +35,8 @@ def get_all_time(data: netCDF4.Dataset) -> NDArray:
         simulation conditions at that time.
     """
 
-    if not isinstance(data, netCDF4.Dataset):
-        raise TypeError("data must be a NetCDF4 object")
+    if not isinstance(data, (netCDF4.Dataset, xr.Dataset)):
+        raise TypeError("data must be a NetCDF4 object or xarray Dataset")
 
     seconds_run = np.ma.getdata(data.variables["time"][:], False)
 
@@ -244,6 +244,10 @@ def get_layer_data(
                 "name": "mesh2d_nLayers",
                 "coords": data.variables["mesh2d_layer_sigma"][:],
             },
+            "mesh2d_face_x mesh2d_face_y mesh2d_layer_sigma": {
+                "name": "mesh2d_nLayers",
+                "coords": data.variables["mesh2d_layer_sigma"][:],
+            },
             "mesh2d_edge_x mesh2d_edge_y": {
                 "name": "mesh2d_nInterfaces",
                 "coords": data.variables["mesh2d_interface_sigma"][:],
@@ -637,6 +641,10 @@ def get_all_data_points(
                 "name": "mesh2d_nLayers",
                 "coords": data.variables["mesh2d_layer_sigma"][:],
             },
+            "mesh2d_face_x mesh2d_face_y mesh2d_layer_sigma": {
+                "name": "mesh2d_nLayers",
+                "coords": data.variables["mesh2d_layer_sigma"][:],
+            },
             "mesh2d_edge_x mesh2d_edge_y": {
                 "name": "mesh2d_nInterfaces",
                 "coords": data.variables["mesh2d_interface_sigma"][:],