adding regrid option

CMCC-Foundation · Jan 27, 2025 · 628a233 · 628a233
1 parent 6523f30
commit 628a233
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/datastore/datastore/datastore.py b/datastore/datastore/datastore.py
@@ -461,6 +461,9 @@ def _process_query(kube, query: GeoQuery, compute: None | bool = False):
         if query.resample:
             Datastore._LOG.debug("Applying resample...")
             kube = kube.resample(**query.resample)
+        if query.regrid:
+            if query.regrid == 'regular':
+                kube = kube.to_regular()
         return kube.compute() if compute else kube
 
     @staticmethod

diff --git a/datastore/geoquery/geoquery.py b/datastore/geoquery/geoquery.py
@@ -17,6 +17,7 @@ class GeoQuery(BaseModel, extra="allow"):
     filters: Optional[Dict]
     format: Optional[str]
     format_args: Optional[Dict]
+    regrid: Optional[str]
 
     # TODO: Check if we are going to allow the vertical coordinates inside both
     # `area`/`location` nad `vertical`