From be7f5d01bbc09bcb77b085c4cdb58e21b6d4bcde Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Wed, 21 Jan 2015 17:02:08 +0300
Subject: [PATCH] Update JetObjectDeclaration after parser changes to parser

---
 .../kotlin/psi/JetObjectDeclaration.java      | 27 +++++--------------
 1 file changed, 6 insertions(+), 21 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/kotlin/psi/JetObjectDeclaration.java b/compiler/frontend/src/org/jetbrains/kotlin/psi/JetObjectDeclaration.java
index 3d1b070316b..b0b1096b3d7 100644
--- a/compiler/frontend/src/org/jetbrains/kotlin/psi/JetObjectDeclaration.java
+++ b/compiler/frontend/src/org/jetbrains/kotlin/psi/JetObjectDeclaration.java
@@ -86,32 +86,12 @@ public class JetObjectDeclaration extends JetNamedDeclarationStub<KotlinObjectSt
         return (JetObjectDeclarationName) findChildByType(JetNodeTypes.OBJECT_DECLARATION_NAME);
     }
 
-    @Override
-    @Nullable
-    public JetModifierList getModifierList() {
-        if (isClassObject()) {
-            PsiElement parent = getParentByStub();
-            assert parent instanceof JetDeclaration;
-            return ((JetDeclaration)parent).getModifierList();
-        }
-        return super.getModifierList();
-    }
-
     public boolean isClassObject() {
         KotlinObjectStub stub = getStub();
         if (stub != null) {
             return stub.isClassObject();
         }
-        PsiElement parent = getParent();
-        return parent != null && parent.getNode().getElementType().equals(JetNodeTypes.CLASS_OBJECT);
-    }
-
-    @Nullable
-    public JetClassObject getClassObjectElement() {
-        if (!isClassObject()) {
-            return null;
-        }
-        return (JetClassObject) getParentByStub();
+        return getClassKeyword() != null;
     }
 
     @Override
@@ -188,6 +168,11 @@ public class JetObjectDeclaration extends JetNamedDeclarationStub<KotlinObjectSt
         return findChildByType(JetTokens.OBJECT_KEYWORD);
     }
 
+    @Nullable
+    public PsiElement getClassKeyword() {
+        return findChildByType(JetTokens.CLASS_KEYWORD);
+    }
+
     @Override
     public void delete() throws IncorrectOperationException {
         JetPsiUtil.deleteClass(this);