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

[IR] Autogenerate IrFunctionWithLateBindingImpl

Browse Source 
^KT-65773 In Progress
This commit is contained in:
Wojciech Litewka
2024-02-16 12:42:01 +01:00
committed by Space Team
parent d5b78c0cdc
commit d97681f2a9
2 changed files with 105 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/ir/ir.tree/tree-generator/src/org/jetbrains/kotlin/ir/generator/ImplementationConfigurator.kt
+55 -1
View File
@@ -5,10 +5,20 @@
package org.jetbrains.kotlin.ir.generator
import org.jetbrains.kotlin.generators.tree.ArbitraryImportable
import org.jetbrains.kotlin.generators.tree.ClassRef
import org.jetbrains.kotlin.generators.tree.ImportCollector
import org.jetbrains.kotlin.generators.tree.StandardTypes
import org.jetbrains.kotlin.generators.tree.Visibility
import org.jetbrains.kotlin.generators.tree.printer.FunctionParameter
import org.jetbrains.kotlin.generators.tree.printer.VariableKind
import org.jetbrains.kotlin.generators.tree.printer.printFunctionWithBlockBody
import org.jetbrains.kotlin.generators.tree.printer.printPropertyDeclaration
import org.jetbrains.kotlin.ir.generator.config.AbstractIrTreeImplementationConfigurator
import org.jetbrains.kotlin.ir.generator.model.Element
import org.jetbrains.kotlin.ir.generator.model.Implementation
import org.jetbrains.kotlin.ir.generator.model.ListField
import org.jetbrains.kotlin.utils.SmartPrinter
object ImplementationConfigurator : AbstractIrTreeImplementationConfigurator() {
override fun configure(model: Model): Unit = with(IrTree) {
@@ -23,7 +33,12 @@ object ImplementationConfigurator : AbstractIrTreeImplementationConfigurator() {
isLateinit("returnType")
}
impl(functionWithLateBinding) {
implementation.doPrint = false
defaultEmptyList("valueParameters")
defaultNull("dispatchReceiverParameter", "extensionReceiverParameter", "body", "correspondingPropertySymbol")
default("contextReceiverParametersCount", "0")
isLateinit("returnType")
defaultNull("containerSource", withGetter = true)
configureDeclarationWithLateBindinig(simpleFunctionSymbolType)
}
impl(constructor) {
@@ -90,6 +105,45 @@ object ImplementationConfigurator : AbstractIrTreeImplementationConfigurator() {
}
}
private fun ImplementationContext.configureDeclarationWithLateBindinig(symbolType: ClassRef<*>) {
default("isBound") {
value = "_symbol != null"
withGetter = true
}
default("symbol") {
value = "_symbol ?: error(\"\$this has not acquired a symbol yet\")"
withGetter = true
}
additionalImports(ArbitraryImportable("org.jetbrains.kotlin.ir.descriptors", "toIrBasedDescriptor"))
default("descriptor") {
value = "_symbol?.descriptor ?: this.toIrBasedDescriptor()"
withGetter = true
}
implementation.generationCallback = {
println()
printPropertyDeclaration(
"_symbol",
symbolType.copy(nullable = true),
VariableKind.VAR,
visibility = Visibility.PRIVATE,
initializer = "null"
)
println()
println()
printFunctionWithBlockBody(
"acquireSymbol",
listOf(FunctionParameter("symbol", symbolType)),
implementation.element,
override = true,
) {
println("assert(_symbol == null) { \"\$this already has symbol _symbol\" }")
println("_symbol = symbol")
println("symbol.bind(this)")
println("return this")
}
}
}
override fun configureAllImplementations(model: Model) {
configureFieldInAllImplementations("parent") {
isLateinit("parent")