| Index: third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_map_field.cc
|
| diff --git a/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_map_field.cc b/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_map_field.cc
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..e5e2f07dd4e13ebdb6df4ebd78566c9c87a27284
|
| --- /dev/null
|
| +++ b/third_party/protobuf/src/google/protobuf/compiler/cpp/cpp_map_field.cc
|
| @@ -0,0 +1,410 @@
|
| +// Protocol Buffers - Google's data interchange format
|
| +// Copyright 2008 Google Inc. All rights reserved.
|
| +// https://developers.google.com/protocol-buffers/
|
| +//
|
| +// Redistribution and use in source and binary forms, with or without
|
| +// modification, are permitted provided that the following conditions are
|
| +// met:
|
| +//
|
| +// * Redistributions of source code must retain the above copyright
|
| +// notice, this list of conditions and the following disclaimer.
|
| +// * Redistributions in binary form must reproduce the above
|
| +// copyright notice, this list of conditions and the following disclaimer
|
| +// in the documentation and/or other materials provided with the
|
| +// distribution.
|
| +// * Neither the name of Google Inc. nor the names of its
|
| +// contributors may be used to endorse or promote products derived from
|
| +// this software without specific prior written permission.
|
| +//
|
| +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
| +// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
| +// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
| +// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
| +// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
| +// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
| +// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
| +// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
| +// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
| +// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
| +// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
| +
|
| +#include <google/protobuf/compiler/cpp/cpp_map_field.h>
|
| +#include <google/protobuf/compiler/cpp/cpp_helpers.h>
|
| +#include <google/protobuf/io/printer.h>
|
| +#include <google/protobuf/wire_format.h>
|
| +#include <google/protobuf/stubs/strutil.h>
|
| +
|
| +namespace google {
|
| +namespace protobuf {
|
| +namespace compiler {
|
| +namespace cpp {
|
| +
|
| +bool IsProto3Field(const FieldDescriptor* field_descriptor) {
|
| + const FileDescriptor* file_descriptor = field_descriptor->file();
|
| + return file_descriptor->syntax() == FileDescriptor::SYNTAX_PROTO3;
|
| +}
|
| +
|
| +void SetMessageVariables(const FieldDescriptor* descriptor,
|
| + map<string, string>* variables,
|
| + const Options& options) {
|
| + SetCommonFieldVariables(descriptor, variables, options);
|
| + (*variables)["type"] = FieldMessageTypeName(descriptor);
|
| + (*variables)["stream_writer"] = (*variables)["declared_type"] +
|
| + (HasFastArraySerialization(descriptor->message_type()->file()) ?
|
| + "MaybeToArray" :
|
| + "");
|
| + (*variables)["full_name"] = descriptor->full_name();
|
| +
|
| + const FieldDescriptor* key =
|
| + descriptor->message_type()->FindFieldByName("key");
|
| + const FieldDescriptor* val =
|
| + descriptor->message_type()->FindFieldByName("value");
|
| + (*variables)["key_cpp"] = PrimitiveTypeName(key->cpp_type());
|
| + switch (val->cpp_type()) {
|
| + case FieldDescriptor::CPPTYPE_MESSAGE:
|
| + (*variables)["val_cpp"] = FieldMessageTypeName(val);
|
| + (*variables)["wrapper"] = "EntryWrapper";
|
| + break;
|
| + case FieldDescriptor::CPPTYPE_ENUM:
|
| + (*variables)["val_cpp"] = ClassName(val->enum_type(), true);
|
| + (*variables)["wrapper"] = "EnumEntryWrapper";
|
| + break;
|
| + default:
|
| + (*variables)["val_cpp"] = PrimitiveTypeName(val->cpp_type());
|
| + (*variables)["wrapper"] = "EntryWrapper";
|
| + }
|
| + (*variables)["key_wire_type"] =
|
| + "::google::protobuf::internal::WireFormatLite::TYPE_" +
|
| + ToUpper(DeclaredTypeMethodName(key->type()));
|
| + (*variables)["val_wire_type"] =
|
| + "::google::protobuf::internal::WireFormatLite::TYPE_" +
|
| + ToUpper(DeclaredTypeMethodName(val->type()));
|
| + (*variables)["map_classname"] = ClassName(descriptor->message_type(), false);
|
| + (*variables)["number"] = SimpleItoa(descriptor->number());
|
| + (*variables)["tag"] = SimpleItoa(internal::WireFormat::MakeTag(descriptor));
|
| +
|
| + if (HasDescriptorMethods(descriptor->file())) {
|
| + (*variables)["lite"] = "";
|
| + } else {
|
| + (*variables)["lite"] = "Lite";
|
| + }
|
| +
|
| + if (!IsProto3Field(descriptor) &&
|
| + val->type() == FieldDescriptor::TYPE_ENUM) {
|
| + const EnumValueDescriptor* default_value = val->default_value_enum();
|
| + (*variables)["default_enum_value"] = Int32ToString(default_value->number());
|
| + } else {
|
| + (*variables)["default_enum_value"] = "0";
|
| + }
|
| +}
|
| +
|
| +MapFieldGenerator::
|
| +MapFieldGenerator(const FieldDescriptor* descriptor,
|
| + const Options& options)
|
| + : descriptor_(descriptor),
|
| + dependent_field_(options.proto_h && IsFieldDependent(descriptor)) {
|
| + SetMessageVariables(descriptor, &variables_, options);
|
| +}
|
| +
|
| +MapFieldGenerator::~MapFieldGenerator() {}
|
| +
|
| +void MapFieldGenerator::
|
| +GeneratePrivateMembers(io::Printer* printer) const {
|
| + printer->Print(variables_,
|
| + "typedef ::google::protobuf::internal::MapEntryLite<\n"
|
| + " $key_cpp$, $val_cpp$,\n"
|
| + " $key_wire_type$,\n"
|
| + " $val_wire_type$,\n"
|
| + " $default_enum_value$ >\n"
|
| + " $map_classname$;\n"
|
| + "::google::protobuf::internal::MapField$lite$<\n"
|
| + " $key_cpp$, $val_cpp$,\n"
|
| + " $key_wire_type$,\n"
|
| + " $val_wire_type$,\n"
|
| + " $default_enum_value$ > $name$_;\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateAccessorDeclarations(io::Printer* printer) const {
|
| + printer->Print(variables_,
|
| + "const ::google::protobuf::Map< $key_cpp$, $val_cpp$ >&\n"
|
| + " $name$() const$deprecation$;\n"
|
| + "::google::protobuf::Map< $key_cpp$, $val_cpp$ >*\n"
|
| + " mutable_$name$()$deprecation$;\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateInlineAccessorDefinitions(io::Printer* printer,
|
| + bool is_inline) const {
|
| + map<string, string> variables(variables_);
|
| + variables["inline"] = is_inline ? "inline" : "";
|
| + printer->Print(variables,
|
| + "$inline$ const ::google::protobuf::Map< $key_cpp$, $val_cpp$ >&\n"
|
| + "$classname$::$name$() const {\n"
|
| + " // @@protoc_insertion_point(field_map:$full_name$)\n"
|
| + " return $name$_.GetMap();\n"
|
| + "}\n"
|
| + "$inline$ ::google::protobuf::Map< $key_cpp$, $val_cpp$ >*\n"
|
| + "$classname$::mutable_$name$() {\n"
|
| + " // @@protoc_insertion_point(field_mutable_map:$full_name$)\n"
|
| + " return $name$_.MutableMap();\n"
|
| + "}\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateClearingCode(io::Printer* printer) const {
|
| + map<string, string> variables(variables_);
|
| + variables["this_message"] = dependent_field_ ? DependentBaseDownCast() : "";
|
| + printer->Print(variables, "$this_message$$name$_.Clear();\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateMergingCode(io::Printer* printer) const {
|
| + printer->Print(variables_, "$name$_.MergeFrom(from.$name$_);\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateSwappingCode(io::Printer* printer) const {
|
| + printer->Print(variables_, "$name$_.Swap(&other->$name$_);\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateConstructorCode(io::Printer* printer) const {
|
| + if (HasDescriptorMethods(descriptor_->file())) {
|
| + printer->Print(variables_,
|
| + "$name$_.SetAssignDescriptorCallback(\n"
|
| + " protobuf_AssignDescriptorsOnce);\n"
|
| + "$name$_.SetEntryDescriptor(\n"
|
| + " &$type$_descriptor_);\n");
|
| + }
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateMergeFromCodedStream(io::Printer* printer) const {
|
| + const FieldDescriptor* value_field =
|
| + descriptor_->message_type()->FindFieldByName("value");
|
| + printer->Print(variables_,
|
| + "::google::protobuf::scoped_ptr<$map_classname$> entry($name$_.NewEntry());\n");
|
| +
|
| + if (IsProto3Field(descriptor_) ||
|
| + value_field->type() != FieldDescriptor::TYPE_ENUM) {
|
| + printer->Print(variables_,
|
| + "DO_(::google::protobuf::internal::WireFormatLite::ReadMessageNoVirtual(\n"
|
| + " input, entry.get()));\n");
|
| + switch (value_field->cpp_type()) {
|
| + case FieldDescriptor::CPPTYPE_MESSAGE:
|
| + printer->Print(variables_,
|
| + "(*mutable_$name$())[entry->key()].Swap("
|
| + "entry->mutable_value());\n");
|
| + break;
|
| + case FieldDescriptor::CPPTYPE_ENUM:
|
| + printer->Print(variables_,
|
| + "(*mutable_$name$())[entry->key()] =\n"
|
| + " static_cast< $val_cpp$ >(*entry->mutable_value());\n");
|
| + break;
|
| + default:
|
| + printer->Print(variables_,
|
| + "(*mutable_$name$())[entry->key()] = *entry->mutable_value();\n");
|
| + break;
|
| + }
|
| + } else {
|
| + printer->Print(variables_,
|
| + "{\n"
|
| + " ::std::string data;\n"
|
| + " DO_(::google::protobuf::internal::WireFormatLite::ReadString(input, &data));\n"
|
| + " DO_(entry->ParseFromString(data));\n"
|
| + " if ($val_cpp$_IsValid(*entry->mutable_value())) {\n"
|
| + " (*mutable_$name$())[entry->key()] =\n"
|
| + " static_cast< $val_cpp$ >(*entry->mutable_value());\n"
|
| + " } else {\n");
|
| + if (HasDescriptorMethods(descriptor_->file())) {
|
| + printer->Print(variables_,
|
| + " mutable_unknown_fields()"
|
| + "->AddLengthDelimited($number$, data);\n");
|
| + } else {
|
| + printer->Print(variables_,
|
| + " unknown_fields_stream.WriteVarint32($tag$);\n"
|
| + " unknown_fields_stream.WriteVarint32(data.size());\n"
|
| + " unknown_fields_stream.WriteString(data);\n");
|
| + }
|
| +
|
| +
|
| + printer->Print(variables_,
|
| + " }\n"
|
| + "}\n");
|
| + }
|
| +
|
| + const FieldDescriptor* key_field =
|
| + descriptor_->message_type()->FindFieldByName("key");
|
| + if (key_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + key_field, true, variables_,
|
| + "entry->key().data(), entry->key().length(),\n", printer);
|
| + }
|
| + if (value_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + value_field, true, variables_,
|
| + "entry->mutable_value()->data(),\n"
|
| + "entry->mutable_value()->length(),\n", printer);
|
| + }
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + "if (entry->GetArena() != NULL) entry.release();\n");
|
| + }
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateSerializeWithCachedSizes(io::Printer* printer) const {
|
| + printer->Print(variables_,
|
| + "{\n"
|
| + " ::google::protobuf::scoped_ptr<$map_classname$> entry;\n"
|
| + " for (::google::protobuf::Map< $key_cpp$, $val_cpp$ >::const_iterator\n"
|
| + " it = this->$name$().begin();\n"
|
| + " it != this->$name$().end(); ++it) {\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print(variables_,
|
| + " entry.reset($name$_.New$wrapper$(it->first, it->second));\n"
|
| + " ::google::protobuf::internal::WireFormatLite::Write$stream_writer$(\n"
|
| + " $number$, *entry, output);\n");
|
| +
|
| + printer->Indent();
|
| + printer->Indent();
|
| +
|
| + const FieldDescriptor* key_field =
|
| + descriptor_->message_type()->FindFieldByName("key");
|
| + const FieldDescriptor* value_field =
|
| + descriptor_->message_type()->FindFieldByName("value");
|
| + if (key_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + key_field, false, variables_,
|
| + "it->first.data(), it->first.length(),\n", printer);
|
| + }
|
| + if (value_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + value_field, false, variables_,
|
| + "it->second.data(), it->second.length(),\n", printer);
|
| + }
|
| +
|
| + printer->Outdent();
|
| + printer->Outdent();
|
| +
|
| + printer->Print(
|
| + " }\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print("}\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateSerializeWithCachedSizesToArray(io::Printer* printer) const {
|
| + printer->Print(variables_,
|
| + "{\n"
|
| + " ::google::protobuf::scoped_ptr<$map_classname$> entry;\n"
|
| + " for (::google::protobuf::Map< $key_cpp$, $val_cpp$ >::const_iterator\n"
|
| + " it = this->$name$().begin();\n"
|
| + " it != this->$name$().end(); ++it) {\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print(variables_,
|
| + " entry.reset($name$_.New$wrapper$(it->first, it->second));\n"
|
| + " target = ::google::protobuf::internal::WireFormatLite::\n"
|
| + " Write$declared_type$NoVirtualToArray(\n"
|
| + " $number$, *entry, target);\n");
|
| +
|
| + printer->Indent();
|
| + printer->Indent();
|
| +
|
| + const FieldDescriptor* key_field =
|
| + descriptor_->message_type()->FindFieldByName("key");
|
| + const FieldDescriptor* value_field =
|
| + descriptor_->message_type()->FindFieldByName("value");
|
| + if (key_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + key_field, false, variables_,
|
| + "it->first.data(), it->first.length(),\n", printer);
|
| + }
|
| + if (value_field->type() == FieldDescriptor::TYPE_STRING) {
|
| + GenerateUtf8CheckCodeForString(
|
| + value_field, false, variables_,
|
| + "it->second.data(), it->second.length(),\n", printer);
|
| + }
|
| +
|
| + printer->Outdent();
|
| + printer->Outdent();
|
| + printer->Print(
|
| + " }\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print("}\n");
|
| +}
|
| +
|
| +void MapFieldGenerator::
|
| +GenerateByteSize(io::Printer* printer) const {
|
| + printer->Print(variables_,
|
| + "total_size += $tag_size$ * this->$name$_size();\n"
|
| + "{\n"
|
| + " ::google::protobuf::scoped_ptr<$map_classname$> entry;\n"
|
| + " for (::google::protobuf::Map< $key_cpp$, $val_cpp$ >::const_iterator\n"
|
| + " it = this->$name$().begin();\n"
|
| + " it != this->$name$().end(); ++it) {\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print(variables_,
|
| + " entry.reset($name$_.New$wrapper$(it->first, it->second));\n"
|
| + " total_size += ::google::protobuf::internal::WireFormatLite::\n"
|
| + " $declared_type$SizeNoVirtual(*entry);\n"
|
| + " }\n");
|
| +
|
| + // If entry is allocated by arena, its desctructor should be avoided.
|
| + if (SupportsArenas(descriptor_)) {
|
| + printer->Print(variables_,
|
| + " if (entry.get() != NULL && entry->GetArena() != NULL) {\n"
|
| + " entry.release();\n"
|
| + " }\n");
|
| + }
|
| +
|
| + printer->Print("}\n");
|
| +}
|
| +
|
| +} // namespace cpp
|
| +} // namespace compiler
|
| +} // namespace protobuf
|
| +} // namespace google
|
|
|