1 files changed, 38 insertions, 2 deletions

diff --git a/src/cuchaz/enigma/convert/ClassIdentity.java b/src/cuchaz/enigma/convert/ClassIdentity.java
index 0a3a449..980f31f 100644
--- a/src/cuchaz/enigma/convert/ClassIdentity.java
+++ b/src/cuchaz/enigma/convert/ClassIdentity.java
@@ -60,6 +60,7 @@ import cuchaz.enigma.mapping.SignatureUpdater.ClassNameUpdater;
 public class ClassIdentity
 {
         private ClassEntry m_classEntry;
+        private String m_rawName;
         private SidedClassNamer m_namer;
         private Set<String> m_fields;
         private Set<String> m_methods;
@@ -70,13 +71,18 @@ public class ClassIdentity
         private Set<String> m_implementations;
         private Set<String> m_references;
         
-        public ClassIdentity( CtClass c, SidedClassNamer namer, JarIndex index, boolean useReferences )
+        public ClassIdentity( CtClass c, SidedClassNamer namer, JarIndex index, boolean useReferences, boolean useRawNames )
         {
                 m_namer = namer;
                 
                 // stuff from the bytecode
                 
                 m_classEntry = new ClassEntry( Descriptor.toJvmName( c.getName() ) );
+                m_rawName = "";
+                if( useRawNames )
+                {
+                        m_rawName = m_classEntry.getName();
+                }
                 m_fields = Sets.newHashSet();
                 for( CtField field : c.getDeclaredFields() )
                 {
@@ -176,8 +182,16 @@ public class ClassIdentity
         {
                 StringBuilder buf = new StringBuilder();
                 buf.append( "class: " );
+                buf.append( m_classEntry.getName() );
+                buf.append( " " );
                 buf.append( hashCode() );
                 buf.append( "\n" );
+                if( m_rawName.length() > 0 )
+                {
+                        buf.append( "\traw name: " );
+                        buf.append( m_rawName );
+                        buf.append( "\n" );
+                }
                 for( String field : m_fields )
                 {
                         buf.append( "\tfield " );
@@ -425,7 +439,8 @@ public class ClassIdentity
         
         public boolean equals( ClassIdentity other )
         {
-                return m_fields.equals( other.m_fields )
+                return m_rawName.equals( other.m_rawName )
+                        && m_fields.equals( other.m_fields )
                         && m_methods.equals( other.m_methods )
                         && m_constructors.equals( other.m_constructors )
                         && m_staticInitializer.equals( other.m_staticInitializer )
@@ -439,6 +454,7 @@ public class ClassIdentity
         public int hashCode( )
         {
                 List<Object> objs = Lists.newArrayList();
+                objs.add( m_rawName );
                 objs.addAll( m_fields );
                 objs.addAll( m_methods );
                 objs.addAll( m_constructors );
@@ -449,4 +465,24 @@ public class ClassIdentity
                 objs.addAll( m_references );
                 return Util.combineHashesOrdered( objs );
         }
+        
+        public int getMatchScore( ClassIdentity other )
+        {
+                return getNumMatches( m_fields, other.m_fields )
+                        + getNumMatches( m_methods, other.m_methods )
+                        + getNumMatches( m_constructors, other.m_constructors );
+        }
+        
+        private int getNumMatches( Set<String> a, Set<String> b ) 
+        {
+                int numMatches = 0;
+                for( String val : a )
+                {
+                        if( b.contains( val ) )
+                        {
+                                numMatches++;
+                        }
+                }
+                return numMatches;
+        }
 }