AE2 Pattern Workstation Compatibility

Author: link-fgfgui

gradle.properties

@@ -30,7 +30,7 @@ loader_version_range=[47,)
 mod_id=emi_patternizer
 mod_name=EMI Patternizer
 mod_license=GNU LGPL 3.0
-mod_version=1.0.1
+mod_version=1.0.1.1
 mod_group_id=io.github.linkfgfgui
 mod_authors=link-fgfgui
 mod_description=Automatically Encoding Patterns from the EMI Recipe Tree.

src/main/java/io/github/linkfgfgui/emi_patternizer/Patternize.java

@@ -1,6 +1,8 @@
 package io.github.linkfgfgui.emi_patternizer;
 
+import appeng.client.gui.AEBaseScreen;
 import appeng.client.gui.me.items.PatternEncodingTermScreen;
+import appeng.menu.AEBaseMenu;
 import appeng.menu.SlotSemantics;
 import appeng.menu.me.items.PatternEncodingTermMenu;
 import com.mojang.blaze3d.platform.InputConstants;
@@ -18,12 +20,16 @@
 import net.minecraft.client.resources.sounds.SimpleSoundInstance;
 import net.minecraft.sounds.SoundEvents;
 import net.minecraft.world.inventory.ClickType;
+import net.minecraft.world.inventory.Slot;
 import net.minecraftforge.client.event.ScreenEvent;
 import org.slf4j.Logger;
 
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
 import java.util.Comparator;
 import java.util.HashSet;
 import java.util.List;
+import java.util.concurrent.Callable;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicLong;
@@ -37,6 +43,8 @@ public class Patternize {
     static long delayPerOperation;
     static long delayAdditionalPerPattern;
     static boolean isPlaySound;
+    static Method encodeMethod = null;
+    static Runnable callEncode = null;
 
     public static boolean containsAllItems(EmiRecipe r) {
         return containsAllItems(r.getOutputs());
@@ -46,7 +54,7 @@ public static boolean containsAllItems(List<EmiStack> stackList) {
         return (stackList.stream().allMatch(emiStack -> EncodedItems.contains(emiStack.getId().toString())));
     }
 
-    public static void Encode(long initDelay, Minecraft minecraft, EmiRecipe recipe, PatternEncodingTermScreen<?> screen, PatternEncodingTermMenu menu, LocalPlayer player, MultiPlayerGameMode gameMode, int encodedPatternSlot) {
+    public static void Encode(long initDelay, Minecraft minecraft, EmiRecipe recipe, AEBaseScreen<?> screen, AEBaseMenu menu, LocalPlayer player, MultiPlayerGameMode gameMode, Integer encodedPatternSlot) {
         CompletableFuture.delayedExecutor(initDelay, TimeUnit.MILLISECONDS).execute(() -> {
             minecraft.execute(() -> {
                 EmiRecipeFiller.performFill(recipe, screen, EmiCraftContext.Type.FILL_BUTTON, EmiCraftContext.Destination.NONE, 1);
@@ -56,17 +64,22 @@ public static void Encode(long initDelay, Minecraft minecraft, EmiRecipe recipe,
             });
             CompletableFuture.delayedExecutor(delayPerOperation, TimeUnit.MILLISECONDS).execute(() -> {
                 minecraft.execute(() -> {
-                    menu.encode();
+                    if(encodeMethod!=null){
+                        callEncode.run();
+                    }
                 });
                 CompletableFuture.delayedExecutor(delayPerOperation, TimeUnit.MILLISECONDS).execute(() ->
-                        minecraft.execute(() ->
-                                gameMode.handleInventoryMouseClick(
-                                        menu.containerId,
-                                        encodedPatternSlot,
-                                        0, // 0 = Left, 1 = Right
-                                        ClickType.QUICK_MOVE,
-                                        player
-                                )
+                        minecraft.execute(() -> {
+                                    if (encodedPatternSlot != null) {
+                                        gameMode.handleInventoryMouseClick(
+                                                menu.containerId,
+                                                encodedPatternSlot,
+                                                0, // 0 = Left, 1 = Right
+                                                ClickType.QUICK_MOVE,
+                                                player
+                                        );
+                                    }
+                                }
                         ));
             });
         });
@@ -89,19 +102,47 @@ public static void LoadConfig() {
 
     public static void onKeyPressed(ScreenEvent.KeyPressed event) {
         if (!operating && Emi_patternizer.PATTERNIZE_MAPPING.get().isActiveAndMatches(InputConstants.getKey(event.getKeyCode(), event.getScanCode()))) {
-            if (event.getScreen() instanceof PatternEncodingTermScreen<?> screen) {
-                PatternEncodingTermMenu menu = screen.getMenu();
+            if (event.getScreen() instanceof AEBaseScreen<?> screen) {
+                AEBaseMenu menu = screen.getMenu();
 //                int blankPatternSlot = ((AEBaseMenuAccessor) menu).getSlotsBySemantic().get(SlotSemantics.BLANK_PATTERN).getFirst().index;
-                int encodedPatternSlot = ((AEBaseMenuAccessor) menu).getSlotsBySemantic().get(SlotSemantics.ENCODED_PATTERN).get(0).index;
+                List<Slot> slots = ((AEBaseMenuAccessor) menu).getSlotsBySemantic().get(SlotSemantics.ENCODED_PATTERN);
+                Integer encodedPatternSlot = slots.isEmpty() ? null : slots.get(0).index;
                 if (BoM.craftingMode && !operating) {
                     LoadConfig();
                     Minecraft minecraft = Minecraft.getInstance();
                     LocalPlayer player = minecraft.player;
                     MultiPlayerGameMode gameMode = minecraft.gameMode;
                     MaterialNode goal = BoM.tree.goal;
+                    encodeMethod=null;
+                    try {
+                        // AE2 Pattern Workstation(1462173) Compatibility
+                        encodeMethod=menu.getClass().getMethod("encode", Long.class);
+                        encodedPatternSlot = null;
+                        callEncode=()->{
+                            try {
+                                encodeMethod.invoke(menu,0L);
+                            } catch (IllegalAccessException | InvocationTargetException e01) {
+                                LOGGER.error(e01.getMessage());
+                            }
+                        };
+                    } catch (NoSuchMethodException e1) {
+                        try{
+                            encodeMethod=menu.getClass().getMethod("encode");
+                            callEncode=()->{
+                                try {
+                                    encodeMethod.invoke(menu);
+                                } catch (IllegalAccessException | InvocationTargetException e11) {
+                                    LOGGER.error(e11.getMessage());
+                                }
+                            };
+                        } catch (NoSuchMethodException e2) {
+                            LOGGER.error(e2.getMessage());
+                        }
+                    }
                     operating = true;
                     LOGGER.debug("operating started");
                     AtomicLong maxDelay = new AtomicLong(0);
+                    Integer finalEncodedPatternSlot = encodedPatternSlot;
                     Stream.of(goal)
                             .flatMap(Patternize::streamTree)
                             .sorted(Comparator.comparing(node -> {
@@ -113,7 +154,7 @@ public static void onKeyPressed(ScreenEvent.KeyPressed event) {
                             }))
                             .forEachOrdered(node -> {
                                 if (node.recipe != null && node.recipe.getId() != null && !containsAllItems(node.recipe)) {
-                                    Encode(maxDelay.get(), minecraft, node.recipe, screen, menu, player, gameMode, encodedPatternSlot);
+                                    Encode(maxDelay.get(), minecraft, node.recipe, screen, menu, player, gameMode, finalEncodedPatternSlot);
                                     node.recipe.getOutputs().forEach(emiStack -> EncodedItems.add(emiStack.getId().toString()));
                                     maxDelay.addAndGet(3 * delayPerOperation + delayAdditionalPerPattern);
                                 }