org.jetbrains.kotlin.ir.expressions.IrConstructorCall.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation
Kotlin Compiler
There is a newer version: 2.0.0
/*
 * Copyright 2010-2019 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.ir.expressions

import org.jetbrains.kotlin.ir.symbols.IrConstructorSymbol
import org.jetbrains.kotlin.ir.types.IrType

abstract class IrConstructorCall(
    typeArgumentsCount: Int,
    valueArgumentsCount: Int,
) : IrFunctionAccessExpression(typeArgumentsCount, valueArgumentsCount) {
    abstract override val symbol: IrConstructorSymbol

    abstract val constructorTypeArgumentsCount: Int

    class ConstructorTypeArguments(internal val irConstructorCall: IrConstructorCall) : AbstractList() {
        override val size: Int
            get() = irConstructorCall.constructorTypeArgumentsCount

        override fun get(index: Int): IrType? =
            if (index >= size)
                throw IndexOutOfBoundsException("index: $index, size: $size")
            else
                irConstructorCall.getConstructorTypeArgument(index)
    }

    class ClassTypeArguments(internal val irConstructorCall: IrConstructorCall) : AbstractList() {
        override val size: Int
            get() = irConstructorCall.classTypeArgumentsCount

        override fun get(index: Int): IrType? =
            if (index >= size)
                throw IndexOutOfBoundsException("index: $index, size: $size")
            else
                irConstructorCall.getTypeArgument(index)
    }
}

fun IrConstructorCall.getConstructorTypeArgumentIndex(constructorTypeArgumentIndex: Int) =
    typeArgumentsCount - constructorTypeArgumentsCount + constructorTypeArgumentIndex

fun IrConstructorCall.getConstructorTypeArgument(index: Int): IrType? =
    getTypeArgument(getConstructorTypeArgumentIndex(index))

fun IrConstructorCall.putConstructorTypeArgument(index: Int, type: IrType?) {
    putTypeArgument(getConstructorTypeArgumentIndex(index), type)
}

operator fun IrConstructorCall.ConstructorTypeArguments.set(index: Int, type: IrType?) {
    if (index >= size) throw IndexOutOfBoundsException("index: $index, size: $size")
    irConstructorCall.putConstructorTypeArgument(index, type)
}

val IrConstructorCall.classTypeArgumentsCount: Int
    get() = typeArgumentsCount - constructorTypeArgumentsCount

fun IrConstructorCall.getClassTypeArgument(index: Int): IrType? =
    getTypeArgument(index)

fun IrConstructorCall.putClassTypeArgument(index: Int, type: IrType?) {
    putTypeArgument(index, type)
}

operator fun IrConstructorCall.ClassTypeArguments.set(index: Int, type: IrType?) {
    if (index >= size) throw IndexOutOfBoundsException("index: $index, size: $size")
    irConstructorCall.putClassTypeArgument(index, type)
}

fun IrConstructorCall.getConstructorTypeArguments() =
    IrConstructorCall.ConstructorTypeArguments(this)

fun IrConstructorCall.getClassTypeArguments() =
    IrConstructorCall.ClassTypeArguments(this)

var IrConstructorCall.outerClassReceiver: IrExpression?
    get() = dispatchReceiver
    set(value) {
        dispatchReceiver = value
    }