KT-52027 Encapsulate and fix calculation of parameter indices

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/InlineCodegen.kt

@@ -230,7 +230,10 @@ abstract class InlineCodegen<out T : BaseExpressionCodegen>(
             return processDefaultMaskOrMethodHandler(stackValue, kind)
         }
 
-        val info = invocationParamBuilder.addNextValueParameter(jvmKotlinType.type, false, null, parameterIndex)
+        val info = when (parameterIndex) {
+            -1 -> invocationParamBuilder.addNextParameter(jvmKotlinType.type, false)
+            else -> invocationParamBuilder.addNextValueParameter(jvmKotlinType.type, false, null, parameterIndex)
+        }
         info.functionalArgument = when (kind) {
             ValueKind.READ_OF_INLINE_LAMBDA_FOR_INLINE_SUSPEND_PARAMETER ->
                 NonInlineArgumentForInlineSuspendParameter.INLINE_LAMBDA_AS_VARIABLE

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/ParametersBuilder.kt

@@ -20,19 +20,17 @@ class ParametersBuilder private constructor() {
     private var nextValueParameterIndex = 0
 
     fun addThis(type: Type, skipped: Boolean): ParameterInfo {
-        return addParameter(ParameterInfo(type, skipped, nextParameterOffset, -1, nextValueParameterIndex))
+        return addParameter(ParameterInfo(type, skipped, nextParameterOffset, -1, nextValueParameterIndex), false)
     }
 
     fun addNextParameter(type: Type, skipped: Boolean, typeOnStack: Type = type): ParameterInfo {
-        return addParameter(ParameterInfo(type, skipped, nextParameterOffset, null, nextValueParameterIndex, typeOnStack))
+        return addParameter(ParameterInfo(type, skipped, nextParameterOffset, null, nextValueParameterIndex, typeOnStack), false)
     }
 
     fun addNextValueParameter(type: Type, skipped: Boolean, remapValue: StackValue?, parameterIndex: Int): ParameterInfo {
         return addParameter(
-            ParameterInfo(
-                type, skipped, nextParameterOffset, remapValue,
-                if (parameterIndex == -1) nextValueParameterIndex else parameterIndex + valueParamFirstIndex
-            )
+            ParameterInfo(type, skipped, nextParameterOffset, remapValue, parameterIndex + valueParamFirstIndex),
+            true
         )
     }
 
@@ -41,15 +39,15 @@ class ParametersBuilder private constructor() {
             original.desc, newFieldName, original.isSkipped, nextParameterOffset, original.index, original.isSkipInConstructor
         )
         info.functionalArgument = original.functionalArgument
-        return addParameter(info)
+        return addParameter(info, false)
     }
 
     fun addCapturedParam(desc: CapturedParamDesc, newFieldName: String, skipInConstructor: Boolean): CapturedParamInfo {
-        return addParameter(CapturedParamInfo(desc, newFieldName, false, nextParameterOffset, -1, skipInConstructor))
+        return addParameter(CapturedParamInfo(desc, newFieldName, false, nextParameterOffset, -1, skipInConstructor), false)
     }
 
     fun addCapturedParamCopy(copyFrom: CapturedParamInfo): CapturedParamInfo {
-        return addParameter(copyFrom.cloneWithNewDeclarationIndex(-1))
+        return addParameter(copyFrom.cloneWithNewDeclarationIndex(-1), false)
     }
 
     fun addCapturedParam(
@@ -66,23 +64,21 @@ class ParametersBuilder private constructor() {
         if (original != null) {
             info.functionalArgument = original.functionalArgument
         }
-        return addParameter(info)
+        return addParameter(info, false)
     }
 
-    private fun <T : ParameterInfo> addParameter(info: T): T {
+    private fun <T : ParameterInfo> addParameter(info: T, isValueParameter: Boolean): T {
         params.add(info)
         nextParameterOffset += info.type.size
         if (info !is CapturedParamInfo) {
             nextValueParameterIndex++
+            if (!isValueParameter) {
+                valueParamFirstIndex++
+            }
         }
         return info
     }
 
-    fun markValueParametersStart(contextReceiversCount: Int) {
-        this.valueParamFirstIndex = params.size - contextReceiversCount
-        this.nextValueParameterIndex = valueParamFirstIndex
-    }
-
     fun listCaptured(): List<CapturedParamInfo> {
         return params.filterIsInstance<CapturedParamInfo>()
     }

compiler/backend/src/org/jetbrains/kotlin/codegen/inline/PsiInlineCodegen.kt

@@ -97,8 +97,7 @@ class PsiInlineCodegen(
             hiddenParameters += invocationParamBuilder.addNextParameter(param.asmType, false) to
                     codegen.frameMap.enterTemp(param.asmType)
         }
-        // TODO: Add context receivers as hiddenParameters and pass their count
-        invocationParamBuilder.markValueParametersStart(0)
+        // TODO: Add context receivers as hiddenParameters
     }
 
     override fun putHiddenParamsIntoLocals() {