101 lines
3.2 KiB
Java
101 lines
3.2 KiB
Java
package com.pahimar.ee3.exchange;
|
|
|
|
import com.google.common.collect.ImmutableMap;
|
|
import com.google.common.collect.ImmutableMultimap;
|
|
import com.google.common.collect.Multimap;
|
|
import com.google.common.collect.TreeMultimap;
|
|
import com.pahimar.ee3.reference.Comparators;
|
|
import com.pahimar.ee3.util.LogHelper;
|
|
import net.minecraft.item.ItemStack;
|
|
import net.minecraftforge.oredict.OreDictionary;
|
|
|
|
import java.util.*;
|
|
|
|
public class CachedOreDictionary
|
|
{
|
|
private static CachedOreDictionary cachedOreDictionary = null;
|
|
private ImmutableMap<Integer, String> idToNameMap;
|
|
private ImmutableMap<String, List<ItemStack>> oreNameToItemStackMap;
|
|
private ImmutableMultimap<WrappedStack, String> itemStackToOreNameMap;
|
|
|
|
private CachedOreDictionary()
|
|
{
|
|
Map<Integer, String> idToOreNameMap = new TreeMap<Integer, String>();
|
|
Map<String, List<ItemStack>> nameToStackMap = new TreeMap<String, List<ItemStack>>(Comparators.stringComparator);
|
|
Multimap<WrappedStack, String> stackToNameMultiMap = TreeMultimap.create(WrappedStack.comparator, Comparators.stringComparator);
|
|
|
|
for (String oreName : OreDictionary.getOreNames())
|
|
{
|
|
idToOreNameMap.put(OreDictionary.getOreID(oreName), oreName);
|
|
|
|
List<ItemStack> oreNameItemStacks = OreDictionary.getOres(oreName);
|
|
nameToStackMap.put(oreName, oreNameItemStacks);
|
|
|
|
for (ItemStack itemStack : oreNameItemStacks)
|
|
{
|
|
stackToNameMultiMap.put(WrappedStack.wrap(itemStack), oreName);
|
|
}
|
|
}
|
|
|
|
idToNameMap = ImmutableMap.copyOf(idToOreNameMap);
|
|
oreNameToItemStackMap = ImmutableMap.copyOf(nameToStackMap);
|
|
itemStackToOreNameMap = ImmutableMultimap.copyOf(stackToNameMultiMap);
|
|
}
|
|
|
|
public static CachedOreDictionary getInstance()
|
|
{
|
|
if (cachedOreDictionary == null)
|
|
{
|
|
cachedOreDictionary = new CachedOreDictionary();
|
|
}
|
|
|
|
return cachedOreDictionary;
|
|
}
|
|
|
|
public Set<String> getOreNames()
|
|
{
|
|
return oreNameToItemStackMap.keySet();
|
|
}
|
|
|
|
public List<ItemStack> getItemStacksForOreName(String oreName)
|
|
{
|
|
if (oreNameToItemStackMap.containsKey(oreName))
|
|
{
|
|
return oreNameToItemStackMap.get(oreName);
|
|
}
|
|
|
|
return new ArrayList<ItemStack>();
|
|
}
|
|
|
|
public List<String> getOreNamesForItemStack(ItemStack itemStack)
|
|
{
|
|
List<String> oreNameList = new ArrayList<String>();
|
|
WrappedStack wrappedStack = WrappedStack.wrap(itemStack);
|
|
|
|
if (itemStackToOreNameMap.containsKey(wrappedStack))
|
|
{
|
|
for (String oreName : itemStackToOreNameMap.get(wrappedStack))
|
|
{
|
|
oreNameList.add(oreName);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (WrappedStack wrappedStack1 : itemStackToOreNameMap.keySet())
|
|
{
|
|
|
|
}
|
|
}
|
|
|
|
return oreNameList;
|
|
}
|
|
|
|
public void dumpCachedOreDictionaryToLog()
|
|
{
|
|
for (String oreName : CachedOreDictionary.getInstance().getOreNames())
|
|
{
|
|
LogHelper.info("OreName: {}, ItemStacks: {}", oreName, CachedOreDictionary.getInstance().getItemStacksForOreName(oreName));
|
|
}
|
|
}
|
|
}
|