From 2b2249e873c4adfd2dd6e8f1f2489ccd9f6aa021 Mon Sep 17 00:00:00 2001
From: gegy1000
Date: Sat, 19 May 2018 17:02:46 +0200
Subject: Initial port to ASM

---
 src/main/java/cuchaz/enigma/mapping/TranslationDirection.java | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

(limited to 'src/main/java/cuchaz/enigma/mapping/TranslationDirection.java')

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;
 		}
 	};
-- 
cgit v1.2.3