IR BE Common: Provisional IrFunctionExpression lowering

Transforms IrFunctionExpression elements to local function references, as it was done before IrFunctionExpression introduction.
2019-07-18 12:29:33 +03:00
parent 92984b2626
commit dd3f8ecaac
3 changed files with 64 additions and 2 deletions
@@ -0,0 +1,46 @@
+/*
+ * Copyright 2010-2019 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.backend.common.lower
+
+import org.jetbrains.kotlin.backend.common.FileLoweringPass
+import org.jetbrains.kotlin.ir.declarations.IrFile
+import org.jetbrains.kotlin.ir.expressions.IrExpression
+import org.jetbrains.kotlin.ir.expressions.IrFunctionExpression
+import org.jetbrains.kotlin.ir.expressions.impl.IrBlockImpl
+import org.jetbrains.kotlin.ir.expressions.impl.IrFunctionReferenceImpl
+import org.jetbrains.kotlin.ir.visitors.IrElementTransformerVoid
+import org.jetbrains.kotlin.ir.visitors.transformChildrenVoid
+
+class ProvisionalFunctionExpressionLowering :
+    IrElementTransformerVoid(),
+    FileLoweringPass {
+
+    override fun lower(irFile: IrFile) {
+        irFile.transformChildrenVoid(this)
+    }
+
+    override fun visitFunctionExpression(expression: IrFunctionExpression): IrExpression {
+        expression.transformChildrenVoid(this)
+
+        val startOffset = expression.startOffset
+        val endOffset = expression.endOffset
+        val type = expression.type
+        val origin = expression.origin
+        val function = expression.function
+
+        return IrBlockImpl(
+            startOffset, endOffset, type, origin,
+            listOf(
+                function,
+                IrFunctionReferenceImpl(
+                    startOffset, endOffset, type,
+                    function.symbol, function.descriptor, 0,
+                    origin
+                )
+            )
+        )
+    }
+}