crs tools

cordex/ccrs.py

@@ -0,0 +1,23 @@
+from cartopy import crs as ccrs
+
+crs_map = {
+    "rotated_latitude_longitude": {
+        "crs": ccrs.RotatedPole,
+        "kwargs": {
+            "pole_longitude": "grid_north_pole_longitude",
+            "pole_latitude": "grid_north_pole_latitude",
+            "central_rotated_longitude": 0.0,
+        },
+    }
+}
+
+
+def _ccrs_from_cf(mapping):
+    crs = crs_map.get(mapping.grid_mapping_name)
+    kwargs = {kw: (mapping.attrs.get(v) or v) for kw, v in crs["kwargs"].items()}
+    return crs["crs"](**kwargs)
+
+
+def get_ccrs(ds):
+    mapping = ds.cf["grid_mapping"]
+    return _ccrs_from_cf(mapping)

cordex/utils.py

@@ -98,3 +98,16 @@ def cell_area(ds, R=6371000, attrs=True):
         da.attrs = {}
 
     return da
+
+
+def create_polygon(obj):
+    """create polygon in rotated pole coords"""
+    from shapely.geometry import Polygon
+
+    coords = [
+        [obj.cf["X"].min(), obj.cf["Y"].min()],
+        [obj.cf["X"].max(), obj.cf["Y"].min()],
+        [obj.cf["X"].max(), obj.cf["Y"].max()],
+        [obj.cf["X"].min(), obj.cf["Y"].max()],
+    ]
+    return Polygon(coords)