001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.descriptors.serialization;
018
019 import org.jetbrains.annotations.NotNull;
020 import org.jetbrains.annotations.Nullable;
021 import org.jetbrains.jet.lang.resolve.name.ClassId;
022 import org.jetbrains.jet.lang.resolve.name.FqName;
023 import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
024 import org.jetbrains.jet.lang.resolve.name.Name;
025
026 import static org.jetbrains.jet.descriptors.serialization.ProtoBuf.QualifiedNameTable.QualifiedName;
027
028 public class NameResolver {
029 private final ProtoBuf.SimpleNameTable simpleNames;
030 private final ProtoBuf.QualifiedNameTable qualifiedNames;
031
032 public NameResolver(
033 @NotNull ProtoBuf.SimpleNameTable simpleNames,
034 @NotNull ProtoBuf.QualifiedNameTable qualifiedNames
035 ) {
036 this.simpleNames = simpleNames;
037 this.qualifiedNames = qualifiedNames;
038 }
039
040 @NotNull
041 public ProtoBuf.SimpleNameTable getSimpleNameTable() {
042 return simpleNames;
043 }
044
045 @NotNull
046 public ProtoBuf.QualifiedNameTable getQualifiedNameTable() {
047 return qualifiedNames;
048 }
049
050 @NotNull
051 public Name getName(int index) {
052 String name = simpleNames.getName(index);
053 return Name.guess(name);
054 }
055
056 @NotNull
057 public ClassId getClassId(int index) {
058 QualifiedName fqNameProto = qualifiedNames.getQualifiedName(index);
059 assert fqNameProto.getKind() == ProtoBuf.QualifiedNameTable.QualifiedName.Kind.CLASS : "Not a class fqName: " + fqNameProto.getKind();
060
061 StringBuilder relativeClassName = new StringBuilder();
062 QualifiedName packageFqNameProto = renderFqName(relativeClassName, fqNameProto, QualifiedName.Kind.CLASS);
063
064 FqName packageFqName;
065 if (packageFqNameProto != null) {
066 StringBuilder sb = new StringBuilder();
067 QualifiedName mustBeNull = renderFqName(sb, packageFqNameProto, QualifiedName.Kind.PACKAGE);
068 assert mustBeNull == null : "Prefix of an fqName must be all of kind PACKAGE";
069
070 packageFqName = new FqName(sb.toString());
071 }
072 else {
073 packageFqName = FqName.ROOT;
074 }
075
076 return new ClassId(packageFqName, new FqNameUnsafe(relativeClassName.toString()));
077 }
078
079 @Nullable
080 private QualifiedName renderFqName(StringBuilder sb, QualifiedName fqNameProto, QualifiedName.Kind kind) {
081 QualifiedName result = null;
082 if (fqNameProto.hasParentQualifiedName()) {
083 QualifiedName parentProto = qualifiedNames.getQualifiedName(fqNameProto.getParentQualifiedName());
084 if (kind == null || parentProto.getKind() == kind) {
085 result = renderFqName(sb, parentProto, kind);
086 sb.append(".");
087 }
088 else {
089 result = parentProto;
090 }
091 }
092 sb.append(simpleNames.getName(fqNameProto.getShortName()));
093 return result;
094 }
095
096 @NotNull
097 public FqName getFqName(int index) {
098 QualifiedName qualifiedName = qualifiedNames.getQualifiedName(index);
099 Name shortName = getName(qualifiedName.getShortName());
100 if (!qualifiedName.hasParentQualifiedName()) {
101 return FqName.topLevel(shortName);
102 }
103 return getFqName(qualifiedName.getParentQualifiedName()).child(shortName);
104 }
105 }