path: root/src/main/java/goodgenerator/loader/ComponentAssemblyLineRecipeLoader.java

package goodgenerator.loader;

import static goodgenerator.util.StackUtils.getTotalItems;
import static goodgenerator.util.StackUtils.mergeStacks;
import static goodgenerator.util.StackUtils.multiplyAndSplitIntoStacks;
import static gregtech.api.util.GTRecipeConstants.COAL_CASING_TIER;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.stream.Collectors;

import javax.annotation.Nullable;

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

import org.apache.commons.lang3.tuple.Pair;

import goodgenerator.api.recipe.GoodGeneratorRecipeMaps;
import gregtech.api.enums.GTValues;
import gregtech.api.enums.ItemList;
import gregtech.api.enums.Materials;
import gregtech.api.enums.MaterialsUEVplus;
import gregtech.api.enums.OrePrefixes;
import gregtech.api.objects.ItemData;
import gregtech.api.recipe.RecipeMaps;
import gregtech.api.util.GTOreDictUnificator;
import gregtech.api.util.GTRecipe;
import gregtech.api.util.GTUtility;
import gregtech.common.items.ItemIntegratedCircuit;

public class ComponentAssemblyLineRecipeLoader {

    private static final String[] compPrefixes = { "Electric_Motor_", "Electric_Piston_", "Electric_Pump_",
        "Robot_Arm_", "Conveyor_Module_", "Emitter_", "Sensor_", "Field_Generator_", };
    private static final String[] blacklistedDictPrefixes = { "circuit" };
    private static final String[] softBlacklistedDictPrefixes = { "Any", "crafting", "nanite" };
    private static final String moltenMHDCSM = "molten.magnetohydrodynamicallyconstrainedstarmatter";

    private static LinkedHashMap<List<GTRecipe>, Pair<ItemList, Integer>> allAssemblerRecipes;
    private static LinkedHashMap<List<GTRecipe.RecipeAssemblyLine>, Pair<ItemList, Integer>> allAsslineRecipes;

    private static final HashMap<OrePrefixes, Double> magnetoConversionMultipliers = new HashMap<>();
    private static final HashMap<OrePrefixes, OrePrefixes> conversion = new HashMap<>();

    private static final int INPUT_MULTIPLIER = 48;
    private static final int OUTPUT_MULTIPLIER = 64;
    private static final int FLUID_CONVERSION_STACKSIZE_THRESHOLD = 64;

    public static void run() {
        ComponentAssemblyLineMiscRecipes.run();
        conversion.put(OrePrefixes.cableGt01, OrePrefixes.cableGt16);
        conversion.put(OrePrefixes.wireGt01, OrePrefixes.wireGt16);
        conversion.put(OrePrefixes.cableGt02, OrePrefixes.cableGt16);
        conversion.put(OrePrefixes.wireGt02, OrePrefixes.wireGt16);
        conversion.put(OrePrefixes.cableGt04, OrePrefixes.cableGt16);
        conversion.put(OrePrefixes.wireGt04, OrePrefixes.wireGt16);
        conversion.put(OrePrefixes.cableGt08, OrePrefixes.cableGt16);
        conversion.put(OrePrefixes.wireGt08, OrePrefixes.wireGt16);
        conversion.put(OrePrefixes.plate, OrePrefixes.plateDense);
        conversion.put(OrePrefixes.foil, OrePrefixes.plate);
        conversion.put(OrePrefixes.stick, OrePrefixes.stickLong);
        conversion.put(OrePrefixes.gearGtSmall, OrePrefixes.gearGt);
        magnetoConversionMultipliers.put(OrePrefixes.frameGt, 1.0);
        magnetoConversionMultipliers.put(OrePrefixes.plate, 1.0);
        magnetoConversionMultipliers.put(OrePrefixes.plateDense, 3.0);
        magnetoConversionMultipliers.put(OrePrefixes.stick, 1.0 / 2.0);
        magnetoConversionMultipliers.put(OrePrefixes.round, 1.0 / 8.0);
        magnetoConversionMultipliers.put(OrePrefixes.bolt, 1.0 / 8.0);
        magnetoConversionMultipliers.put(OrePrefixes.screw, 1.0 / 8.0);
        magnetoConversionMultipliers.put(OrePrefixes.ring, 1.0 / 4.0);
        magnetoConversionMultipliers.put(OrePrefixes.foil, 1.0 / 8.0);
        magnetoConversionMultipliers.put(OrePrefixes.gearGtSmall, 1.0);
        magnetoConversionMultipliers.put(OrePrefixes.rotor, 2.0);
        magnetoConversionMultipliers.put(OrePrefixes.stickLong, 1.0);
        magnetoConversionMultipliers.put(OrePrefixes.gearGt, 2.0);
        magnetoConversionMultipliers.put(OrePrefixes.wireFine, 1.0 / 8.0);
        findAllRecipes();
        generateAssemblerRecipes();
        generateAsslineRecipes();
    }

    /** Normal assembler recipes (LV-IV) */
    private static void generateAssemblerRecipes() {
        allAssemblerRecipes.forEach((recipeList, info) -> {
            for (GTRecipe recipe : recipeList) {
                if (recipe != null) {
                    ArrayList<ItemStack> fixedInputs = new ArrayList<>();
                    ArrayList<FluidStack> fixedFluids = new ArrayList<>();

                    for (int j = 0; j < recipe.mInputs.length; j++) {
                        ItemStack input = recipe.mInputs[j];
                        if (GTUtility.isStackValid(input) && !(input.getItem() instanceof ItemIntegratedCircuit))
                            fixedInputs.addAll(multiplyAndSplitIntoStacks(input, INPUT_MULTIPLIER));
                    }
                    for (int j = 0; j < recipe.mFluidInputs.length; j++) {
                        FluidStack currFluid = recipe.mFluidInputs[j].copy();
                        currFluid.amount = currFluid.amount * INPUT_MULTIPLIER;
                        fixedFluids.add(currFluid);
                    }

                    fixedInputs = compactItems(fixedInputs, info.getRight());
                    replaceIntoFluids(fixedInputs, fixedFluids, 64);
                    int tier = info.getRight();
                    int energy = (int) Math.min(Integer.MAX_VALUE - 7, GTValues.VP[tier - 1]);
                    GTValues.RA.stdBuilder()
                        .itemInputs(fixedInputs.toArray(new ItemStack[0]))
                        .itemOutputs(
                            info.getLeft()
                                .get(OUTPUT_MULTIPLIER))
                        .fluidInputs(fixedFluids.toArray(new FluidStack[0]))
                        .duration(recipe.mDuration * INPUT_MULTIPLIER)
                        .eut(energy)
                        .metadata(COAL_CASING_TIER, info.getRight())
                        .noOptimize()
                        .addTo(GoodGeneratorRecipeMaps.componentAssemblyLineRecipes);
                }
            }
        });
    }

    /** Assembly Line Recipes (LuV+) **/
    private static void generateAsslineRecipes() {
        allAsslineRecipes.forEach((recipeList, info) -> {
            for (GTRecipe.RecipeAssemblyLine recipe : recipeList) {
                if (recipe != null) {
                    int componentCircuit = -1;
                    for (int i = 0; i < compPrefixes.length; i++) if (info.getLeft()
                        .toString()
                        .startsWith(compPrefixes[i])) componentCircuit = i + 1;
                    if (componentCircuit == -1) {
                        throw new NullPointerException(
                            "Wrong circuit. Comp: " + info.getLeft()
                                .toString());
                    }
                    final boolean addProgrammedCircuit = componentCircuit <= 7;
                    // Arrays of the item and fluid inputs, that are updated to be multiplied and/or condensed in the
                    // following code
                    ArrayList<ItemStack> fixedInputs = new ArrayList<>();
                    ArrayList<FluidStack> fixedFluids = new ArrayList<>();

                    // Multiplies the original fluid inputs
                    for (int j = 0; j < recipe.mFluidInputs.length; j++) {
                        FluidStack currFluid = recipe.mFluidInputs[j].copy();
                        currFluid.amount *= (double) INPUT_MULTIPLIER;
                        fixedFluids.add(currFluid);
                    }

                    // First pass.
                    for (ItemStack input : recipe.mInputs) {
                        if (GTUtility.isStackValid(input)) {
                            int count = input.stackSize;
                            // Mulitplies the input by its multiplier, and adjusts the stacks accordingly
                            if (!(input.getItem() instanceof ItemIntegratedCircuit)) {

                                ItemData data = GTOreDictUnificator.getAssociation(input);
                                // trying to fix some circuit oredicting issues

                                if (data != null && data.mPrefix == OrePrefixes.circuit) fixedInputs.addAll(
                                    multiplyAndSplitIntoStacks(
                                        GTOreDictUnificator.get(data.mPrefix, data.mMaterial.mMaterial, count),
                                        INPUT_MULTIPLIER));
                                else fixedInputs.addAll(multiplyAndSplitIntoStacks(input, INPUT_MULTIPLIER));
                            }
                        }
                    }
                    fixedInputs = compactItems(fixedInputs, info.getRight());
                    replaceIntoFluids(fixedInputs, fixedFluids, FLUID_CONVERSION_STACKSIZE_THRESHOLD);
                    // If it overflows then it tries REALLY HARD to cram as much stuff into there.
                    if (fixedInputs.size() > (addProgrammedCircuit ? 8 : 9))
                        replaceIntoFluids(fixedInputs, fixedFluids, FLUID_CONVERSION_STACKSIZE_THRESHOLD / 2);
                    if (addProgrammedCircuit) fixedInputs.add(GTUtility.getIntegratedCircuit(componentCircuit));

                    addEternityForMHDCSM(fixedFluids);
                    GTValues.RA.stdBuilder()
                        .itemInputs(fixedInputs.toArray(new ItemStack[0]))
                        .itemOutputs(
                            info.getLeft()
                                .get(OUTPUT_MULTIPLIER))
                        .fluidInputs(fixedFluids.toArray(new FluidStack[0]))
                        .duration(recipe.mDuration * INPUT_MULTIPLIER)
                        .eut(recipe.mEUt)
                        .metadata(COAL_CASING_TIER, info.getRight())
                        .noOptimize()
                        .addTo(GoodGeneratorRecipeMaps.componentAssemblyLineRecipes);

                    // Add a second recipe using Styrene-Butadiene
                    // Rubber instead of Silicone Rubber.
                    // This relies on silicone rubber being first in
                    // @allSyntheticRubber so it's quite fragile, but
                    // it's also the least invasive change.
                    if (swapSiliconeForStyreneButadiene(fixedFluids)) {
                        GTValues.RA.stdBuilder()
                            .itemInputs(fixedInputs.toArray(new ItemStack[0]))
                            .itemOutputs(
                                info.getLeft()
                                    .get(OUTPUT_MULTIPLIER))
                            .fluidInputs(fixedFluids.toArray(new FluidStack[0]))
                            .duration(recipe.mDuration * INPUT_MULTIPLIER)
                            .eut(recipe.mEUt)
                            .metadata(COAL_CASING_TIER, info.getRight())
                            .noOptimize()
                            .addTo(GoodGeneratorRecipeMaps.componentAssemblyLineRecipes);
                    }
                }
            }
        });
    }

    /**
     * Looks for a matching FluidStack and merges the amount of the converted fluid with the one it found. Otherwise, it
     * will add the converted to the fluid inputs.
     */
    private static void replaceIntoFluids(List<ItemStack> inputs, List<FluidStack> fluidOutputs, int threshold) {
        HashMap<ItemStack, Integer> totals = getTotalItems(inputs.toArray(new ItemStack[0]));
        ArrayList<ItemStack> newInputs = new ArrayList<>();
        for (ItemStack input : totals.keySet()) {
            int count = totals.get(input);
            boolean isConverted = false;
            if (OreDictionary.getOreIDs(input).length > 0 && count > threshold) {
                FluidStack foundFluidStack = tryConvertItemStackToFluidMaterial(input);

                ItemData data = GTOreDictUnificator.getAssociation(input);

                // Prevents the uncraftable molten magnetic samarium from being converted into fluid during auto
                // generation
                if (data != null && data.mMaterial.mMaterial == Materials.SamariumMagnetic) {
                    input = GTOreDictUnificator.get(data.mPrefix, Materials.Samarium, 1);
                    foundFluidStack = tryConvertItemStackToFluidMaterial(input);
                } else if (data != null && data.mMaterial.mMaterial == Materials.TengamAttuned) {
                    input = GTOreDictUnificator.get(data.mPrefix, Materials.TengamPurified, 1);
                    foundFluidStack = tryConvertItemStackToFluidMaterial(input);
                }

                if (foundFluidStack != null) {
                    foundFluidStack.amount *= count;
                    boolean alreadyHasFluid = false;
                    for (FluidStack fluidstack : fluidOutputs) {
                        if (foundFluidStack.getFluid()
                            .equals(fluidstack.getFluid())) {
                            fluidstack.amount += foundFluidStack.amount;
                            alreadyHasFluid = true;
                            break;
                        }
                    }
                    if (!alreadyHasFluid) {
                        fluidOutputs.add(foundFluidStack);
                    }
                    isConverted = true;
                }
            }
            if (!isConverted) {
                newInputs.add(GTUtility.copyAmountUnsafe(count, input));
            }
        }
        inputs.clear();
        inputs.addAll(newInputs);
    }

    /**
     * Tries to convert {@code input} into its molten form. Because the internal names for material fluids in GT5u,
     * GT++, and BartWorks follow the same naming scheme, this method should work for any {@code ItemStack} from any of
     * the 3 material systems.
     */
    @Nullable
    private static FluidStack tryConvertItemStackToFluidMaterial(ItemStack input) {
        ArrayList<String> oreDicts = new ArrayList<>();
        for (int id : OreDictionary.getOreIDs(input)) {
            oreDicts.add(OreDictionary.getOreName(id));
        }
        oreDictLoop: for (String dict : oreDicts) {
            for (String blacklistedPrefix : blacklistedDictPrefixes) {
                if (dict.startsWith(blacklistedPrefix)) {
                    return null;
                }
            }
            for (String blacklistedPrefix : softBlacklistedDictPrefixes) {
                if (dict.startsWith(blacklistedPrefix)) {
                    continue oreDictLoop;
                }
            }
            OrePrefixes orePrefix;
            try {
                orePrefix = OrePrefixes.valueOf(findBestPrefix(dict));
            } catch (Exception e) {
                continue;
            }

            String strippedOreDict = dict.substring(
                orePrefix.toString()
                    .length());

            // Prevents things like AnyCopper or AnyIron from messing the search up.
            if (strippedOreDict.contains("Any")) continue;
            if (strippedOreDict.contains("PTMEG")) return FluidRegistry.getFluidStack(
                "molten.silicone",
                (int) (orePrefix.mMaterialAmount / (GTValues.M / 144)) * input.stackSize);
            return FluidRegistry.getFluidStack(
                "molten." + strippedOreDict.toLowerCase(),
                (int) (orePrefix.mMaterialAmount / (GTValues.M / 144)) * input.stackSize);
        }
        return null;
    }

    /**
     * Gives the longest Ore Prefix that the OreDictionary string starts with. This makes it the most accurate prefix.
     * For example: If your OreDictionary is something like {@code gearGtSmallSpaceTime}, a conventional search would
     * return something like {@code gearGt} instead of {@code gearGtSmall}. This makes the longer String the most
     * accurate.
     *
     * @param oreDict The Ore Dictionary entry
     * @return The longest ore prefix that the OreDict string starts with. This makes it the most accurate prefix.
     */
    private static String findBestPrefix(String oreDict) {
        int longestPrefixLength = 0;
        String matchingPrefix = null;
        for (OrePrefixes prefix : OrePrefixes.values()) {
            String name = prefix.toString();
            if (oreDict.startsWith(name)) {
                if (name.length() > longestPrefixLength) {
                    longestPrefixLength = name.length();
                    matchingPrefix = name;
                }
            }
        }
        return matchingPrefix;
    }

    /**
     * Transforms each {@code ItemStack}, if possible, into a more compact form. For example, a stack of 16 1x cables,
     * when passed into the {@code items} array, will be converted into a single 16x cable. Also handles GraviStar and
     * neutronium nanite conversion.
     */
    private static ArrayList<ItemStack> compactItems(List<ItemStack> items, int tier) {
        ArrayList<ItemStack> stacks = new ArrayList<>();
        HashMap<ItemStack, Integer> totals = getTotalItems(items);
        for (ItemStack itemstack : totals.keySet()) {
            int totalItems = totals.get(itemstack);
            ItemData data = GTOreDictUnificator.getAssociation(itemstack);
            boolean isCompacted = false;

            for (String dict : Arrays.stream(OreDictionary.getOreIDs(itemstack))
                .mapToObj(OreDictionary::getOreName)
                .collect(Collectors.toList())) {
                if (dict.startsWith("circuit")) {
                    stacks.addAll(getWrappedCircuits(itemstack, totalItems, dict));
                    isCompacted = true;
                    break;
                }
                if (dict.contains("Magneto")) {
                    stacks.addAll(getMagnetoConversion(itemstack, totalItems));
                    isCompacted = true;
                    break;
                }
            }

            if (data != null && !isCompacted) {
                OrePrefixes goInto = conversion.get(data.mPrefix);
                if (goInto != null && GTOreDictUnificator.get(goInto, data.mMaterial.mMaterial, 1) != null) {
                    compactorHelper(data, goInto, stacks, totalItems);
                    isCompacted = true;
                }
            }
            if (GTUtility.areStacksEqual(itemstack, ItemList.Gravistar.get(1)) && tier >= 9) {
                stacks.addAll(multiplyAndSplitIntoStacks(ItemList.NuclearStar.get(1), totalItems / 16));
                isCompacted = true;
            }
            if (GTUtility
                .areStacksEqual(itemstack, GTOreDictUnificator.get(OrePrefixes.nanite, Materials.Neutronium, 1))) {
                stacks.addAll(
                    multiplyAndSplitIntoStacks(
                        GTOreDictUnificator.get(OrePrefixes.nanite, Materials.Gold, 1),
                        totalItems / 16));
                isCompacted = true;
            }
            if (!isCompacted) stacks.addAll(multiplyAndSplitIntoStacks(itemstack, totalItems));
        }
        stacks = mergeStacks(stacks);
        return stacks;
    }

    /** A helper method for compacting items */
    private static void compactorHelper(ItemData data, OrePrefixes compactInto, ArrayList<ItemStack> output,
        int total) {
        int materialRatio = (int) ((double) compactInto.mMaterialAmount / data.mPrefix.mMaterialAmount);
        output.addAll(
            multiplyAndSplitIntoStacks(
                GTOreDictUnificator.get(compactInto, data.mMaterial.mMaterial, 1),
                total / materialRatio));
    }

    /**
     * Searches the Assembler and Assembly line registry for all the base component recipes.
     */
    private static void findAllRecipes() {
        allAssemblerRecipes = new LinkedHashMap<>();
        allAsslineRecipes = new LinkedHashMap<>();
        for (String compPrefix : compPrefixes) {
            for (int t = 1; t <= 13; t++) {
                String vName = GTValues.VN[t];
                ItemList currentComponent = ItemList.valueOf(compPrefix + vName);
                if (currentComponent.hasBeenSet()) {
                    if (t < 6) {
                        ArrayList<GTRecipe> foundRecipes = new ArrayList<>();
                        for (GTRecipe recipe : RecipeMaps.assemblerRecipes.getAllRecipes()) {
                            if (GTUtility.areStacksEqual(currentComponent.get(1), recipe.mOutputs[0])) {
                                foundRecipes.add(recipe);
                            }
                        }
                        allAssemblerRecipes.put(foundRecipes, Pair.of(currentComponent, t));
                    } else {
                        ArrayList<GTRecipe.RecipeAssemblyLine> foundRecipes = new ArrayList<>();
                        for (GTRecipe.RecipeAssemblyLine recipe : GTRecipe.RecipeAssemblyLine.sAssemblylineRecipes) {
                            if (GTUtility.areStacksEqual(currentComponent.get(1), recipe.mOutput)) {
                                foundRecipes.add(recipe);
                            }
                        }
                        allAsslineRecipes.put(foundRecipes, Pair.of(currentComponent, t));
                    }
                }
            }
        }
    }

    private static List<ItemStack> getWrappedCircuits(ItemStack item, int total, String oreDict) {
        ArrayList<ItemStack> stacks = new ArrayList<>();
        String circuitMaterial = oreDict.substring(7);
        int tier = ComponentAssemblyLineMiscRecipes.NameToTier.get(circuitMaterial);
        if (total >= 16)
            stacks.addAll(multiplyAndSplitIntoStacks(new ItemStack(Loaders.circuitWrap, 1, tier), total / 16));
        else stacks.addAll(multiplyAndSplitIntoStacks(item, total));

        return stacks;
    }

    private static List<ItemStack> getMagnetoConversion(ItemStack item, int total) {
        ArrayList<ItemStack> stacks = new ArrayList<>();
        ItemData data = GTOreDictUnificator.getAssociation(item);
        if (data == null) {
            return stacks;
        }
        if (total >= FLUID_CONVERSION_STACKSIZE_THRESHOLD) {
            double multiplier = magnetoConversionMultipliers.get(data.mPrefix);
            stacks.addAll(
                getWrappedCircuits(
                    GTOreDictUnificator.get(OrePrefixes.circuit, Materials.UHV, 1),
                    (int) (total * multiplier),
                    "circuitInfinite"));
        }
        stacks.addAll(multiplyAndSplitIntoStacks(item, total));
        return stacks;
    }

    private static void addEternityForMHDCSM(ArrayList<FluidStack> fluidInputs) {
        boolean eternity = false;
        boolean mhdcsm = false;
        int mhdcsmAmount = 0;

        for (FluidStack fluidstack : fluidInputs) {
            if (fluidstack.getFluid()
                .equals(FluidRegistry.getFluid(moltenMHDCSM))) {
                mhdcsm = true;
                mhdcsmAmount = fluidstack.amount;
            }
            if (fluidstack.getFluid()
                .equals(FluidRegistry.getFluid("molten.eternity"))) {
                eternity = true;
            }
        }

        if (mhdcsm && !eternity) {
            // 576 * 48 is substracted because uxv parts have 576L mhdcsm fluid (not solid, so no EIC conversion needed)
            // in their assline recipes and each CoAl recipe has 48x recipe inputs
            fluidInputs.add(MaterialsUEVplus.Eternity.getMolten(mhdcsmAmount - 576 * 48));
        }
    }

    private static boolean swapSiliconeForStyreneButadiene(ArrayList<FluidStack> fluidInputs) {
        for (int i = 0; i < fluidInputs.size(); i++) {
            FluidStack fluidstack = fluidInputs.get(i);
            if (fluidstack.getFluid()
                .equals(FluidRegistry.getFluid("molten.silicone"))) {
                fluidInputs.set(i, FluidRegistry.getFluidStack("molten.styrenebutadienerubber", fluidstack.amount));
                return true;
            }
        }
        return false;
    }
}