From 72115a6e4c83422b7359a9ae4d60badc244b55ff Mon Sep 17 00:00:00 2001
From: Thog
Date: Wed, 19 Oct 2016 17:44:04 +0200
Subject: Starting implementing local variables (#33)

TODO:
- Store format (need to be defined)
- Implement some translate operations

This commit also fix some cases where argument tokens are not selected
---
 src/main/java/cuchaz/enigma/analysis/TranslationIndex.java | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'src/main/java/cuchaz/enigma/analysis/TranslationIndex.java')

diff --git a/src/main/java/cuchaz/enigma/analysis/TranslationIndex.java b/src/main/java/cuchaz/enigma/analysis/TranslationIndex.java
index 25edaef..d51131f 100644
--- a/src/main/java/cuchaz/enigma/analysis/TranslationIndex.java
+++ b/src/main/java/cuchaz/enigma/analysis/TranslationIndex.java
@@ -181,6 +181,8 @@ public class TranslationIndex {
             return behaviorExists((BehaviorEntry) entry);
         } else if (entry instanceof ArgumentEntry) {
             return behaviorExists(((ArgumentEntry) entry).getBehaviorEntry());
+        } else if (entry instanceof LocalVariableEntry) {
+            return behaviorExists(((LocalVariableEntry) entry).getBehaviorEntry());
         }
         throw new IllegalArgumentException("Cannot check existence for " + entry.getClass());
     }
-- 
cgit v1.2.3