azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Commonizer] Reworked functions for making nullable CIR types

Browse Source
This commit is contained in:
Dmitriy Dolovov
2021-03-02 02:05:43 +03:00
parent f2db93a9d2
commit 43d9c6cc90
2 changed files with 34 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
native/commonizer/src/org/jetbrains/kotlin/descriptors/commonizer/cir/factory/CirTypeFactory.kt
+33 -20
View File
@@ -67,7 +67,7 @@ object CirTypeFactory {
val expandedType = extractExpandedType(abbreviatedType)
val cirExpandedType = create(expandedType) as CirClassOrTypeAliasType
val cirExpandedTypeWithProperNullability = if (source.isMarkedNullable) makeNullable(cirExpandedType) else cirExpandedType
val cirExpandedTypeWithProperNullability = makeNullableIfNecessary(cirExpandedType, source.isMarkedNullable)
createTypeAliasType(
typeAliasId = classifierDescriptor.classifierId,
@@ -128,25 +128,38 @@ object CirTypeFactory {
)
}
fun makeNullable(classOrTypeAliasType: CirClassOrTypeAliasType): CirClassOrTypeAliasType =
if (classOrTypeAliasType.isMarkedNullable)
classOrTypeAliasType
else
when (classOrTypeAliasType) {
is CirClassType -> createClassType(
classId = classOrTypeAliasType.classifierId,
outerType = classOrTypeAliasType.outerType,
visibility = classOrTypeAliasType.visibility,
arguments = classOrTypeAliasType.arguments,
isMarkedNullable = true
)
is CirTypeAliasType -> createTypeAliasType(
typeAliasId = classOrTypeAliasType.classifierId,
underlyingType = makeNullable(classOrTypeAliasType.underlyingType),
arguments = classOrTypeAliasType.arguments,
isMarkedNullable = true
)
}
fun <T : CirSimpleType> makeNullable(type: T): T {
if (type.isMarkedNullable)
return type
val result = when (type) {
is CirClassType -> createClassType(
classId = type.classifierId,
outerType = type.outerType,
visibility = type.visibility,
arguments = type.arguments,
isMarkedNullable = true
)
is CirTypeAliasType -> createTypeAliasType(
typeAliasId = type.classifierId,
underlyingType = makeNullable(type.underlyingType),
arguments = type.arguments,
isMarkedNullable = true
)
is CirTypeParameterType -> createTypeParameterType(
index = type.index,
isMarkedNullable = true
)
else -> error("Unsupported type: $type")
}
@Suppress("UNCHECKED_CAST")
return result as T
}
@Suppress("NOTHING_TO_INLINE")
inline fun <T : CirSimpleType> makeNullableIfNecessary(type: T, necessary: Boolean): T =
if (!necessary) type else makeNullable(type)
fun unabbreviate(type: CirClassOrTypeAliasType): CirClassType = when (type) {
is CirClassType -> {
native/commonizer/src/org/jetbrains/kotlin/descriptors/commonizer/core/TypeCommonizer.kt
+1 -5
View File
@@ -143,11 +143,7 @@ private class TypeAliasTypeCommonizer(private val classifiers: CirKnownClassifie
// type alias don't needs to be commonized because it is from the standard library
fun forKnownUnderlyingType(underlyingType: CirClassOrTypeAliasType) = object : CommonizedTypeAliasTypeBuilder {
override fun build(typeAliasId: CirEntityId, arguments: List<CirTypeProjection>, isMarkedNullable: Boolean): CirTypeAliasType {
val underlyingTypeWithProperNullability = if (isMarkedNullable && !underlyingType.isMarkedNullable)
CirTypeFactory.makeNullable(underlyingType)
else
underlyingType
val underlyingTypeWithProperNullability = CirTypeFactory.makeNullableIfNecessary(underlyingType, isMarkedNullable)
return CirTypeFactory.createTypeAliasType(
typeAliasId = typeAliasId,
underlyingType = underlyingTypeWithProperNullability, // TODO replace arguments???