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

1 files changed, 2 insertions, 0 deletions

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());
     }