[FIR] Fix FirExpressionWithSmartcastImpl.acceptChildren

This commit is contained in:
Dmitriy Novozhilov
2021-04-22 17:36:42 +03:00
committed by TeamCityServer
parent 12331cb10c
commit 1afb844e1a
2 changed files with 6 additions and 3 deletions
@@ -44,7 +44,7 @@ internal class FirExpressionWithSmartcastImpl(
} }
override fun <R, D> acceptChildren(visitor: FirVisitor<R, D>, data: D) { override fun <R, D> acceptChildren(visitor: FirVisitor<R, D>, data: D) {
originalExpression.acceptChildren(visitor, data) originalExpression.accept(visitor, data)
} }
override fun <D> transformCalleeReference(transformer: FirTransformer<D>, data: D): FirExpressionWithSmartcast { override fun <D> transformCalleeReference(transformer: FirTransformer<D>, data: D): FirExpressionWithSmartcast {
@@ -146,8 +146,11 @@ class FirDiagnosticsHandler(testServices: TestServices) : FirAnalysisHandler(tes
) )
) )
} }
if (element is FirExpressionWithSmartcast) {
element.acceptChildren(this) element.originalExpression.acceptChildren(this)
} else {
element.acceptChildren(this)
}
} }
override fun visitFunctionCall(functionCall: FirFunctionCall) { override fun visitFunctionCall(functionCall: FirFunctionCall) {