Add retain exact modes to fluid regulators and robot arms

Author: Zorbatron

src/main/java/gregtech/api/mui/GTGuiTextures.java

@@ -184,11 +184,11 @@ public static class IDs {
             18, 18 * 2, 18, 18, true);
 
     public static final UITexture[] TRANSFER_MODE_OVERLAY = slice("textures/gui/overlay/transfer_mode_overlay.png",
-            18, 18 * 3, 18, 18, true);
+            18, 18 * 4, 18, 18, true);
 
     public static final UITexture[] FLUID_TRANSFER_MODE_OVERLAY = slice(
             "textures/gui/overlay/fluid_transfer_mode_overlay.png",
-            18, 18 * 3, 18, 18, true);
+            18, 18 * 4, 18, 18, true);
 
     public static final UITexture[] DISTRIBUTION_MODE_OVERLAY = slice(
             "textures/gui/widget/button_distribution_mode.png",

src/main/java/gregtech/common/covers/CoverFluidRegulator.java

@@ -65,11 +65,14 @@ protected int doTransferFluidsInternal(IFluidHandler myFluidHandler, IFluidHandl
         return switch (transferMode) {
             case TRANSFER_ANY -> GTTransferUtils.transferFluids(sourceHandler, destHandler, transferLimit,
                     fluidFilterContainer::test);
-            case KEEP_EXACT -> doKeepExact(transferLimit, sourceHandler, destHandler,
-                    fluidFilterContainer::test,
-                    this.fluidFilterContainer.getTransferSize());
             case TRANSFER_EXACT -> doTransferExact(transferLimit, sourceHandler, destHandler,
                     fluidFilterContainer::test, this.fluidFilterContainer.getTransferSize());
+            case KEEP_EXACT -> doKeepExact(transferLimit, sourceHandler, destHandler,
+                    fluidFilterContainer::test,
+                    this.fluidFilterContainer.getTransferSize(), true);
+            case RETAIN_EXACT -> doKeepExact(transferLimit, sourceHandler, destHandler,
+                    fluidFilterContainer::test,
+                    this.fluidFilterContainer.getTransferSize(), false);
         };
     }
 
@@ -107,7 +110,7 @@ protected int doKeepExact(final int transferLimit,
                               final IFluidHandler sourceHandler,
                               final IFluidHandler destHandler,
                               final Predicate<FluidStack> fluidFilter,
-                              int keepAmount) {
+                              int keepAmount, boolean direction) {
         if (sourceHandler == null || destHandler == null || fluidFilter == null)
             return 0;
 
@@ -127,11 +130,12 @@ protected int doKeepExact(final int transferLimit,
 
             // if fluid needs to be moved to meet the Keep Exact value
             int amountInDest;
-            if ((amountInDest = destFluids.getOrDefault(fluidStack, 0)) < keepAmount) {
+            if (direction ? (amountInDest = destFluids.getOrDefault(fluidStack, 0)) < keepAmount :
+                    (amountInDest = sourceFluids.getOrDefault(fluidStack, 0)) > keepAmount) {
 
                 // move the lesser of the remaining transfer limit and the difference in actual vs keep exact amount
                 int amountToMove = Math.min(transferLimit - transferred,
-                        keepAmount - amountInDest);
+                        direction ? (keepAmount - amountInDest) : (amountInDest - keepAmount));
 
                 // Nothing to do here, try the next fluid.
                 if (amountToMove <= 0)
@@ -281,7 +285,7 @@ public int getMaxTransferRate() {
         return switch (this.transferMode) {
             case TRANSFER_ANY -> 1;
             case TRANSFER_EXACT -> maxFluidTransferRate;
-            case KEEP_EXACT -> Integer.MAX_VALUE;
+            case KEEP_EXACT, RETAIN_EXACT -> Integer.MAX_VALUE;
         };
     }
 

src/main/java/gregtech/common/covers/CoverRoboticArm.java

@@ -69,7 +69,8 @@ protected int doTransferItems(IItemHandler itemHandler, IItemHandler myItemHandl
         return switch (transferMode) {
             case TRANSFER_ANY -> doTransferItemsAny(itemHandler, myItemHandler, maxTransferAmount);
             case TRANSFER_EXACT -> doTransferExact(itemHandler, myItemHandler, maxTransferAmount);
-            case KEEP_EXACT -> doKeepExact(itemHandler, myItemHandler, maxTransferAmount);
+            case KEEP_EXACT -> doKeepExact(itemHandler, myItemHandler, maxTransferAmount, true);
+            case RETAIN_EXACT -> doKeepExact(itemHandler, myItemHandler, maxTransferAmount, false);
         };
     }
 
@@ -122,7 +123,8 @@ protected int doTransferExact(IItemHandler itemHandler, IItemHandler myItemHandl
         return Math.min(itemsTransferred, maxTransferAmount);
     }
 
-    protected int doKeepExact(IItemHandler itemHandler, IItemHandler myItemHandler, int maxTransferAmount) {
+    protected int doKeepExact(IItemHandler itemHandler, IItemHandler myItemHandler, int maxTransferAmount,
+                              boolean direction) {
         Map<Integer, GroupItemInfo> currentItemAmount = doCountDestinationInventoryItemsByMatchIndex(itemHandler,
                 myItemHandler);
         Map<Integer, GroupItemInfo> sourceItemAmounts = doCountDestinationInventoryItemsByMatchIndex(myItemHandler,
@@ -145,12 +147,20 @@ protected int doKeepExact(IItemHandler itemHandler, IItemHandler myItemHandler,
             }
 
             int itemAmount = 0;
-            if (currentItemAmount.containsKey(filterSlotIndex)) {
-                GroupItemInfo destItemInfo = currentItemAmount.get(filterSlotIndex);
-                itemAmount = destItemInfo.totalCount;
+            if (direction) {
+                if (currentItemAmount.containsKey(filterSlotIndex)) {
+                    GroupItemInfo destItemInfo = currentItemAmount.get(filterSlotIndex);
+                    itemAmount = destItemInfo.totalCount;
+                }
+            } else {
+                if (sourceItemAmounts.containsKey(filterSlotIndex)) {
+                    GroupItemInfo sourceItemInfo = sourceItemAmounts.get(filterSlotIndex);
+                    itemAmount = sourceItemInfo.totalCount;
+                }
             }
-            if (itemAmount < itemToKeepAmount) {
-                sourceInfo.totalCount = itemToKeepAmount - itemAmount;
+
+            if (direction ? (itemAmount < itemToKeepAmount) : (itemAmount > itemToKeepAmount)) {
+                sourceInfo.totalCount = direction ? (itemToKeepAmount - itemAmount) : (itemAmount - itemToKeepAmount);
             } else {
                 iterator.remove();
             }

src/main/java/gregtech/common/covers/TransferMode.java

@@ -8,7 +8,8 @@ public enum TransferMode implements IStringSerializable {
 
     TRANSFER_ANY("cover.robotic_arm.transfer_mode.transfer_any", 1),
     TRANSFER_EXACT("cover.robotic_arm.transfer_mode.transfer_exact", 1024),
-    KEEP_EXACT("cover.robotic_arm.transfer_mode.keep_exact", Integer.MAX_VALUE);
+    KEEP_EXACT("cover.robotic_arm.transfer_mode.keep_exact", Integer.MAX_VALUE),
+    RETAIN_EXACT("cover.robotic_arm.transfer_mode.retain_exact", Integer.MAX_VALUE);
 
     public static final TransferMode[] VALUES = values();
     public final String localeName;

src/main/resources/assets/gregtech/lang/en_us.lang

@@ -1335,6 +1335,7 @@ cover.robotic_arm.exact=§7items
 cover.robotic_arm.transfer_mode.transfer_any=Transfer Any
 cover.robotic_arm.transfer_mode.transfer_exact=Supply Exact
 cover.robotic_arm.transfer_mode.keep_exact=Keep Exact
+cover.robotic_arm.transfer_mode.retain_exact=Retain Exact
 cover.robotic_arm.transfer_mode.description=§eTransfer Any§r - in this mode, cover will transfer as many items matching its filter as possible./n§eSupply Exact§r - in this mode, cover will supply items in portions specified in item filter slots (or variable under this button for ore dictionary filter). If amount of items is less than portion size, items won't be moved. If there's a §bSmart Item Filter§r, the variable under this button will act as a multiplier instead./n§eKeep Exact§r - in this mode, cover will keep specified amount of items in the destination inventory, supplying additional amount of items if required./n§7Tip: left/right click on filter slots to change item amount,  use shift clicking to change amount faster.
 
 cover.pump.title=Pump Cover Settings (%s)