添加方块战利品表

Anvil-Dev · Mar 13, 2024 · e136cd0 · e136cd0
1 parent a44d568
commit e136cd0
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/src/main/java/dev/dubhe/anvilcraft/data/generator/AnvilCraftDataGenerator.java b/src/main/java/dev/dubhe/anvilcraft/data/generator/AnvilCraftDataGenerator.java
@@ -2,13 +2,15 @@
 
 import net.fabricmc.fabric.api.datagen.v1.DataGeneratorEntrypoint;
 import net.fabricmc.fabric.api.datagen.v1.FabricDataGenerator;
+import org.jetbrains.annotations.NotNull;
 
 public class AnvilCraftDataGenerator implements DataGeneratorEntrypoint {
     @Override
-    public void onInitializeDataGenerator(FabricDataGenerator fabricDataGenerator) {
+    public void onInitializeDataGenerator(@NotNull FabricDataGenerator fabricDataGenerator) {
         FabricDataGenerator.Pack pack = fabricDataGenerator.createPack();
         pack.addProvider(MyBlockTagGenerator::new);
         pack.addProvider(MyItemTagGenerator::new);
         pack.addProvider(MyRecipesGenerator::new);
+        pack.addProvider(MyBlockLootGenerator::new);
     }
 }
diff --git a/src/main/java/dev/dubhe/anvilcraft/data/generator/MyBlockLootGenerator.java b/src/main/java/dev/dubhe/anvilcraft/data/generator/MyBlockLootGenerator.java
@@ -0,0 +1,31 @@
+package dev.dubhe.anvilcraft.data.generator;
+
+import dev.dubhe.anvilcraft.init.ModBlocks;
+import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput;
+import net.fabricmc.fabric.api.datagen.v1.provider.FabricBlockLootTableProvider;
+import net.minecraft.world.item.Items;
+import net.minecraft.world.level.block.Block;
+import net.minecraft.world.level.block.Blocks;
+
+public class MyBlockLootGenerator extends FabricBlockLootTableProvider {
+
+    protected MyBlockLootGenerator(FabricDataOutput dataOutput) {
+        super(dataOutput);
+    }
+
+    @Override
+    public void generate() {
+        this.dropSelf(ModBlocks.ROYAL_ANVIL);
+        this.dropSelf(ModBlocks.MAGNET_BLOCK);
+        this.dropSelf(ModBlocks.HOLLOW_MAGNET_BLOCK);
+        this.dropSelf(ModBlocks.FERRITE_CORE_MAGNET_BLOCK);
+        this.dropSelf(ModBlocks.INTERACT_MACHINE);
+        this.dropSelf(ModBlocks.AUTO_CRAFTER);
+        this.dropSelf(ModBlocks.ROYAL_STEEL_BLOCK);
+        this.dropSelf(ModBlocks.SMOOTH_ROYAL_STEEL_BLOCK);
+        this.dropSelf(ModBlocks.CUT_ROYAL_STEEL_BLOCK);
+        this.add(ModBlocks.CUT_ROYAL_STEEL_SLAB, this::createSlabItemTable);
+        this.dropSelf(ModBlocks.CUT_ROYAL_STEEL_STAIRS);
+        this.dropOther(ModBlocks.LAVA_CAULDRON, Items.CAULDRON);
+    }
+}