Research android target support

Author: kx412764776

examples/kotlin-multiplatform/build.gradle

@@ -52,8 +52,7 @@ kotlin {
     linuxX64()
     mingwX64()
 
-    androidTarget {
-    }
+    androidTarget {}
 
     applyDefaultHierarchyTemplate()
 

examples/kotlin-multiplatform/src/androidMain/kotlin/AndroidTestBenchmark.kt

@@ -0,0 +1,26 @@
+package test
+
+import kotlinx.benchmark.*
+import kotlin.math.*
+
+@State(Scope.Benchmark)
+class AndroidTestBenchmark {
+    private var data = 0.0
+
+    @Setup
+    fun setUp() {
+        data = 3.0
+    }
+
+    @Benchmark
+    fun sqrtBenchmark(): Double {
+        return sqrt(data)
+    }
+
+    @Benchmark
+    fun cosBenchmark(): Double {
+        return cos(data)
+    }
+}
+
+

plugin/main/src/kotlinx/benchmark/gradle/BenchmarkConfiguration.kt

@@ -2,6 +2,7 @@ package kotlinx.benchmark.gradle
 
 import kotlinx.benchmark.gradle.internal.KotlinxBenchmarkPluginInternalApi
 import org.gradle.api.tasks.*
+import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinAndroidTarget
 import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinJvmCompilation
 import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinNativeCompilation
 import org.jetbrains.kotlin.gradle.plugin.mpp.NativeBuildType
@@ -126,3 +127,11 @@ constructor(
 ) : BenchmarkTarget(extension, name) {
     var buildType: NativeBuildType = NativeBuildType.RELEASE
 }
+
+class AndroidBenchmarkTarget
+@KotlinxBenchmarkPluginInternalApi
+constructor(
+    extension: BenchmarksExtension,
+    name: String,
+    val target: KotlinAndroidTarget
+) : BenchmarkTarget(extension, name)

plugin/main/src/kotlinx/benchmark/gradle/BenchmarksExtension.kt

@@ -8,9 +8,7 @@ import org.gradle.api.provider.*
 import org.jetbrains.kotlin.gradle.dsl.KotlinMultiplatformExtension
 import org.jetbrains.kotlin.gradle.plugin.KotlinCompilation
 import org.jetbrains.kotlin.gradle.plugin.KotlinPlatformType
-import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinJsCompilation
-import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinJvmCompilation
-import org.jetbrains.kotlin.gradle.plugin.mpp.KotlinNativeCompilation
+import org.jetbrains.kotlin.gradle.plugin.mpp.*
 import org.jetbrains.kotlin.gradle.targets.js.ir.KotlinJsIrCompilation
 
 fun Project.benchmark(configure: Action<BenchmarksExtension>) {
@@ -63,10 +61,17 @@ constructor(
             when {
                 multiplatform != null -> {
                     val target = multiplatform.targets.findByName(name)
+                    println("target: $target")
+                    if (target is KotlinAndroidTarget) {
+                        return@container AndroidBenchmarkTarget(this, name, target)
+                    }
+                    target?.compilations?.findByName(KotlinCompilation.MAIN_COMPILATION_NAME)
+                        .let { println("compilation: $it") }
                     // We allow the name to be either a target or a source set
                     when (val compilation = target?.compilations?.findByName(KotlinCompilation.MAIN_COMPILATION_NAME)
                         ?: multiplatform.targets.flatMap { it.compilations }
-                            .find { it.defaultSourceSet.name == name }) {
+                            .find { it.defaultSourceSet.name == name }
+                    ) {
                         null -> {
                             project.logger.warn("Warning: Cannot find a benchmark compilation '$name', ignoring.")
                             BenchmarkTarget(this, name) // ignore

plugin/main/src/kotlinx/benchmark/gradle/BenchmarksPlugin.kt

@@ -119,6 +119,14 @@ constructor(
                 is JsBenchmarkTarget -> processJsCompilation(config)
                 is WasmBenchmarkTarget -> processWasmCompilation(config)
                 is NativeBenchmarkTarget -> processNativeCompilation(config)
+                // is AndroidBenchmarkTarget -> processAndroidCompilation(config)
+                is AndroidBenchmarkTarget -> {
+                    println("processConfigurations: AndroidBenchmarkTarget")
+                    config.target.compilations.all { compilation ->
+                        // This block is called for each compilation when they are materialized
+                        println("handling compilation: $compilation")
+                    }
+                }
             }
         }
     }