WASM: Lower expression bodies into block bodies

This commit is contained in:
Igor Laevsky
2021-09-02 15:48:43 +03:00
committed by TeamCityServer
parent b79719d6f5
commit 37874b1937
3 changed files with 49 additions and 22 deletions
@@ -22,28 +22,6 @@ import org.jetbrains.kotlin.ir.visitors.IrElementVisitorVoid
import org.jetbrains.kotlin.ir.visitors.acceptChildrenVoid
import org.jetbrains.kotlin.ir.visitors.acceptVoid
internal class ExpressionBodyTransformer(val context: Context) : FileLoweringPass {
override fun lower(irFile: IrFile) {
irFile.acceptVoid(object : IrElementVisitorVoid {
override fun visitElement(element: IrElement) {
element.acceptChildrenVoid(this)
}
override fun visitFunction(declaration: IrFunction) {
declaration.acceptChildrenVoid(this)
context.createIrBuilder(declaration.symbol, declaration.endOffset, declaration.endOffset).run {
val body = declaration.body
if (body is IrExpressionBody)
declaration.body = IrBlockBodyImpl(body.startOffset, body.endOffset) {
statements += irReturn(body.expression)
}
}
}
})
}
}
internal class ReturnsInsertionLowering(val context: Context) : FileLoweringPass {
private val symbols = context.ir.symbols