Don't save local variable entries with no mapping

1 files changed, 2 insertions, 8 deletions

diff --git a/src/main/java/cuchaz/enigma/translation/mapping/serde/EnigmaMappingsWriter.java b/src/main/java/cuchaz/enigma/translation/mapping/serde/EnigmaMappingsWriter.java
index 1d44b6e8..8c55fa99 100644
--- a/src/main/java/cuchaz/enigma/translation/mapping/serde/EnigmaMappingsWriter.java
+++ b/src/main/java/cuchaz/enigma/translation/mapping/serde/EnigmaMappingsWriter.java
@@ -165,7 +165,7 @@ public enum EnigmaMappingsWriter implements MappingsWriter {
                 } else if (entry instanceof FieldEntry) {
                         String line = writeField((FieldEntry) entry, mapping);
                         writer.println(indent(line, depth));
-                } else if (entry instanceof LocalVariableEntry) {
+                } else if (entry instanceof LocalVariableEntry && mapping != null) {
                         String line = writeArgument((LocalVariableEntry) entry, mapping);
                         writer.println(indent(line, depth));
                 }
@@ -231,13 +231,7 @@ public enum EnigmaMappingsWriter implements MappingsWriter {
         }
 
         protected String writeArgument(LocalVariableEntry entry, EntryMapping mapping) {
-                StringBuilder builder = new StringBuilder("ARG ");
-                builder.append(entry.getIndex()).append(' ');
-
-                String mappedName = mapping != null ? mapping.getTargetName() : entry.getName();
-                builder.append(mappedName);
-
-                return builder.toString();
+                return "ARG " + entry.getIndex() + ' ' + mapping.getTargetName();
         }
 
         private void writeMapping(StringBuilder builder, EntryMapping mapping) {