Move shared code

Author: jchung01

src/main/java/github/kasuminova/mmce/common/util/AEFluidInventoryUpgradeable.java

@@ -7,6 +7,7 @@
 import appeng.fluids.util.IAEFluidTank;
 import appeng.util.Platform;
 import github.kasuminova.mmce.common.util.concurrent.ReadWriteLockProvider;
+import hellfirepvp.modularmachinery.common.util.HybridFluidUtils;
 import net.minecraft.nbt.NBTTagCompound;
 import net.minecraftforge.fluids.FluidStack;
 import net.minecraftforge.fluids.capability.IFluidTankProperties;
@@ -204,18 +205,9 @@ public int fill(final FluidStack fluid, final boolean doFill) {
         try {
             (doFill ? rwLock.writeLock() : rwLock.readLock()).lock();
 
-            if (oneFluidOneSlot) {
-                int found = -1;
-                for (int i = 0; i < fluids.length; i++) {
-                    final IAEFluidStack fluidInSlot = getFluid(i);
-                    if (fluidInSlot != null && fluidInSlot.getFluid() == insert.getFluid()) {
-                        found = i;
-                        break;
-                    }
-                }
-                if (found != -1) {
-                    return this.fill(found, insert, doFill);
-                }
+            int found = HybridFluidUtils.findSlotWithFluid(this, getTankProperties(), insert);
+            if (found >= 0) {
+                return this.fill(found, insert, doFill);
             }
 
             int totalFillAmount = 0;

src/main/java/github/kasuminova/mmce/common/util/MultiFluidTank.java

@@ -1,5 +1,6 @@
 package github.kasuminova.mmce.common.util;
 
+import hellfirepvp.modularmachinery.common.util.HybridFluidUtils;
 import net.minecraft.nbt.NBTTagCompound;
 import net.minecraftforge.fluids.FluidStack;
 import net.minecraftforge.fluids.capability.IFluidHandler;
@@ -10,7 +11,7 @@
 import java.util.Objects;
 
 @SuppressWarnings("unused")
-public class MultiFluidTank implements IFluidHandler {
+public class MultiFluidTank implements IFluidHandler, IOneToOneFluidHandler {
     private final FluidStack[] contents;
     private final IFluidTankProperties[] props;
     private int capacity;
@@ -66,6 +67,11 @@ public MultiFluidTank setCapacity(final int capacity) {
         return this;
     }
 
+    @Override
+    public boolean isOneFluidOneSlot() {
+        return oneToOne;
+    }
+
     @Override
     public IFluidTankProperties[] getTankProperties() {
         return props;
@@ -79,20 +85,9 @@ public synchronized int fill(final FluidStack fluid, final boolean doFill) {
 
         final FluidStack insert = fluid.copy();
 
-        if (oneToOne) {
-            // Find the distinct slot
-            int foundSlot = -1;
-            for (int i = 0; i < props.length; i++) {
-                FluidStack fluidInSlot = props[i].getContents();
-                if (fluidInSlot != null && fluidInSlot.getFluid() == fluid.getFluid()) {
-                    foundSlot = i;
-                    break;
-                }
-            }
-            if (foundSlot >= 0) {
-                return fill(foundSlot, insert, doFill);
-            }
-            // Didn't find existing fluid, resume normal logic.
+        int foundSlot = HybridFluidUtils.findSlotWithFluid(this, props, fluid);
+        if (foundSlot >= 0) {
+            return fill(foundSlot, insert, doFill);
         }
 
         int totalFillAmount = 0;

src/main/java/hellfirepvp/modularmachinery/common/util/HybridFluidUtils.java

@@ -3,6 +3,7 @@
 import com.google.common.collect.Lists;
 import github.kasuminova.mmce.common.concurrent.Sync;
 import github.kasuminova.mmce.common.util.IExtendedGasHandler;
+import github.kasuminova.mmce.common.util.IOneToOneFluidHandler;
 import github.kasuminova.mmce.common.util.MultiFluidTank;
 import github.kasuminova.mmce.common.util.MultiGasTank;
 import hellfirepvp.modularmachinery.common.crafting.helper.ProcessingComponent;
@@ -12,6 +13,7 @@
 import mekanism.api.gas.IGasHandler;
 import net.minecraftforge.fluids.FluidStack;
 import net.minecraftforge.fluids.capability.IFluidHandler;
+import net.minecraftforge.fluids.capability.IFluidTankProperties;
 import net.minecraftforge.fml.common.Optional;
 
 import javax.annotation.Nonnull;
@@ -135,6 +137,20 @@ public static void doDrainOrFill(final GasStack drainOrFill, final long maxDrain
         }
     }
 
+    public static int findSlotWithFluid(final IOneToOneFluidHandler handler, final IFluidTankProperties[] props, final FluidStack fluid) {
+        int found = -1;
+        if (handler.isOneFluidOneSlot()) {
+            for (int i = 0; i < props.length; i++) {
+                FluidStack fluidInSlot = props[i].getContents();
+                if (fluidInSlot != null && fluidInSlot.getFluid() == fluid.getFluid()) {
+                    found = i;
+                    break;
+                }
+            }
+        }
+        return found;
+    }
+
     @Nonnull
     public static List<IFluidHandler> castFluidHandlerComponents(final List<ProcessingComponent<?>> components) {
         if (components.size() == 1) {