66from zarr .codecs .sharding import ShardingCodec , ShardingCodecIndexLocation
77from zarr .codecs .transpose import TransposeCodec
88from zarr .codecs .zstd import ZstdCodec
9- import zarrita
109import numcodecs
1110
1211def parse_codecs_zarr_python (codec_string : str , param_string : str , zarr_version : int = 3 ):
@@ -51,33 +50,4 @@ def parse_codecs_zarr_python(codec_string: str, param_string: str, zarr_version:
5150 else :
5251 raise ValueError (f"Invalid codec: { codec_string } , zarr_version: { zarr_version } " )
5352
54- return compressor , serializer , filters
55-
56- def parse_codecs_zarrita (codec_string : str , param_string : str ):
57- codec = []
58- if codec_string == "blosc" :
59- cname , shuffle , clevel = param_string .split ("_" )
60- codec = [zarrita .codecs .bytes_codec (),
61- zarrita .codecs .blosc_codec (typesize = 4 , cname = cname , shuffle = shuffle , clevel = int (clevel ))]
62- elif codec_string == "gzip" :
63- codec = [zarrita .codecs .bytes_codec (), zarrita .codecs .gzip_codec (level = int (param_string ))]
64- elif codec_string == "zstd" :
65- level , checksum = param_string .split ("_" )
66- codec = [zarrita .codecs .bytes_codec (), zarrita .codecs .zstd_codec (checksum = checksum == 'true' , level = int (level ))]
67- elif codec_string == "bytes" :
68- codec = [zarrita .codecs .bytes_codec (endian = param_string .lower ())]
69- elif codec_string == "transpose" :
70- codec = [zarrita .codecs .transpose_codec ((1 , 0 , 2 )), zarrita .codecs .bytes_codec ()]
71- elif codec_string == "sharding" :
72- codec = zarrita .codecs .sharding_codec (chunk_shape = (2 , 2 , 4 ), codecs = [zarrita .codecs .bytes_codec ("little" )],
73- index_location = zarrita .metadata .ShardingCodecIndexLocation .start if param_string == "start"
74- else zarrita .metadata .ShardingCodecIndexLocation .end ),
75- elif codec_string == "sharding_nested" :
76- codec = zarrita .codecs .sharding_codec (chunk_shape = (2 , 2 , 4 ),
77- codecs = [zarrita .codecs .sharding_codec (chunk_shape = (2 , 1 , 2 ), codecs = [
78- zarrita .codecs .bytes_codec ("little" )])]),
79- elif codec_string == "crc32c" :
80- codec = [zarrita .codecs .bytes_codec (), zarrita .codecs .crc32c_codec ()]
81- else :
82- raise ValueError (f"Invalid codec: { codec_string } " )
83- return codec
53+ return compressor , serializer , filters
0 commit comments