minecraft-src/net/minecraft/world/inventory/BeaconMenu.java
2025-07-04 01:41:11 +03:00

184 lines
5.3 KiB
Java

package net.minecraft.world.inventory;
import java.util.Optional;
import net.minecraft.core.Holder;
import net.minecraft.core.registries.BuiltInRegistries;
import net.minecraft.tags.ItemTags;
import net.minecraft.world.Container;
import net.minecraft.world.SimpleContainer;
import net.minecraft.world.effect.MobEffect;
import net.minecraft.world.entity.player.Player;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.level.Level;
import net.minecraft.world.level.block.Blocks;
import org.jetbrains.annotations.Nullable;
public class BeaconMenu extends AbstractContainerMenu {
private static final int PAYMENT_SLOT = 0;
private static final int SLOT_COUNT = 1;
private static final int DATA_COUNT = 3;
private static final int INV_SLOT_START = 1;
private static final int INV_SLOT_END = 28;
private static final int USE_ROW_SLOT_START = 28;
private static final int USE_ROW_SLOT_END = 37;
private static final int NO_EFFECT = 0;
private final Container beacon = new SimpleContainer(1) {
@Override
public boolean canPlaceItem(int slot, ItemStack stack) {
return stack.is(ItemTags.BEACON_PAYMENT_ITEMS);
}
@Override
public int getMaxStackSize() {
return 1;
}
};
private final BeaconMenu.PaymentSlot paymentSlot;
private final ContainerLevelAccess access;
private final ContainerData beaconData;
public BeaconMenu(int containerId, Container container) {
this(containerId, container, new SimpleContainerData(3), ContainerLevelAccess.NULL);
}
public BeaconMenu(int containerId, Container container, ContainerData beaconData, ContainerLevelAccess access) {
super(MenuType.BEACON, containerId);
checkContainerDataCount(beaconData, 3);
this.beaconData = beaconData;
this.access = access;
this.paymentSlot = new BeaconMenu.PaymentSlot(this.beacon, 0, 136, 110);
this.addSlot(this.paymentSlot);
this.addDataSlots(beaconData);
int i = 36;
int j = 137;
for (int k = 0; k < 3; k++) {
for (int l = 0; l < 9; l++) {
this.addSlot(new Slot(container, l + k * 9 + 9, 36 + l * 18, 137 + k * 18));
}
}
for (int k = 0; k < 9; k++) {
this.addSlot(new Slot(container, k, 36 + k * 18, 195));
}
}
@Override
public void removed(Player player) {
super.removed(player);
if (!player.level().isClientSide) {
ItemStack itemStack = this.paymentSlot.remove(this.paymentSlot.getMaxStackSize());
if (!itemStack.isEmpty()) {
player.drop(itemStack, false);
}
}
}
@Override
public boolean stillValid(Player player) {
return stillValid(this.access, player, Blocks.BEACON);
}
@Override
public void setData(int id, int data) {
super.setData(id, data);
this.broadcastChanges();
}
@Override
public ItemStack quickMoveStack(Player player, int index) {
ItemStack itemStack = ItemStack.EMPTY;
Slot slot = this.slots.get(index);
if (slot != null && slot.hasItem()) {
ItemStack itemStack2 = slot.getItem();
itemStack = itemStack2.copy();
if (index == 0) {
if (!this.moveItemStackTo(itemStack2, 1, 37, true)) {
return ItemStack.EMPTY;
}
slot.onQuickCraft(itemStack2, itemStack);
} else if (!this.paymentSlot.hasItem() && this.paymentSlot.mayPlace(itemStack2) && itemStack2.getCount() == 1) {
if (!this.moveItemStackTo(itemStack2, 0, 1, false)) {
return ItemStack.EMPTY;
}
} else if (index >= 1 && index < 28) {
if (!this.moveItemStackTo(itemStack2, 28, 37, false)) {
return ItemStack.EMPTY;
}
} else if (index >= 28 && index < 37) {
if (!this.moveItemStackTo(itemStack2, 1, 28, false)) {
return ItemStack.EMPTY;
}
} else if (!this.moveItemStackTo(itemStack2, 1, 37, false)) {
return ItemStack.EMPTY;
}
if (itemStack2.isEmpty()) {
slot.setByPlayer(ItemStack.EMPTY);
} else {
slot.setChanged();
}
if (itemStack2.getCount() == itemStack.getCount()) {
return ItemStack.EMPTY;
}
slot.onTake(player, itemStack2);
}
return itemStack;
}
public int getLevels() {
return this.beaconData.get(0);
}
public static int encodeEffect(@Nullable Holder<MobEffect> effect) {
return effect == null ? 0 : BuiltInRegistries.MOB_EFFECT.asHolderIdMap().getId(effect) + 1;
}
@Nullable
public static Holder<MobEffect> decodeEffect(int effectId) {
return effectId == 0 ? null : BuiltInRegistries.MOB_EFFECT.asHolderIdMap().byId(effectId - 1);
}
@Nullable
public Holder<MobEffect> getPrimaryEffect() {
return decodeEffect(this.beaconData.get(1));
}
@Nullable
public Holder<MobEffect> getSecondaryEffect() {
return decodeEffect(this.beaconData.get(2));
}
public void updateEffects(Optional<Holder<MobEffect>> primaryEffect, Optional<Holder<MobEffect>> secondaryEffect) {
if (this.paymentSlot.hasItem()) {
this.beaconData.set(1, encodeEffect((Holder<MobEffect>)primaryEffect.orElse(null)));
this.beaconData.set(2, encodeEffect((Holder<MobEffect>)secondaryEffect.orElse(null)));
this.paymentSlot.remove(1);
this.access.execute(Level::blockEntityChanged);
}
}
public boolean hasPayment() {
return !this.beacon.getItem(0).isEmpty();
}
class PaymentSlot extends Slot {
public PaymentSlot(final Container container, final int containerIndex, final int xPosition, final int yPosition) {
super(container, containerIndex, xPosition, yPosition);
}
@Override
public boolean mayPlace(ItemStack stack) {
return stack.is(ItemTags.BEACON_PAYMENT_ITEMS);
}
@Override
public int getMaxStackSize() {
return 1;
}
}
}