[SIR] Add SirTransformerVoid class

This is convenient for cases when the transformer has no context to pass
to each method
This commit is contained in:
Sergej Jaskiewicz
2023-12-11 19:41:19 +01:00
committed by Space Team
parent 1bc2ac2c87
commit 1bed6a063f
9 changed files with 304 additions and 34 deletions
@@ -29,6 +29,7 @@ fun main(args: Array<String>) {
elementVisitorType to ::VisitorPrinter,
elementVisitorVoidType to ::VisitorVoidPrinter,
elementTransformerType to ::TransformerPrinter.bind(model.rootElement),
elementTransformerVoidType to ::TransformerVoidPrinter,
),
ImplementationConfigurator,
BuilderConfigurator(model.elements),
@@ -20,6 +20,7 @@ private const val VISITORS_PACKAGE = "$BASE_PACKAGE.visitors"
val elementVisitorType = type(VISITORS_PACKAGE, "SirVisitor", TypeKind.Class)
val elementVisitorVoidType = type(VISITORS_PACKAGE, "SirVisitorVoid", TypeKind.Class)
val elementTransformerType = type(VISITORS_PACKAGE, "SirTransformer", TypeKind.Class)
val elementTransformerVoidType = type(VISITORS_PACKAGE, "SirTransformerVoid", TypeKind.Class)
val swiftIrImplementationDetailAnnotation = type(BASE_PACKAGE, "SirImplementationDetail", TypeKind.Class)
val swiftIrBuilderDslAnnotation = type(BASE_PACKAGE, "SirBuilderDsl", TypeKind.Class)
@@ -7,10 +7,7 @@ package org.jetbrains.kotlin.sir.tree.generator.printer
import org.jetbrains.kotlin.generators.tree.*
import org.jetbrains.kotlin.generators.tree.printer.*
import org.jetbrains.kotlin.sir.tree.generator.SwiftIrTree
import org.jetbrains.kotlin.sir.tree.generator.elementTransformerType
import org.jetbrains.kotlin.sir.tree.generator.elementVisitorType
import org.jetbrains.kotlin.sir.tree.generator.elementVisitorVoidType
import org.jetbrains.kotlin.sir.tree.generator.*
import org.jetbrains.kotlin.sir.tree.generator.model.Element
import org.jetbrains.kotlin.sir.tree.generator.model.Field
import org.jetbrains.kotlin.utils.SmartPrinter
@@ -42,8 +39,12 @@ internal class ElementPrinter(printer: SmartPrinter) : AbstractElementPrinter<El
println()
println()
printAcceptChildrenVoidMethod(elementVisitorVoidType)
println()
printTransformVoidMethod(element, elementTransformerVoidType, treeName)
printTransformChildrenMethod(element, elementTransformerType, StandardTypes.unit)
println()
println()
printTransformChildrenVoidMethod(element, elementTransformerVoidType, StandardTypes.unit)
}
}
}
@@ -0,0 +1,23 @@
/*
* Copyright 2010-2023 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.sir.tree.generator.printer
import org.jetbrains.kotlin.generators.tree.AbstractTransformerVoidPrinter
import org.jetbrains.kotlin.generators.tree.ClassRef
import org.jetbrains.kotlin.generators.tree.PositionTypeParameterRef
import org.jetbrains.kotlin.sir.tree.generator.elementTransformerType
import org.jetbrains.kotlin.sir.tree.generator.model.Element
import org.jetbrains.kotlin.sir.tree.generator.model.Field
import org.jetbrains.kotlin.utils.SmartPrinter
internal class TransformerVoidPrinter(
printer: SmartPrinter,
override val visitorType: ClassRef<*>,
) : AbstractTransformerVoidPrinter<Element, Field>(printer) {
override val transformerSuperClass: ClassRef<PositionTypeParameterRef>
get() = elementTransformerType
}