add default compute_encoded_size implementations

Author: d-v-b

src/zarr/codecs/numcodecs/_codecs.py

@@ -421,6 +421,9 @@ def _encode(self, chunk_data: Buffer, prototype: BufferPrototype) -> Buffer:
     async def _encode_single(self, chunk_data: Buffer, chunk_spec: ArraySpec) -> Buffer:
         return await asyncio.to_thread(self._encode, chunk_data, chunk_spec.prototype)
 
+    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
+        raise NotImplementedError
+
 
 class _NumcodecsArrayArrayCodec(_NumcodecsCodec, ArrayArrayCodec):
     async def _decode_single(self, chunk_data: NDBuffer, chunk_spec: ArraySpec) -> NDBuffer:
@@ -433,6 +436,9 @@ async def _encode_single(self, chunk_data: NDBuffer, chunk_spec: ArraySpec) -> N
         out = await asyncio.to_thread(self._codec.encode, chunk_ndarray)
         return chunk_spec.prototype.nd_buffer.from_ndarray_like(out)
 
+    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
+        raise NotImplementedError
+
 
 class _NumcodecsArrayBytesCodec(_NumcodecsCodec, ArrayBytesCodec):
     def __init__(self, **codec_config: JSON) -> None:
@@ -448,6 +454,9 @@ async def _encode_single(self, chunk_data: NDBuffer, chunk_spec: ArraySpec) -> B
         out = await asyncio.to_thread(self._codec.encode, chunk_ndarray)
         return chunk_spec.prototype.buffer.from_bytes(out)
 
+    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
+        raise NotImplementedError
+
 
 # bytes-to-bytes codecs
 class Blosc(_NumcodecsBytesBytesCodec):
@@ -475,9 +484,6 @@ def to_json(self, zarr_format: ZarrFormat) -> LZ4JSON_V2 | LZ4JSON_V3:
         _warn_unstable_specification(self)
         return super().to_json(zarr_format)  # type: ignore[return-value]
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class Zstd(_NumcodecsBytesBytesCodec):
     codec_name = "numcodecs.zstd"
@@ -566,9 +572,6 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> Self:
             return type(self)(**{**self.codec_config, "elementsize": dtype.itemsize})
         return self  # pragma: no cover
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 # array-to-array codecs ("filters")
 class Delta(_NumcodecsArrayArrayCodec):
@@ -595,9 +598,6 @@ def resolve_metadata(self, chunk_spec: ArraySpec) -> ArraySpec:
             return replace(chunk_spec, dtype=dtype)
         return chunk_spec
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class BitRound(_NumcodecsArrayArrayCodec):
     codec_name = "numcodecs.bitround"
@@ -612,9 +612,6 @@ def to_json(self, zarr_format: ZarrFormat) -> BitRoundJSON_V2 | BitRoundJSON_V3:
         _warn_unstable_specification(self)
         return super().to_json(zarr_format)  # type: ignore[return-value]
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class FixedScaleOffset(_NumcodecsArrayArrayCodec):
     codec_name = "numcodecs.fixedscaleoffset"
@@ -641,9 +638,6 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> Self:
             return type(self)(**{**self.codec_config, "dtype": str(dtype)})
         return self
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class Quantize(_NumcodecsArrayArrayCodec):
     codec_name = "numcodecs.quantize"
@@ -664,9 +658,6 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> Self:
             return type(self)(**{**self.codec_config, "dtype": str(dtype)})
         return self
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class PackBits(_NumcodecsArrayArrayCodec):
     codec_name = "numcodecs.packbits"
@@ -696,9 +687,6 @@ def validate(self, *, dtype: ZDType[Any, Any], **_kwargs: Any) -> None:
         if _dtype != np.dtype("bool"):
             raise ValueError(f"Packbits filter requires bool dtype. Got {dtype}.")
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 class AsType(_NumcodecsArrayArrayCodec):
     codec_name = "numcodecs.astype"
@@ -724,9 +712,6 @@ def evolve_from_array_spec(self, array_spec: ArraySpec) -> AsType:
             return AsType(**{**self.codec_config, "decode_dtype": str(dtype)})  # pragma: no cover
         return self
 
-    def compute_encoded_size(self, input_byte_length: int, chunk_spec: ArraySpec) -> int:
-        raise NotImplementedError
-
 
 # bytes-to-bytes checksum codecs
 class _NumcodecsChecksumCodec(_NumcodecsBytesBytesCodec):