4 files changed, 34 insertions, 56 deletions

diff --git a/src/main/java/cuchaz/enigma/gui/CodeReader.java b/src/main/java/cuchaz/enigma/gui/CodeReader.java
index f76dc89..97deaf3 100644
--- a/src/main/java/cuchaz/enigma/gui/CodeReader.java
+++ b/src/main/java/cuchaz/enigma/gui/CodeReader.java
@@ -162,7 +162,7 @@ public class CodeReader extends JEditorPane {
                         // 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)) {
+                                if (entry.getOwnerClassEntry().equals(classEntry)) {
                                         token = sourceIndex.getDeclarationToken(entry);
                                         break;
                                 }
diff --git a/src/main/java/cuchaz/enigma/gui/Gui.java b/src/main/java/cuchaz/enigma/gui/Gui.java
index 4a891cf..5d9f7cb 100644
--- a/src/main/java/cuchaz/enigma/gui/Gui.java
+++ b/src/main/java/cuchaz/enigma/gui/Gui.java
@@ -35,7 +35,6 @@ import cuchaz.enigma.mapping.*;
 import cuchaz.enigma.throwables.IllegalNameException;
 import cuchaz.enigma.utils.Utils;
 import de.sciss.syntaxpane.DefaultSyntaxKit;
-import javassist.bytecode.Descriptor;
 
 import javax.swing.*;
 import javax.swing.text.BadLocationException;
@@ -48,8 +47,10 @@ import java.awt.*;
 import java.awt.event.*;
 import java.io.File;
 import java.io.IOException;
-import java.util.*;
+import java.util.Collection;
+import java.util.Collections;
 import java.util.List;
+import java.util.Vector;
 import java.util.function.Function;
 
 public class Gui {
@@ -438,14 +439,10 @@ public class Gui {
                         showFieldEntry((FieldEntry) this.reference.entry);
                 } else if (this.reference.entry instanceof MethodEntry) {
                         showMethodEntry((MethodEntry) this.reference.entry);
-                } else if (this.reference.entry instanceof ConstructorEntry) {
-                        showConstructorEntry((ConstructorEntry) this.reference.entry);
-                } else if (this.reference.entry instanceof ArgumentEntry) {
-                        showArgumentEntry((ArgumentEntry) this.reference.entry);
                 } else if (this.reference.entry instanceof LocalVariableEntry) {
                         showLocalVariableEntry((LocalVariableEntry) this.reference.entry);
                 } else {
-                        throw new Error("Unknown entry type: " + this.reference.entry.getClass().getName());
+                        throw new Error("Unknown entry desc: " + this.reference.entry.getClass().getName());
                 }
 
                 redraw();
@@ -453,10 +450,9 @@ public class Gui {
 
         private void showLocalVariableEntry(LocalVariableEntry entry) {
                 addNameValue(infoPanel, "Variable", entry.getName());
-                addNameValue(infoPanel, "Class", entry.getClassEntry().getName());
-                addNameValue(infoPanel, "Method", entry.getBehaviorEntry().getName());
+                addNameValue(infoPanel, "Class", entry.getOwnerClassEntry().getName());
+                addNameValue(infoPanel, "Method", entry.getOwnerEntry().getName());
                 addNameValue(infoPanel, "Index", Integer.toString(entry.getIndex()));
-                addNameValue(infoPanel, "Type", entry.getType().toString());
         }
 
         private void showClassEntry(ClassEntry entry) {
@@ -466,32 +462,20 @@ public class Gui {
 
         private void showFieldEntry(FieldEntry entry) {
                 addNameValue(infoPanel, "Field", entry.getName());
-                addNameValue(infoPanel, "Class", entry.getClassEntry().getName());
-                addNameValue(infoPanel, "Type", entry.getType().toString());
+                addNameValue(infoPanel, "Class", entry.getOwnerClassEntry().getName());
+                addNameValue(infoPanel, "TypeDescriptor", entry.getDesc().toString());
                 addModifierComboBox(infoPanel, "Modifier", entry);
         }
 
         private void showMethodEntry(MethodEntry entry) {
-                addNameValue(infoPanel, "Method", entry.getName());
-                addNameValue(infoPanel, "Class", entry.getClassEntry().getName());
-                addNameValue(infoPanel, "Signature", entry.getSignature().toString());
-                addModifierComboBox(infoPanel, "Modifier", entry);
-
-        }
-
-        private void showConstructorEntry(ConstructorEntry entry) {
-                addNameValue(infoPanel, "Constructor", entry.getClassEntry().getName());
-                if (!entry.isStatic()) {
-                        addNameValue(infoPanel, "Signature", entry.getSignature().toString());
-                        addModifierComboBox(infoPanel, "Modifier", entry);
+                if (entry.isConstructor()) {
+                        addNameValue(infoPanel, "Constructor", entry.getOwnerClassEntry().getName());
+                } else {
+                        addNameValue(infoPanel, "Method", entry.getName());
+                        addNameValue(infoPanel, "Class", entry.getOwnerClassEntry().getName());
                 }
-        }
-
-        private void showArgumentEntry(ArgumentEntry entry) {
-                addNameValue(infoPanel, "Argument", entry.getName());
-                addNameValue(infoPanel, "Class", entry.getClassEntry().getName());
-                addNameValue(infoPanel, "Method", entry.getBehaviorEntry().getName());
-                addNameValue(infoPanel, "Index", Integer.toString(entry.getIndex()));
+                addNameValue(infoPanel, "MethodDescriptor", entry.getDesc().toString());
+                addModifierComboBox(infoPanel, "Modifier", entry);
         }
 
         private void addNameValue(JPanel container, String name, String value) {
@@ -532,8 +516,8 @@ public class Gui {
                 reference = this.controller.getDeobfReference(token);
                 boolean isClassEntry = isToken && reference.entry instanceof ClassEntry;
                 boolean isFieldEntry = isToken && reference.entry instanceof FieldEntry;
-                boolean isMethodEntry = isToken && reference.entry instanceof MethodEntry;
-                boolean isConstructorEntry = isToken && reference.entry instanceof ConstructorEntry;
+                boolean isMethodEntry = isToken && reference.entry instanceof MethodEntry && !((MethodEntry) reference.entry).isConstructor();
+                boolean isConstructorEntry = isToken && reference.entry instanceof MethodEntry && ((MethodEntry) reference.entry).isConstructor();
                 boolean isInJar = isToken && this.controller.entryIsInJar(reference.entry);
                 boolean isRenameable = isToken && this.controller.referenceIsRenameable(reference);
 
@@ -710,16 +694,13 @@ public class Gui {
                 if (reference.entry instanceof ClassEntry) {
                         // look for calls to the default constructor
                         // TODO: get a list of all the constructors and find calls to all of them
-                        BehaviorReferenceTreeNode node = this.controller.getMethodReferences(new ConstructorEntry((ClassEntry) reference.entry, new Signature("()V")));
+                        MethodReferenceTreeNode node = this.controller.getMethodReferences(new MethodEntry((ClassEntry) reference.entry, "<init>", new MethodDescriptor("()V")));
                         callsTree.setModel(new DefaultTreeModel(node));
                 } else if (reference.entry instanceof FieldEntry) {
                         FieldReferenceTreeNode node = this.controller.getFieldReferences((FieldEntry) reference.entry);
                         callsTree.setModel(new DefaultTreeModel(node));
                 } else if (reference.entry instanceof MethodEntry) {
-                        BehaviorReferenceTreeNode node = this.controller.getMethodReferences((MethodEntry) reference.entry);
-                        callsTree.setModel(new DefaultTreeModel(node));
-                } else if (reference.entry instanceof ConstructorEntry) {
-                        BehaviorReferenceTreeNode node = this.controller.getMethodReferences((ConstructorEntry) reference.entry);
+                        MethodReferenceTreeNode node = this.controller.getMethodReferences((MethodEntry) reference.entry);
                         callsTree.setModel(new DefaultTreeModel(node));
                 }
 
@@ -790,7 +771,6 @@ public class Gui {
                 // package rename
                 if (data instanceof String) {
                         for (int i = 0; i < node.getChildCount(); i++) {
-                                data = Descriptor.toJvmName((String) data);
                                 DefaultMutableTreeNode childNode = (DefaultMutableTreeNode) node.getChildAt(i);
                                 ClassEntry prevDataChild = (ClassEntry) childNode.getUserObject();
                                 ClassEntry dataChild = new ClassEntry(data + "/" + prevDataChild.getSimpleName());
@@ -807,15 +787,15 @@ public class Gui {
         }
 
         public void moveClassTree(EntryReference<Entry, Entry> deobfReference, String newName) {
-                String oldEntry = deobfReference.entry.getClassEntry().getPackageName();
-                String newEntry = new ClassEntry(Descriptor.toJvmName(newName)).getPackageName();
+                String oldEntry = deobfReference.entry.getOwnerClassEntry().getPackageName();
+                String newEntry = new ClassEntry(newName).getPackageName();
                 moveClassTree(deobfReference, newName, oldEntry == null,
                         newEntry == null);
         }
 
         public void moveClassTree(EntryReference<Entry, Entry> deobfReference, String newName, boolean isOldOb, boolean isNewOb) {
-                ClassEntry oldEntry = deobfReference.entry.getClassEntry();
-                ClassEntry newEntry = new ClassEntry(Descriptor.toJvmName(newName));
+                ClassEntry oldEntry = deobfReference.entry.getOwnerClassEntry();
+                ClassEntry newEntry = new ClassEntry(newName);
 
                 // Ob -> deob
                 if (isOldOb && !isNewOb) {
diff --git a/src/main/java/cuchaz/enigma/gui/GuiController.java b/src/main/java/cuchaz/enigma/gui/GuiController.java
index 6d98743..a5528a7 100644
--- a/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -51,7 +51,7 @@ public class GuiController {
                 return this.isDirty;
         }
 
-        public void openJar(final JarFile jar) {
+        public void openJar(final JarFile jar) throws IOException {
                 this.gui.onStartOpenJar();
                 this.deobfuscator = new Deobfuscator(jar);
                 this.gui.onFinishOpenJar(this.deobfuscator.getJarName());
@@ -161,24 +161,24 @@ public class GuiController {
 
         public ClassInheritanceTreeNode getClassInheritance(ClassEntry deobfClassEntry) {
                 ClassEntry obfClassEntry = this.deobfuscator.obfuscateEntry(deobfClassEntry);
-                ClassInheritanceTreeNode rootNode = this.deobfuscator.getJarIndex().getClassInheritance(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfClassEntry);
+                ClassInheritanceTreeNode rootNode = this.deobfuscator.getJarIndex().getClassInheritance(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfClassEntry);
                 return ClassInheritanceTreeNode.findNode(rootNode, obfClassEntry);
         }
 
         public ClassImplementationsTreeNode getClassImplementations(ClassEntry deobfClassEntry) {
                 ClassEntry obfClassEntry = this.deobfuscator.obfuscateEntry(deobfClassEntry);
-                return this.deobfuscator.getJarIndex().getClassImplementations(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfClassEntry);
+                return this.deobfuscator.getJarIndex().getClassImplementations(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfClassEntry);
         }
 
         public MethodInheritanceTreeNode getMethodInheritance(MethodEntry deobfMethodEntry) {
                 MethodEntry obfMethodEntry = this.deobfuscator.obfuscateEntry(deobfMethodEntry);
-                MethodInheritanceTreeNode rootNode = this.deobfuscator.getJarIndex().getMethodInheritance(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfMethodEntry);
+                MethodInheritanceTreeNode rootNode = this.deobfuscator.getJarIndex().getMethodInheritance(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfMethodEntry);
                 return MethodInheritanceTreeNode.findNode(rootNode, obfMethodEntry);
         }
 
         public MethodImplementationsTreeNode getMethodImplementations(MethodEntry deobfMethodEntry) {
                 MethodEntry obfMethodEntry = this.deobfuscator.obfuscateEntry(deobfMethodEntry);
-                List<MethodImplementationsTreeNode> rootNodes = this.deobfuscator.getJarIndex().getMethodImplementations(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfMethodEntry);
+                List<MethodImplementationsTreeNode> rootNodes = this.deobfuscator.getJarIndex().getMethodImplementations(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfMethodEntry);
                 if (rootNodes.isEmpty()) {
                         return null;
                 }
@@ -190,14 +190,14 @@ public class GuiController {
 
         public FieldReferenceTreeNode getFieldReferences(FieldEntry deobfFieldEntry) {
                 FieldEntry obfFieldEntry = this.deobfuscator.obfuscateEntry(deobfFieldEntry);
-                FieldReferenceTreeNode rootNode = new FieldReferenceTreeNode(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfFieldEntry);
+                FieldReferenceTreeNode rootNode = new FieldReferenceTreeNode(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfFieldEntry);
                 rootNode.load(this.deobfuscator.getJarIndex(), true);
                 return rootNode;
         }
 
-        public BehaviorReferenceTreeNode getMethodReferences(BehaviorEntry deobfBehaviorEntry) {
-                BehaviorEntry obfBehaviorEntry = this.deobfuscator.obfuscateEntry(deobfBehaviorEntry);
-                BehaviorReferenceTreeNode rootNode = new BehaviorReferenceTreeNode(this.deobfuscator.getTranslator(TranslationDirection.Deobfuscating), obfBehaviorEntry);
+        public MethodReferenceTreeNode getMethodReferences(MethodEntry deobfMethodEntry) {
+                MethodEntry obfMethodEntry = this.deobfuscator.obfuscateEntry(deobfMethodEntry);
+                MethodReferenceTreeNode rootNode = new MethodReferenceTreeNode(this.deobfuscator.getTranslator(TranslationDirection.DEOBFUSCATING), obfMethodEntry);
                 rootNode.load(this.deobfuscator.getJarIndex(), true);
                 return rootNode;
         }
diff --git a/src/main/java/cuchaz/enigma/gui/node/ClassSelectorPackageNode.java b/src/main/java/cuchaz/enigma/gui/node/ClassSelectorPackageNode.java
index f80abba..caa985c 100644
--- a/src/main/java/cuchaz/enigma/gui/node/ClassSelectorPackageNode.java
+++ b/src/main/java/cuchaz/enigma/gui/node/ClassSelectorPackageNode.java
@@ -11,8 +11,6 @@
 
 package cuchaz.enigma.gui.node;
 
-import javassist.bytecode.Descriptor;
-
 import javax.swing.tree.DefaultMutableTreeNode;
 
 public class ClassSelectorPackageNode extends DefaultMutableTreeNode {
@@ -41,7 +39,7 @@ public class ClassSelectorPackageNode extends DefaultMutableTreeNode {
 
         @Override
         public String toString() {
-                return !packageName.equals("(none)") ? Descriptor.toJavaName(this.packageName) : "(none)";
+                return !packageName.equals("(none)") ? this.packageName : "(none)";
         }
 
         @Override