Don't go through MachineRecipeType caching on the client side

Technici4n · Technici4n · commit c723cc367c27 · 2024-11-25T01:03:58.000+01:00
diff --git a/src/client/java/aztech/modern_industrialization/compat/viewer/usage/MachineCategory.java b/src/client/java/aztech/modern_industrialization/compat/viewer/usage/MachineCategory.java
@@ -35,7 +35,6 @@
 import aztech.modern_industrialization.machines.gui.MachineScreen;
 import aztech.modern_industrialization.machines.guicomponents.EnergyBarClient;
 import aztech.modern_industrialization.machines.guicomponents.ProgressBarClient;
-import aztech.modern_industrialization.machines.init.MIMachineRecipeTypes;
 import aztech.modern_industrialization.machines.init.MachineTier;
 import aztech.modern_industrialization.machines.recipe.MachineRecipe;
 import aztech.modern_industrialization.thirdparty.fabrictransfer.api.fluid.FluidVariant;
@@ -93,8 +92,7 @@ public void buildWorkstations(WorkstationConsumer consumer) {
 
     @Override
     public void buildRecipes(RecipeManager recipeManager, RegistryAccess registryAccess, Consumer<RecipeHolder<MachineRecipe>> consumer) {
-        MIMachineRecipeTypes.getRecipeTypes().stream()
-                .flatMap(type -> type.getRecipes(Minecraft.getInstance().level).stream())
+        params.recipeType.getRecipesWithoutCache(Minecraft.getInstance().level).stream()
                 .filter(recipe -> params.recipePredicate.test(recipe.value()))
                 .sorted(Comparator.comparing(RecipeHolder::id))
                 .forEach(consumer);
diff --git a/src/main/java/aztech/modern_industrialization/compat/rei/machines/MachineCategoryParams.java b/src/main/java/aztech/modern_industrialization/compat/rei/machines/MachineCategoryParams.java
@@ -26,6 +26,7 @@
 import aztech.modern_industrialization.inventory.SlotPositions;
 import aztech.modern_industrialization.machines.guicomponents.ProgressBar;
 import aztech.modern_industrialization.machines.recipe.MachineRecipe;
+import aztech.modern_industrialization.machines.recipe.MachineRecipeType;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.function.Predicate;
@@ -39,21 +40,23 @@ public class MachineCategoryParams {
     public final SlotPositions fluidInputs;
     public final SlotPositions fluidOutputs;
     public final ProgressBar.Parameters progressBarParams;
+    public final MachineRecipeType recipeType;
     public final Predicate<MachineRecipe> recipePredicate;
     public final boolean isMultiblock;
     public final SteamMode steamMode;
     public final List<ResourceLocation> workstations = new ArrayList<>();
 
     public MachineCategoryParams(String englishName, ResourceLocation category, SlotPositions itemInputs, SlotPositions itemOutputs,
-            SlotPositions fluidInputs, SlotPositions fluidOutputs, ProgressBar.Parameters progressBarParams, Predicate<MachineRecipe> recipePredicate,
-            boolean isMultiblock, SteamMode steamMode) {
+            SlotPositions fluidInputs, SlotPositions fluidOutputs, ProgressBar.Parameters progressBarParams, MachineRecipeType recipeType,
+            Predicate<MachineRecipe> recipePredicate, boolean isMultiblock, SteamMode steamMode) {
         this.englishName = englishName;
         this.category = category;
         this.itemInputs = itemInputs;
         this.itemOutputs = itemOutputs;
         this.fluidInputs = fluidInputs;
         this.fluidOutputs = fluidOutputs;
         this.progressBarParams = progressBarParams;
+        this.recipeType = recipeType;
         this.recipePredicate = recipePredicate;
         this.isMultiblock = isMultiblock;
         this.steamMode = steamMode;
diff --git a/src/main/java/aztech/modern_industrialization/machines/blockentities/AbstractCraftingMachineBlockEntity.java b/src/main/java/aztech/modern_industrialization/machines/blockentities/AbstractCraftingMachineBlockEntity.java
@@ -39,7 +39,7 @@
 import aztech.modern_industrialization.machines.recipe.MachineRecipeType;
 import aztech.modern_industrialization.util.Tickable;
 import java.util.UUID;
-import net.minecraft.world.level.Level;
+import net.minecraft.server.level.ServerLevel;
 import org.jetbrains.annotations.Nullable;
 
 public abstract class AbstractCraftingMachineBlockEntity extends MachineBlockEntity implements CrafterComponent.Behavior, Tickable,
@@ -106,8 +106,8 @@ public MIInventory getInventory() {
     }
 
     @Override
-    public Level getCrafterWorld() {
-        return level;
+    public ServerLevel getCrafterWorld() {
+        return (ServerLevel) level;
     }
 
     @Override
diff --git a/src/main/java/aztech/modern_industrialization/machines/blockentities/multiblocks/AbstractElectricCraftingMultiblockBlockEntity.java b/src/main/java/aztech/modern_industrialization/machines/blockentities/multiblocks/AbstractElectricCraftingMultiblockBlockEntity.java
@@ -35,10 +35,10 @@
 import java.util.List;
 import java.util.UUID;
 import net.minecraft.core.Direction;
+import net.minecraft.server.level.ServerLevel;
 import net.minecraft.world.InteractionHand;
 import net.minecraft.world.ItemInteractionResult;
 import net.minecraft.world.entity.player.Player;
-import net.minecraft.world.level.Level;
 import org.jetbrains.annotations.Nullable;
 
 public abstract class AbstractElectricCraftingMultiblockBlockEntity extends AbstractCraftingMultiblockBlockEntity
@@ -109,8 +109,8 @@ public final long consumeEu(long max, Simulation simulation) {
     }
 
     @Override
-    public final Level getCrafterWorld() {
-        return level;
+    public final ServerLevel getCrafterWorld() {
+        return (ServerLevel) level;
     }
 
     @Override
diff --git a/src/main/java/aztech/modern_industrialization/machines/blockentities/multiblocks/SteamCraftingMultiblockBlockEntity.java b/src/main/java/aztech/modern_industrialization/machines/blockentities/multiblocks/SteamCraftingMultiblockBlockEntity.java
@@ -38,10 +38,10 @@
 import java.util.UUID;
 import net.minecraft.core.Direction;
 import net.minecraft.network.chat.Component;
+import net.minecraft.server.level.ServerLevel;
 import net.minecraft.world.InteractionHand;
 import net.minecraft.world.ItemInteractionResult;
 import net.minecraft.world.entity.player.Player;
-import net.minecraft.world.level.Level;
 import org.jetbrains.annotations.Nullable;
 
 public class SteamCraftingMultiblockBlockEntity extends AbstractCraftingMultiblockBlockEntity {
@@ -105,8 +105,8 @@ public long getMaxRecipeEu() {
         }
 
         @Override
-        public Level getCrafterWorld() {
-            return level;
+        public ServerLevel getCrafterWorld() {
+            return (ServerLevel) level;
         }
 
         @Override
diff --git a/src/main/java/aztech/modern_industrialization/machines/components/CrafterComponent.java b/src/main/java/aztech/modern_industrialization/machines/components/CrafterComponent.java
@@ -57,7 +57,6 @@
 import net.minecraft.world.item.Item;
 import net.minecraft.world.item.ItemStack;
 import net.minecraft.world.item.crafting.RecipeHolder;
-import net.minecraft.world.level.Level;
 import net.minecraft.world.level.material.Fluid;
 import org.jetbrains.annotations.Nullable;
 
@@ -104,7 +103,7 @@ default boolean isOverdriving() {
         }
 
         // can't use getWorld() or the remapping will fail
-        Level getCrafterWorld();
+        ServerLevel getCrafterWorld();
 
         default int getMaxFluidOutputs() {
             return Integer.MAX_VALUE;
@@ -627,7 +626,7 @@ protected void clearLocks() {
 
     public void lockRecipe(ResourceLocation recipeId, net.minecraft.world.entity.player.Inventory inventory) {
         // Find MachineRecipe
-        Optional<RecipeHolder<MachineRecipe>> optionalMachineRecipe = behavior.recipeType().getRecipes(behavior.getCrafterWorld()).stream()
+        Optional<RecipeHolder<MachineRecipe>> optionalMachineRecipe = behavior.recipeType().getRecipesWithCache(behavior.getCrafterWorld()).stream()
                 .filter(recipe -> recipe.id().equals(recipeId)).findFirst();
         if (optionalMachineRecipe.isEmpty())
             return;
diff --git a/src/main/java/aztech/modern_industrialization/machines/init/MultiblockMachines.java b/src/main/java/aztech/modern_industrialization/machines/init/MultiblockMachines.java
@@ -790,7 +790,7 @@ public final void register() {
 
             ReiMachineRecipes.registerCategory(category, new MachineCategoryParams(englishName, category,
                     itemInputs.build(), itemOutputs.build(), fluidInputs.build(), fluidOutputs.build(),
-                    progressBarParams, recipe -> recipe.getType() == recipeType && extraTest.test(recipe), true, steamMode));
+                    progressBarParams, recipeType, extraTest, true, steamMode));
             for (ResourceLocation workstation : workstations) {
                 ReiMachineRecipes.registerWorkstation(category, workstation);
                 ReiMachineRecipes.registerRecipeCategoryForMachine(workstation, category, ReiMachineRecipes.MachineScreenPredicate.MULTIBLOCK);
diff --git a/src/main/java/aztech/modern_industrialization/machines/init/SingleBlockCraftingMachines.java b/src/main/java/aztech/modern_industrialization/machines/init/SingleBlockCraftingMachines.java
@@ -245,7 +245,7 @@ public static void registerMachineTiers(String englishName, String machine, Mach
                 new MachineCategoryParams(null, null, items.sublist(0, itemInputCount),
                         items.sublist(itemInputCount, itemInputCount + itemOutputCount),
                         fluids.sublist(0, fluidInputCount), fluids.sublist(fluidInputCount, fluidInputCount + fluidOutputCount), progressBarParams,
-                        null, false, SteamMode.BOTH),
+                        null, null, false, SteamMode.BOTH),
                 tiers);
     }
 
@@ -265,7 +265,7 @@ private static void registerReiTiers(String englishName, String machine, Machine
                 MachineCategoryParams category = new MachineCategoryParams(fullEnglishName, MI.id(itemId), categoryParams.itemInputs,
                         categoryParams.itemOutputs,
                         categoryParams.fluidInputs, categoryParams.fluidOutputs, categoryParams.progressBarParams,
-                        recipe -> recipe.getType() == recipeType && minEu <= recipe.eu && recipe.eu <= maxEu, false,
+                        recipeType, recipe -> minEu <= recipe.eu && recipe.eu <= maxEu, false,
                         i < 2 ? SteamMode.BOTH : SteamMode.ELECTRIC_ONLY);
                 ReiMachineRecipes.registerCategory(MI.id(itemId), category);
                 ReiMachineRecipes.registerMachineClickArea(MI.id(itemId), categoryParams.progressBarParams.toRectangle());
diff --git a/src/main/java/aztech/modern_industrialization/machines/recipe/CentrifugeMachineRecipeType.java b/src/main/java/aztech/modern_industrialization/machines/recipe/CentrifugeMachineRecipeType.java
@@ -23,7 +23,9 @@
  */
 package aztech.modern_industrialization.machines.recipe;
 
+import java.util.List;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.crafting.RecipeHolder;
 import net.minecraft.world.level.Level;
 import net.minecraft.world.level.block.ComposterBlock;
 
@@ -34,7 +36,7 @@ public CentrifugeMachineRecipeType(ResourceLocation id) {
     }
 
     @Override
-    protected void fillRecipeList(Level world) {
+    protected void fillRecipeList(Level world, List<RecipeHolder<MachineRecipe>> recipeList) {
         recipeList.addAll(getManagerRecipes(world));
 
         for (var itemCompostable : ComposterBlock.COMPOSTABLES.keySet()) {
diff --git a/src/main/java/aztech/modern_industrialization/machines/recipe/CuttingMachineRecipeType.java b/src/main/java/aztech/modern_industrialization/machines/recipe/CuttingMachineRecipeType.java
@@ -23,7 +23,9 @@
  */
 package aztech.modern_industrialization.machines.recipe;
 
+import java.util.List;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.crafting.RecipeHolder;
 import net.minecraft.world.item.crafting.RecipeType;
 import net.minecraft.world.level.Level;
 
@@ -33,7 +35,7 @@ public CuttingMachineRecipeType(ResourceLocation id) {
     }
 
     @Override
-    protected void fillRecipeList(Level world) {
+    protected void fillRecipeList(Level world, List<RecipeHolder<MachineRecipe>> recipeList) {
         // Add all regular cutting machine recipes
         recipeList.addAll(getManagerRecipes(world));
         // Add all stone cutter recipes
diff --git a/src/main/java/aztech/modern_industrialization/machines/recipe/FurnaceRecipeProxy.java b/src/main/java/aztech/modern_industrialization/machines/recipe/FurnaceRecipeProxy.java
@@ -26,6 +26,7 @@
 import aztech.modern_industrialization.MI;
 import java.util.*;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.world.item.crafting.RecipeHolder;
 import net.minecraft.world.item.crafting.RecipeType;
 import net.minecraft.world.level.Level;
 
@@ -34,7 +35,7 @@ public FurnaceRecipeProxy(ResourceLocation id) {
         super(id);
     }
 
-    protected void fillRecipeList(Level world) {
+    protected void fillRecipeList(Level world, List<RecipeHolder<MachineRecipe>> recipeList) {
         for (var smeltingRecipe : world.getRecipeManager().getAllRecipesFor(RecipeType.SMELTING)) {
             var recipe = RecipeConversions.ofSmelting(smeltingRecipe, this, world.registryAccess());
             recipeList.add(recipe);
diff --git a/src/main/java/aztech/modern_industrialization/machines/recipe/MachineRecipeType.java b/src/main/java/aztech/modern_industrialization/machines/recipe/MachineRecipeType.java
@@ -23,7 +23,6 @@
  */
 package aztech.modern_industrialization.machines.recipe;
 
-import com.google.gson.*;
 import com.mojang.serialization.DataResult;
 import com.mojang.serialization.MapCodec;
 import java.util.*;
@@ -59,17 +58,21 @@ public MachineRecipeType(ResourceLocation id) {
     /**
      * Never modify or store the result!
      */
-    protected Collection<RecipeHolder<MachineRecipe>> getManagerRecipes(Level world) {
-        return world.getRecipeManager().getAllRecipesFor(this);
+    protected Collection<RecipeHolder<MachineRecipe>> getManagerRecipes(Level level) {
+        return level.getRecipeManager().getAllRecipesFor(this);
     }
 
-    public Collection<RecipeHolder<MachineRecipe>> getRecipes(Level world) {
-        return getManagerRecipes(world);
+    public Collection<RecipeHolder<MachineRecipe>> getRecipesWithoutCache(Level level) {
+        return getManagerRecipes(level);
+    }
+
+    public Collection<RecipeHolder<MachineRecipe>> getRecipesWithCache(ServerLevel level) {
+        return getManagerRecipes(level);
     }
 
     @Nullable
-    public RecipeHolder<MachineRecipe> getRecipe(Level world, ResourceLocation id) {
-        return getRecipes(world).stream().filter(r -> r.id().equals(id)).findFirst().orElse(null);
+    public RecipeHolder<MachineRecipe> getRecipe(ServerLevel world, ResourceLocation id) {
+        return getRecipesWithCache(world).stream().filter(r -> r.id().equals(id)).findFirst().orElse(null);
     }
 
     /*
@@ -97,7 +100,7 @@ private void updateRecipeCache(ServerLevel world) {
         lastUpdate = time;
         recipeCache.clear();
         fluidOnlyRecipes.clear();
-        for (RecipeHolder<MachineRecipe> recipe : getRecipes(world)) {
+        for (RecipeHolder<MachineRecipe> recipe : getRecipesWithCache(world)) {
             if (recipe.value().itemInputs.size() == 0) {
                 if (recipe.value().fluidInputs.size() > 0) {
                     fluidOnlyRecipes.add(recipe);
diff --git a/src/main/java/aztech/modern_industrialization/machines/recipe/ProxyableMachineRecipeType.java b/src/main/java/aztech/modern_industrialization/machines/recipe/ProxyableMachineRecipeType.java
@@ -27,6 +27,7 @@
 import java.util.Collection;
 import java.util.List;
 import net.minecraft.resources.ResourceLocation;
+import net.minecraft.server.level.ServerLevel;
 import net.minecraft.world.item.crafting.RecipeHolder;
 import net.minecraft.world.level.Level;
 
@@ -40,17 +41,24 @@ public ProxyableMachineRecipeType(ResourceLocation id) {
 
     private long lastUpdate = 0;
     private static final long UPDATE_INTERVAL = 20 * 1000;
-    protected List<RecipeHolder<MachineRecipe>> recipeList = new ArrayList<>();
+    private final List<RecipeHolder<MachineRecipe>> recipeList = new ArrayList<>();
 
-    protected abstract void fillRecipeList(Level world);
+    protected abstract void fillRecipeList(Level world, List<RecipeHolder<MachineRecipe>> recipeList);
 
     @Override
-    public Collection<RecipeHolder<MachineRecipe>> getRecipes(Level world) {
+    public final Collection<RecipeHolder<MachineRecipe>> getRecipesWithoutCache(Level level) {
+        var recipes = new ArrayList<RecipeHolder<MachineRecipe>>();
+        fillRecipeList(level, recipes);
+        return recipes;
+    }
+
+    @Override
+    public final Collection<RecipeHolder<MachineRecipe>> getRecipesWithCache(ServerLevel level) {
         long time = System.currentTimeMillis();
         if (time - lastUpdate > UPDATE_INTERVAL) {
             lastUpdate = time;
             recipeList.clear();
-            fillRecipeList(world);
+            fillRecipeList(level, recipeList);
         }
         return recipeList;
     }
