From b274c7567eec3a202cfa5b05eabdbea3fb326c78 Mon Sep 17 00:00:00 2001
From: Dmitriy Novozhilov <dmitriy.novozhilov@jetbrains.com>
Date: Wed, 7 Aug 2019 14:16:39 +0300
Subject: [PATCH] [FIR] Minor. Add `transformResult` to `FirWhenBranch`

---
 .../org/jetbrains/kotlin/fir/expressions/FirWhenBranch.kt  | 2 ++
 .../kotlin/fir/expressions/impl/FirWhenBranchImpl.kt       | 7 ++++++-
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/FirWhenBranch.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/FirWhenBranch.kt
index 1bbb54532cf..703756a80db 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/FirWhenBranch.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/FirWhenBranch.kt
@@ -29,4 +29,6 @@ interface FirWhenBranch : FirElement {
     }
 
     fun <D> transformCondition(transformer: FirTransformer<D>, data: D): FirWhenBranch
+
+    fun <D> transformResult(transformer: FirTransformer<D>, data: D): FirWhenBranch
 }
\ No newline at end of file
diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirWhenBranchImpl.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirWhenBranchImpl.kt
index f8dd537441c..d81d47d20d7 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirWhenBranchImpl.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/expressions/impl/FirWhenBranchImpl.kt
@@ -21,7 +21,7 @@ class FirWhenBranchImpl(
 ) : FirAbstractElement(psi), FirWhenBranch {
     override fun <D> transformChildren(transformer: FirTransformer<D>, data: D): FirElement {
         transformCondition(transformer, data)
-        result = result.transformSingle(transformer, data)
+        transformResult(transformer, data)
         return this
     }
 
@@ -29,4 +29,9 @@ class FirWhenBranchImpl(
         condition = condition.transformSingle(transformer, data)
         return this
     }
+
+    override fun <D> transformResult(transformer: FirTransformer<D>, data: D): FirWhenBranch {
+        result = result.transformSingle(transformer, data)
+        return this
+    }
 }
\ No newline at end of file