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/MethodMapping.java | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

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

diff --git a/src/cuchaz/enigma/mapping/MethodMapping.java b/src/cuchaz/enigma/mapping/MethodMapping.java
index bf8a94f..bf6dacc 100644
--- a/src/cuchaz/enigma/mapping/MethodMapping.java
+++ b/src/cuchaz/enigma/mapping/MethodMapping.java
@@ -16,7 +16,7 @@ import java.util.Map.Entry;
 
 import com.google.common.collect.Maps;
 
-public class MethodMapping implements Serializable, Comparable<MethodMapping> {
+public class MethodMapping implements Serializable, Comparable<MethodMapping>, MemberMapping<BehaviorEntry> {
 	
 	private static final long serialVersionUID = -4409570216084263978L;
 	
@@ -170,4 +170,13 @@ public class MethodMapping implements Serializable, Comparable<MethodMapping> {
 		}
 		return false;
 	}
+
+	@Override
+	public BehaviorEntry getObfEntry(ClassEntry classEntry) {
+		if (isConstructor()) {
+			return new ConstructorEntry(classEntry, m_obfSignature);
+		} else {
+			return new MethodEntry(classEntry, m_obfName, m_obfSignature);
+		}
+	}
 }
-- 
cgit v1.2.3