1 files changed, 20 insertions, 0 deletions

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;
+                        }
+                } );
+        }
 }