2 files changed, 21 insertions, 2 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/JarIndex.java b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
index 820ecbb..361c8e7 100644
--- a/src/main/java/cuchaz/enigma/analysis/JarIndex.java
+++ b/src/main/java/cuchaz/enigma/analysis/JarIndex.java
@@ -447,8 +447,22 @@ public class JarIndex {
                 return this.methodsReferencingClasses.get(classEntry);
         }
 
+        @Deprecated
         public Collection<EntryReference<MethodEntry, MethodDefEntry>> getMethodsReferencing(MethodEntry methodEntry) {
-                return this.methodsReferencing.get(methodEntry);
+                return getMethodsReferencing(methodEntry, false);
+        }
+
+        public Collection<EntryReference<MethodEntry, MethodDefEntry>> getMethodsReferencing(MethodEntry methodEntry, boolean recurse) {
+                if (!recurse) {
+                        return this.methodsReferencing.get(methodEntry);
+                }
+
+                List<EntryReference<MethodEntry, MethodDefEntry>> references = new ArrayList<>();
+                Set<MethodEntry> methodEntries = getRelatedMethodImplementations(methodEntry);
+                for (MethodEntry entry : methodEntries) {
+                        references.addAll(getMethodsReferencing(entry, false));
+                }
+                return references;
         }
 
         public Collection<MethodEntry> getReferencedMethods(MethodDefEntry methodEntry) {
diff --git a/src/main/java/cuchaz/enigma/analysis/MethodReferenceTreeNode.java b/src/main/java/cuchaz/enigma/analysis/MethodReferenceTreeNode.java
index 0eae5cc..ac05acd 100644
--- a/src/main/java/cuchaz/enigma/analysis/MethodReferenceTreeNode.java
+++ b/src/main/java/cuchaz/enigma/analysis/MethodReferenceTreeNode.java
@@ -63,9 +63,14 @@ public class MethodReferenceTreeNode extends DefaultMutableTreeNode
                 return this.deobfuscatingTranslator.getTranslatedMethod(this.entry).getName();
         }
 
+        @Deprecated
         public void load(JarIndex index, boolean recurse) {
+                load(index, recurse, false);
+        }
+
+        public void load(JarIndex index, boolean recurse, boolean recurseMethod) {
                 // get all the child nodes
-                for (EntryReference<MethodEntry, MethodDefEntry> reference : index.getMethodsReferencing(this.entry)) {
+                for (EntryReference<MethodEntry, MethodDefEntry> reference : index.getMethodsReferencing(this.entry, recurseMethod)) {
                         add(new MethodReferenceTreeNode(this.deobfuscatingTranslator, reference, index.getAccessFlags(this.entry)));
                 }