[FIR/IR generator] Extract visitFunctionName to AbstractElement

compiler/ir/ir.tree/tree-generator/src/org/jetbrains/kotlin/ir/generator/IrTree.kt

@@ -71,7 +71,7 @@ object IrTree : AbstractTreeBuilder() {
         skipInIrFactory()
     }
 
-    override val rootElement: Element by element(Other, name = "element") {
+    override val rootElement: Element by element(Other, name = "Element") {
         hasAcceptMethod = true
         hasTransformMethod = true
         transformByChildren = true
@@ -111,7 +111,7 @@ object IrTree : AbstractTreeBuilder() {
         transformByChildren = true
         transformerReturnType = statement
         parentInVisitor = rootElement
-        visitorName = "declaration"
+        nameInVisitorMethod = "Declaration"
 
         parent(declaration)
     }
@@ -731,7 +731,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val memberAccessExpression: Element by element(Expression) {
         parentInVisitor = declarationReference
-        visitorName = "memberAccess"
+        nameInVisitorMethod = "MemberAccess"
         transformerReturnType = rootElement
         val s = +param("S", symbolType)
 
@@ -813,7 +813,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val functionAccessExpression: Element by element(Expression) {
         parentInVisitor = memberAccessExpression
-        visitorName = "functionAccess"
+        nameInVisitorMethod = "FunctionAccess"
         transformerReturnType = rootElement
 
         parent(memberAccessExpression.withArgs("S" to functionSymbolType))
@@ -834,7 +834,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val getSingletonValue: Element by element(Expression) {
         parentInVisitor = declarationReference
-        visitorName = "SingletonReference"
+        nameInVisitorMethod = "SingletonReference"
 
         parent(declarationReference)
     }
@@ -1092,7 +1092,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val fieldAccessExpression: Element by element(Expression) {
         parentInVisitor = declarationReference
-        visitorName = "fieldAccess"
+        nameInVisitorMethod = "FieldAccess"
         ownsChildren = false
 
         parent(declarationReference)
@@ -1228,7 +1228,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val typeOperatorCall: Element by element(Expression) {
         parentInVisitor = expression
-        visitorName = "typeOperator"
+        nameInVisitorMethod = "TypeOperator"
 
         parent(expression)
 
@@ -1238,7 +1238,7 @@ object IrTree : AbstractTreeBuilder() {
     }
     val valueAccessExpression: Element by element(Expression) {
         parentInVisitor = declarationReference
-        visitorName = "valueAccess"
+        nameInVisitorMethod = "ValueAccess"
 
         parent(declarationReference)
 

compiler/ir/ir.tree/tree-generator/src/org/jetbrains/kotlin/ir/generator/config/AbstractTreeBuilder.kt

@@ -122,7 +122,7 @@ class ElementDelegate(
 
     override fun provideDelegate(thisRef: AbstractTreeBuilder, property: KProperty<*>): ElementDelegate {
         val path = thisRef.javaClass.name + "." + property.name
-        element = Element(name ?: property.name, path, category)
+        element = Element(name ?: property.name.replaceFirstChar(Char::uppercaseChar), path, category)
         return this
     }
 }

compiler/ir/ir.tree/tree-generator/src/org/jetbrains/kotlin/ir/generator/model/Model.kt

@@ -13,10 +13,10 @@ import org.jetbrains.kotlin.generators.tree.ElementOrRef as GenericElementOrRef
 import org.jetbrains.kotlin.generators.tree.ElementRef as GenericElementRef
 
 class Element(
-    override val name: String,
+    name: String,
     override val propertyName: String,
     category: Category,
-) : AbstractElement<Element, Field>() {
+) : AbstractElement<Element, Field>(name) {
 
     enum class Category(private val packageDir: String, val defaultVisitorParam: String) {
         Expression("expressions", "expression"),
@@ -67,8 +67,7 @@ class Element(
 
     override var kind: ImplementationKind? = null
 
-    override val typeName
-        get() = "Ir" + name.replaceFirstChar(Char::uppercaseChar)
+    override val typeName = "Ir$name"
 
     /**
      * Whether this element is semantically a leaf element in the hierarchy.
@@ -85,11 +84,6 @@ class Element(
     override var walkableChildren: List<Field> = emptyList()
     override val transformableChildren get() = walkableChildren.filter { it.transformable }
 
-    var visitorName: String? = null
-
-    override val visitFunctionName: String
-        get() = "visit" + (visitorName ?: name).replaceFirstChar(Char::uppercaseChar)
-
     override var visitorParameterName = category.defaultVisitorParam
 
     override var hasAcceptMethod = false // By default, accept is generated only for leaves.