116 lines
4.3 KiB
Java
116 lines
4.3 KiB
Java
package net.minecraft.world.inventory;
|
|
|
|
import net.minecraft.resources.ResourceLocation;
|
|
import net.minecraft.tags.EntityTypeTags;
|
|
import net.minecraft.world.Container;
|
|
import net.minecraft.world.entity.EquipmentSlot;
|
|
import net.minecraft.world.entity.animal.horse.AbstractHorse;
|
|
import net.minecraft.world.entity.animal.horse.Llama;
|
|
import net.minecraft.world.entity.player.Inventory;
|
|
import net.minecraft.world.entity.player.Player;
|
|
import net.minecraft.world.item.ItemStack;
|
|
|
|
public class HorseInventoryMenu extends AbstractContainerMenu {
|
|
private static final ResourceLocation SADDLE_SLOT_SPRITE = ResourceLocation.withDefaultNamespace("container/slot/saddle");
|
|
private static final ResourceLocation LLAMA_ARMOR_SLOT_SPRITE = ResourceLocation.withDefaultNamespace("container/slot/llama_armor");
|
|
private static final ResourceLocation ARMOR_SLOT_SPRITE = ResourceLocation.withDefaultNamespace("container/slot/horse_armor");
|
|
private final Container horseContainer;
|
|
private final AbstractHorse horse;
|
|
private static final int SLOT_SADDLE = 0;
|
|
private static final int SLOT_BODY_ARMOR = 1;
|
|
private static final int SLOT_HORSE_INVENTORY_START = 2;
|
|
|
|
public HorseInventoryMenu(int containerId, Inventory inventory, Container horseContainer, AbstractHorse horse, int columns) {
|
|
super(null, containerId);
|
|
this.horseContainer = horseContainer;
|
|
this.horse = horse;
|
|
horseContainer.startOpen(inventory.player);
|
|
Container container = horse.createEquipmentSlotContainer(EquipmentSlot.SADDLE);
|
|
this.addSlot(new ArmorSlot(container, horse, EquipmentSlot.SADDLE, 0, 8, 18, SADDLE_SLOT_SPRITE) {
|
|
@Override
|
|
public boolean isActive() {
|
|
return horse.canUseSlot(EquipmentSlot.SADDLE) && horse.getType().is(EntityTypeTags.CAN_EQUIP_SADDLE);
|
|
}
|
|
});
|
|
final boolean bl = horse instanceof Llama;
|
|
ResourceLocation resourceLocation = bl ? LLAMA_ARMOR_SLOT_SPRITE : ARMOR_SLOT_SPRITE;
|
|
Container container2 = horse.createEquipmentSlotContainer(EquipmentSlot.BODY);
|
|
this.addSlot(new ArmorSlot(container2, horse, EquipmentSlot.BODY, 0, 8, 36, resourceLocation) {
|
|
@Override
|
|
public boolean isActive() {
|
|
return horse.canUseSlot(EquipmentSlot.BODY) && (horse.getType().is(EntityTypeTags.CAN_WEAR_HORSE_ARMOR) || bl);
|
|
}
|
|
});
|
|
if (columns > 0) {
|
|
for (int i = 0; i < 3; i++) {
|
|
for (int j = 0; j < columns; j++) {
|
|
this.addSlot(new Slot(horseContainer, j + i * columns, 80 + j * 18, 18 + i * 18));
|
|
}
|
|
}
|
|
}
|
|
|
|
this.addStandardInventorySlots(inventory, 8, 84);
|
|
}
|
|
|
|
@Override
|
|
public boolean stillValid(Player player) {
|
|
return !this.horse.hasInventoryChanged(this.horseContainer)
|
|
&& this.horseContainer.stillValid(player)
|
|
&& this.horse.isAlive()
|
|
&& player.canInteractWithEntity(this.horse, 4.0);
|
|
}
|
|
|
|
@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();
|
|
int i = 2 + this.horseContainer.getContainerSize();
|
|
if (index < i) {
|
|
if (!this.moveItemStackTo(itemStack2, i, this.slots.size(), true)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
} else if (this.getSlot(1).mayPlace(itemStack2) && !this.getSlot(1).hasItem()) {
|
|
if (!this.moveItemStackTo(itemStack2, 1, 2, false)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
} else if (this.getSlot(0).mayPlace(itemStack2) && !this.getSlot(0).hasItem()) {
|
|
if (!this.moveItemStackTo(itemStack2, 0, 1, false)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
} else if (this.horseContainer.getContainerSize() == 0 || !this.moveItemStackTo(itemStack2, 2, i, false)) {
|
|
int j = i + 27;
|
|
int l = j + 9;
|
|
if (index >= j && index < l) {
|
|
if (!this.moveItemStackTo(itemStack2, i, j, false)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
} else if (index >= i && index < j) {
|
|
if (!this.moveItemStackTo(itemStack2, j, l, false)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
} else if (!this.moveItemStackTo(itemStack2, j, j, false)) {
|
|
return ItemStack.EMPTY;
|
|
}
|
|
|
|
return ItemStack.EMPTY;
|
|
}
|
|
|
|
if (itemStack2.isEmpty()) {
|
|
slot.setByPlayer(ItemStack.EMPTY);
|
|
} else {
|
|
slot.setChanged();
|
|
}
|
|
}
|
|
|
|
return itemStack;
|
|
}
|
|
|
|
@Override
|
|
public void removed(Player player) {
|
|
super.removed(player);
|
|
this.horseContainer.stopOpen(player);
|
|
}
|
|
}
|