starting on field matching gui

5 files changed, 338 insertions, 162 deletions

diff --git a/src/cuchaz/enigma/gui/ClassMatchingGui.java b/src/cuchaz/enigma/gui/ClassMatchingGui.java
index ff7cda9..b674451 100644
--- a/src/cuchaz/enigma/gui/ClassMatchingGui.java
+++ b/src/cuchaz/enigma/gui/ClassMatchingGui.java
@@ -15,7 +15,6 @@ import javax.swing.BoxLayout;
 import javax.swing.ButtonGroup;
 import javax.swing.JButton;
 import javax.swing.JCheckBox;
-import javax.swing.JEditorPane;
 import javax.swing.JFrame;
 import javax.swing.JLabel;
 import javax.swing.JPanel;
@@ -28,22 +27,18 @@ import javax.swing.WindowConstants;
 import com.beust.jcommander.internal.Lists;
 import com.beust.jcommander.internal.Maps;
 import com.google.common.collect.BiMap;
-import com.strobel.decompiler.languages.java.ast.CompilationUnit;
 
 import cuchaz.enigma.Constants;
 import cuchaz.enigma.Deobfuscator;
-import cuchaz.enigma.analysis.SourceIndex;
-import cuchaz.enigma.analysis.Token;
 import cuchaz.enigma.convert.ClassIdentifier;
 import cuchaz.enigma.convert.ClassIdentity;
 import cuchaz.enigma.convert.ClassMatch;
+import cuchaz.enigma.convert.ClassMatches;
 import cuchaz.enigma.convert.ClassMatching;
 import cuchaz.enigma.convert.ClassNamer;
 import cuchaz.enigma.convert.MappingsConverter;
-import cuchaz.enigma.convert.Matches;
 import cuchaz.enigma.gui.ClassSelector.ClassSelectionListener;
 import cuchaz.enigma.mapping.ClassEntry;
-import cuchaz.enigma.mapping.Entry;
 import cuchaz.enigma.mapping.Mappings;
 import cuchaz.enigma.mapping.MappingsChecker;
 import de.sciss.syntaxpane.DefaultSyntaxKit;
@@ -55,21 +50,21 @@ public class ClassMatchingGui {
                 Matched {
                         
                         @Override
-                        public Collection<ClassEntry> getSourceClasses(Matches matches) {
+                        public Collection<ClassEntry> getSourceClasses(ClassMatches matches) {
                                 return matches.getUniqueMatches().keySet();
                         }
                 },
                 Unmatched {
                         
                         @Override
-                        public Collection<ClassEntry> getSourceClasses(Matches matches) {
+                        public Collection<ClassEntry> getSourceClasses(ClassMatches matches) {
                                 return matches.getUnmatchedSourceClasses();
                         }
                 },
                 Ambiguous {
                         
                         @Override
-                        public Collection<ClassEntry> getSourceClasses(Matches matches) {
+                        public Collection<ClassEntry> getSourceClasses(ClassMatches matches) {
                                 return matches.getAmbiguouslyMatchedSourceClasses();
                         }
                 };
@@ -82,7 +77,7 @@ public class ClassMatchingGui {
                         return button;
                 }
                 
-                public abstract Collection<ClassEntry> getSourceClasses(Matches matches);
+                public abstract Collection<ClassEntry> getSourceClasses(ClassMatches matches);
                 
                 public static SourceType getDefault() {
                         return values()[0];
@@ -90,23 +85,22 @@ public class ClassMatchingGui {
         }
         
         public static interface SaveListener {
-                public void save(Matches matches);
+                public void save(ClassMatches matches);
         }
         
         // controls
         private JFrame m_frame;
         private ClassSelector m_sourceClasses;
         private ClassSelector m_destClasses;
-        private JEditorPane m_sourceReader;
-        private JEditorPane m_destReader;
+        private CodeReader m_sourceReader;
+        private CodeReader m_destReader;
         private JLabel m_sourceClassLabel;
         private JLabel m_destClassLabel;
         private JButton m_matchButton;
         private Map<SourceType,JRadioButton> m_sourceTypeButtons;
         private JCheckBox m_advanceCheck;
-        private SelectionHighlightPainter m_selectionHighlightPainter;
         
-        private Matches m_matches;
+        private ClassMatches m_classMatches;
         private Deobfuscator m_sourceDeobfuscator;
         private Deobfuscator m_destDeobfuscator;
         private ClassEntry m_sourceClass;
@@ -114,14 +108,14 @@ public class ClassMatchingGui {
         private SourceType m_sourceType;
         private SaveListener m_saveListener;
 
-        public ClassMatchingGui(Matches matches, Deobfuscator sourceDeobfuscator, Deobfuscator destDeobfuscator) {
+        public ClassMatchingGui(ClassMatches matches, Deobfuscator sourceDeobfuscator, Deobfuscator destDeobfuscator) {
                 
-                m_matches = matches;
+                m_classMatches = matches;
                 m_sourceDeobfuscator = sourceDeobfuscator;
                 m_destDeobfuscator = destDeobfuscator;
                 
                 // init frame
-                m_frame = new JFrame(Constants.Name);
+                m_frame = new JFrame(Constants.Name + " - Class Matcher");
                 final Container pane = m_frame.getContentPane();
                 pane.setLayout(new BorderLayout());
                 
@@ -188,8 +182,8 @@ public class ClassMatchingGui {
                 
                 // init source panels
                 DefaultSyntaxKit.initKit();
-                m_sourceReader = makeReader();
-                m_destReader = makeReader();
+                m_sourceReader = new CodeReader();
+                m_destReader = new CodeReader();
                 
                 // init all the splits
                 JSplitPane splitLeft = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, true, sourcePanel, new JScrollPane(m_sourceReader));
@@ -238,8 +232,6 @@ public class ClassMatchingGui {
                 m_frame.setVisible(true);
                 m_frame.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
                 
-                m_selectionHighlightPainter = new SelectionHighlightPainter();
-                
                 // init state
                 updateDestMappings();
                 setSourceType(SourceType.getDefault());
@@ -247,19 +239,6 @@ public class ClassMatchingGui {
                 m_saveListener = null;
         }
         
-        private JEditorPane makeReader() {
-                
-                JEditorPane reader = new JEditorPane();
-                reader.setEditable(false);
-                reader.setContentType("text/java");
-                
-                // turn off token highlighting (it's wrong most of the time anyway...)
-                DefaultSyntaxKit kit = (DefaultSyntaxKit)reader.getEditorKit();
-                kit.toggleComponent(reader, "de.sciss.syntaxpane.components.TokenMarker");
-                
-                return reader;
-        }
-
         public void setSaveListener(SaveListener val) {
                 m_saveListener = val;
         }
@@ -267,7 +246,7 @@ public class ClassMatchingGui {
         private void updateDestMappings() {
                 
                 Mappings newMappings = MappingsConverter.newMappings(
-                        m_matches,
+                        m_classMatches,
                         m_sourceDeobfuscator.getMappings(),
                         m_sourceDeobfuscator,
                         m_destDeobfuscator
@@ -294,13 +273,13 @@ public class ClassMatchingGui {
                 
                 // show the source classes
                 m_sourceType = val;
-                m_sourceClasses.setClasses(deobfuscateClasses(m_sourceType.getSourceClasses(m_matches), m_sourceDeobfuscator));
+                m_sourceClasses.setClasses(deobfuscateClasses(m_sourceType.getSourceClasses(m_classMatches), m_sourceDeobfuscator));
                 
                 // update counts
                 for (SourceType sourceType : SourceType.values()) {
                         m_sourceTypeButtons.get(sourceType).setText(String.format("%s (%d)",
                                 sourceType.name(),
-                                sourceType.getSourceClasses(m_matches).size()
+                                sourceType.getSourceClasses(m_classMatches).size()
                         ));
                 }
         }
@@ -345,7 +324,7 @@ public class ClassMatchingGui {
                 if (m_sourceClass != null) {
                         
                         // show the dest class(es)
-                        ClassMatch match = m_matches.getMatchBySource(m_sourceDeobfuscator.obfuscateEntry(m_sourceClass));
+                        ClassMatch match = m_classMatches.getMatchBySource(m_sourceDeobfuscator.obfuscateEntry(m_sourceClass));
                         assert(match != null);
                         if (match.destClasses.isEmpty()) {
                                 
@@ -376,7 +355,12 @@ public class ClassMatchingGui {
                 }
                 
                 setDestClass(null);
-                decompileClass(m_sourceClass, m_sourceDeobfuscator, m_sourceReader);
+                m_sourceReader.decompileClass(m_sourceClass, m_sourceDeobfuscator, new Runnable() {
+                        @Override
+                        public void run() {
+                                m_sourceReader.navigateToClassDeclaration(m_sourceClass);
+                        }
+                });
                 
                 updateMatchButton();
         }
@@ -386,7 +370,7 @@ public class ClassMatchingGui {
                 ClassEntry obfSourceClass = m_sourceDeobfuscator.obfuscateEntry(sourceClass);
                 
                 // set up identifiers
-                ClassNamer namer = new ClassNamer(m_matches.getUniqueMatches());
+                ClassNamer namer = new ClassNamer(m_classMatches.getUniqueMatches());
                 ClassIdentifier sourceIdentifier = new ClassIdentifier(
                         m_sourceDeobfuscator.getJar(), m_sourceDeobfuscator.getJarIndex(),
                         namer.getSourceNamer(), true
@@ -401,7 +385,7 @@ public class ClassMatchingGui {
                         // rank all the unmatched dest classes against the source class
                         ClassIdentity sourceIdentity = sourceIdentifier.identify(obfSourceClass);
                         List<ClassEntry> scoredDestClasses = Lists.newArrayList();
-                        for (ClassEntry unmatchedDestClass : m_matches.getUnmatchedDestClasses()) {
+                        for (ClassEntry unmatchedDestClass : m_classMatches.getUnmatchedDestClasses()) {
                                 ClassIdentity destIdentity = destIdentifier.identify(unmatchedDestClass);
                                 float score = 100.0f*(sourceIdentity.getMatchScore(destIdentity) + destIdentity.getMatchScore(sourceIdentity))
                                         /(sourceIdentity.getMaxMatchScore() + destIdentity.getMaxMatchScore());
@@ -420,59 +404,14 @@ public class ClassMatchingGui {
                 m_destClass = classEntry;
                 m_destClassLabel.setText(m_destClass != null ? m_destClass.getName() : "");
                 
-                decompileClass(m_destClass, m_destDeobfuscator, m_destReader);
-                
-                updateMatchButton();
-        }
-
-        protected void decompileClass(final ClassEntry classEntry, final Deobfuscator deobfuscator, final JEditorPane reader) {
-                
-                if (classEntry == null) {
-                        reader.setText(null);
-                        return;
-                }
-                
-                reader.setText("(decompiling...)");
-
-                // run in a separate thread to keep ui responsive
-                new Thread() {
+                m_destReader.decompileClass(m_destClass, m_destDeobfuscator, new Runnable() {
                         @Override
                         public void run() {
-                                
-                                // get the outermost class
-                                ClassEntry outermostClassEntry = classEntry;
-                                while (outermostClassEntry.isInnerClass()) {
-                                        outermostClassEntry = outermostClassEntry.getOuterClassEntry();
-                                }
-                                
-                                // decompile it
-                                CompilationUnit sourceTree = deobfuscator.getSourceTree(outermostClassEntry.getName());
-                                String source = deobfuscator.getSource(sourceTree);
-                                reader.setText(source);
-                                SourceIndex sourceIndex = deobfuscator.getSourceIndex(sourceTree, source);
-                                
-                                // navigate to the class declaration
-                                Token token = sourceIndex.getDeclarationToken(classEntry);
-                                if (token == null) {
-                                        // couldn't find the class declaration token, might be an anonymous class
-                                        // look for any declaration in that class instead
-                                        for (Entry entry : sourceIndex.declarations()) {
-                                                if (entry.getClassEntry().equals(classEntry)) {
-                                                        token = sourceIndex.getDeclarationToken(entry);
-                                                        break;
-                                                }
-                                        }
-                                }
-                                
-                                if (token != null) {
-                                        GuiTricks.navigateToToken(reader, token, m_selectionHighlightPainter);
-                                } else {
-                                        // couldn't find anything =(
-                                        System.out.println("Unable to find declaration in source for " + classEntry);
-                                }
-                                
+                                m_destReader.navigateToClassDeclaration(m_destClass);
                         }
-                }.start();
+                });
+                
+                updateMatchButton();
         }
         
         private void updateMatchButton() {
@@ -480,7 +419,7 @@ public class ClassMatchingGui {
                 ClassEntry obfSource = m_sourceDeobfuscator.obfuscateEntry(m_sourceClass);
                 ClassEntry obfDest = m_destDeobfuscator.obfuscateEntry(m_destClass);
                 
-                BiMap<ClassEntry,ClassEntry> uniqueMatches = m_matches.getUniqueMatches();
+                BiMap<ClassEntry,ClassEntry> uniqueMatches = m_classMatches.getUniqueMatches();
                 boolean twoSelected = m_sourceClass != null && m_destClass != null;
                 boolean isMatched = uniqueMatches.containsKey(obfSource) && uniqueMatches.containsValue(obfDest);
                 boolean canMatch = !uniqueMatches.containsKey(obfSource) && ! uniqueMatches.containsValue(obfDest);
@@ -529,11 +468,11 @@ public class ClassMatchingGui {
                 ClassEntry obfDest = m_destDeobfuscator.obfuscateEntry(m_destClass);
                 
                 // remove the classes from their match
-                m_matches.removeSource(obfSource);
-                m_matches.removeDest(obfDest);
+                m_classMatches.removeSource(obfSource);
+                m_classMatches.removeDest(obfDest);
                 
                 // add them as matched classes
-                m_matches.add(new ClassMatch(obfSource, obfDest));
+                m_classMatches.add(new ClassMatch(obfSource, obfDest));
                 
                 ClassEntry nextClass = null;
                 if (m_advanceCheck.isSelected()) {
@@ -554,8 +493,8 @@ public class ClassMatchingGui {
                 ClassEntry obfSource = m_sourceDeobfuscator.obfuscateEntry(m_sourceClass);
                 
                 // remove the source to break the match, then add the source back as unmatched
-                m_matches.removeSource(obfSource);
-                m_matches.add(new ClassMatch(obfSource, null));
+                m_classMatches.removeSource(obfSource);
+                m_classMatches.add(new ClassMatch(obfSource, null));
                 
                 save();
                 updateMatches();
@@ -577,7 +516,7 @@ public class ClassMatchingGui {
         
         private void save() {
                 if (m_saveListener != null) {
-                        m_saveListener.save(m_matches);
+                        m_saveListener.save(m_classMatches);
                 }
         }
         
@@ -589,15 +528,15 @@ public class ClassMatchingGui {
                 ClassMatching matching = MappingsConverter.computeMatching(
                         m_sourceDeobfuscator.getJar(), m_sourceDeobfuscator.getJarIndex(),
                         m_destDeobfuscator.getJar(), m_destDeobfuscator.getJarIndex(),
-                        m_matches.getUniqueMatches()
+                        m_classMatches.getUniqueMatches()
                 );
-                Matches newMatches = new Matches(matching.matches());
+                ClassMatches newMatches = new ClassMatches(matching.matches());
                 System.out.println(String.format("Automatch found %d new matches",
-                        newMatches.getUniqueMatches().size() - m_matches.getUniqueMatches().size()
+                        newMatches.getUniqueMatches().size() - m_classMatches.getUniqueMatches().size()
                 ));
                 
                 // update the current matches
-                m_matches = newMatches;
+                m_classMatches = newMatches;
                 save();
                 updateMatches();
         }
diff --git a/src/cuchaz/enigma/gui/CodeReader.java b/src/cuchaz/enigma/gui/CodeReader.java
new file mode 100644
index 0000000..05feb59
--- /dev/null
+++ b/src/cuchaz/enigma/gui/CodeReader.java
@@ -0,0 +1,164 @@
+package cuchaz.enigma.gui;
+
+import java.awt.Rectangle;
+import java.awt.event.ActionEvent;
+import java.awt.event.ActionListener;
+
+import javax.swing.JEditorPane;
+import javax.swing.SwingUtilities;
+import javax.swing.Timer;
+import javax.swing.text.BadLocationException;
+import javax.swing.text.Highlighter.HighlightPainter;
+
+import com.strobel.decompiler.languages.java.ast.CompilationUnit;
+
+import cuchaz.enigma.Deobfuscator;
+import cuchaz.enigma.analysis.SourceIndex;
+import cuchaz.enigma.analysis.Token;
+import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.Entry;
+import de.sciss.syntaxpane.DefaultSyntaxKit;
+
+
+public class CodeReader extends JEditorPane {
+        
+        private static final long serialVersionUID = 3673180950485748810L;
+        
+        private static final Object m_lock = new Object();
+        
+        private SelectionHighlightPainter m_highlightPainter;
+        private SourceIndex m_sourceIndex;
+        
+        public CodeReader() {
+                
+                setEditable(false);
+                setContentType("text/java");
+                
+                // turn off token highlighting (it's wrong most of the time anyway...)
+                DefaultSyntaxKit kit = (DefaultSyntaxKit)getEditorKit();
+                kit.toggleComponent(this, "de.sciss.syntaxpane.components.TokenMarker");
+                
+                m_highlightPainter = new SelectionHighlightPainter();
+                m_sourceIndex = null;
+        }
+        
+        public void setCode(String code) {
+                // sadly, the java lexer is not thread safe, so we have to serialize all these calls
+                synchronized (m_lock) {
+                        setText(code);
+                }
+        }
+        
+        public void decompileClass(ClassEntry classEntry, Deobfuscator deobfuscator) {
+                decompileClass(classEntry, deobfuscator, null);
+        }
+        
+        public void decompileClass(final ClassEntry classEntry, final Deobfuscator deobfuscator, final Runnable callback) {
+                
+                if (classEntry == null) {
+                        setCode(null);
+                        return;
+                }
+                
+                setCode("(decompiling...)");
+
+                // run decompilation in a separate thread to keep ui responsive
+                new Thread() {
+                        @Override
+                        public void run() {
+                                
+                                // get the outermost class
+                                ClassEntry outermostClassEntry = classEntry;
+                                while (outermostClassEntry.isInnerClass()) {
+                                        outermostClassEntry = outermostClassEntry.getOuterClassEntry();
+                                }
+                                
+                                // decompile it
+                                CompilationUnit sourceTree = deobfuscator.getSourceTree(outermostClassEntry.getName());
+                                String source = deobfuscator.getSource(sourceTree);
+                                setCode(source);
+                                m_sourceIndex = deobfuscator.getSourceIndex(sourceTree, source);
+                                
+                                if (callback != null) {
+                                        callback.run();
+                                }
+                        }
+                }.start();
+        }
+        
+        public void navigateToClassDeclaration(ClassEntry classEntry) {
+
+                // navigate to the class declaration
+                Token token = m_sourceIndex.getDeclarationToken(classEntry);
+                if (token == null) {
+                        // couldn't find the class declaration token, might be an anonymous class
+                        // look for any declaration in that class instead
+                        for (Entry entry : m_sourceIndex.declarations()) {
+                                if (entry.getClassEntry().equals(classEntry)) {
+                                        token = m_sourceIndex.getDeclarationToken(entry);
+                                        break;
+                                }
+                        }
+                }
+                
+                if (token != null) {
+                        navigateToToken(token);
+                } else {
+                        // couldn't find anything =(
+                        System.out.println("Unable to find declaration in source for " + classEntry);
+                }
+        }
+        
+        public void navigateToToken(final Token token) {
+                navigateToToken(this, token, m_highlightPainter);
+        }
+        
+        // HACKHACK: someday we can update the main GUI to use this code reader
+        public static void navigateToToken(final JEditorPane editor, final Token token, final HighlightPainter highlightPainter) {
+                
+                // set the caret position to the token
+                editor.setCaretPosition(token.start);
+                editor.grabFocus();
+                
+                try {
+                        // make sure the token is visible in the scroll window
+                        Rectangle start = editor.modelToView(token.start);
+                        Rectangle end = editor.modelToView(token.end);
+                        final Rectangle show = start.union(end);
+                        show.grow(start.width * 10, start.height * 6);
+                        SwingUtilities.invokeLater(new Runnable() {
+                                @Override
+                                public void run() {
+                                        editor.scrollRectToVisible(show);
+                                }
+                        });
+                } catch (BadLocationException ex) {
+                        throw new Error(ex);
+                }
+                
+                // highlight the token momentarily
+                final Timer timer = new Timer(200, new ActionListener() {
+                        private int m_counter = 0;
+                        private Object m_highlight = null;
+                        
+                        @Override
+                        public void actionPerformed(ActionEvent event) {
+                                if (m_counter % 2 == 0) {
+                                        try {
+                                                m_highlight = editor.getHighlighter().addHighlight(token.start, token.end, highlightPainter);
+                                        } catch (BadLocationException ex) {
+                                                // don't care
+                                        }
+                                } else if (m_highlight != null) {
+                                        editor.getHighlighter().removeHighlight(m_highlight);
+                                }
+                                
+                                if (m_counter++ > 6) {
+                                        Timer timer = (Timer)event.getSource();
+                                        timer.stop();
+                                }
+                        }
+                });
+                timer.start();
+        }
+}
diff --git a/src/cuchaz/enigma/gui/FieldMatchingGui.java b/src/cuchaz/enigma/gui/FieldMatchingGui.java
new file mode 100644
index 0000000..de9ba14
--- /dev/null
+++ b/src/cuchaz/enigma/gui/FieldMatchingGui.java
@@ -0,0 +1,131 @@
+package cuchaz.enigma.gui;
+
+import java.awt.BorderLayout;
+import java.awt.Container;
+import java.awt.Dimension;
+import java.awt.FlowLayout;
+import java.util.Map;
+
+import javax.swing.BoxLayout;
+import javax.swing.JFrame;
+import javax.swing.JLabel;
+import javax.swing.JPanel;
+import javax.swing.JScrollPane;
+import javax.swing.JSplitPane;
+import javax.swing.WindowConstants;
+
+import cuchaz.enigma.Constants;
+import cuchaz.enigma.Deobfuscator;
+import cuchaz.enigma.convert.ClassMatches;
+import cuchaz.enigma.convert.FieldMatches;
+import cuchaz.enigma.gui.ClassSelector.ClassSelectionListener;
+import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.FieldEntry;
+import cuchaz.enigma.mapping.FieldMapping;
+import de.sciss.syntaxpane.DefaultSyntaxKit;
+
+
+public class FieldMatchingGui {
+        
+        public static interface SaveListener {
+                public void save(FieldMatches matches);
+        }
+        
+        // controls
+        private JFrame m_frame;
+        private ClassSelector m_sourceClasses;
+        private CodeReader m_sourceReader;
+        private CodeReader m_destReader;
+
+        private ClassMatches m_classMatches;
+        private FieldMatches m_fieldMatches;
+        private Map<FieldEntry,FieldMapping> m_droppedFieldMappings;
+        private Deobfuscator m_sourceDeobfuscator;
+        private Deobfuscator m_destDeobfuscator;
+        private SaveListener m_saveListener;
+
+        public FieldMatchingGui(ClassMatches classMatches, FieldMatches fieldMatches, Map<FieldEntry,FieldMapping> droppedFieldMappings, Deobfuscator sourceDeobfuscator, Deobfuscator destDeobfuscator) {
+                
+                m_classMatches = classMatches;
+                m_fieldMatches = fieldMatches;
+                m_droppedFieldMappings = droppedFieldMappings;
+                m_sourceDeobfuscator = sourceDeobfuscator;
+                m_destDeobfuscator = destDeobfuscator;
+                
+                // init frame
+                m_frame = new JFrame(Constants.Name + " - Field Matcher");
+                final Container pane = m_frame.getContentPane();
+                pane.setLayout(new BorderLayout());
+                
+                // init classes side
+                JPanel classesPanel = new JPanel();
+                classesPanel.setLayout(new BoxLayout(classesPanel, BoxLayout.PAGE_AXIS));
+                classesPanel.setPreferredSize(new Dimension(200, 0));
+                pane.add(classesPanel, BorderLayout.WEST);
+                classesPanel.add(new JLabel("Classes"));
+                
+                m_sourceClasses = new ClassSelector(ClassSelector.DeobfuscatedClassEntryComparator);
+                m_sourceClasses.setListener(new ClassSelectionListener() {
+                        @Override
+                        public void onSelectClass(ClassEntry classEntry) {
+                                setSourceClass(classEntry);
+                        }
+                });
+                JScrollPane sourceScroller = new JScrollPane(m_sourceClasses);
+                classesPanel.add(sourceScroller);
+
+                // init fields side
+                JPanel fieldsPanel = new JPanel();
+                fieldsPanel.setLayout(new BoxLayout(fieldsPanel, BoxLayout.PAGE_AXIS));
+                fieldsPanel.setPreferredSize(new Dimension(200, 0));
+                pane.add(fieldsPanel, BorderLayout.WEST);
+                fieldsPanel.add(new JLabel("Destination Fields"));
+                
+                // init readers
+                DefaultSyntaxKit.initKit();
+                m_sourceReader = new CodeReader();
+                m_destReader = new CodeReader();
+
+                // init all the splits
+                JSplitPane splitLeft = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, true, classesPanel, new JScrollPane(m_sourceReader));
+                splitLeft.setResizeWeight(0); // let the right side take all the slack
+                JSplitPane splitRight = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, true, new JScrollPane(m_destReader), fieldsPanel);
+                splitRight.setResizeWeight(1); // let the left side take all the slack
+                JSplitPane splitCenter = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT, true, splitLeft, splitRight);
+                splitCenter.setResizeWeight(0.5); // resize 50:50
+                pane.add(splitCenter, BorderLayout.CENTER);
+                splitCenter.resetToPreferredSizes();
+                
+                // init bottom panel
+                JPanel bottomPanel = new JPanel();
+                bottomPanel.setLayout(new FlowLayout());
+
+                // show the frame
+                pane.doLayout();
+                m_frame.setSize(1024, 576);
+                m_frame.setMinimumSize(new Dimension(640, 480));
+                m_frame.setVisible(true);
+                m_frame.setDefaultCloseOperation(WindowConstants.DISPOSE_ON_CLOSE);
+
+                // init state
+                m_saveListener = null;
+                m_fieldMatches.addUnmatchedSourceFields(droppedFieldMappings.keySet());
+                m_sourceClasses.setClasses(m_fieldMatches.getSourceClassesWithUnmatchedFields());
+        }
+
+        public void setSaveListener(SaveListener val) {
+                m_saveListener = val;
+        }
+        
+        protected void setSourceClass(ClassEntry obfSourceClass) {
+                
+                // get the matched dest class
+                final ClassEntry obfDestClass = m_classMatches.getUniqueMatches().get(obfSourceClass);
+                if (obfDestClass == null) {
+                        throw new Error("No matching dest class for source class: " + obfSourceClass);
+                }
+
+                m_sourceReader.decompileClass(obfSourceClass, m_sourceDeobfuscator);
+                m_destReader.decompileClass(obfDestClass, m_destDeobfuscator);
+        }
+}
diff --git a/src/cuchaz/enigma/gui/Gui.java b/src/cuchaz/enigma/gui/Gui.java
index 38a6ee9..ea05d25 100644
--- a/src/cuchaz/enigma/gui/Gui.java
+++ b/src/cuchaz/enigma/gui/Gui.java
@@ -737,7 +737,7 @@ public class Gui {
                 if (token == null) {
                         throw new IllegalArgumentException("Token cannot be null!");
                 }
-                GuiTricks.navigateToToken(m_editor, token, m_selectionHighlightPainter);
+                CodeReader.navigateToToken(m_editor, token, m_selectionHighlightPainter);
                 redraw();
         }
         
diff --git a/src/cuchaz/enigma/gui/GuiTricks.java b/src/cuchaz/enigma/gui/GuiTricks.java
index 5a3a01d..df9e221 100644
--- a/src/cuchaz/enigma/gui/GuiTricks.java
+++ b/src/cuchaz/enigma/gui/GuiTricks.java
@@ -11,21 +11,11 @@
 package cuchaz.enigma.gui;
 
 import java.awt.Font;
-import java.awt.Rectangle;
-import java.awt.event.ActionEvent;
-import java.awt.event.ActionListener;
 import java.awt.event.MouseEvent;
 
 import javax.swing.JComponent;
-import javax.swing.JEditorPane;
 import javax.swing.JLabel;
-import javax.swing.SwingUtilities;
-import javax.swing.Timer;
 import javax.swing.ToolTipManager;
-import javax.swing.text.BadLocationException;
-import javax.swing.text.Highlighter.HighlightPainter;
-
-import cuchaz.enigma.analysis.Token;
 
 public class GuiTricks {
         
@@ -43,52 +33,4 @@ public class GuiTricks {
                 manager.mouseMoved(new MouseEvent(component, MouseEvent.MOUSE_MOVED, System.currentTimeMillis(), 0, 0, 0, 0, false));
                 manager.setInitialDelay(oldDelay);
         }
-        
-        public static void navigateToToken(final JEditorPane editor, final Token token, final HighlightPainter highlightPainter) {
-                
-                // set the caret position to the token
-                editor.setCaretPosition(token.start);
-                editor.grabFocus();
-                
-                try {
-                        // make sure the token is visible in the scroll window
-                        Rectangle start = editor.modelToView(token.start);
-                        Rectangle end = editor.modelToView(token.end);
-                        final Rectangle show = start.union(end);
-                        show.grow(start.width * 10, start.height * 6);
-                        SwingUtilities.invokeLater(new Runnable() {
-                                @Override
-                                public void run() {
-                                        editor.scrollRectToVisible(show);
-                                }
-                        });
-                } catch (BadLocationException ex) {
-                        throw new Error(ex);
-                }
-                
-                // highlight the token momentarily
-                final Timer timer = new Timer(200, new ActionListener() {
-                        private int m_counter = 0;
-                        private Object m_highlight = null;
-                        
-                        @Override
-                        public void actionPerformed(ActionEvent event) {
-                                if (m_counter % 2 == 0) {
-                                        try {
-                                                m_highlight = editor.getHighlighter().addHighlight(token.start, token.end, highlightPainter);
-                                        } catch (BadLocationException ex) {
-                                                // don't care
-                                        }
-                                } else if (m_highlight != null) {
-                                        editor.getHighlighter().removeHighlight(m_highlight);
-                                }
-                                
-                                if (m_counter++ > 6) {
-                                        Timer timer = (Timer)event.getSource();
-                                        timer.stop();
-                                }
-                        }
-                });
-                timer.start();
-        }
 }