support zlib compression

NeurodataWithoutBorders · Dec 3, 2024 · 90725f8 · 90725f8
1 parent a478191
commit 90725f8
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/lindi/LindiH5pyFile/writers/LindiH5pyGroupWriter.py b/lindi/LindiH5pyFile/writers/LindiH5pyGroupWriter.py
@@ -82,6 +82,14 @@ def create_dataset(
                 else:
                     raise Exception(f'Unexpected type for compression_opts: {type(compression_opts)}')
                 _zarr_compressor = numcodecs.GZip(level=level)
+            elif compression == 'zlib':
+                if compression_opts is None:
+                    level = 4
+                elif isinstance(compression_opts, int):
+                    level = compression_opts
+                else:
+                    raise Exception(f'Unexpected type for compression_opts: {type(compression_opts)}')
+                _zarr_compressor = numcodecs.Zlib(level=level)
             else:
                 raise Exception(f'Compression {compression} is not supported')
         elif compression is None:

diff --git a/lindi/conversion/h5_filters_to_codecs.py b/lindi/conversion/h5_filters_to_codecs.py
@@ -57,6 +57,8 @@ def h5_filters_to_codecs(h5obj: h5py.Dataset) -> Union[List[Codec], None]:
             filters.append(numcodecs.Zstd(level=properties[0]))
         elif str(filter_id) == "gzip":
             filters.append(numcodecs.Zlib(level=properties))
+        elif str(filter_id) == "zlib":
+            filters.append(numcodecs.Zlib(level=properties))
         elif str(filter_id) == "32004":
             raise RuntimeError(
                 f"{h5obj.name} uses lz4 compression - not supported"