|
17 | 17 |
|
18 | 18 | package de.siphalor.nbtcrafting.mixin; |
19 | 19 |
|
20 | | -import java.util.ArrayList; |
21 | | -import java.util.List; |
22 | | -import java.util.Map; |
| 20 | +import de.siphalor.nbtcrafting.api.RecipeUtil; |
| 21 | +import de.siphalor.nbtcrafting.api.recipe.NBTCRecipe; |
| 22 | +import de.siphalor.nbtcrafting.ingredient.IIngredient; |
23 | 23 |
|
24 | 24 | import net.minecraft.inventory.Inventory; |
25 | 25 | import net.minecraft.item.ItemStack; |
|
28 | 28 | import net.minecraft.util.collection.DefaultedList; |
29 | 29 | import org.apache.commons.lang3.ArrayUtils; |
30 | 30 | import org.spongepowered.asm.mixin.Mixin; |
31 | | -import org.spongepowered.asm.mixin.Overwrite; |
32 | 31 | import org.spongepowered.asm.mixin.Shadow; |
| 32 | +import org.spongepowered.asm.mixin.injection.At; |
| 33 | +import org.spongepowered.asm.mixin.injection.Inject; |
| 34 | +import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable; |
| 35 | +import org.spongepowered.asm.mixin.injection.callback.LocalCapture; |
33 | 36 |
|
34 | | -import de.siphalor.nbtcrafting.api.RecipeUtil; |
35 | | -import de.siphalor.nbtcrafting.api.recipe.NBTCRecipe; |
36 | | -import de.siphalor.nbtcrafting.ingredient.IIngredient; |
| 37 | +import java.util.ArrayList; |
| 38 | +import java.util.List; |
| 39 | +import java.util.Map; |
37 | 40 |
|
38 | 41 | @Mixin(Recipe.class) |
39 | 42 | public interface MixinRecipe { |
40 | 43 | @Shadow |
41 | 44 | DefaultedList<Ingredient> getIngredients(); |
42 | 45 |
|
43 | | - /** |
44 | | - * @reason Returns the recipe remainders. Sadly has to overwrite since this is an interface. |
45 | | - * @author Siphalor |
46 | | - */ |
47 | | - @Overwrite |
48 | | - default DefaultedList<ItemStack> getRemainder(Inventory inventory) { |
49 | | - final DefaultedList<ItemStack> stackList = DefaultedList.ofSize(inventory.size(), ItemStack.EMPTY); |
50 | | - Map<String, Object> reference; |
| 46 | + @Inject(method = "getRemainder", at = @At("RETURN"), locals = LocalCapture.CAPTURE_FAILSOFT) |
| 47 | + default void modifyRemainingStacks(Inventory inventory, CallbackInfoReturnable<DefaultedList<ItemStack>> cir, DefaultedList<ItemStack> stackList) { |
| 48 | + // The stackList is already pre-populated with Vanilla and Fabric API remainders |
51 | 49 | List<Ingredient> ingredients; |
52 | | - int[] resolvedIngredientStacks; |
53 | | - if (this instanceof NBTCRecipe) { |
| 50 | + boolean customRecipe = this instanceof NBTCRecipe; |
| 51 | + if (customRecipe) { |
54 | 52 | ingredients = new ArrayList<>(((NBTCRecipe<?>) this).getIngredients()); |
55 | | - // noinspection unchecked |
56 | | - reference = ((NBTCRecipe<Inventory>) this).buildDollarReference(inventory); |
57 | | - resolvedIngredientStacks = RecipeUtil.resolveIngredients(ingredients, inventory); |
58 | 53 | } else { |
59 | 54 | ingredients = getIngredients(); |
60 | | - resolvedIngredientStacks = RecipeUtil.resolveIngredients(ingredients, inventory); |
| 55 | + } |
| 56 | + |
| 57 | + boolean shallContinue = false; |
| 58 | + for (Ingredient ingredient : ingredients) { |
| 59 | + if (((IIngredient) (Object) ingredient).nbtCrafting$isAdvanced()) { |
| 60 | + shallContinue = true; |
| 61 | + break; |
| 62 | + } |
| 63 | + } |
| 64 | + if (!shallContinue) { |
| 65 | + return; |
| 66 | + } |
| 67 | + |
| 68 | + // Resolve the ingredient indexes to the belonging stack indices |
| 69 | + int[] resolvedIngredientStacks = RecipeUtil.resolveIngredients(ingredients, inventory); |
| 70 | + Map<String, Object> reference; |
| 71 | + |
| 72 | + if (customRecipe) { |
| 73 | + // noinspection unchecked |
| 74 | + reference = ((NBTCRecipe<Inventory>) this).buildDollarReference(inventory, resolvedIngredientStacks); |
| 75 | + } else { |
61 | 76 | reference = RecipeUtil.buildReferenceMapFromResolvedIngredients(resolvedIngredientStacks, inventory); |
62 | 77 | } |
63 | 78 |
|
64 | 79 | for (int i = 0; i < stackList.size(); ++i) { |
65 | 80 | ItemStack stack = inventory.getStack(i); |
66 | 81 | int ingredientIndex = ArrayUtils.indexOf(resolvedIngredientStacks, i); |
67 | 82 | if (ingredientIndex >= 0) { |
68 | | - ItemStack remainder = ((IIngredient) (Object) ingredients.get(ingredientIndex)).nbtCrafting$getRecipeRemainder(stack, reference); |
| 83 | + IIngredient ingredient = (IIngredient) (Object) ingredients.get(ingredientIndex); |
| 84 | + // Simple, Vanilla-ish entries should already be set |
| 85 | + if (!ingredient.nbtCrafting$isAdvanced()) { |
| 86 | + continue; |
| 87 | + } |
| 88 | + |
| 89 | + ItemStack remainder = ingredient.nbtCrafting$getRecipeRemainder(stack, reference); |
69 | 90 | if (remainder != null) { |
70 | 91 | stackList.set(i, remainder); |
71 | | - continue; |
72 | 92 | } |
73 | 93 | } |
74 | | - if (stack.getItem().hasRecipeRemainder()) { |
75 | | - stackList.set(i, new ItemStack(stack.getItem().getRecipeRemainder())); |
76 | | - } |
77 | 94 | } |
78 | | - return stackList; |
79 | 95 | } |
80 | 96 | } |
0 commit comments