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

FIR: generate FirDefaultVisitor and FirDefaultVisitorVoid...

Browse Source 
so they won't be outdated when FIR class tree is updated.
This commit is contained in:
Tianyu Geng
2021-09-28 16:20:06 -07:00
committed by TeamCityServer
parent 8f3b06ac06
commit 263f6b1fd8
7 changed files with 876 additions and 206 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/printer/main.kt
+3 -1
View File
@@ -27,8 +27,10 @@ fun generateElements(builder: AbstractFirTreeBuilder, generationPath: File): Lis
builder.elements.flatMap { it.allImplementations }.mapNotNull { it.builder }.mapTo(generatedFiles) { it.generateCode(generationPath) }
builder.intermediateBuilders.mapTo(generatedFiles) { it.generateCode(generationPath) }
generatedFiles += printVisitor(builder.elements, generationPath)
generatedFiles += printVisitor(builder.elements, generationPath, false)
generatedFiles += printVisitorVoid(builder.elements, generationPath)
generatedFiles += printVisitor(builder.elements, generationPath, true)
generatedFiles += printDefaultVisitorVoid(builder.elements, generationPath)
generatedFiles += printTransformer(builder.elements, generationPath)
return generatedFiles
}
compiler/fir/tree/tree-generator/src/org/jetbrains/kotlin/fir/tree/generator/printer/visitor.kt
+53 -5
View File
@@ -11,9 +11,10 @@ import org.jetbrains.kotlin.util.SmartPrinter
import org.jetbrains.kotlin.util.withIndent
import java.io.File
fun printVisitor(elements: List<Element>, generationPath: File): GeneratedFile {
fun printVisitor(elements: List<Element>, generationPath: File, visitSuperTypeByDefault: Boolean): GeneratedFile {
val className = if (visitSuperTypeByDefault) "FirDefaultVisitor" else "FirVisitor"
val dir = File(generationPath, VISITOR_PACKAGE.replace(".", "/"))
val file = File(dir, "FirVisitor.kt")
val file = File(dir, "$className.kt")
val stringBuilder = StringBuilder()
SmartPrinter(stringBuilder).apply {
printCopyright()
@@ -23,15 +24,32 @@ fun printVisitor(elements: List<Element>, generationPath: File): GeneratedFile {
println()
printGeneratedMessage()
println("abstract class FirVisitor<out R, in D> {")
print("abstract class $className<out R, in D> ")
if (visitSuperTypeByDefault) {
print(": FirVisitor<R, D>() ")
}
println("{")
pushIndent()
println("abstract fun visitElement(element: FirElement, data: D): R\n")
if (!visitSuperTypeByDefault) {
println("abstract fun visitElement(element: FirElement, data: D): R\n")
}
for (element in elements) {
if (element == AbstractFirTreeBuilder.baseFirElement) continue
with(element) {
val varName = safeDecapitalizedName
println("open fun ${typeParameters}visit$name($varName: $typeWithArguments, data: D): R${multipleUpperBoundsList()} = visitElement($varName, data)")
if (visitSuperTypeByDefault) {
print("override")
} else {
print("open")
}
print(" fun ${typeParameters}visit$name($varName: $typeWithArguments, data: D): R${multipleUpperBoundsList()} = visit")
if (visitSuperTypeByDefault) {
print(parents.first().name)
} else {
print("Element")
}
println("($varName, data)")
println()
}
}
@@ -88,3 +106,33 @@ fun printVisitorVoid(elements: List<Element>, generationPath: File): GeneratedFi
}
return GeneratedFile(file, stringBuilder.toString())
}
fun printDefaultVisitorVoid(elements: List<Element>, generationPath: File): GeneratedFile {
val className = "FirDefaultVisitorVoid"
val dir = File(generationPath, VISITOR_PACKAGE.replace(".", "/"))
val file = File(dir, "$className.kt")
val stringBuilder = StringBuilder()
SmartPrinter(stringBuilder).apply {
printCopyright()
println("package $VISITOR_PACKAGE")
println()
elements.forEach { println("import ${it.fullQualifiedName}") }
println()
printGeneratedMessage()
println("abstract class $className : FirVisitorVoid() {")
pushIndent()
for (element in elements) {
if (element == AbstractFirTreeBuilder.baseFirElement) continue
with(element) {
val varName = safeDecapitalizedName
println("override fun ${typeParameters}visit$name($varName: $typeWithArguments)${multipleUpperBoundsList()} = visit${parents.first().name}($varName)")
println()
}
}
popIndent()
println("}")
}
return GeneratedFile(file, stringBuilder.toString())
}