diff --git a/backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/lower/FunctionInlining.kt b/backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/lower/FunctionInlining.kt index 5a9ddfee787..eb28182e540 100644 --- a/backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/lower/FunctionInlining.kt +++ b/backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/lower/FunctionInlining.kt @@ -6,7 +6,7 @@ import org.jetbrains.kotlin.backend.konan.ir.IrInlineFunctionBody import org.jetbrains.kotlin.backend.konan.ir.getArguments import org.jetbrains.kotlin.descriptors.FunctionDescriptor import org.jetbrains.kotlin.descriptors.ParameterDescriptor -import org.jetbrains.kotlin.descriptors.impl.TypeParameterDescriptorImpl +import org.jetbrains.kotlin.descriptors.TypeParameterDescriptor import org.jetbrains.kotlin.ir.IrElement import org.jetbrains.kotlin.ir.IrStatement import org.jetbrains.kotlin.ir.builders.Scope @@ -21,7 +21,6 @@ import org.jetbrains.kotlin.ir.util.DeepCopyIrTree import org.jetbrains.kotlin.ir.visitors.IrElementTransformerVoid import org.jetbrains.kotlin.types.KotlinType - //-----------------------------------------------------------------------------// internal class FunctionInlining(val context: Context): IrElementTransformerVoid() { @@ -30,6 +29,14 @@ internal class FunctionInlining(val context: Context): IrElementTransformerVoid( //-------------------------------------------------------------------------// + var fqName: String? = null + override fun visitFile(declaration: IrFile): IrFile { + fqName = declaration.packageFragmentDescriptor.fqName.asString() + return super.visitFile(declaration) + } + + //-------------------------------------------------------------------------// + fun isLambdaExpression(expression: IrExpression) : Boolean { if (expression !is IrContainerExpressionBase) return false if (expression.origin != IrStatementOrigin.LAMBDA) return false @@ -68,7 +75,6 @@ internal class FunctionInlining(val context: Context): IrElementTransformerVoid( .functions[functionDescriptor.original] // Get FunctionDeclaration by FunctionDescriptor. if (functionDeclaration == null) return super.visitCall(irCall) // Function is declared in another module. - val copyFuncDeclaration = functionDeclaration.accept(DeepCopyIrTree(), // Create copy of the function. null) as IrFunction @@ -84,16 +90,18 @@ internal class FunctionInlining(val context: Context): IrElementTransformerVoid( inlineBody.accept(lambdaInliner, null) val transformer = ParametersTransformer(parameters, irCall) inlineBody.accept(transformer, null) // Replace parameters with expression. - return inlineBody + return super.visitBlock(inlineBody) } //-------------------------------------------------------------------------// override fun visitCall(expression: IrCall): IrExpression { - val functionDescriptor = expression.descriptor as FunctionDescriptor - if (!functionDescriptor.name.asString().contains("foo")) return super.visitCall(expression) + if(fqName!!.contains("kotlin")) return super.visitCall(expression) + + val functionDescriptor = expression.descriptor as FunctionDescriptor if (functionDescriptor.isInline) return inlineFunction(expression) // Return newly created IrInlineBody instead of IrCall. + return super.visitCall(expression) } @@ -194,8 +202,9 @@ internal class ParametersTransformer(val parameterToArgument: List Int): Int { + return body(i1) +} + +fun bar(i0: Int): Int { + return foo1(i0) { foo2(it) + 1 } +} + +fun main(args: Array) { + println(bar(2).toString()) +} diff --git a/backend.native/tests/codegen/inline/inline16.kt b/backend.native/tests/codegen/inline/inline16.kt new file mode 100644 index 00000000000..9c5255ea714 --- /dev/null +++ b/backend.native/tests/codegen/inline/inline16.kt @@ -0,0 +1,17 @@ +@Suppress("NOTHING_TO_INLINE") +inline fun > foo(destination: C, predicate: (T) -> Boolean): C { + for (element in destination) { + val value = element as T + if (!predicate(value)) destination.add(value) + } + return destination +} + +fun bar(): Boolean { + val result = foo > (mutableListOf(1, 2, 3)) { true } + return result.isEmpty() +} + +fun main(args: Array) { + println(bar().toString()) +} diff --git a/backend.native/tests/codegen/inline/inline17.kt b/backend.native/tests/codegen/inline/inline17.kt new file mode 100644 index 00000000000..1de814adfc4 --- /dev/null +++ b/backend.native/tests/codegen/inline/inline17.kt @@ -0,0 +1,12 @@ +@Suppress("NOTHING_TO_INLINE") +inline fun foo(i1: T, i2: T): List { + return listOf(i1, i2) +} + +fun bar(): List { + return foo (1, 2) +} + +fun main(args: Array) { + println(bar().toString()) +}