From eede79b1f210eec76f3b223353ab06a3398919ef Mon Sep 17 00:00:00 2001
From: Glease <4586901+Glease@users.noreply.github.com>
Date: Tue, 9 Feb 2021 02:14:26 +0800
Subject: Fix size 0 fluid stack

Signed-off-by: Glease <4586901+Glease@users.noreply.github.com>
---
 src/main/java/gregtech/api/gui/GT_Container_BasicMachine.java | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/main/java/gregtech')

diff --git a/src/main/java/gregtech/api/gui/GT_Container_BasicMachine.java b/src/main/java/gregtech/api/gui/GT_Container_BasicMachine.java
index 7e4e105353..da528d67fc 100644
--- a/src/main/java/gregtech/api/gui/GT_Container_BasicMachine.java
+++ b/src/main/java/gregtech/api/gui/GT_Container_BasicMachine.java
@@ -188,6 +188,7 @@ public class GT_Container_BasicMachine extends GT_Container_BasicTank {
     public ItemStack slotClick(int aSlotIndex, int aMouseclick, int aShifthold, EntityPlayer aPlayer) {
         GT_MetaTileEntity_BasicMachine machine = (GT_MetaTileEntity_BasicMachine) mTileEntity.getMetaTileEntity();
         if (machine == null) return null;
+        ItemStack tResultStack;
         switch (aSlotIndex) {
             case 0:
                 machine.mFluidTransfer = !machine.mFluidTransfer;
@@ -197,7 +198,10 @@ public class GT_Container_BasicMachine extends GT_Container_BasicTank {
                 machine.mItemTransfer = !machine.mItemTransfer;
                 return null;
             case 2:
-                return pickupFluid(machine.getDrainableStack(), aPlayer);
+                tResultStack = pickupFluid(machine.getDrainableStack(), aPlayer);
+                if (machine.getDrainableStack().amount == 0)
+                    machine.setDrainableStack(null);
+                return tResultStack;
             default:
                 if (aSlotIndex == OTHER_SLOT_COUNT + 1 + machine.mInputSlotCount + machine.mOutputItems.length) {
                     // input fluid slot
@@ -216,7 +220,10 @@ public class GT_Container_BasicMachine extends GT_Container_BasicTank {
                             // both nonnull. actually both pickup and fill is reasonable, but I'll go with fill here
                             return fillFluid(machine, aPlayer, tFluidHeld);
                         } else {
-                            return pickupFluid(tInputFluid, aPlayer);
+                            tResultStack = pickupFluid(tInputFluid, aPlayer);
+                            if (tInputFluid.amount == 0)
+                                machine.setFillableStack(null);
+                            return tResultStack;
                         }
                     }
                 } else {
-- 
cgit