added methods for condensing an item inventory

GregTech-Intergalactical · Dec 10, 2023 · f9a28e6 · f9a28e6
1 parent 43778a0
commit f9a28e6
Showing 1 changed file with 29 additions and 0 deletions.
diff --git a/common/src/main/java/muramasa/antimatter/util/Utils.java b/common/src/main/java/muramasa/antimatter/util/Utils.java
@@ -45,6 +45,7 @@
 import net.minecraft.server.level.ServerPlayer;
 import net.minecraft.tags.BlockTags;
 import net.minecraft.tags.TagKey;
+import net.minecraft.util.Mth;
 import net.minecraft.world.InteractionHand;
 import net.minecraft.world.entity.EquipmentSlot;
 import net.minecraft.world.entity.LivingEntity;
@@ -173,6 +174,34 @@ public static List<ItemStack> mergeItems(List<ItemStack> a, List<ItemStack> b) {
         return splitStacks(a);
     }
 
+    public static void tryCondenseInventory(PlatformItemHandler itemHandler){
+        tryCondenseInventory(itemHandler, 0, itemHandler.getSlots());
+    }
+
+    public static void tryCondenseInventory(PlatformItemHandler tile, int startSlot, int endSlot) {
+        for (int i = startSlot; i < endSlot; ++i) {
+            for (int j = startSlot; j < endSlot; ++j) {
+                if (i == j) {
+                    continue;
+                }
+                ItemStack stack1 = tile.getStackInSlot(i);
+                ItemStack stack2 = tile.getStackInSlot(j);
+                if (!stack1.isEmpty() && !stack2.isEmpty()
+                        && (Utils.equals(stack1, stack2) && stack1.getCount() < stack1.getMaxStackSize())) {
+                    int max = stack1.getMaxStackSize() - stack1.getCount();
+                    int available = stack2.getCount();
+                    int size = Mth.clamp(available, 1, max);
+                    stack1.grow(size);
+                    stack2.shrink(size);
+                }
+                if (stack2.isEmpty() && !stack1.isEmpty() && j < i) {
+                    tile.setStackInSlot(j, stack1.copy());
+                    tile.setStackInSlot(i, ItemStack.EMPTY);
+                }
+            }
+        }
+    }
+
     public static List<ItemStack> splitStacks(List<ItemStack> stacks){
         List<ItemStack> returned = new ArrayList<>();
         for (ItemStack stack : stacks) {