healpix moc index

docs/api.rst

@@ -38,6 +38,11 @@ Parameters
 
    Dataset.dggs.grid_info
    Dataset.dggs.params
+
+.. autosummary::
+   :toctree: generated
+   :template: autosummary/accessor_method.rst
+
    Dataset.dggs.decode
 
 Indexing
@@ -72,7 +77,12 @@ Parameters
 
    DataArray.dggs.grid_info
    DataArray.dggs.params
-   DataArray.dggs.decode
+
+.. autosummary::
+   :toctree: generated
+   :template: autosummary/accessor_method.rst
+
+   Dataset.dggs.decode
 
 Indexing
 --------

docs/conf.py

@@ -115,6 +115,7 @@
     "healpy": ("https://healpy.readthedocs.io/en/latest", None),
     "cdshealpix-python": ("https://cds-astro.github.io/cds-healpix-python", None),
     "shapely": ("https://shapely.readthedocs.io/en/stable", None),
+    "healpix-geo": ("https://healpix-geo.readthedocs.io/en/latest", None),
 }
 
 # -- myst-nb options ---------------------------------------------------------

pyproject.toml

@@ -24,7 +24,9 @@ classifiers = [
 requires-python = ">=3.11"
 dependencies = [
   "xarray",
+  "numpy>=2.0",
   "cdshealpix",
+  "healpix-geo>=0.0.6",
   "h3ronpy",
   "lonboard>=0.9.3",
   "pyproj>=3.3",
@@ -119,7 +121,7 @@ filterwarnings = [
 
 [tool.pixi.workspace]
 channels = ["conda-forge"]
-platforms = ["linux-64", "osx-arm64", "win-64"]
+platforms = ["linux-64", "win-64", "osx-arm64"]
 
 [tool.pixi.system-requirements]
 libc = { family = "glibc", version = "2.34" }
@@ -138,6 +140,7 @@ pyproj = ">=3.3"
 matplotlib-base = ">=3.10.5"
 arro3-core = ">=0.4.0"
 pooch = ">=1.8.2"
+healpix-geo = ">=0.0.6"
 
 [tool.pixi.feature.tests.dependencies]
 pytest = "*"
@@ -148,6 +151,7 @@ netcdf4 = "*"
 hypothesis = "*"
 geoarrow-pyarrow = "*"
 shapely = ">=2.1.1,<3"
+dask = ">=2025.7.0"
 
 [tool.pixi.feature.tests.tasks]
 tests = "pytest -n auto --cov=xdggs"
@@ -164,6 +168,7 @@ pandas = "*"
 pyarrow = "*"
 h3ronpy = { git = "git+https://github.com/keewis/h3ronpy.git", subdirectory = "h3ronpy", rev = "version" }
 cdshealpix = { git = "git+https://github.com/cds-astro/cds-healpix-python.git" }
+healpix-geo = { git = "git+https://github.com/eopf-dggs/healpix-geo.git" }
 astropy = { git = "git+https://github.com/astropy/astropy.git" }
 
 [tool.pixi.feature.nightly.dependencies]

xdggs/accessor.py

@@ -29,7 +29,9 @@ def __init__(self, obj: xr.Dataset | xr.DataArray):
         self._name = name
         self._index = index
 
-    def decode(self, grid_info=None, *, name="cell_ids") -> xr.Dataset | xr.DataArray:
+    def decode(
+        self, grid_info=None, *, name="cell_ids", index_options=None, **index_kwargs
+    ) -> xr.Dataset | xr.DataArray:
         """decode the DGGS cell ids
 
         Parameters
@@ -39,6 +41,8 @@ def decode(self, grid_info=None, *, name="cell_ids") -> xr.Dataset | xr.DataArra
             the dataset.
         name : str, default: "cell_ids"
             The name of the coordinate containing the cell ids.
+        index_options, **index_kwargs : dict, optional
+            Additional options to forward to the index.
 
         Returns
         -------
@@ -51,7 +55,12 @@ def decode(self, grid_info=None, *, name="cell_ids") -> xr.Dataset | xr.DataArra
         if isinstance(grid_info, dict):
             var.attrs = grid_info
 
-        return self._obj.drop_indexes(name, errors="ignore").set_xindex(name, DGGSIndex)
+        if index_options is None:
+            index_options = {}
+
+        return self._obj.drop_indexes(name, errors="ignore").set_xindex(
+            name, DGGSIndex, **(index_options | index_kwargs)
+        )
 
     @property
     def index(self) -> DGGSIndex:

xdggs/h3.py

@@ -18,7 +18,6 @@
         cells_to_wkb_polygons,
         coordinates_to_cells,
     )
-from xarray.indexes import PandasIndex
 
 from xdggs.grid import DGGSInfo, translate_parameters
 from xdggs.index import DGGSIndex
@@ -203,7 +202,7 @@ class H3Index(DGGSIndex):
 
     def __init__(
         self,
-        cell_ids: Any | PandasIndex,
+        cell_ids: Any | xr.Index,
         dim: str,
         grid_info: DGGSInfo,
     ):
@@ -226,8 +225,8 @@ def from_variables(
     def grid_info(self) -> H3Info:
         return self._grid
 
-    def _replace(self, new_pd_index: PandasIndex):
-        return type(self)(new_pd_index, self._dim, self._grid)
+    def _replace(self, new_index: xr.Index):
+        return type(self)(new_index, self._dim, self._grid)
 
     def _repr_inline_(self, max_width: int):
         return f"H3Index(level={self._grid.level})"