|
| 1 | +package com.tcm.MineTale.datagen; |
| 2 | + |
| 3 | +import com.tcm.MineTale.registry.ModBlocks; |
| 4 | +import com.tcm.MineTale.registry.ModItems; |
| 5 | +import net.fabricmc.fabric.api.datagen.v1.FabricDataOutput; |
| 6 | +import net.fabricmc.fabric.api.datagen.v1.provider.FabricBlockLootTableProvider; |
| 7 | +import net.minecraft.core.HolderLookup; |
| 8 | +import net.minecraft.core.registries.Registries; |
| 9 | +import net.minecraft.world.item.Item; |
| 10 | +import net.minecraft.world.item.enchantment.Enchantment; |
| 11 | +import net.minecraft.world.item.enchantment.Enchantments; |
| 12 | +import net.minecraft.world.level.block.Block; |
| 13 | +import net.minecraft.world.level.storage.loot.LootTable; |
| 14 | +import net.minecraft.world.level.storage.loot.entries.LootItem; |
| 15 | +import net.minecraft.world.level.storage.loot.entries.LootPoolSingletonContainer; |
| 16 | +import net.minecraft.world.level.storage.loot.functions.ApplyBonusCount; |
| 17 | +import net.minecraft.world.level.storage.loot.functions.SetItemCountFunction; |
| 18 | +import net.minecraft.world.level.storage.loot.providers.number.ConstantValue; |
| 19 | +import net.minecraft.world.level.storage.loot.providers.number.UniformGenerator; |
| 20 | + |
| 21 | +import java.util.concurrent.CompletableFuture; |
| 22 | + |
| 23 | +public class ModLootTableProvider extends FabricBlockLootTableProvider { |
| 24 | + public ModLootTableProvider(FabricDataOutput dataOutput, CompletableFuture<HolderLookup.Provider> registryLookup) { |
| 25 | + super(dataOutput, registryLookup); |
| 26 | + } |
| 27 | + |
| 28 | + @Override |
| 29 | + public void generate() { |
| 30 | + ///Block Drops Itself |
| 31 | + //Bug!! Workbenches currently drop items based on the amount of blocks |
| 32 | + //the multiblock is. For example the Furnaces currently drop 4 furnaces. |
| 33 | + dropSelf(ModBlocks.FURNACE_WORKBENCH_BLOCK_T1); |
| 34 | + dropSelf(ModBlocks.FURNACE_WORKBENCH_BLOCK_T1); |
| 35 | + dropSelf(ModBlocks.ARMORERS_WORKBENCH_BLOCK); |
| 36 | + dropSelf(ModBlocks.CAMPFIRE_WORKBENCH_BLOCK); |
| 37 | + dropSelf(ModBlocks.WORKBENCH_WORKBENCH_BLOCK); |
| 38 | + } |
| 39 | + |
| 40 | + |
| 41 | +/// For Ore Drops |
| 42 | +/** Ex: |
| 43 | + * add(ModBlocks.IRON_ORE_SHALE, AverageOreDrops(ModBlocks.IRON_ORE_SHALE, Items.RAW_IRON)); |
| 44 | + * **/ |
| 45 | + public LootTable.Builder AverageOreDrops(Block drop, Item item) { |
| 46 | + HolderLookup.RegistryLookup<Enchantment> impl = this.registries.lookupOrThrow(Registries.ENCHANTMENT); |
| 47 | + return this.createSilkTouchDispatchTable(drop, this.applyExplosionDecay(drop, ((LootPoolSingletonContainer.Builder<?>) |
| 48 | + LootItem.lootTableItem(item).apply(SetItemCountFunction.setCount(UniformGenerator.between(2, 5)))) |
| 49 | + .apply(ApplyBonusCount.addOreBonusCount(impl.getOrThrow(Enchantments.FORTUNE))))); |
| 50 | + } |
| 51 | + |
| 52 | + public LootTable.Builder SingleOreDrops(Block drop, Item item) { |
| 53 | + HolderLookup.RegistryLookup<Enchantment> impl = this.registries.lookupOrThrow(Registries.ENCHANTMENT); |
| 54 | + return this.createSilkTouchDispatchTable(drop, this.applyExplosionDecay(drop, ((LootPoolSingletonContainer.Builder<?>) |
| 55 | + LootItem.lootTableItem(item).apply(SetItemCountFunction.setCount(ConstantValue.exactly(1)))) |
| 56 | + .apply(ApplyBonusCount.addOreBonusCount(impl.getOrThrow(Enchantments.FORTUNE))))); |
| 57 | + } |
| 58 | + |
| 59 | + public LootTable.Builder LightOreDrops(Block drop, Item item) { |
| 60 | + HolderLookup.RegistryLookup<Enchantment> impl = this.registries.lookupOrThrow(Registries.ENCHANTMENT); |
| 61 | + return this.createSilkTouchDispatchTable(drop, this.applyExplosionDecay(drop, ((LootPoolSingletonContainer.Builder<?>) |
| 62 | + LootItem.lootTableItem(item).apply(SetItemCountFunction.setCount(UniformGenerator.between(1, 2)))) |
| 63 | + .apply(ApplyBonusCount.addOreBonusCount(impl.getOrThrow(Enchantments.FORTUNE))))); |
| 64 | + } |
| 65 | +} |
0 commit comments