5 files changed, 27 insertions, 14 deletions

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java
index 7044106b..57b23c9a 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java
@@ -361,6 +361,12 @@ public class ClassSelector extends JTree {
                 }
         }
 
+        public void collapseAll() {
+                for (ClassSelectorPackageNode packageNode : packageNodes()) {
+                        collapsePath(new TreePath(new Object[]{getModel().getRoot(), packageNode}));
+                }
+        }
+
         public ClassEntry getFirstClass() {
                 ClassSelectorPackageNode packageNode = packageNodes().get(0);
                 if (packageNode != null) {
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/config/Decompiler.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/config/Decompiler.java
index 0f9e41c6..8232ab6e 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/config/Decompiler.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/config/Decompiler.java
@@ -4,8 +4,8 @@ import cuchaz.enigma.source.DecompilerService;
 import cuchaz.enigma.source.Decompilers;
 
 public enum Decompiler {
-        PROCYON("Procyon", Decompilers.PROCYON),
-        CFR("CFR", Decompilers.CFR);
+        CFR("CFR", Decompilers.CFR),
+        PROCYON("Procyon", Decompilers.PROCYON);
 
         public final DecompilerService service;
         public final String name;
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/DeobfPanelPopupMenu.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/DeobfPanelPopupMenu.java
index c888e4f7..9481412e 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/DeobfPanelPopupMenu.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/DeobfPanelPopupMenu.java
@@ -11,15 +11,22 @@ import java.awt.*;
 public class DeobfPanelPopupMenu {
 
     private final JPopupMenu ui;
-    private final JMenuItem renamePackage;
-    private final JMenuItem renameClass;
+    private final JMenuItem renamePackage = new JMenuItem();
+    private final JMenuItem renameClass = new JMenuItem();
+    private final JMenuItem expandAll = new JMenuItem();
+    private final JMenuItem collapseAll = new JMenuItem();
 
     public DeobfPanelPopupMenu(Gui gui) {
         this.ui = new JPopupMenu();
 
+        this.ui.add(this.renamePackage);
+        this.ui.add(this.renameClass);
+        this.ui.addSeparator();
+        this.ui.add(this.expandAll);
+        this.ui.add(this.collapseAll);
+
         ClassSelector deobfClasses = gui.getDeobfPanel().deobfClasses;
 
-        this.renamePackage = new JMenuItem();
         this.renamePackage.addActionListener(a -> {
             TreePath path;
 
@@ -33,11 +40,9 @@ public class DeobfPanelPopupMenu {
 
             deobfClasses.getUI().startEditingAtPath(deobfClasses, path);
         });
-        this.ui.add(this.renamePackage);
-
-        this.renameClass = new JMenuItem();
         this.renameClass.addActionListener(a -> deobfClasses.getUI().startEditingAtPath(deobfClasses, deobfClasses.getSelectionPath()));
-        this.ui.add(this.renameClass);
+        this.expandAll.addActionListener(a -> deobfClasses.expandAll());
+        this.collapseAll.addActionListener(a -> deobfClasses.collapseAll());
 
         this.retranslateUi();
     }
@@ -52,5 +57,7 @@ public class DeobfPanelPopupMenu {
     public void retranslateUi() {
         this.renamePackage.setText(I18n.translate("popup_menu.deobf_panel.rename_package"));
         this.renameClass.setText(I18n.translate("popup_menu.deobf_panel.rename_class"));
+        this.expandAll.setText(I18n.translate("popup_menu.deobf_panel.expand_all"));
+        this.collapseAll.setText(I18n.translate("popup_menu.deobf_panel.collapse_all"));
     }
 }
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
index 6603dd55..a0b2a52b 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/elements/MenuBar.java
@@ -57,9 +57,9 @@ public class MenuBar {
         private final JMenuItem customScaleItem = new JMenuItem();
 
         private final JMenu searchMenu = new JMenu();
-        private final JMenuItem searchClassItem = new JMenuItem();
-        private final JMenuItem searchMethodItem = new JMenuItem();
-        private final JMenuItem searchFieldItem = new JMenuItem();
+        private final JMenuItem searchClassItem = new JMenuItem(GuiUtil.CLASS_ICON);
+        private final JMenuItem searchMethodItem = new JMenuItem(GuiUtil.METHOD_ICON);
+        private final JMenuItem searchFieldItem = new JMenuItem(GuiUtil.FIELD_ICON);
 
         private final JMenu collabMenu = new JMenu();
         private final JMenuItem connectItem = new JMenuItem();
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/stats/StatsMember.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/stats/StatsMember.java
index 70b4f40d..0e2452fa 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/stats/StatsMember.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/stats/StatsMember.java
@@ -1,8 +1,8 @@
 package cuchaz.enigma.gui.stats;
 
 public enum StatsMember {
+    CLASSES,
     METHODS,
     FIELDS,
-    PARAMETERS,
-    CLASSES
+    PARAMETERS
 }