diff options
2 files changed, 14 insertions, 0 deletions
diff --git a/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalInstanceStackMap.java b/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalInstanceStackMap.java index 79c7d30bca..b0edc510b6 100644 --- a/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalInstanceStackMap.java +++ b/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalInstanceStackMap.java @@ -429,4 +429,10 @@ public final class cElementalInstanceStackMap implements Comparable<cElementalIn } return build.toString(); } + + public void cleanUp(){ + for(Map.Entry<iElementalDefinition, cElementalInstanceStack> entry:map.entrySet()){ + if(entry.getValue().amount<=0) map.remove(entry.getKey()); + } + } } diff --git a/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalStackMap.java b/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalStackMap.java index 2a11e908f7..3be0889671 100644 --- a/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalStackMap.java +++ b/src/main/java/com/github/technus/tectech/elementalMatter/core/cElementalStackMap.java @@ -5,6 +5,7 @@ import com.github.technus.tectech.elementalMatter.core.interfaces.iElementalDefi import net.minecraft.nbt.NBTTagCompound; import net.minecraft.util.EnumChatFormatting; +import java.util.Map; import java.util.TreeMap; /** @@ -27,6 +28,7 @@ abstract class cElementalStackMap implements Comparable<cElementalStackMap> { public final cElementalDefinitionStack getLast(){ return map.lastEntry().getValue(); } + public final cElementalDefinitionStack getDefinitionStack(iElementalDefinition def) { return map.get(def); } @@ -123,4 +125,10 @@ abstract class cElementalStackMap implements Comparable<cElementalStackMap> { } return hash; } + + public void cleanUp(){ + for(Map.Entry<iElementalDefinition, cElementalDefinitionStack> entry:map.entrySet()){ + if(entry.getValue().amount<=0) map.remove(entry.getKey()); + } + } } |