1 files changed, 22 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/gui/ClassInheritanceTreeNode.java b/src/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java
index 295bebe..2ed141f 100644
--- a/src/cuchaz/enigma/gui/ClassInheritanceTreeNode.java
+++ b/src/cuchaz/enigma/analysis/ClassInheritanceTreeNode.java
@@ -8,7 +8,7 @@
  * Contributors:
  *     Jeff Martin - initial API and implementation
  ******************************************************************************/
-package cuchaz.enigma.gui;
+package cuchaz.enigma.analysis;
 
 import java.util.List;
 
@@ -16,7 +16,7 @@ import javax.swing.tree.DefaultMutableTreeNode;
 
 import com.google.common.collect.Lists;
 
-import cuchaz.enigma.mapping.Ancestries;
+import cuchaz.enigma.mapping.ClassEntry;
 import cuchaz.enigma.mapping.Translator;
 
 public class ClassInheritanceTreeNode extends DefaultMutableTreeNode
@@ -76,4 +76,24 @@ public class ClassInheritanceTreeNode extends DefaultMutableTreeNode
                         }
                 }
         }
+
+        public static ClassInheritanceTreeNode findNode( ClassInheritanceTreeNode node, ClassEntry entry )
+        {
+                // is this the node?
+                if( node.getObfClassName().equals( entry.getName() ) )
+                {
+                        return node;
+                }
+                
+                // recurse
+                for( int i=0; i<node.getChildCount(); i++ )
+                {
+                        ClassInheritanceTreeNode foundNode = findNode( (ClassInheritanceTreeNode)node.getChildAt( i ), entry );
+                        if( foundNode != null )
+                        {
+                                return foundNode;
+                        }
+                }
+                return null;
+        }
 }