Merge branch 'structure-panel' of https://github.com/YanisBft/Enigma into structure-panel

2 files changed, 6 insertions, 1 deletions

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/StructurePanel.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/StructurePanel.java
index 32f803f8..d6044dc7 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/StructurePanel.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/panels/StructurePanel.java
@@ -89,7 +89,7 @@ public class StructurePanel extends JPanel {
             if (entry instanceof ClassEntry) {
                 this.label.setIcon(GuiUtil.CLASS_ICON);
             } else if (entry instanceof MethodEntry) {
-                this.label.setIcon(GuiUtil.METHOD_ICON);
+                this.label.setIcon(isConstructor(entry) ? GuiUtil.CONSTRUCTOR_ICON : GuiUtil.METHOD_ICON);
             } else if (entry instanceof FieldEntry) {
                 this.label.setIcon(GuiUtil.FIELD_ICON);
             }
@@ -98,5 +98,9 @@ public class StructurePanel extends JPanel {
 
             return this.label;
         }
+
+        private boolean isConstructor(ParentedEntry<?> entry) {
+            return "<init>".equals(entry.getName());
+        }
     }
 }
diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/util/GuiUtil.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/util/GuiUtil.java
index 666fc0aa..6393913d 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/util/GuiUtil.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/util/GuiUtil.java
@@ -18,6 +18,7 @@ public class GuiUtil {
     public static final Icon CLASS_ICON = loadIcon("class");
     public static final Icon METHOD_ICON = loadIcon("method");
     public static final Icon FIELD_ICON = loadIcon("field");
+    public static final Icon CONSTRUCTOR_ICON = loadIcon("constructor");
 
     public static void openUrl(String url) {
         try {