add methods for better runtime obfuscation in M3L

2 files changed, 20 insertions, 10 deletions

diff --git a/src/cuchaz/enigma/analysis/TranslationIndex.java b/src/cuchaz/enigma/analysis/TranslationIndex.java
index 8651ebd..bd77344 100644
--- a/src/cuchaz/enigma/analysis/TranslationIndex.java
+++ b/src/cuchaz/enigma/analysis/TranslationIndex.java
@@ -84,6 +84,10 @@ public class TranslationIndex implements Serializable {
         }
         
         public void indexClass(CtClass c) {
+                indexClass(c, true);
+        }
+        
+        public void indexClass(CtClass c, boolean indexMembers) {
                 
                 ClassEntry classEntry = EntryFactory.getClassEntry(c);
                 
@@ -93,16 +97,18 @@ public class TranslationIndex implements Serializable {
                         m_superclasses.put(classEntry, superclassEntry);
                 }
                 
-                // add fields
-                for (CtField field : c.getDeclaredFields()) {
-                        FieldEntry fieldEntry = EntryFactory.getFieldEntry(field);
-                        m_fieldEntries.put(fieldEntry.getClassEntry(), fieldEntry);
-                }
-                
-                // add behaviors
-                for (CtBehavior behavior : c.getDeclaredBehaviors()) {
-                        BehaviorEntry behaviorEntry = EntryFactory.getBehaviorEntry(behavior);
-                        m_behaviorEntries.put(behaviorEntry.getClassEntry(), behaviorEntry);
+                if (indexMembers) {
+                        // add fields
+                        for (CtField field : c.getDeclaredFields()) {
+                                FieldEntry fieldEntry = EntryFactory.getFieldEntry(field);
+                                m_fieldEntries.put(fieldEntry.getClassEntry(), fieldEntry);
+                        }
+                        
+                        // add behaviors
+                        for (CtBehavior behavior : c.getDeclaredBehaviors()) {
+                                BehaviorEntry behaviorEntry = EntryFactory.getBehaviorEntry(behavior);
+                                m_behaviorEntries.put(behaviorEntry.getClassEntry(), behaviorEntry);
+                        }
                 }
         }
         
diff --git a/src/cuchaz/enigma/mapping/Translator.java b/src/cuchaz/enigma/mapping/Translator.java
index dcb7435..5d17494 100644
--- a/src/cuchaz/enigma/mapping/Translator.java
+++ b/src/cuchaz/enigma/mapping/Translator.java
@@ -43,6 +43,10 @@ public class Translator {
                 m_index = index;
         }
         
+        public TranslationDirection getDirection() {
+                return m_direction;
+        }
+        
         public TranslationIndex getTranslationIndex() {
                 return m_index;
         }