package com.forgerscraft.screen; import com.forgerscraft.ModScreenHandlers; import com.forgerscraft.recipe.AnvilRecipe; import com.forgerscraft.recipe.AnvilRecipeRegistry; import net.minecraft.entity.player.PlayerEntity; import net.minecraft.entity.player.PlayerInventory; import net.minecraft.inventory.Inventory; import net.minecraft.inventory.SimpleInventory; import net.minecraft.item.ItemStack; import net.minecraft.screen.ScreenHandler; import net.minecraft.screen.ScreenHandlerContext; import net.minecraft.screen.slot.Slot; import java.util.Optional; public class ForgerersAnvilScreenHandler extends ScreenHandler { private final Inventory inventory; private final SimpleInventory outputInventory = new SimpleInventory(1); private final ScreenHandlerContext context; public ForgerersAnvilScreenHandler(int syncId, PlayerInventory playerInventory) { this(syncId, playerInventory, new SimpleInventory(10), ScreenHandlerContext.EMPTY); } public ForgerersAnvilScreenHandler(int syncId, PlayerInventory playerInventory, Inventory inventory, ScreenHandlerContext context) { super(ModScreenHandlers.FORGERERS_ANVIL, syncId); this.inventory = inventory; this.context = context; checkSize(inventory, 10); inventory.onOpen(playerInventory.player); this.addSlot(new HammerSlot(inventory, 0, 8, 35)); for (int row = 0; row < 3; row++) { for (int col = 0; col < 3; col++) { this.addSlot(new Slot(inventory, 1 + row * 3 + col, 44 + col * 18, 17 + row * 18)); } } this.addSlot(new AnvilOutputSlot(this, outputInventory, 0, 116, 35)); for (int row = 0; row < 3; row++) { for (int col = 0; col < 9; col++) { this.addSlot(new Slot(playerInventory, col + row * 9 + 9, 8 + col * 18, 84 + row * 18)); } } for (int col = 0; col < 9; col++) { this.addSlot(new Slot(playerInventory, col, 8 + col * 18, 142)); } updateResult(); } @Override public void onContentChanged(Inventory changedInventory) { super.onContentChanged(changedInventory); if (changedInventory == this.inventory) { updateResult(); } } private void updateResult() { ItemStack[] grid = getGrid(); ItemStack utility = inventory.getStack(0); Optional match = AnvilRecipeRegistry.findMatch(grid, utility); if (match.isPresent()) { outputInventory.setStack(0, match.get().getResult().copy()); } else { outputInventory.setStack(0, ItemStack.EMPTY); } sendContentUpdates(); } public void onOutputTaken(PlayerEntity player, ItemStack taken) { ItemStack[] grid = getGrid(); ItemStack utility = inventory.getStack(0); Optional matchOpt = AnvilRecipeRegistry.findMatch(grid, utility); if (matchOpt.isEmpty()) return; AnvilRecipe recipe = matchOpt.get(); if (recipe.isRequiresHammer()) { int newDamage = utility.getDamage() + 1; if (newDamage >= utility.getMaxDamage()) { inventory.setStack(0, ItemStack.EMPTY); } else { utility.setDamage(newDamage); } } else { utility.decrement(1); } if (recipe.getType() == AnvilRecipe.Type.SINGLE_ANYWHERE) { for (int i = 1; i <= 9; i++) { ItemStack slot = inventory.getStack(i); if (!slot.isEmpty()) { slot.decrement(1); break; } } } else { String[] pattern = recipe.getPattern(); for (int row = 0; row < 3; row++) { for (int col = 0; col < 3; col++) { char c = pattern[row].charAt(col); if (c != ' ') { inventory.getStack(1 + row * 3 + col).decrement(1); } } } } updateResult(); } private ItemStack[] getGrid() { ItemStack[] grid = new ItemStack[9]; for (int i = 0; i < 9; i++) { grid[i] = inventory.getStack(i + 1); } return grid; } @Override public boolean canUse(PlayerEntity player) { return this.context.get((world, pos) -> { if (!(world.getBlockState(pos).getBlock() instanceof com.forgerscraft.block.ForgerersAnvilBlock)) { return false; } return player.squaredDistanceTo(pos.getX() + 0.5, pos.getY() + 0.5, pos.getZ() + 0.5) < 64.0; }, true); } @Override public ItemStack quickMove(PlayerEntity player, int slotIndex) { ItemStack newStack = ItemStack.EMPTY; Slot slot = this.slots.get(slotIndex); if (slot.hasStack()) { ItemStack originalStack = slot.getStack(); newStack = originalStack.copy(); if (slotIndex == 10) { if (!this.insertItem(originalStack, 11, 47, true)) { return ItemStack.EMPTY; } slot.onQuickTransfer(originalStack, newStack); } else if (slotIndex >= 11) { if (!this.insertItem(originalStack, 0, 1, false)) { if (!this.insertItem(originalStack, 1, 10, false)) { if (slotIndex < 38) { if (!this.insertItem(originalStack, 38, 47, false)) return ItemStack.EMPTY; } else { if (!this.insertItem(originalStack, 11, 38, false)) return ItemStack.EMPTY; } } } } else { if (!this.insertItem(originalStack, 11, 47, false)) { return ItemStack.EMPTY; } } if (originalStack.isEmpty()) { slot.setStack(ItemStack.EMPTY); } else { slot.markDirty(); } if (originalStack.getCount() == newStack.getCount()) { return ItemStack.EMPTY; } slot.onTakeItem(player, originalStack); } return newStack; } @Override public void onClosed(PlayerEntity player) { super.onClosed(player); this.inventory.onClose(player); } }