From f28144ff9cace90943981b2eac57c9cf4e8ebc64 Mon Sep 17 00:00:00 2001
From: Mikhail Glukhikh <mikhail.glukhikh@jetbrains.com>
Date: Mon, 3 Feb 2020 17:14:39 +0300
Subject: [PATCH] Minor FirRenderer code cleanup

---
 .../org/jetbrains/kotlin/fir/FirRenderer.kt   | 26 ++++++++++---------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirRenderer.kt b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirRenderer.kt
index 1f5eee9fdcb..1b7c77ac375 100644
--- a/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirRenderer.kt
+++ b/compiler/fir/tree/src/org/jetbrains/kotlin/fir/FirRenderer.kt
@@ -240,20 +240,22 @@ class FirRenderer(builder: StringBuilder) : FirVisitorVoid() {
         }
 
         visitDeclaration(memberDeclaration)
-        if (memberDeclaration !is FirCallableDeclaration<*>) { // Handled by visitCallableDeclaration
-            if (memberDeclaration is FirRegularClass) {
-                print(" " + memberDeclaration.name)
-            }
-            if (memberDeclaration is FirTypeAlias) {
-                print(" " + memberDeclaration.name)
-            }
-            if (memberDeclaration is FirTypeParametersOwner) {
+        when (memberDeclaration) {
+            is FirClassLikeDeclaration<*> -> {
+                if (memberDeclaration is FirRegularClass) {
+                    print(" " + memberDeclaration.name)
+                }
+                if (memberDeclaration is FirTypeAlias) {
+                    print(" " + memberDeclaration.name)
+                }
                 memberDeclaration.typeParameters.renderTypeParameters()
             }
-        } else {
-            if (memberDeclaration.typeParameters.isNotEmpty()) {
-                print(" ")
-                memberDeclaration.typeParameters.renderTypeParameters()
+            is FirCallableDeclaration<*> -> {
+                // Name is handled by visitCallableDeclaration
+                if (memberDeclaration.typeParameters.isNotEmpty()) {
+                    print(" ")
+                    memberDeclaration.typeParameters.renderTypeParameters()
+                }
             }
         }
     }