zarr_open

NAMESPACE

@@ -33,6 +33,7 @@ export(zarr_create_array)
 export(zarr_create_empty)
 export(zarr_create_group)
 export(zarr_create_zeros)
+export(zarr_open)
 export(zarr_open_array)
 export(zarr_open_group)
 export(zarr_save_array)

R/creation.R

@@ -790,4 +790,50 @@ zarr_save_array <- function(store, arr, ...) {
     # Reference: https://github.com/zarr-developers/zarr-python/blob/5dd4a0e6cdc04c6413e14f57f61d389972ea937c/zarr/convenience.py#L112
     store <- normalize_store_arg(store)
     zarr_create_array(data=arr$get_item("..."), shape=arr$get_shape(), store=store, ...)
-}
+}
+
+#' Convenience function to open a group or array using file-mode-like semantics.
+#' @param store : MutableMapping or string, optional
+#'     Store or path to directory in file system or name of zip file.
+#' @param mode : {'r', 'r+', 'a', 'w', 'w-'}, optional
+#'     Persistence mode: 'r' means read only (must exist); 'r+' means
+#'     read/write (must exist); 'a' means read/write (create if doesn't
+#'     exist); 'w' means create (overwrite if exists); 'w-' means create
+#'     (fail if exists).
+#' @param path : str or NA, optional
+#'     The path within the store to open.
+#' @param ... Additional arguments to pass to zarr_open_array or zarr_open_group.
+#' @returns ZarrArray or ZarrGroup
+#' @export
+zarr_open <- function(store = NA, mode = NA, path = NA, ...) {
+    kwargs <- list(...)
+
+    if(is_na(mode)) {
+        mode <- "a"
+    }
+
+    store <- normalize_store_arg(store)
+    path <- normalize_storage_path(path)
+
+    if(mode %in% c("w", "w-", "x")) {
+        if("shape" %in% names(kwargs)) {
+            return(zarr_open_array(store=store, mode=mode, path=path, ...))
+        } else {
+            return(zarr_open_group(store=store, mode=mode, path=path, ...))
+        }
+    } else if(mode == "a") {
+        if("shape" %in% names(kwargs) || contains_array(store, path)) {
+            return(zarr_open_array(store=store, mode=mode, path=path, ...))
+        } else {
+            return(zarr_open_group(store=store, mode=mode, path=path, ...))
+        }
+    } else {
+        if(contains_array(store, path)) {
+            return(zarr_open_array(store=store, mode=mode, path=path, ...))
+        } else if(contains_group(store, path)) {
+            return(zarr_open_group(store=store, mode=mode, path=path, ...))
+        } else {
+            stop("PathNotFoundError(path)")
+        }
+    }
+}

pkgdown/_pkgdown.yml

@@ -62,6 +62,7 @@ reference:
   - zarr_create_zeros
   - zarr_create_array
   - zarr_create_group
+  - zarr_open
   - zarr_open_group
   - zarr_open_array
   - zarr_save_array

tests/testthat/test-compat.R

@@ -10,6 +10,16 @@ test_that("Can open Zarr group using convenience function", {
     expect_equal(a$get_shape(), c(4))
 })
 
+test_that("Can open Zarr group or array using convenience function", {
+
+    root <- system.file("extdata", "fixtures", "v2", "data.zarr", package="pizzarr")
+    g <- zarr_open(root)
+    a <- zarr_open(root, path="1d.contiguous.lz4.i2")
+
+    expect_equal(class(g)[1], "ZarrGroup")
+    expect_equal(class(a)[1], "ZarrArray")
+})
+
 test_that("Can open Zarr group and read a 1D 2-byte integer array with LZ4 compression", {
 
     root <- system.file("extdata", "fixtures", "v2", "data.zarr", package="pizzarr")