From 37b0fb65413f13ff70d9f9747799d11cbf62a0d9 Mon Sep 17 00:00:00 2001
From: Juuxel
Date: Mon, 14 Dec 2020 23:13:57 +0200
Subject: Use class icon in class tree (ClassSelector) (#2)

---
 enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'enigma-swing/src/main/java')

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 488d04e..b27832b 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/ClassSelector.java
@@ -27,6 +27,7 @@ import com.google.common.collect.Maps;
 import com.google.common.collect.Multimap;
 import cuchaz.enigma.gui.node.ClassSelectorClassNode;
 import cuchaz.enigma.gui.node.ClassSelectorPackageNode;
+import cuchaz.enigma.gui.util.GuiUtil;
 import cuchaz.enigma.translation.Translator;
 import cuchaz.enigma.translation.representation.entry.ClassEntry;
 import cuchaz.enigma.utils.validation.ValidationContext;
@@ -69,10 +70,13 @@ public class ClassSelector extends JTree {
 			}
 		});
 
+		final DefaultTreeCellRenderer renderer = new DefaultTreeCellRenderer();
+		renderer.setLeafIcon(GuiUtil.CLASS_ICON);
+		setCellRenderer(renderer);
+
 		final JTree tree = this;
 
-		final DefaultTreeCellEditor editor = new DefaultTreeCellEditor(tree,
-				(DefaultTreeCellRenderer) tree.getCellRenderer()) {
+		final DefaultTreeCellEditor editor = new DefaultTreeCellEditor(tree, renderer) {
 			@Override
 			public boolean isCellEditable(EventObject event) {
 				return isRenamable && !(event instanceof MouseEvent) && super.isCellEditable(event);
-- 
cgit v1.2.3