From 5ad1fac2b299e4473babece4e4cd7d04087073fd Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Wed, 9 Apr 2014 18:00:05 +0400
Subject: [PATCH] JetPropertyAccessor#isGetter(), isSetter(), hasBody(),
 getReturnTypeReference() by stub

---
 .../jet/lang/psi/JetPropertyAccessor.java       | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPropertyAccessor.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPropertyAccessor.java
index b8cd5f85ba4..f4060054992 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPropertyAccessor.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPropertyAccessor.java
@@ -21,7 +21,6 @@ import com.intellij.psi.PsiElement;
 import com.intellij.psi.util.PsiTreeUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
-import org.jetbrains.jet.JetNodeTypes;
 import org.jetbrains.jet.lang.psi.stubs.PsiJetPropertyAccessorStub;
 import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementTypes;
 import org.jetbrains.jet.lexer.JetTokens;
@@ -45,16 +44,24 @@ public class JetPropertyAccessor extends JetDeclarationStub<PsiJetPropertyAccess
     }
 
     public boolean isSetter() {
+        PsiJetPropertyAccessorStub stub = getStub();
+        if (stub != null) {
+            return !stub.isGetter();
+        }
         return findChildByType(JetTokens.SET_KEYWORD) != null;
     }
 
     public boolean isGetter() {
+        PsiJetPropertyAccessorStub stub = getStub();
+        if (stub != null) {
+            return stub.isGetter();
+        }
         return findChildByType(JetTokens.GET_KEYWORD) != null;
     }
 
     @Nullable
     public JetParameter getParameter() {
-        JetParameterList parameterList = (JetParameterList) findChildByType(JetNodeTypes.VALUE_PARAMETER_LIST);
+        JetParameterList parameterList = getStubOrPsiChild(JetStubElementTypes.VALUE_PARAMETER_LIST);
         if (parameterList == null) return null;
         List<JetParameter> parameters = parameterList.getParameters();
         if (parameters.isEmpty()) return null;
@@ -84,6 +91,10 @@ public class JetPropertyAccessor extends JetDeclarationStub<PsiJetPropertyAccess
 
     @Override
     public boolean hasBody() {
+        PsiJetPropertyAccessorStub stub = getStub();
+        if (stub != null) {
+            return stub.hasBody();
+        }
         return getBodyExpression() != null;
     }
 
@@ -105,7 +116,7 @@ public class JetPropertyAccessor extends JetDeclarationStub<PsiJetPropertyAccess
 
     @Nullable
     public JetTypeReference getReturnTypeReference() {
-        return findChildByClass(JetTypeReference.class);
+        return getStubOrPsiChild(JetStubElementTypes.TYPE_REFERENCE);
     }
 
     @NotNull