Files
kotlin-fork/compiler/testData/codegen/box/when/enumOptimization/bigEnum.kt
T
Sergej Jaskiewicz 55ae6d1f3e [JS IR] Optimize pattern-matching of enums into comparing their ordinals
For this code:
```
enum class Season {
    WINTER,
    SPRING,
    SUMMER,
    AUTUMN
}

fun bar1(x : Season) : String {
    return when (x) {
        Season.WINTER, Season.SPRING -> "winter_spring"
        Season.SUMMER -> "summer"
        else -> "autumn"
    }
}
```

previously we generated this:
```
function foo(x) {
    var tmp0_subject = x;
    return (tmp0_subject.equals(Season_WINTER_getInstance())
        ? true
        : tmp0_subject.equals(Season_SPRING_getInstance()))
            ? 'winter_spring'
            : tmp0_subject.equals(Season_SUMMER_getInstance())
                ? 'summer'
                : 'autumn';
}
```

And now we generated this:
```
function bar2(x) {
  var tmp0_subject = x;
  var tmp0 = tmp0_subject._get_ordinal__0_k$();
  var tmp;
  switch (tmp0) {
    case 0:
    case 1:
      tmp = 'winter_spring';
      break;
    case 2:
      tmp = 'summer';
      break;
    case 3:
      tmp = 'autumn';
      break;
    default:
      noWhenBranchMatchedException();
      break;
  }
  return tmp;
}
```
2021-10-14 11:44:01 +00:00

57 lines
1.5 KiB
Kotlin
Vendored

// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar1 count=6
// CHECK_IF_COUNT: function=bar1 count=0
// CHECK_CASES_COUNT: function=bar2 count=6
// CHECK_IF_COUNT: function=bar2 count=0
import kotlin.test.assertEquals
enum class BigEnum {
ITEM1, ITEM2, ITEM3, ITEM4, ITEM5, ITEM6, ITEM7, ITEM8, ITEM9, ITEM10,
ITEM11, ITEM12, ITEM13, ITEM14, ITEM15, ITEM16, ITEM17, ITEM18, ITEM19, ITEM20
}
fun bar1(x : BigEnum) : String {
when (x) {
BigEnum.ITEM1, BigEnum.ITEM2, BigEnum.ITEM3 -> return "123"
BigEnum.ITEM4, BigEnum.ITEM5, BigEnum.ITEM6 -> return "456"
}
return "-1";
}
fun bar2(x : BigEnum) : String {
when (x) {
BigEnum.ITEM7, BigEnum.ITEM8, BigEnum.ITEM9 -> return "789"
BigEnum.ITEM10 -> return "10"
BigEnum.ITEM11, BigEnum.ITEM12 -> return "1112"
else -> return "-1"
}
}
fun box() : String {
//bar1
assertEquals("123", bar1(BigEnum.ITEM1))
assertEquals("123", bar1(BigEnum.ITEM2))
assertEquals("123", bar1(BigEnum.ITEM3))
assertEquals("456", bar1(BigEnum.ITEM4))
assertEquals("456", bar1(BigEnum.ITEM5))
assertEquals("456", bar1(BigEnum.ITEM6))
assertEquals("-1", bar1(BigEnum.ITEM7))
//bar2
assertEquals("789", bar2(BigEnum.ITEM7))
assertEquals("789", bar2(BigEnum.ITEM8))
assertEquals("789", bar2(BigEnum.ITEM9))
assertEquals("10", bar2(BigEnum.ITEM10))
assertEquals("1112", bar2(BigEnum.ITEM11))
assertEquals("1112", bar2(BigEnum.ITEM12))
return "OK"
}