From 295eceece371b516e771de93b6127bf728999483 Mon Sep 17 00:00:00 2001
From: jeff
Date: Sat, 26 Jul 2014 19:24:00 -0400
Subject: initial commit

so far source analysis is working. =)
---
 src/cuchaz/enigma/analysis/SourcedAst.java | 111 +++++++++++++++++++++++++++++
 1 file changed, 111 insertions(+)
 create mode 100644 src/cuchaz/enigma/analysis/SourcedAst.java

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

diff --git a/src/cuchaz/enigma/analysis/SourcedAst.java b/src/cuchaz/enigma/analysis/SourcedAst.java
new file mode 100644
index 0000000..04c6f03
--- /dev/null
+++ b/src/cuchaz/enigma/analysis/SourcedAst.java
@@ -0,0 +1,111 @@
+/*******************************************************************************
+ * Copyright (c) 2014 Jeff Martin.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the GNU Public License v3.0
+ * which accompanies this distribution, and is available at
+ * http://www.gnu.org/licenses/gpl.html
+ * 
+ * Contributors:
+ *     Jeff Martin - initial API and implementation
+ ******************************************************************************/
+package cuchaz.enigma.analysis;
+
+import java.io.IOException;
+import java.util.HashMap;
+
+import com.google.common.collect.Maps;
+import com.sun.source.tree.CompilationUnitTree;
+import com.sun.source.tree.ImportTree;
+import com.sun.source.tree.Tree;
+import com.sun.source.util.SourcePositions;
+import com.sun.source.util.Trees;
+
+public class SourcedAst
+{
+	private CompilationUnitTree m_tree;
+	private Trees m_trees;
+	private SourcePositions m_positions;
+	private HashMap<String,String> m_classNameIndex;
+	
+	public SourcedAst( CompilationUnitTree tree, Trees trees )
+	{
+		m_tree = tree;
+		m_trees = trees;
+		m_positions = m_trees.getSourcePositions();
+		m_classNameIndex = Maps.newHashMap();
+		
+		// index all the class names
+		for( ImportTree importTree : m_tree.getImports() )
+		{
+			// ignore static imports for now
+			if( importTree.isStatic() )
+			{
+				continue;
+			}
+			
+			// get the full and simple class names
+			String fullName = importTree.getQualifiedIdentifier().toString();
+			String simpleName = fullName;
+			String[] parts = fullName.split( "\\." );
+			if( parts.length > 0 )
+			{
+				simpleName = parts[parts.length - 1];
+			}
+			
+			m_classNameIndex.put( simpleName, fullName );
+		}
+	}
+
+	public int getStart( Tree node )
+	{
+		return (int)m_positions.getStartPosition( m_tree, node );
+	}
+
+	public int getEnd( Tree node )
+	{
+		return (int)m_positions.getEndPosition( m_tree, node );
+	}
+	
+	public int getLine( Tree node )
+	{
+		return getLine( getStart( node ) );
+	}
+	
+	public int getLine( int pos )
+	{
+		return (int)m_tree.getLineMap().getLineNumber( pos );
+	}
+	
+	public CharSequence getSource( )
+	{
+		try
+		{
+			return m_tree.getSourceFile().getCharContent( true );
+		}
+		catch( IOException ex )
+		{
+			throw new Error( ex );
+		}
+	}
+	
+	public CharSequence getSource( Tree node )
+	{
+		return getSource().subSequence( getStart( node ), getEnd( node ) );
+	}
+
+	public void visit( TreeVisitor visitor )
+	{
+		m_tree.accept( visitor, this );
+	}
+	
+	public String getFullClassName( String simpleClassName )
+	{
+		String fullClassName = m_classNameIndex.get( simpleClassName );
+		if( fullClassName == null )
+		{
+			// no mapping was found, the name is probably already fully-qualified
+			fullClassName = simpleClassName;
+		}
+		return fullClassName;
+	}
+}
-- 
cgit v1.2.3