add zstd to v2

Author: brokkoli71

src/main/java/dev/zarr/zarrjava/core/codec/core/ZstdCodec.java

@@ -0,0 +1,40 @@
+package dev.zarr.zarrjava.core.codec.core;
+
+import com.github.luben.zstd.Zstd;
+import com.github.luben.zstd.ZstdCompressCtx;
+import dev.zarr.zarrjava.ZarrException;
+import dev.zarr.zarrjava.core.codec.BytesBytesCodec;
+
+import java.nio.ByteBuffer;
+
+public abstract class ZstdCodec extends BytesBytesCodec {
+
+    protected abstract int getLevel();
+
+    protected abstract boolean getChecksum();
+
+    @Override
+    public ByteBuffer decode(ByteBuffer compressedBytes) throws ZarrException {
+        byte[] compressedArray = compressedBytes.array();
+
+        long originalSize = Zstd.getFrameContentSize(compressedArray);
+        if (originalSize == 0) {
+            throw new ZarrException("Failed to get decompressed size");
+        }
+
+        byte[] decompressed = Zstd.decompress(compressedArray, (int) originalSize);
+        return ByteBuffer.wrap(decompressed);
+    }
+
+    @Override
+    public ByteBuffer encode(ByteBuffer chunkBytes) throws ZarrException {
+        byte[] arr = chunkBytes.array();
+        byte[] compressed;
+        try (ZstdCompressCtx ctx = new ZstdCompressCtx()) {
+            ctx.setLevel(getLevel());
+            ctx.setChecksum(getChecksum());
+            compressed = ctx.compress(arr);
+        }
+        return ByteBuffer.wrap(compressed);
+    }
+}

src/main/java/dev/zarr/zarrjava/v2/codec/CodecRegistry.java

@@ -3,6 +3,7 @@
 import com.fasterxml.jackson.databind.jsontype.NamedType;
 import dev.zarr.zarrjava.v2.codec.core.BloscCodec;
 import dev.zarr.zarrjava.v2.codec.core.ZlibCodec;
+import dev.zarr.zarrjava.v2.codec.core.ZstdCodec;
 
 import java.util.HashMap;
 import java.util.Map;
@@ -14,6 +15,7 @@ public class CodecRegistry {
     static {
         addType("blosc", BloscCodec.class);
         addType("zlib", ZlibCodec.class);
+        addType("zstd", ZstdCodec.class);
     }
 
     public static void addType(String name, Class<? extends Codec> codecClass) {

src/main/java/dev/zarr/zarrjava/v2/codec/core/ZstdCodec.java

@@ -0,0 +1,43 @@
+package dev.zarr.zarrjava.v2.codec.core;
+
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonIgnore;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import dev.zarr.zarrjava.ZarrException;
+import dev.zarr.zarrjava.core.ArrayMetadata;
+import dev.zarr.zarrjava.v2.codec.Codec;
+
+public class ZstdCodec extends dev.zarr.zarrjava.core.codec.core.ZstdCodec implements Codec {
+
+    @JsonIgnore
+    public final String id = "zstd";
+    public final int level;
+    public final boolean checksum;
+
+    @JsonCreator(mode = JsonCreator.Mode.PROPERTIES)
+    public ZstdCodec(
+            @JsonProperty(value = "level", defaultValue = "0") int level,
+            @JsonProperty(value = "checksum", defaultValue = "false") boolean checksum)
+            throws ZarrException {
+        if (level < -131072 || level > 22) {
+            throw new ZarrException("'level' needs to be between -131072 and 22.");
+        }
+        this.level = level;
+        this.checksum = checksum;
+    }
+
+    @Override
+    protected int getLevel() {
+        return level;
+    }
+
+    @Override
+    protected boolean getChecksum() {
+        return checksum;
+    }
+
+    @Override
+    public Codec evolveFromCoreArrayMetadata(ArrayMetadata.CoreArrayMetadata arrayMetadata) {
+        return this;
+    }
+}

src/main/java/dev/zarr/zarrjava/v3/codec/core/ZstdCodec.java

@@ -3,17 +3,13 @@
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonIgnore;
 import com.fasterxml.jackson.annotation.JsonProperty;
-import com.github.luben.zstd.Zstd;
-import com.github.luben.zstd.ZstdCompressCtx;
 import dev.zarr.zarrjava.ZarrException;
-import dev.zarr.zarrjava.core.codec.BytesBytesCodec;
 import dev.zarr.zarrjava.v3.ArrayMetadata;
 import dev.zarr.zarrjava.v3.codec.Codec;
 
 import javax.annotation.Nonnull;
-import java.nio.ByteBuffer;
 
-public class ZstdCodec extends BytesBytesCodec implements Codec {
+public class ZstdCodec extends dev.zarr.zarrjava.core.codec.core.ZstdCodec implements Codec {
 
     @JsonIgnore
     public final String name = "zstd";
@@ -27,28 +23,13 @@ public ZstdCodec(
     }
 
     @Override
-    public ByteBuffer decode(ByteBuffer compressedBytes) throws ZarrException {
-        byte[] compressedArray = compressedBytes.array();
-
-        long originalSize = Zstd.getFrameContentSize(compressedArray);
-        if (originalSize == 0) {
-            throw new ZarrException("Failed to get decompressed size");
-        }
-
-        byte[] decompressed = Zstd.decompress(compressedArray, (int) originalSize);
-        return ByteBuffer.wrap(decompressed);
+    protected int getLevel() {
+        return configuration.level;
     }
 
     @Override
-    public ByteBuffer encode(ByteBuffer chunkBytes) throws ZarrException {
-        byte[] arr = chunkBytes.array();
-        byte[] compressed;
-        try (ZstdCompressCtx ctx = new ZstdCompressCtx()) {
-            ctx.setLevel(configuration.level);
-            ctx.setChecksum(configuration.checksum);
-            compressed = ctx.compress(arr);
-        }
-        return ByteBuffer.wrap(compressed);
+    protected boolean getChecksum() {
+        return configuration.checksum;
     }
 
     @Override