From 36bc9f2bcbf7f5181f47d4b492d0b2f8aa28d77a Mon Sep 17 00:00:00 2001
From: Konstantin Bulenkov <kb@jetbrains.com>
Date: Mon, 9 Nov 2009 16:12:14 +0300
Subject: [PATCH] getIcon() in PsiViewerExtention

---
 .../internal/psiView/PsiViewerCodeFragmentExtension.java      | 11 +++++++++--
 .../internal/psiView/PsiViewerExpressionExtension.java        | 11 +++++++++--
 .../intellij/internal/psiView/PsiViewerMethodExtension.java   | 11 +++++++++--
 .../src/com/intellij/internal/psiView/PsiViewerExtension.java |  6 +++++-
 4 files changed, 32 insertions(+), 7 deletions(-)

diff --git a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerCodeFragmentExtension.java b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerCodeFragmentExtension.java
index 6c31250821..9a2a645a84 100644
--- a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerCodeFragmentExtension.java
+++ b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerCodeFragmentExtension.java
@@ -15,9 +15,12 @@
  */
 package com.intellij.internal.psiView;
 
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.JavaPsiFacade;
 import com.intellij.openapi.project.Project;
+import com.intellij.psi.JavaPsiFacade;
+import com.intellij.psi.PsiElement;
+import com.intellij.util.Icons;
+
+import javax.swing.*;
 
 /**
  * @author yole
@@ -27,6 +30,10 @@ public class PsiViewerCodeFragmentExtension implements PsiViewerExtension {
     return "Java Code Block";
   }
 
+  public Icon getIcon() {
+    return Icons.CLASS_INITIALIZER;
+  }
+
   public PsiElement createElement(Project project, String text) {
     return JavaPsiFacade.getElementFactory(project).createCodeBlockFromText(text, null);
   }
diff --git a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerExpressionExtension.java b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerExpressionExtension.java
index 430433337f..6122df00ec 100644
--- a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerExpressionExtension.java
+++ b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerExpressionExtension.java
@@ -15,9 +15,12 @@
  */
 package com.intellij.internal.psiView;
 
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.JavaPsiFacade;
 import com.intellij.openapi.project.Project;
+import com.intellij.psi.JavaPsiFacade;
+import com.intellij.psi.PsiElement;
+import com.intellij.util.Icons;
+
+import javax.swing.*;
 
 /**
  * @author yole
@@ -27,6 +30,10 @@ public class PsiViewerExpressionExtension implements PsiViewerExtension {
     return "Java Expression";
   }
 
+  public Icon getIcon() {
+    return Icons.CLASS_INITIALIZER;
+  }
+
   public PsiElement createElement(Project project, String text) {
     return JavaPsiFacade.getElementFactory(project).createExpressionFromText(text, null);
   }
diff --git a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerMethodExtension.java b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerMethodExtension.java
index 2ea51dba18..8067797cda 100644
--- a/java/java-impl/src/com/intellij/internal/psiView/PsiViewerMethodExtension.java
+++ b/java/java-impl/src/com/intellij/internal/psiView/PsiViewerMethodExtension.java
@@ -15,9 +15,12 @@
  */
 package com.intellij.internal.psiView;
 
-import com.intellij.psi.PsiElement;
-import com.intellij.psi.JavaPsiFacade;
 import com.intellij.openapi.project.Project;
+import com.intellij.psi.JavaPsiFacade;
+import com.intellij.psi.PsiElement;
+import com.intellij.util.Icons;
+
+import javax.swing.*;
 
 /**
  * @author yole
@@ -27,6 +30,10 @@ public class PsiViewerMethodExtension implements PsiViewerExtension {
     return "Java Method";
   }
 
+  public Icon getIcon() {
+    return Icons.METHOD_ICON;
+  }
+
   public PsiElement createElement(Project project, String text) {
     return JavaPsiFacade.getInstance(project).getElementFactory().createMethodFromText(text, null);
   }
diff --git a/platform/lang-impl/src/com/intellij/internal/psiView/PsiViewerExtension.java b/platform/lang-impl/src/com/intellij/internal/psiView/PsiViewerExtension.java
index cb06f0f251..9aa8d777d8 100644
--- a/platform/lang-impl/src/com/intellij/internal/psiView/PsiViewerExtension.java
+++ b/platform/lang-impl/src/com/intellij/internal/psiView/PsiViewerExtension.java
@@ -16,16 +16,20 @@
 
 package com.intellij.internal.psiView;
 
-import com.intellij.openapi.project.Project;
 import com.intellij.openapi.extensions.ExtensionPointName;
+import com.intellij.openapi.project.Project;
 import com.intellij.psi.PsiElement;
 
+import javax.swing.*;
+
 /**
  * @author yole
+ * @author Konstantin Bulenkov
  */
 public interface PsiViewerExtension {
   ExtensionPointName<PsiViewerExtension> EP_NAME = ExtensionPointName.create("com.intellij.psiViewerExtension");
 
   String getName();
+  Icon getIcon();
   PsiElement createElement(Project project, String text);
 }
-- 
2.11.4.GIT