add cleanup

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());
+        }
+    }
 }