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

1 files changed, 18 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/gui/GuiController.java b/src/cuchaz/enigma/gui/GuiController.java
index 098e065..3adaf91 100644
--- a/src/cuchaz/enigma/gui/GuiController.java
+++ b/src/cuchaz/enigma/gui/GuiController.java
@@ -249,6 +249,24 @@ public class GuiController
                 refreshCurrentClass( obfReference );
         }
         
+        public void removeMapping( EntryReference<Entry,Entry> deobfReference )
+        {
+                EntryReference<Entry,Entry> obfReference = m_deobfuscator.obfuscateReference( deobfReference );
+                m_deobfuscator.removeMapping( obfReference.entry );
+                m_isDirty = true;
+                refreshClasses();
+                refreshCurrentClass( obfReference );
+        }
+        
+        public void markAsDeobfuscated( EntryReference<Entry,Entry> deobfReference )
+        {
+                EntryReference<Entry,Entry> obfReference = m_deobfuscator.obfuscateReference( deobfReference );
+                m_deobfuscator.markAsDeobfuscated( obfReference.entry );
+                m_isDirty = true;
+                refreshClasses();
+                refreshCurrentClass( obfReference );
+        }
+        
         public void openDeclaration( Entry deobfEntry )
         {
                 if( deobfEntry == null )