Add the Zlib codec

juntyr · Aug 8, 2024 · b387190 · b387190
1 parent 7dcc47e
commit b387190
Show file tree

Hide file tree

Showing 6 changed files with 238 additions and 139 deletions.
diff --git a/codecs/bit-round/src/lib.rs b/codecs/bit-round/src/lib.rs
@@ -60,12 +60,18 @@ impl Codec for BitRoundCodec {
         encoded: AnyArrayView,
         mut decoded: AnyArrayViewMut,
     ) -> Result<(), Self::Error> {
-        fn shape_checked_assign<T: Copy>(encoded: &ArrayViewD<T>, decoded: &mut ArrayViewMutD<T>) -> Result<(), BitRoundCodecError> {
+        fn shape_checked_assign<T: Copy>(
+            encoded: &ArrayViewD<T>,
+            decoded: &mut ArrayViewMutD<T>,
+        ) -> Result<(), BitRoundCodecError> {
             #[allow(clippy::unit_arg)]
             if encoded.shape() == decoded.shape() {
                 Ok(decoded.assign(encoded))
             } else {
-                Err(BitRoundCodecError::MismatchedDecodeIntoShape { decoded: encoded.shape().to_vec(), provided: decoded.shape().to_vec() })
+                Err(BitRoundCodecError::MismatchedDecodeIntoShape {
+                    decoded: encoded.shape().to_vec(),
+                    provided: decoded.shape().to_vec(),
+                })
             }
         }
 

diff --git a/codecs/identity/src/lib.rs b/codecs/identity/src/lib.rs
@@ -45,12 +45,18 @@ impl Codec for IdentityCodec {
         encoded: AnyArrayView,
         mut decoded: AnyArrayViewMut,
     ) -> Result<(), Self::Error> {
-        fn shape_checked_assign<T: Copy>(encoded: &ArrayViewD<T>, decoded: &mut ArrayViewMutD<T>) -> Result<(), IdentityCodecError> {
+        fn shape_checked_assign<T: Copy>(
+            encoded: &ArrayViewD<T>,
+            decoded: &mut ArrayViewMutD<T>,
+        ) -> Result<(), IdentityCodecError> {
             #[allow(clippy::unit_arg)]
             if encoded.shape() == decoded.shape() {
                 Ok(decoded.assign(encoded))
             } else {
-                Err(IdentityCodecError::MismatchedDecodeIntoShape { decoded: encoded.shape().to_vec(), provided: decoded.shape().to_vec() })
+                Err(IdentityCodecError::MismatchedDecodeIntoShape {
+                    decoded: encoded.shape().to_vec(),
+                    provided: decoded.shape().to_vec(),
+                })
             }
         }
 

diff --git a/codecs/uniform-noise/src/lib.rs b/codecs/uniform-noise/src/lib.rs
@@ -73,12 +73,18 @@ impl Codec for UniformNoiseCodec {
         encoded: AnyArrayView,
         mut decoded: AnyArrayViewMut,
     ) -> Result<(), Self::Error> {
-        fn shape_checked_assign<T: Copy>(encoded: &ArrayViewD<T>, decoded: &mut ArrayViewMutD<T>) -> Result<(), UniformNoiseCodecError> {
+        fn shape_checked_assign<T: Copy>(
+            encoded: &ArrayViewD<T>,
+            decoded: &mut ArrayViewMutD<T>,
+        ) -> Result<(), UniformNoiseCodecError> {
             #[allow(clippy::unit_arg)]
             if encoded.shape() == decoded.shape() {
                 Ok(decoded.assign(encoded))
             } else {
-                Err(UniformNoiseCodecError::MismatchedDecodeIntoShape { decoded: encoded.shape().to_vec(), provided: decoded.shape().to_vec() })
+                Err(UniformNoiseCodecError::MismatchedDecodeIntoShape {
+                    decoded: encoded.shape().to_vec(),
+                    provided: decoded.shape().to_vec(),
+                })
             }
         }
 
@@ -89,14 +95,18 @@ impl Codec for UniformNoiseCodec {
             (AnyArrayView::F64(encoded), AnyArrayViewMut::F64(decoded)) => {
                 shape_checked_assign(encoded, decoded)
             }
-            (AnyArrayView::F32(_), decoded) => Err(UniformNoiseCodecError::MismatchedDecodeIntoDtype {
-                decoded: AnyArrayDType::F32,
-                provided: decoded.dtype(),
-            }),
-            (AnyArrayView::F64(_), decoded) => Err(UniformNoiseCodecError::MismatchedDecodeIntoDtype {
-                decoded: AnyArrayDType::F64,
-                provided: decoded.dtype(),
-            }),
+            (AnyArrayView::F32(_), decoded) => {
+                Err(UniformNoiseCodecError::MismatchedDecodeIntoDtype {
+                    decoded: AnyArrayDType::F32,
+                    provided: decoded.dtype(),
+                })
+            }
+            (AnyArrayView::F64(_), decoded) => {
+                Err(UniformNoiseCodecError::MismatchedDecodeIntoDtype {
+                    decoded: AnyArrayDType::F64,
+                    provided: decoded.dtype(),
+                })
+            }
             (encoded, _decoded) => Err(UniformNoiseCodecError::UnsupportedDtype(encoded.dtype())),
         }
     }

diff --git a/codecs/zlib/Cargo.toml b/codecs/zlib/Cargo.toml
@@ -17,7 +17,7 @@ keywords = ["zlib", "numcodecs", "compression", "encoding"]
 [dependencies]
 ndarray = { workspace = true }
 numcodecs = { workspace = true }
-miniz_oxide = { workspace = true, features = ["with-alloc"] }
+miniz_oxide = { workspace = true, features = ["std", "with-alloc"] }
 postcard = { workspace = true }
 serde = { workspace = true, features = ["std", "derive"] }
 serde_repr = { workspace = true }