1 files changed, 29 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/ClassMapping.java b/src/cuchaz/enigma/mapping/ClassMapping.java
index 5936512..095cb38 100644
--- a/src/cuchaz/enigma/mapping/ClassMapping.java
+++ b/src/cuchaz/enigma/mapping/ClassMapping.java
@@ -11,6 +11,7 @@
 package cuchaz.enigma.mapping;
 
 import java.io.Serializable;
+import java.util.ArrayList;
 import java.util.Map;
 
 import com.google.common.collect.Maps;
@@ -299,4 +300,32 @@ public class ClassMapping implements Serializable, Comparable<ClassMapping>
         {
                 return m_obfName.compareTo( other.m_obfName );
         }
+        
+        public void renameObfClasses( Map<String,String> nameMap )
+        {
+                // rename self
+                {
+                        String newName = nameMap.get( m_obfName );
+                        if( newName != null )
+                        {
+                                m_obfName = newName;
+                        }
+                }
+                
+                // rename inner classes
+                for( ClassMapping classMapping : new ArrayList<ClassMapping>( m_innerClassesByObf.values() ) )
+                {
+                        m_innerClassesByObf.remove( classMapping.getObfName() );
+                        classMapping.renameObfClasses( nameMap );
+                        m_innerClassesByObf.put( classMapping.getObfName(), classMapping );
+                }
+                
+                // rename method signatures
+                for( MethodMapping methodMapping : new ArrayList<MethodMapping>( m_methodsByObf.values() ) )
+                {
+                        m_methodsByObf.remove( getMethodKey( methodMapping.getObfName(), methodMapping.getObfSignature() ) );
+                        methodMapping.renameObfClasses( nameMap );
+                        m_methodsByObf.put( getMethodKey( methodMapping.getObfName(), methodMapping.getObfSignature() ), methodMapping );
+                }
+        }
 }