Fix for 'getNthResult' in IfThenElse expression impl

2016-09-19 11:44:02 +03:00
parent f4aee6cab5
commit fe51c35298
1 changed files with 10 additions and 1 deletions
@@ -45,7 +45,16 @@ class IrIfThenElseImpl(
            if (n == 0) condition else null

    override fun getNthResult(n: Int): IrExpression? =
-            if (n == 0) thenBranch else null
+            when (n) {
+                0 -> {
+                    thenBranch
+                }
+                1 -> {
+                    elseBranch
+                }
+                else -> null
+            }
+

    override fun putNthCondition(n: Int, expression: IrExpression) {
        if (n == 0) condition = expression