From e7febe4549c9fcdf1e82239959b3c6a83fad8934 Mon Sep 17 00:00:00 2001
From: jeff
Date: Sun, 10 Aug 2014 19:29:43 -0400
Subject: added go to entry feature

---
 src/cuchaz/enigma/analysis/SourceIndex.java | 11 ++++++++++-
 1 file changed, 10 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 7981f87..ad94cf0 100644
--- a/src/cuchaz/enigma/analysis/SourceIndex.java
+++ b/src/cuchaz/enigma/analysis/SourceIndex.java
@@ -25,12 +25,14 @@ public class SourceIndex
 {
 	private String m_source;
 	private TreeMap<Token,Entry> m_tokens;
+	private Map<Entry,Token> m_declarations;
 	private List<Integer> m_lineOffsets;
 	
 	public SourceIndex( String source )
 	{
 		m_source = source;
 		m_tokens = Maps.newTreeMap();
+		m_declarations = Maps.newHashMap();
 		m_lineOffsets = Lists.newArrayList();
 		
 		// count the lines
@@ -82,9 +84,11 @@ public class SourceIndex
 		m_tokens.put( getToken( node ), entry );
 	}
 	
-	public void add( Token token, Entry entry )
+	public void addDeclaration( AstNode node, Entry entry )
 	{
+		Token token = getToken( node );
 		m_tokens.put( token, entry );
+		m_declarations.put( entry, token );
 	}
 	
 	public Token getToken( int pos )
@@ -116,6 +120,11 @@ public class SourceIndex
 		return m_tokens.keySet();
 	}
 	
+	public Token getDeclarationToken( Entry entry )
+	{
+		return m_declarations.get( entry );
+	}
+	
 	private int toPos( int line, int col )
 	{
 		// line and col are 1-based
-- 
cgit v1.2.3