org.jetbrains.kotlin.codegen.forLoop.ForInOptimizedIndicesLoopGenerator.kt Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler Show documentation

the Kotlin compiler

There is a newer version: 2.0.0

/*
 * Copyright 2010-2017 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.jetbrains.kotlin.codegen.forLoop

import org.jetbrains.kotlin.codegen.ExpressionCodegen
import org.jetbrains.kotlin.codegen.StackValue
import org.jetbrains.kotlin.psi.KtForExpression
import org.jetbrains.kotlin.resolve.calls.model.ResolvedCall
import org.jetbrains.kotlin.resolve.scopes.receivers.ReceiverValue
import org.jetbrains.kotlin.types.KotlinType
import org.jetbrains.org.objectweb.asm.Type

abstract class ForInOptimizedIndicesLoopGenerator(
        codegen: ExpressionCodegen,
        forExpression: KtForExpression,
        loopRangeCall: ResolvedCall<*>
) : AbstractForInExclusiveRangeLoopGenerator(codegen, forExpression) {
    protected val receiverValue: ReceiverValue = loopRangeCall.extensionReceiver!!
    protected val expectedReceiverType: KotlinType = ExpressionCodegen.getExpectedReceiverType(loopRangeCall)

    override fun generateFrom(): StackValue =
            StackValue.constant(0, asmElementType)

    override fun generateTo(): StackValue =
            StackValue.operation(Type.INT_TYPE) { v ->
                codegen.generateReceiverValue(receiverValue, false).put(codegen.asmType(expectedReceiverType), v)
                getReceiverSizeAsInt()
            }

    /**
     * `(receiver -> size:I)`
     */
    protected abstract fun getReceiverSizeAsInt()
}