2 files changed, 65 insertions, 0 deletions
diff --git a/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/DataStickSlotWidget.java b/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/DataStickSlotWidget.java
new file mode 100644
index 0000000000..dbda217c27
--- /dev/null
+++ b/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/DataStickSlotWidget.java
@@ -0,0 +1,28 @@
+package gtPlusPlus.xmod.gregtech.api.gui.widget;
+
+import com.gtnewhorizons.modularui.api.ModularUITextures;
+import com.gtnewhorizons.modularui.api.forge.IItemHandlerModifiable;
+import com.gtnewhorizons.modularui.common.internal.wrapper.BaseSlot;
+import com.gtnewhorizons.modularui.common.widget.SlotWidget;
+import gregtech.api.enums.ItemList;
+import gregtech.api.gui.modularui.GT_UITextures;
+import gregtech.api.util.GT_Utility;
+
+public class DataStickSlotWidget extends SlotWidget {
+
+    public DataStickSlotWidget(IItemHandlerModifiable handler, int index) {
+        this(new BaseSlot(handler, index) {
+            @Override
+            public int getSlotStackLimit() {
+                return 1;
+            }
+        });
+    }
+
+    private DataStickSlotWidget(BaseSlot slot) {
+        super(slot);
+        setFilter(stack -> GT_Utility.areStacksEqual(stack, ItemList.Tool_DataStick.get(1), true)
+                || GT_Utility.areStacksEqual(stack, ItemList.Tool_DataOrb.get(1), true));
+        setBackground(ModularUITextures.ITEM_SLOT, GT_UITextures.OVERLAY_SLOT_DATA_ORB);
+    }
+}
diff --git a/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/ElectricSlotWidget.java b/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/ElectricSlotWidget.java
new file mode 100644
index 0000000000..b575c3c4bd
--- /dev/null
+++ b/src/main/java/gtPlusPlus/xmod/gregtech/api/gui/widget/ElectricSlotWidget.java
@@ -0,0 +1,37 @@
+package gtPlusPlus.xmod.gregtech.api.gui.widget;
+
+import com.gtnewhorizons.modularui.api.forge.IItemHandlerModifiable;
+import com.gtnewhorizons.modularui.common.internal.wrapper.BaseSlot;
+import com.gtnewhorizons.modularui.common.widget.SlotWidget;
+import gregtech.api.items.GT_MetaGenerated_Tool;
+import ic2.api.info.Info;
+import ic2.api.item.ElectricItem;
+import ic2.api.item.IElectricItem;
+import net.minecraft.item.ItemStack;
+
+public class ElectricSlotWidget extends SlotWidget {
+
+    public ElectricSlotWidget(IItemHandlerModifiable handler, int index) {
+        this(new BaseSlot(handler, index, false) {
+            @Override
+            public int getSlotStackLimit() {
+                return 1;
+            }
+        });
+    }
+
+    private ElectricSlotWidget(BaseSlot slot) {
+        super(slot);
+        setFilter(stack -> (accepts(stack))
+                || (stack.getItem() instanceof GT_MetaGenerated_Tool)
+                || (stack.getItem() instanceof IElectricItem));
+    }
+
+    private boolean accepts(final ItemStack stack) {
+        if (stack == null) {
+            return false;
+        }
+        return (Info.itemEnergy.getEnergyValue(stack) > 0.0D)
+                || (ElectricItem.manager.discharge(stack, (1.0D / 0.0D), 4, true, true, true) > 0.0D);
+    }
+}