From dc2120999c137aa4763ea2358b8df83f4098d280 Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 11 Mar 2015 11:44:24 -0400
Subject: working on writing mappings based on all the matches

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

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

diff --git a/src/cuchaz/enigma/mapping/ClassMapping.java b/src/cuchaz/enigma/mapping/ClassMapping.java
index 07fed32..ac70df0 100644
--- a/src/cuchaz/enigma/mapping/ClassMapping.java
+++ b/src/cuchaz/enigma/mapping/ClassMapping.java
@@ -243,6 +243,15 @@ public class ClassMapping implements Serializable, Comparable<ClassMapping> {
 		}
 	}
 	
+	public void setFieldObfName(String oldObfName, Type obfType, String newObfName) {
+		assert(newObfName != null);
+		FieldMapping fieldMapping = m_fieldsByObf.remove(getFieldKey(oldObfName, obfType));
+		assert(fieldMapping != null);
+		fieldMapping.setObfName(newObfName);
+		boolean obfWasAdded = m_fieldsByObf.put(getFieldKey(newObfName, obfType), fieldMapping) == null;
+		assert(obfWasAdded);
+	}
+	
 	
 	//// METHODS ////////
 	
@@ -319,6 +328,15 @@ public class ClassMapping implements Serializable, Comparable<ClassMapping> {
 		}
 	}
 	
+	public void setMethodObfName(String oldObfName, Signature obfSignature, String newObfName) {
+		assert(newObfName != null);
+		MethodMapping methodMapping = m_methodsByObf.remove(getMethodKey(oldObfName, obfSignature));
+		assert(methodMapping != null);
+		methodMapping.setObfName(newObfName);
+		boolean obfWasAdded = m_methodsByObf.put(getMethodKey(newObfName, obfSignature), methodMapping) == null;
+		assert(obfWasAdded);
+	}
+	
 	//// ARGUMENTS ////////
 	
 	public void setArgumentName(String obfMethodName, Signature obfMethodSignature, int argumentIndex, String argumentName) {
-- 
cgit v1.2.3