add covers that can hold some items (#2593)

* add chest cover

* Spotless apply for branch feature/chestcover for #2593 (#2594)

* create window on the left

* sa

* add slot migration

* add texture and T2&3

* add texture and T2&3

* spotless

* fix bugs

* add chest cover

* small fix and spotless

Co-Authored-By: GitHub GTNH Actions <>

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Martin Robertz <dream-master@gmx.net>
Co-authored-by: Nxer <43300390+Nxer@users.noreply.github.com>
Co-authored-by: Nxer <306803661@qq.com>

1 files changed, 274 insertions, 0 deletions

diff --git a/src/main/java/gregtech/common/covers/GT_Cover_Chest.java b/src/main/java/gregtech/common/covers/GT_Cover_Chest.java
new file mode 100644
index 0000000000..36e441102b
--- /dev/null
+++ b/src/main/java/gregtech/common/covers/GT_Cover_Chest.java
@@ -0,0 +1,274 @@
+package gregtech.common.covers;
+
+import static gregtech.api.objects.XSTR.XSTR_INSTANCE;
+
+import net.minecraft.entity.item.EntityItem;
+import net.minecraft.entity.player.EntityPlayerMP;
+import net.minecraft.item.ItemStack;
+import net.minecraft.nbt.NBTBase;
+import net.minecraft.nbt.NBTTagCompound;
+import net.minecraftforge.common.util.ForgeDirection;
+
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+
+import com.google.common.io.ByteArrayDataInput;
+import com.gtnewhorizons.modularui.api.forge.ItemStackHandler;
+import com.gtnewhorizons.modularui.api.math.Alignment;
+import com.gtnewhorizons.modularui.api.math.Size;
+import com.gtnewhorizons.modularui.api.screen.ModularWindow;
+import com.gtnewhorizons.modularui.common.widget.SlotGroup;
+
+import cpw.mods.fml.common.network.ByteBufUtils;
+import gregtech.api.gui.modularui.GT_CoverUIBuildContext;
+import gregtech.api.interfaces.ITexture;
+import gregtech.api.interfaces.tileentity.ICoverable;
+import gregtech.api.util.GT_CoverBehaviorBase;
+import gregtech.api.util.ISerializableObject;
+import gregtech.common.gui.modularui.widget.CoverDataControllerWidget;
+import io.netty.buffer.ByteBuf;
+
+public class GT_Cover_Chest extends GT_CoverBehaviorBase<GT_Cover_Chest.ChestInventory> {
+
+    private final int slots;
+    private final int stackSizeLimit = 1;
+
+    public GT_Cover_Chest(int slots, ITexture coverTexture) {
+        super(ChestInventory.class, coverTexture);
+        if (slots <= 0) throw new IllegalArgumentException("slots must be greater than 0");
+        this.slots = slots;
+    }
+
+    @Override
+    public ChestInventory createDataObject(int aLegacyData) {
+        return new ChestInventory(slots, stackSizeLimit);
+    }
+
+    @Override
+    public ChestInventory createDataObject() {
+        return new ChestInventory(slots, stackSizeLimit);
+    }
+
+    @Override
+    public boolean hasCoverGUI() {
+        return true;
+    }
+
+    @Override
+    public boolean isSimpleCover() {
+        return true;
+    }
+
+    @Override
+    public boolean allowsCopyPasteTool() {
+        return false;
+    }
+
+    @Override
+    public boolean allowsTickRateAddition() {
+        return false;
+    }
+
+    @Override
+    protected void onDroppedImpl(ForgeDirection side, int aCoverID, ChestInventory aCoverVariable,
+        ICoverable aTileEntity) {
+        if (aTileEntity.getWorld().isRemote) return;
+        aCoverVariable.dropAll(aTileEntity, side);
+    }
+
+    @Override
+    protected int getTickRateImpl(ForgeDirection side, int aCoverID, ChestInventory aCoverVariable,
+        ICoverable aTileEntity) {
+        return aCoverVariable.firstTick ? 1 : 0;
+    }
+
+    @Override
+    protected ChestInventory doCoverThingsImpl(ForgeDirection side, byte aInputRedstone, int aCoverID,
+        ChestInventory aCoverVariable, ICoverable aTileEntity, long aTimer) {
+        // migrate slots. mostly needed while in development. still can be useful if we ever resize the inventory in the
+        // future
+        if (aCoverVariable.items.getSlots() != slots) {
+            if (aCoverVariable.items.getSlots() > slots) {
+                for (int i = slots; i < aCoverVariable.items.getSlots(); i++) {
+                    ItemStack item = aCoverVariable.items.getStackInSlot(i);
+                    if (item != null) {
+                        dropItem(aTileEntity, side, item);
+                    }
+                }
+            }
+
+            ChestInventory newData = createDataObject();
+            int toCopy = Math.min(newData.items.getSlots(), aCoverVariable.items.getSlots());
+            for (int i = 0; i < toCopy; i++) {
+                newData.items.setStackInSlot(i, aCoverVariable.items.getStackInSlot(i));
+            }
+            return newData;
+        }
+        aCoverVariable.firstTick = false;
+        return super.doCoverThingsImpl(side, aInputRedstone, aCoverID, aCoverVariable, aTileEntity, aTimer);
+    }
+
+    @Override
+    public ModularWindow createWindow(GT_CoverUIBuildContext buildContext) {
+        return new ChestUIFactory(buildContext).createWindow();
+    }
+
+    public class ChestUIFactory extends UIFactory {
+
+        private static final int spaceX = 18;
+        private static final int spaceY = 18;
+
+        public ChestUIFactory(GT_CoverUIBuildContext buildContext) {
+            super(buildContext);
+        }
+
+        @Override
+        protected int getGUIHeight() {
+            int height = slots / 3 * 18 + 8;
+            if (!getUIBuildContext().isAnotherWindow()) {
+                // player inv is 4 row
+                return height + 4 * 18 + 14;
+            }
+            return height;
+        }
+
+        @Override
+        protected void addTitleToUI(ModularWindow.Builder builder) {}
+
+        @Override
+        protected int getGUIWidth() {
+            if (getUIBuildContext().isAnotherWindow()) {
+                return 18 * 3 + 20;
+            } else {
+                return 18 * 9 + 20;
+            }
+        }
+
+        @Override
+        protected void addUIWidgets(ModularWindow.Builder builder) {
+            CoverDataControllerWidget<ChestInventory> w = new CoverDataControllerWidget<>(
+                this::getCoverData,
+                this::setCoverData,
+                GT_Cover_Chest.this);
+            ChestInventory d = getCoverData();
+            LimitingItemStackHandler h;
+            if (d == null) {
+                // ???
+                return;
+            }
+            h = d.items;
+            SlotGroup slotGroup = SlotGroup.ofItemHandler(h, 3)
+                .build();
+            if (getUIBuildContext().isAnotherWindow()) {
+                slotGroup.setPos(4, 4);
+            } else {
+                slotGroup.setPos(getGUIWidth() / 2 - 18 * 3 / 2, 6);
+            }
+            w.addChild(slotGroup);
+            builder.widget(w);
+
+            builder.setPos(
+                (p, t) -> Alignment.Center.getAlignedPos(
+                    new Size(getUIBuildContext().isAnotherWindow() ? t.getPos().x : p.width, p.height),
+                    new Size(getGUIWidth(), getGUIHeight())));
+        }
+
+        @Override
+        protected boolean doesBindPlayerInventory() {
+            return true;
+        }
+    }
+
+    private static class LimitingItemStackHandler extends ItemStackHandler {
+
+        private final int slotLimit;
+
+        private LimitingItemStackHandler(int slots, int slotLimit) {
+            super(slots);
+            this.slotLimit = slotLimit;
+        }
+
+        @Override
+        public int getSlotLimit(int slot) {
+            return slotLimit;
+        }
+    }
+
+    public static class ChestInventory implements ISerializableObject {
+
+        final LimitingItemStackHandler items;
+        boolean firstTick;
+
+        public ChestInventory(int slots, int stackSize) {
+            items = new LimitingItemStackHandler(slots, stackSize);
+        }
+
+        @NotNull
+        @Override
+        public ISerializableObject readFromPacket(ByteArrayDataInput aBuf, @Nullable EntityPlayerMP aPlayer) {
+            items.deserializeNBT(ISerializableObject.readCompoundTagFromGreggyByteBuf(aBuf));
+            return this;
+        }
+
+        @Override
+        public void writeToByteBuf(ByteBuf aBuf) {
+            ByteBufUtils.writeTag(aBuf, items.serializeNBT());
+        }
+
+        @Override
+        public void loadDataFromNBT(NBTBase aNBT) {
+            if (!(aNBT instanceof NBTTagCompound)) return;
+            items.deserializeNBT((NBTTagCompound) aNBT);
+            firstTick = true;
+        }
+
+        @NotNull
+        @Override
+        public NBTBase saveDataToNBT() {
+            return items.serializeNBT();
+        }
+
+        @NotNull
+        @Override
+        public ISerializableObject copy() {
+            ChestInventory copy = new ChestInventory(items.getSlots(), items.getSlotLimit(0));
+            for (int i = 0; i < items.getSlots(); i++) {
+                copy.items.setStackInSlot(i, items.getStackInSlot(i));
+            }
+            return copy;
+        }
+
+        public void dropAll(ICoverable coverable, ForgeDirection direction) {
+            for (int i = 0; i < items.getSlots(); i++) {
+                ItemStack tItem = items.getStackInSlot(i);
+                if (tItem == null) {
+                    continue;
+                }
+                dropItem(coverable, direction, tItem);
+                items.setStackInSlot(i, null);
+            }
+        }
+    }
+
+    private static void dropItem(ICoverable coverable, ForgeDirection direction, ItemStack tItem) {
+        final EntityItem tItemEntity = new EntityItem(
+            coverable.getWorld(),
+            coverable.getXCoord() + XSTR_INSTANCE.nextFloat() * 0.8F + 0.1F + direction.offsetX,
+            coverable.getYCoord() + XSTR_INSTANCE.nextFloat() * 0.8F + 0.1F + direction.offsetY,
+            coverable.getZCoord() + XSTR_INSTANCE.nextFloat() * 0.8F + 0.1F + direction.offsetZ,
+            new ItemStack(tItem.getItem(), tItem.stackSize, tItem.getItemDamage()));
+        if (tItem.hasTagCompound()) {
+            tItemEntity.getEntityItem()
+                .setTagCompound(
+                    (NBTTagCompound) tItem.getTagCompound()
+                        .copy());
+        }
+        tItemEntity.motionX = (XSTR_INSTANCE.nextGaussian() * 0.05D);
+        tItemEntity.motionY = (XSTR_INSTANCE.nextGaussian() * 0.05D + 0.2D);
+        tItemEntity.motionZ = (XSTR_INSTANCE.nextGaussian() * 0.05D);
+        tItemEntity.hurtResistantTime = 999999;
+        tItemEntity.lifespan = 60000;
+        coverable.getWorld()
+            .spawnEntityInWorld(tItemEntity);
+    }
+}