fix up class matcher a bit

2 files changed, 30 insertions, 8 deletions

diff --git a/src/cuchaz/enigma/convert/ClassIdentity.java b/src/cuchaz/enigma/convert/ClassIdentity.java
index 2e164ae..d9ed08e 100644
--- a/src/cuchaz/enigma/convert/ClassIdentity.java
+++ b/src/cuchaz/enigma/convert/ClassIdentity.java
@@ -117,7 +117,7 @@ public class ClassIdentity {
                 
                 // stuff from the bytecode
                 
-                m_classEntry = new ClassEntry(Descriptor.toJvmName(c.getName()));
+                m_classEntry = EntryFactory.getClassEntry(c);
                 m_fields = HashMultiset.create();
                 for (CtField field : c.getDeclaredFields()) {
                         m_fields.add(scrubType(field.getSignature()));
@@ -180,7 +180,10 @@ public class ClassIdentity {
                         }
                 }
                 
-                m_outer = EntryFactory.getClassEntry(c).getOuterClassName();
+                m_outer = null;
+                if (m_classEntry.isInnerClass()) {
+                        m_outer = m_classEntry.getOuterClassName();
+                }
         }
         
         private void addReference(EntryReference<? extends Entry,BehaviorEntry> reference) {
@@ -460,7 +463,9 @@ public class ClassIdentity {
         }
         
         private int getNumMatches(String a, String b) {
-                if (a.equals(b)) {
+                if (a == null && b == null) {
+                        return 1;
+                } else if (a != null && b != null && a.equals(b)) {
                         return 1;
                 }
                 return 0;
diff --git a/src/cuchaz/enigma/convert/MappingsConverter.java b/src/cuchaz/enigma/convert/MappingsConverter.java
index b457d6c..b404e8f 100644
--- a/src/cuchaz/enigma/convert/MappingsConverter.java
+++ b/src/cuchaz/enigma/convert/MappingsConverter.java
@@ -207,18 +207,35 @@ public class MappingsConverter {
                         newClassMapping = new ClassMapping(newObfClass.getName());
                 }
                 
-                // copy fields
+                // migrate fields
                 for (FieldMapping fieldMapping : oldClassMapping.fields()) {
-                        newClassMapping.addFieldMapping(new FieldMapping(fieldMapping, replacer));
+                        if (canMigrate(fieldMapping.getObfType(), replacer)) {
+                                newClassMapping.addFieldMapping(new FieldMapping(fieldMapping, replacer));
+                        }
                 }
                 
-                // copy methods
-                for (MethodMapping methodMapping : oldClassMapping.methods()) {
-                        newClassMapping.addMethodMapping(new MethodMapping(methodMapping, replacer));
+                // migrate methods
+                for (MethodMapping oldMethodMapping : oldClassMapping.methods()) {
+                        if (canMigrate(oldMethodMapping.getObfSignature(), replacer)) {
+                                newClassMapping.addMethodMapping(new MethodMapping(oldMethodMapping, replacer));
+                        }
                 }
                 
                 return newClassMapping;
         }
+        
+        private static boolean canMigrate(Signature obfSignature, ClassNameReplacer replacer) {
+                for (Type type : obfSignature.types()) {
+                        if (!canMigrate(type, replacer)) {
+                                return false;
+                        }
+                }
+                return true;
+        }
+
+        private static boolean canMigrate(Type type, ClassNameReplacer replacer) {
+                return !type.hasClass() || replacer.replace(type.getClassEntry().getClassName()) != null;
+        }
 
         public static void convertMappings(Mappings mappings, BiMap<ClassEntry,ClassEntry> changes) {