Tianyu Geng
fb14b03824
Currently, FIR reports errors caused by previous resolution failure. For example with unresolved `a` and `b` in code `a.b`, both `a` and `b` are highlighted. FE1.0 only highlights `a` since it's the root cause. This change applies this heuristics when reporting FirDiagnostics.
41 lines
2.0 KiB
Kotlin
Vendored
41 lines
2.0 KiB
Kotlin
Vendored
package org.jetbrains.kotlin.codegen.range.inExpression
|
|
|
|
interface ExpressionCodegen
|
|
interface KtSimpleNameExpression
|
|
interface InExpressionGenerator
|
|
interface StackValue
|
|
open class BranchedValue
|
|
interface Type
|
|
interface KotlinType
|
|
interface Label
|
|
interface InstructionAdapter
|
|
|
|
class CallBasedInExpressionGenerator(
|
|
val codegen: ExpressionCodegen,
|
|
operatorReference: KtSimpleNameExpression
|
|
) : InExpressionGenerator {
|
|
private val resolvedCall = operatorReference.<!UNRESOLVED_REFERENCE!>getResolvedCallWithAssert<!>(codegen.<!UNRESOLVED_REFERENCE!>bindingContext<!>)
|
|
private val isInverted = operatorReference.<!UNRESOLVED_REFERENCE!>getReferencedNameElementType<!>() == <!UNRESOLVED_REFERENCE!>KtTokens<!>.NOT_IN
|
|
|
|
override fun generate(argument: StackValue): BranchedValue =
|
|
gen(argument).let { if (isInverted) <!UNRESOLVED_REFERENCE!>Invert<!>(it) else it }
|
|
|
|
private fun gen(argument: StackValue): BranchedValue =
|
|
object : <!INAPPLICABLE_CANDIDATE{LT}!><!INAPPLICABLE_CANDIDATE{PSI}!>BranchedValue<!>(argument, null, argument.<!UNRESOLVED_REFERENCE!>type<!>, <!UNRESOLVED_REFERENCE!>Opcodes<!>.IFEQ)<!> {
|
|
override fun putSelector(type: Type, kotlinType: KotlinType?, v: InstructionAdapter) {
|
|
invokeFunction(v)
|
|
<!UNRESOLVED_REFERENCE!>coerceTo<!>(type, kotlinType, v)
|
|
}
|
|
|
|
override fun condJump(jumpLabel: Label, v: InstructionAdapter, jumpIfFalse: Boolean) {
|
|
invokeFunction(v)
|
|
v.<!UNRESOLVED_REFERENCE!>visitJumpInsn<!>(if (jumpIfFalse) <!UNRESOLVED_REFERENCE!>Opcodes<!>.IFEQ else <!UNRESOLVED_REFERENCE!>Opcodes<!>.IFNE, jumpLabel)
|
|
}
|
|
|
|
private fun invokeFunction(v: InstructionAdapter) {
|
|
val result = codegen.<!UNRESOLVED_REFERENCE!>invokeFunction<!>(resolvedCall.<!UNRESOLVED_REFERENCE!>call<!>, resolvedCall, <!UNRESOLVED_REFERENCE!>none<!>())
|
|
result.<!UNRESOLVED_REFERENCE!>put<!>(result.<!UNRESOLVED_REFERENCE!>type<!>, result.<!UNRESOLVED_REFERENCE!>kotlinType<!>, v)
|
|
}
|
|
}
|
|
}
|