RecipeRegistry.java

package com.pahimar.ee.recipe;

import com.google.common.collect.ImmutableMultimap;
import com.google.common.collect.Multimap;
import com.google.common.collect.TreeMultimap;
import com.pahimar.ee.exchange.WrappedStack;
import com.pahimar.ee.reference.Comparators;
import com.pahimar.ee.util.LogHelper;
import net.minecraftforge.fml.common.Loader;
import org.apache.logging.log4j.Marker;
import org.apache.logging.log4j.MarkerManager;

import java.util.Collection;
import java.util.Set;
import java.util.TreeSet;

public class RecipeRegistry {

    public static final Marker RECIPE_MARKER = MarkerManager.getMarker("EE_RECIPE", LogHelper.MOD_MARKER);
    public static final RecipeRegistry INSTANCE = new RecipeRegistry();

    private Multimap<WrappedStack, Set<WrappedStack>> recipeMap;
    private ImmutableMultimap<WrappedStack, Set<WrappedStack>> immutableRecipeMap;

    private RecipeRegistry() {
        recipeMap = TreeMultimap.create(WrappedStack.COMPARATOR, Comparators.WRAPPED_STACK_SET_COMPARATOR);
    }

    /**
     * TODO Finish JavaDoc
     *
     * @param recipeOutput
     * @param recipeInputList
     */
    public void addRecipe(Object recipeOutput, Collection<?> recipeInputList) {

        // Wrap the recipe output
        WrappedStack wrappedRecipeOutput = WrappedStack.build(recipeOutput);
        if (wrappedRecipeOutput == null) {
            return;
        }

        Set<WrappedStack> wrappedRecipeInputList = new TreeSet<>();
        StringBuilder stringBuilder = new StringBuilder();

        for (Object recipeInputObject : recipeInputList) {

            WrappedStack wrappedInputObject = WrappedStack.build(recipeInputObject);

            if (wrappedInputObject != null) {
                wrappedRecipeInputList.add(wrappedInputObject);
                stringBuilder.append(wrappedInputObject);
                stringBuilder.append(" ");
            }
            else {
                return;
            }
        }

        // Check to see if we already have this recipe in the map
        boolean existsAlready = false;
        for (Set<WrappedStack> recipeInputs : recipeMap.get(wrappedRecipeOutput)) {
            if (recipeInputs.containsAll(wrappedRecipeInputList) && wrappedRecipeInputList.containsAll(recipeInputs)) {
                existsAlready = true;
            }
        }

        // Add the recipe mapping only if we don't already have it
        if (!existsAlready) {
            LogHelper.trace(RECIPE_MARKER, "[{}] Mod with ID '{}' added recipe (Output: {}, Inputs: {})", Loader.instance().getLoaderState(), Loader.instance().activeModContainer().getModId(), wrappedRecipeOutput, stringBuilder.toString().trim());
            recipeMap.put(wrappedRecipeOutput, wrappedRecipeInputList);
        }
    }

    /**
     * TODO Finish JavaDoc
     */
    public void registerVanillaRecipes() {

        RecipesVanilla.registerRecipes();
        RecipesFluidContainers.registerRecipes();
        RecipesPotions.registerRecipes();
        RecipesArrows.registerRecipes();
    }

    /**
     * TODO Finish JavaDoc
     *
     * @return
     */
    public Multimap<WrappedStack, Set<WrappedStack>> getRecipeMappings() {

        if (immutableRecipeMap == null) {
            immutableRecipeMap = ImmutableMultimap.copyOf(INSTANCE.recipeMap);
        }

        return immutableRecipeMap;
    }
}