From 959cb5fd4f9586ec3bd265b452fe25fe1db82e3f Mon Sep 17 00:00:00 2001
From: jeff
Date: Tue, 13 Jan 2015 23:25:04 -0500
Subject: source format change don't hate me too much if you were planning a
 big merge. =P

---
 src/cuchaz/enigma/analysis/SourceIndex.java | 144 ++++++++++++----------------
 1 file changed, 59 insertions(+), 85 deletions(-)

(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 0e33de0..b43ab61 100644
--- a/src/cuchaz/enigma/analysis/SourceIndex.java
+++ b/src/cuchaz/enigma/analysis/SourceIndex.java
@@ -25,16 +25,15 @@ import com.strobel.decompiler.languages.java.ast.Identifier;
 
 import cuchaz.enigma.mapping.Entry;
 
-public class SourceIndex
-{
+public class SourceIndex {
+	
 	private String m_source;
 	private TreeMap<Token,EntryReference<Entry,Entry>> m_tokenToReference;
 	private Multimap<EntryReference<Entry,Entry>,Token> m_referenceToTokens;
 	private Map<Entry,Token> m_declarationToToken;
 	private List<Integer> m_lineOffsets;
 	
-	public SourceIndex( String source )
-	{
+	public SourceIndex(String source) {
 		m_source = source;
 		m_tokenToReference = Maps.newTreeMap();
 		m_referenceToTokens = HashMultimap.create();
@@ -42,142 +41,119 @@ public class SourceIndex
 		m_lineOffsets = Lists.newArrayList();
 		
 		// count the lines
-		m_lineOffsets.add( 0 );
-		for( int i=0; i<source.length(); i++ )
-		{
-			if( source.charAt( i ) == '\n' )
-			{
-				m_lineOffsets.add( i + 1 );
+		m_lineOffsets.add(0);
+		for (int i = 0; i < source.length(); i++) {
+			if (source.charAt(i) == '\n') {
+				m_lineOffsets.add(i + 1);
 			}
 		}
 	}
 	
-	public String getSource( )
-	{
+	public String getSource() {
 		return m_source;
 	}
 	
-	public Token getToken( AstNode node )
-	{
+	public Token getToken(AstNode node) {
+		
 		// get the text of the node
 		String name = "";
-		if( node instanceof Identifier )
-		{
+		if (node instanceof Identifier) {
 			name = ((Identifier)node).getName();
 		}
 		
 		// get a token for this node's region
 		Region region = node.getRegion();
-		if( region.getBeginLine() == 0 || region.getEndLine() == 0 )
-		{
+		if (region.getBeginLine() == 0 || region.getEndLine() == 0) {
 			// DEBUG
-			System.err.println( String.format( "WARNING: %s \"%s\" has invalid region: %s", node.getNodeType(), name, region ) );
+			System.err.println(String.format("WARNING: %s \"%s\" has invalid region: %s", node.getNodeType(), name, region));
 			return null;
 		}
 		Token token = new Token(
-			toPos( region.getBeginLine(), region.getBeginColumn() ),
-			toPos( region.getEndLine(), region.getEndColumn() ),
+			toPos(region.getBeginLine(), region.getBeginColumn()),
+			toPos(region.getEndLine(), region.getEndColumn()),
 			m_source
 		);
-		if( token.start == 0 )
-		{
+		if (token.start == 0) {
 			// DEBUG
-			System.err.println( String.format( "WARNING: %s \"%s\" has invalid start: %s", node.getNodeType(), name, region ) );
+			System.err.println(String.format("WARNING: %s \"%s\" has invalid start: %s", node.getNodeType(), name, region));
 			return null;
 		}
 		
 		// DEBUG
-		//System.out.println( String.format( "%s \"%s\" region: %s", node.getNodeType(), name, region ) );
+		// System.out.println( String.format( "%s \"%s\" region: %s", node.getNodeType(), name, region ) );
 		
 		// for tokens representing inner classes, make sure we only get the simple name
-		int pos = name.lastIndexOf( '$' );
-		if( pos >= 0 )
-		{
+		int pos = name.lastIndexOf('$');
+		if (pos >= 0) {
 			token.end -= pos + 1;
 		}
 		
 		return token;
 	}
 	
-	public void addReference( AstNode node, Entry deobfEntry, Entry deobfContext )
-	{
-		Token token = getToken( node );
-		if( token != null )
-		{
-			EntryReference<Entry,Entry> deobfReference = new EntryReference<Entry,Entry>( deobfEntry, token.text, deobfContext );
-			m_tokenToReference.put( token, deobfReference );
-			m_referenceToTokens.put( deobfReference, token );
+	public void addReference(AstNode node, Entry deobfEntry, Entry deobfContext) {
+		Token token = getToken(node);
+		if (token != null) {
+			EntryReference<Entry,Entry> deobfReference = new EntryReference<Entry,Entry>(deobfEntry, token.text, deobfContext);
+			m_tokenToReference.put(token, deobfReference);
+			m_referenceToTokens.put(deobfReference, token);
 		}
 	}
 	
-	public void addDeclaration( AstNode node, Entry deobfEntry )
-	{
-		Token token = getToken( node );
-		if( token != null )
-		{
-			EntryReference<Entry,Entry> reference = new EntryReference<Entry,Entry>( deobfEntry, token.text );
-			m_tokenToReference.put( token, reference );
-			m_referenceToTokens.put( reference, token );
-			m_declarationToToken.put( deobfEntry, token );
+	public void addDeclaration(AstNode node, Entry deobfEntry) {
+		Token token = getToken(node);
+		if (token != null) {
+			EntryReference<Entry,Entry> reference = new EntryReference<Entry,Entry>(deobfEntry, token.text);
+			m_tokenToReference.put(token, reference);
+			m_referenceToTokens.put(reference, token);
+			m_declarationToToken.put(deobfEntry, token);
 		}
 	}
 	
-	public Token getReferenceToken( int pos )
-	{
-		Token token = m_tokenToReference.floorKey( new Token( pos, pos, null ) );
-		if( token != null && token.contains( pos ) )
-		{
+	public Token getReferenceToken(int pos) {
+		Token token = m_tokenToReference.floorKey(new Token(pos, pos, null));
+		if (token != null && token.contains(pos)) {
 			return token;
 		}
 		return null;
 	}
 	
-	public Collection<Token> getReferenceTokens( EntryReference<Entry,Entry> deobfReference )
-	{
-		return m_referenceToTokens.get( deobfReference );
+	public Collection<Token> getReferenceTokens(EntryReference<Entry,Entry> deobfReference) {
+		return m_referenceToTokens.get(deobfReference);
 	}
 	
-	public EntryReference<Entry,Entry> getDeobfReference( Token token )
-	{
-		if( token == null )
-		{
+	public EntryReference<Entry,Entry> getDeobfReference(Token token) {
+		if (token == null) {
 			return null;
 		}
-		return m_tokenToReference.get( token );
+		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 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( )
-	{
+	public Iterable<Token> referenceTokens() {
 		return m_tokenToReference.keySet();
 	}
 	
-	public Iterable<Token> declarationTokens( )
-	{
+	public Iterable<Token> declarationTokens() {
 		return m_declarationToToken.values();
 	}
 	
-	public Token getDeclarationToken( Entry deobfEntry )
-	{
-		return m_declarationToToken.get( deobfEntry );
+	public Token getDeclarationToken(Entry deobfEntry) {
+		return m_declarationToToken.get(deobfEntry);
 	}
 	
-	public int getLineNumber( int pos )
-	{
+	public int getLineNumber(int pos) {
 		// line number is 1-based
 		int line = 0;
-		for( Integer offset : m_lineOffsets )
-		{
-			if( offset > pos )
-			{
+		for (Integer offset : m_lineOffsets) {
+			if (offset > pos) {
 				break;
 			}
 			line++;
@@ -185,15 +161,13 @@ public class SourceIndex
 		return line;
 	}
 	
-	public int getColumnNumber( int pos )
-	{
+	public int getColumnNumber(int pos) {
 		// column number is 1-based
-		return pos - m_lineOffsets.get( getLineNumber( pos ) - 1 ) + 1;
+		return pos - m_lineOffsets.get(getLineNumber(pos) - 1) + 1;
 	}
-
-	private int toPos( int line, int col )
-	{
+	
+	private int toPos(int line, int col) {
 		// line and col are 1-based
-		return m_lineOffsets.get( line - 1 ) + col - 1;
+		return m_lineOffsets.get(line - 1) + col - 1;
 	}
 }
-- 
cgit v1.2.3