Enhance radar and geo coordinates detection for stacked datasets

AlexeyPechnikov · Jul 9, 2024 · 2c41377 · 2c41377
1 parent 88697f5
commit 2c41377
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pygmtsar/pygmtsar/datagrid.py b/pygmtsar/pygmtsar/datagrid.py
@@ -130,6 +130,9 @@ def is_ra(grid):
             True if the grid is in radar coordinates, False otherwise.
         """
         dims = grid.dims
+        coords = grid.coords
+        if 'stack' in dims and 'y' in coords and 'x' in coords:
+            return True
         if 'y' in dims and 'x' in dims:
             return True
         return False
@@ -150,6 +153,9 @@ def is_geo(grid):
             True if the grid is in geographic coordinates, False otherwise.
         """
         dims = grid.dims
+        coords = grid.coords
+        if 'stack' in dims and 'lat' in coords and 'lon' in coords:
+            return True
         if 'lat' in dims and 'lon' in dims:
             return True
         return False