package com.pahimar.ee3.exchange; import com.google.common.collect.ImmutableSortedMap; import com.pahimar.ee3.api.EnergyValue; import com.pahimar.ee3.api.IEnergyValueProvider; import com.pahimar.ee3.recipe.RecipeRegistry; import com.pahimar.ee3.reference.Files; import com.pahimar.ee3.reference.Reference; import com.pahimar.ee3.reference.Settings; import com.pahimar.ee3.util.EnergyValueHelper; import com.pahimar.ee3.util.INBTTaggable; import com.pahimar.ee3.util.LogHelper; import com.pahimar.ee3.util.SerializationHelper; import cpw.mods.fml.common.FMLCommonHandler; import net.minecraft.item.Item; import net.minecraft.item.ItemStack; import net.minecraft.nbt.NBTTagCompound; import net.minecraft.nbt.NBTTagList; import net.minecraftforge.fluids.FluidStack; import net.minecraftforge.oredict.OreDictionary; import java.io.File; import java.util.*; public class EnergyValueRegistry implements INBTTaggable { private boolean shouldRegenNextRestart = false; private static EnergyValueRegistry energyValueRegistry = null; private static String modListMD5; private static Map preAssignedMappings; private static Map postAssignedMappings; private ImmutableSortedMap stackMappings; private ImmutableSortedMap> valueMappings; private EnergyValueRegistry() { modListMD5 = SerializationHelper.getModListMD5(); } public static EnergyValueRegistry getInstance() { if (energyValueRegistry == null) { energyValueRegistry = new EnergyValueRegistry(); } return energyValueRegistry; } public void addPreAssignedEnergyValue(Object object, float energyValue) { addPreAssignedEnergyValue(object, new EnergyValue(energyValue)); } public void addPreAssignedEnergyValue(Object object, EnergyValue energyValue) { if (preAssignedMappings == null) { preAssignedMappings = new HashMap(); } if (WrappedStack.canBeWrapped(object) && energyValue != null && Float.compare(energyValue.getEnergyValue(), 0f) > 0) { WrappedStack wrappedStack = new WrappedStack(object); if (wrappedStack.getStackSize() > 0) { WrappedStack factoredWrappedStack = new WrappedStack(wrappedStack, 1); EnergyValue factoredEnergyValue = EnergyValueHelper.factorEnergyValue(energyValue, wrappedStack.getStackSize()); if (preAssignedMappings.containsKey(factoredWrappedStack)) { if (factoredEnergyValue.compareTo(preAssignedMappings.get(factoredWrappedStack)) < 0) { preAssignedMappings.put(factoredWrappedStack, factoredEnergyValue); } } else { preAssignedMappings.put(factoredWrappedStack, factoredEnergyValue); } } } } public void addPostAssignedEnergyValue(Object object, float energyValue) { if (postAssignedMappings == null) { postAssignedMappings = new HashMap(); } if (WrappedStack.canBeWrapped(object) && Float.compare(energyValue, 0f) > 0) { WrappedStack wrappedStack = new WrappedStack(object); if (wrappedStack.getStackSize() > 0) { WrappedStack factoredWrappedStack = new WrappedStack(wrappedStack, 1); EnergyValue factoredEnergyValue = new EnergyValue(energyValue * 1f / wrappedStack.getStackSize(), EnergyValue.EnergyType.CORPOREAL); postAssignedMappings.put(factoredWrappedStack, factoredEnergyValue); } } } public void addPostAssignedEnergyValue(Object object, EnergyValue energyValue) { if (postAssignedMappings == null) { postAssignedMappings = new HashMap(); } if (WrappedStack.canBeWrapped(object) && energyValue != null && Float.compare(energyValue.getEnergyValue(), 0f) > 0) { WrappedStack wrappedStack = new WrappedStack(object); if (wrappedStack.getStackSize() > 0) { WrappedStack factoredWrappedStack = new WrappedStack(wrappedStack, 1); EnergyValue factoredEnergyValue = EnergyValueHelper.factorEnergyValue(energyValue, wrappedStack.getStackSize()); postAssignedMappings.put(factoredWrappedStack, factoredEnergyValue); } } } public boolean hasEnergyValue(Object object, boolean strict) { if (WrappedStack.canBeWrapped(object)) { WrappedStack stack = new WrappedStack(object); if (stack.getWrappedStack() instanceof ItemStack && ((ItemStack) stack.getWrappedStack()).getItem() instanceof IEnergyValueProvider) { EnergyValue energyValue = ((IEnergyValueProvider) ((ItemStack) stack.getWrappedStack()).getItem()).getEnergyValue((ItemStack) stack.getWrappedStack()); if (energyValue != null && energyValue.getEnergyValue() > 0f) { return true; } } else { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(stack.getWrappedStack()))) { return true; } else { if (!strict) { if (stack.getWrappedStack() instanceof ItemStack) { ItemStack wrappedItemStack = (ItemStack) stack.getWrappedStack(); // If its an OreDictionary item, scan its siblings for values if (OreDictionary.getOreIDs(wrappedItemStack).length > 0) { OreStack oreStack = new OreStack(wrappedItemStack); if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(oreStack))) { return true; } else { for (int oreId : OreDictionary.getOreIDs(wrappedItemStack)) { for (ItemStack itemStack : OreDictionary.getOres(OreDictionary.getOreName(oreId))) { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(itemStack))) { return true; } } } } } // Else, scan for if there is a wildcard value for it else { for (WrappedStack valuedStack : energyValueRegistry.stackMappings.keySet()) { if (valuedStack.getWrappedStack() instanceof ItemStack) { ItemStack valuedItemStack = (ItemStack) valuedStack.getWrappedStack(); if (Item.getIdFromItem(valuedItemStack.getItem()) == Item.getIdFromItem(wrappedItemStack.getItem())) { if (valuedItemStack.getItemDamage() == OreDictionary.WILDCARD_VALUE || wrappedItemStack.getItemDamage() == OreDictionary.WILDCARD_VALUE) { return true; } else if (wrappedItemStack.getItem().isDamageable() && wrappedItemStack.isItemDamaged()) { return true; } } } } } } else if (stack.getWrappedStack() instanceof OreStack) { OreStack oreStack = (OreStack) stack.getWrappedStack(); for (ItemStack oreItemStack : OreDictionary.getOres(oreStack.oreName)) { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(oreItemStack))) { return true; } } } } } } } return false; } public boolean hasEnergyValue(Object object) { return hasEnergyValue(object, false); } public EnergyValue getEnergyValue(Object object) { return getEnergyValue(object, false); } public EnergyValue getEnergyValue(Object object, boolean strict) { if (WrappedStack.canBeWrapped(object)) { WrappedStack stack = new WrappedStack(object); if (stack.getWrappedStack() instanceof ItemStack && ((ItemStack) stack.getWrappedStack()).getItem() instanceof IEnergyValueProvider) { EnergyValue energyValue = ((IEnergyValueProvider) ((ItemStack) stack.getWrappedStack()).getItem()).getEnergyValue((ItemStack) stack.getWrappedStack()); if (energyValue != null && energyValue.getEnergyValue() > 0f) { return energyValue; } } else { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(stack.getWrappedStack()))) { return energyValueRegistry.stackMappings.get(new WrappedStack(stack.getWrappedStack())); } else { if (!strict) { if (stack.getWrappedStack() instanceof ItemStack) { EnergyValue lowestValue = null; ItemStack wrappedItemStack = (ItemStack) stack.getWrappedStack(); if (OreDictionary.getOreIDs(wrappedItemStack).length > 0) { OreStack oreStack = new OreStack(wrappedItemStack); if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(oreStack))) { return energyValueRegistry.stackMappings.get(new WrappedStack(oreStack)); } else { for (int oreId : OreDictionary.getOreIDs(wrappedItemStack)) { for (ItemStack itemStack : OreDictionary.getOres(OreDictionary.getOreName(oreId))) { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(itemStack))) { if (lowestValue == null) { lowestValue = energyValueRegistry.stackMappings.get(new WrappedStack(itemStack)); } else { EnergyValue itemValue = energyValueRegistry.stackMappings.get(new WrappedStack(itemStack)); if (itemValue.compareTo(lowestValue) < 0) { lowestValue = itemValue; } } } } } return lowestValue; } } else { for (WrappedStack valuedStack : energyValueRegistry.stackMappings.keySet()) { if (valuedStack.getWrappedStack() instanceof ItemStack) { ItemStack valuedItemStack = (ItemStack) valuedStack.getWrappedStack(); if (Item.getIdFromItem(valuedItemStack.getItem()) == Item.getIdFromItem(wrappedItemStack.getItem())) { if (valuedItemStack.getItemDamage() == OreDictionary.WILDCARD_VALUE || wrappedItemStack.getItemDamage() == OreDictionary.WILDCARD_VALUE) { EnergyValue stackValue = energyValueRegistry.stackMappings.get(valuedStack); if (stackValue.compareTo(lowestValue) < 0) { lowestValue = stackValue; } } else if (wrappedItemStack.getItem().isDamageable() && wrappedItemStack.isItemDamaged()) { EnergyValue stackValue = new EnergyValue(energyValueRegistry.stackMappings.get(valuedStack).getEnergyValue() * (1 - (wrappedItemStack.getItemDamage() * 1.0F / wrappedItemStack.getMaxDamage()))); if (stackValue.compareTo(lowestValue) < 0) { lowestValue = stackValue; } } } } } return lowestValue; } } else if (stack.getWrappedStack() instanceof OreStack) { OreStack oreStack = (OreStack) stack.getWrappedStack(); for (ItemStack oreItemStack : OreDictionary.getOres(oreStack.oreName)) { if (energyValueRegistry.stackMappings.containsKey(new WrappedStack(oreItemStack))) { return energyValueRegistry.stackMappings.get(new WrappedStack(oreItemStack)); } } } } } } } return null; } protected final void init() { if (!loadEnergyValueRegistryFromFile()) { runDynamicEnergyValueResolution(); } this.shouldRegenNextRestart = false; } private void runDynamicEnergyValueResolution() { HashMap stackValueMap = new HashMap(); /* * Pre-assigned values */ stackValueMap.putAll(preAssignedMappings); // Grab custom pre-assigned values from file Map preAssignedValueMap = SerializationHelper.readEnergyValueStackMapFromJsonFile(Files.PRE_ASSIGNED_ENERGY_VALUES); stackValueMap.putAll(preAssignedValueMap); /* * Auto-assignment */ // Initialize the maps for the first pass to happen ImmutableSortedMap.Builder stackMappingsBuilder = ImmutableSortedMap.naturalOrder(); stackMappingsBuilder.putAll(stackValueMap); stackMappings = stackMappingsBuilder.build(); Map computedStackValues = computeStackMappings(); // Initialize the pass counter int passNumber = 0; long computationStartTime = System.currentTimeMillis(); long passStartTime; int computedValueCount; int totalComputedValueCount = 0; LogHelper.info("Beginning dynamic value computation"); while ((computedStackValues.size() > 0) && (passNumber < 16)) { computedValueCount = 0; passStartTime = System.currentTimeMillis(); // Increment the pass counter passNumber++; // Set the values for getEnergyValue calls in the auto-assignment computation stackMappingsBuilder = ImmutableSortedMap.naturalOrder(); stackMappingsBuilder.putAll(stackValueMap); stackMappings = stackMappingsBuilder.build(); // Compute stack mappings from existing stack mappings computedStackValues = computeStackMappings(); for (WrappedStack keyStack : computedStackValues.keySet()) { EnergyValue factoredExchangeEnergyValue = null; WrappedStack factoredKeyStack = null; if (keyStack != null && keyStack.getWrappedStack() != null && keyStack.getStackSize() > 0) { if (computedStackValues.get(keyStack) != null && Float.compare(computedStackValues.get(keyStack).getEnergyValue(), 0f) > 0) { factoredExchangeEnergyValue = EnergyValueHelper.factorEnergyValue(computedStackValues.get(keyStack), keyStack.getStackSize()); factoredKeyStack = new WrappedStack(keyStack, 1); } } if (factoredExchangeEnergyValue != null) { if (stackValueMap.containsKey(factoredKeyStack)) { if (factoredExchangeEnergyValue.compareTo(stackValueMap.get(factoredKeyStack)) == -1) { stackValueMap.put(factoredKeyStack, factoredExchangeEnergyValue); } } else { stackValueMap.put(factoredKeyStack, factoredExchangeEnergyValue); computedValueCount++; totalComputedValueCount++; } } } LogHelper.info(String.format("Pass %s: Computed %s values for obejcts in %s ms", passNumber, computedValueCount, System.currentTimeMillis() - passStartTime)); } LogHelper.info(String.format("Finished dynamic value computation (computed %s values for objects in %s ms)", totalComputedValueCount, System.currentTimeMillis() - computationStartTime)); /* * Post-assigned values */ if (postAssignedMappings != null) { for (WrappedStack wrappedStack : postAssignedMappings.keySet()) { stackValueMap.put(wrappedStack, postAssignedMappings.get(wrappedStack)); } } else { postAssignedMappings = new HashMap(); } // Grab custom post-assigned values from file Map postAssignedValueMap = SerializationHelper.readEnergyValueStackMapFromJsonFile(Files.POST_ASSIGNED_ENERGY_VALUES); stackValueMap.putAll(postAssignedValueMap); /** * Finalize the stack to value map */ stackMappingsBuilder = ImmutableSortedMap.naturalOrder(); stackMappingsBuilder.putAll(stackValueMap); stackMappings = stackMappingsBuilder.build(); /** * Value map resolution */ SortedMap> tempValueMappings = new TreeMap>(); for (WrappedStack stack : stackMappings.keySet()) { if (stack != null) { EnergyValue value = stackMappings.get(stack); if (value != null) { if (tempValueMappings.containsKey(value)) { if (!(tempValueMappings.get(value).contains(stack))) { tempValueMappings.get(value).add(stack); } } else { tempValueMappings.put(value, new ArrayList(Arrays.asList(stack))); } } } } valueMappings = ImmutableSortedMap.copyOf(tempValueMappings); // Serialize values to disk LogHelper.info("Saving energy values to disk"); saveEnergyValueRegistryToFile(); } private Map computeStackMappings() { Map computedStackMap = new HashMap(); for (WrappedStack recipeOutput : RecipeRegistry.getInstance().getRecipeMappings().keySet()) { if (!hasEnergyValue(recipeOutput.getWrappedStack(), false) && !computedStackMap.containsKey(recipeOutput)) { EnergyValue lowestValue = null; for (List recipeInputs : RecipeRegistry.getInstance().getRecipeMappings().get(recipeOutput)) { EnergyValue computedValue = EnergyValueHelper.computeEnergyValueFromList(recipeInputs); computedValue = EnergyValueHelper.factorEnergyValue(computedValue, recipeOutput.getStackSize()); if (computedValue != null) { if (computedValue.compareTo(lowestValue) < 0) { lowestValue = computedValue; } } } if ((lowestValue != null) && (lowestValue.getEnergyValue() > 0f)) { computedStackMap.put(new WrappedStack(recipeOutput.getWrappedStack()), lowestValue); } } } return computedStackMap; } public List getStacksInRange(int start, int finish) { return getStacksInRange(new EnergyValue(start), new EnergyValue(finish)); } public List getStacksInRange(float start, float finish) { return getStacksInRange(new EnergyValue(start), new EnergyValue(finish)); } public List getStacksInRange(EnergyValue start, EnergyValue finish) { List stacksInRange = new ArrayList(); SortedMap> tailMap = energyValueRegistry.valueMappings.tailMap(start); SortedMap> headMap = energyValueRegistry.valueMappings.headMap(finish); SortedMap> smallerMap; SortedMap> biggerMap; if (!tailMap.isEmpty() && !headMap.isEmpty()) { if (tailMap.size() <= headMap.size()) { smallerMap = tailMap; biggerMap = headMap; } else { smallerMap = headMap; biggerMap = tailMap; } for (EnergyValue value : smallerMap.keySet()) { if (biggerMap.containsKey(value)) { for (WrappedStack wrappedStack : energyValueRegistry.valueMappings.get(value)) { if (wrappedStack.getWrappedStack() instanceof ItemStack || wrappedStack.getWrappedStack() instanceof FluidStack) { stacksInRange.add(wrappedStack.getWrappedStack()); } else if (wrappedStack.getWrappedStack() instanceof OreStack) { for (ItemStack itemStack : OreDictionary.getOres(((OreStack) wrappedStack.getWrappedStack()).oreName)) { stacksInRange.add(itemStack); } } } } } } return stacksInRange; } @Override public void readFromNBT(NBTTagCompound nbtTagCompound) { if (nbtTagCompound != null && nbtTagCompound.hasKey("modListMD5")) { this.modListMD5 = nbtTagCompound.getString("modListMD5"); } if (nbtTagCompound != null && nbtTagCompound.hasKey("stackMappingList")) { HashMap stackValueMap = new HashMap(); /** * Read stack value mappings from NBTTagCompound */ NBTTagList stackMappingTagList = nbtTagCompound.getTagList("stackMappingList", 10); for (int i = 0; i < stackMappingTagList.tagCount(); i++) { NBTTagCompound tagCompound = stackMappingTagList.getCompoundTagAt(i); WrappedStack wrappedStack = WrappedStack.fromNBTTagCompound(tagCompound.getCompoundTag("wrappedStack")); EnergyValue energyValue = EnergyValue.loadEnergyValueFromNBT(tagCompound.getCompoundTag("energyValue")); stackValueMap.put(wrappedStack, energyValue); } ImmutableSortedMap.Builder stackMappingsBuilder = ImmutableSortedMap.naturalOrder(); stackMappingsBuilder.putAll(stackValueMap); stackMappings = stackMappingsBuilder.build(); /** * Resolve value stack mappings from the newly loaded stack mappings */ SortedMap> tempValueMappings = new TreeMap>(); for (WrappedStack stack : stackMappings.keySet()) { if (stack != null) { EnergyValue value = stackMappings.get(stack); if (value != null) { if (tempValueMappings.containsKey(value)) { if (!(tempValueMappings.get(value).contains(stack))) { tempValueMappings.get(value).add(stack); } } else { tempValueMappings.put(value, new ArrayList(Arrays.asList(stack))); } } } } valueMappings = ImmutableSortedMap.copyOf(tempValueMappings); } } @Override public void writeToNBT(NBTTagCompound nbtTagCompound) { NBTTagList stackMappingTagList = new NBTTagList(); for (WrappedStack wrappedStack : stackMappings.keySet()) { if (wrappedStack != null && stackMappings.get(wrappedStack) != null) { NBTTagCompound stackMappingCompound = new NBTTagCompound(); stackMappingCompound.setTag("wrappedStack", WrappedStack.toNBTTagCompound(wrappedStack)); stackMappingCompound.setTag("energyValue", EnergyValue.writeEnergyValueToNBT(stackMappings.get(wrappedStack))); stackMappingTagList.appendTag(stackMappingCompound); } } nbtTagCompound.setTag("stackMappingList", stackMappingTagList); nbtTagCompound.setString("modListMD5", SerializationHelper.getModListMD5()); } @Override public String getTagLabel() { return "EnergyValueRegistry"; } public void setEnergyValue(WrappedStack wrappedStack, EnergyValue energyValue) { if (wrappedStack != null && energyValue != null && Float.compare(energyValue.getEnergyValue(), 0f) > 0) { HashMap stackValueMap = new HashMap(); /** * Read stack value mappings from NBTTagCompound */ stackValueMap.putAll(stackMappings); stackValueMap.put(wrappedStack, energyValue); ImmutableSortedMap.Builder stackMappingsBuilder = ImmutableSortedMap.naturalOrder(); stackMappingsBuilder.putAll(stackValueMap); stackMappings = stackMappingsBuilder.build(); /** * Resolve value stack mappings from the newly loaded stack mappings */ SortedMap> tempValueMappings = new TreeMap>(); for (WrappedStack stack : stackMappings.keySet()) { if (stack != null) { EnergyValue value = stackMappings.get(stack); if (value != null) { if (tempValueMappings.containsKey(value)) { if (!(tempValueMappings.get(value).contains(stack))) { tempValueMappings.get(value).add(stack); } } else { tempValueMappings.put(value, new ArrayList(Arrays.asList(stack))); } } } } valueMappings = ImmutableSortedMap.copyOf(tempValueMappings); } } public boolean getShouldRegenNextRestart() { return shouldRegenNextRestart; } public void setShouldRegenNextRestart(boolean shouldRegenNextRestart) { this.shouldRegenNextRestart = shouldRegenNextRestart; } public void saveEnergyValueRegistryToFile() { File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues"); energyValuesDataDirectory.mkdirs(); if (shouldRegenNextRestart) { File staticEnergyValuesFile = new File(energyValuesDataDirectory, Files.STATIC_ENERGY_VALUES); File md5EnergyValuesFile = new File(energyValuesDataDirectory, SerializationHelper.getModListMD5() + ".dat"); if (staticEnergyValuesFile.exists()) { staticEnergyValuesFile.delete(); } if (md5EnergyValuesFile.exists()) { md5EnergyValuesFile.delete(); } shouldRegenNextRestart = false; } else { SerializationHelper.writeNBTToFile(energyValuesDataDirectory, Files.STATIC_ENERGY_VALUES, this); SerializationHelper.writeNBTToFile(energyValuesDataDirectory, SerializationHelper.getModListMD5() + ".dat", this); } } public boolean loadEnergyValueRegistryFromFile() { File energyValuesDataDirectory = new File(FMLCommonHandler.instance().getMinecraftServerInstance().getEntityWorld().getSaveHandler().getWorldDirectory(), "data" + File.separator + Reference.LOWERCASE_MOD_ID + File.separator + "energyvalues"); energyValuesDataDirectory.mkdirs(); File staticEnergyValuesFile = new File(energyValuesDataDirectory, Files.STATIC_ENERGY_VALUES); File md5EnergyValuesFile = new File(energyValuesDataDirectory, SerializationHelper.getModListMD5() + ".dat"); NBTTagCompound nbtTagCompound = null; if (Settings.DynamicEnergyValueGeneration.regenerateEnergyValuesWhen.equalsIgnoreCase("Never")) { if (staticEnergyValuesFile.exists()) { LogHelper.info("Attempting to load energy values from file: " + staticEnergyValuesFile.getAbsolutePath()); nbtTagCompound = SerializationHelper.readNBTFromFile(staticEnergyValuesFile); } else if (md5EnergyValuesFile.exists()) { LogHelper.info("Attempting to load energy values from file: " + md5EnergyValuesFile.getAbsolutePath()); nbtTagCompound = SerializationHelper.readNBTFromFile(md5EnergyValuesFile); } } else if (md5EnergyValuesFile.exists()) { LogHelper.info("Attempting to load energy values from file: " + md5EnergyValuesFile.getAbsolutePath()); nbtTagCompound = SerializationHelper.readNBTFromFile(md5EnergyValuesFile); } if (nbtTagCompound != null) { energyValueRegistry.readFromNBT(nbtTagCompound); LogHelper.info("Successfully loaded energy values from file"); return true; } else { LogHelper.info("No energy value file to load values from, generating new values"); return false; } } }