From 343ec04252ea876c11a1bdb1e9833bd5f2db784d Mon Sep 17 00:00:00 2001
From: Ben Gruver <bgruv@google.com>
Date: Fri, 6 Mar 2015 20:00:16 -0800
Subject: [PATCH] Associate the smali icon with the SmaliClass object

This ensures that the icon is shown next to the class in the project view
---
 .../src/main/java/org/jf/smalidea/psi/impl/SmaliClass.java  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/smalidea/src/main/java/org/jf/smalidea/psi/impl/SmaliClass.java b/smalidea/src/main/java/org/jf/smalidea/psi/impl/SmaliClass.java
index 84720f39..b4c473fe 100644
--- a/smalidea/src/main/java/org/jf/smalidea/psi/impl/SmaliClass.java
+++ b/smalidea/src/main/java/org/jf/smalidea/psi/impl/SmaliClass.java
@@ -48,11 +48,13 @@ import com.sun.jdi.ReferenceType;
 import org.jetbrains.annotations.NonNls;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
+import org.jf.smalidea.SmaliIcons;
 import org.jf.smalidea.psi.SmaliElementTypes;
 import org.jf.smalidea.psi.iface.SmaliModifierListOwner;
 import org.jf.smalidea.psi.stub.SmaliClassStub;
 
 import javax.annotation.Nonnull;
+import javax.swing.*;
 import java.util.Collection;
 import java.util.List;
 
@@ -330,4 +332,8 @@ public class SmaliClass extends SmaliStubBasedPsiElement<SmaliClassStub> impleme
         return PsiClassImplUtil.processDeclarationsInClass(this, processor, state, null, lastParent, place,
                 PsiUtil.getLanguageLevel(place), false);
     }
+
+    @Nullable @Override protected Icon getElementIcon(@IconFlags int flags) {
+        return SmaliIcons.SmaliIcon;
+    }
 }
\ No newline at end of file