azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

JetNamedFunction#getReceiverTypeRef and getReturnTypeRef by stub

Browse Source
This commit is contained in:
Pavel V. Talanov
2014-03-26 16:17:19 +04:00
parent 56e3500eaa
commit 581ecda0f2
1 changed files with 33 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/frontend/src/org/jetbrains/jet/lang/psi/JetNamedFunction.java
+33 -2
View File
@@ -24,7 +24,6 @@ import com.intellij.psi.tree.IElementType;
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.PsiJetFunctionStub;
import org.jetbrains.jet.lang.psi.stubs.elements.JetStubElementTypes;
import org.jetbrains.jet.lexer.JetTokens;
@@ -82,7 +81,7 @@ public class JetNamedFunction extends JetTypeParameterListOwnerStub<PsiJetFuncti
@Override
@Nullable
public JetParameterList getValueParameterList() {
return (JetParameterList) findChildByType(JetNodeTypes.VALUE_PARAMETER_LIST);
return getStubOrPsiChild(JetStubElementTypes.VALUE_PARAMETER_LIST);
}
@Override
@@ -106,6 +105,24 @@ public class JetNamedFunction extends JetTypeParameterListOwnerStub<PsiJetFuncti
@Override
@Nullable
public JetTypeReference getReceiverTypeRef() {
PsiJetFunctionStub stub = getStub();
if (stub != null) {
if (!stub.isExtension()) {
return null;
}
List<JetTypeReference> childTypeReferences = getStubOrPsiChildrenAsList(JetStubElementTypes.TYPE_REFERENCE);
if (!childTypeReferences.isEmpty()) {
return childTypeReferences.get(0);
}
else {
return null;
}
}
return getReceiverTypeRefByTree();
}
@Nullable
private JetTypeReference getReceiverTypeRefByTree() {
PsiElement child = getFirstChild();
while (child != null) {
IElementType tt = child.getNode().getElementType();
@@ -122,6 +139,20 @@ public class JetNamedFunction extends JetTypeParameterListOwnerStub<PsiJetFuncti
@Override
@Nullable
public JetTypeReference getReturnTypeRef() {
PsiJetFunctionStub stub = getStub();
if (stub != null) {
List<JetTypeReference> typeReferences = getStubOrPsiChildrenAsList(JetStubElementTypes.TYPE_REFERENCE);
int returnTypeIndex = stub.isExtension() ? 1 : 0;
if (returnTypeIndex >= typeReferences.size()) {
return null;
}
return typeReferences.get(returnTypeIndex);
}
return getReturnTypeRefByPsi();
}
@Nullable
private JetTypeReference getReturnTypeRefByPsi() {
boolean colonPassed = false;
PsiElement child = getFirstChild();
while (child != null) {