Merge branch 'appcode10' into merge_appcode10

[idea/community.git] / platform / core-api / src / com / intellij / psi / util / PsiTreeUtil.java

diff --git a/platform/core-api/src/com/intellij/psi/util/PsiTreeUtil.java b/platform/core-api/src/com/intellij/psi/util/PsiTreeUtil.java
index fafa4d00570ce753c6f8667abd03380221201849..af7d0f557c8c10174e3ae25a45007b5cbc766399 100644 (file)
--- a/platform/core-api/src/com/intellij/psi/util/PsiTreeUtil.java
+++ b/platform/core-api/src/com/intellij/psi/util/PsiTreeUtil.java
@@ -733,6 +733,32 @@ public class PsiTreeUtil {
     return result;
   }
 
+  @Nullable
+  public static <T extends PsiElement> T findElementOfClassAtOffsetWithStopSet(@NotNull PsiFile file,
+                                                                               int offset,
+                                                                               @NotNull Class<T> clazz,
+                                                                               boolean strictStart,
+                                                                               @NotNull Class<? extends PsiElement>... stopAt) {
+    final PsiElement[] psiRoots = file.getPsiRoots();
+    T result = null;
+    for (PsiElement root : psiRoots) {
+      final PsiElement elementAt = root.findElementAt(offset);
+      if (elementAt != null) {
+        final T parent = getParentOfType(elementAt, clazz, strictStart, stopAt);
+        if (parent != null) {
+          final TextRange range = parent.getTextRange();
+          if (!strictStart || range.getStartOffset() == offset) {
+            if (result == null || result.getTextRange().getEndOffset() > range.getEndOffset()) {
+              result = parent;
+            }
+          }
+        }
+      }
+    }
+
+    return result;
+  }
+
   /**
    * @return maximal element of specified Class starting at startOffset exactly and ending not farther than endOffset
    */