generalized field matching

added method matching

1 files changed, 12 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/EntryFactory.java b/src/cuchaz/enigma/mapping/EntryFactory.java
index 333bb09..7bc6183 100644
--- a/src/cuchaz/enigma/mapping/EntryFactory.java
+++ b/src/cuchaz/enigma/mapping/EntryFactory.java
@@ -138,6 +138,10 @@ public class EntryFactory {
         public static BehaviorEntry getBehaviorEntry(String className, String behaviorName, String behaviorSignature) {
                 return getBehaviorEntry(new ClassEntry(className), behaviorName, new Signature(behaviorSignature));
         }
+
+        public static BehaviorEntry getBehaviorEntry(String className, String behaviorName) {
+                return getBehaviorEntry(new ClassEntry(className), behaviorName);
+        }
         
         public static BehaviorEntry getBehaviorEntry(ClassEntry classEntry, String behaviorName, Signature behaviorSignature) {
                 if (behaviorName.equals("<init>")) {
@@ -149,6 +153,14 @@ public class EntryFactory {
                 }
         }
         
+        public static BehaviorEntry getBehaviorEntry(ClassEntry classEntry, String behaviorName) {
+                if(behaviorName.equals("<clinit>")) {
+                        return new ConstructorEntry(classEntry);
+                } else {
+                        throw new IllegalArgumentException("Only class initializers don't have signatures");
+                }
+        }
+        
         public static BehaviorEntry getBehaviorEntry(MethodDefinition def) {
                 if (def.isConstructor() || def.isTypeInitializer()) {
                         return getConstructorEntry(def);