1 files changed, 10 insertions, 8 deletions
diff --git a/forge/src/main/java/me/shedaniel/rei/plugin/common/displays/crafting/forge/DefaultCraftingDisplayImpl.java b/forge/src/main/java/me/shedaniel/rei/plugin/common/displays/crafting/forge/DefaultCraftingDisplayImpl.java
index 4f809e306..9f5faa713 100644
--- a/forge/src/main/java/me/shedaniel/rei/plugin/common/displays/crafting/forge/DefaultCraftingDisplayImpl.java
+++ b/forge/src/main/java/me/shedaniel/rei/plugin/common/displays/crafting/forge/DefaultCraftingDisplayImpl.java
@@ -23,16 +23,18 @@
 
 package me.shedaniel.rei.plugin.common.displays.crafting.forge;
 
-import net.minecraft.world.item.crafting.Recipe;
+import me.shedaniel.rei.plugin.common.displays.crafting.CraftingRecipeSizeProvider;
+import me.shedaniel.rei.plugin.common.displays.crafting.DefaultCraftingDisplay;
 import net.minecraftforge.common.crafting.IShapedRecipe;
-import org.apache.commons.lang3.tuple.Pair;
 
 public class DefaultCraftingDisplayImpl {
-    public static Pair<Integer, Integer> getSize(Recipe<?> recipe) {
-        if (recipe instanceof IShapedRecipe) {
-            return Pair.of(((IShapedRecipe<?>) recipe).getRecipeWidth(), ((IShapedRecipe<?>) recipe).getRecipeHeight());
-        }
-        
-        return null;
+    public static void registerPlatformSizeProvider() {
+        DefaultCraftingDisplay.registerSizeProvider(recipe -> {
+            if (recipe instanceof IShapedRecipe) {
+                return new CraftingRecipeSizeProvider.Size(((IShapedRecipe<?>) recipe).getRecipeWidth(), ((IShapedRecipe<?>) recipe).getRecipeHeight());
+            }
+            
+            return null;
+        });
     }
 }