Add generate sphere

Author: TPexcoTblu

src/client/java/com/tpexcotblu/litematicgen/LitematicGeneratorClient.java

@@ -6,6 +6,6 @@
 public class LitematicGeneratorClient implements ClientModInitializer {
 	@Override
 	public void onInitializeClient() {
-		SchemeCommandRegistrar.registerCircleCommand();
+		SchemeCommandRegistrar.registerAllCommands();
 	}
 }

src/client/java/com/tpexcotblu/litematicgen/commands/SchemeCommandRegistrar.java

@@ -4,6 +4,7 @@
 import com.mojang.brigadier.arguments.StringArgumentType;
 import com.mojang.brigadier.suggestion.SuggestionProvider;
 import com.tpexcotblu.litematicgen.generators.SchemeGenerator;
+import com.tpexcotblu.litematicgen.generators.SphereGenerator;
 import net.fabricmc.fabric.api.client.command.v2.ClientCommandRegistrationCallback;
 import net.fabricmc.fabric.api.client.command.v2.FabricClientCommandSource;
 import net.minecraft.nbt.NbtCompound;
@@ -58,7 +59,7 @@ private static int saveScheme(FabricClientCommandSource source, SchemeGenerator
         return 1;
     }
 
-    public static void registerCircleCommand() {
+    private static void registerCircleCommand() {
         SuggestionProvider<FabricClientCommandSource> orientationSuggestions = (context, builder) -> {
             return builder.suggest("horizontal").suggest("vertical").buildFuture();
         };
@@ -80,4 +81,24 @@ public static void registerCircleCommand() {
             );
         });
     }
+
+    private static void registerSphereCommand() {
+        ClientCommandRegistrationCallback.EVENT.register((dispatcher, registryAccess) -> {
+            dispatcher.register(
+                    literal("generatesphere")
+                            .then(argument("radius", IntegerArgumentType.integer(1, 128))
+                                    .executes(ctx -> {
+                                        int radius = IntegerArgumentType.getInteger(ctx, "radius");
+                                        SphereGenerator generator = new SphereGenerator(radius);
+                                        return saveScheme(ctx.getSource(), generator);
+                                    })
+                            )
+            );
+        });
+    }
+
+    public static void registerAllCommands() {
+        registerCircleCommand();
+        registerSphereCommand();
+    }
 }

src/client/java/com/tpexcotblu/litematicgen/generators/SphereGenerator.java

@@ -0,0 +1,63 @@
+package com.tpexcotblu.litematicgen.generators;
+
+import net.minecraft.nbt.NbtCompound;
+import net.minecraft.nbt.NbtInt;
+import net.minecraft.nbt.NbtList;
+
+public class SphereGenerator implements SchemeGenerator {
+    private final int radius;
+
+    public SphereGenerator(int radius) {
+        this.radius = radius;
+    }
+
+    @Override
+    public NbtList generateBlocks() {
+        int size = radius * 2 + 1;
+        NbtList blocks = new NbtList();
+
+        for (int y = 0; y < size; y++) {
+            for (int z = 0; z < size; z++) {
+                for (int x = 0; x < size; x++) {
+                    int dx = x - radius;
+                    int dy = y - radius;
+                    int dz = z - radius;
+                    double dist = Math.sqrt(dx * dx + dy * dy + dz * dz);
+
+                    if (Math.abs(dist - radius) < 0.5) {
+                        NbtCompound block = new NbtCompound();
+                        NbtList posList = new NbtList();
+                        posList.add(NbtInt.of(x));
+                        posList.add(NbtInt.of(y));
+                        posList.add(NbtInt.of(z));
+                        block.put("pos", posList);
+                        block.putInt("state", 1);
+                        blocks.add(block);
+                    }
+                }
+            }
+        }
+
+        return blocks;
+    }
+
+    @Override
+    public NbtList generateSize() {
+        int size = radius * 2 + 1;
+        NbtList sizeList = new NbtList();
+        sizeList.add(NbtInt.of(size));
+        sizeList.add(NbtInt.of(size));
+        sizeList.add(NbtInt.of(size));
+        return sizeList;
+    }
+
+    @Override
+    public String getFilename() {
+        return "sphere_" + radius + ".nbt";
+    }
+
+    @Override
+    public String getSuccessMessage() {
+        return "The sphere is saved";
+    }
+}