Use reference map instead of EnumMap for nullable key. Closes #5504

mekanism · Jun 7, 2019 · 28c13c8 · 28c13c8
1 parent 9132aa7
commit 28c13c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/mekanism/common/capabilities/CapabilityWrapperManager.java b/src/main/java/mekanism/common/capabilities/CapabilityWrapperManager.java
@@ -1,12 +1,12 @@
 package mekanism.common.capabilities;
 
-import java.util.EnumMap;
+import it.unimi.dsi.fastutil.objects.Reference2ObjectArrayMap;
 import java.util.Map;
 import net.minecraft.util.EnumFacing;
 
 public class CapabilityWrapperManager<IMPL, WRAPPER> {
 
-    private Map<EnumFacing, WRAPPER> wrappers = new EnumMap<>(EnumFacing.class);
+    private Map<EnumFacing, WRAPPER> wrappers = new Reference2ObjectArrayMap<>(EnumFacing.values().length+1);
     private Class<IMPL> typeClass;
     private Class<WRAPPER> wrapperClass;