Namer added to abstract over naming conventions on different platforms

This commit is contained in:
Andrey Breslav
2013-05-22 16:45:29 +04:00
committed by Alexander Udalov
parent 347e2c27aa
commit b90522b223
3 changed files with 36 additions and 10 deletions
@@ -27,13 +27,11 @@ import java.util.Collection;
public class DescriptorSerializer {
// TODO: flags
private final NameTable nameTable;
private final Interner<TypeParameterDescriptor> typeParameters;
public DescriptorSerializer() {
this(new NameTable(), new Interner<TypeParameterDescriptor>());
public DescriptorSerializer(@NotNull NameTable.Namer namer) {
this(new NameTable(namer), new Interner<TypeParameterDescriptor>());
}
private DescriptorSerializer(NameTable nameTable, Interner<TypeParameterDescriptor> typeParameters) {
@@ -51,9 +51,23 @@ public class NameTable {
}
};
public interface Namer {
@NotNull
Name getClassName(@NotNull ClassDescriptor classDescriptor);
@NotNull
Name getPackageName(@NotNull NamespaceDescriptor namespaceDescriptor);
}
private final Namer namer;
private final Interner<String> simpleNames = new Interner<String>();
private final Interner<QualifiedName.Builder> qualifiedNames = new Interner<QualifiedName.Builder>(QUALIFIED_NAME_BUILDER_HASHING);
public NameTable(@NotNull Namer namer) {
this.namer = namer;
}
@NotNull
public List<String> getSimpleNames() {
return simpleNames.getAllInternedObjects();
@@ -71,7 +85,7 @@ public class NameTable {
public int getFqNameIndex(@NotNull ClassDescriptor classDescriptor) {
QualifiedName.Builder builder = QualifiedName.newBuilder();
builder.setKind(QualifiedName.Kind.CLASS);
builder.setShortName(getSimpleNameIndex(classDescriptor.getName()));
builder.setShortName(getSimpleNameIndex(namer.getClassName(classDescriptor)));
DeclarationDescriptor containingDeclaration = classDescriptor.getContainingDeclaration();
if (containingDeclaration instanceof NamespaceDescriptor) {
@@ -89,7 +103,7 @@ public class NameTable {
public int getFqNameIndex(@NotNull NamespaceDescriptor namespaceDescriptor) {
QualifiedName.Builder builder = QualifiedName.newBuilder();
//default: builder.setKind(QualifiedNameTable.QualifiedName.Kind.PACKAGE);
builder.setShortName(getSimpleNameIndex(namespaceDescriptor.getName()));
builder.setShortName(getSimpleNameIndex(namer.getPackageName(namespaceDescriptor)));
NamespaceDescriptorParent containingDeclaration = namespaceDescriptor.getContainingDeclaration();
if (containingDeclaration instanceof NamespaceDescriptor) {
@@ -105,6 +119,4 @@ public class NameTable {
return qualifiedNames.intern(builder);
}
}
@@ -57,6 +57,22 @@ import java.util.*;
public abstract class AbstractDescriptorSerializationTest extends KotlinTestWithEnvironment {
public static final Name TEST_PACKAGE_NAME = Name.identifier("test");
public static final NameTable.Namer JAVA_NAMER = new NameTable.Namer() {
@NotNull
@Override
public Name getClassName(@NotNull ClassDescriptor classDescriptor) {
if (classDescriptor.getKind() == ClassKind.CLASS_OBJECT) {
return Name.identifier(JvmAbi.CLASS_OBJECT_CLASS_NAME);
}
return classDescriptor.getName();
}
@NotNull
@Override
public Name getPackageName(@NotNull NamespaceDescriptor namespaceDescriptor) {
return namespaceDescriptor.getName();
}
};
@Override
protected JetCoreEnvironment createEnvironment() {
@@ -212,7 +228,7 @@ public abstract class AbstractDescriptorSerializationTest extends KotlinTestWith
serializeClasses(classes, serializedClasses);
DescriptorSerializer descriptorSerializer = new DescriptorSerializer();
DescriptorSerializer descriptorSerializer = new DescriptorSerializer(JAVA_NAMER);
List<MessageLite> messages = Lists.newArrayList();
for (CallableMemberDescriptor callable : callables) {
messages.add(descriptorSerializer.callableProto(callable).build());
@@ -227,7 +243,7 @@ public abstract class AbstractDescriptorSerializationTest extends KotlinTestWith
private static void serializeClasses(Collection<ClassDescriptor> classes, Map<ClassDescriptor, byte[]> serializedClasses) throws IOException {
for (ClassDescriptor classDescriptor : classes) {
DescriptorSerializer descriptorSerializer = new DescriptorSerializer();
DescriptorSerializer descriptorSerializer = new DescriptorSerializer(JAVA_NAMER);
ByteArrayOutputStream bytes = new ByteArrayOutputStream();
ProtoBuf.Class classProto = descriptorSerializer.classProto(classDescriptor).build();