enable zarr backend, fix #20.

kaizhang · Dec 9, 2024 · 329978a · 329978a
1 parent 39611b7
commit 329978a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/snapatac2-python/Cargo.toml b/snapatac2-python/Cargo.toml
@@ -14,6 +14,7 @@ keywords = ["single-cell", "biology"]
 snapatac2-core = { workspace = true }
 anndata = "0.6.1"
 anndata-hdf5 = "0.5"
+anndata-zarr = "0.1"
 pyanndata = "0.5.1"
 anyhow = "1.0"
 bed-utils = "0.7.1"

diff --git a/snapatac2-python/src/utils/anndata.rs b/snapatac2-python/src/utils/anndata.rs
@@ -214,12 +214,18 @@ macro_rules! with_anndata {
                 H5::NAME => {
                     $fun!(x.inner_ref::<H5>().deref())
                 }
+                anndata_zarr::Zarr::NAME => {
+                    $fun!(x.inner_ref::<anndata_zarr::Zarr>().deref())
+                }
                 x => panic!("Unsupported backend: {}", x),
             },
             AnnDataLike::AnnDataSet(x) => match x.backend().as_str() {
                 H5::NAME => {
                     $fun!(x.inner_ref::<H5>().deref())
                 }
+                anndata_zarr::Zarr::NAME => {
+                    $fun!(x.inner_ref::<anndata_zarr::Zarr>().deref())
+                }
                 x => panic!("Unsupported backend: {}", x),
             },
             AnnDataLike::PyAnnData(x) => {
@@ -264,12 +270,18 @@ macro_rules! with_rs_anndata {
                 H5::NAME => {
                     $fun!(x.inner_ref::<H5>().deref())
                 }
+                anndata_zarr::Zarr::NAME => {
+                    $fun!(x.inner_ref::<anndata_zarr::Zarr>().deref())
+                }
                 x => panic!("Unsupported backend: {}", x),
             },
             RustAnnDataLike::AnnDataSet(x) => match x.backend().as_str() {
                 H5::NAME => {
                     $fun!(x.inner_ref::<H5>().deref())
                 }
+                anndata_zarr::Zarr::NAME => {
+                    $fun!(x.inner_ref::<anndata_zarr::Zarr>().deref())
+                }
                 x => panic!("Unsupported backend: {}", x),
             },
         }