Extendable zarr arrays

docs/source/reference/python-client.md

@@ -165,6 +165,7 @@ Tiled currently includes two clients for each structure family:
    tiled.client.array.DaskArrayClient.export
    tiled.client.array.DaskArrayClient.write
    tiled.client.array.DaskArrayClient.write_block
+   tiled.client.array.DaskArrayClient.patch
 ```
 
 ```{eval-rst}

docs/source/tutorials/writing.md

@@ -60,6 +60,19 @@ Write array and tabular data.
 >>> client.write_array(numpy.array([4, 5, 6]), metadata={"color": "blue", "barcode": 11})
 <ArrayClient shape=(3,) chunks=((3,),) dtype=int64>
 
+# Create an array and grow it by one
+>>> new_array = client.write_array([1, 2, 3])
+>>> new_array
+<ArrayClient shape=(3,) chunks=((3,),) dtype=int64>
+
+# Extend the array. This array has only one dimension, here we extend by one
+# along that dimension.
+>>> new_array.patch([4], offset=(3,), extend=True)
+>>> new_array
+<ArrayClient shape=(4,) chunks=((3, 1),) dtype=int64>
+>>> new_array.read()
+array([1, 2, 3, 4])
+
 # Write a table (DataFrame).
 >>> import pandas
 >>> client.write_dataframe(pandas.DataFrame({"x": [1, 2, 3], "y": [4, 5, 6]}), metadata={"color": "green", "barcode": 12})

tiled/_tests/test_protocols.py

@@ -18,13 +18,13 @@
     SparseAdapter,
     TableAdapter,
 )
-from tiled.adapters.type_alliases import JSON, Filters, NDSlice, Scopes
 from tiled.server.schemas import Principal, PrincipalType
 from tiled.structures.array import ArrayStructure, BuiltinDtype
 from tiled.structures.awkward import AwkwardStructure
 from tiled.structures.core import Spec, StructureFamily
 from tiled.structures.sparse import COOStructure
 from tiled.structures.table import TableStructure
+from tiled.type_aliases import JSON, Filters, NDSlice, Scopes
 
 
 class CustomArrayAdapter:

tiled/_tests/test_writing.py

@@ -121,6 +121,55 @@ def test_write_array_chunked(tree):
         assert result.specs == specs
 
 
+def test_extend_array(tree):
+    "Extend an array with additional data, expanding its shape."
+    with Context.from_app(
+        build_app(tree, validation_registry=validation_registry)
+    ) as context:
+        client = from_context(context)
+
+        a = numpy.ones((3, 2, 2))
+        new_data = numpy.ones((1, 2, 2)) * 2
+        full_array = numpy.concatenate((a, new_data), axis=0)
+
+        # Upload a (3, 2, 2) array.
+        ac = client.write_array(a)
+        assert ac.shape == a.shape
+
+        # Patching data into a region beyond the current extent of the array
+        # raises a ValueError (catching a 409 from the server).
+        with pytest.raises(ValueError):
+            ac.patch(new_data, offset=(3,))
+        # With extend=True, the array is expanded.
+        ac.patch(new_data, offset=(3,), extend=True)
+        # The local cache of the structure is updated.
+        assert ac.shape == full_array.shape
+        actual = ac.read()
+        # The array has the expected shape and data.
+        assert actual.shape == full_array.shape
+        numpy.testing.assert_equal(actual, full_array)
+
+        # Overwrite data (do not extend).
+        revised_data = numpy.ones((1, 2, 2)) * 3
+        revised_array = full_array.copy()
+        revised_array[3, :, :] = 3
+        ac.patch(revised_data, offset=(3,))
+        numpy.testing.assert_equal(ac.read(), revised_array)
+
+        # Extend out of order.
+        ones = numpy.ones((1, 2, 2))
+        ac.patch(ones * 7, offset=(7,), extend=True)
+        ac.patch(ones * 5, offset=(5,), extend=True)
+        ac.patch(ones * 6, offset=(6,), extend=True)
+        numpy.testing.assert_equal(ac[5:6], ones * 5)
+        numpy.testing.assert_equal(ac[6:7], ones * 6)
+        numpy.testing.assert_equal(ac[7:8], ones * 7)
+
+        # Offset given as an int is acceptable.
+        ac.patch(ones * 8, offset=8, extend=True)
+        numpy.testing.assert_equal(ac[8:9], ones * 8)
+
+
 def test_write_dataframe_full(tree):
     with Context.from_app(
         build_app(tree, validation_registry=validation_registry)

tiled/adapters/array.py

@@ -5,8 +5,8 @@
 
 from ..structures.array import ArrayStructure
 from ..structures.core import Spec, StructureFamily
+from ..type_aliases import JSON, NDSlice
 from .protocols import AccessPolicy
-from .type_alliases import JSON, NDSlice
 
 
 class ArrayAdapter:

tiled/adapters/arrow.py

@@ -9,10 +9,10 @@
 from ..structures.core import Spec, StructureFamily
 from ..structures.data_source import Asset, DataSource, Management
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from ..utils import ensure_uri, path_from_uri
 from .array import ArrayAdapter
 from .protocols import AccessPolicy
-from .type_alliases import JSON
 
 
 class ArrowAdapter:

tiled/adapters/awkward.py

@@ -6,9 +6,9 @@
 
 from ..structures.awkward import AwkwardStructure
 from ..structures.core import Spec, StructureFamily
+from ..type_aliases import JSON
 from .awkward_directory_container import DirectoryContainer
 from .protocols import AccessPolicy
-from .type_alliases import JSON
 
 
 class AwkwardAdapter:

tiled/adapters/awkward_buffers.py

@@ -9,11 +9,11 @@
 from ..server.schemas import Asset
 from ..structures.awkward import AwkwardStructure
 from ..structures.core import Spec, StructureFamily
+from ..type_aliases import JSON
 from ..utils import path_from_uri
 from .awkward import AwkwardAdapter
 from .awkward_directory_container import DirectoryContainer
 from .protocols import AccessPolicy
-from .type_alliases import JSON
 
 
 class AwkwardBuffersAdapter(AwkwardAdapter):

tiled/adapters/csv.py

@@ -7,12 +7,12 @@
 from ..structures.core import Spec, StructureFamily
 from ..structures.data_source import Asset, DataSource, Management
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from ..utils import ensure_uri, path_from_uri
 from .array import ArrayAdapter
 from .dataframe import DataFrameAdapter
 from .protocols import AccessPolicy
 from .table import TableAdapter
-from .type_alliases import JSON
 
 
 def read_csv(

tiled/adapters/hdf5.py

@@ -14,11 +14,11 @@
 from ..structures.array import ArrayStructure
 from ..structures.core import Spec, StructureFamily
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from ..utils import node_repr, path_from_uri
 from .array import ArrayAdapter
 from .protocols import AccessPolicy
 from .resource_cache import with_resource_cache
-from .type_alliases import JSON
 
 SWMR_DEFAULT = bool(int(os.getenv("TILED_HDF5_SWMR_DEFAULT", "0")))
 INLINED_DEPTH = int(os.getenv("TILED_HDF5_INLINED_CONTENTS_MAX_DEPTH", "7"))

tiled/adapters/jpeg.py

@@ -7,11 +7,11 @@
 
 from ..structures.array import ArrayStructure, BuiltinDtype
 from ..structures.core import Spec, StructureFamily
+from ..type_aliases import JSON, NDSlice
 from ..utils import path_from_uri
 from .protocols import AccessPolicy
 from .resource_cache import with_resource_cache
 from .sequence import FileSequenceAdapter
-from .type_alliases import JSON, NDSlice
 
 
 class JPEGAdapter:

tiled/adapters/mapping.py

@@ -38,9 +38,9 @@
 from ..server.schemas import SortingItem
 from ..structures.core import Spec, StructureFamily
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from ..utils import UNCHANGED, Sentinel
 from .protocols import AccessPolicy, AnyAdapter
-from .type_alliases import JSON
 from .utils import IndexersMixin
 
 if sys.version_info < (3, 9):

tiled/adapters/parquet.py

@@ -7,10 +7,10 @@
 from ..server.schemas import Asset
 from ..structures.core import Spec, StructureFamily
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from ..utils import path_from_uri
 from .dataframe import DataFrameAdapter
 from .protocols import AccessPolicy
-from .type_alliases import JSON
 
 
 class ParquetDatasetAdapter:

tiled/adapters/protocols.py

@@ -14,8 +14,8 @@
 from ..structures.core import Spec, StructureFamily
 from ..structures.sparse import SparseStructure
 from ..structures.table import TableStructure
+from ..type_aliases import JSON, Filters, NDSlice, Scopes
 from .awkward_directory_container import DirectoryContainer
-from .type_alliases import JSON, Filters, NDSlice, Scopes
 
 
 class BaseAdapter(Protocol):

tiled/adapters/sequence.py

@@ -11,9 +11,9 @@
 
 from ..structures.array import ArrayStructure, BuiltinDtype
 from ..structures.core import Spec
+from ..type_aliases import JSON, NDSlice
 from ..utils import path_from_uri
 from .protocols import AccessPolicy
-from .type_alliases import JSON, NDSlice
 
 
 def force_reshape(arr: np.array, desired_shape: Tuple[int, ...]) -> np.array:

tiled/adapters/sparse.py

@@ -8,9 +8,9 @@
 
 from ..structures.core import Spec, StructureFamily
 from ..structures.sparse import COOStructure
+from ..type_aliases import JSON, NDSlice
 from .array import slice_and_shape_from_block_and_chunks
 from .protocols import AccessPolicy
-from .type_alliases import JSON, NDSlice
 
 
 class COOAdapter:

tiled/adapters/sparse_blocks_parquet.py

@@ -12,9 +12,9 @@
 from ..server.schemas import Asset
 from ..structures.core import Spec, StructureFamily
 from ..structures.sparse import COOStructure
+from ..type_aliases import JSON, NDSlice
 from ..utils import path_from_uri
 from .protocols import AccessPolicy
-from .type_alliases import JSON, NDSlice
 
 
 def load_block(uri: str) -> Tuple[List[int], Tuple[NDArray[Any], Any]]:
@@ -113,18 +113,13 @@ def write_block(
         self,
         data: Union[dask.dataframe.DataFrame, pandas.DataFrame],
         block: Tuple[int, ...],
+        slice: NDSlice = ...,
     ) -> None:
-        """
-
-        Parameters
-        ----------
-        data :
-        block :
-
-        Returns
-        -------
-
-        """
+        if slice != ...:
+            raise NotImplementedError(
+                "Writing into a slice of a sparse block is not yet supported."
+            )
+        "Write into a block of the array."
         uri = self.blocks[block]
         data.to_parquet(path_from_uri(uri))
 

tiled/adapters/table.py

@@ -6,9 +6,9 @@
 
 from ..structures.core import Spec, StructureFamily
 from ..structures.table import TableStructure
+from ..type_aliases import JSON
 from .array import ArrayAdapter
 from .protocols import AccessPolicy
-from .type_alliases import JSON
 
 
 class TableAdapter:

tiled/adapters/tiff.py

@@ -6,11 +6,11 @@
 
 from ..structures.array import ArrayStructure, BuiltinDtype
 from ..structures.core import Spec, StructureFamily
+from ..type_aliases import JSON, NDSlice
 from ..utils import path_from_uri
 from .protocols import AccessPolicy
 from .resource_cache import with_resource_cache
 from .sequence import FileSequenceAdapter
-from .type_alliases import JSON, NDSlice
 
 
 class TiffAdapter: