stuff

1 files changed, 64 insertions, 0 deletions

diff --git a/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultShapelessDisplay.java b/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultShapelessDisplay.java
new file mode 100644
index 000000000..ca2f7db4a
--- /dev/null
+++ b/src/main/java/me/shedaniel/rei/plugin/crafting/DefaultShapelessDisplay.java
@@ -0,0 +1,64 @@
+/*
+ * Roughly Enough Items by Danielshe.
+ * Licensed under the MIT License.
+ */
+
+package me.shedaniel.rei.plugin.crafting;
+
+import net.minecraft.item.ItemStack;
+import net.minecraft.recipe.ShapelessRecipe;
+import net.minecraft.util.Identifier;
+
+import java.util.Arrays;
+import java.util.Collections;
+import java.util.List;
+import java.util.Optional;
+import java.util.stream.Collectors;
+
+public class DefaultShapelessDisplay implements DefaultCraftingDisplay {
+    
+    private ShapelessRecipe display;
+    private List<List<ItemStack>> input;
+    private List<ItemStack> output;
+    
+    public DefaultShapelessDisplay(ShapelessRecipe recipe) {
+        this.display = recipe;
+        this.input = recipe.getPreviewInputs().stream().map(i -> Arrays.asList(i.getStackArray())).collect(Collectors.toList());
+        this.output = Collections.singletonList(recipe.getOutput());
+    }
+    
+    @Override
+    public Optional<Identifier> getRecipeLocation() {
+        return Optional.ofNullable(display).map(ShapelessRecipe::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() {
+        if (display.getPreviewInputs().size() > 4)
+            return 3;
+        return 2;
+    }
+    
+    @Override
+    public int getHeight() {
+        if (display.getPreviewInputs().size() > 4)
+            return 3;
+        return 2;
+    }
+    
+}