[Wasm] Move intrinsic generators to generators/wasm

Reason: avoid kotlin-stdlib-gen dependency on kotlinStdlib() via wasm.ir
2020-12-15 17:39:24 +03:00
parent efeabac2c5
commit 602ed42b99
6 changed files with 58 additions and 28 deletions
@@ -24,6 +24,7 @@ fun extraSourceSet(name: String, extendMain: Boolean = true): Pair<SourceSet, Co
 val (builtinsSourceSet, builtinsApi) = extraSourceSet("builtins", extendMain = false)
 val (evaluateSourceSet, evaluateApi) = extraSourceSet("evaluate")
 val (interpreterSourceSet, interpreterApi) = extraSourceSet("interpreter")
+val (wasmSourceSet, wasmApi) = extraSourceSet("wasm")

 dependencies {
    // for GeneratorsFileUtil
@@ -32,6 +33,8 @@ dependencies {

    builtinsApi("org.jetbrains.kotlin:kotlin-stdlib:$bootstrapKotlinVersion") { isTransitive = false }
    evaluateApi(project(":core:deserialization"))
+    wasmApi(project(":wasm:wasm.ir"))
+    wasmApi(kotlinStdlib())
    interpreterApi(project(":compiler:ir.tree"))
    interpreterApi(project(":compiler:ir.tree.impl"))

@@ -96,5 +99,6 @@ val generateKeywordStrings by generator("org.jetbrains.kotlin.generators.fronten
 val generateBuiltins by generator("org.jetbrains.kotlin.generators.builtins.generateBuiltIns.GenerateBuiltInsKt", builtinsSourceSet)
 val generateOperationsMap by generator("org.jetbrains.kotlin.generators.evaluate.GenerateOperationsMapKt", evaluateSourceSet)
 val generateInterpreterMap by generator("org.jetbrains.kotlin.generators.interpreter.GenerateInterpreterMapKt", interpreterSourceSet)
+val generateWasmIntrinsics by generator("org.jetbrains.kotlin.generators.wasm.WasmIntrinsicGeneratorKt", wasmSourceSet)

 testsJar()
@@ -0,0 +1,125 @@
+/*
+ * Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.generators.wasm
+
+import org.jetbrains.kotlin.wasm.ir.WasmOp
+import java.io.File
+import java.io.FileWriter
+
+
+fun FileWriter.generateStandardWasmInternalHeader() {
+    appendLine(File("license/COPYRIGHT.txt").readText())
+    appendLine("package kotlin.wasm.internal")
+    appendLine()
+    appendLine("//")
+    appendLine("// NOTE: THIS FILE IS AUTO-GENERATED by the generators/wasm/WasmIntrinsicGenerator.kt")
+    appendLine("//")
+    appendLine()
+}
+
+fun generateWasmOps(targetDir: File) {
+    FileWriter(targetDir.resolve("_WasmOp.kt")).use { writer ->
+        writer.generateStandardWasmInternalHeader()
+        writer.appendLine(
+            """
+            @ExcludedFromCodegen
+            @Suppress("unused")
+            @Target(AnnotationTarget.FUNCTION)
+            @Retention(AnnotationRetention.BINARY)
+            internal annotation class WasmOp(val name: String) {
+                companion object {
+            """.trimIndent()
+        )
+        WasmOp.values().forEach { op ->
+            writer.appendLine("        const val $op = \"$op\"")
+        }
+        writer.appendLine(
+            """
+                }
+            }
+            """.trimIndent()
+        )
+    }
+}
+
+fun generateWasmArrays(targetDir: File) {
+    FileWriter(targetDir.resolve("_WasmArrays.kt")).use { writer ->
+        writer.generateStandardWasmInternalHeader()
+
+        writer.appendLine(wasmArrayForType("Any", true))
+        val types = listOf(
+            "Byte",
+            "Char",
+            "Short",
+            "Int",
+            "Long",
+            "Float",
+            "Double"
+        )
+
+        types.forEach { primitive ->
+            val isPacked = primitive in setOf(
+                "Byte",
+                "Char",
+                "Short",
+            )
+            val isUnsigned = primitive == "Char"
+
+            writer.appendLine(
+                wasmArrayForType(
+                    primitive,
+                    false, isPacked, isUnsigned
+                )
+            )
+        }
+    }
+}
+
+fun wasmArrayForType(
+    klass: String,
+    isNullable: Boolean,
+    isPacked: Boolean = false,
+    isUnsigned: Boolean = false,
+): String {
+    val type = klass + if (isNullable) "?" else ""
+    val name = "Wasm${klass}Array"
+    val getSuffix = when {
+        isPacked && isUnsigned -> "_U"
+        isPacked && !isUnsigned -> "_S"
+        else -> ""
+    }
+    return """
+            @WasmArrayOf($klass::class, isNullable = $isNullable)
+            internal class $name(size: Int) {
+                @WasmOp(WasmOp.ARRAY_GET${getSuffix})
+                fun get(index: Int): $type =
+                    implementedAsIntrinsic
+    
+                @WasmOp(WasmOp.ARRAY_SET)
+                fun set(index: Int, value: $type): Unit =
+                    implementedAsIntrinsic
+    
+                @WasmOp(WasmOp.ARRAY_LEN)
+                fun len(): Int =
+                    implementedAsIntrinsic
+            }
+    
+            internal inline fun $name.fill(size: Int, init: (Int) -> $type) {
+                var i = 0
+                while (i < size) {
+                    set(i, init(i))
+                    i++
+                }
+            }
+            
+            """.trimIndent()
+}
+
+fun main() {
+    val targetDir = File("libraries/stdlib/wasm/src/generated")
+    generateWasmOps(targetDir)
+    generateWasmArrays(targetDir)
+}