[IR] Add special preprocessors for ir interpreter

These preprocessors allow us to modify IR at first and only after that try to evaluate. With this we can drop `KCallableNamePropertyLowering`.
2023-04-28 13:18:06 +02:00
parent c8371a339f
commit e58e20fc7b
10 changed files with 126 additions and 115 deletions
@@ -1,85 +0,0 @@
-/*
- * Copyright 2010-2022 JetBrains s.r.o. and Kotlin Programming Language contributors.
- * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
- */
-
-package org.jetbrains.kotlin.backend.common.lower
-
-import org.jetbrains.kotlin.backend.common.BackendContext
-import org.jetbrains.kotlin.backend.common.FileLoweringPass
-import org.jetbrains.kotlin.backend.common.phaser.makeIrFilePhase
-import org.jetbrains.kotlin.ir.declarations.IrClass
-import org.jetbrains.kotlin.ir.declarations.IrFile
-import org.jetbrains.kotlin.ir.declarations.IrProperty
-import org.jetbrains.kotlin.ir.declarations.IrSimpleFunction
-import org.jetbrains.kotlin.ir.expressions.*
-import org.jetbrains.kotlin.ir.expressions.impl.IrCompositeImpl
-import org.jetbrains.kotlin.ir.expressions.impl.IrConstImpl
-import org.jetbrains.kotlin.ir.util.isSubclassOf
-import org.jetbrains.kotlin.ir.util.render
-import org.jetbrains.kotlin.ir.visitors.IrElementTransformerVoid
-import org.jetbrains.kotlin.ir.visitors.transformChildrenVoid
-import org.jetbrains.kotlin.name.Name
-
-val kCallableNamePropertyPhase = makeIrFilePhase(
-    ::KCallableNamePropertyLowering,
-    name = "KCallableNameProperty",
-    description = "Replace name references for callables with constants"
-)
-
-private class KCallableNamePropertyLowering(val context: BackendContext) : FileLoweringPass {
-    override fun lower(irFile: IrFile) {
-        irFile.transformChildrenVoid(KCallableNamePropertyTransformer(this))
-    }
-}
-
-private class KCallableNamePropertyTransformer(val lower: KCallableNamePropertyLowering) : IrElementTransformerVoid() {
-    private fun nameForCallableMember(reference: IrCallableReference<*>): Name {
-        return when (reference) {
-            is IrFunctionReference -> reference.symbol.owner.name
-            is IrPropertyReference -> reference.symbol.owner.name
-            is IrLocalDelegatedPropertyReference -> reference.symbol.owner.name
-            else -> error("Unexpected callable reference type ${reference.render()}")
-        }
-    }
-
-    override fun visitCall(expression: IrCall): IrExpression {
-        val callableReference = expression.dispatchReceiver as? IrCallableReference<*>
-            ?: return super.visitCall(expression)
-
-        val directMember = expression.symbol.owner.let {
-            (it as? IrSimpleFunction)?.correspondingPropertySymbol?.owner ?: it
-        }
-
-        val irClass = directMember.parent as? IrClass
-            ?: return super.visitCall(expression)
-        if (!irClass.isSubclassOf(lower.context.irBuiltIns.kCallableClass.owner)) {
-            return super.visitCall(expression)
-        }
-
-        val name = when (directMember) {
-            is IrSimpleFunction -> directMember.name
-            is IrProperty -> directMember.name
-            else -> throw AssertionError("Should be IrSimpleFunction or IrProperty, got $directMember")
-        }
-        if (name.asString() != "name") return expression
-
-        val receiver = callableReference.dispatchReceiver ?: callableReference.extensionReceiver
-
-        return IrCompositeImpl(expression.startOffset, expression.endOffset, lower.context.irBuiltIns.stringType).apply {
-            receiver?.let {
-                //put receiver for bound callable reference
-                statements.add(it)
-            }
-
-            statements.add(
-                IrConstImpl.string(
-                    expression.startOffset,
-                    expression.endOffset,
-                    lower.context.irBuiltIns.stringType,
-                    nameForCallableMember(callableReference).asString()
-                )
-            )
-        }
-    }
-}