implemented mark-as-deobfuscated and reset-to-obfuscated

1 files changed, 11 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/mapping/Mappings.java b/src/cuchaz/enigma/mapping/Mappings.java
index 0b4e7f3..45b41bc 100644
--- a/src/cuchaz/enigma/mapping/Mappings.java
+++ b/src/cuchaz/enigma/mapping/Mappings.java
@@ -94,6 +94,17 @@ public class Mappings implements Serializable
                 }
         }
         
+        public void removeClassMapping( ClassMapping classMapping )
+        {
+                boolean obfWasRemoved = m_classesByObf.remove( classMapping.getObfName() ) != null;
+                assert( obfWasRemoved );
+                if( classMapping.getDeobfName() != null )
+                {
+                        boolean deobfWasRemoved = m_classesByDeobf.remove( classMapping.getDeobfName() ) != null;
+                        assert( deobfWasRemoved );
+                }
+        }
+        
         public ClassMapping getClassByObf( ClassEntry entry )
         {
                 return getClassByObf( entry.getName() );