1 files changed, 9 insertions, 0 deletions

diff --git a/src/cuchaz/enigma/analysis/SourceIndex.java b/src/cuchaz/enigma/analysis/SourceIndex.java
index 49451b9..a5d1460 100644
--- a/src/cuchaz/enigma/analysis/SourceIndex.java
+++ b/src/cuchaz/enigma/analysis/SourceIndex.java
@@ -126,6 +126,15 @@ public class SourceIndex
                 return m_tokenToReference.get( token );
         }
         
+        public void replaceDeobfReference( Token token, EntryReference<Entry,Entry> newDeobfReference )
+        {
+                EntryReference<Entry,Entry> oldDeobfReference = m_tokenToReference.get( token );
+                m_tokenToReference.put( token, newDeobfReference );
+                Collection<Token> tokens = m_referenceToTokens.get( oldDeobfReference );
+                m_referenceToTokens.removeAll( oldDeobfReference );
+                m_referenceToTokens.putAll( newDeobfReference, tokens );
+        }
+        
         public Iterable<Token> referenceTokens( )
         {
                 return m_tokenToReference.keySet();