finished class matching for now, need to work on class member matching

1 files changed, 25 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/Mappings.java b/src/cuchaz/enigma/mapping/Mappings.java
index f3b8fad..70bea25 100644
--- a/src/cuchaz/enigma/mapping/Mappings.java
+++ b/src/cuchaz/enigma/mapping/Mappings.java
@@ -14,6 +14,7 @@ import java.io.IOException;
 import java.io.InputStream;
 import java.io.ObjectInputStream;
 import java.io.Serializable;
+import java.util.ArrayList;
 import java.util.Map;
 import java.util.zip.GZIPInputStream;
 
@@ -136,4 +137,28 @@ public class Mappings implements Serializable
                 }
                 return buf.toString();
         }
+        
+        public void renameObfClasses( Map<String,String> nameMap )
+        {
+                for( ClassMapping classMapping : new ArrayList<ClassMapping>( m_classesByObf.values() ) )
+                {
+                        String newName = nameMap.get( classMapping.getObfName() );
+                        if( newName != null )
+                        {
+                                m_classesByObf.remove( classMapping.getObfName() );
+                                classMapping.renameObfClasses( nameMap );
+                                m_classesByObf.put( classMapping.getObfName(), classMapping );
+                        }
+                }
+        }
+
+        public void removeClassByObfName( String obfName )
+        {
+                ClassMapping classMapping = m_classesByObf.get( obfName );
+                if( classMapping != null )
+                {
+                        m_classesByObf.remove( classMapping.getObfName() );
+                        m_classesByDeobf.remove( classMapping.getDeobfName() );
+                }
+        }
 }