| // 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. |
| |
| // Author: kenton@google.com (Kenton Varda) |
| // Based on original Protocol Buffers design by |
| // Sanjay Ghemawat, Jeff Dean, and others. |
| |
| #include <google/protobuf/compiler/cpp/cpp_message_field.h> |
| #include <google/protobuf/compiler/cpp/cpp_helpers.h> |
| #include <google/protobuf/io/printer.h> |
| |
| #include <google/protobuf/stubs/strutil.h> |
| |
| namespace google { |
| namespace protobuf { |
| namespace compiler { |
| namespace cpp { |
| |
| namespace { |
| |
| // When we are generating code for implicit weak fields, we need to insert some |
| // additional casts. These functions return the casted expression if |
| // implicit_weak_field is true but otherwise return the original expression. |
| // Ordinarily a static_cast is enough to cast google::protobuf::MessageLite* to a class |
| // deriving from it, but we need a reinterpret_cast in cases where the generated |
| // message is forward-declared but its full definition is not visible. |
| string StaticCast(const string& type, const string& expression, |
| bool implicit_weak_field) { |
| if (implicit_weak_field) { |
| return "static_cast< " + type + " >(" + expression + ")"; |
| } else { |
| return expression; |
| } |
| } |
| |
| string ReinterpretCast(const string& type, const string& expression, |
| bool implicit_weak_field) { |
| if (implicit_weak_field) { |
| return "reinterpret_cast< " + type + " >(" + expression + ")"; |
| } else { |
| return expression; |
| } |
| } |
| |
| void SetMessageVariables(const FieldDescriptor* descriptor, |
| const Options& options, bool implicit_weak, |
| std::map<string, string>* variables) { |
| SetCommonFieldVariables(descriptor, variables, options); |
| (*variables)["type"] = FieldMessageTypeName(descriptor); |
| (*variables)["casted_member"] = ReinterpretCast( |
| (*variables)["type"] + "*", (*variables)["name"] + "_", implicit_weak); |
| (*variables)["type_default_instance"] = |
| DefaultInstanceName(descriptor->message_type()); |
| (*variables)["type_reference_function"] = |
| implicit_weak |
| ? (" " + ReferenceFunctionName(descriptor->message_type()) + "();\n") |
| : ""; |
| (*variables)["stream_writer"] = |
| (*variables)["declared_type"] + |
| (HasFastArraySerialization(descriptor->message_type()->file(), options) |
| ? "MaybeToArray" |
| : ""); |
| // NOTE: Escaped here to unblock proto1->proto2 migration. |
| // TODO(liujisi): Extend this to apply for other conflicting methods. |
| (*variables)["release_name"] = |
| SafeFunctionName(descriptor->containing_type(), |
| descriptor, "release_"); |
| (*variables)["full_name"] = descriptor->full_name(); |
| } |
| |
| } // namespace |
| |
| // =================================================================== |
| |
| MessageFieldGenerator::MessageFieldGenerator(const FieldDescriptor* descriptor, |
| const Options& options, |
| SCCAnalyzer* scc_analyzer) |
| : FieldGenerator(options), |
| descriptor_(descriptor), |
| implicit_weak_field_( |
| IsImplicitWeakField(descriptor, options, scc_analyzer)) { |
| SetMessageVariables(descriptor, options, implicit_weak_field_, &variables_); |
| } |
| |
| MessageFieldGenerator::~MessageFieldGenerator() {} |
| |
| void MessageFieldGenerator:: |
| GeneratePrivateMembers(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, "::google::protobuf::MessageLite* $name$_;\n"); |
| } else { |
| printer->Print(variables_, "$type$* $name$_;\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateAccessorDeclarations(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| // These private accessors are used by MergeFrom and |
| // MergePartialFromCodedStream, and their purpose is to provide access to |
| // the field without creating a strong dependency on the message type. |
| printer->Print(variables_, |
| "private:\n" |
| "const ::google::protobuf::MessageLite& _internal_$name$() const;\n" |
| "::google::protobuf::MessageLite* _internal_mutable_$name$();\n" |
| "public:\n"); |
| } else { |
| // This inline accessor directly returns member field and is used in |
| // Serialize such that AFDO profile correctly captures access information to |
| // message fields under serialize. |
| printer->Print(variables_, |
| "private:\n" |
| "const $type$& _internal_$name$() const;\n" |
| "public:\n"); |
| } |
| printer->Print(variables_, |
| "$deprecated_attr$const $type$& $name$() const;\n"); |
| printer->Annotate("name", descriptor_); |
| printer->Print(variables_, "$deprecated_attr$$type$* $release_name$();\n"); |
| printer->Annotate("release_name", descriptor_); |
| printer->Print(variables_, |
| "$deprecated_attr$$type$* ${$mutable_$name$$}$();\n"); |
| printer->Annotate("{", "}", descriptor_); |
| printer->Print(variables_, |
| "$deprecated_attr$void ${$set_allocated_$name$$}$" |
| "($type$* $name$);\n"); |
| printer->Annotate("{", "}", descriptor_); |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| "$deprecated_attr$void " |
| "${$unsafe_arena_set_allocated_$name$$}$(\n" |
| " $type$* $name$);\n"); |
| printer->Annotate("{", "}", descriptor_); |
| printer->Print( |
| variables_, |
| "$deprecated_attr$$type$* ${$unsafe_arena_release_$name$$}$();\n"); |
| printer->Annotate("{", "}", descriptor_); |
| } |
| } |
| |
| void MessageFieldGenerator::GenerateNonInlineAccessorDefinitions( |
| io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "const ::google::protobuf::MessageLite& $classname$::_internal_$name$() const {\n" |
| " if ($name$_ != NULL) {\n" |
| " return *$name$_;\n" |
| " } else if (&$type_default_instance$ != NULL) {\n" |
| " return *reinterpret_cast<const ::google::protobuf::MessageLite*>(\n" |
| " &$type_default_instance$);\n" |
| " } else {\n" |
| " return " |
| "*::google::protobuf::internal::ImplicitWeakMessage::default_instance();\n" |
| " }\n" |
| "}\n"); |
| } |
| if (SupportsArenas(descriptor_)) { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "::google::protobuf::MessageLite* $classname$::_internal_mutable_$name$() {\n" |
| " $set_hasbit$\n" |
| " if ($name$_ == NULL) {\n" |
| " if (&$type_default_instance$ == NULL) {\n" |
| " $name$_ = ::google::protobuf::Arena::CreateMessage<\n" |
| " ::google::protobuf::internal::ImplicitWeakMessage>(\n" |
| " GetArenaNoVirtual());\n" |
| " } else {\n" |
| " $name$_ = reinterpret_cast<const ::google::protobuf::MessageLite*>(\n" |
| " &$type_default_instance$)->New(GetArenaNoVirtual());\n" |
| " }\n" |
| " }\n" |
| " return $name$_;\n" |
| "}\n"); |
| } |
| |
| printer->Print(variables_, |
| "void $classname$::unsafe_arena_set_allocated_$name$(\n" |
| " $type$* $name$) {\n" |
| // If we're not on an arena, free whatever we were holding before. |
| // (If we are on arena, we can just forget the earlier pointer.) |
| " if (GetArenaNoVirtual() == NULL) {\n" |
| " delete $name$_;\n" |
| " }\n" |
| " $name$_ = $name$;\n" |
| " if ($name$) {\n" |
| " $set_hasbit$\n" |
| " } else {\n" |
| " $clear_hasbit$\n" |
| " }\n" |
| " // @@protoc_insertion_point(field_unsafe_arena_set_allocated" |
| ":$full_name$)\n" |
| "}\n"); |
| } else if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "::google::protobuf::MessageLite* $classname$::_internal_mutable_$name$() {\n" |
| " $set_hasbit$\n" |
| " if ($name$_ == NULL) {\n" |
| " if (&$type_default_instance$ == NULL) {\n" |
| " $name$_ = new ::google::protobuf::internal::ImplicitWeakMessage;\n" |
| " } else {\n" |
| " $name$_ = reinterpret_cast<const ::google::protobuf::MessageLite*>(\n" |
| " &$type_default_instance$)->New();\n" |
| " }\n" |
| " }\n" |
| " return $name$_;\n" |
| "}\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateInlineAccessorDefinitions(io::Printer* printer) const { |
| if (!implicit_weak_field_) { |
| printer->Print(variables_, |
| "inline const $type$& $classname$::_internal_$name$() const {\n" |
| " return *$field_member$;\n" |
| "}\n"); |
| } |
| printer->Print(variables_, |
| "inline const $type$& $classname$::$name$() const {\n" |
| " const $type$* p = $casted_member$;\n" |
| " // @@protoc_insertion_point(field_get:$full_name$)\n" |
| " return p != NULL ? *p : *reinterpret_cast<const $type$*>(\n" |
| " &$type_default_instance$);\n" |
| "}\n"); |
| |
| printer->Print(variables_, |
| "inline $type$* $classname$::$release_name$() {\n" |
| " // @@protoc_insertion_point(field_release:$full_name$)\n" |
| "$type_reference_function$" |
| " $clear_hasbit$\n" |
| " $type$* temp = $casted_member$;\n"); |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| " if (GetArenaNoVirtual() != NULL) {\n" |
| " temp = ::google::protobuf::internal::DuplicateIfNonNull(temp);\n" |
| " }\n"); |
| } |
| printer->Print(variables_, |
| " $name$_ = NULL;\n" |
| " return temp;\n" |
| "}\n"); |
| |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| "inline $type$* $classname$::unsafe_arena_release_$name$() {\n" |
| " // @@protoc_insertion_point(field_unsafe_arena_release:$full_name$)\n" |
| "$type_reference_function$" |
| " $clear_hasbit$\n" |
| " $type$* temp = $casted_member$;\n" |
| " $name$_ = NULL;\n" |
| " return temp;\n" |
| "}\n"); |
| } |
| |
| printer->Print(variables_, |
| "inline $type$* $classname$::mutable_$name$() {\n" |
| " $set_hasbit$\n" |
| " if ($name$_ == NULL) {\n" |
| " auto* p = CreateMaybeMessage<$type$>(GetArenaNoVirtual());\n"); |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| " $name$_ = reinterpret_cast<::google::protobuf::MessageLite*>(p);\n"); |
| } else { |
| printer->Print(variables_, |
| " $name$_ = p;\n"); |
| } |
| printer->Print(variables_, |
| " }\n" |
| " // @@protoc_insertion_point(field_mutable:$full_name$)\n" |
| " return $casted_member$;\n" |
| "}\n"); |
| |
| // We handle the most common case inline, and delegate less common cases to |
| // the slow fallback function. |
| printer->Print(variables_, |
| "inline void $classname$::set_allocated_$name$($type$* $name$) {\n" |
| " ::google::protobuf::Arena* message_arena = GetArenaNoVirtual();\n"); |
| printer->Print(variables_, |
| " if (message_arena == NULL) {\n"); |
| if (IsCrossFileMessage(descriptor_)) { |
| printer->Print(variables_, |
| " delete reinterpret_cast< ::google::protobuf::MessageLite*>($name$_);\n"); |
| } else { |
| printer->Print(variables_, |
| " delete $name$_;\n"); |
| } |
| printer->Print(variables_, |
| " }\n" |
| " if ($name$) {\n"); |
| if (SupportsArenas(descriptor_->message_type()) && |
| IsCrossFileMessage(descriptor_)) { |
| // We have to read the arena through the virtual method, because the type |
| // isn't defined in this file. |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena =\n" |
| " reinterpret_cast<::google::protobuf::MessageLite*>($name$)->GetArena();\n"); |
| } else if (!SupportsArenas(descriptor_->message_type())) { |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena = NULL;\n"); |
| } else { |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena =\n" |
| " ::google::protobuf::Arena::GetArena($name$);\n"); |
| } |
| printer->Print(variables_, |
| " if (message_arena != submessage_arena) {\n" |
| " $name$ = ::google::protobuf::internal::GetOwnedMessage(\n" |
| " message_arena, $name$, submessage_arena);\n" |
| " }\n" |
| " $set_hasbit$\n" |
| " } else {\n" |
| " $clear_hasbit$\n" |
| " }\n"); |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| " $name$_ = reinterpret_cast<MessageLite*>($name$);\n"); |
| } else { |
| printer->Print(variables_, |
| " $name$_ = $name$;\n"); |
| } |
| printer->Print(variables_, |
| " // @@protoc_insertion_point(field_set_allocated:$full_name$)\n" |
| "}\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateClearingCode(io::Printer* printer) const { |
| if (!HasFieldPresence(descriptor_->file())) { |
| // If we don't have has-bits, message presence is indicated only by ptr != |
| // NULL. Thus on clear, we need to delete the object. |
| printer->Print(variables_, |
| "if (GetArenaNoVirtual() == NULL && $name$_ != NULL) {\n" |
| " delete $name$_;\n" |
| "}\n" |
| "$name$_ = NULL;\n"); |
| } else { |
| printer->Print(variables_, |
| "if ($name$_ != NULL) $name$_->Clear();\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateMessageClearingCode(io::Printer* printer) const { |
| if (!HasFieldPresence(descriptor_->file())) { |
| // If we don't have has-bits, message presence is indicated only by ptr != |
| // NULL. Thus on clear, we need to delete the object. |
| printer->Print(variables_, |
| "if (GetArenaNoVirtual() == NULL && $name$_ != NULL) {\n" |
| " delete $name$_;\n" |
| "}\n" |
| "$name$_ = NULL;\n"); |
| } else { |
| printer->Print(variables_, |
| "GOOGLE_DCHECK($name$_ != NULL);\n" |
| "$name$_->Clear();\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateMergingCode(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "_internal_mutable_$name$()->CheckTypeAndMergeFrom(\n" |
| " from._internal_$name$());\n"); |
| } else { |
| printer->Print(variables_, |
| "mutable_$name$()->$type$::MergeFrom(from.$name$());\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateSwappingCode(io::Printer* printer) const { |
| printer->Print(variables_, "swap($name$_, other->$name$_);\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateDestructorCode(io::Printer* printer) const { |
| // TODO(gerbens) Remove this when we don't need to destruct default instances. |
| // In google3 a default instance will never get deleted so we don't need to |
| // worry about that but in opensource protobuf default instances are deleted |
| // in shutdown process and we need to take special care when handling them. |
| printer->Print(variables_, |
| "if (this != internal_default_instance()) "); |
| printer->Print(variables_, "delete $name$_;\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateConstructorCode(io::Printer* printer) const { |
| printer->Print(variables_, "$name$_ = NULL;\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateCopyConstructorCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (from.has_$name$()) {\n" |
| " $name$_ = new $type$(*from.$name$_);\n" |
| "} else {\n" |
| " $name$_ = NULL;\n" |
| "}\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateMergeFromCodedStream(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::ReadMessage(\n" |
| " input, _internal_mutable_$name$()));\n"); |
| } else if (descriptor_->type() == FieldDescriptor::TYPE_MESSAGE) { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::ReadMessage(\n" |
| " input, mutable_$name$()));\n"); |
| } else { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::ReadGroup(\n" |
| " $number$, input, mutable_$name$()));\n"); |
| } |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateSerializeWithCachedSizes(io::Printer* printer) const { |
| printer->Print(variables_, |
| "::google::protobuf::internal::WireFormatLite::Write$stream_writer$(\n" |
| " $number$, this->_internal_$name$(), output);\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateSerializeWithCachedSizesToArray(io::Printer* printer) const { |
| printer->Print(variables_, |
| "target = ::google::protobuf::internal::WireFormatLite::\n" |
| " InternalWrite$declared_type$ToArray(\n" |
| " $number$, this->_internal_$name$(), deterministic, target);\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateByteSize(io::Printer* printer) const { |
| printer->Print(variables_, |
| "total_size += $tag_size$ +\n" |
| " ::google::protobuf::internal::WireFormatLite::$declared_type$Size(\n" |
| " *$field_member$);\n"); |
| } |
| |
| // =================================================================== |
| |
| MessageOneofFieldGenerator::MessageOneofFieldGenerator( |
| const FieldDescriptor* descriptor, const Options& options, |
| SCCAnalyzer* scc_analyzer) |
| : MessageFieldGenerator(descriptor, options, scc_analyzer) { |
| SetCommonOneofFieldVariables(descriptor, &variables_); |
| } |
| |
| MessageOneofFieldGenerator::~MessageOneofFieldGenerator() {} |
| |
| void MessageOneofFieldGenerator::GenerateNonInlineAccessorDefinitions( |
| io::Printer* printer) const { |
| printer->Print(variables_, |
| "void $classname$::set_allocated_$name$($type$* $name$) {\n" |
| " ::google::protobuf::Arena* message_arena = GetArenaNoVirtual();\n" |
| " clear_$oneof_name$();\n" |
| " if ($name$) {\n"); |
| if (SupportsArenas(descriptor_->message_type()) && |
| descriptor_->file() != descriptor_->message_type()->file()) { |
| // We have to read the arena through the virtual method, because the type |
| // isn't defined in this file. |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena =\n" |
| " reinterpret_cast<::google::protobuf::MessageLite*>($name$)->GetArena();\n"); |
| } else if (!SupportsArenas(descriptor_->message_type())) { |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena = NULL;\n"); |
| } else { |
| printer->Print(variables_, |
| " ::google::protobuf::Arena* submessage_arena =\n" |
| " ::google::protobuf::Arena::GetArena($name$);\n"); |
| } |
| printer->Print(variables_, |
| " if (message_arena != submessage_arena) {\n" |
| " $name$ = ::google::protobuf::internal::GetOwnedMessage(\n" |
| " message_arena, $name$, submessage_arena);\n" |
| " }\n" |
| " set_has_$name$();\n" |
| " $field_member$ = $name$;\n" |
| " }\n" |
| " // @@protoc_insertion_point(field_set_allocated:$full_name$)\n" |
| "}\n"); |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateInlineAccessorDefinitions(io::Printer* printer) const { |
| if (!implicit_weak_field_) { |
| printer->Print(variables_, |
| "inline const $type$& $classname$::_internal_$name$() const {\n" |
| " return *$field_member$;\n" |
| "}\n"); |
| } |
| printer->Print(variables_, |
| "inline $type$* $classname$::$release_name$() {\n" |
| " // @@protoc_insertion_point(field_release:$full_name$)\n" |
| " if (has_$name$()) {\n" |
| " clear_has_$oneof_name$();\n" |
| " $type$* temp = $field_member$;\n"); |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| " if (GetArenaNoVirtual() != NULL) {\n" |
| " temp = ::google::protobuf::internal::DuplicateIfNonNull(temp);\n" |
| " }\n"); |
| } |
| printer->Print(variables_, |
| " $field_member$ = NULL;\n" |
| " return temp;\n" |
| " } else {\n" |
| " return NULL;\n" |
| " }\n" |
| "}\n"); |
| |
| printer->Print(variables_, |
| "inline const $type$& $classname$::$name$() const {\n" |
| " // @@protoc_insertion_point(field_get:$full_name$)\n" |
| " return has_$name$()\n" |
| " ? *$field_member$\n" |
| " : *reinterpret_cast< $type$*>(&$type_default_instance$);\n" |
| "}\n"); |
| |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| "inline $type$* $classname$::unsafe_arena_release_$name$() {\n" |
| " // @@protoc_insertion_point(field_unsafe_arena_release" |
| ":$full_name$)\n" |
| " if (has_$name$()) {\n" |
| " clear_has_$oneof_name$();\n" |
| " $type$* temp = $field_member$;\n" |
| " $field_member$ = NULL;\n" |
| " return temp;\n" |
| " } else {\n" |
| " return NULL;\n" |
| " }\n" |
| "}\n" |
| "inline void $classname$::unsafe_arena_set_allocated_$name$" |
| "($type$* $name$) {\n" |
| // We rely on the oneof clear method to free the earlier contents of this |
| // oneof. We can directly use the pointer we're given to set the new |
| // value. |
| " clear_$oneof_name$();\n" |
| " if ($name$) {\n" |
| " set_has_$name$();\n" |
| " $field_member$ = $name$;\n" |
| " }\n" |
| " // @@protoc_insertion_point(field_unsafe_arena_set_allocated:" |
| "$full_name$)\n" |
| "}\n"); |
| } |
| |
| printer->Print(variables_, |
| "inline $type$* $classname$::mutable_$name$() {\n" |
| " if (!has_$name$()) {\n" |
| " clear_$oneof_name$();\n" |
| " set_has_$name$();\n" |
| " $field_member$ = CreateMaybeMessage< $type$ >(\n" |
| " GetArenaNoVirtual());\n" |
| " }\n" |
| " // @@protoc_insertion_point(field_mutable:$full_name$)\n" |
| " return $field_member$;\n" |
| "}\n"); |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateClearingCode(io::Printer* printer) const { |
| if (SupportsArenas(descriptor_)) { |
| printer->Print(variables_, |
| "if (GetArenaNoVirtual() == NULL) {\n" |
| " delete $field_member$;\n" |
| "}\n"); |
| } else { |
| printer->Print(variables_, |
| "delete $field_member$;\n"); |
| } |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateMessageClearingCode(io::Printer* printer) const { |
| GenerateClearingCode(printer); |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateSwappingCode(io::Printer* printer) const { |
| // Don't print any swapping code. Swapping the union will swap this field. |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateDestructorCode(io::Printer* printer) const { |
| // We inherit from MessageFieldGenerator, so we need to override the default |
| // behavior. |
| } |
| |
| void MessageOneofFieldGenerator:: |
| GenerateConstructorCode(io::Printer* printer) const { |
| // Don't print any constructor code. The field is in a union. We allocate |
| // space only when this field is used. |
| } |
| |
| // =================================================================== |
| |
| RepeatedMessageFieldGenerator::RepeatedMessageFieldGenerator( |
| const FieldDescriptor* descriptor, const Options& options, |
| SCCAnalyzer* scc_analyzer) |
| : FieldGenerator(options), |
| descriptor_(descriptor), |
| implicit_weak_field_( |
| IsImplicitWeakField(descriptor, options, scc_analyzer)) { |
| SetMessageVariables(descriptor, options, implicit_weak_field_, &variables_); |
| } |
| |
| RepeatedMessageFieldGenerator::~RepeatedMessageFieldGenerator() {} |
| |
| void RepeatedMessageFieldGenerator:: |
| GeneratePrivateMembers(io::Printer* printer) const { |
| printer->Print(variables_, |
| "::google::protobuf::RepeatedPtrField< $type$ > $name$_;\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateAccessorDeclarations(io::Printer* printer) const { |
| printer->Print(variables_, |
| "$deprecated_attr$$type$* ${$mutable_$name$$}$(int index);\n"); |
| printer->Annotate("{", "}", descriptor_); |
| printer->Print(variables_, |
| "$deprecated_attr$::google::protobuf::RepeatedPtrField< $type$ >*\n" |
| " ${$mutable_$name$$}$();\n"); |
| printer->Annotate("{", "}", descriptor_); |
| |
| printer->Print(variables_, |
| "$deprecated_attr$const $type$& $name$(int index) const;\n"); |
| printer->Annotate("name", descriptor_); |
| printer->Print(variables_, "$deprecated_attr$$type$* ${$add_$name$$}$();\n"); |
| printer->Annotate("{", "}", descriptor_); |
| printer->Print(variables_, |
| "$deprecated_attr$const ::google::protobuf::RepeatedPtrField< $type$ >&\n" |
| " $name$() const;\n"); |
| printer->Annotate("name", descriptor_); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateInlineAccessorDefinitions(io::Printer* printer) const { |
| printer->Print(variables_, |
| "inline $type$* $classname$::mutable_$name$(int index) {\n" |
| // TODO(dlj): move insertion points |
| " // @@protoc_insertion_point(field_mutable:$full_name$)\n" |
| "$type_reference_function$" |
| " return $name$_.Mutable(index);\n" |
| "}\n" |
| "inline ::google::protobuf::RepeatedPtrField< $type$ >*\n" |
| "$classname$::mutable_$name$() {\n" |
| " // @@protoc_insertion_point(field_mutable_list:$full_name$)\n" |
| "$type_reference_function$" |
| " return &$name$_;\n" |
| "}\n"); |
| |
| if (options_.safe_boundary_check) { |
| printer->Print(variables_, |
| "inline const $type$& $classname$::$name$(int index) const {\n" |
| " // @@protoc_insertion_point(field_get:$full_name$)\n" |
| " return $name$_.InternalCheckedGet(index,\n" |
| " *reinterpret_cast<const $type$*>(&$type_default_instance$));\n" |
| "}\n"); |
| } else { |
| printer->Print(variables_, |
| "inline const $type$& $classname$::$name$(int index) const {\n" |
| " // @@protoc_insertion_point(field_get:$full_name$)\n" |
| "$type_reference_function$" |
| " return $name$_.Get(index);\n" |
| "}\n"); |
| } |
| |
| printer->Print(variables_, |
| "inline $type$* $classname$::add_$name$() {\n" |
| " // @@protoc_insertion_point(field_add:$full_name$)\n" |
| " return $name$_.Add();\n" |
| "}\n"); |
| |
| printer->Print(variables_, |
| "inline const ::google::protobuf::RepeatedPtrField< $type$ >&\n" |
| "$classname$::$name$() const {\n" |
| " // @@protoc_insertion_point(field_list:$full_name$)\n" |
| "$type_reference_function$" |
| " return $name$_;\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateClearingCode(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print( |
| variables_, |
| "CastToBase(&$name$_)->Clear<" |
| "::google::protobuf::internal::ImplicitWeakTypeHandler<$type$>>();\n"); |
| } else { |
| printer->Print(variables_, "$name$_.Clear();\n"); |
| } |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateMergingCode(io::Printer* printer) const { |
| if (implicit_weak_field_) { |
| printer->Print( |
| variables_, |
| "CastToBase(&$name$_)->MergeFrom<" |
| "::google::protobuf::internal::ImplicitWeakTypeHandler<$type$>>(CastToBase(" |
| "from.$name$_));\n"); |
| } else { |
| printer->Print(variables_, "$name$_.MergeFrom(from.$name$_);\n"); |
| } |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateSwappingCode(io::Printer* printer) const { |
| printer->Print( |
| variables_, |
| "CastToBase(&$name$_)->InternalSwap(CastToBase(&other->$name$_));\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateConstructorCode(io::Printer* printer) const { |
| // Not needed for repeated fields. |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateMergeFromCodedStream(io::Printer* printer) const { |
| if (descriptor_->type() == FieldDescriptor::TYPE_MESSAGE) { |
| if (implicit_weak_field_) { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::" |
| "ReadMessage(input, CastToBase(&$name$_)->AddWeak(\n" |
| " reinterpret_cast<const ::google::protobuf::MessageLite*>(\n" |
| " &$type_default_instance$))));\n"); |
| } else { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::" |
| "ReadMessage(\n" |
| " input, add_$name$()));\n"); |
| } |
| } else { |
| printer->Print(variables_, |
| "DO_(::google::protobuf::internal::WireFormatLite::" |
| "ReadGroup($number$, input, add_$name$()));\n"); |
| } |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateSerializeWithCachedSizes(io::Printer* printer) const { |
| printer->Print(variables_, |
| "for (unsigned int i = 0,\n" |
| " n = static_cast<unsigned int>(this->$name$_size()); i < n; i++) {\n" |
| " ::google::protobuf::internal::WireFormatLite::Write$stream_writer$(\n" |
| " $number$,\n"); |
| if (implicit_weak_field_) { |
| printer->Print( |
| variables_, |
| " CastToBase($name$_).Get<" |
| "::google::protobuf::internal::ImplicitWeakTypeHandler<$type$>>(" |
| "static_cast<int>(i)),\n"); |
| } else { |
| printer->Print(variables_, |
| " this->$name$(static_cast<int>(i)),\n"); |
| } |
| printer->Print(variables_, |
| " output);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateSerializeWithCachedSizesToArray(io::Printer* printer) const { |
| printer->Print(variables_, |
| "for (unsigned int i = 0,\n" |
| " n = static_cast<unsigned int>(this->$name$_size()); i < n; i++) {\n" |
| " target = ::google::protobuf::internal::WireFormatLite::\n" |
| " InternalWrite$declared_type$ToArray(\n" |
| " $number$, this->$name$(static_cast<int>(i)), deterministic, target);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateByteSize(io::Printer* printer) const { |
| printer->Print(variables_, |
| "{\n" |
| " unsigned int count = static_cast<unsigned int>(this->$name$_size());\n"); |
| printer->Indent(); |
| printer->Print(variables_, |
| "total_size += $tag_size$UL * count;\n" |
| "for (unsigned int i = 0; i < count; i++) {\n" |
| " total_size +=\n" |
| " ::google::protobuf::internal::WireFormatLite::$declared_type$Size(\n"); |
| if (implicit_weak_field_) { |
| printer->Print( |
| variables_, |
| " CastToBase($name$_).Get<" |
| "::google::protobuf::internal::ImplicitWeakTypeHandler<$type$>>(" |
| "static_cast<int>(i)));\n"); |
| } else { |
| printer->Print(variables_, |
| " this->$name$(static_cast<int>(i)));\n"); |
| } |
| printer->Print(variables_, "}\n"); |
| printer->Outdent(); |
| printer->Print("}\n"); |
| } |
| |
| } // namespace cpp |
| } // namespace compiler |
| } // namespace protobuf |
| } // namespace google |
