From c9decd5dc98bce152ed23c74214075f2bb06f932 Mon Sep 17 00:00:00 2001
From: Ilya Kirillov <ilya.kirillov@jetbrains.com>
Date: Thu, 16 May 2019 14:37:39 +0300
Subject: [PATCH] New J2K: correctly retrieve all type variables in printing

---
 .../jetbrains/kotlin/nj2k/nullabilityAnalysis/printing.kt   | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/nullabilityAnalysis/printing.kt b/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/nullabilityAnalysis/printing.kt
index 37d7d0455ca..2e592082093 100644
--- a/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/nullabilityAnalysis/printing.kt
+++ b/nj2k/nj2k-services/src/org/jetbrains/kotlin/nj2k/nullabilityAnalysis/printing.kt
@@ -71,8 +71,10 @@ internal class Printer(private val analysisContext: AnalysisContext) {
 
 private class Namer(analysisContext: AnalysisContext) {
     val names = run {
-        val typeVariables = (analysisContext.typeElementToTypeVariable.values +
-                analysisContext.declarationToTypeVariable.values)
+        val typeVariables =
+            analysisContext.typeVariableOwners.flatMap { owner ->
+                owner.allTypeVariables.flatMap { it.innerTypeVariables() + it }
+            }
         typeVariables.mapIndexed { index, typeVariable ->
             typeVariable to "T$index"
         }.toMap()