From e33b4003a5c423894e7aef575faff359dd1d33b1 Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 11 Mar 2015 11:03:16 -0400
Subject: generalized field matching added method matching

---
 src/cuchaz/enigma/mapping/EntryFactory.java | 12 ++++++++++++
 1 file changed, 12 insertions(+)

(limited to 'src/cuchaz/enigma/mapping/EntryFactory.java')

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);
-- 
cgit v1.2.3