zarr support

radiocosmology · Apr 19, 2021 · 1a61bcc · 1a61bcc
1 parent 0a0655a
commit 1a61bcc
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/caput/fileformats.py b/caput/fileformats.py
@@ -1,8 +1,10 @@
-import bitshuffle.h5
 import h5py
 import numcodecs
 import zarr
 
+# Copied from https://github.com/kiyo-masui/bitshuffle/blob/master/src/bshuf_h5filter.h
+BSHUF_H5FILTER = 32008
+BSHUF_H5_COMPRESS_LZ4 = 2
 
 class FileFormat:
     """Abstract base class for file formats supported by this module."""
@@ -68,9 +70,9 @@ def compression_kwargs(compression=None, compression_opts=None, compressor=None)
         if compression:
             if compression == "gzip":
                 return {"compressor": numcodecs.gzip.GZip(level=compression_opts)}
-            elif compression == bitshuffle.h5.H5FILTER:
+            elif compression == BSHUF_H5FILTER:
                 blocksize, c = compression_opts
-                if c == bitshuffle.h5.H5_COMPRESS_LZ4:
+                if c == BSHUF_H5_COMPRESS_LZ4:
                     cname = "lz4"
                 else:
                     raise ValueError(