From f187c4f093413ef52c7930ed81b13f12219a8b1c Mon Sep 17 00:00:00 2001 From: Svetlana Isakova Date: Wed, 14 May 2014 16:59:40 +0400 Subject: [PATCH] Minor: added annotation, used util method --- .../org/jetbrains/jet/lang/psi/JetPrefixExpression.java | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPrefixExpression.java b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPrefixExpression.java index 2345f694f68..160b13afe5f 100644 --- a/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPrefixExpression.java +++ b/compiler/frontend/src/org/jetbrains/jet/lang/psi/JetPrefixExpression.java @@ -17,7 +17,7 @@ package org.jetbrains.jet.lang.psi; import com.intellij.lang.ASTNode; -import com.intellij.psi.PsiElement; +import com.intellij.psi.util.PsiTreeUtil; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; @@ -31,12 +31,9 @@ public class JetPrefixExpression extends JetUnaryExpression { return visitor.visitPrefixExpression(this, data); } + @Override @Nullable @IfNotParsed public JetExpression getBaseExpression() { - PsiElement expression = getOperationReference().getNextSibling(); - while (expression != null && !(expression instanceof JetExpression)) { - expression = expression.getNextSibling(); - } - return (JetExpression) expression; + return PsiTreeUtil.getNextSiblingOfType(getOperationReference(), JetExpression.class); } }