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