check sharding bounds

Author: brokkoli71

src/main/java/dev/zarr/zarrjava/v3/ArrayMetadataBuilder.java

@@ -11,6 +11,9 @@
 import dev.zarr.zarrjava.v3.codec.CodecBuilder;
 import dev.zarr.zarrjava.v3.codec.core.BytesCodec;
 import dev.zarr.zarrjava.v3.codec.core.BytesCodec.Endian;
+import dev.zarr.zarrjava.v3.codec.core.ShardingIndexedCodec;
+
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.function.Function;
@@ -123,6 +126,22 @@ public ArrayMetadataBuilder withAttributes(Map<String, Object> attributes) {
     return this;
   }
 
+  public void verifyShardingBounds(int[] outerChunkShape, Codec[] codecs) throws ZarrException {
+    for (Codec codec : codecs) {
+      if (codec instanceof ShardingIndexedCodec) {
+        ShardingIndexedCodec.Configuration shardingConfig = ((ShardingIndexedCodec) codec).configuration;
+        int[] innerChunkShape = shardingConfig.chunkShape;
+        if (outerChunkShape.length != innerChunkShape.length)
+          throw new ZarrException("Sharding dimensions mismatch of outer chunk shape " + Arrays.toString(outerChunkShape) + " and inner chunk shape" + Arrays.toString(innerChunkShape));
+        for (int i = 0; i < outerChunkShape.length; i++) {
+          if (outerChunkShape[i] < innerChunkShape[i])
+            throw new ZarrException("Sharding outer chunk shape " + Arrays.toString(outerChunkShape) + " can not contain inner chunk shape " + Arrays.toString(innerChunkShape));
+        }
+        verifyShardingBounds(innerChunkShape, shardingConfig.codecs);
+      }
+    }
+  }
+
   public ArrayMetadata build() throws ZarrException {
     if (shape == null) {
       throw new ZarrException("Shape needs to be provided. Please call `.withShape`.");
@@ -133,12 +152,21 @@ public ArrayMetadata build() throws ZarrException {
     if (chunkGrid == null) {
       throw new ZarrException("Chunk grid needs to be provided. Please call `.withChunkShape`.");
     }
-    if (chunkGrid instanceof RegularChunkGrid
-        && shape.length != ((RegularChunkGrid) chunkGrid).configuration.chunkShape.length) {
-      throw new ZarrException("Shape (ndim=" + shape.length + ") and chunk shape (ndim=" +
-          ((RegularChunkGrid) chunkGrid).configuration.chunkShape.length +
-          ") need to have the same number of dimensions.");
+    if (chunkGrid instanceof RegularChunkGrid) {
+      int[] chunkShape = ((RegularChunkGrid) chunkGrid).configuration.chunkShape;
+      if (shape.length != chunkShape.length) {
+        throw new ZarrException("Shape (ndim=" + shape.length + ") and chunk shape (ndim=" +
+            chunkShape.length + ") need to have the same number of dimensions.");
+      }
+      for (int i = 0; i < shape.length; i++) {
+        if (shape[i] < chunkShape[i]) {
+          throw new ZarrException("Shape " + Arrays.toString(shape) + " can not contain chunk shape "
+              + Arrays.toString(chunkShape));
+        }
+      }
+      verifyShardingBounds(chunkShape, codecs);
     }
+
     return new ArrayMetadata(shape, dataType, chunkGrid, chunkKeyEncoding, fillValue, codecs,
         dimensionNames,
         attributes