1 files changed, 56 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/ClassInheritanceTreeNode.java b/src/cuchaz/enigma/gui/ClassInheritanceTreeNode.java
new file mode 100644
index 0000000..921a1e9
--- /dev/null
+++ b/src/cuchaz/enigma/gui/ClassInheritanceTreeNode.java
@@ -0,0 +1,56 @@
+package cuchaz.enigma.gui;
+
+import java.util.List;
+
+import javax.swing.tree.DefaultMutableTreeNode;
+
+import com.beust.jcommander.internal.Lists;
+
+import cuchaz.enigma.mapping.Ancestries;
+
+public class ClassInheritanceTreeNode extends DefaultMutableTreeNode
+{
+        private static final long serialVersionUID = 4432367405826178490L;
+        
+        String m_className;
+        
+        public ClassInheritanceTreeNode( String className )
+        {
+                m_className = className;
+        }
+        
+        public String getClassName( )
+        {
+                return m_className;
+        }
+        
+        @Override
+        public String toString( )
+        {
+                return m_className;
+        }
+        
+        public void load( Ancestries ancestries, boolean recurse )
+        {
+                // get all the child nodes
+                List<ClassInheritanceTreeNode> nodes = Lists.newArrayList();
+                for( String subclassName : ancestries.getSubclasses( m_className ) )
+                {
+                        nodes.add( new ClassInheritanceTreeNode( subclassName ) );
+                }
+                
+                // add then to this node
+                for( ClassInheritanceTreeNode node : nodes )
+                {
+                        this.add( node );
+                }
+                
+                if( recurse )
+                {
+                        for( ClassInheritanceTreeNode node : nodes )
+                        {
+                                node.load( ancestries, true );
+                        }
+                }
+        }
+}