From 8776a8ba38123c822530e5f659c626c8db616217 Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 24 Sep 2014 01:00:54 -0400
Subject: HOW DO I WRITE SO MANY BUGS?!?

---
 src/cuchaz/enigma/analysis/SourceIndex.java | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'src/cuchaz/enigma/analysis/SourceIndex.java')

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();
-- 
cgit v1.2.3