org.jetbrains.kotlin.codegen.inline.SMAPAndMethodNode.kt Maven / Gradle / Ivy

Go to download

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.1.20-Beta1

/*
 * Copyright 2010-2015 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.inline

import org.jetbrains.kotlin.codegen.optimization.common.InsnSequence
import org.jetbrains.org.objectweb.asm.tree.LineNumberNode
import org.jetbrains.org.objectweb.asm.tree.MethodNode
import java.util.*

//TODO comment
class SMAPAndMethodNode(val node: MethodNode, val classSMAP: SMAP) {
    val ranges = createLineNumberSequence(node, classSMAP).map { it.mapper }.distinct().toList()

    fun copyWithNewNode(newMethodNode: MethodNode) = SMAPAndMethodNode(newMethodNode, classSMAP)
}

private fun createLineNumberSequence(node: MethodNode, classSMAP: SMAP): Sequence {
    return InsnSequence(node.instructions.first, null).filterIsInstance().map { lineNumber ->
        val index = classSMAP.intervals.binarySearch(RangeMapping(lineNumber.line, lineNumber.line, 1), Comparator {
            value, key ->
            if (key.dest in value) 0 else RangeMapping.Comparator.compare(value, key)
        })
        if (index < 0) {
            error("Unmapped label in inlined function $node ${lineNumber.line}")
        }
        LabelAndMapping(lineNumber, classSMAP.intervals[index])
    }
}

class LabelAndMapping(val lineNumberNode: LineNumberNode, val mapper: RangeMapping)