From d71c0af8ed298bfb4b35b4e3e61b5678bc1c7d9f Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 6 Aug 2014 00:22:25 -0400
Subject: added simple class inheritance browsing

---
 src/cuchaz/enigma/mapping/Ancestries.java | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

(limited to 'src/cuchaz/enigma/mapping')

diff --git a/src/cuchaz/enigma/mapping/Ancestries.java b/src/cuchaz/enigma/mapping/Ancestries.java
index b7a5e24..f77a00e 100644
--- a/src/cuchaz/enigma/mapping/Ancestries.java
+++ b/src/cuchaz/enigma/mapping/Ancestries.java
@@ -26,6 +26,7 @@ import javassist.CtClass;
 import javassist.NotFoundException;
 import javassist.bytecode.Descriptor;
 
+import com.beust.jcommander.internal.Lists;
 import com.google.common.collect.Maps;
 
 import cuchaz.enigma.Constants;
@@ -121,11 +122,30 @@ public class Ancestries implements Serializable
 		while( className != null )
 		{
 			className = getSuperclassName( className );
-			ancestors.add( className );
+			if( className != null )
+			{
+				ancestors.add( className );
+			}
 		}
 		return ancestors;
 	}
 	
+	public List<String> getSubclasses( String className )
+	{
+		// linear search is fast enough for now
+		List<String> subclasses = Lists.newArrayList();
+		for( Map.Entry<String,String> entry : m_superclasses.entrySet() )
+		{
+			String subclass = entry.getKey();
+			String superclass = entry.getValue();
+			if( className.equals( superclass ) )
+			{
+				subclasses.add( subclass );
+			}
+		}
+		return subclasses;
+	}
+	
 	private boolean isJre( String className )
 	{
 		return className.startsWith( "java/" )
-- 
cgit v1.2.3