[Wasm] Support enumValues and enumValueOf functions
^KT-57276 Fixed Merge-request: KT-MR-9169 Merged-by: Svyatoslav Kuzmich <svyatoslav.kuzmich@jetbrains.com>
This commit is contained in:
committed by
Space Team
parent
bcf7c38288
commit
a99ea1d3ce
+7
-5
@@ -19,7 +19,7 @@ import org.jetbrains.kotlin.ir.util.isEnumClass
|
||||
import org.jetbrains.kotlin.name.Name
|
||||
|
||||
|
||||
class EnumIntrinsicsTransformer(private val context: JsIrBackendContext) : CallsTransformer {
|
||||
object EnumIntrinsicsUtils {
|
||||
private fun transformEnumTopLevelIntrinsic(
|
||||
call: IrFunctionAccessExpression,
|
||||
staticMethodPredicate: (IrSimpleFunction) -> Boolean
|
||||
@@ -36,19 +36,21 @@ class EnumIntrinsicsTransformer(private val context: JsIrBackendContext) : Calls
|
||||
return irCall(call, staticMethod.symbol)
|
||||
}
|
||||
|
||||
private fun transformEnumValueOfIntrinsic(call: IrFunctionAccessExpression) = transformEnumTopLevelIntrinsic(call) {
|
||||
fun transformEnumValueOfIntrinsic(call: IrFunctionAccessExpression) = transformEnumTopLevelIntrinsic(call) {
|
||||
it.name == Name.identifier("valueOf") &&
|
||||
it.valueParameters.count() == 1 &&
|
||||
it.valueParameters[0].type.isString()
|
||||
}
|
||||
|
||||
private fun transformEnumValuesIntrinsic(call: IrFunctionAccessExpression) = transformEnumTopLevelIntrinsic(call) {
|
||||
fun transformEnumValuesIntrinsic(call: IrFunctionAccessExpression) = transformEnumTopLevelIntrinsic(call) {
|
||||
it.name == Name.identifier("values") && it.valueParameters.count() == 0
|
||||
}
|
||||
}
|
||||
|
||||
class EnumIntrinsicsTransformer(private val context: JsIrBackendContext) : CallsTransformer {
|
||||
override fun transformFunctionAccess(call: IrFunctionAccessExpression, doNotIntrinsify: Boolean) = when (call.symbol) {
|
||||
context.intrinsics.enumValueOfIntrinsic -> transformEnumValueOfIntrinsic(call)
|
||||
context.intrinsics.enumValuesIntrinsic -> transformEnumValuesIntrinsic(call)
|
||||
context.intrinsics.enumValueOfIntrinsic -> EnumIntrinsicsUtils.transformEnumValueOfIntrinsic(call)
|
||||
context.intrinsics.enumValuesIntrinsic -> EnumIntrinsicsUtils.transformEnumValuesIntrinsic(call)
|
||||
else -> call
|
||||
}
|
||||
}
|
||||
|
||||
@@ -111,6 +111,9 @@ class WasmSymbols(
|
||||
.find { it.valueParameters.firstOrNull()?.type?.isFunctionType == true }
|
||||
.let { symbolTable.referenceSimpleFunction(it!!) }
|
||||
|
||||
val enumValueOfIntrinsic = getInternalFunction("enumValueOfIntrinsic")
|
||||
val enumValuesIntrinsic = getInternalFunction("enumValuesIntrinsic")
|
||||
|
||||
val coroutineEmptyContinuation: IrPropertySymbol = symbolTable.referenceProperty(
|
||||
getProperty(FqName.fromSegments(listOf("kotlin", "wasm", "internal", "EmptyContinuation")))
|
||||
)
|
||||
|
||||
+5
@@ -12,6 +12,7 @@ import org.jetbrains.kotlin.backend.common.lower.createIrBuilder
|
||||
import org.jetbrains.kotlin.backend.wasm.WasmBackendContext
|
||||
import org.jetbrains.kotlin.config.AnalysisFlags
|
||||
import org.jetbrains.kotlin.config.languageVersionSettings
|
||||
import org.jetbrains.kotlin.ir.backend.js.lower.calls.EnumIntrinsicsUtils
|
||||
import org.jetbrains.kotlin.ir.backend.js.utils.erasedUpperBound
|
||||
import org.jetbrains.kotlin.ir.backend.js.utils.isEqualsInheritedFromAny
|
||||
import org.jetbrains.kotlin.ir.builders.*
|
||||
@@ -172,6 +173,10 @@ class BuiltInsLowering(val context: WasmBackendContext) : FileLoweringPass {
|
||||
}
|
||||
}
|
||||
}
|
||||
symbols.enumValueOfIntrinsic ->
|
||||
return EnumIntrinsicsUtils.transformEnumValueOfIntrinsic(call)
|
||||
symbols.enumValuesIntrinsic ->
|
||||
return EnumIntrinsicsUtils.transformEnumValuesIntrinsic(call)
|
||||
}
|
||||
|
||||
return call
|
||||
|
||||
Reference in New Issue
Block a user