From 4ceb8d490058e48df666bf7227ce020e60928be5 Mon Sep 17 00:00:00 2001
From: jeff
Date: Sun, 8 Mar 2015 20:48:30 -0400
Subject: more tweaks, improvements, and bug fixes

---
 src/cuchaz/enigma/mapping/MethodMapping.java | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

(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 1704428..bf8a94f 100644
--- a/src/cuchaz/enigma/mapping/MethodMapping.java
+++ b/src/cuchaz/enigma/mapping/MethodMapping.java
@@ -12,7 +12,9 @@ package cuchaz.enigma.mapping;
 
 import java.io.Serializable;
 import java.util.Map;
-import java.util.TreeMap;
+import java.util.Map.Entry;
+
+import com.google.common.collect.Maps;
 
 public class MethodMapping implements Serializable, Comparable<MethodMapping> {
 	
@@ -37,9 +39,19 @@ public class MethodMapping implements Serializable, Comparable<MethodMapping> {
 		m_obfName = obfName;
 		m_deobfName = NameValidator.validateMethodName(deobfName);
 		m_obfSignature = obfSignature;
-		m_arguments = new TreeMap<Integer,ArgumentMapping>();
+		m_arguments = Maps.newTreeMap();
 	}
 	
+	public MethodMapping(MethodMapping other, ClassNameReplacer obfClassNameReplacer) {
+		m_obfName = other.m_obfName;
+		m_deobfName = other.m_deobfName;
+		m_obfSignature = new Signature(other.m_obfSignature, obfClassNameReplacer);
+		m_arguments = Maps.newTreeMap();
+		for (Entry<Integer,ArgumentMapping> entry : other.m_arguments.entrySet()) {
+			m_arguments.put(entry.getKey(), new ArgumentMapping(entry.getValue()));
+		}
+	}
+
 	public String getObfName() {
 		return m_obfName;
 	}
-- 
cgit v1.2.3