Handle withIndex() on Iterables (including progressions) and Sequences

in ForLoopsLowering.

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInDownToWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (7 downTo 4).withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(7, 6, 5, 4), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInIndicesWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in listOf(4, 5, 6, 7).indices.withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(0, 1, 2, 3), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInRangeToWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4..7).withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(4, 5, 6, 7), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInReversedStepWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in ((4..11).reversed() step 2).withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(11, 9, 7, 5), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInReversedWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4..7).reversed().withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(7, 6, 5, 4), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInStepReversedWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4..11 step 2).reversed().withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(10, 8, 6, 4), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInStepWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4..11 step 2).withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(4, 6, 8, 10), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInUntilWithIndex.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4 until 8).withIndex()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(4, 5, 6, 7), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInWithIndexNoIndexOrElementVar.kt

@@ -0,0 +1,15 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    var count = 0
+    for ((_, _) in (4..7).withIndex()) {
+        count++
+    }
+    assertEquals(4, count)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInWithIndexNotDestructured.kt

@@ -0,0 +1,19 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for (iv in (4..7).withIndex()) {
+        val (i, v) = iv
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(4, 5, 6, 7), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInWithIndexReversed.kt

@@ -0,0 +1,18 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((i, v) in (4..7).withIndex().reversed()) {
+        indexList += i
+        valueList += v
+    }
+    assertEquals(listOf(3, 2, 1, 0), indexList)
+    assertEquals(listOf(7, 6, 5, 4), valueList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInWithIndexWithDestructuringInLoop.kt

@@ -0,0 +1,23 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val indexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    val valueAndIndexList = mutableListOf<Int>()
+    for ((i, v) in (4..7).withIndex()) {
+        val (v2, i2) = Pair(v, i)
+        indexList += i
+        valueList += v
+        valueAndIndexList += v2
+        valueAndIndexList += i2
+    }
+    assertEquals(listOf(0, 1, 2, 3), indexList)
+    assertEquals(listOf(4, 5, 6, 7), valueList)
+    assertEquals(listOf(4, 0, 5, 1, 6, 2, 7, 3), valueAndIndexList)
+
+    return "OK"
+}

compiler/testData/codegen/box/ranges/forInProgressionWithIndex/forInWithIndexWithIndex.kt

@@ -0,0 +1,22 @@
+// IGNORE_BACKEND_FIR: JVM_IR
+// KJS_WITH_FULL_RUNTIME
+// WITH_RUNTIME
+
+import kotlin.test.assertEquals
+
+fun box(): String {
+    val outerIndexList = mutableListOf<Int>()
+    val innerIndexList = mutableListOf<Int>()
+    val valueList = mutableListOf<Int>()
+    for ((outer, iv) in (4..7).withIndex().withIndex()) {
+        outerIndexList += outer
+        val (inner, v) = iv
+        innerIndexList += inner
+        valueList += v
+    }
+    assertEquals(listOf(0, 1, 2, 3), outerIndexList)
+    assertEquals(listOf(0, 1, 2, 3), innerIndexList)
+    assertEquals(listOf(4, 5, 6, 7), valueList)
+
+    return "OK"
+}