-
Notifications
You must be signed in to change notification settings - Fork 61
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
5 changed files
with
177 additions
and
4 deletions.
There are no files selected for viewing
7 changes: 7 additions & 0 deletions
7
src/main/java/com/glodblock/github/nei/object/IRecipeExtractorLegacy.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,7 @@ | ||
| package com.glodblock.github.nei.object; | ||
|
|
||
| public interface IRecipeExtractorLegacy extends IRecipeExtractor { | ||
|
|
||
| String getClassName(); | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
129 changes: 129 additions & 0 deletions
129
src/main/java/com/glodblock/github/nei/recipes/extractor/ForestryRecipeExtractor.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,129 @@ | ||
| package com.glodblock.github.nei.recipes.extractor; | ||
|
|
||
| import codechicken.nei.PositionedStack; | ||
| import codechicken.nei.recipe.IRecipeHandler; | ||
| import codechicken.nei.recipe.TemplateRecipeHandler; | ||
| import com.glodblock.github.nei.object.IRecipeExtractorLegacy; | ||
| import com.glodblock.github.nei.object.OrderStack; | ||
| import crazypants.enderio.nei.VatRecipeHandler; | ||
| import forestry.core.recipes.nei.PositionedFluidTank; | ||
| import forestry.core.recipes.nei.RecipeHandlerBase; | ||
| import forestry.factory.recipes.nei.NEIHandlerFabricator; | ||
| import forestry.factory.recipes.nei.NEIHandlerSqueezer; | ||
| import net.minecraft.item.ItemStack; | ||
| import net.minecraftforge.fluids.FluidStack; | ||
|
|
||
| import java.util.ArrayList; | ||
| import java.util.LinkedList; | ||
| import java.util.List; | ||
| import java.util.Objects; | ||
| import java.util.stream.Collectors; | ||
|
|
||
| public class ForestryRecipeExtractor implements IRecipeExtractorLegacy { | ||
|
|
||
| private final IRecipeHandler handler; | ||
|
|
||
| public ForestryRecipeExtractor(IRecipeHandler recipes) { | ||
| handler = recipes; | ||
| } | ||
|
|
||
| @Override | ||
| public String getClassName() { | ||
| return handler.getHandlerId(); | ||
| } | ||
|
|
||
| @Override | ||
| public List<OrderStack<?>> getInputIngredients(List<PositionedStack> rawInputs) { | ||
| List<OrderStack<?>> tmp = new LinkedList<>(); | ||
| for (int i = 0; i < rawInputs.size(); i ++) { | ||
| if (rawInputs.get(i) == null) continue; | ||
| OrderStack<?> stack = OrderStack.pack(rawInputs.get(i), i); | ||
| if (stack != null) tmp.add(stack); | ||
| } | ||
| return tmp; | ||
| } | ||
|
|
||
| @Override | ||
| public List<OrderStack<?>> getOutputIngredients(List<PositionedStack> rawOutputs) { | ||
| List<OrderStack<?>> tmp = new LinkedList<>(); | ||
| for (int i = 0; i < rawOutputs.size(); i ++) { | ||
| if (rawOutputs.get(i) == null) continue; | ||
| OrderStack<?> stack = OrderStack.pack(rawOutputs.get(i), i); | ||
| if (stack != null) tmp.add(stack); | ||
| } | ||
| return tmp; | ||
| } | ||
|
|
||
| @Override | ||
| public List<OrderStack<?>> getInputIngredients(List<PositionedStack> rawInputs, IRecipeHandler recipe, int index) { | ||
| TemplateRecipeHandler tRecipe = (TemplateRecipeHandler) recipe; | ||
| List<OrderStack<?>> tmp = new ArrayList<>(); | ||
| List<PositionedStack> compressed = compress(rawInputs); | ||
| if (tRecipe.arecipes.get(index) instanceof RecipeHandlerBase.CachedBaseRecipe) { | ||
| tmp = getInputIngredients(compressed); | ||
| List<PositionedFluidTank> tanks = ((RecipeHandlerBase.CachedBaseRecipe) tRecipe.arecipes.get(index)).getFluidTanks(); | ||
| if (tanks.size() > 0 && !(handler instanceof NEIHandlerSqueezer)) { | ||
| FluidStack fluid = tanks.get(0).tank.getFluid(); | ||
| if (fluid != null) { | ||
| tmp.add(new OrderStack<>(fluid, compressed.size())); | ||
| } | ||
| } | ||
| } | ||
| return tmp; | ||
| } | ||
|
|
||
| @Override | ||
| public List<OrderStack<?>> getOutputIngredients(List<PositionedStack> rawOutputs, IRecipeHandler recipe, int index) { | ||
| TemplateRecipeHandler tRecipe = (TemplateRecipeHandler) recipe; | ||
| removeGlass(rawOutputs); | ||
| List<OrderStack<?>> tmp = new ArrayList<>(); | ||
| List<PositionedStack> compressed = compress(rawOutputs); | ||
| if (tRecipe.arecipes.get(index) instanceof RecipeHandlerBase.CachedBaseRecipe) { | ||
| tmp = getOutputIngredients(compressed); | ||
| List<PositionedFluidTank> tanks = ((RecipeHandlerBase.CachedBaseRecipe) tRecipe.arecipes.get(index)).getFluidTanks(); | ||
| if (tanks.size() > 0 && handler instanceof NEIHandlerSqueezer) { | ||
| FluidStack fluid = tanks.get(0).tank.getFluid(); | ||
| if (fluid != null) { | ||
| tmp.add(new OrderStack<>(fluid, compressed.size())); | ||
| } | ||
| } | ||
| else if (tanks.size() > 1) { | ||
| FluidStack fluid = tanks.get(1).tank.getFluid(); | ||
| if (fluid != null) { | ||
| tmp.add(new OrderStack<>(fluid, compressed.size())); | ||
| } | ||
| } | ||
| } | ||
| return tmp; | ||
| } | ||
|
|
||
| private List<PositionedStack> compress(List<PositionedStack> list) { | ||
| List<PositionedStack> comp = new LinkedList<>(); | ||
| for (PositionedStack positionedStack : list) { | ||
| if (positionedStack == null) continue; | ||
| ItemStack currentStack = positionedStack.items[0].copy(); | ||
| if (currentStack.stackSize == 0) continue; | ||
| boolean find = false; | ||
| for (PositionedStack storedStack : comp) { | ||
| if (storedStack == null) continue; | ||
| ItemStack firstStack = storedStack.items[0].copy(); | ||
| boolean areItemStackEqual = firstStack.isItemEqual(currentStack) && ItemStack.areItemStackTagsEqual(firstStack, currentStack); | ||
| if (areItemStackEqual && (firstStack.stackSize + currentStack.stackSize) <= firstStack.getMaxStackSize()) { | ||
| find = true; | ||
| storedStack.items[0].stackSize = firstStack.stackSize + currentStack.stackSize; | ||
| } | ||
| } | ||
| if (!find) { | ||
| comp.add(positionedStack.copy()); | ||
| } | ||
| } | ||
| return comp.stream().filter(Objects::nonNull).collect(Collectors.toList()); | ||
| } | ||
|
|
||
| private void removeGlass(List<PositionedStack> list) { | ||
| if (handler instanceof NEIHandlerFabricator) { | ||
| list.remove(list.size() - 1); | ||
| } | ||
| } | ||
|
|
||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters