254 lines
17 KiB
Java
254 lines
17 KiB
Java
package org.betterx.betterend.util;
|
|
|
|
import net.minecraft.advancements.critereon.EntityPredicate;
|
|
import net.minecraft.advancements.critereon.FishingHookPredicate;
|
|
import net.minecraft.advancements.critereon.LocationPredicate;
|
|
import net.minecraft.world.level.Level;
|
|
import net.minecraft.world.level.storage.loot.BuiltInLootTables;
|
|
import net.minecraft.world.level.storage.loot.LootContext;
|
|
import net.minecraft.world.level.storage.loot.entries.*;
|
|
import net.minecraft.world.level.storage.loot.functions.EnchantWithLevelsFunction;
|
|
import net.minecraft.world.level.storage.loot.functions.SetItemDamageFunction;
|
|
import net.minecraft.world.level.storage.loot.predicates.LocationCheck;
|
|
import net.minecraft.world.level.storage.loot.predicates.LootItemCondition;
|
|
import net.minecraft.world.level.storage.loot.predicates.LootItemEntityPropertyCondition;
|
|
import org.betterx.bclib.BCLib;
|
|
import org.betterx.bclib.api.v2.levelgen.biomes.BCLBiome;
|
|
import org.betterx.bclib.api.v2.levelgen.biomes.BCLBiomeRegistry;
|
|
import org.betterx.bclib.api.v2.levelgen.biomes.BiomeAPI;
|
|
import org.betterx.bclib.complexmaterials.WoodenComplexMaterial;
|
|
import org.betterx.betterend.BetterEnd;
|
|
import org.betterx.betterend.registry.EndBiomes;
|
|
import org.betterx.betterend.registry.EndBlocks;
|
|
import org.betterx.betterend.registry.EndItems;
|
|
|
|
import net.minecraft.core.Holder;
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.world.item.Items;
|
|
import net.minecraft.world.level.biome.Biome;
|
|
import net.minecraft.world.level.block.Blocks;
|
|
import net.minecraft.world.level.storage.loot.LootPool;
|
|
import net.minecraft.world.level.storage.loot.LootTable;
|
|
import net.minecraft.world.level.storage.loot.predicates.LootItemRandomChanceCondition;
|
|
import net.minecraft.world.level.storage.loot.providers.number.ConstantValue;
|
|
import net.minecraft.world.level.storage.loot.providers.number.UniformGenerator;
|
|
|
|
import net.fabricmc.fabric.api.loot.v2.LootTableEvents;
|
|
import org.betterx.betterend.world.biome.EndBiome;
|
|
|
|
public class LootTableUtil {
|
|
private static final ResourceLocation END_CITY_TREASURE_ID = new ResourceLocation("chests/end_city_treasure");
|
|
private static final ResourceLocation COMMON = BetterEnd.makeID("chests/common");
|
|
private static final ResourceLocation FOGGY_MUSHROOMLAND = BetterEnd.makeID("chests/foggy_mushroomland");
|
|
private static final ResourceLocation CHORUS_FOREST = BetterEnd.makeID("chests/chorus_forest");
|
|
private static final ResourceLocation SHADOW_FOREST = BetterEnd.makeID("chests/shadow_forest");
|
|
private static final ResourceLocation LANTERN_WOODS = BetterEnd.makeID("chests/lantern_woods");
|
|
private static final ResourceLocation UMBRELLA_JUNGLE = BetterEnd.makeID("chests/umbrella_jungle");
|
|
private static final ResourceLocation FISHING_FISH = BetterEnd.makeID("gameplay/fishing/fish");
|
|
private static final ResourceLocation FISHING_TREASURE = BetterEnd.makeID("gameplay/fishing/treasure");
|
|
private static final ResourceLocation FISHING_JUNK = BetterEnd.makeID("gameplay/fishing/junk");
|
|
|
|
private static final LootItemCondition.Builder IN_END
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setDimension(Level.END));
|
|
private static final LootItemCondition.Builder IN_FOGGY_MUSHROOMLAND
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.FOGGY_MUSHROOMLAND.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_CHORUS_FOREST
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.CHORUS_FOREST.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_AMBER_LAND
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.AMBER_LAND.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_GLOWING_GRASSLANDS
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.GLOWING_GRASSLANDS.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_LANTERN_WOODS
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.LANTERN_WOODS.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_MEGALAKE
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.MEGALAKE.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_MEGALAKE_GROVE
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.MEGALAKE_GROVE.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_NEON_OASIS
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.NEON_OASIS.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_SHADOW_FOREST
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.SHADOW_FOREST.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_SULPHUR_SPRINGS
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.SULPHUR_SPRINGS.getBiomeKey()));
|
|
private static final LootItemCondition.Builder IN_UMBRELLA_JUNGLE
|
|
= LocationCheck.checkLocation(LocationPredicate.Builder.location().setBiome(EndBiomes.UMBRELLA_JUNGLE.getBiomeKey()));
|
|
|
|
public static void init() {
|
|
LootTableEvents.MODIFY.register((resourceManager, lootManager, id, table, setter) -> {
|
|
if (END_CITY_TREASURE_ID.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(ConstantValue.exactly(1));
|
|
builder.when(LootItemRandomChanceCondition.randomChance(0.2f));
|
|
builder.add(LootItem.lootTableItem(Items.GHAST_TEAR));
|
|
table.withPool(builder);
|
|
|
|
builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(0, 3));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_STRANGE_AND_ALIEN));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_GRASPING_AT_STARS));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_ENDSEEKER));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_EO_DRACONA));
|
|
table.withPool(builder);
|
|
} else if (id.getNamespace().equals(BetterEnd.MOD_ID)) {
|
|
if (FISHING_FISH.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool()
|
|
.add(LootItem.lootTableItem(EndItems.END_FISH_RAW));
|
|
table.withPool(builder);
|
|
return;
|
|
} else if (FISHING_JUNK.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool()
|
|
.add(LootItem.lootTableItem(EndItems.END_LILY_LEAF))
|
|
.add(LootItem.lootTableItem(Items.ENDER_PEARL))
|
|
.add(LootItem.lootTableItem(Items.CHORUS_FRUIT))
|
|
.add(LootItem.lootTableItem(EndItems.GELATINE))
|
|
.add(LootItem.lootTableItem(EndItems.CRYSTAL_SHARDS));
|
|
addCharnia(builder);
|
|
table.withPool(builder);
|
|
return;
|
|
} else if (FISHING_TREASURE.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool()
|
|
.add(LootItem.lootTableItem(EndItems.AETERNIUM_SWORD_BLADE))
|
|
.add(LootItem.lootTableItem(EndItems.AETERNIUM_FORGED_PLATE))
|
|
.add(LootItem.lootTableItem(EndBlocks.MENGER_SPONGE))
|
|
.add(LootItem.lootTableItem(Items.BOW)
|
|
.apply(SetItemDamageFunction.setDamage(UniformGenerator.between(0.0F, 0.25F)))
|
|
.apply(EnchantWithLevelsFunction.enchantWithLevels(ConstantValue.exactly(30.0F)).allowTreasure()))
|
|
.add(LootItem.lootTableItem(Items.FISHING_ROD)
|
|
.apply(SetItemDamageFunction.setDamage(UniformGenerator.between(0.0F, 0.25F)))
|
|
.apply(EnchantWithLevelsFunction.enchantWithLevels(ConstantValue.exactly(30.0F)).allowTreasure()))
|
|
.add(LootItem.lootTableItem(Items.BOOK)
|
|
.apply(EnchantWithLevelsFunction.enchantWithLevels(ConstantValue.exactly(30.0F)).allowTreasure()));
|
|
table.withPool(builder);
|
|
return;
|
|
}
|
|
addCommonItems(table);
|
|
if (FOGGY_MUSHROOMLAND.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.MOSSY_GLOWSHROOM.getBlock(WoodenComplexMaterial.BLOCK_PLANKS)));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.MOSSY_GLOWSHROOM_SAPLING));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.BLUE_VINE_SEED));
|
|
table.withPool(builder);
|
|
} else if (CHORUS_FOREST.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.PYTHADENDRON.getBlock(WoodenComplexMaterial.BLOCK_PLANKS)));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.PYTHADENDRON_SAPLING));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.CHORUS_MUSHROOM));
|
|
table.withPool(builder);
|
|
} else if (SHADOW_FOREST.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.DRAGON_TREE.getBlock(WoodenComplexMaterial.BLOCK_PLANKS)));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.DRAGON_TREE_SAPLING));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.SHADOW_BERRY));
|
|
builder.add(LootItem.lootTableItem(EndItems.SHADOW_BERRY_RAW));
|
|
table.withPool(builder);
|
|
} else if (LANTERN_WOODS.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.LUCERNIA.getBlock(WoodenComplexMaterial.BLOCK_PLANKS)));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.LUCERNIA_SAPLING));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.BOLUX_MUSHROOM));
|
|
table.withPool(builder);
|
|
} else if (UMBRELLA_JUNGLE.equals(id)) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.UMBRELLA_TREE.getBlock(WoodenComplexMaterial.BLOCK_PLANKS)));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.UMBRELLA_TREE_SAPLING));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.SMALL_JELLYSHROOM));
|
|
table.withPool(builder);
|
|
}
|
|
}
|
|
});
|
|
|
|
LootTableEvents.REPLACE.register((resourceManager, lootManager, id, originalTable, setter) -> {
|
|
if (BuiltInLootTables.FISHING.equals(id)) {
|
|
LootTable.Builder builder = LootTable.lootTable()
|
|
.withPool(LootPool.lootPool().when(IN_END.invert()).setRolls(ConstantValue.exactly(1.0F))
|
|
.add(LootTableReference.lootTableReference(BuiltInLootTables.FISHING_JUNK).setWeight(10).setQuality(-2))
|
|
.add(LootTableReference.lootTableReference(BuiltInLootTables.FISHING_TREASURE).setWeight(5).setQuality(2)
|
|
.when(LootItemEntityPropertyCondition.hasProperties(LootContext.EntityTarget.THIS,
|
|
EntityPredicate.Builder.entity().subPredicate(FishingHookPredicate.inOpenWater(true)))))
|
|
.add(LootTableReference.lootTableReference(BuiltInLootTables.FISHING_FISH).setWeight(85).setQuality(-1)))
|
|
.withPool(LootPool.lootPool().when(IN_END).setRolls(ConstantValue.exactly(1.0F))
|
|
.add(LootTableReference.lootTableReference(FISHING_FISH).setWeight(85).setQuality(-1))
|
|
.add(LootTableReference.lootTableReference(FISHING_TREASURE).setWeight(5).setQuality(2))
|
|
.add(LootTableReference.lootTableReference(FISHING_JUNK).setWeight(10).setQuality(-2)));
|
|
return builder.build();
|
|
}
|
|
return null;
|
|
});
|
|
}
|
|
|
|
public static ResourceLocation getTable(Holder<Biome> biome) {
|
|
BCLBiome bclBiome = BiomeAPI.getBiome(biome.value());
|
|
if (bclBiome == EndBiomes.FOGGY_MUSHROOMLAND) {
|
|
return FOGGY_MUSHROOMLAND;
|
|
} else if (bclBiome == EndBiomes.CHORUS_FOREST) {
|
|
return CHORUS_FOREST;
|
|
} else if (bclBiome == EndBiomes.SHADOW_FOREST) {
|
|
return SHADOW_FOREST;
|
|
} else if (bclBiome == EndBiomes.LANTERN_WOODS) {
|
|
return LANTERN_WOODS;
|
|
} else if (bclBiome == EndBiomes.UMBRELLA_JUNGLE) {
|
|
return UMBRELLA_JUNGLE;
|
|
}
|
|
return COMMON;
|
|
}
|
|
|
|
private static void addCharnia(LootPool.Builder pool) {
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_CYAN)
|
|
.when(IN_GLOWING_GRASSLANDS.or(IN_MEGALAKE).or(IN_MEGALAKE_GROVE).or(IN_NEON_OASIS)));
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_LIGHT_BLUE)
|
|
.when(IN_FOGGY_MUSHROOMLAND.or(IN_GLOWING_GRASSLANDS).or(IN_MEGALAKE).or(IN_MEGALAKE_GROVE).or(IN_UMBRELLA_JUNGLE)));
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_GREEN)
|
|
.when(IN_GLOWING_GRASSLANDS.or(IN_NEON_OASIS).or(IN_SULPHUR_SPRINGS).or(IN_UMBRELLA_JUNGLE)));
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_RED)
|
|
.when(IN_AMBER_LAND.or(IN_LANTERN_WOODS).or(IN_NEON_OASIS)));
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_ORANGE)
|
|
.when(IN_AMBER_LAND.or(IN_LANTERN_WOODS).or(IN_SULPHUR_SPRINGS)));
|
|
pool.add(LootItem.lootTableItem(EndBlocks.CHARNIA_PURPLE)
|
|
.when(IN_CHORUS_FOREST.or(IN_SHADOW_FOREST)));
|
|
}
|
|
|
|
private static void addCommonItems(LootTable.Builder table) {
|
|
LootPool.Builder builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(0, 2));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_STRANGE_AND_ALIEN));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_GRASPING_AT_STARS));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_ENDSEEKER));
|
|
builder.add(LootItem.lootTableItem(EndItems.MUSIC_DISC_EO_DRACONA));
|
|
table.withPool(builder);
|
|
|
|
builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(4, 8));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.ingot));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.rawOre));
|
|
builder.add(LootItem.lootTableItem(Items.ENDER_PEARL));
|
|
table.withPool(builder);
|
|
|
|
builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(2, 4));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.TERMINITE.ingot));
|
|
builder.add(LootItem.lootTableItem(EndItems.ENDER_SHARD));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.AURORA_CRYSTAL));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.axe));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.pickaxe));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.hoe));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.sword));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.THALLASIUM.shovel));
|
|
builder.add(LootItem.lootTableItem(Items.ENDER_EYE));
|
|
builder.add(LootItem.lootTableItem(Blocks.OBSIDIAN));
|
|
table.withPool(builder);
|
|
|
|
builder = LootPool.lootPool();
|
|
builder.setRolls(UniformGenerator.between(0, 4));
|
|
builder.add(LootItem.lootTableItem(EndBlocks.FLAVOLITE_RUNED));
|
|
builder.add(LootItem.lootTableItem(EndItems.AETERNIUM_INGOT));
|
|
builder.add(LootItem.lootTableItem(EndItems.AMBER_GEM));
|
|
builder.add(LootItem.lootTableItem(Items.END_CRYSTAL));
|
|
builder.add(LootItem.lootTableItem(Items.GHAST_TEAR));
|
|
table.withPool(builder);
|
|
}
|
|
}
|