field matcher is starting to be useful

1 files changed, 16 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/EntryFactory.java b/src/cuchaz/enigma/mapping/EntryFactory.java
index f4d62c8..333bb09 100644
--- a/src/cuchaz/enigma/mapping/EntryFactory.java
+++ b/src/cuchaz/enigma/mapping/EntryFactory.java
@@ -26,6 +26,10 @@ public class EntryFactory {
                 return obfClassEntry.buildClassEntry(jarIndex.getObfClassChain(obfClassEntry));
         }
         
+        private static ClassEntry getObfClassEntry(ClassMapping classMapping) {
+                return new ClassEntry(classMapping.getObfFullName());
+        }
+        
         public static ClassEntry getDeobfClassEntry(ClassMapping classMapping) {
                 return new ClassEntry(classMapping.getDeobfName());
         }
@@ -50,6 +54,14 @@ public class EntryFactory {
                 );
         }
         
+        public static FieldEntry getObfFieldEntry(ClassMapping classMapping, FieldMapping fieldMapping) {
+                return new FieldEntry(
+                        getObfClassEntry(classMapping),
+                        fieldMapping.getObfName(),
+                        fieldMapping.getObfType()
+                );
+        }
+        
         public static MethodEntry getMethodEntry(CtMethod method) {
                 return new MethodEntry(
                         getClassEntry(method.getDeclaringClass()),
@@ -148,4 +160,8 @@ public class EntryFactory {
         public static BehaviorEntry getObfBehaviorEntry(ClassEntry classEntry, MethodMapping methodMapping) {
                 return getBehaviorEntry(classEntry, methodMapping.getObfName(), methodMapping.getObfSignature());
         }
+        
+        public static BehaviorEntry getObfBehaviorEntry(ClassMapping classMapping, MethodMapping methodMapping) {
+                return getObfBehaviorEntry(getObfClassEntry(classMapping), methodMapping);
+        }
 }