numcodecs-mask 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,6 @@
1
+ version: 2
2
+ updates:
3
+ - package-ecosystem: "github-actions"
4
+ directory: "/"
5
+ schedule:
6
+ interval: "weekly"
@@ -0,0 +1,66 @@
1
+ name: CI
2
+
3
+ on:
4
+ push:
5
+ branches:
6
+ - main
7
+ pull_request:
8
+ branches:
9
+ - '*'
10
+
11
+ jobs:
12
+ check:
13
+ name: Check
14
+ runs-on: ubuntu-latest
15
+ steps:
16
+ - name: Checkout the Repository
17
+ uses: actions/checkout@v6
18
+
19
+ - name: Check with ruff
20
+ uses: astral-sh/ruff-action@v3
21
+ with:
22
+ args: check
23
+
24
+ fmt:
25
+ name: Formatting
26
+ runs-on: ubuntu-latest
27
+ steps:
28
+ - name: Checkout the Repository
29
+ uses: actions/checkout@v6
30
+
31
+ - name: Check with ruff
32
+ uses: astral-sh/ruff-action@v3
33
+ with:
34
+ args: "format --diff --check"
35
+
36
+ test:
37
+ name: Test
38
+ runs-on: ubuntu-latest
39
+ steps:
40
+ - name: Checkout the Repository
41
+ uses: actions/checkout@v6
42
+
43
+ - name: Install uv
44
+ uses: astral-sh/setup-uv@v7
45
+
46
+ - name: Setup the Python venv
47
+ run: uv sync --all-extras --dev
48
+
49
+ - name: Run tests
50
+ run: uv run pytest -v
51
+
52
+ mypy:
53
+ name: Typecheck
54
+ runs-on: ubuntu-latest
55
+ steps:
56
+ - name: Checkout the Repository
57
+ uses: actions/checkout@v6
58
+
59
+ - name: Install uv
60
+ uses: astral-sh/setup-uv@v7
61
+
62
+ - name: Setup the Python venv
63
+ run: uv sync --all-extras --dev
64
+
65
+ - name: Typecheck numcodecs-mask
66
+ run: uv run mypy -p numcodecs_mask
@@ -0,0 +1,27 @@
1
+ name: Publish
2
+
3
+ on:
4
+ release:
5
+ branch: main
6
+ types: [released]
7
+
8
+ jobs:
9
+ publish:
10
+ name: Publish to PyPi
11
+ runs-on: ubuntu-latest
12
+ environment: publish
13
+ permissions:
14
+ contents: read
15
+ id-token: write
16
+ steps:
17
+ - name: Checkout the Repository
18
+ uses: actions/checkout@v6
19
+
20
+ - name: Install uv
21
+ uses: astral-sh/setup-uv@v7
22
+
23
+ - name: Build the package
24
+ run: uv build
25
+
26
+ - name: Publish the package to PyPi
27
+ run: uv publish
@@ -0,0 +1,11 @@
1
+ /.venv
2
+ /.mypy_cache
3
+ /.pytest_cache
4
+ /.ruff_cache
5
+ /uv.lock
6
+
7
+ /build
8
+ /site
9
+
10
+ **/__pycache__
11
+ **/*.egg-info
@@ -0,0 +1 @@
1
+ 3.10
@@ -0,0 +1,18 @@
1
+ version: 2
2
+
3
+ build:
4
+ os: ubuntu-24.04
5
+ tools:
6
+ python: "3.12"
7
+ jobs:
8
+ pre_create_environment:
9
+ - asdf plugin add uv
10
+ - asdf install uv latest
11
+ - asdf global uv latest
12
+ create_environment:
13
+ - uv venv "${READTHEDOCS_VIRTUALENV_PATH}"
14
+ install:
15
+ - UV_PROJECT_ENVIRONMENT="${READTHEDOCS_VIRTUALENV_PATH}" uv sync --group docs --link-mode=copy
16
+ build:
17
+ html:
18
+ - python -m properdocs build --clean --site-dir "${READTHEDOCS_OUTPUT}/html"
@@ -0,0 +1,375 @@
1
+ Copyright (c) 2026, Juniper Tyree
2
+
3
+ Mozilla Public License Version 2.0
4
+ ==================================
5
+
6
+ 1. Definitions
7
+ --------------
8
+
9
+ 1.1. "Contributor"
10
+ means each individual or legal entity that creates, contributes to
11
+ the creation of, or owns Covered Software.
12
+
13
+ 1.2. "Contributor Version"
14
+ means the combination of the Contributions of others (if any) used
15
+ by a Contributor and that particular Contributor's Contribution.
16
+
17
+ 1.3. "Contribution"
18
+ means Covered Software of a particular Contributor.
19
+
20
+ 1.4. "Covered Software"
21
+ means Source Code Form to which the initial Contributor has attached
22
+ the notice in Exhibit A, the Executable Form of such Source Code
23
+ Form, and Modifications of such Source Code Form, in each case
24
+ including portions thereof.
25
+
26
+ 1.5. "Incompatible With Secondary Licenses"
27
+ means
28
+
29
+ (a) that the initial Contributor has attached the notice described
30
+ in Exhibit B to the Covered Software; or
31
+
32
+ (b) that the Covered Software was made available under the terms of
33
+ version 1.1 or earlier of the License, but not also under the
34
+ terms of a Secondary License.
35
+
36
+ 1.6. "Executable Form"
37
+ means any form of the work other than Source Code Form.
38
+
39
+ 1.7. "Larger Work"
40
+ means a work that combines Covered Software with other material, in
41
+ a separate file or files, that is not Covered Software.
42
+
43
+ 1.8. "License"
44
+ means this document.
45
+
46
+ 1.9. "Licensable"
47
+ means having the right to grant, to the maximum extent possible,
48
+ whether at the time of the initial grant or subsequently, any and
49
+ all of the rights conveyed by this License.
50
+
51
+ 1.10. "Modifications"
52
+ means any of the following:
53
+
54
+ (a) any file in Source Code Form that results from an addition to,
55
+ deletion from, or modification of the contents of Covered
56
+ Software; or
57
+
58
+ (b) any new file in Source Code Form that contains any Covered
59
+ Software.
60
+
61
+ 1.11. "Patent Claims" of a Contributor
62
+ means any patent claim(s), including without limitation, method,
63
+ process, and apparatus claims, in any patent Licensable by such
64
+ Contributor that would be infringed, but for the grant of the
65
+ License, by the making, using, selling, offering for sale, having
66
+ made, import, or transfer of either its Contributions or its
67
+ Contributor Version.
68
+
69
+ 1.12. "Secondary License"
70
+ means either the GNU General Public License, Version 2.0, the GNU
71
+ Lesser General Public License, Version 2.1, the GNU Affero General
72
+ Public License, Version 3.0, or any later versions of those
73
+ licenses.
74
+
75
+ 1.13. "Source Code Form"
76
+ means the form of the work preferred for making modifications.
77
+
78
+ 1.14. "You" (or "Your")
79
+ means an individual or a legal entity exercising rights under this
80
+ License. For legal entities, "You" includes any entity that
81
+ controls, is controlled by, or is under common control with You. For
82
+ purposes of this definition, "control" means (a) the power, direct
83
+ or indirect, to cause the direction or management of such entity,
84
+ whether by contract or otherwise, or (b) ownership of more than
85
+ fifty percent (50%) of the outstanding shares or beneficial
86
+ ownership of such entity.
87
+
88
+ 2. License Grants and Conditions
89
+ --------------------------------
90
+
91
+ 2.1. Grants
92
+
93
+ Each Contributor hereby grants You a world-wide, royalty-free,
94
+ non-exclusive license:
95
+
96
+ (a) under intellectual property rights (other than patent or trademark)
97
+ Licensable by such Contributor to use, reproduce, make available,
98
+ modify, display, perform, distribute, and otherwise exploit its
99
+ Contributions, either on an unmodified basis, with Modifications, or
100
+ as part of a Larger Work; and
101
+
102
+ (b) under Patent Claims of such Contributor to make, use, sell, offer
103
+ for sale, have made, import, and otherwise transfer either its
104
+ Contributions or its Contributor Version.
105
+
106
+ 2.2. Effective Date
107
+
108
+ The licenses granted in Section 2.1 with respect to any Contribution
109
+ become effective for each Contribution on the date the Contributor first
110
+ distributes such Contribution.
111
+
112
+ 2.3. Limitations on Grant Scope
113
+
114
+ The licenses granted in this Section 2 are the only rights granted under
115
+ this License. No additional rights or licenses will be implied from the
116
+ distribution or licensing of Covered Software under this License.
117
+ Notwithstanding Section 2.1(b) above, no patent license is granted by a
118
+ Contributor:
119
+
120
+ (a) for any code that a Contributor has removed from Covered Software;
121
+ or
122
+
123
+ (b) for infringements caused by: (i) Your and any other third party's
124
+ modifications of Covered Software, or (ii) the combination of its
125
+ Contributions with other software (except as part of its Contributor
126
+ Version); or
127
+
128
+ (c) under Patent Claims infringed by Covered Software in the absence of
129
+ its Contributions.
130
+
131
+ This License does not grant any rights in the trademarks, service marks,
132
+ or logos of any Contributor (except as may be necessary to comply with
133
+ the notice requirements in Section 3.4).
134
+
135
+ 2.4. Subsequent Licenses
136
+
137
+ No Contributor makes additional grants as a result of Your choice to
138
+ distribute the Covered Software under a subsequent version of this
139
+ License (see Section 10.2) or under the terms of a Secondary License (if
140
+ permitted under the terms of Section 3.3).
141
+
142
+ 2.5. Representation
143
+
144
+ Each Contributor represents that the Contributor believes its
145
+ Contributions are its original creation(s) or it has sufficient rights
146
+ to grant the rights to its Contributions conveyed by this License.
147
+
148
+ 2.6. Fair Use
149
+
150
+ This License is not intended to limit any rights You have under
151
+ applicable copyright doctrines of fair use, fair dealing, or other
152
+ equivalents.
153
+
154
+ 2.7. Conditions
155
+
156
+ Sections 3.1, 3.2, 3.3, and 3.4 are conditions of the licenses granted
157
+ in Section 2.1.
158
+
159
+ 3. Responsibilities
160
+ -------------------
161
+
162
+ 3.1. Distribution of Source Form
163
+
164
+ All distribution of Covered Software in Source Code Form, including any
165
+ Modifications that You create or to which You contribute, must be under
166
+ the terms of this License. You must inform recipients that the Source
167
+ Code Form of the Covered Software is governed by the terms of this
168
+ License, and how they can obtain a copy of this License. You may not
169
+ attempt to alter or restrict the recipients' rights in the Source Code
170
+ Form.
171
+
172
+ 3.2. Distribution of Executable Form
173
+
174
+ If You distribute Covered Software in Executable Form then:
175
+
176
+ (a) such Covered Software must also be made available in Source Code
177
+ Form, as described in Section 3.1, and You must inform recipients of
178
+ the Executable Form how they can obtain a copy of such Source Code
179
+ Form by reasonable means in a timely manner, at a charge no more
180
+ than the cost of distribution to the recipient; and
181
+
182
+ (b) You may distribute such Executable Form under the terms of this
183
+ License, or sublicense it under different terms, provided that the
184
+ license for the Executable Form does not attempt to limit or alter
185
+ the recipients' rights in the Source Code Form under this License.
186
+
187
+ 3.3. Distribution of a Larger Work
188
+
189
+ You may create and distribute a Larger Work under terms of Your choice,
190
+ provided that You also comply with the requirements of this License for
191
+ the Covered Software. If the Larger Work is a combination of Covered
192
+ Software with a work governed by one or more Secondary Licenses, and the
193
+ Covered Software is not Incompatible With Secondary Licenses, this
194
+ License permits You to additionally distribute such Covered Software
195
+ under the terms of such Secondary License(s), so that the recipient of
196
+ the Larger Work may, at their option, further distribute the Covered
197
+ Software under the terms of either this License or such Secondary
198
+ License(s).
199
+
200
+ 3.4. Notices
201
+
202
+ You may not remove or alter the substance of any license notices
203
+ (including copyright notices, patent notices, disclaimers of warranty,
204
+ or limitations of liability) contained within the Source Code Form of
205
+ the Covered Software, except that You may alter any license notices to
206
+ the extent required to remedy known factual inaccuracies.
207
+
208
+ 3.5. Application of Additional Terms
209
+
210
+ You may choose to offer, and to charge a fee for, warranty, support,
211
+ indemnity or liability obligations to one or more recipients of Covered
212
+ Software. However, You may do so only on Your own behalf, and not on
213
+ behalf of any Contributor. You must make it absolutely clear that any
214
+ such warranty, support, indemnity, or liability obligation is offered by
215
+ You alone, and You hereby agree to indemnify every Contributor for any
216
+ liability incurred by such Contributor as a result of warranty, support,
217
+ indemnity or liability terms You offer. You may include additional
218
+ disclaimers of warranty and limitations of liability specific to any
219
+ jurisdiction.
220
+
221
+ 4. Inability to Comply Due to Statute or Regulation
222
+ ---------------------------------------------------
223
+
224
+ If it is impossible for You to comply with any of the terms of this
225
+ License with respect to some or all of the Covered Software due to
226
+ statute, judicial order, or regulation then You must: (a) comply with
227
+ the terms of this License to the maximum extent possible; and (b)
228
+ describe the limitations and the code they affect. Such description must
229
+ be placed in a text file included with all distributions of the Covered
230
+ Software under this License. Except to the extent prohibited by statute
231
+ or regulation, such description must be sufficiently detailed for a
232
+ recipient of ordinary skill to be able to understand it.
233
+
234
+ 5. Termination
235
+ --------------
236
+
237
+ 5.1. The rights granted under this License will terminate automatically
238
+ if You fail to comply with any of its terms. However, if You become
239
+ compliant, then the rights granted under this License from a particular
240
+ Contributor are reinstated (a) provisionally, unless and until such
241
+ Contributor explicitly and finally terminates Your grants, and (b) on an
242
+ ongoing basis, if such Contributor fails to notify You of the
243
+ non-compliance by some reasonable means prior to 60 days after You have
244
+ come back into compliance. Moreover, Your grants from a particular
245
+ Contributor are reinstated on an ongoing basis if such Contributor
246
+ notifies You of the non-compliance by some reasonable means, this is the
247
+ first time You have received notice of non-compliance with this License
248
+ from such Contributor, and You become compliant prior to 30 days after
249
+ Your receipt of the notice.
250
+
251
+ 5.2. If You initiate litigation against any entity by asserting a patent
252
+ infringement claim (excluding declaratory judgment actions,
253
+ counter-claims, and cross-claims) alleging that a Contributor Version
254
+ directly or indirectly infringes any patent, then the rights granted to
255
+ You by any and all Contributors for the Covered Software under Section
256
+ 2.1 of this License shall terminate.
257
+
258
+ 5.3. In the event of termination under Sections 5.1 or 5.2 above, all
259
+ end user license agreements (excluding distributors and resellers) which
260
+ have been validly granted by You or Your distributors under this License
261
+ prior to termination shall survive termination.
262
+
263
+ ************************************************************************
264
+ * *
265
+ * 6. Disclaimer of Warranty *
266
+ * ------------------------- *
267
+ * *
268
+ * Covered Software is provided under this License on an "as is" *
269
+ * basis, without warranty of any kind, either expressed, implied, or *
270
+ * statutory, including, without limitation, warranties that the *
271
+ * Covered Software is free of defects, merchantable, fit for a *
272
+ * particular purpose or non-infringing. The entire risk as to the *
273
+ * quality and performance of the Covered Software is with You. *
274
+ * Should any Covered Software prove defective in any respect, You *
275
+ * (not any Contributor) assume the cost of any necessary servicing, *
276
+ * repair, or correction. This disclaimer of warranty constitutes an *
277
+ * essential part of this License. No use of any Covered Software is *
278
+ * authorized under this License except under this disclaimer. *
279
+ * *
280
+ ************************************************************************
281
+
282
+ ************************************************************************
283
+ * *
284
+ * 7. Limitation of Liability *
285
+ * -------------------------- *
286
+ * *
287
+ * Under no circumstances and under no legal theory, whether tort *
288
+ * (including negligence), contract, or otherwise, shall any *
289
+ * Contributor, or anyone who distributes Covered Software as *
290
+ * permitted above, be liable to You for any direct, indirect, *
291
+ * special, incidental, or consequential damages of any character *
292
+ * including, without limitation, damages for lost profits, loss of *
293
+ * goodwill, work stoppage, computer failure or malfunction, or any *
294
+ * and all other commercial damages or losses, even if such party *
295
+ * shall have been informed of the possibility of such damages. This *
296
+ * limitation of liability shall not apply to liability for death or *
297
+ * personal injury resulting from such party's negligence to the *
298
+ * extent applicable law prohibits such limitation. Some *
299
+ * jurisdictions do not allow the exclusion or limitation of *
300
+ * incidental or consequential damages, so this exclusion and *
301
+ * limitation may not apply to You. *
302
+ * *
303
+ ************************************************************************
304
+
305
+ 8. Litigation
306
+ -------------
307
+
308
+ Any litigation relating to this License may be brought only in the
309
+ courts of a jurisdiction where the defendant maintains its principal
310
+ place of business and such litigation shall be governed by laws of that
311
+ jurisdiction, without reference to its conflict-of-law provisions.
312
+ Nothing in this Section shall prevent a party's ability to bring
313
+ cross-claims or counter-claims.
314
+
315
+ 9. Miscellaneous
316
+ ----------------
317
+
318
+ This License represents the complete agreement concerning the subject
319
+ matter hereof. If any provision of this License is held to be
320
+ unenforceable, such provision shall be reformed only to the extent
321
+ necessary to make it enforceable. Any law or regulation which provides
322
+ that the language of a contract shall be construed against the drafter
323
+ shall not be used to construe this License against a Contributor.
324
+
325
+ 10. Versions of the License
326
+ ---------------------------
327
+
328
+ 10.1. New Versions
329
+
330
+ Mozilla Foundation is the license steward. Except as provided in Section
331
+ 10.3, no one other than the license steward has the right to modify or
332
+ publish new versions of this License. Each version will be given a
333
+ distinguishing version number.
334
+
335
+ 10.2. Effect of New Versions
336
+
337
+ You may distribute the Covered Software under the terms of the version
338
+ of the License under which You originally received the Covered Software,
339
+ or under the terms of any subsequent version published by the license
340
+ steward.
341
+
342
+ 10.3. Modified Versions
343
+
344
+ If you create software not governed by this License, and you want to
345
+ create a new license for such software, you may create and use a
346
+ modified version of this License if you rename the license and remove
347
+ any references to the name of the license steward (except to note that
348
+ such modified license differs from this License).
349
+
350
+ 10.4. Distributing Source Code Form that is Incompatible With Secondary
351
+ Licenses
352
+
353
+ If You choose to distribute Source Code Form that is Incompatible With
354
+ Secondary Licenses under the terms of this version of the License, the
355
+ notice described in Exhibit B of this License must be attached.
356
+
357
+ Exhibit A - Source Code Form License Notice
358
+ -------------------------------------------
359
+
360
+ This Source Code Form is subject to the terms of the Mozilla Public
361
+ License, v. 2.0. If a copy of the MPL was not distributed with this
362
+ file, You can obtain one at https://mozilla.org/MPL/2.0/.
363
+
364
+ If it is not possible or desirable to put the notice in a particular
365
+ file, then You may include the notice in a location (such as a LICENSE
366
+ file in a relevant directory) where a recipient would be likely to look
367
+ for such a notice.
368
+
369
+ You may add additional accurate notices of copyright ownership.
370
+
371
+ Exhibit B - "Incompatible With Secondary Licenses" Notice
372
+ ---------------------------------------------------------
373
+
374
+ This Source Code Form is "Incompatible With Secondary Licenses", as
375
+ defined by the Mozilla Public License, v. 2.0.
@@ -0,0 +1,36 @@
1
+ Metadata-Version: 2.4
2
+ Name: numcodecs-mask
3
+ Version: 0.1.0
4
+ Summary: Masking codecs for the `numcodecs` buffer compression API
5
+ License-Expression: MPL-2.0
6
+ License-File: LICENSE
7
+ Requires-Python: >=3.10
8
+ Requires-Dist: leb128~=1.0.8
9
+ Requires-Dist: numcodecs-combinators~=0.2.15
10
+ Requires-Dist: numcodecs<0.17,>=0.13.0
11
+ Requires-Dist: numpy~=2.0
12
+ Requires-Dist: typing-extensions~=4.6
13
+ Description-Content-Type: text/markdown
14
+
15
+ [![image](https://img.shields.io/github/actions/workflow/status/juntyr/numcodecs-mask/ci.yml?branch=main)](https://github.com/juntyr/numcodecs-mask/actions/workflows/ci.yml?query=branch%3Amain)
16
+ [![image](https://img.shields.io/pypi/v/numcodecs-mask.svg)](https://pypi.python.org/pypi/numcodecs-mask)
17
+ [![image](https://img.shields.io/pypi/l/numcodecs-mask.svg)](https://github.com/juntyr/numcodecs-mask/blob/main/LICENSE)
18
+ [![image](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fjuntyr%2Fnumcodecs-mask%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://pypi.python.org/pypi/numcodecs-mask)
19
+ [![image](https://readthedocs.org/projects/numcodecs-mask/badge/?version=latest)](https://numcodecs-mask.readthedocs.io/en/latest/?badge=latest)
20
+
21
+ # numcodecs-mask
22
+
23
+ Masking codecs for the [`numcodecs`] buffer compression API.
24
+
25
+ [`numcodecs`]: https://numcodecs.readthedocs.io/en/stable/
26
+
27
+ ## License
28
+
29
+ Licensed under the Mozilla Public License, Version 2.0 ([LICENSE](LICENSE) or https://www.mozilla.org/en-US/MPL/2.0/).
30
+
31
+
32
+ ## Funding
33
+
34
+ The `numcodecs-mask` package has been developed as part of [ESiWACE3](https://www.esiwace.eu), the third phase of the Centre of Excellence in Simulation of Weather and Climate in Europe.
35
+
36
+ Funded by the European Union. This work has received funding from the European High Performance Computing Joint Undertaking (JU) under grant agreement No 101093054.
@@ -0,0 +1,22 @@
1
+ [![image](https://img.shields.io/github/actions/workflow/status/juntyr/numcodecs-mask/ci.yml?branch=main)](https://github.com/juntyr/numcodecs-mask/actions/workflows/ci.yml?query=branch%3Amain)
2
+ [![image](https://img.shields.io/pypi/v/numcodecs-mask.svg)](https://pypi.python.org/pypi/numcodecs-mask)
3
+ [![image](https://img.shields.io/pypi/l/numcodecs-mask.svg)](https://github.com/juntyr/numcodecs-mask/blob/main/LICENSE)
4
+ [![image](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fjuntyr%2Fnumcodecs-mask%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://pypi.python.org/pypi/numcodecs-mask)
5
+ [![image](https://readthedocs.org/projects/numcodecs-mask/badge/?version=latest)](https://numcodecs-mask.readthedocs.io/en/latest/?badge=latest)
6
+
7
+ # numcodecs-mask
8
+
9
+ Masking codecs for the [`numcodecs`] buffer compression API.
10
+
11
+ [`numcodecs`]: https://numcodecs.readthedocs.io/en/stable/
12
+
13
+ ## License
14
+
15
+ Licensed under the Mozilla Public License, Version 2.0 ([LICENSE](LICENSE) or https://www.mozilla.org/en-US/MPL/2.0/).
16
+
17
+
18
+ ## Funding
19
+
20
+ The `numcodecs-mask` package has been developed as part of [ESiWACE3](https://www.esiwace.eu), the third phase of the Centre of Excellence in Simulation of Weather and Climate in Europe.
21
+
22
+ Funded by the European Union. This work has received funding from the European High Performance Computing Joint Undertaking (JU) under grant agreement No 101093054.
@@ -0,0 +1,15 @@
1
+ [![image](https://img.shields.io/github/actions/workflow/status/juntyr/numcodecs-mask/ci.yml?branch=main)](https://github.com/juntyr/numcodecs-mask/actions/workflows/ci.yml?query=branch%3Amain)
2
+ [![image](https://img.shields.io/pypi/v/numcodecs-mask.svg)](https://pypi.python.org/pypi/numcodecs-mask)
3
+ [![image](https://img.shields.io/pypi/l/numcodecs-mask.svg)](https://github.com/juntyr/numcodecs-mask/blob/main/LICENSE)
4
+ [![image](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fjuntyr%2Fnumcodecs-mask%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://pypi.python.org/pypi/numcodecs-mask)
5
+ [![image](https://readthedocs.org/projects/numcodecs-mask/badge/?version=latest)](https://numcodecs-mask.readthedocs.io/en/latest/?badge=latest)
6
+
7
+ # numcodecs-mask
8
+
9
+ Masking codecs for the [`numcodecs`][numcodecs] buffer compression API.
10
+
11
+ ## Funding
12
+
13
+ The `numcodecs-mask` package has been developed as part of [ESiWACE3](https://www.esiwace.eu), the third phase of the Centre of Excellence in Simulation of Weather and Climate in Europe.
14
+
15
+ Funded by the European Union. This work has received funding from the European High Performance Computing Joint Undertaking (JU) under grant agreement No 101093054.
@@ -0,0 +1,8 @@
1
+ mkdocs
2
+ mkdocstrings[python]
3
+ mkdocs-exclude
4
+ mkdocstrings-python-generator==1.0.0rc1
5
+
6
+ Pygments
7
+
8
+ ruff
@@ -0,0 +1,62 @@
1
+ site_name: numcodecs-mask
2
+
3
+ strict: true
4
+
5
+ markdown_extensions:
6
+ - footnotes
7
+ - pymdownx.superfences
8
+ - pymdownx.highlight:
9
+ use_pygments: true
10
+ noclasses: true
11
+
12
+ theme:
13
+ name: readthedocs
14
+ highlightjs: false
15
+
16
+ nav:
17
+ - Home: index.md
18
+ - Links:
19
+ - GitHub: https://github.com/juntyr/numcodecs-mask
20
+ - PyPI: https://pypi.org/project/numcodecs-mask/
21
+
22
+ plugins:
23
+ - mkdocstrings-python-generator:
24
+ source_dirs:
25
+ - nav_heading: [Documentation]
26
+ base: src
27
+ - exclude:
28
+ glob:
29
+ - requirements.txt
30
+ - search
31
+ - mkdocstrings:
32
+ enable_inventory: true
33
+ handlers:
34
+ python:
35
+ options:
36
+ docstring_section_style: list
37
+ docstring_style: numpy
38
+ show_if_no_docstring: true
39
+ filters: ["!^_$", "!^_[^_]", "!^__", "__init__"]
40
+ members_order: source
41
+ group_by_category: false
42
+ show_source: false
43
+ separate_signature: true
44
+ show_signature_annotations: true
45
+ signature_crossrefs: true
46
+ relative_crossrefs: true
47
+ show_root_toc_entry: false
48
+ merge_init_into_class: true
49
+ summary:
50
+ attributes: false
51
+ classes: true
52
+ functions: true
53
+ modules: true
54
+ inventories:
55
+ - https://docs.python.org/3.12/objects.inv
56
+ - https://numpy.org/doc/2.2/objects.inv
57
+ - https://numcodecs.readthedocs.io/en/v0.16.0/objects.inv
58
+ - https://numcodecs-combinators.readthedocs.io/en/v0.2.15/objects.inv
59
+ - https://numcodecs-replace.readthedocs.io/en/0.1.2/objects.inv
60
+
61
+ watch:
62
+ - src
@@ -0,0 +1,50 @@
1
+ [build-system]
2
+ requires = ["hatchling"]
3
+ build-backend = "hatchling.build"
4
+
5
+ [project]
6
+ name = "numcodecs-mask"
7
+ version = "0.1.0"
8
+ description = "Masking codecs for the `numcodecs` buffer compression API"
9
+ readme = "README.md"
10
+ license = "MPL-2.0"
11
+ requires-python = ">=3.10"
12
+ dependencies = [
13
+ "leb128~=1.0.8",
14
+ "numcodecs>=0.13.0,<0.17",
15
+ "numcodecs-combinators~=0.2.15",
16
+ "numpy~=2.0",
17
+ "typing-extensions~=4.6",
18
+ ]
19
+
20
+ [dependency-groups]
21
+ dev = ["mypy~=1.14", "pytest~=8.3"]
22
+ docs = [
23
+ "mkdocstrings[python]~=1.0",
24
+ "mkdocstrings-python-generator==1.0.0rc1",
25
+ "mkdocs-exclude~=1.0",
26
+ "properdocs~=1.6.5",
27
+ "properdocs-theme-readthedocs~=1.6.5",
28
+ "Pygments~=2.19",
29
+ "ruff~=0.14",
30
+ ]
31
+
32
+ [project.entry-points."numcodecs.codecs"]
33
+ "mask.meta" = "numcodecs_mask:MaskMetaCodec"
34
+
35
+ [tool.setuptools.packages.find]
36
+ where = ["src"]
37
+
38
+ [tool.setuptools.package-data]
39
+ "numcodecs_mask" = ["py.typed"]
40
+
41
+ [tool.ruff.lint]
42
+ select = ["E4", "E7", "E9", "F", "I"]
43
+
44
+ [tool.pytest.ini_options]
45
+ addopts = ["--import-mode=importlib"]
46
+ xfail_strict = true
47
+
48
+ [[tool.mypy.overrides]]
49
+ module = ["numcodecs.*"]
50
+ follow_untyped_imports = true
@@ -0,0 +1,297 @@
1
+ """
2
+ Masking codecs for the [`numcodecs`][numcodecs] buffer compression API.
3
+ """
4
+
5
+ __all__ = ["MaskMetaCodec"]
6
+
7
+ from collections.abc import Callable
8
+ from functools import reduce
9
+ from io import BytesIO
10
+
11
+ import leb128
12
+ import numcodecs.compat
13
+ import numcodecs.registry
14
+ import numpy as np
15
+ from numcodecs.abc import Codec
16
+ from numcodecs_combinators.abc import CodecCombinatorMixin
17
+ from typing_extensions import Buffer # MSPV 3.12
18
+
19
+
20
+ class MaskMetaCodec(Codec, CodecCombinatorMixin):
21
+ """
22
+ Meta-codec that masks a value during encoding and restores it during decoding.
23
+
24
+ The data is encoded with the `codec`, the [boolean][numpy.bool] mask of
25
+ where values need to be restored is encoded with the `bitmap_codec`.
26
+
27
+ Multiple [`MaskMetaCodec`][.]s can be nested to mask several values.
28
+ The masked value can be replaced with a fill value before further encoding
29
+ by including a
30
+ [`numcodecs_replace.ReplaceFilterCodec`][numcodecs_replace.ReplaceFilterCodec]
31
+ in the `codec`, e.g. by stacking using the
32
+ [`numcodecs-combinators`](https://numcodecs-combinators.readthedocs.io)
33
+ package.
34
+
35
+ Parameters
36
+ ----------
37
+ mask : int | float
38
+ The value to be masked.
39
+ codec : dict | Codec
40
+ The configuration or instantiated codec that encodes the data.
41
+ bitmap_codec : dict | Codec
42
+ The configuration or instantiated codec that encodes the
43
+ [boolean][numpy.bool] mask of where values need to be restored.
44
+
45
+ For instance, the [`numcodecs.PackBits`][numcodecs.packbits.PackBits]
46
+ codec can be used to pack the mask into a byte array.
47
+ """
48
+
49
+ __slots__: tuple[str, ...] = ("_mask", "_codec", "_bitmap_codec")
50
+ _mask: int | float
51
+ _codec: Codec
52
+ _bitmap_codec: Codec
53
+
54
+ codec_id: str = "mask.meta" # type: ignore
55
+
56
+ def __init__(
57
+ self,
58
+ *,
59
+ mask: int | float,
60
+ codec: dict | Codec,
61
+ bitmap_codec: dict | Codec,
62
+ ) -> None:
63
+ self._mask = mask
64
+ self._codec = (
65
+ codec if isinstance(codec, Codec) else numcodecs.registry.get_codec(codec)
66
+ )
67
+ self._bitmap_codec = (
68
+ bitmap_codec
69
+ if isinstance(bitmap_codec, Codec)
70
+ else numcodecs.registry.get_codec(bitmap_codec)
71
+ )
72
+
73
+ def encode(self, buf: Buffer) -> Buffer:
74
+ """Encode the data in `buf`.
75
+
76
+ Parameters
77
+ ----------
78
+ buf : Buffer
79
+ Data to be encoded. May be any object supporting the new-style
80
+ buffer protocol.
81
+
82
+ Returns
83
+ -------
84
+ enc : Buffer
85
+ Encoded data. May be any object supporting the new-style buffer
86
+ protocol.
87
+ """
88
+
89
+ a = np.copy(numcodecs.compat.ensure_ndarray(buf))
90
+ dtype, shape = a.dtype, a.shape
91
+
92
+ if isinstance(self._mask, int) or not np.isnan(self._mask):
93
+ is_masked = a == self._mask
94
+ else:
95
+ is_masked = np.isnan(a)
96
+
97
+ # message: dtype shape encoded-dtype encoded-shape [padding] encoded
98
+ # bitmap-dtype bitmap-shape [padding] bitmap
99
+ message: list[bytes | bytearray] = []
100
+
101
+ message.append(leb128.u.encode(len(dtype.str)))
102
+ message.append(dtype.str.encode("ascii"))
103
+
104
+ message.append(leb128.u.encode(len(shape)))
105
+ for s in shape:
106
+ message.append(leb128.u.encode(s))
107
+
108
+ encoded = self._codec.encode(a)
109
+ encoded = numcodecs.compat.ensure_ndarray(encoded)
110
+
111
+ message.append(leb128.u.encode(len(encoded.dtype.str)))
112
+ message.append(encoded.dtype.str.encode("ascii"))
113
+
114
+ message.append(leb128.u.encode(encoded.ndim))
115
+ for s in encoded.shape:
116
+ message.append(leb128.u.encode(s))
117
+
118
+ # insert padding to align with encoded itemsize
119
+ message.append(
120
+ b"\0"
121
+ * (
122
+ encoded.dtype.itemsize
123
+ - (sum(len(m) for m in message) % encoded.itemsize)
124
+ )
125
+ )
126
+
127
+ # ensure that the encoded values are encoded in little endian binary
128
+ message.append(encoded.astype(encoded.dtype.newbyteorder("<")).tobytes())
129
+
130
+ bitmap = self._bitmap_codec.encode(is_masked)
131
+ bitmap = numcodecs.compat.ensure_ndarray(bitmap)
132
+
133
+ message.append(leb128.u.encode(len(bitmap.dtype.str)))
134
+ message.append(bitmap.dtype.str.encode("ascii"))
135
+
136
+ message.append(leb128.u.encode(bitmap.ndim))
137
+ for s in bitmap.shape:
138
+ message.append(leb128.u.encode(s))
139
+
140
+ # insert padding to align with bitmap itemsize
141
+ message.append(
142
+ b"\0"
143
+ * (bitmap.dtype.itemsize - (sum(len(m) for m in message) % bitmap.itemsize))
144
+ )
145
+
146
+ # ensure that the bitmap values are encoded in little endian binary
147
+ message.append(bitmap.astype(bitmap.dtype.newbyteorder("<")).tobytes())
148
+
149
+ return b"".join(message)
150
+
151
+ def decode(self, buf: Buffer, out: None | Buffer = None) -> Buffer:
152
+ """
153
+ Decode the data in `buf`.
154
+
155
+ Parameters
156
+ ----------
157
+ buf : Buffer
158
+ Encoded data. May be any object supporting the new-style buffer
159
+ protocol.
160
+ out : Buffer, optional
161
+ Writeable buffer to store decoded data. N.B. if provided, this buffer must
162
+ be exactly the right size to store the decoded data.
163
+
164
+ Returns
165
+ -------
166
+ dec : Buffer
167
+ Decoded data. May be any object supporting the new-style buffer
168
+ protocol.
169
+ """
170
+
171
+ b = numcodecs.compat.ensure_bytes(buf)
172
+ b_io = BytesIO(b)
173
+
174
+ # message: dtype shape encoded-dtype encoded-shape [padding] encoded
175
+ # bitmap-dtype bitmap-shape [padding] bitmap
176
+ dtype = np.dtype(b_io.read(leb128.u.decode_reader(b_io)[0]).decode("ascii"))
177
+ shape = tuple(
178
+ leb128.u.decode_reader(b_io)[0]
179
+ for _ in range(leb128.u.decode_reader(b_io)[0])
180
+ )
181
+
182
+ encoded_dtype = np.dtype(
183
+ b_io.read(leb128.u.decode_reader(b_io)[0]).decode("ascii")
184
+ )
185
+ encoded_shape = tuple(
186
+ leb128.u.decode_reader(b_io)[0]
187
+ for _ in range(leb128.u.decode_reader(b_io)[0])
188
+ )
189
+ encoded_size = reduce(lambda a, b: a * b, encoded_shape, 1)
190
+
191
+ # remove padding to align with encoded itemsize
192
+ b_io.read(encoded_dtype.itemsize - (b_io.tell() % encoded_dtype.itemsize))
193
+
194
+ encoded = (
195
+ np.frombuffer(
196
+ b_io.read(encoded_size * encoded_dtype.itemsize),
197
+ dtype=encoded_dtype.newbyteorder("<"),
198
+ count=encoded_size,
199
+ )
200
+ .astype(encoded_dtype)
201
+ .reshape(encoded_shape)
202
+ )
203
+
204
+ decoded = np.empty(shape, dtype=dtype)
205
+ self._codec.decode(encoded, out=decoded)
206
+
207
+ bitmap_dtype = np.dtype(
208
+ b_io.read(leb128.u.decode_reader(b_io)[0]).decode("ascii")
209
+ )
210
+ bitmap_shape = tuple(
211
+ leb128.u.decode_reader(b_io)[0]
212
+ for _ in range(leb128.u.decode_reader(b_io)[0])
213
+ )
214
+ bitmap_size = reduce(lambda a, b: a * b, bitmap_shape, 1)
215
+
216
+ # remove padding to align with bitmap itemsize
217
+ b_io.read(bitmap_dtype.itemsize - (b_io.tell() % bitmap_dtype.itemsize))
218
+
219
+ bitmap = (
220
+ np.frombuffer(
221
+ b_io.read(bitmap_size * bitmap_dtype.itemsize),
222
+ dtype=bitmap_dtype.newbyteorder("<"),
223
+ count=bitmap_size,
224
+ )
225
+ .astype(bitmap_dtype)
226
+ .reshape(bitmap_shape)
227
+ )
228
+
229
+ is_masked = np.empty(shape, dtype=np.bool)
230
+ self._bitmap_codec.decode(bitmap, out=is_masked)
231
+
232
+ decoded[is_masked] = self._mask
233
+
234
+ return numcodecs.compat.ndarray_copy(decoded, out) # type: ignore
235
+
236
+ def get_config(self) -> dict:
237
+ """
238
+ Returns the configuration of this mask meta-codec.
239
+
240
+ [`numcodecs.registry.get_codec(config)`][numcodecs.registry.get_codec]
241
+ can be used to reconstruct this codec from the returned config.
242
+
243
+ Returns
244
+ -------
245
+ config : dict
246
+ Configuration of this mask meta-codec.
247
+ """
248
+
249
+ return dict(
250
+ id=type(self).codec_id,
251
+ mask=self._mask,
252
+ codec=self._codec.get_config(),
253
+ bitmap_codec=self._bitmap_codec.get_config(),
254
+ )
255
+
256
+ def __repr__(self) -> str:
257
+ return f"{type(self).__name__}(mask={self._mask!r}, codec={self._codec!r}, bitmap_codec={self._bitmap_codec!r})"
258
+
259
+ def map(self, mapper: Callable[[Codec], Codec]) -> "MaskMetaCodec":
260
+ """
261
+ Apply the `mapper` to this mask meta-codec.
262
+
263
+ In the returned [`MaskMetaCodec`][..], the `codec` and
264
+ `bitmap_codec` are replaced by their mapped codecs.
265
+
266
+ The `mapper` should recursively apply itself to any inner codecs that
267
+ also implement the
268
+ [`CodecCombinatorMixin`][numcodecs_combinators.abc.CodecCombinatorMixin]
269
+ mixin.
270
+
271
+ To automatically handle the recursive application as a caller, you can
272
+ use
273
+ ```python
274
+ numcodecs_combinators.map_codec(codec, mapper)
275
+ ```
276
+ instead.
277
+
278
+ Parameters
279
+ ----------
280
+ mapper : Callable[[Codec], Codec]
281
+ The callable that should be applied to the wrapped `codec` and
282
+ `bitmap_codec` to map over this mask meta-codec.
283
+
284
+ Returns
285
+ -------
286
+ mapped : MaskMetaCodec
287
+ The mapped mask meta-codec.
288
+ """
289
+
290
+ return MaskMetaCodec(
291
+ mask=self._mask,
292
+ codec=mapper(self._codec),
293
+ bitmap_codec=mapper(self._bitmap_codec),
294
+ )
295
+
296
+
297
+ numcodecs.registry.register_codec(MaskMetaCodec)
File without changes
@@ -0,0 +1,115 @@
1
+ import json
2
+ from typing import ClassVar
3
+
4
+ import numcodecs
5
+ import numcodecs.compat
6
+ import numcodecs.registry
7
+ import numpy as np
8
+ import pytest
9
+ from numcodecs.abc import Codec
10
+
11
+
12
+ def test_from_config():
13
+ config = dict(
14
+ id="mask.meta",
15
+ mask=np.nan,
16
+ codec=dict(id="zlib", level=1),
17
+ bitmap_codec=dict(id="packbits"),
18
+ )
19
+
20
+ codec = numcodecs.registry.get_codec(config)
21
+ assert codec.__class__.__name__ == "MaskMetaCodec"
22
+ assert codec.__class__.__module__ == "numcodecs_mask"
23
+
24
+ assert (
25
+ repr(codec)
26
+ == "MaskMetaCodec(mask=nan, codec=Zlib(level=1), bitmap_codec=PackBits())"
27
+ )
28
+
29
+ assert json.dumps(codec.get_config(), sort_keys=True) == json.dumps(
30
+ config, sort_keys=True
31
+ )
32
+
33
+
34
+ def check_roundtrip(data: np.ndarray, mask: int | float):
35
+ config = dict(
36
+ id="mask.meta",
37
+ mask=mask,
38
+ codec=dict(id="noise", seed=42),
39
+ bitmap_codec=dict(id="packbits"),
40
+ )
41
+
42
+ codec = numcodecs.registry.get_codec(config)
43
+
44
+ assert (
45
+ repr(codec)
46
+ == f"MaskMetaCodec(mask={mask!r}, codec=NoiseCodec(seed=42), bitmap_codec=PackBits())"
47
+ )
48
+
49
+ assert json.dumps(codec.get_config(), sort_keys=True) == json.dumps(
50
+ config, sort_keys=True
51
+ )
52
+
53
+ encoded = codec.encode(data)
54
+ decoded = codec.decode(encoded)
55
+
56
+ assert decoded.dtype == data.dtype
57
+ assert decoded.shape == data.shape
58
+
59
+ if isinstance(mask, int) or not np.isnan(mask):
60
+ assert np.all((decoded == mask) | (data != mask))
61
+ else:
62
+ assert np.all(np.isnan(decoded) | ~np.isnan(data))
63
+
64
+
65
+ @np.errstate(invalid="ignore")
66
+ @pytest.mark.parametrize(
67
+ "mask", [0.0, -0.0, 1.0, -1.0, np.nan, -np.nan, np.inf, -np.inf, 4.2, -2.4]
68
+ )
69
+ def test_roundtrip(mask):
70
+ check_roundtrip(np.zeros(tuple()), mask=mask)
71
+ check_roundtrip(np.zeros((0,)), mask=mask)
72
+ if np.isfinite(mask):
73
+ check_roundtrip(np.arange(1000).reshape(10, 10, 10), mask=mask)
74
+ check_roundtrip(np.array([4.2, -2.4, np.nan, -np.nan, 0.0, -0.0]), mask=mask)
75
+ check_roundtrip(np.array([np.inf, -np.inf, np.nan, -np.nan, 0.0, -0.0]), mask=mask)
76
+ check_roundtrip(
77
+ np.array(
78
+ [np.inf, -np.inf, np.nan, -np.nan, 0.0, -0.0],
79
+ dtype=np.dtype(np.float64).newbyteorder("<"),
80
+ ),
81
+ mask=mask,
82
+ )
83
+ check_roundtrip(
84
+ np.array(
85
+ [np.inf, -np.inf, np.nan, -np.nan, 0.0, -0.0],
86
+ dtype=np.dtype(np.float64).newbyteorder(">"),
87
+ ),
88
+ mask=mask,
89
+ )
90
+
91
+
92
+ class NoiseCodec(Codec):
93
+ __slots__: tuple[str, ...] = ("_seed",)
94
+ _seed: int
95
+
96
+ codec_id: ClassVar[str] = "noise" # type: ignore
97
+
98
+ def __init__(self, *, seed: int):
99
+ self._seed = seed
100
+
101
+ def encode(self, buf):
102
+ rng = np.random.Generator(np.random.PCG64(seed=self._seed))
103
+ return buf + rng.normal(scale=0.1, size=buf.shape).astype(buf.dtype)
104
+
105
+ def decode(self, buf, out=None):
106
+ return numcodecs.compat.ndarray_copy(buf, out)
107
+
108
+ def get_config(self) -> dict:
109
+ return dict(id=type(self).codec_id, seed=self._seed)
110
+
111
+ def __repr__(self) -> str:
112
+ return f"NoiseCodec(seed={self._seed!r})"
113
+
114
+
115
+ numcodecs.registry.register_codec(NoiseCodec)