azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[JS IR] Optimize pattern-matching of enums into comparing their ordinals

Browse Source 
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;
}
```
This commit is contained in:
Sergej Jaskiewicz
2021-10-12 15:50:49 +03:00
committed by Space
parent 533eb589cb
commit 55ae6d1f3e
12 changed files with 45 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/testData/codegen/box/when/enumOptimization/bigEnum.kt
Vendored
+4 -8
View File
@@ -1,12 +1,8 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar1 count=6 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar1 count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar1 count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar1 count=2 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=6 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=3 IGNORED_BACKENDS=JS
// 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
compiler/testData/codegen/box/when/enumOptimization/duplicatingItems.kt
Vendored
+2 -3
View File
@@ -1,8 +1,7 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar count=3 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar count=2 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar count=4 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar count=0
import kotlin.test.assertEquals
compiler/testData/codegen/box/when/enumOptimization/enumInsideClassObject.kt
Vendored
+1 -2
View File
@@ -1,6 +1,5 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=foo count=3 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=foo count=0 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=foo count=3
// CHECK_IF_COUNT: function=foo count=0
import kotlin.test.assertEquals
compiler/testData/codegen/box/when/enumOptimization/expression.kt
Vendored
+3 -6
View File
@@ -1,11 +1,8 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar1 count=3 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar1 count=0 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar1 count=3
// CHECK_IF_COUNT: function=bar1 count=0
// CHECK_CASES_COUNT: function=bar2 count=4 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=3 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=4
// CHECK_IF_COUNT: function=bar2 count=0
import kotlin.test.assertEquals
compiler/testData/codegen/box/when/enumOptimization/kt14597.kt
Vendored
+2 -4
View File
@@ -1,7 +1,5 @@
// CHECK_CASES_COUNT: function=box count=6 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=box count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=box count=1 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=box count=7 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=box count=6
// CHECK_IF_COUNT: function=box count=1
enum class En { A, B, C }
compiler/testData/codegen/box/when/enumOptimization/kt14597_full.kt
Vendored
+2 -4
View File
@@ -1,7 +1,5 @@
// CHECK_CASES_COUNT: function=box count=18 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=box count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=box count=3 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=box count=21 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=box count=18
// CHECK_IF_COUNT: function=box count=3
enum class En { A, B, C }
compiler/testData/codegen/box/when/enumOptimization/nullability.kt
Vendored
+8 -4
View File
@@ -1,8 +1,12 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=foo1 count=0
// CHECK_IF_COUNT: function=foo1 count=2
// CHECK_CASES_COUNT: function=foo2 count=0
// CHECK_IF_COUNT: function=foo2 count=2
// CHECK_CASES_COUNT: function=foo1 count=0 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=foo1 count=4 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=foo1 count=2 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=foo1 count=0 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=foo2 count=0 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=foo2 count=3 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=foo2 count=2 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=foo2 count=0 IGNORED_BACKENDS=JS
import kotlin.test.assertEquals
compiler/testData/codegen/box/when/enumOptimization/nullableEnum.kt
Vendored
+4 -2
View File
@@ -1,5 +1,7 @@
// CHECK_CASES_COUNT: function=test count=0
// CHECK_IF_COUNT: function=test count=3
// CHECK_CASES_COUNT: function=test count=0 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=test count=3 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=test count=3 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=test count=0 IGNORED_BACKENDS=JS
enum class E {
A,
compiler/testData/codegen/box/when/enumOptimization/withoutElse.kt
Vendored
+4 -6
View File
@@ -1,12 +1,10 @@
// WITH_RUNTIME
// CHECK_CASES_COUNT: function=bar1 count=3 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar1 count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar1 count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar1 count=3 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar1 count=4 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar1 count=0
// CHECK_CASES_COUNT: function=bar2 count=4 TARGET_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=0 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=0 TARGET_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=4 IGNORED_BACKENDS=JS
// CHECK_CASES_COUNT: function=bar2 count=5 IGNORED_BACKENDS=JS
// CHECK_IF_COUNT: function=bar2 count=0
import kotlin.test.assertEquals