From 6cc0322aacd36bbf325d3e08b7fa0b76d2eb0e5a Mon Sep 17 00:00:00 2001 From: shedaniel Date: Tue, 16 Apr 2024 18:32:47 +0900 Subject: Add anvil recipes, Close #1381 --- .../rei/plugin/client/DefaultClientPlugin.java | 63 ++++++++++++++++++++++ .../categories/anvil/DefaultAnvilCategory.java | 16 +++++- .../plugin/common/displays/anvil/AnvilRecipe.java | 11 ++++ .../common/displays/anvil/DefaultAnvilDisplay.java | 55 ++++++++++++++++--- 4 files changed, 137 insertions(+), 8 deletions(-) (limited to 'default-plugin/src/main') diff --git a/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/DefaultClientPlugin.java b/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/DefaultClientPlugin.java index 239b95e2e..b3b94b9e3 100644 --- a/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/DefaultClientPlugin.java +++ b/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/DefaultClientPlugin.java @@ -96,6 +96,7 @@ import net.minecraft.world.item.alchemy.Potion; import net.minecraft.world.item.alchemy.PotionBrewing; import net.minecraft.world.item.alchemy.PotionUtils; import net.minecraft.world.item.crafting.*; +import net.minecraft.world.item.enchantment.EnchantmentInstance; import net.minecraft.world.level.GameType; import net.minecraft.world.level.block.Block; import net.minecraft.world.level.block.ComposterBlock; @@ -104,11 +105,13 @@ import net.minecraft.world.level.block.entity.AbstractFurnaceBlockEntity; import net.minecraft.world.level.block.state.BlockState; import net.minecraft.world.level.material.Fluid; import net.minecraft.world.level.material.FluidState; +import org.apache.commons.lang3.tuple.Pair; import org.jetbrains.annotations.ApiStatus; import java.util.*; import java.util.function.Supplier; import java.util.function.UnaryOperator; +import java.util.stream.IntStream; import java.util.stream.Stream; @Environment(EnvType.CLIENT) @@ -355,6 +358,66 @@ public class DefaultClientPlugin implements REIClientPlugin, BuiltinClientPlugin registerForgePotions(registry, this); } + for (Item item : Registry.ITEM) { + ItemStack stack = item.getDefaultInstance(); + if (!stack.isDamageableItem()) continue; + EntryIngredient repairMaterialBase = null; + if (item instanceof TieredItem tieredItem) { + Tier tier = tieredItem.getTier(); + repairMaterialBase = EntryIngredients.ofIngredient(tier.getRepairIngredient()); + } else if (item instanceof ArmorItem armorItem) { + ArmorMaterial material = armorItem.getMaterial(); + repairMaterialBase = EntryIngredients.ofIngredient(material.getRepairIngredient()); + } else if (item instanceof ShieldItem shieldItem) { + repairMaterialBase = EntryIngredients.ofItemTag(ItemTags.PLANKS); + repairMaterialBase.filter(s -> shieldItem.isValidRepairItem(stack, s.castValue())); + } else if (item instanceof ElytraItem elytraItem) { + repairMaterialBase = EntryIngredients.of(Items.PHANTOM_MEMBRANE); + repairMaterialBase.filter(s -> elytraItem.isValidRepairItem(stack, s.castValue())); + } + if (repairMaterialBase == null || repairMaterialBase.isEmpty()) continue; + for (int[] i = {1}; i[0] <= 4; i[0]++) { + ItemStack baseStack = item.getDefaultInstance(); + int toRepair = i[0] == 4 ? item.getMaxDamage() : baseStack.getMaxDamage() / 4 * i[0]; + baseStack.setDamageValue(toRepair); + EntryIngredient repairMaterial = repairMaterialBase.map(s -> { + EntryStack newStack = s.copy(); + newStack.castValue().setCount(i[0]); + return newStack; + }); + Optional> output = DefaultAnvilDisplay.calculateOutput(baseStack, repairMaterial.get(0).castValue()); + if (output.isEmpty()) continue; + registry.add(new DefaultAnvilDisplay(List.of(EntryIngredients.of(baseStack), repairMaterial), + Collections.singletonList(EntryIngredients.of(output.get().getLeft())), Optional.empty(), OptionalInt.of(output.get().getRight()))); + } + } + List> enchantmentBooks = Registry.ENCHANTMENT.stream() + .flatMap(enchantment -> { + if (enchantment.getMaxLevel() - enchantment.getMinLevel() >= 10) { + return IntStream.of(enchantment.getMinLevel(), enchantment.getMaxLevel()) + .mapToObj(lvl -> new EnchantmentInstance(enchantment, lvl)); + } else { + return IntStream.rangeClosed(enchantment.getMinLevel(), enchantment.getMaxLevel()) + .mapToObj(lvl -> new EnchantmentInstance(enchantment, lvl)); + } + }) + .map(instance -> { + return Pair.of(instance, EnchantedBookItem.createForEnchantment(instance)); + }) + .toList(); + EntryRegistry.getInstance().getEntryStacks().forEach(stack -> { + if (stack.getType() != VanillaEntryTypes.ITEM) return; + ItemStack itemStack = stack.castValue(); + if (!itemStack.isEnchantable()) return; + for (Pair pair : enchantmentBooks) { + if (!pair.getKey().enchantment.canEnchant(itemStack)) continue; + Optional> output = DefaultAnvilDisplay.calculateOutput(itemStack, pair.getValue()); + if (output.isEmpty()) continue; + registry.add(new DefaultAnvilDisplay(List.of(EntryIngredients.of(itemStack), EntryIngredients.of(pair.getValue())), + Collections.singletonList(EntryIngredients.of(output.get().getLeft())), Optional.empty(), OptionalInt.of(output.get().getRight()))); + } + }); + for (Registry reg : Registry.REGISTRY) { reg.getTags().forEach(tagPair -> registry.add(tagPair.getFirst())); } diff --git a/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/categories/anvil/DefaultAnvilCategory.java b/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/categories/anvil/DefaultAnvilCategory.java index f8c16278e..efcd41b1f 100644 --- a/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/categories/anvil/DefaultAnvilCategory.java +++ b/default-plugin/src/main/java/me/shedaniel/rei/plugin/client/categories/anvil/DefaultAnvilCategory.java @@ -34,6 +34,9 @@ import me.shedaniel.rei.api.common.category.CategoryIdentifier; import me.shedaniel.rei.api.common.util.EntryStacks; import me.shedaniel.rei.plugin.common.BuiltinPlugin; import me.shedaniel.rei.plugin.common.displays.anvil.DefaultAnvilDisplay; +import net.minecraft.client.Minecraft; +import net.minecraft.client.gui.Font; +import net.minecraft.client.gui.GuiComponent; import net.minecraft.network.chat.Component; import net.minecraft.world.level.block.Blocks; @@ -57,7 +60,7 @@ public class DefaultAnvilCategory implements DisplayCategory setupDisplay(DefaultAnvilDisplay display, Rectangle bounds) { - Point startPoint = new Point(bounds.getCenterX() - 31, bounds.getCenterY() - 13); + Point startPoint = new Point(bounds.getCenterX() - 31, bounds.getCenterY() - (display.getCost().isPresent() ? 20 : 13)); List widgets = Lists.newArrayList(); widgets.add(Widgets.createRecipeBase(bounds)); widgets.add(Widgets.createArrow(new Point(startPoint.x + 27, startPoint.y + 4))); @@ -65,11 +68,20 @@ public class DefaultAnvilCategory implements DisplayCategory { + Font font = Minecraft.getInstance().font; + Component component = Component.translatable("container.repair.cost", display.getCost().getAsInt()); + int x = startPoint.x + 102 - font.width(component) - 2; + GuiComponent.fill(matrices, x - 2, startPoint.y + 28, startPoint.x + 102, startPoint.y + 28 + 12, 0x4f000000); + font.drawShadow(matrices, component, x, startPoint.y + 28 + 2, 0x80ff20); + })); + } return widgets; } @Override public int getDisplayHeight() { - return 36; + return 48; } } diff --git a/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/AnvilRecipe.java b/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/AnvilRecipe.java index ba4f70458..694868145 100644 --- a/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/AnvilRecipe.java +++ b/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/AnvilRecipe.java @@ -28,6 +28,7 @@ import net.minecraft.world.item.ItemStack; import org.jetbrains.annotations.Nullable; import java.util.List; +import java.util.OptionalInt; public class AnvilRecipe { @Nullable @@ -35,12 +36,18 @@ public class AnvilRecipe { private final List leftInput; private final List rightInputs; private final List outputs; + private final OptionalInt cost; public AnvilRecipe(@Nullable ResourceLocation id, List leftInput, List rightInputs, List outputs) { + this(id, leftInput, rightInputs, outputs, OptionalInt.empty()); + } + + public AnvilRecipe(@Nullable ResourceLocation id, List leftInput, List rightInputs, List outputs, OptionalInt cost) { this.id = id; this.leftInput = leftInput; this.rightInputs = rightInputs; this.outputs = outputs; + this.cost = cost; } public ResourceLocation getId() { @@ -58,4 +65,8 @@ public class AnvilRecipe { public List getOutputs() { return outputs; } + + public OptionalInt getCost() { + return cost; + } } diff --git a/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/DefaultAnvilDisplay.java b/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/DefaultAnvilDisplay.java index 0c2b5e655..befc8d143 100644 --- a/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/DefaultAnvilDisplay.java +++ b/default-plugin/src/main/java/me/shedaniel/rei/plugin/common/displays/anvil/DefaultAnvilDisplay.java @@ -28,14 +28,20 @@ import me.shedaniel.rei.api.common.display.basic.BasicDisplay; import me.shedaniel.rei.api.common.entry.EntryIngredient; import me.shedaniel.rei.api.common.util.EntryIngredients; import me.shedaniel.rei.plugin.common.BuiltinPlugin; +import net.minecraft.client.Minecraft; +import net.minecraft.nbt.CompoundTag; import net.minecraft.resources.ResourceLocation; +import net.minecraft.world.entity.player.Inventory; +import net.minecraft.world.inventory.AnvilMenu; +import net.minecraft.world.item.ItemStack; +import org.apache.commons.lang3.tuple.Pair; +import org.jetbrains.annotations.ApiStatus; -import java.util.Arrays; -import java.util.Collections; -import java.util.List; -import java.util.Optional; +import java.util.*; public class DefaultAnvilDisplay extends BasicDisplay { + private final OptionalInt cost; + public DefaultAnvilDisplay(AnvilRecipe recipe) { this( Arrays.asList( @@ -43,12 +49,22 @@ public class DefaultAnvilDisplay extends BasicDisplay { EntryIngredients.ofItemStacks(recipe.getRightInputs()) ), Collections.singletonList(EntryIngredients.ofItemStacks(recipe.getOutputs())), - Optional.ofNullable(recipe.getId()) + Optional.ofNullable(recipe.getId()), + recipe.getCost() ); } public DefaultAnvilDisplay(List inputs, List outputs, Optional location) { + this(inputs, outputs, location, OptionalInt.empty()); + } + + public DefaultAnvilDisplay(List inputs, List outputs, Optional location, CompoundTag tag) { + this(inputs, outputs, location, tag.contains("Cost") ? OptionalInt.of(tag.getInt("Cost")) : OptionalInt.empty()); + } + + public DefaultAnvilDisplay(List inputs, List outputs, Optional location, OptionalInt cost) { super(inputs, outputs, location); + this.cost = cost; } @Override @@ -56,7 +72,34 @@ public class DefaultAnvilDisplay extends BasicDisplay { return BuiltinPlugin.ANVIL; } + public OptionalInt getCost() { + return cost; + } + public static BasicDisplay.Serializer serializer() { - return BasicDisplay.Serializer.ofSimple(DefaultAnvilDisplay::new); + return BasicDisplay.Serializer.of(DefaultAnvilDisplay::new, (display, tag) -> { + if (display.getCost().isPresent()) { + tag.putInt("Cost", display.getCost().getAsInt()); + } + }); + } + + @ApiStatus.Experimental + @ApiStatus.Internal + public static Optional> calculateOutput(ItemStack left, ItemStack right) { + try { + if (Minecraft.getInstance().player == null) return Optional.empty(); + AnvilMenu menu = new AnvilMenu(0, new Inventory(Minecraft.getInstance().player)); + menu.setItem(0, menu.incrementStateId(), left); + menu.setItem(1, menu.incrementStateId(), right); + ItemStack output = menu.getSlot(2).getItem().copy(); + if (!output.isEmpty()) { + return Optional.of(Pair.of(output, menu.getCost())); + } else { + return Optional.empty(); + } + } catch (Throwable ignored) { + return Optional.empty(); + } } } -- cgit