Mekanism-tilera-Edition/src/main/java/mekanism/common/OreDictCache.java

package mekanism.common;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import mekanism.api.ItemInfo;
import mekanism.common.util.MekanismUtils;
import mekanism.common.util.StackUtils;
import net.minecraft.item.ItemBlock;
import net.minecraft.item.ItemStack;
import net.minecraftforge.oredict.OreDictionary;

public final class OreDictCache
{
	public static HashMap<ItemInfo, List<String>> cachedKeys = new HashMap<ItemInfo, List<String>>();
	public static HashMap<String, List<ItemStack>> oreDictStacks = new HashMap<String, List<ItemStack>>();
	public static HashMap<String, List<ItemStack>> modIDStacks = new HashMap<String, List<ItemStack>>();

	public static List<String> getOreDictName(ItemStack check)
	{
		ItemInfo info = ItemInfo.get(check);

		if(cachedKeys.get(info) != null)
		{
			return cachedKeys.get(info);
		}

		List<Integer> idsFound = new ArrayList<Integer>();
		HashMap<Integer, ArrayList<ItemStack>> oreStacks = (HashMap<Integer, ArrayList<ItemStack>>)MekanismUtils.getPrivateValue(null, OreDictionary.class, new String[] {"oreStacks"});
		oreStacks = (HashMap<Integer, ArrayList<ItemStack>>)oreStacks.clone();

		for(Map.Entry<Integer, ArrayList<ItemStack>> entry : oreStacks.entrySet())
		{
			for(ItemStack stack : entry.getValue())
			{
				if(StackUtils.equalsWildcard(stack, check))
				{
					idsFound.add(entry.getKey());
					break;
				}
			}
		}

		List<String> ret = new ArrayList<String>();

		for(Integer id : idsFound)
		{
			ret.add(OreDictionary.getOreName(id));
		}

		cachedKeys.put(info, ret);

		return ret;
	}
	
	public static List<ItemStack> getOreDictStacks(String oreName, boolean forceBlock)
	{
		if(oreDictStacks.get(oreName) != null)
		{
			return oreDictStacks.get(oreName);
		}

		List<String> keys = new ArrayList<String>();

		for(String s : OreDictionary.getOreNames())
		{
			if(oreName.equals(s) || oreName.equals("*"))
			{
				keys.add(s);
			}
			else if(oreName.endsWith("*") && !oreName.startsWith("*"))
			{
				if(s.startsWith(oreName.substring(0, oreName.length()-1)))
				{
					keys.add(s);
				}
			}
			else if(oreName.startsWith("*") && !oreName.endsWith("*"))
			{
				if(s.endsWith(oreName.substring(1)))
				{
					keys.add(s);
				}
			}
			else if(oreName.startsWith("*") && oreName.endsWith("*"))
			{
				if(s.contains(oreName.substring(1, oreName.length()-1)))
				{
					keys.add(s);
				}
			}
		}
		
		List<ItemStack> stacks = new ArrayList<ItemStack>();

		for(String key : keys)
		{
			for(ItemStack stack : OreDictionary.getOres(key))
			{
				ItemStack toAdd = stack.copy();

				if(!stacks.contains(stack) && (!forceBlock || toAdd.getItem() instanceof ItemBlock))
				{
					stacks.add(stack.copy());
				}
			}
		}
		
		oreDictStacks.put(oreName, stacks);
		
		return stacks;
	}
	
	public static List<ItemStack> getModIDStacks(String modName, boolean forceBlock)
	{
		if(modIDStacks.get(modName) != null)
		{
			return modIDStacks.get(modName);
		}
		
		List<ItemStack> stacks = new ArrayList<ItemStack>();

		for(String key : OreDictionary.getOreNames())
		{
			for(ItemStack stack : OreDictionary.getOres(key))
			{
				ItemStack toAdd = stack.copy();
				String s = MekanismUtils.getMod(toAdd);

				if(!stacks.contains(stack) && toAdd.getItem() instanceof ItemBlock)
				{
					if(modName.equals(s) || modName.equals("*"))
					{
						stacks.add(stack.copy());
					}
					else if(modName.endsWith("*") && !modName.startsWith("*"))
					{
						if(s.startsWith(modName.substring(0, modName.length()-1)))
						{
							stacks.add(stack.copy());
						}
					}
					else if(modName.startsWith("*") && !modName.endsWith("*"))
					{
						if(s.endsWith(modName.substring(1)))
						{
							stacks.add(stack.copy());
						}
					}
					else if(modName.startsWith("*") && modName.endsWith("*"))
					{
						if(s.contains(modName.substring(1, modName.length()-1)))
						{
							stacks.add(stack.copy());
						}
					}
				}
			}
		}
		
		modIDStacks.put(modName, stacks);
		
		return stacks;
	}
}