commit 1f3ea690daf7c8b41161496f0a4a3f04cfbf1bdb
author Jinseong Jeon <jsjeon@google.com> Fri Dec 23 17:27:48 2022 -0800
committer Ilya Kirillov <ilya.kirillov@jetbrains.com> Mon Jan 16 20:10:35 2023 +0100
tree ca498f40feccbbf5435d2cd2fa5272e1238444e2
parent 37a94239c4c9547308d9b564b50bb0bb9f1a8f0f

Introduce a util to copy TypeMappingMode for UAST

compiler/backend.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingModeExtensions.kt

diff --git a/compiler/backend.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingModeExtensions.kt b/compiler/backend.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingModeExtensions.kt
index 81d5cd3..5b45206 100644
--- a/compiler/backend.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingModeExtensions.kt
+++ b/compiler/backend.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingModeExtensions.kt
@@ -9,26 +9,23 @@
 import org.jetbrains.kotlin.types.model.KotlinTypeMarker
 
 fun TypeSystemCommonBackendContext.getOptimalModeForValueParameter(
-    type: KotlinTypeMarker,
-    isForUast: Boolean = false,
-): TypeMappingMode = getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = true, isForUast)
+    type: KotlinTypeMarker
+): TypeMappingMode = getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = true)
 
 fun TypeSystemCommonBackendContext.getOptimalModeForReturnType(
     type: KotlinTypeMarker,
-    isAnnotationMethod: Boolean,
-    isForUast: Boolean = false,
+    isAnnotationMethod: Boolean
 ): TypeMappingMode {
     return if (isAnnotationMethod)
         TypeMappingMode.VALUE_FOR_ANNOTATION
     else
-        getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = false, isForUast)
+        getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = false)
 }
 
 @OptIn(TypeMappingModeInternals::class)
 private fun TypeSystemCommonBackendContext.getOptimalModeForSignaturePart(
     type: KotlinTypeMarker,
-    canBeUsedInSupertypePosition: Boolean,
-    isForUast: Boolean = false,
+    canBeUsedInSupertypePosition: Boolean
 ): TypeMappingMode {
     if (type.argumentsCount() == 0) return TypeMappingMode.DEFAULT
 
@@ -36,7 +33,7 @@
     if (isInlineClassType && shouldUseUnderlyingType(type)) {
         val underlyingType = computeUnderlyingType(type)
         if (underlyingType != null) {
-            return getOptimalModeForSignaturePart(underlyingType, canBeUsedInSupertypePosition, isForUast).dontWrapInlineClassesMode()
+            return getOptimalModeForSignaturePart(underlyingType, canBeUsedInSupertypePosition).dontWrapInlineClassesMode()
         }
     }
 
@@ -48,17 +45,15 @@
 
     val invariantArgumentMode =
         if (canBeUsedInSupertypePosition)
-            getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = false, isForUast)
+            getOptimalModeForSignaturePart(type, canBeUsedInSupertypePosition = false)
         else
             null
 
     return TypeMappingMode(
         skipDeclarationSiteWildcards = !canBeUsedInSupertypePosition,
         skipDeclarationSiteWildcardsIfPossible = true,
-        genericArgumentMode = if (isForUast) TypeMappingMode.GENERIC_ARGUMENT_UAST else null,
         genericContravariantArgumentMode = contravariantArgumentMode,
         genericInvariantArgumentMode = invariantArgumentMode,
-        needInlineClassWrapping = !isInlineClassType,
-        mapTypeAliases = isForUast
+        needInlineClassWrapping = !isInlineClassType
     )
 }

core/compiler.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingMode.kt

diff --git a/core/compiler.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingMode.kt b/core/compiler.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingMode.kt
index f9eeed54..0516819 100644
--- a/core/compiler.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingMode.kt
+++ b/core/compiler.common.jvm/src/org/jetbrains/kotlin/load/kotlin/TypeMappingMode.kt
@@ -142,4 +142,20 @@
             needPrimitiveBoxing, false, isForAnnotationParameter, skipDeclarationSiteWildcards, skipDeclarationSiteWildcardsIfPossible,
             genericArgumentMode, kotlinCollectionsToJavaCollections, genericContravariantArgumentMode, genericInvariantArgumentMode
         )
+
+    fun mapTypeAliases(
+        genericArgumentMode: TypeMappingMode? = null
+    ): TypeMappingMode =
+        TypeMappingMode(
+            needPrimitiveBoxing,
+            needInlineClassWrapping,
+            isForAnnotationParameter,
+            skipDeclarationSiteWildcards,
+            skipDeclarationSiteWildcardsIfPossible,
+            genericArgumentMode ?: this.genericArgumentMode,
+            kotlinCollectionsToJavaCollections,
+            genericContravariantArgumentMode,
+            genericInvariantArgumentMode,
+            mapTypeAliases = true
+        )
 }