From 430df87ba5d855ca29bc53a5765a2862d2209098 Mon Sep 17 00:00:00 2001
From: jeff
Date: Tue, 10 Mar 2015 00:55:03 -0400
Subject: tweaks and improvements to field matching gui

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

(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 b3fb751..8f751ef 100644
--- a/src/cuchaz/enigma/analysis/SourceIndex.java
+++ b/src/cuchaz/enigma/analysis/SourceIndex.java
@@ -32,9 +32,15 @@ public class SourceIndex {
 	private Multimap<EntryReference<Entry,Entry>,Token> m_referenceToTokens;
 	private Map<Entry,Token> m_declarationToToken;
 	private List<Integer> m_lineOffsets;
+	private boolean m_ignoreBadTokens;
 	
 	public SourceIndex(String source) {
+		this(source, true);
+	}
+	
+	public SourceIndex(String source, boolean ignoreBadTokens) {
 		m_source = source;
+		m_ignoreBadTokens = ignoreBadTokens;
 		m_tokenToReference = Maps.newTreeMap();
 		m_referenceToTokens = HashMultimap.create();
 		m_declarationToToken = Maps.newHashMap();
@@ -83,7 +89,7 @@ public class SourceIndex {
 		// System.out.println( String.format( "%s \"%s\" region: %s", node.getNodeType(), name, region ) );
 		
 		// if the token has a $ in it, something's wrong. Ignore this token
-		if (name.lastIndexOf('$') >= 0) {
+		if (name.lastIndexOf('$') >= 0 && m_ignoreBadTokens) {
 			// DEBUG
 			System.err.println(String.format("WARNING: %s \"%s\" is probably a bad token. It was ignored", node.getNodeType(), name));
 			return null;
-- 
cgit v1.2.3