1 files changed, 79 insertions, 0 deletions
diff --git a/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultCustomDisplay.java b/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultCustomDisplay.java
new file mode 100644
index 000000000..4bc83f233
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultCustomDisplay.java
@@ -0,0 +1,79 @@
+/*
+ * Roughly Enough Items by Danielshe.
+ * Licensed under the MIT License.
+ */
+
+package me.shedaniel.rei.plugin.crafting;
+
+import com.google.common.collect.Lists;
+import net.minecraft.item.ItemStack;
+import net.minecraft.recipe.Recipe;
+import net.minecraft.util.Identifier;
+
+import java.util.List;
+import java.util.Optional;
+
+public class DefaultCustomDisplay implements DefaultCraftingDisplay {
+    
+    private List<List<ItemStack>> input;
+    private List<ItemStack> output;
+    private Recipe possibleRecipe;
+    private int width, height;
+    
+    public DefaultCustomDisplay(List<List<ItemStack>> input, List<ItemStack> output, Recipe possibleRecipe) {
+        this.input = input;
+        this.output = output;
+        this.possibleRecipe = possibleRecipe;
+        List<Boolean> row = Lists.newArrayList(false, false, false);
+        List<Boolean> column = Lists.newArrayList(false, false, false);
+        for(int i = 0; i < 9; i++)
+            if (i < input.size()) {
+                List<ItemStack> stacks = input.get(i);
+                if (stacks.stream().filter(stack -> !stack.isEmpty()).count() > 0) {
+                    row.set((i - (i % 3)) / 3, true);
+                    column.set(i % 3, true);
+                }
+            }
+        this.width = (int) column.stream().filter(Boolean::booleanValue).count();
+        this.height = (int) row.stream().filter(Boolean::booleanValue).count();
+    }
+    
+    public DefaultCustomDisplay(List<List<ItemStack>> input, List<ItemStack> output) {
+        this(input, output, null);
+    }
+    
+    public Recipe getPossibleRecipe() {
+        return possibleRecipe;
+    }
+    
+    @Override
+    public Optional<Identifier> getRecipeLocation() {
+        return Optional.ofNullable(possibleRecipe).map(Recipe::getId);
+    }
+    
+    @Override
+    public List<List<ItemStack>> getInput() {
+        return input;
+    }
+    
+    @Override
+    public List<ItemStack> getOutput() {
+        return output;
+    }
+    
+    @Override
+    public List<List<ItemStack>> getRequiredItems() {
+        return input;
+    }
+    
+    @Override
+    public int getWidth() {
+        return width;
+    }
+    
+    @Override
+    public int getHeight() {
+        return height;
+    }
+    
+}