Refactor getKotlinTypeWithPossibleSmartCastToFP and related parts

2018-10-10 15:25:33 +03:00
parent bbac1d802f
commit 01091ba1aa
3 changed files with 72 additions and 84 deletions
@@ -0,0 +1,52 @@
+/*
+ * Copyright 2010-2018 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license
+ * that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.resolve.calls.smartcasts
+
+import org.jetbrains.kotlin.builtins.KotlinBuiltIns
+import org.jetbrains.kotlin.config.LanguageVersionSettings
+import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
+import org.jetbrains.kotlin.psi.KtExpression
+import org.jetbrains.kotlin.psi.KtProperty
+import org.jetbrains.kotlin.resolve.BindingContext
+import org.jetbrains.kotlin.resolve.bindingContextUtil.getDataFlowInfoBefore
+import org.jetbrains.kotlin.types.KotlinType
+import org.jetbrains.kotlin.utils.addToStdlib.firstNotNullResult
+
+fun KtExpression.getKotlinTypeForComparison(bindingContext: BindingContext): KotlinType? =
+    when {
+        this is KtProperty -> bindingContext[BindingContext.VARIABLE, this]?.type
+        else -> bindingContext.getType(this)
+    }
+
+fun KtExpression?.getKotlinTypeWithPossibleSmartCastToFP(
+    bindingContext: BindingContext,
+    descriptor: DeclarationDescriptor?,
+    languageVersionSettings: LanguageVersionSettings,
+    dataFlowValueFactory: DataFlowValueFactory
+): KotlinType? {
+    val givenType = this?.getKotlinTypeForComparison(bindingContext) ?: return null
+
+    if (KotlinBuiltIns.isDoubleOrNullableDouble(givenType)) {
+        return givenType
+    }
+
+    if (KotlinBuiltIns.isFloatOrNullableFloat(givenType)) {
+        return givenType
+    }
+
+    if (descriptor != null) {
+        val dataFlow = dataFlowValueFactory.createDataFlowValue(this, givenType, bindingContext, descriptor)
+        val stableTypes = bindingContext.getDataFlowInfoBefore(this).getStableTypes(dataFlow, languageVersionSettings)
+        stableTypes.firstNotNullResult {
+            when {
+                KotlinBuiltIns.isDoubleOrNullableDouble(it) -> it
+                KotlinBuiltIns.isFloatOrNullableFloat(it) -> it
+                else -> null
+            }
+        }?.let { return it }
+    }
+    return givenType
+}