Add EBCC to the benchmark

pyproject.toml

@@ -19,6 +19,7 @@ dependencies = [
     "numcodecs-safeguards==0.1.0b2",
     "numcodecs-wasm==0.2.2",
     "numcodecs-wasm-bit-round==0.4.0",
+    "numcodecs-wasm-ebcc==0.3.0a0",
     "numcodecs-wasm-fixed-offset-scale==0.4.0",
     "numcodecs-wasm-jpeg2000==0.3.0",
     "numcodecs-wasm-pco==0.3.0",

src/climatebenchpress/compressor/compressors/__init__.py

@@ -1,6 +1,7 @@
 __all__ = [
     "BitRound",
     "BitRoundPco",
+    "Ebcc",
     "Jpeg2000",
     "SafeguardedBitRoundPco",
     "SafeguardedSperr",
@@ -20,6 +21,7 @@
 from . import abc as abc
 from .bitround import BitRound
 from .bitround_pco import BitRoundPco
+from .ebcc import Ebcc
 from .jpeg2000 import Jpeg2000
 from .safeguarded import (
     SafeguardedBitRoundPco,

src/climatebenchpress/compressor/compressors/ebcc.py

@@ -0,0 +1,52 @@
+__all__ = ["Ebcc"]
+
+import numcodecs.astype
+import numcodecs_wasm_ebcc
+from numcodecs_combinators.stack import CodecStack
+
+from .abc import Compressor
+
+
+class Ebcc(Compressor):
+    """EBCC compressor."""
+
+    name = "ebcc"
+    description = "EBCC"
+
+    @staticmethod
+    def abs_bound_codec(error_bound, dtype=None, **kwargs):
+        assert dtype is not None, "dtype must be provided"
+
+        return CodecStack(
+            # EBCC only supports float32 data
+            numcodecs.astype.AsType(
+                encode_dtype="float32",
+                decode_dtype=dtype.name,
+            ),
+            numcodecs_wasm_ebcc.Ebcc(
+                # reasonable default recommended by Langwen Huang
+                base_cr=100,
+                residual="absolute",
+                error=error_bound,
+                chunk_shape="auto",
+            ),
+        )
+
+    @staticmethod
+    def rel_bound_codec(error_bound, dtype=None, **kwargs):
+        assert dtype is not None, "dtype must be provided"
+
+        return CodecStack(
+            # EBCC only supports float32 data
+            numcodecs.astype.AsType(
+                encode_dtype="float32",
+                decode_dtype=dtype.name,
+            ),
+            numcodecs_wasm_ebcc.Ebcc(
+                # reasonable default recommended by Langwen Huang
+                base_cr=100,
+                residual="relative",
+                error=error_bound,
+                chunk_shape="auto",
+            ),
+        )