Minor: added annotation, used util method

This commit is contained in:
Svetlana Isakova
2014-05-14 16:59:40 +04:00
parent 07c0d87a8f
commit f187c4f093
@@ -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);
}
}