| // Protocol Buffers - Google's data interchange format |
| // Copyright 2008 Google Inc. |
| // http://code.google.com/p/protobuf/ |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| // Author: kenton@google.com (Kenton Varda) |
| // Based on original Protocol Buffers design by |
| // Sanjay Ghemawat, Jeff Dean, and others. |
| |
| #include <map> |
| #include <string> |
| |
| #include <google/protobuf/compiler/java/java_message_field.h> |
| #include <google/protobuf/compiler/java/java_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 java { |
| |
| namespace { |
| |
| // TODO(kenton): Factor out a "SetCommonFieldVariables()" to get rid of |
| // repeat code between this and the other field types. |
| void SetMessageVariables(const FieldDescriptor* descriptor, |
| map<string, string>* variables) { |
| (*variables)["name"] = |
| UnderscoresToCamelCase(descriptor); |
| (*variables)["capitalized_name"] = |
| UnderscoresToCapitalizedCamelCase(descriptor); |
| (*variables)["number"] = SimpleItoa(descriptor->number()); |
| (*variables)["type"] = ClassName(descriptor->message_type()); |
| (*variables)["group_or_message"] = |
| (descriptor->type() == FieldDescriptor::TYPE_GROUP) ? |
| "Group" : "Message"; |
| } |
| |
| } // namespace |
| |
| // =================================================================== |
| |
| MessageFieldGenerator:: |
| MessageFieldGenerator(const FieldDescriptor* descriptor) |
| : descriptor_(descriptor) { |
| SetMessageVariables(descriptor, &variables_); |
| } |
| |
| MessageFieldGenerator::~MessageFieldGenerator() {} |
| |
| void MessageFieldGenerator:: |
| GenerateMembers(io::Printer* printer) const { |
| printer->Print(variables_, |
| "private boolean has$capitalized_name$;\n" |
| "private $type$ $name$_ = $type$.getDefaultInstance();\n" |
| "public boolean has$capitalized_name$() { return has$capitalized_name$; }\n" |
| "public $type$ get$capitalized_name$() { return $name$_; }\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateBuilderMembers(io::Printer* printer) const { |
| printer->Print(variables_, |
| "public boolean has$capitalized_name$() {\n" |
| " return result.has$capitalized_name$();\n" |
| "}\n" |
| "public $type$ get$capitalized_name$() {\n" |
| " return result.get$capitalized_name$();\n" |
| "}\n" |
| "public Builder set$capitalized_name$($type$ value) {\n" |
| " result.has$capitalized_name$ = true;\n" |
| " result.$name$_ = value;\n" |
| " return this;\n" |
| "}\n" |
| "public Builder set$capitalized_name$($type$.Builder builderForValue) {\n" |
| " result.has$capitalized_name$ = true;\n" |
| " result.$name$_ = builderForValue.build();\n" |
| " return this;\n" |
| "}\n" |
| "public Builder merge$capitalized_name$($type$ value) {\n" |
| " if (result.has$capitalized_name$() &&\n" |
| " result.$name$_ != $type$.getDefaultInstance()) {\n" |
| " result.$name$_ =\n" |
| " $type$.newBuilder(result.$name$_).mergeFrom(value).buildPartial();\n" |
| " } else {\n" |
| " result.$name$_ = value;\n" |
| " }\n" |
| " result.has$capitalized_name$ = true;\n" |
| " return this;\n" |
| "}\n" |
| "public Builder clear$capitalized_name$() {\n" |
| " result.has$capitalized_name$ = false;\n" |
| " result.$name$_ = $type$.getDefaultInstance();\n" |
| " return this;\n" |
| "}\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateMergingCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (other.has$capitalized_name$()) {\n" |
| " merge$capitalized_name$(other.get$capitalized_name$());\n" |
| "}\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateBuildingCode(io::Printer* printer) const { |
| // Nothing to do for singular fields. |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateParsingCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "$type$.Builder subBuilder = $type$.newBuilder();\n" |
| "if (has$capitalized_name$()) {\n" |
| " subBuilder.mergeFrom(get$capitalized_name$());\n" |
| "}\n"); |
| |
| if (descriptor_->type() == FieldDescriptor::TYPE_GROUP) { |
| printer->Print(variables_, |
| "input.readGroup($number$, subBuilder, extensionRegistry);\n"); |
| } else { |
| printer->Print(variables_, |
| "input.readMessage(subBuilder, extensionRegistry);\n"); |
| } |
| |
| printer->Print(variables_, |
| "set$capitalized_name$(subBuilder.buildPartial());\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateSerializationCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (has$capitalized_name$()) {\n" |
| " output.write$group_or_message$($number$, get$capitalized_name$());\n" |
| "}\n"); |
| } |
| |
| void MessageFieldGenerator:: |
| GenerateSerializedSizeCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (has$capitalized_name$()) {\n" |
| " size += com.google.protobuf.CodedOutputStream\n" |
| " .compute$group_or_message$Size($number$, get$capitalized_name$());\n" |
| "}\n"); |
| } |
| |
| string MessageFieldGenerator::GetBoxedType() const { |
| return ClassName(descriptor_->message_type()); |
| } |
| |
| // =================================================================== |
| |
| RepeatedMessageFieldGenerator:: |
| RepeatedMessageFieldGenerator(const FieldDescriptor* descriptor) |
| : descriptor_(descriptor) { |
| SetMessageVariables(descriptor, &variables_); |
| } |
| |
| RepeatedMessageFieldGenerator::~RepeatedMessageFieldGenerator() {} |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateMembers(io::Printer* printer) const { |
| printer->Print(variables_, |
| "private java.util.List<$type$> $name$_ =\n" |
| " java.util.Collections.emptyList();\n" |
| "public java.util.List<$type$> get$capitalized_name$List() {\n" |
| " return $name$_;\n" // note: unmodifiable list |
| "}\n" |
| "public int get$capitalized_name$Count() { return $name$_.size(); }\n" |
| "public $type$ get$capitalized_name$(int index) {\n" |
| " return $name$_.get(index);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateBuilderMembers(io::Printer* printer) const { |
| printer->Print(variables_, |
| // Note: We return an unmodifiable list because otherwise the caller |
| // could hold on to the returned list and modify it after the message |
| // has been built, thus mutating the message which is supposed to be |
| // immutable. |
| "public java.util.List<$type$> get$capitalized_name$List() {\n" |
| " return java.util.Collections.unmodifiableList(result.$name$_);\n" |
| "}\n" |
| "public int get$capitalized_name$Count() {\n" |
| " return result.get$capitalized_name$Count();\n" |
| "}\n" |
| "public $type$ get$capitalized_name$(int index) {\n" |
| " return result.get$capitalized_name$(index);\n" |
| "}\n" |
| "public Builder set$capitalized_name$(int index, $type$ value) {\n" |
| " result.$name$_.set(index, value);\n" |
| " return this;\n" |
| "}\n" |
| "public Builder set$capitalized_name$(int index, " |
| "$type$.Builder builderForValue) {\n" |
| " result.$name$_.set(index, builderForValue.build());\n" |
| " return this;\n" |
| "}\n" |
| "public Builder add$capitalized_name$($type$ value) {\n" |
| " if (result.$name$_.isEmpty()) {\n" |
| " result.$name$_ = new java.util.ArrayList<$type$>();\n" |
| " }\n" |
| " result.$name$_.add(value);\n" |
| " return this;\n" |
| "}\n" |
| "public Builder add$capitalized_name$($type$.Builder builderForValue) {\n" |
| " if (result.$name$_.isEmpty()) {\n" |
| " result.$name$_ = new java.util.ArrayList<$type$>();\n" |
| " }\n" |
| " result.$name$_.add(builderForValue.build());\n" |
| " return this;\n" |
| "}\n" |
| "public Builder addAll$capitalized_name$(\n" |
| " java.lang.Iterable<? extends $type$> values) {\n" |
| " if (result.$name$_.isEmpty()) {\n" |
| " result.$name$_ = new java.util.ArrayList<$type$>();\n" |
| " }\n" |
| " super.addAll(values, result.$name$_);\n" |
| " return this;\n" |
| "}\n" |
| "public Builder clear$capitalized_name$() {\n" |
| " result.$name$_ = java.util.Collections.emptyList();\n" |
| " return this;\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateMergingCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (!other.$name$_.isEmpty()) {\n" |
| " if (result.$name$_.isEmpty()) {\n" |
| " result.$name$_ = new java.util.ArrayList<$type$>();\n" |
| " }\n" |
| " result.$name$_.addAll(other.$name$_);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateBuildingCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "if (result.$name$_ != java.util.Collections.EMPTY_LIST) {\n" |
| " result.$name$_ =\n" |
| " java.util.Collections.unmodifiableList(result.$name$_);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateParsingCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "$type$.Builder subBuilder = $type$.newBuilder();\n"); |
| |
| if (descriptor_->type() == FieldDescriptor::TYPE_GROUP) { |
| printer->Print(variables_, |
| "input.readGroup($number$, subBuilder, extensionRegistry);\n"); |
| } else { |
| printer->Print(variables_, |
| "input.readMessage(subBuilder, extensionRegistry);\n"); |
| } |
| |
| printer->Print(variables_, |
| "add$capitalized_name$(subBuilder.buildPartial());\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateSerializationCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "for ($type$ element : get$capitalized_name$List()) {\n" |
| " output.write$group_or_message$($number$, element);\n" |
| "}\n"); |
| } |
| |
| void RepeatedMessageFieldGenerator:: |
| GenerateSerializedSizeCode(io::Printer* printer) const { |
| printer->Print(variables_, |
| "for ($type$ element : get$capitalized_name$List()) {\n" |
| " size += com.google.protobuf.CodedOutputStream\n" |
| " .compute$group_or_message$Size($number$, element);\n" |
| "}\n"); |
| } |
| |
| string RepeatedMessageFieldGenerator::GetBoxedType() const { |
| return ClassName(descriptor_->message_type()); |
| } |
| |
| } // namespace java |
| } // namespace compiler |
| } // namespace protobuf |
| } // namespace google |