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.FqName;
022 import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
023 import org.jetbrains.jet.lang.resolve.name.Name;
024
025 import static org.jetbrains.jet.descriptors.serialization.ProtoBuf.QualifiedNameTable.QualifiedName;
026
027 public class NameResolver {
028 private final ProtoBuf.SimpleNameTable simpleNames;
029 private final ProtoBuf.QualifiedNameTable qualifiedNames;
030
031 public NameResolver(
032 @NotNull ProtoBuf.SimpleNameTable simpleNames,
033 @NotNull ProtoBuf.QualifiedNameTable qualifiedNames
034 ) {
035 this.simpleNames = simpleNames;
036 this.qualifiedNames = qualifiedNames;
037 }
038
039 @NotNull
040 public ProtoBuf.SimpleNameTable getSimpleNameTable() {
041 return simpleNames;
042 }
043
044 @NotNull
045 public ProtoBuf.QualifiedNameTable getQualifiedNameTable() {
046 return qualifiedNames;
047 }
048
049 @NotNull
050 public Name getName(int index) {
051 String name = simpleNames.getName(index);
052 return Name.guess(name);
053 }
054
055 @NotNull
056 public ClassId getClassId(int index) {
057 QualifiedName fqNameProto = qualifiedNames.getQualifiedName(index);
058 assert fqNameProto.getKind() == ProtoBuf.QualifiedNameTable.QualifiedName.Kind.CLASS : "Not a class fqName: " + fqNameProto.getKind();
059
060 StringBuilder relativeClassName = new StringBuilder();
061 QualifiedName packageFqNameProto = renderFqName(relativeClassName, fqNameProto, QualifiedName.Kind.CLASS);
062
063 FqName packageFqName;
064 if (packageFqNameProto != null) {
065 StringBuilder sb = new StringBuilder();
066 QualifiedName mustBeNull = renderFqName(sb, packageFqNameProto, QualifiedName.Kind.PACKAGE);
067 assert mustBeNull == null : "Prefix of an fqName must be all of kind PACKAGE";
068
069 packageFqName = new FqName(sb.toString());
070 }
071 else {
072 packageFqName = FqName.ROOT;
073 }
074
075 return new ClassId(packageFqName, new FqNameUnsafe(relativeClassName.toString()));
076 }
077
078 @Nullable
079 private QualifiedName renderFqName(StringBuilder sb, QualifiedName fqNameProto, QualifiedName.Kind kind) {
080 QualifiedName result = null;
081 if (fqNameProto.hasParentQualifiedName()) {
082 QualifiedName parentProto = qualifiedNames.getQualifiedName(fqNameProto.getParentQualifiedName());
083 if (kind == null || parentProto.getKind() == kind) {
084 result = renderFqName(sb, parentProto, kind);
085 sb.append(".");
086 }
087 else {
088 result = parentProto;
089 }
090 }
091 sb.append(simpleNames.getName(fqNameProto.getShortName()));
092 return result;
093 }
094
095 @NotNull
096 public FqName getFqName(int index) {
097 QualifiedName qualifiedName = qualifiedNames.getQualifiedName(index);
098 Name shortName = getName(qualifiedName.getShortName());
099 if (!qualifiedName.hasParentQualifiedName()) {
100 return FqName.topLevel(shortName);
101 }
102 return getFqName(qualifiedName.getParentQualifiedName()).child(shortName);
103 }
104 }