[IR] Inliner: supported recursion in default arguments

2020-10-29 13:14:25 +05:00
parent 5f8fabed61
commit 94145f880c
2 changed files with 12 additions and 6 deletions
@@ -162,11 +162,17 @@ class FunctionInlining(
            callee: IrFunction,
            performRecursiveInline: Boolean
        ): IrReturnableBlock {
-            val copiedCallee = copyIrElement.copy(callee).let {
-                (it as IrFunction).parent = callee.parent
-                if (performRecursiveInline)
-                    visitElement(it) as IrFunction
-                else it
+            val copiedCallee = (copyIrElement.copy(callee) as IrFunction).apply {
+                parent = callee.parent
+                if (performRecursiveInline) {
+                    body?.transformChildrenVoid()
+                    valueParameters.forEachIndexed { index, param ->
+                        if (callSite.getValueArgument(index) == null) {
+                            // Default values can recursively reference [callee] - transform only needed.
+                            param.defaultValue = param.defaultValue?.transform(this@FunctionInlining, null)
+                        }
+                    }
+                }
            }

            val evaluationStatements = evaluateArguments(callSite, copiedCallee)