Binnie/src/main/java/binnie/genetics/nei/InoculatorRecipeHandler.java

package binnie.genetics.nei;

import static binnie.genetics.item.GeneticsItems.serum;
import static binnie.genetics.item.GeneticsItems.serumArray;

import java.awt.Rectangle;
import java.util.ArrayList;
import java.util.List;

import net.minecraft.item.ItemStack;
import net.minecraft.util.StatCollector;
import net.minecraftforge.fluids.FluidStack;
import net.minecraftforge.oredict.OreDictionary;

import binnie.botany.items.BotanyItems;
import binnie.core.BinnieCore;
import binnie.core.Mods;
import binnie.core.nei.NEIUtils;
import binnie.core.nei.PositionedFluidTank;
import binnie.core.nei.RecipeHandlerBase;
import binnie.genetics.api.IItemSerum;
import binnie.genetics.item.GeneticLiquid;
import codechicken.lib.gui.GuiDraw;
import codechicken.nei.NEIServerUtils;
import codechicken.nei.PositionedStack;
import forestry.api.genetics.AlleleManager;
import forestry.api.genetics.ISpeciesRoot;

public class InoculatorRecipeHandler extends RecipeHandlerBase {

    private static final List<InoculatorRecipe> recipes = new ArrayList<>();

    private void createInoculatorRecipeForNEI(ItemStack dnaManipulable) {
        recipes.add(new InoculatorRecipe(dnaManipulable, new ItemStack(serum, 1, OreDictionary.WILDCARD_VALUE)));
        recipes.add(new InoculatorRecipe(dnaManipulable, new ItemStack(serumArray, 1, OreDictionary.WILDCARD_VALUE)));
    }

    @Override
    public void prepare() {
        if (BinnieCore.isApicultureActive())
            createInoculatorRecipeForNEI(Mods.forestry.stack("beeLarvaeGE", 1, OreDictionary.WILDCARD_VALUE));
        if (BinnieCore.isArboricultureActive())
            createInoculatorRecipeForNEI(Mods.forestry.stack("pollenFertile", 1, OreDictionary.WILDCARD_VALUE));
        if (BinnieCore.isLepidopteryActive())
            createInoculatorRecipeForNEI(Mods.forestry.stack("serumGE", 1, OreDictionary.WILDCARD_VALUE));

        createInoculatorRecipeForNEI(new ItemStack(BotanyItems.pollen, 1, OreDictionary.WILDCARD_VALUE));
    }

    @Override
    public String getOverlayIdentifier() {
        return "genetics.inoculator";
    }

    @Override
    public String getGuiTexture() {
        return "genetics:textures/gui/nei/inoculator.png";
    }

    @Override
    public String getRecipeName() {
        return StatCollector.translateToLocal("genetics.machine.machine.inoculator");
    }

    @Override
    public void loadTransferRects() {
        this.addTransferRect(75, 27, 24, 17);
    }

    @Override
    public void drawBackground(int recipe) {
        changeToGuiTexture();
        GuiDraw.drawTexturedModalRect(0, 0, 5, 11, 166, 74);
    }

    @Override
    public void drawExtras(int recipe) {
        this.drawProgressBar(75, 27, 176, 0, 24, 17, 40, 0);
    }

    @Override
    public void loadAllRecipes() {
        for (InoculatorRecipe recipe : recipes) {
            this.arecipes.add(new CachedInoculatorRecipe(recipe));
        }
    }

    @Override
    public void loadCraftingRecipes(ItemStack result) {
        for (InoculatorRecipe recipe : recipes) {
            if (NEIServerUtils.areStacksSameTypeCrafting(recipe.getDnaManipulable(), result)) {
                this.arecipes.add(new CachedInoculatorRecipe(recipe));
            }
        }
    }

    @Override
    public void loadUsageRecipes(FluidStack ingredient) {
        if (NEIUtils.areFluidsSameType(GeneticLiquid.BacteriaVector.get(0), ingredient)) {
            this.loadAllRecipes();
        }
    }

    @Override
    public void loadUsageRecipes(ItemStack ingredient) {
        for (InoculatorRecipe recipe : recipes) {
            if (NEIServerUtils.areStacksSameTypeCrafting(recipe.getDnaManipulable(), ingredient)
                    || NEIServerUtils.areStacksSameTypeCrafting(recipe.getSerum(), ingredient)) {
                this.arecipes.add(new CachedInoculatorRecipe(recipe));
            }
        }
    }

    private static class InoculatorRecipe {

        private final ItemStack dnaManipulable;
        private final ItemStack serum;

        public InoculatorRecipe(ItemStack dnaManipulable, ItemStack serum) {
            this.dnaManipulable = dnaManipulable;
            this.serum = serum;
        }

        public FluidStack getBacteriaVector() {
            return GeneticLiquid.BacteriaVector.get(15);
        }

        public ItemStack getDnaManipulable() {
            return this.dnaManipulable;
        }

        public ItemStack getSerum() {
            return this.serum;
        }
    }

    public class CachedInoculatorRecipe extends CachedBaseRecipe {

        public PositionedFluidTank bacteriaTank;
        public PositionedStack serum;
        public PositionedStack input;
        public PositionedStack output;
        public List<PositionedStack> ingredients = new ArrayList<>();

        public CachedInoculatorRecipe(InoculatorRecipe recipe) {

            if (recipe.getSerum() != null && recipe.getDnaManipulable() != null) {
                ISpeciesRoot root1 = AlleleManager.alleleRegistry.getSpeciesRoot(recipe.getDnaManipulable());
                List<ItemStack> serums = new ArrayList<>(200);
                List<ItemStack> allSerumList = new ArrayList<>(1000);
                recipe.serum.getItem().getSubItems(recipe.serum.getItem(), null, allSerumList);
                for (ItemStack serumStack : allSerumList) {
                    if (serumStack.getTagCompound() != null) {
                        IItemSerum itemSerum = (IItemSerum) serumStack.getItem();
                        ISpeciesRoot root2 = itemSerum.getSpeciesRoot(serumStack);
                        if (root1 == root2) serums.add(serumStack.copy());
                    }
                }

                this.bacteriaTank = new PositionedFluidTank(
                        recipe.getBacteriaVector(),
                        100,
                        new Rectangle(28, 6, 16, 58));
                this.serum = new PositionedStack(serums, 53, 6);
                this.input = new PositionedStack(recipe.getDnaManipulable(), 53, 27);
                this.output = new PositionedStack(recipe.getDnaManipulable(), 105, 27);

                this.ingredients.add(this.serum);
                this.ingredients.add(this.input);
            }
        }

        @Override
        public List<PositionedStack> getIngredients() {
            this.input.setPermutationToRender(cycleticks / 40 % this.input.items.length);
            this.serum.setPermutationToRender(cycleticks / 40 % this.serum.items.length);
            return this.ingredients;
        }

        @Override
        public PositionedStack getResult() {
            this.output.setPermutationToRender(cycleticks / 40 % this.output.items.length);
            return this.output;
        }

        @Override
        public PositionedFluidTank getFluidTank() {
            return this.bacteriaTank;
        }
    }
}