Edit light tree to FIR benchmarks to support off stub mode

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/AbstractBenchmark.kt

@@ -24,6 +24,9 @@ abstract class AbstractBenchmark {
     private val files = mutableMapOf<File, String>()
     abstract val generator: TreeGenerator
 
+    @Param("true", "false")
+    var stubMode: Boolean = false
+
     protected fun readFiles(ignoreTestData: Boolean, path: String) {
         println("BASE PATH: $path")
         val saveText: (File) -> Unit = { file ->

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/AbstractBenchmarkForGivenPath.kt

@@ -30,6 +30,6 @@ abstract class AbstractBenchmarkForGivenPath(
 
     @Benchmark
     fun testBuildFirForGivenPath() {
-        forEachFile { text, file -> generator.generateFir(text, file) }
+        forEachFile { text, file -> generator.generateFir(text, file, stubMode) }
     }
 }

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/createdFiles/AbstractKotlinBenchmark.kt

@@ -53,7 +53,7 @@ abstract class AbstractKotlinBenchmark : AbstractBenchmark() {
 
     @Benchmark
     fun testCreatedFiles() {
-        generator.generateFir(text, file)
+        generator.generateFir(text, file, stubMode)
     }
 }
 

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/generators/LightTree2FirGenerator.kt

@@ -16,25 +16,20 @@ import java.io.File
 
 @State(Scope.Benchmark)
 open class LightTree2FirGenerator : TreeGenerator, AbstractRawFirBuilderTestCase() {
-    private lateinit var lightTreeConverter: LightTree2Fir
-
     override fun generateBaseTree(text: String, file: File) {
+        val lightTreeConverter = LightTree2Fir(false, myProject)
         val lightTree = lightTreeConverter.buildLightTree(text)
         DebugUtil.lightTreeToString(lightTree, false)
     }
 
-    override fun generateFir(text: String, file: File) {
+    override fun generateFir(text: String, file: File, stubMode: Boolean) {
+        val lightTreeConverter = LightTree2Fir(stubMode, myProject)
         val firFile = lightTreeConverter.buildFirFile(text, file.name)
         StringBuilder().also { FirRenderer(it).visitFile(firFile) }.toString()
     }
 
-    private fun createConverter() {
-        lightTreeConverter = LightTree2Fir(true, myProject)
-    }
-
     override fun setUp() {
         super.setUp()
-        createConverter()
     }
 
     override fun tearDown() {

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/generators/Psi2FirGenerator.kt

@@ -22,9 +22,9 @@ open class Psi2FirGenerator : TreeGenerator, AbstractRawFirBuilderTestCase() {
         DebugUtil.psiTreeToString(ktFile, false)
     }
 
-    override fun generateFir(text: String, file: File) {
+    override fun generateFir(text: String, file: File, stubMode: Boolean) {
         val ktFile = createPsiFile(FileUtil.getNameWithoutExtension(PathUtil.getFileName(file.path)), text) as KtFile
-        val firFile = ktFile.toFirFile(stubMode = true)
+        val firFile = ktFile.toFirFile(stubMode)
         StringBuilder().also { FirRenderer(it).visitFile(firFile) }.toString()
     }
 

compiler/fir/lightTree/tests/org/jetbrains/kotlin/fir/lightTree/benchmark/generators/TreeGenerator.kt

@@ -9,7 +9,7 @@ import java.io.File
 
 interface TreeGenerator {
     fun generateBaseTree(text: String, file: File)
-    fun generateFir(text: String, file: File)
+    fun generateFir(text: String, file: File, stubMode: Boolean)
     fun setUp()
     fun tearDown()
 }