added iterator override to support custom recipes.

GregTechCEu · Feb 4, 2024 · 3e0b96b · 3e0b96b
1 parent 89f3b5a
commit 3e0b96b
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 0 deletions.
diff --git a/src/main/java/gregtech/api/recipes/machines/RecipeMapFluidCanner.java b/src/main/java/gregtech/api/recipes/machines/RecipeMapFluidCanner.java
@@ -16,6 +16,8 @@
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
+import java.util.Collections;
+import java.util.Iterator;
 import java.util.List;
 
 @ApiStatus.Internal
@@ -83,4 +85,11 @@ public Recipe findRecipe(long voltage, List<ItemStack> inputs, List<FluidStack>
         }
         return null;
     }
+
+    @Override
+    @NotNull
+    public Iterator<Recipe> getRecipeIterator(long voltage, List<ItemStack> inputs, List<FluidStack> fluidInputs,
+                                              boolean exactVoltage) {
+        return Collections.singleton(this.findRecipe(voltage, inputs, fluidInputs, exactVoltage)).iterator();
+    }
 }
diff --git a/src/main/java/gregtech/api/recipes/machines/RecipeMapFormingPress.java b/src/main/java/gregtech/api/recipes/machines/RecipeMapFormingPress.java
@@ -17,6 +17,8 @@
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
+import java.util.Collections;
+import java.util.Iterator;
 import java.util.List;
 
 @ApiStatus.Internal
@@ -77,4 +79,11 @@ public Recipe findRecipe(long voltage, List<ItemStack> inputs, List<FluidStack>
         }
         return recipe;
     }
+
+    @Override
+    @NotNull
+    public Iterator<Recipe> getRecipeIterator(long voltage, List<ItemStack> inputs, List<FluidStack> fluidInputs,
+                                              boolean exactVoltage) {
+        return Collections.singleton(this.findRecipe(voltage, inputs, fluidInputs, exactVoltage)).iterator();
+    }
 }
diff --git a/src/main/java/gregtech/api/recipes/machines/RecipeMapScanner.java b/src/main/java/gregtech/api/recipes/machines/RecipeMapScanner.java
@@ -14,6 +14,8 @@
 import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Iterator;
 import java.util.List;
 
 @ApiStatus.Internal
@@ -60,4 +62,11 @@ public Recipe findRecipe(long voltage, List<ItemStack> inputs, List<FluidStack>
         }
         return null;
     }
+
+    @Override
+    @NotNull
+    public Iterator<Recipe> getRecipeIterator(long voltage, List<ItemStack> inputs, List<FluidStack> fluidInputs,
+                                              boolean exactVoltage) {
+        return Collections.singleton(this.findRecipe(voltage, inputs, fluidInputs, exactVoltage)).iterator();
+    }
 }