Fixed a bug, now can automatically complete any forging recipes

Author: Eternal130

gradle.properties

@@ -48,7 +48,7 @@ mod_name=TFC:TNG Auto Forging
 # The license of the mod. Review your options at https://choosealicense.com/. All Rights Reserved is the default.
 mod_license=All Rights Reserved
 # The mod version. See https://semver.org/
-mod_version=1.20.1-1.0
+mod_version=1.20.1-1.1
 # The group ID for the mod. It is only important when publishing as an artifact to a Maven repository.
 # This should match the base package used for the mod sources.
 # See https://maven.apache.org/guides/mini/guide-naming-conventions.html

src/main/java/com/eternal130/tfcaf/TFCAutoForging.java

@@ -2,34 +2,15 @@
 
 import com.mojang.logging.LogUtils;
 import net.minecraft.client.KeyMapping;
-import net.minecraft.client.Minecraft;
-import net.minecraft.core.registries.Registries;
-import net.minecraft.world.food.FoodProperties;
-import net.minecraft.world.item.BlockItem;
-import net.minecraft.world.item.CreativeModeTab;
-import net.minecraft.world.item.CreativeModeTabs;
-import net.minecraft.world.item.Item;
-import net.minecraft.world.level.block.Block;
-import net.minecraft.world.level.block.Blocks;
-import net.minecraft.world.level.block.state.BlockBehaviour;
-import net.minecraft.world.level.material.MapColor;
-import net.minecraftforge.api.distmarker.Dist;
-import net.minecraftforge.client.event.RegisterKeyMappingsEvent;
 import net.minecraftforge.common.MinecraftForge;
-import net.minecraftforge.event.BuildCreativeModeTabContentsEvent;
 import net.minecraftforge.event.server.ServerStartingEvent;
 import net.minecraftforge.eventbus.api.IEventBus;
 import net.minecraftforge.eventbus.api.SubscribeEvent;
 import net.minecraftforge.fml.ModLoadingContext;
 import net.minecraftforge.fml.common.Mod;
 import net.minecraftforge.fml.config.ModConfig;
 import net.minecraftforge.fml.event.lifecycle.FMLClientSetupEvent;
-import net.minecraftforge.fml.event.lifecycle.FMLCommonSetupEvent;
 import net.minecraftforge.fml.javafmlmod.FMLJavaModLoadingContext;
-import net.minecraftforge.registries.DeferredRegister;
-import net.minecraftforge.registries.ForgeRegistries;
-import net.minecraftforge.registries.RegistryObject;
-import org.lwjgl.glfw.GLFW;
 import org.slf4j.Logger;
 
 // The value here should match an entry in the META-INF/mods.toml file

src/main/java/com/eternal130/tfcaf/forgeEvent.java

@@ -4,9 +4,11 @@
 import static com.eternal130.tfcaf.ConfigFile.enableForgingTip;
 
 import java.lang.reflect.Field;
+import java.util.Arrays;
 
 import net.dries007.tfc.client.screen.AnvilScreen;
 import net.dries007.tfc.common.blockentities.AnvilBlockEntity;
+import net.dries007.tfc.common.capabilities.forge.ForgeRule;
 import net.dries007.tfc.common.capabilities.forge.ForgeSteps;
 import net.dries007.tfc.common.capabilities.forge.Forging;
 import net.dries007.tfc.common.recipes.AnvilRecipe;
@@ -62,14 +64,14 @@ public static void operationHighlight(ScreenEvent.Render.Post event) {
                     // 下面那个变量内存有最后三步
                     ForgeSteps steps = forging.getSteps();
                     // 用于存储锻造要求，其中的值为Util类中operations的索引
-                    int[] lastOperations = new int[3];
+                    int[] lastOperations = getRules(AnvilRecipe.getRules());
                     for (int i = 0; i < 3; i++) {
-                        try {
-                            lastOperations[i] = Util.operationsTfc.get(AnvilRecipe.getRules()[i].ordinal());
-                        } catch (Exception e) {
-                            //这里的异常是索引越界异常,因为锻造需求相比于1.7,没有any,长度可以小于3
-                            lastOperations[i] = 4;
-                        }
+//                        try {
+//                            lastOperations[i] = Util.operationsTfc.get(AnvilRecipe.getRules()[i].ordinal());
+//                        } catch (Exception e) {
+//                            //这里的异常是索引越界异常,因为锻造需求相比于1.7,没有any,长度可以小于3
+//                            lastOperations[i] = 4;
+//                        }
 
                         ruleOffset += Util.operations[lastOperations[i]];
                     }
@@ -234,4 +236,64 @@ private static AnvilBlockEntity getTEAnvilTFC(AnvilScreen gui) throws NoSuchFiel
         fields.setAccessible(true);
         return (AnvilBlockEntity) fields.get(gui);
     }
+    private static int[] getRules(ForgeRule[] rules) {
+        // 相对于1.7版本,没有any类型,每种步骤也只有五种位置,少了LastTwo这种类型,因此少遍历一次
+        int[] lastOperations = new int[3];
+        // 将锻造要求初始化为-1,表示没有要求
+        Arrays.fill(lastOperations, -1);
+        // 标志该位置要求是否已经被填充
+        boolean[] flag = new boolean[3];
+        // 首先遍历一次锻造目标,将确定位置的步骤填充到lastOperations中,例如Hit_Last,Hit_Second_Last,Hit_Third_Last
+        // 因为hit的last和notlast相比于其他步骤是反序的,因此单独摘出来判断
+        for (ForgeRule rule : rules) {
+            // 这三种序号对5取余后分别是1,3,4,Hit_Last的序号是2,Hit_Not_Last的序号是1,单独摘出来判断
+            if ((rule.ordinal() != 1 && rule.ordinal() % 5 == 1 && !flag[0]) || rule.ordinal() == 2) {
+                lastOperations[0] = Util.operationsTfc.get(rule.ordinal());
+                flag[0] = true;
+            } else if (rule.ordinal() % 5 == 3 && !flag[1]) {
+                lastOperations[1] = Util.operationsTfc.get(rule.ordinal());
+                flag[1] = true;
+            } else if (rule.ordinal() % 5 == 4 && !flag[2]) {
+                lastOperations[2] = Util.operationsTfc.get(rule.ordinal());
+                flag[2] = true;
+            }
+        }
+        // 第二次遍历,填充可以位于倒数第二步和倒数第三步的步骤,例如Hit_Not_Last
+        // 其他步骤的Not_last序号对5取余后是2，Hit_Not_Last的序号是1，因此单独摘出来判断
+        for (ForgeRule rule : rules) {
+            // 这一步的序号对5取余后是2
+            if ((rule.ordinal() != 2 && rule.ordinal() % 5 == 2) || rule.ordinal() == 1){
+                // 如果倒数第三步已经填充,说明倒数第三步是已经定死的步骤,不可以更改,如果能进入这里的循环并且两个位置都已经填满,说明该锻造配方无法完成
+                // 所以当倒数第三步已经填充,就将倒数第二步填充为当前步骤,否则填充最后一步
+                if (flag[2]) {
+                    lastOperations[1] = Util.operationsTfc.get(rule.ordinal());
+                    flag[1] = true;
+                } else {
+                    lastOperations[2] = Util.operationsTfc.get(rule.ordinal());
+                    flag[2] = true;
+                }
+            }
+        }
+        // 最后一次遍历,填充剩余的步骤,这里的步骤是可以位于任意位置的步骤,例如BendAny
+        o: for (ForgeRule rule : rules) {
+            // 这个步骤的序号对5取余后是0
+            if (rule.ordinal() % 5 == 0) {
+                // 遍历lastOperations,如果有空位就填充,并且因为锻造需求里每步出现一次,所以只填充一次,跳出大循环
+                for (int i = 0; i < 3; i++) {
+                    if (!flag[i]) {
+                        lastOperations[i] = Util.operationsTfc.get(rule.ordinal());
+                        flag[i] = true;
+                        break o;
+                    }
+                }
+            }
+        }
+        // 如果还有空位,说明锻造需求不满3个,这时将空位填入4,对应锻造数值是2,此时未遍历的需求只有Any,对于Any,填入的数值依然是4
+        for (int i = 0; i < 3; i++) {
+            if (!flag[i]) {
+                lastOperations[i] = 4;
+            }
+        }
+        return lastOperations;
+    }
 }