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

FIR: Support fake elements for light tree

Browse Source
This commit is contained in:
Denis Zharkov
2020-07-17 11:53:16 +03:00
parent fea1fb0fc3
commit dfc75f3447
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirSourceElement.kt
+3 -4
View File
@@ -167,7 +167,7 @@ class FirFakeSourceElement<out P : PsiElement>(psi: P, override val kind: FirFak
fun FirSourceElement.fakeElement(newKind: FirFakeSourceElementKind): FirSourceElement { fun FirSourceElement.fakeElement(newKind: FirFakeSourceElementKind): FirSourceElement {
return when (this) { return when (this) {
is FirLightSourceElement -> this is FirLightSourceElement -> FirLightSourceElement(element, startOffset, endOffset, tree, newKind)
is FirPsiSourceElement<*> -> FirFakeSourceElement(psi, newKind) is FirPsiSourceElement<*> -> FirFakeSourceElement(psi, newKind)
} }
} }
@@ -176,12 +176,11 @@ class FirLightSourceElement(
val element: LighterASTNode, val element: LighterASTNode,
override val startOffset: Int, override val startOffset: Int,
override val endOffset: Int, override val endOffset: Int,
val tree: FlyweightCapableTreeStructure<LighterASTNode> val tree: FlyweightCapableTreeStructure<LighterASTNode>,
override val kind: FirSourceElementKind = FirRealSourceElementKind,
) : FirSourceElement() { ) : FirSourceElement() {
override val elementType: IElementType override val elementType: IElementType
get() = element.tokenType get() = element.tokenType
override val kind: FirSourceElementKind get() = FirRealSourceElementKind
} }
val FirSourceElement?.psi: PsiElement? get() = (this as? FirPsiSourceElement<*>)?.psi val FirSourceElement?.psi: PsiElement? get() = (this as? FirPsiSourceElement<*>)?.psi