feat(mod.datagen): HBlockTagsProvider

auioc · Oct 6, 2022 · 93b98ab · 93b98ab
1 parent 6ea3abd
commit 93b98ab
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 2 deletions.
diff --git a/src/main/java/org/auioc/mcmod/arnicalib/mod/datagen/AHDataGenerators.java b/src/main/java/org/auioc/mcmod/arnicalib/mod/datagen/AHDataGenerators.java
@@ -1,5 +1,6 @@
 package org.auioc.mcmod.arnicalib.mod.datagen;
 
+import org.auioc.mcmod.arnicalib.mod.datagen.provider.HBlockTagsProvider;
 import org.auioc.mcmod.arnicalib.mod.datagen.provider.HEntityTypeTagsProvider;
 import net.minecraftforge.eventbus.api.SubscribeEvent;
 import net.minecraftforge.fml.common.Mod;
@@ -13,9 +14,11 @@ public static void dataGen(GatherDataEvent event) {
         var generator = event.getGenerator();
         var existingFileHelper = event.getExistingFileHelper();
 
-        generator.addProvider(new HEntityTypeTagsProvider(generator, existingFileHelper));
-        // var blockTags = new HBlockTagsProvider(generator, existingFileHelper);
+        var blockTags = new HBlockTagsProvider(generator, existingFileHelper);
+        generator.addProvider(blockTags);
         // generator.addProvider(new HItemTagsProvider(generator, blockTags, existingFileHelper));
+
+        generator.addProvider(new HEntityTypeTagsProvider(generator, existingFileHelper));
     }
 
 }
diff --git a/src/main/java/org/auioc/mcmod/arnicalib/mod/datagen/provider/HBlockTagsProvider.java b/src/main/java/org/auioc/mcmod/arnicalib/mod/datagen/provider/HBlockTagsProvider.java
@@ -0,0 +1,41 @@
+package org.auioc.mcmod.arnicalib.mod.datagen.provider;
+
+import javax.annotation.Nonnull;
+import javax.annotation.Nullable;
+import org.auioc.mcmod.arnicalib.ArnicaLib;
+import org.auioc.mcmod.arnicalib.game.datagen.tag.IHTagsProvider;
+import org.auioc.mcmod.arnicalib.game.tag.HBlockTags;
+import net.minecraft.data.DataGenerator;
+import net.minecraft.data.tags.BlockTagsProvider;
+import net.minecraft.world.level.block.Block;
+import net.minecraftforge.common.data.ExistingFileHelper;
+import net.minecraftforge.registries.ForgeRegistries;
+import net.minecraftforge.registries.IForgeRegistry;
+
+public class HBlockTagsProvider extends BlockTagsProvider implements IHTagsProvider<Block> {
+
+    public HBlockTagsProvider(DataGenerator generator, @Nullable ExistingFileHelper existingFileHelper) {
+        super(generator, ArnicaLib.MOD_ID, existingFileHelper);
+    }
+
+    @Override
+    public String getName() {
+        return "HBlocksTags";
+    }
+
+    @Nonnull
+    @Override
+    public IForgeRegistry<Block> getRegistry() {
+        return ForgeRegistries.BLOCKS;
+    }
+
+    @Override
+    @SuppressWarnings("deprecation")
+    protected void addTags() {
+        addFromRegistry(tag(HBlockTags.INSTABREAKABLE), (b) -> b.defaultDestroyTime() == 0.0F);
+        addFromRegistry(tag(HBlockTags.RANDOMLY_TICKABLE), (b) -> b.defaultBlockState().isRandomlyTicking());
+        addFromRegistry(tag(HBlockTags.LIGHT), (b) -> b.defaultBlockState().getLightEmission() > 0);
+    }
+
+}
+