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

[tree generators] Factor out common transformer printing logic

Browse Source
This commit is contained in:
Sergej Jaskiewicz
2023-12-11 18:04:03 +01:00
committed by Space Team
parent 9670de74f1
commit 80c9eae409
10 changed files with 89 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files
native/swift/sir/tree-generator/src/org/jetbrains/kotlin/sir/tree/generator/model/Element.kt
+2 -4
View File
@@ -20,10 +20,8 @@ class Element(name: String, override val propertyName: String) : AbstractElement
override val otherParents: MutableList<ClassRef<*>> = mutableListOf()
override val visitorParameterName: String = when (val parameterName = name.replaceFirstChar(Char::lowercaseChar)) {
"class" -> "klass"
else -> parameterName
}
override val visitorParameterName: String
get() = safeDecapitalizedName
override val hasAcceptMethod: Boolean
get() = true
native/swift/sir/tree-generator/src/org/jetbrains/kotlin/sir/tree/generator/printer/TransformerPrinter.kt
+1 -61
View File
@@ -5,12 +5,7 @@
package org.jetbrains.kotlin.sir.tree.generator.printer
import org.jetbrains.kotlin.descriptors.Modality
import org.jetbrains.kotlin.generators.tree.*
import org.jetbrains.kotlin.generators.tree.printer.FunctionParameter
import org.jetbrains.kotlin.generators.tree.printer.printBlock
import org.jetbrains.kotlin.generators.tree.printer.printFunctionDeclaration
import org.jetbrains.kotlin.generators.tree.printer.printFunctionWithBlockBody
import org.jetbrains.kotlin.sir.tree.generator.elementVisitorType
import org.jetbrains.kotlin.sir.tree.generator.model.Element
import org.jetbrains.kotlin.sir.tree.generator.model.Field
@@ -20,7 +15,7 @@ internal class TransformerPrinter(
printer: SmartPrinter,
override val visitorType: ClassRef<*>,
private val rootElement: Element,
) : AbstractVisitorPrinter<Element, Field>(printer) {
) : AbstractTransformerPrinter<Element, Field>(printer) {
override val visitorSuperType: ClassRef<PositionTypeParameterRef>
get() = elementVisitorType.withArgs(rootElement, dataTypeVariable)
@@ -30,59 +25,4 @@ internal class TransformerPrinter(
override val visitorDataType: TypeRef
get() = dataTypeVariable
override fun visitMethodReturnType(element: Element) = element.transformerClass
context(ImportCollector)
override fun printMethodsForElement(element: Element) {
// FIXME: This code is copy-pasted from the FIR generator. Factor it out.
printer.run {
println()
val elementParameterName = element.visitorParameterName
if (element.isRootElement) {
val elementTP = TypeVariable("E", listOf(element))
printFunctionDeclaration(
name = "transformElement",
parameters = listOf(
FunctionParameter(elementParameterName, elementTP),
FunctionParameter("data", dataTypeVariable)
),
returnType = elementTP,
typeParameters = listOf(elementTP),
modality = Modality.ABSTRACT,
)
println()
} else {
val parentInVisitor = element.parentInVisitor ?: return
printFunctionWithBlockBody(
name = "transform" + element.name,
parameters = listOf(
FunctionParameter(elementParameterName, element),
FunctionParameter("data", dataTypeVariable)
),
returnType = visitMethodReturnType(element),
typeParameters = element.params,
modality = Modality.OPEN,
) {
println("return ", "transform", parentInVisitor.name, "(", elementParameterName, ", data)")
}
}
println()
printVisitMethodDeclaration(
element = element,
modality = Modality.FINAL,
override = true,
)
printBlock {
println(
"return transform",
element.name,
"(",
elementParameterName,
", ",
"data)"
)
}
}
}
}