|
6 | 6 |
|
7 | 7 | from io import BytesIO |
8 | 8 |
|
| 9 | +import leb128 |
9 | 10 | import numcodecs.compat |
10 | 11 | import numcodecs.registry |
11 | 12 | import numpy as np |
12 | | -import varint |
13 | 13 | from numcodecs.abc import Codec |
14 | 14 |
|
15 | 15 | from .typing import S, T, U |
@@ -60,14 +60,14 @@ def encode( |
60 | 60 | # message: dtype shape [padding] table indices |
61 | 61 | message = [] |
62 | 62 |
|
63 | | - message.append(varint.encode(len(dtype.str))) |
| 63 | + message.append(leb128.u.encode(len(dtype.str))) |
64 | 64 | message.append(dtype.str.encode("ascii")) |
65 | 65 |
|
66 | | - message.append(varint.encode(len(shape))) |
| 66 | + message.append(leb128.u.encode(len(shape))) |
67 | 67 | for s in shape: |
68 | | - message.append(varint.encode(s)) |
| 68 | + message.append(leb128.u.encode(s)) |
69 | 69 |
|
70 | | - message.append(varint.encode(unique.size)) |
| 70 | + message.append(leb128.u.encode(unique.size)) |
71 | 71 |
|
72 | 72 | # select the smallest output data type that can encode the indices |
73 | 73 | utype: np.dtype[np.unsignedinteger] |
@@ -132,13 +132,13 @@ def decode( |
132 | 132 | b = numcodecs.compat.ensure_bytes(buf) |
133 | 133 | b_io = BytesIO(b) |
134 | 134 |
|
135 | | - dtype = np.dtype(b_io.read(varint.decode_stream(b_io)).decode("ascii")) |
| 135 | + dtype = np.dtype(b_io.read(leb128.u.decode_reader(b_io)).decode("ascii")) |
136 | 136 |
|
137 | 137 | shape = tuple( |
138 | | - varint.decode_stream(b_io) for _ in range(varint.decode_stream(b_io)) |
| 138 | + leb128.u.decode_reader(b_io) for _ in range(leb128.u.decode_reader(b_io)) |
139 | 139 | ) |
140 | 140 |
|
141 | | - table_len = varint.decode_stream(b_io) |
| 141 | + table_len = leb128.u.decode_reader(b_io) |
142 | 142 |
|
143 | 143 | # select the smallest output data type that can encode the indices |
144 | 144 | utype: np.dtype[np.unsignedinteger] |
|
0 commit comments