From 5ee56d4ce7d29bfb47dc5d2156dedc408a549f85 Mon Sep 17 00:00:00 2001
From: Maxim Shafirov <max@jetbrains.com>
Date: Tue, 1 Feb 2005 14:09:50 +0300
Subject: [PATCH] SourceTreeToPsiMap.hasTreeElement() optimization

---
 source/com/intellij/psi/impl/source/SourceTreeToPsiMap.java | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/source/com/intellij/psi/impl/source/SourceTreeToPsiMap.java b/source/com/intellij/psi/impl/source/SourceTreeToPsiMap.java
index ff7057004e7b..85f225e8dd9c 100644
--- a/source/com/intellij/psi/impl/source/SourceTreeToPsiMap.java
+++ b/source/com/intellij/psi/impl/source/SourceTreeToPsiMap.java
@@ -2,6 +2,8 @@ package com.intellij.psi.impl.source;
 
 import com.intellij.lang.ASTNode;
 import com.intellij.psi.PsiElement;
+import com.intellij.ide.util.treeView.smartTree.TreeElement;
+import com.intellij.extapi.psi.ASTWrapperPsiElement;
 
 public class SourceTreeToPsiMap {
   public static PsiElement treeElementToPsi(ASTNode element) {
@@ -15,6 +17,6 @@ public class SourceTreeToPsiMap {
   }
 
   public static boolean hasTreeElement(PsiElement psiElement) {
-    return psiElement != null && psiElement.getNode() != null;
+    return psiElement instanceof TreeElement || psiElement instanceof SrcRepositoryPsiElement || psiElement instanceof ASTWrapperPsiElement;
   }
 }