improve GhostCircuitSlotWidget popup

Author: ghzdude

src/main/java/gregtech/api/mui/widget/GhostCircuitSlotWidget.java

@@ -1,11 +1,14 @@
 package gregtech.api.mui.widget;
 
+import com.cleanroommc.modularui.value.sync.SyncHandler;
+
 import gregtech.api.capability.impl.GhostCircuitItemStackHandler;
 import gregtech.api.mui.GTGuiTextures;
 import gregtech.api.mui.GTGuis;
 import gregtech.api.recipes.ingredients.IntCircuitIngredient;
 import gregtech.client.utils.TooltipHelper;
 
+import net.minecraft.item.ItemStack;
 import net.minecraft.network.PacketBuffer;
 import net.minecraftforge.items.IItemHandler;
 
@@ -23,6 +26,7 @@
 import com.cleanroommc.modularui.widgets.layout.Grid;
 import com.cleanroommc.modularui.widgets.slot.ModularSlot;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import java.io.IOException;
 import java.util.ArrayList;
@@ -31,6 +35,9 @@
 public class GhostCircuitSlotWidget extends ItemSlot {
 
     private static final int SYNC_CIRCUIT_INDEX = 10;
+    @Nullable
+    private IPanelHandler selectorPanel;
+    private GhostCircuitSyncHandler syncHandler;
 
     public GhostCircuitSlotWidget() {
         super();
@@ -41,7 +48,7 @@ public GhostCircuitSlotWidget() {
     public @NotNull Result onMousePressed(int mouseButton) {
         if (!isSelectorPanelOpen()) {
             if (mouseButton == 0 && TooltipHelper.isShiftDown()) {
-                createSelectorPanel();
+                this.getSelectorPanel().openPanel();
             } else {
                 MouseData mouseData = MouseData.create(mouseButton);
                 getSyncHandler().syncToServer(2, mouseData::writeToPacket);
@@ -58,17 +65,24 @@ public boolean onMouseScroll(ModularScreen.UpOrDown scrollDirection, int amount)
         return true;
     }
 
+    @Override
+    public boolean isValidSyncHandler(SyncHandler syncHandler) {
+        this.syncHandler = castIfTypeElseNull(syncHandler, GhostCircuitSyncHandler.class);
+        if (this.syncHandler == null) return false;
+        return super.isValidSyncHandler(syncHandler);
+    }
+
     @Override
     public ItemSlot slot(ModularSlot slot) {
-        ItemSlotSH sh = new GhostCircuitSyncHandler(slot);
-        isValidSyncHandler(sh);
-        setSyncHandler(sh);
+        this.syncHandler = new GhostCircuitSyncHandler(slot);
+        isValidSyncHandler(this.syncHandler);
+        setSyncHandler(this.syncHandler);
         return this;
     }
 
     protected void getCircuitSlotTooltip(@NotNull RichTooltip tooltip) {
         String configString;
-        int value = getSyncHandler().getGhostCircuitHandler().getCircuitValue();
+        int value = this.syncHandler.getCircuitValue();
         if (value == GhostCircuitItemStackHandler.NO_CONFIG) {
             configString = IKey.lang("gregtech.gui.configurator_slot.no_value").get();
         } else {
@@ -78,11 +92,6 @@ protected void getCircuitSlotTooltip(@NotNull RichTooltip tooltip) {
         tooltip.addLine(IKey.lang("gregtech.gui.configurator_slot.tooltip", configString));
     }
 
-    @Override
-    public @NotNull GhostCircuitSyncHandler getSyncHandler() {
-        return (GhostCircuitSyncHandler) super.getSyncHandler();
-    }
-
     @Override
     public void onMouseDrag(int mouseButton, long timeSinceClick) {}
 
@@ -92,44 +101,39 @@ public boolean onMouseRelease(int mouseButton) {
     }
 
     private boolean isSelectorPanelOpen() {
-        return getPanel().getScreen().isPanelOpen("circuit_selector");
+        return this.getSelectorPanel().isPanelOpen();
     }
 
-    private void createSelectorPanel() {
-        ItemDrawable circuitPreview = new ItemDrawable(getSyncHandler().getSlot().getStack());
-
-        IPanelHandler.simple(getPanel(), (mainPanel, player) -> {
-            var panel = GTGuis.createPopupPanel("circuit_selector", 176, 120);
-            List<List<IWidget>> options = new ArrayList<>();
-            for (int i = 0; i < 4; i++) {
-                options.add(new ArrayList<>());
-                for (int j = 0; j < 9; j++) {
-                    int index = i * 9 + j;
-                    if (index > 32) break;
-                    options.get(i).add(new ButtonWidget<>()
-                            .size(18)
-                            .background(GTGuiTextures.SLOT, new ItemDrawable(
-                                    IntCircuitIngredient.getIntegratedCircuit(index)).asIcon())
-                            .disableHoverBackground()
-                            .onMousePressed(mouseButton -> {
-                                getSyncHandler().syncToServer(SYNC_CIRCUIT_INDEX, buf -> buf.writeShort(index));
-                                circuitPreview.setItem(IntCircuitIngredient.getIntegratedCircuit(index));
-                                if (Interactable.hasShiftDown()) panel.animateClose();
-                                return true;
-                            }));
-                }
-            }
-            return panel.child(IKey.lang("metaitem.circuit.integrated.gui").asWidget().pos(5, 5))
-                    .child(circuitPreview.asIcon().size(16).asWidget()
-                            .size(18)
-                            .top(19).alignX(0.5f)
-                            .background(GTGuiTextures.SLOT, GTGuiTextures.INT_CIRCUIT_OVERLAY))
-                    .child(new Grid()
-                            .left(7).right(7).top(41).height(4 * 18)
-                            .matrix(options)
-                            .minColWidth(18).minRowHeight(18)
-                            .minElementMargin(0, 0));
-        }, true).openPanel();
+    @NotNull
+    private IPanelHandler getSelectorPanel() {
+        if (this.selectorPanel == null) {
+            this.selectorPanel = IPanelHandler.simple(getPanel(), (mainPanel, player) -> {
+                ItemDrawable circuitPreview = new ItemDrawable(this.syncHandler.getCircuitStack());
+
+                return GTGuis.createPopupPanel("circuit_selector", 176, 120)
+                        .child(IKey.lang("metaitem.circuit.integrated.gui").asWidget().pos(5, 5))
+                        .child(circuitPreview.asIcon().size(16).asWidget()
+                                .size(18)
+                                .top(19).alignX(0.5f)
+                                .background(GTGuiTextures.SLOT, GTGuiTextures.INT_CIRCUIT_OVERLAY))
+                        .child(new Grid()
+                                .left(7).right(7).top(41).height(4 * 18)
+                                .mapTo(9, 33, value -> new ButtonWidget<>()
+                                        .size(18)
+                                        .background(GTGuiTextures.SLOT, new ItemDrawable(
+                                                IntCircuitIngredient.getIntegratedCircuit(value)).asIcon())
+                                        .disableHoverBackground()
+                                        .onMousePressed(mouseButton -> {
+                                            getSyncHandler().syncToServer(SYNC_CIRCUIT_INDEX, buf -> buf.writeShort(value));
+                                            circuitPreview.setItem(IntCircuitIngredient.getIntegratedCircuit(value));
+                                            if (Interactable.hasShiftDown()) this.selectorPanel.closePanel();
+                                            return true;
+                                        }))
+                                .minColWidth(18).minRowHeight(18)
+                                .minElementMargin(0, 0));
+            }, true);
+        }
+        return this.selectorPanel;
     }
 
     private static class GhostCircuitSyncHandler extends ItemSlotSH {
@@ -169,6 +173,14 @@ private void setCircuitValue(int value) {
             }
         }
 
+        public int getCircuitValue() {
+            return getGhostCircuitHandler().getCircuitValue();
+        }
+
+        public ItemStack getCircuitStack() {
+            return getSlot().getStack();
+        }
+
         @Override
         public void readOnServer(int id, PacketBuffer buf) throws IOException {
             if (id == SYNC_CIRCUIT_INDEX) {