From 7e56311d6f68360fc95a0da6d359879c6a4da7d4 Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Mon, 31 Mar 2014 14:09:17 +0400
Subject: [PATCH] Add isTopLevel() to PsiJetClassStub

---
 .../jet/lang/psi/stubs/PsiJetClassOrObjectStub.java    |  2 ++
 .../jetbrains/jet/lang/psi/stubs/PsiJetObjectStub.java |  1 -
 .../lang/psi/stubs/elements/JetClassElementType.java   |  7 ++++---
 .../jet/lang/psi/stubs/impl/PsiJetClassStubImpl.java   | 10 +++++++++-
 4 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetClassOrObjectStub.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetClassOrObjectStub.java
index 7ab610fa1f8..825ce57ac75 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetClassOrObjectStub.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetClassOrObjectStub.java
@@ -26,4 +26,6 @@ public interface PsiJetClassOrObjectStub<T extends JetClassOrObject> extends Psi
 
     @NotNull
     List<String> getSuperNames();
+
+    boolean isTopLevel();
 }
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetObjectStub.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetObjectStub.java
index 6547314c890..e92dcf10719 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetObjectStub.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/PsiJetObjectStub.java
@@ -19,7 +19,6 @@ package org.jetbrains.jet.lang.psi.stubs;
 import org.jetbrains.jet.lang.psi.JetObjectDeclaration;
 
 public interface PsiJetObjectStub extends PsiJetClassOrObjectStub<JetObjectDeclaration> {
-    boolean isTopLevel();
     boolean isClassObject();
     boolean isObjectLiteral();
 }
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetClassElementType.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetClassElementType.java
index 96fe0758f6e..b47edc67d5b 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetClassElementType.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/elements/JetClassElementType.java
@@ -67,8 +67,7 @@ public class JetClassElementType extends JetStubElementType<PsiJetClassStub, Jet
         return new PsiJetClassStubImpl(
                 getStubType(isEnumEntry), parentStub, StringRef.fromString(fqName != null ? fqName.asString() : null),
                 StringRef.fromString(psi.getName()), Utils.instance$.wrapStrings(superNames), psi.isTrait(), psi.isEnum(), isEnumEntry,
-                psi.isAnnotation(), psi.isInner(), JetPsiUtil.isLocal(psi)
-        );
+                psi.isAnnotation(), psi.isInner(), JetPsiUtil.isLocal(psi), psi.isTopLevel());
     }
 
     @Override
@@ -82,6 +81,7 @@ public class JetClassElementType extends JetStubElementType<PsiJetClassStub, Jet
         dataStream.writeBoolean(stub.isAnnotation());
         dataStream.writeBoolean(stub.isInner());
         dataStream.writeBoolean(stub.isLocal());
+        dataStream.writeBoolean(stub.isTopLevel());
 
         List<String> superNames = stub.getSuperNames();
         dataStream.writeVarInt(superNames.size());
@@ -101,6 +101,7 @@ public class JetClassElementType extends JetStubElementType<PsiJetClassStub, Jet
         boolean isAnnotation = dataStream.readBoolean();
         boolean isInner = dataStream.readBoolean();
         boolean isLocal = dataStream.readBoolean();
+        boolean isTopLevel = dataStream.readBoolean();
 
         int superCount = dataStream.readVarInt();
         StringRef[] superNames = StringRef.createArray(superCount);
@@ -109,7 +110,7 @@ public class JetClassElementType extends JetStubElementType<PsiJetClassStub, Jet
         }
 
         return new PsiJetClassStubImpl(getStubType(isEnumEntry), parentStub, qualifiedName, name, superNames,
-                                       isTrait, isEnumClass, isEnumEntry, isAnnotation, isInner, isLocal);
+                                       isTrait, isEnumClass, isEnumEntry, isAnnotation, isInner, isLocal, isTopLevel);
     }
 
     @Override
diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetClassStubImpl.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetClassStubImpl.java
index 2d949156bea..27aedb3052e 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetClassStubImpl.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/stubs/impl/PsiJetClassStubImpl.java
@@ -41,6 +41,7 @@ public class PsiJetClassStubImpl extends StubBase<JetClass> implements PsiJetCla
     private final boolean isAnnotation;
     private final boolean isInner;
     private final boolean isLocal;
+    private final boolean isTopLevel;
 
     public PsiJetClassStubImpl(
             JetClassElementType type,
@@ -53,7 +54,8 @@ public class PsiJetClassStubImpl extends StubBase<JetClass> implements PsiJetCla
             boolean isEnumEntry,
             boolean isAnnotation,
             boolean isInner,
-            boolean isLocal
+            boolean isLocal,
+            boolean isTopLevel
     ) {
         super(parent, type);
         this.qualifiedName = qualifiedName;
@@ -65,6 +67,7 @@ public class PsiJetClassStubImpl extends StubBase<JetClass> implements PsiJetCla
         this.isAnnotation = isAnnotation;
         this.isInner = isInner;
         this.isLocal = isLocal;
+        this.isTopLevel = isTopLevel;
     }
 
     @Override
@@ -121,6 +124,11 @@ public class PsiJetClassStubImpl extends StubBase<JetClass> implements PsiJetCla
         return result;
     }
 
+    @Override
+    public boolean isTopLevel() {
+        return isTopLevel;
+    }
+
     @Override
     public String toString() {
         StringBuilder builder = new StringBuilder();