From aaf8aabb725dcfed74c02204227d43093085f791 Mon Sep 17 00:00:00 2001
From: Eugene Vigdorchik <no_reply@jetbrains.com>
Date: Thu, 24 Feb 2005 15:05:29 +0300
Subject: [PATCH] (no message)

---
 .../psi/impl/search/PsiSearchHelperImpl.java         | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/source/com/intellij/psi/impl/search/PsiSearchHelperImpl.java b/source/com/intellij/psi/impl/search/PsiSearchHelperImpl.java
index 1fdffba6d3b4..7013fdf3801d 100644
--- a/source/com/intellij/psi/impl/search/PsiSearchHelperImpl.java
+++ b/source/com/intellij/psi/impl/search/PsiSearchHelperImpl.java
@@ -201,10 +201,16 @@ public class PsiSearchHelperImpl implements PsiSearchHelper {
   }
 
   private static PsiClass getTopLevelClass(PsiElement element) {
-    while (!(element.getParent() instanceof PsiFile)) {
-      element = element.getParent();
+    LOG.assertTrue(element.isValid());
+    final PsiFile file = element.getContainingFile();
+    if (file instanceof PsiJavaFile) {
+      final PsiClass[] classes = ((PsiJavaFile)file).getClasses();
+      for (int i = 0; i < classes.length; i++) {
+        PsiClass aClass = classes[i];
+        if (PsiTreeUtil.isAncestor(aClass, element, false)) return aClass;
+      }
     }
-    return element instanceof PsiClass ? (PsiClass)element : null;
+    return null;
   }
 
   public interface CustomSearchHelper {