Support GPU on Windows

Author: igordmn

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

@@ -154,11 +154,11 @@ suspend fun runBenchmark(
     }
 }
 
-suspend fun runBenchmarks(
+private suspend fun runBenchmarks(
     width: Int = 1920,
     height: Int = 1080,
     targetFps: Int = 120,
-    graphicsContext: GraphicsContext? = null
+    graphicsContext: GraphicsContext?
 ) {
     println()
     println("Running emulating $targetFps FPS")
@@ -168,4 +168,14 @@ suspend fun runBenchmarks(
     runBenchmark("VisualEffects", width, height, targetFps, 1000, graphicsContext) { NYContent(width, height) }
     runBenchmark("LazyList", width, height, targetFps, 1000, graphicsContext) { MainUiNoImageUseModel()}
     runBenchmark("Example1", width, height, targetFps, 1000, graphicsContext) { Example1() }
-}
+}
+
+suspend fun benchmarksMain(
+    graphicsContext: GraphicsContext? = null
+) {
+    try {
+        runBenchmarks(graphicsContext = graphicsContext)
+    } finally {
+        graphicsContext?.close()
+    }
+}

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

@@ -17,6 +17,8 @@ interface GraphicsContext {
     fun surface(width: Int, height: Int): Surface
 
     suspend fun awaitGPUCompletion()
+
+    fun close() = Unit
 }
 
 expect fun runGC()

benchmarks/multiplatform/benchmarks/src/desktopMain/kotlin/GraphicsContext.desktop.kt

@@ -0,0 +1,46 @@
+import org.jetbrains.skia.ColorSpace
+import org.jetbrains.skia.PixelGeometry
+import org.jetbrains.skia.Surface
+import org.jetbrains.skia.SurfaceColorFormat
+import org.jetbrains.skia.SurfaceOrigin
+import org.jetbrains.skia.SurfaceProps
+import org.jetbrains.skiko.ExperimentalSkikoApi
+import org.jetbrains.skiko.graphicapi.DirectXOffscreenContext
+import org.jetbrains.skiko.hostOs
+
+@OptIn(ExperimentalSkikoApi::class)
+fun graphicsContext(): GraphicsContext? = when {
+    hostOs.isWindows -> DirectXGraphicsContext()
+    else -> {
+        println("Unsupported desktop host OS: $hostOs. Using non-GPU graphic context")
+        null
+    }
+}
+
+@OptIn(ExperimentalSkikoApi::class)
+class DirectXGraphicsContext() : GraphicsContext {
+    private val context = DirectXOffscreenContext()
+    private var texture: DirectXOffscreenContext.Texture? = null
+
+    override fun surface(width: Int, height: Int): Surface {
+        texture?.close()
+        texture = context.Texture(width, height)
+        return Surface.makeFromBackendRenderTarget(
+            context.directContext,
+            texture!!.backendRenderTarget,
+            SurfaceOrigin.TOP_LEFT,
+            SurfaceColorFormat.BGRA_8888,
+            ColorSpace.sRGB,
+            SurfaceProps(pixelGeometry = PixelGeometry.UNKNOWN)
+        ) ?: throw IllegalStateException("Can't create Surface")
+    }
+
+    override suspend fun awaitGPUCompletion() {
+        texture?.waitForCompletion()
+    }
+
+    override fun close() {
+        texture?.close()
+        context.close()
+    }
+}

benchmarks/multiplatform/benchmarks/src/desktopMain/kotlin/main.desktop.kt

@@ -8,5 +8,5 @@ import kotlinx.coroutines.runBlocking
 
 fun main(args: Array<String>) {
     Args.parseArgs(args)
-    runBlocking(Dispatchers.Main) { runBenchmarks() }
+    runBlocking(Dispatchers.Main) { benchmarksMain(graphicsContext = graphicsContext()) }
 }

benchmarks/multiplatform/benchmarks/src/iosMain/kotlin/main.ios.kt

@@ -7,5 +7,5 @@ import kotlinx.coroutines.runBlocking
 
 fun main(args : List<String>) {
     Args.parseArgs(args.toTypedArray())
-    runBlocking { runBenchmarks(graphicsContext = graphicsContext()) }
+    runBlocking { benchmarksMain(graphicsContext = graphicsContext()) }
 }

benchmarks/multiplatform/benchmarks/src/macosMain/kotlin/main.macos.kt

@@ -6,5 +6,5 @@ import kotlinx.coroutines.runBlocking
 
 fun main(args : Array<String>) {
     Args.parseArgs(args)
-    runBlocking { runBenchmarks(graphicsContext = graphicsContext()) }
+    runBlocking { benchmarksMain(graphicsContext = graphicsContext()) }
 }

benchmarks/multiplatform/benchmarks/src/wasmJsMain/kotlin/main.wasmJs.kt

@@ -9,7 +9,7 @@ fun main() {
     val args = generateSequence { urlParams.get("arg${i++}") }.toList().toTypedArray()
     Args.parseArgs(args)
     MainScope().launch {
-        runBenchmarks()
+        benchmarksMain()
         println("Completed!")
     }
 }