From 63172120a39a315e29bc38ea6634741797b3dcab Mon Sep 17 00:00:00 2001
From: jeff
Date: Sat, 30 Aug 2014 14:14:54 -0400
Subject: finished class matching for now, need to work on class member
 matching

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

(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 6e6bec4..fe4e29b 100644
--- a/src/cuchaz/enigma/mapping/MethodMapping.java
+++ b/src/cuchaz/enigma/mapping/MethodMapping.java
@@ -14,6 +14,8 @@ import java.io.Serializable;
 import java.util.Map;
 import java.util.TreeMap;
 
+import cuchaz.enigma.mapping.SignatureUpdater.ClassNameUpdater;
+
 public class MethodMapping implements Serializable, Comparable<MethodMapping>
 {
 	private static final long serialVersionUID = -4409570216084263978L;
@@ -139,4 +141,22 @@ public class MethodMapping implements Serializable, Comparable<MethodMapping>
 	{
 		return ( m_obfName + m_obfSignature ).compareTo( ( other.m_obfName + other.m_obfSignature ) );
 	}
+
+	public void renameObfClasses( final Map<String,String> nameMap )
+	{
+		// rename obf classes in the signature
+		m_obfSignature = SignatureUpdater.update( m_obfSignature, new ClassNameUpdater( )
+		{
+			@Override
+			public String update( String className )
+			{
+				String newName = nameMap.get( className );
+				if( newName != null )
+				{
+					return newName;
+				}
+				return className;
+			}
+		} );
+	}
 }
-- 
cgit v1.2.3