show button only for ICellRestriction

GTNewHorizons · Jan 24, 2025 · 7c32862 · 7c32862
1 parent eb1e61c
commit 7c32862
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/main/java/appeng/client/gui/implementations/GuiCellWorkbench.java b/src/main/java/appeng/client/gui/implementations/GuiCellWorkbench.java
@@ -183,7 +183,7 @@ protected void handleButtonVisibility() {
         }
         this.fuzzyMode.setVisibility(!hasOreFilter && hasFuzzy);
         this.oreFilter.setVisibility(hasOreFilter);
-        this.cellRestriction.setVisibility(this.workbench.haveCell());
+        this.cellRestriction.setVisibility(this.workbench.haveCellRestrictAble());
     }
 
     @Override

diff --git a/src/main/java/appeng/container/implementations/ContainerCellWorkbench.java b/src/main/java/appeng/container/implementations/ContainerCellWorkbench.java
@@ -32,6 +32,7 @@
 import appeng.container.slot.OptionalSlotRestrictedInput;
 import appeng.container.slot.SlotFakeTypeOnly;
 import appeng.container.slot.SlotRestrictedInput;
+import appeng.helpers.ICellRestriction;
 import appeng.tile.inventory.AppEngNullInventory;
 import appeng.tile.misc.TileCellWorkbench;
 import appeng.util.IterationCounter;
@@ -182,6 +183,10 @@ public boolean haveCell() {
         return this.workBench.getCell() != null;
     }
 
+    public boolean haveCellRestrictAble() {
+        return haveCell() && this.workBench.getCell() instanceof ICellRestriction;
+    }
+
     @Override
     public void onUpdate(final String field, final Object oldValue, final Object newValue) {
         if (field.equals("copyMode")) {