2 files changed, 115 insertions, 0 deletions
diff --git a/src/main/java/gregtech/common/tileentities/casings/functional/MotorCasing.java b/src/main/java/gregtech/common/tileentities/casings/functional/MotorCasing.java
new file mode 100644
index 0000000000..32a53d418a
--- /dev/null
+++ b/src/main/java/gregtech/common/tileentities/casings/functional/MotorCasing.java
@@ -0,0 +1,11 @@
+package gregtech.common.tileentities.casings.functional;
+
+import gregtech.api.multitileentity.multiblock.casing.FunctionalCasing;
+
+public class MotorCasing extends FunctionalCasing {
+
+    @Override
+    public String getTileEntityName() {
+        return "gt.multitileentity.multiblock.functional.motor";
+    }
+}
diff --git a/src/main/java/gregtech/common/tileentities/casings/upgrade/InventoryUpgrade.java b/src/main/java/gregtech/common/tileentities/casings/upgrade/InventoryUpgrade.java
new file mode 100644
index 0000000000..de144caf11
--- /dev/null
+++ b/src/main/java/gregtech/common/tileentities/casings/upgrade/InventoryUpgrade.java
@@ -0,0 +1,104 @@
+package gregtech.common.tileentities.casings.upgrade;
+
+import java.util.UUID;
+
+import net.minecraft.nbt.NBTTagCompound;
+
+import com.gtnewhorizons.modularui.api.screen.ModularWindow.Builder;
+import com.gtnewhorizons.modularui.api.screen.UIBuildContext;
+import com.gtnewhorizons.modularui.common.widget.textfield.TextFieldWidget;
+
+import gregtech.api.enums.GT_Values.NBT;
+import gregtech.api.multitileentity.interfaces.IMultiBlockController;
+import gregtech.api.multitileentity.multiblock.casing.UpgradeCasing;
+import gregtech.api.net.GT_Packet_MultiTileEntity;
+
+public class InventoryUpgrade extends UpgradeCasing {
+
+    public UUID mInventoryID;
+    public static final int INPUT = 0;
+    public static final int OUTPUT = 1;
+    public static final int BOTH = 2;
+    private String mInventoryName = "inventory";
+    private int mInventorySize;
+    private int mType = BOTH;
+
+    public String getInventoryName() {
+        return mInventoryName;
+    }
+
+    public void setInventoryName(String aInventoryName) {
+        mInventoryName = aInventoryName;
+    }
+
+    @Override
+    protected void customWork(IMultiBlockController aTarget) {
+        int tInvSize = mInventorySize;
+        if (mType == BOTH) {
+            tInvSize /= 2;
+        }
+        aTarget.registerInventory(mInventoryName, mInventoryID.toString(), tInvSize, mType);
+    }
+
+    @Override
+    public String getTileEntityName() {
+        return "gt.multitileentity.multiblock.inventory";
+    }
+
+    @Override
+    public void readMultiTileNBT(NBTTagCompound aNBT) {
+        super.readMultiTileNBT(aNBT);
+        if (aNBT.hasKey(NBT.UPGRADE_INVENTORY_UUID)) {
+            mInventoryID = UUID.fromString(aNBT.getString(NBT.UPGRADE_INVENTORY_UUID));
+        } else {
+            mInventoryID = UUID.randomUUID();
+        }
+        mInventorySize = aNBT.getInteger(NBT.UPGRADE_INVENTORY_SIZE);
+        mInventoryName = aNBT.getString(NBT.UPGRADE_INVENTORY_NAME);
+
+    }
+
+    @Override
+    public void writeMultiTileNBT(NBTTagCompound aNBT) {
+        super.writeMultiTileNBT(aNBT);
+        aNBT.setString(NBT.UPGRADE_INVENTORY_UUID, mInventoryID.toString());
+        aNBT.setString(NBT.UPGRADE_INVENTORY_NAME, mInventoryName);
+    }
+
+    @Override
+    protected void onBaseTEDestroyed() {
+        super.onBaseTEDestroyed();
+        unregisterInventories();
+    }
+
+    private void unregisterInventories() {
+        final IMultiBlockController controller = getTarget(false);
+        if (controller != null) {
+            controller.unregisterInventory(mInventoryName, mInventoryID.toString(), mType);
+        }
+    }
+
+    @Override
+    public boolean hasGui(byte aSide) {
+        return true;
+    }
+
+    @Override
+    public void addUIWidgets(Builder builder, UIBuildContext buildContext) {
+        builder.widget(new TextFieldWidget().setGetter(() -> mInventoryName).setSetter((val) -> {
+            mInventoryName = val;
+            final IMultiBlockController controller = getTarget(false);
+            if (controller != null) {
+                controller.changeInventoryName(mInventoryName, mInventoryID.toString(), mType);
+            }
+        }).setSize(100, 25).setPos(50, 30));
+    }
+
+    @Override
+    public GT_Packet_MultiTileEntity getClientDataPacket() {
+        final GT_Packet_MultiTileEntity packet = super.getClientDataPacket();
+        String name = getInventoryName();
+        packet.setInventoryName(name);
+        return packet;
+    }
+}