Implement NameResolverImpl#getPackageFqName(index)

This commit is contained in:
Pavel V. Talanov
2015-12-07 16:41:03 +03:00
parent 5846769b43
commit 34c3c04ead
@@ -14,22 +14,6 @@
* limitations under the License.
*/
/*
* 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.serialization.deserialization
import org.jetbrains.kotlin.name.ClassId
@@ -37,58 +21,56 @@ import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.serialization.ProtoBuf
import org.jetbrains.kotlin.serialization.ProtoBuf.QualifiedNameTable.QualifiedName
import org.jetbrains.kotlin.utils.rethrow
import java.io.IOException
import java.io.InputStream
import java.util.*
class NameResolverImpl private constructor(
private val strings: ProtoBuf.StringTable,
private val qualifiedNames: ProtoBuf.QualifiedNameTable) : NameResolver {
private val qualifiedNames: ProtoBuf.QualifiedNameTable
) : NameResolver {
override fun getString(index: Int): String {
return strings.getString(index)
}
override fun getString(index: Int) = strings.getString(index)
override fun getName(index: Int): Name {
return Name.guess(strings.getString(index))
}
override fun getName(index: Int) = Name.guess(strings.getString(index))
override fun getClassId(index: Int): ClassId {
var index = index
val packageFqName = LinkedList<String>()
val relativeClassName = LinkedList<String>()
val (packageFqNameSegments, relativeClassNameSegments, isLocal) = traverseIds(index)
return ClassId(FqName.fromSegments(packageFqNameSegments), FqName.fromSegments(relativeClassNameSegments), isLocal)
}
fun getPackageFqName(index: Int): FqName {
val packageNameSegments = traverseIds(index).first
return FqName.fromSegments(packageNameSegments)
}
private fun traverseIds(startingIndex: Int): Triple<List<String>, List<String>, Boolean> {
var index = startingIndex
val packageNameSegments = LinkedList<String>()
val relativeClassNameSegments = LinkedList<String>()
var local = false
while (index != -1) {
val proto = qualifiedNames.getQualifiedName(index)
val shortName = strings.getString(proto.shortName)
when (proto.kind!!) {
QualifiedName.Kind.CLASS -> relativeClassName.addFirst(shortName)
QualifiedName.Kind.PACKAGE -> packageFqName.addFirst(shortName)
QualifiedName.Kind.CLASS -> relativeClassNameSegments.addFirst(shortName)
QualifiedName.Kind.PACKAGE -> packageNameSegments.addFirst(shortName)
QualifiedName.Kind.LOCAL -> {
relativeClassName.addFirst(shortName)
relativeClassNameSegments.addFirst(shortName)
local = true
}
}
index = proto.parentQualifiedName
}
return ClassId(FqName.fromSegments(packageFqName), FqName.fromSegments(relativeClassName), local)
return Triple(packageNameSegments, relativeClassNameSegments, local)
}
companion object {
fun read(stream: InputStream): NameResolverImpl {
try {
val simpleNames = ProtoBuf.StringTable.parseDelimitedFrom(stream)
val qualifiedNames = ProtoBuf.QualifiedNameTable.parseDelimitedFrom(stream)
return NameResolverImpl(simpleNames, qualifiedNames)
}
catch (e: IOException) {
throw rethrow(e)
}
val simpleNames = ProtoBuf.StringTable.parseDelimitedFrom(stream)
val qualifiedNames = ProtoBuf.QualifiedNameTable.parseDelimitedFrom(stream)
return NameResolverImpl(simpleNames, qualifiedNames)
}
}
}