From 1a61bccf8b0956264b4a71067ab298c735e954df Mon Sep 17 00:00:00 2001 From: Rick Nitsche Date: Mon, 19 Apr 2021 12:53:19 -0700 Subject: [PATCH] zarr support --- caput/fileformats.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/caput/fileformats.py b/caput/fileformats.py index d8707d81..18e39d55 100644 --- 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(