Merge pull request #62 from OpenModLoader/asm

ASM based class translator

1 files changed, 8 insertions, 2 deletions

diff --git a/src/main/java/cuchaz/enigma/mapping/TranslationDirection.java b/src/main/java/cuchaz/enigma/mapping/TranslationDirection.java
index 17e3187..4bbde54 100644
--- a/src/main/java/cuchaz/enigma/mapping/TranslationDirection.java
+++ b/src/main/java/cuchaz/enigma/mapping/TranslationDirection.java
@@ -13,15 +13,21 @@ package cuchaz.enigma.mapping;
 
 public enum TranslationDirection {
 
-        Deobfuscating {
+        DEOBFUSCATING {
                 @Override
                 public <T> T choose(T deobfChoice, T obfChoice) {
+                        if (deobfChoice == null) {
+                                return obfChoice;
+                        }
                         return deobfChoice;
                 }
         },
-        Obfuscating {
+        OBFUSCATING {
                 @Override
                 public <T> T choose(T deobfChoice, T obfChoice) {
+                        if (obfChoice == null) {
+                                return deobfChoice;
+                        }
                         return obfChoice;
                 }
         };