141 lines
4.5 KiB
Kotlin
141 lines
4.5 KiB
Kotlin
package org.neoflock.neocomputers.utils
|
|
|
|
// based off the ImplementedContainer of https://docs.fabricmc.net/develop/blocks/block-containers
|
|
import net.minecraft.client.gui.GuiGraphics
|
|
import net.minecraft.client.gui.screens.inventory.AbstractContainerScreen
|
|
import net.minecraft.world.Container;
|
|
import net.minecraft.core.NonNullList;
|
|
import net.minecraft.network.chat.Component
|
|
import net.minecraft.resources.ResourceLocation
|
|
import net.minecraft.server.packs.resources.Resource
|
|
import net.minecraft.world.ContainerHelper
|
|
import net.minecraft.world.entity.player.Inventory
|
|
import net.minecraft.world.entity.player.Player
|
|
import net.minecraft.world.inventory.AbstractContainerMenu
|
|
import net.minecraft.world.inventory.MenuType
|
|
import net.minecraft.world.inventory.Slot
|
|
import net.minecraft.world.item.ItemStack
|
|
|
|
// Common container interface, assumes the entire purpose is purely raw item storage
|
|
interface GenericContainer : Container {
|
|
fun getItems(): NonNullList<ItemStack>
|
|
|
|
override fun getContainerSize(): Int {
|
|
return getItems().size
|
|
}
|
|
|
|
override fun isEmpty(): Boolean {
|
|
return getItems().all { it.isEmpty }
|
|
}
|
|
|
|
override fun getItem(i: Int): ItemStack {
|
|
return getItems()[i]
|
|
}
|
|
|
|
override fun removeItem(slot: Int, count: Int): ItemStack {
|
|
val res = ContainerHelper.removeItem(getItems(), slot, count)
|
|
if (!res.isEmpty) setChanged()
|
|
return res
|
|
}
|
|
|
|
override fun setItem(slot: Int, itemStack: ItemStack) {
|
|
getItems()[slot] = itemStack
|
|
|
|
// in case of bullshit
|
|
if(itemStack.count > itemStack.maxStackSize) {
|
|
// rip items
|
|
itemStack.count = itemStack.maxStackSize
|
|
}
|
|
}
|
|
|
|
override fun removeItemNoUpdate(i: Int): ItemStack {
|
|
return ContainerHelper.takeItem(getItems(), i)
|
|
}
|
|
|
|
override fun clearContent() {
|
|
getItems().clear()
|
|
}
|
|
}
|
|
|
|
abstract class GenericContainerMenu(menuType: MenuType<*>, id: Int, var container: Container): AbstractContainerMenu(menuType, id) {
|
|
fun addInventorySlots(inventory: Inventory, x: Int, y: Int) {
|
|
// Based off the code in ChestMenu
|
|
for (i in 0..2) {
|
|
for (j in 0..8) {
|
|
this.addSlot(Slot(inventory, j + i * 9 + 9, 8 + j * 18, 84 + i * 18))
|
|
}
|
|
}
|
|
|
|
addInventoryHotbar(inventory, x, y + 3 * 18 + 4)
|
|
}
|
|
|
|
fun addInventoryHotbar(inventory: Inventory, x: Int, y: Int) {
|
|
for (i in 0..8) {
|
|
this.addSlot(Slot(inventory, i, x + i * 18, y))
|
|
}
|
|
}
|
|
|
|
// taken from https://docs.fabricmc.net/develop/blocks/container-menus
|
|
override fun quickMoveStack(player: Player, i: Int): ItemStack? {
|
|
val slot = slots[i]
|
|
|
|
if(!slot.hasItem()) return ItemStack.EMPTY
|
|
|
|
val stack = slot.item
|
|
val copied = stack.copy()
|
|
val contSize = container.containerSize
|
|
|
|
if(i < contSize) {
|
|
if(!this.moveItemStackTo(stack, contSize, slots.size, true)) {
|
|
return ItemStack.EMPTY
|
|
}
|
|
} else if(!this.moveItemStackTo(stack, 0, contSize, false)) {
|
|
return ItemStack.EMPTY
|
|
}
|
|
|
|
if(stack.isEmpty) {
|
|
slot.setByPlayer(ItemStack.EMPTY)
|
|
} else {
|
|
slot.setChanged()
|
|
}
|
|
|
|
return copied
|
|
}
|
|
|
|
override fun stillValid(player: Player): Boolean {
|
|
return container.stillValid(player)
|
|
}
|
|
}
|
|
|
|
abstract class GenericContainerScreen<T: GenericContainerMenu>(menu: T, inventory: Inventory, component: Component): AbstractContainerScreen<T>(menu, inventory, component) {
|
|
open fun shouldCenterTitle() = true
|
|
open fun shouldRenderTooltip() = true
|
|
open fun findMenuTexture(): ResourceLocation? = null
|
|
|
|
val imageX: Int
|
|
get() = (width - imageWidth) / 2
|
|
|
|
val imageY: Int
|
|
get() = (height - imageHeight) / 2
|
|
|
|
override fun init() {
|
|
super.init()
|
|
|
|
if(shouldCenterTitle()) this.titleLabelX = (this.imageWidth - this.font.width(this.title)) / 2
|
|
}
|
|
|
|
override fun renderBg(guiGraphics: GuiGraphics, f: Float, i: Int, j: Int) {
|
|
val menuTex = findMenuTexture()
|
|
if(menuTex != null) {
|
|
val cx = (width - imageWidth) / 2
|
|
val cy = (height - imageHeight) / 2
|
|
|
|
guiGraphics.blit(menuTex, imageX, imageY, 0, 0, imageWidth, imageHeight)
|
|
}
|
|
}
|
|
|
|
override fun render(graphics: GuiGraphics, mouseX: Int, mouseY: Int, something: Float) {
|
|
super.render(graphics, mouseX, mouseY, something)
|
|
if(shouldRenderTooltip()) super.renderTooltip(graphics, mouseX, mouseY)
|
|
}
|
|
} |