Adding mesh_mask_edges file for AA in nemo_curvilinear

OceanParcels · Aug 27, 2024 · f557e3c · f557e3c
1 parent 4505636
commit f557e3c
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 0 deletions.
diff --git a/docs/examples/example_nemo_curvilinear.py b/docs/examples/example_nemo_curvilinear.py
@@ -34,6 +34,34 @@ def run_nemo_curvilinear(mode, outfile, advtype="RK4"):
     fieldset = parcels.FieldSet.from_nemo(
         filenames, variables, dimensions, chunksize=chunksize
     )
+    if advtype == "AA":
+        mesh_mask = f"{data_folder}/mesh_mask_edges.nc"
+        fieldset.add_field(
+            parcels.Field.from_netcdf(
+                mesh_mask, "e2u", dimensions, interp_method="nearest"
+            )
+        )
+        fieldset.add_field(
+            parcels.Field.from_netcdf(
+                mesh_mask, "e1v", dimensions, interp_method="nearest"
+            )
+        )
+        fieldset.add_field(
+            parcels.Field.from_netcdf(
+                mesh_mask, "e1t", dimensions, interp_method="nearest"
+            )
+        )
+        fieldset.add_field(
+            parcels.Field.from_netcdf(
+                mesh_mask, "e2t", dimensions, interp_method="nearest"
+            )
+        )
+        fieldset.add_field(
+            parcels.Field(
+                "e3t", np.array(1), lon=0, lat=0, depth=0, interp_method="nearest"
+            )
+        )
+
     assert fieldset.U.chunksize == chunksize
 
     # Now run particles as normal

diff --git a/parcels/tools/exampledata_utils.py b/parcels/tools/exampledata_utils.py
@@ -34,6 +34,7 @@
         "U_purely_zonal-ORCA025_grid_U.nc4",
         "V_purely_zonal-ORCA025_grid_V.nc4",
         "mesh_mask.nc4",
+        "mesh_mask_edges.nc",
     ],
     "NemoNorthSeaORCA025-N006_data": [
         "ORCA025-N06_20000104d05U.nc",