got simple method call graph working!

1 files changed, 11 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/GuiController.java b/src/cuchaz/enigma/gui/GuiController.java
index 880f001..b54aeba 100644
--- a/src/cuchaz/enigma/gui/GuiController.java
+++ b/src/cuchaz/enigma/gui/GuiController.java
@@ -21,6 +21,7 @@ import com.google.common.collect.Lists;
 
 import cuchaz.enigma.Deobfuscator;
 import cuchaz.enigma.analysis.ClassInheritanceTreeNode;
+import cuchaz.enigma.analysis.MethodCallsTreeNode;
 import cuchaz.enigma.analysis.MethodInheritanceTreeNode;
 import cuchaz.enigma.analysis.SourceIndex;
 import cuchaz.enigma.analysis.Token;
@@ -148,6 +149,16 @@ public class GuiController
                 return MethodInheritanceTreeNode.findNode( rootNode, obfMethodEntry );
         }
         
+        public MethodCallsTreeNode getMethodCalls( MethodEntry obfMethodEntry )
+        {
+                MethodCallsTreeNode rootNode = new MethodCallsTreeNode(
+                        m_deobfuscator.getTranslator( TranslationDirection.Deobfuscating ),
+                        obfMethodEntry
+                );
+                rootNode.load( m_deobfuscator.getJarIndex(), true );
+                return rootNode;
+        }
+        
         public void rename( Entry obfEntry, String newName )
         {
                 m_deobfuscator.rename( obfEntry, newName );