Add ability to specify frame counts to measure as a benchmarks launch parameter (#5204)

## Release Notes
N/A

Author: pjBooms

benchmarks/multiplatform/benchmarks/src/commonMain/kotlin/Args.kt

@@ -7,12 +7,21 @@ enum class Mode {
 object Args {
     private val modes = mutableSetOf<Mode>()
 
-    private val benchmarks = mutableSetOf<String>()
+    private val benchmarks = mutableMapOf<String, Int>()
 
     private fun argToSet(arg: String): Set<String> = arg.substring(arg.indexOf('=') + 1)
         .split(",").filter{!it.isEmpty()}.map{it.uppercase()}.toSet()
 
 
+    private fun argToMap(arg: String): Map<String, Int> = arg.substring(arg.indexOf('=') + 1)
+        .split(",").filter { !it.isEmpty() }.map { it.uppercase() }.associate {
+            if (it.contains('(') && it.contains(')')) {
+                it.substringBefore('(') to it.substringAfter('(').substringBefore(')').toInt()
+            } else {
+                it to -1
+            }
+        }
+
     /**
      * Parses command line arguments to determine modes and benchmarks settings.
      *
@@ -25,12 +34,17 @@ object Args {
             if (arg.startsWith("modes=", ignoreCase = true)) {
                 modes.addAll(argToSet(arg).map { Mode.valueOf(it) })
             } else if (arg.startsWith("benchmarks=", ignoreCase = true)) {
-                benchmarks.addAll(argToSet(arg))
+                benchmarks += argToMap(arg)
             }
         }
     }
 
     fun isModeEnabled(mode: Mode): Boolean = modes.isEmpty() || modes.contains(mode)
 
     fun isBenchmarkEnabled(benchmark: String): Boolean = benchmarks.isEmpty() || benchmarks.contains(benchmark.uppercase())
+
+    fun getBenchmarkProblemSize(benchmark: String, default: Int): Int {
+        val result = benchmarks[benchmark.uppercase()]?: -1
+        return if (result == -1) default else result
+    }
 }

benchmarks/multiplatform/benchmarks/src/commonMain/kotlin/Benchmarks.kt

@@ -148,7 +148,7 @@ suspend fun runBenchmark(
 ) {
     if (Args.isBenchmarkEnabled(name)) {
         println("$name:")
-        val stats = measureComposable(warmupCount, frameCount, width, height, targetFps, graphicsContext, content).generateStats()
+        val stats = measureComposable(warmupCount, Args.getBenchmarkProblemSize(name, frameCount), width, height, targetFps, graphicsContext, content).generateStats()
         stats.prettyPrint()
     }
 }