[FIR] Pass all FirFiles to FirGlobalResolveProcessor.process

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirResolveProcessor.kt

@@ -16,7 +16,7 @@ annotation class AdapterForResolveProcessor
 sealed class FirResolveProcessor(val session: FirSession, val scopeSession: ScopeSession)
 
 abstract class FirGlobalResolveProcessor(session: FirSession, scopeSession: ScopeSession) : FirResolveProcessor(session, scopeSession) {
-    abstract fun process()
+    abstract fun process(files: Collection<FirFile>)
 }
 
 abstract class FirTransformerBasedResolveProcessor(
@@ -28,4 +28,4 @@ abstract class FirTransformerBasedResolveProcessor(
     fun processFile(file: FirFile) {
         file.transform<FirFile, Nothing?>(transformer, null)
     }
-}
+}

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/FirTotalResolveProcessor.kt

@@ -24,7 +24,7 @@ class FirTotalResolveProcessor(session: FirSession) {
                     }
                 }
                 is FirGlobalResolveProcessor -> {
-                    processor.process()
+                    processor.process(files)
                 }
             }
         }

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/plugin/FirGlobalClassGenerationProcessor.kt

@@ -20,7 +20,7 @@ class FirGlobalClassGenerationProcessor(
     session: FirSession,
     scopeSession: ScopeSession
 ) : FirGlobalResolveProcessor(session, scopeSession) {
-    override fun process() {
+    override fun process(files: Collection<FirFile>) {
         val extensions = session.extensionService.declarationGenerators
         if (extensions.isEmpty()) return
         val provider = session.predicateBasedProvider

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/plugin/FirGlobalExtensionStatusProcessor.kt

@@ -24,7 +24,7 @@ class FirGlobalExtensionStatusProcessor(
     session: FirSession,
     scopeSession: ScopeSession
 ) : FirGlobalResolveProcessor(session, scopeSession) {
-    override fun process() {
+    override fun process(files: Collection<FirFile>) {
         val extensions = session.extensionService.statusTransformerExtensions
         if (extensions.isEmpty()) return
         val provider = session.predicateBasedProvider

compiler/fir/resolve/src/org/jetbrains/kotlin/fir/resolve/transformers/plugin/FirGlobalNewMemberGenerationProcessor.kt

@@ -26,7 +26,7 @@ class FirGlobalNewMemberGenerationProcessor(
     private val index = session.generatedClassIndex
     private val provider = session.predicateBasedProvider
 
-    override fun process() {
+    override fun process(files: Collection<FirFile>) {
         val extensions = session.extensionService.declarationGenerators
         if (extensions.isEmpty()) return
         for (extension in extensions) {
@@ -64,4 +64,4 @@ class FirGlobalNewMemberGenerationProcessor(
             }
         }
     }
-}
+}

compiler/tests-common/tests/org/jetbrains/kotlin/fir/FirResolveBench.kt

@@ -159,7 +159,7 @@ class FirResolveBench(val withProgress: Boolean) {
     private fun runStage(processor: FirResolveProcessor, firFileSequence: Sequence<FirFile>) {
         when (processor) {
             is FirTransformerBasedResolveProcessor -> runStage(processor, firFileSequence)
-            is FirGlobalResolveProcessor -> runStage(processor)
+            is FirGlobalResolveProcessor -> runStage(processor, firFileSequence.toList())
         }
     }
 
@@ -182,10 +182,10 @@ class FirResolveBench(val withProgress: Boolean) {
         }
     }
 
-    private fun runStage(processor: FirGlobalResolveProcessor) {
+    private fun runStage(processor: FirGlobalResolveProcessor, firFiles: List<FirFile>) {
         processWithTimeMeasure(
             processor::class,
-            { processor.process() }
+            { processor.process(firFiles) }
         ) { e ->
             val message = "Fail on stage ${processor::class}"
             println(message)