1 files changed, 20 insertions, 26 deletions

diff --git a/test/cuchaz/enigma/TokenChecker.java b/test/cuchaz/enigma/TokenChecker.java
index c0852f3..524c5ec 100644
--- a/test/cuchaz/enigma/TokenChecker.java
+++ b/test/cuchaz/enigma/TokenChecker.java
@@ -23,47 +23,41 @@ import cuchaz.enigma.analysis.SourceIndex;
 import cuchaz.enigma.analysis.Token;
 import cuchaz.enigma.mapping.Entry;
 
-public class TokenChecker
-{
+public class TokenChecker {
+        
         private Deobfuscator m_deobfuscator;
-                
-        protected TokenChecker( File jarFile )
-        throws IOException
-        {
-                m_deobfuscator = new Deobfuscator( jarFile );
+        
+        protected TokenChecker(File jarFile) throws IOException {
+                m_deobfuscator = new Deobfuscator(jarFile);
         }
         
-        protected String getDeclarationToken( Entry entry )
-        {
+        protected String getDeclarationToken(Entry entry) {
                 // decompile the class
-                CompilationUnit tree = m_deobfuscator.getSourceTree( entry.getClassName() );
+                CompilationUnit tree = m_deobfuscator.getSourceTree(entry.getClassName());
                 // DEBUG
-                //tree.acceptVisitor( new TreeDumpVisitor( new File( "tree." + entry.getClassName().replace( '/', '.' ) + ".txt" ) ), null );
-                String source = m_deobfuscator.getSource( tree );
-                SourceIndex index = m_deobfuscator.getSourceIndex( tree, source );
+                // tree.acceptVisitor( new TreeDumpVisitor( new File( "tree." + entry.getClassName().replace( '/', '.' ) + ".txt" ) ), null );
+                String source = m_deobfuscator.getSource(tree);
+                SourceIndex index = m_deobfuscator.getSourceIndex(tree, source);
                 
                 // get the token value
-                Token token = index.getDeclarationToken( entry );
-                if( token == null )
-                {
+                Token token = index.getDeclarationToken(entry);
+                if (token == null) {
                         return null;
                 }
-                return source.substring( token.start, token.end );
+                return source.substring(token.start, token.end);
         }
         
-        @SuppressWarnings( "unchecked" )
-        protected Collection<String> getReferenceTokens( EntryReference<? extends Entry,? extends Entry> reference )
-        {
+        @SuppressWarnings("unchecked")
+        protected Collection<String> getReferenceTokens(EntryReference<? extends Entry,? extends Entry> reference) {
                 // decompile the class
-                CompilationUnit tree = m_deobfuscator.getSourceTree( reference.context.getClassName() );
-                String source = m_deobfuscator.getSource( tree );
-                SourceIndex index = m_deobfuscator.getSourceIndex( tree, source );
+                CompilationUnit tree = m_deobfuscator.getSourceTree(reference.context.getClassName());
+                String source = m_deobfuscator.getSource(tree);
+                SourceIndex index = m_deobfuscator.getSourceIndex(tree, source);
                 
                 // get the token values
                 List<String> values = Lists.newArrayList();
-                for( Token token : index.getReferenceTokens( (EntryReference<Entry,Entry>)reference ) )
-                {
-                        values.add( source.substring( token.start, token.end ) );
+                for (Token token : index.getReferenceTokens((EntryReference<Entry,Entry>)reference)) {
+                        values.add(source.substring(token.start, token.end));
                 }
                 return values;
         }