| Index: third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| index 903b0a9bdf329e16f5b040fa550fd7477b20c097..9b147c7766eb8a02682760b9950b7b96bc2f1c4b 100644
|
| --- a/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| +++ b/third_party/protobuf/src/google/protobuf/compiler/java/java_extension.cc
|
| @@ -86,108 +86,123 @@ ExtensionGenerator::ExtensionGenerator(const FieldDescriptor* descriptor)
|
|
|
| ExtensionGenerator::~ExtensionGenerator() {}
|
|
|
| -void ExtensionGenerator::Generate(io::Printer* printer) {
|
| - map<string, string> vars;
|
| - vars["name"] = UnderscoresToCamelCase(descriptor_);
|
| - vars["containing_type"] = ClassName(descriptor_->containing_type());
|
| - vars["number"] = SimpleItoa(descriptor_->number());
|
| - vars["constant_name"] = FieldConstantName(descriptor_);
|
| - vars["lite"] = HasDescriptorMethods(descriptor_->file()) ? "" : "Lite";
|
| +// Initializes the vars referenced in the generated code templates.
|
| +void InitTemplateVars(const FieldDescriptor* descriptor,
|
| + const string& scope,
|
| + map<string, string>* vars_pointer) {
|
| + map<string, string> &vars = *vars_pointer;
|
| + vars["scope"] = scope;
|
| + vars["name"] = UnderscoresToCamelCase(descriptor);
|
| + vars["containing_type"] = ClassName(descriptor->containing_type());
|
| + vars["number"] = SimpleItoa(descriptor->number());
|
| + vars["constant_name"] = FieldConstantName(descriptor);
|
| + vars["index"] = SimpleItoa(descriptor->index());
|
| + vars["default"] =
|
| + descriptor->is_repeated() ? "" : DefaultValue(descriptor);
|
| + vars["type_constant"] = TypeName(GetType(descriptor));
|
| + vars["packed"] = descriptor->options().packed() ? "true" : "false";
|
| + vars["enum_map"] = "null";
|
| + vars["prototype"] = "null";
|
|
|
| - JavaType java_type = GetJavaType(descriptor_);
|
| + JavaType java_type = GetJavaType(descriptor);
|
| string singular_type;
|
| switch (java_type) {
|
| case JAVATYPE_MESSAGE:
|
| - vars["type"] = ClassName(descriptor_->message_type());
|
| + singular_type = ClassName(descriptor->message_type());
|
| + vars["prototype"] = singular_type + ".getDefaultInstance()";
|
| break;
|
| case JAVATYPE_ENUM:
|
| - vars["type"] = ClassName(descriptor_->enum_type());
|
| + singular_type = ClassName(descriptor->enum_type());
|
| + vars["enum_map"] = singular_type + ".internalGetValueMap()";
|
| break;
|
| default:
|
| - vars["type"] = BoxedPrimitiveTypeName(java_type);
|
| + singular_type = BoxedPrimitiveTypeName(java_type);
|
| break;
|
| }
|
| -
|
| - printer->Print(vars,
|
| - "public static final int $constant_name$ = $number$;\n");
|
| - if (descriptor_->is_repeated()) {
|
| - printer->Print(vars,
|
| - "public static final\n"
|
| - " com.google.protobuf.GeneratedMessage$lite$.GeneratedExtension<\n"
|
| - " $containing_type$,\n"
|
| - " java.util.List<$type$>> $name$ =\n"
|
| - " com.google.protobuf.GeneratedMessage$lite$\n"
|
| - " .newGeneratedExtension();\n");
|
| - } else {
|
| - printer->Print(vars,
|
| - "public static final\n"
|
| - " com.google.protobuf.GeneratedMessage$lite$.GeneratedExtension<\n"
|
| - " $containing_type$,\n"
|
| - " $type$> $name$ =\n"
|
| - " com.google.protobuf.GeneratedMessage$lite$\n"
|
| - " .newGeneratedExtension();\n");
|
| - }
|
| + vars["type"] = descriptor->is_repeated() ?
|
| + "java.util.List<" + singular_type + ">" : singular_type;
|
| + vars["singular_type"] = singular_type;
|
| }
|
|
|
| -void ExtensionGenerator::GenerateInitializationCode(io::Printer* printer) {
|
| +void ExtensionGenerator::Generate(io::Printer* printer) {
|
| map<string, string> vars;
|
| - vars["name"] = UnderscoresToCamelCase(descriptor_);
|
| - vars["scope"] = scope_;
|
| - vars["index"] = SimpleItoa(descriptor_->index());
|
| - vars["extendee"] = ClassName(descriptor_->containing_type());
|
| - vars["default"] = descriptor_->is_repeated() ? "" : DefaultValue(descriptor_);
|
| - vars["number"] = SimpleItoa(descriptor_->number());
|
| - vars["type_constant"] = TypeName(GetType(descriptor_));
|
| - vars["packed"] = descriptor_->options().packed() ? "true" : "false";
|
| - vars["enum_map"] = "null";
|
| - vars["prototype"] = "null";
|
| -
|
| - JavaType java_type = GetJavaType(descriptor_);
|
| - string singular_type;
|
| - switch (java_type) {
|
| - case JAVATYPE_MESSAGE:
|
| - vars["type"] = ClassName(descriptor_->message_type());
|
| - vars["prototype"] = ClassName(descriptor_->message_type()) +
|
| - ".getDefaultInstance()";
|
| - break;
|
| - case JAVATYPE_ENUM:
|
| - vars["type"] = ClassName(descriptor_->enum_type());
|
| - vars["enum_map"] = ClassName(descriptor_->enum_type()) +
|
| - ".internalGetValueMap()";
|
| - break;
|
| - default:
|
| - vars["type"] = BoxedPrimitiveTypeName(java_type);
|
| - break;
|
| - }
|
| + InitTemplateVars(descriptor_, scope_, &vars);
|
| + printer->Print(vars,
|
| + "public static final int $constant_name$ = $number$;\n");
|
|
|
| if (HasDescriptorMethods(descriptor_->file())) {
|
| - printer->Print(vars,
|
| - "$scope$.$name$.internalInit(\n"
|
| - " $scope$.getDescriptor().getExtensions().get($index$),\n"
|
| - " $type$.class);\n");
|
| + // Non-lite extensions
|
| + if (descriptor_->extension_scope() == NULL) {
|
| + // Non-nested
|
| + printer->Print(
|
| + vars,
|
| + "public static final\n"
|
| + " com.google.protobuf.GeneratedMessage.GeneratedExtension<\n"
|
| + " $containing_type$,\n"
|
| + " $type$> $name$ = com.google.protobuf.GeneratedMessage\n"
|
| + " .newFileScopedGeneratedExtension(\n"
|
| + " $singular_type$.class,\n"
|
| + " $prototype$);\n");
|
| + } else {
|
| + // Nested
|
| + printer->Print(
|
| + vars,
|
| + "public static final\n"
|
| + " com.google.protobuf.GeneratedMessage.GeneratedExtension<\n"
|
| + " $containing_type$,\n"
|
| + " $type$> $name$ = com.google.protobuf.GeneratedMessage\n"
|
| + " .newMessageScopedGeneratedExtension(\n"
|
| + " $scope$.getDefaultInstance(),\n"
|
| + " $index$,\n"
|
| + " $singular_type$.class,\n"
|
| + " $prototype$);\n");
|
| + }
|
| } else {
|
| + // Lite extensions
|
| if (descriptor_->is_repeated()) {
|
| - printer->Print(vars,
|
| - "$scope$.$name$.internalInitRepeated(\n"
|
| - " $extendee$.getDefaultInstance(),\n"
|
| - " $prototype$,\n"
|
| - " $enum_map$,\n"
|
| - " $number$,\n"
|
| - " com.google.protobuf.WireFormat.FieldType.$type_constant$,\n"
|
| - " $packed$);\n");
|
| + printer->Print(
|
| + vars,
|
| + "public static final\n"
|
| + " com.google.protobuf.GeneratedMessageLite.GeneratedExtension<\n"
|
| + " $containing_type$,\n"
|
| + " $type$> $name$ = com.google.protobuf.GeneratedMessageLite\n"
|
| + " .newRepeatedGeneratedExtension(\n"
|
| + " $containing_type$.getDefaultInstance(),\n"
|
| + " $prototype$,\n"
|
| + " $enum_map$,\n"
|
| + " $number$,\n"
|
| + " com.google.protobuf.WireFormat.FieldType.$type_constant$,\n"
|
| + " $packed$);\n");
|
| } else {
|
| - printer->Print(vars,
|
| - "$scope$.$name$.internalInitSingular(\n"
|
| - " $extendee$.getDefaultInstance(),\n"
|
| - " $default$,\n"
|
| - " $prototype$,\n"
|
| - " $enum_map$,\n"
|
| - " $number$,\n"
|
| - " com.google.protobuf.WireFormat.FieldType.$type_constant$);\n");
|
| + printer->Print(
|
| + vars,
|
| + "public static final\n"
|
| + " com.google.protobuf.GeneratedMessageLite.GeneratedExtension<\n"
|
| + " $containing_type$,\n"
|
| + " $type$> $name$ = com.google.protobuf.GeneratedMessageLite\n"
|
| + " .newSingularGeneratedExtension(\n"
|
| + " $containing_type$.getDefaultInstance(),\n"
|
| + " $default$,\n"
|
| + " $prototype$,\n"
|
| + " $enum_map$,\n"
|
| + " $number$,\n"
|
| + " com.google.protobuf.WireFormat.FieldType.$type_constant$);\n");
|
| }
|
| }
|
| }
|
|
|
| +void ExtensionGenerator::GenerateNonNestedInitializationCode(
|
| + io::Printer* printer) {
|
| + if (descriptor_->extension_scope() == NULL &&
|
| + HasDescriptorMethods(descriptor_->file())) {
|
| + // Only applies to non-nested, non-lite extensions.
|
| + printer->Print(
|
| + "$name$.internalInit(descriptor.getExtensions().get($index$));\n",
|
| + "name", UnderscoresToCamelCase(descriptor_),
|
| + "index", SimpleItoa(descriptor_->index()));
|
| + }
|
| +}
|
| +
|
| void ExtensionGenerator::GenerateRegistrationCode(io::Printer* printer) {
|
| printer->Print(
|
| "registry.add($scope$.$name$);\n",
|
|
|