optimized loot provider

Author: DevDyna

src/main/java/com/devdyna/modname_id/datagen/Controller.java

@@ -4,18 +4,21 @@
 
 import java.util.concurrent.CompletableFuture;
 
+import java.util.List;
+import java.util.Set;
+import net.minecraft.data.PackOutput;
+import net.minecraft.data.loot.LootTableProvider;
+import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
 import com.devdyna.modname_id.datagen.client.DataBlockModelState;
 import com.devdyna.modname_id.datagen.client.DataItemModel;
 import com.devdyna.modname_id.datagen.client.DataLang;
 import com.devdyna.modname_id.datagen.server.DataBlockTag;
 import com.devdyna.modname_id.datagen.server.DataItemTag;
 import com.devdyna.modname_id.datagen.server.DataLoot;
 import com.devdyna.modname_id.datagen.server.DataRecipe;
-
 import net.minecraft.core.HolderLookup;
 import net.minecraft.data.DataGenerator;
 import net.minecraft.data.DataProvider;
-import net.minecraft.data.PackOutput;
 import net.neoforged.bus.api.SubscribeEvent;
 import net.neoforged.fml.common.EventBusSubscriber;
 import net.neoforged.neoforge.common.data.ExistingFileHelper;
@@ -40,7 +43,11 @@ public static void gatherData(GatherDataEvent e) {
         DataBlockTag blocktag = new DataBlockTag(po, pr, f);
         providerGen(e, g, blocktag);
         providerGen(e, g, new DataItemTag(po, pr, blocktag.contentsGetter()));
-        providerGen(e, g, new DataLoot(po, pr));
+        providerGen(e, g, new LootTableProvider(po, Set.of(),
+                List.of(
+                        // list of any loot table entries inside ./server/tables/
+                        new LootTableProvider.SubProviderEntry(DataLoot::new, LootContextParamSets.BLOCK)),
+                pr));
         providerGen(e, g, new DataRecipe(po, pr));
 
     }

src/main/java/com/devdyna/modname_id/datagen/server/DataLoot.java

@@ -1,24 +1,43 @@
 package com.devdyna.modname_id.datagen.server;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Set;
-import java.util.concurrent.CompletableFuture;
 
-import com.devdyna.modname_id.datagen.server.tables.BlockDrop;
+import com.devdyna.modname_id.init.types.zBlocks;
 
-import net.minecraft.core.HolderLookup.Provider;
-import net.minecraft.data.PackOutput;
-import net.minecraft.data.loot.LootTableProvider;
-import net.minecraft.world.level.storage.loot.parameters.LootContextParamSets;
+import net.minecraft.core.HolderLookup;
+import net.minecraft.data.loot.BlockLootSubProvider;
+import net.minecraft.world.flag.FeatureFlags;
+import net.minecraft.world.level.block.Block;
+import net.neoforged.neoforge.registries.DeferredHolder;
+import net.neoforged.neoforge.registries.DeferredRegister;
 
-public class DataLoot extends LootTableProvider {
+public class DataLoot extends BlockLootSubProvider {
 
-    public DataLoot(PackOutput o, CompletableFuture<Provider> r) {
-        super(o, Set.of(),
-                List.of(
-                    //list of any loot table entries inside ./server/tables/
-                        new LootTableProvider.SubProviderEntry(BlockDrop::new, LootContextParamSets.BLOCK)),
-                r);
-    }
+        public DataLoot(HolderLookup.Provider l) {
+                super(Set.of(), FeatureFlags.DEFAULT_FLAGS, l);
+        }
+
+        @Override
+        protected Iterable<Block> getKnownBlocks() {
+                List<Block> blocks = new ArrayList<>();
+                blocks.addAll(getList(zBlocks.zBlock));
+                blocks.addAll(getList(zBlocks.zBlockItem));
+                return blocks;
+        }
+
+        @SuppressWarnings("unchecked")
+        private List<Block> getList(DeferredRegister.Blocks c) {
+                return (List<Block>) c.getEntries().stream().map(DeferredHolder::get).toList();
+        }
+
+        @Override
+        protected void generate() {
+                // dropSelf(Blocks.BK.get());
+
+                // Blocks.zBlock.getEntries().forEach(b -> dropSelf(b.get()));
+
+        }
 
 }