1 // Protocol Buffers - Google's data interchange format
2 // Copyright 2008 Google Inc. All rights reserved.
3 // https://developers.google.com/protocol-buffers/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are
7 // met:
8 //
9 // * Redistributions of source code must retain the above copyright
10 // notice, this list of conditions and the following disclaimer.
11 // * Redistributions in binary form must reproduce the above
12 // copyright notice, this list of conditions and the following disclaimer
13 // in the documentation and/or other materials provided with the
14 // distribution.
15 // * Neither the name of Google Inc. nor the names of its
16 // contributors may be used to endorse or promote products derived from
17 // this software without specific prior written permission.
18 //
19 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
22 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
23 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
24 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
25 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
26 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
27 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
28 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
29 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
30
31 #include <sstream>
32
33 #include <google/protobuf/compiler/code_generator.h>
34 #include <google/protobuf/descriptor.h>
35 #include <google/protobuf/descriptor.pb.h>
36 #include <google/protobuf/io/printer.h>
37 #include <google/protobuf/io/zero_copy_stream.h>
38
39 #include <google/protobuf/compiler/csharp/csharp_doc_comment.h>
40 #include <google/protobuf/compiler/csharp/csharp_helpers.h>
41 #include <google/protobuf/compiler/csharp/csharp_repeated_message_field.h>
42 #include <google/protobuf/compiler/csharp/csharp_message_field.h>
43 #include <google/protobuf/compiler/csharp/csharp_wrapper_field.h>
44
45 namespace google {
46 namespace protobuf {
47 namespace compiler {
48 namespace csharp {
49
RepeatedMessageFieldGenerator(const FieldDescriptor * descriptor,int presenceIndex,const Options * options)50 RepeatedMessageFieldGenerator::RepeatedMessageFieldGenerator(
51 const FieldDescriptor* descriptor, int presenceIndex, const Options *options)
52 : FieldGeneratorBase(descriptor, presenceIndex, options) {
53 }
54
~RepeatedMessageFieldGenerator()55 RepeatedMessageFieldGenerator::~RepeatedMessageFieldGenerator() {
56
57 }
58
GenerateMembers(io::Printer * printer)59 void RepeatedMessageFieldGenerator::GenerateMembers(io::Printer* printer) {
60 printer->Print(
61 variables_,
62 "private static readonly pb::FieldCodec<$type_name$> _repeated_$name$_codec\n"
63 " = ");
64 // Don't want to duplicate the codec code here... maybe we should have a
65 // "create single field generator for this repeated field"
66 // function, but it doesn't seem worth it for just this.
67 if (IsWrapperType(descriptor_)) {
68 std::unique_ptr<FieldGeneratorBase> single_generator(
69 new WrapperFieldGenerator(descriptor_, presenceIndex_, this->options()));
70 single_generator->GenerateCodecCode(printer);
71 } else {
72 std::unique_ptr<FieldGeneratorBase> single_generator(
73 new MessageFieldGenerator(descriptor_, presenceIndex_, this->options()));
74 single_generator->GenerateCodecCode(printer);
75 }
76 printer->Print(";\n");
77 printer->Print(
78 variables_,
79 "private readonly pbc::RepeatedField<$type_name$> $name$_ = new pbc::RepeatedField<$type_name$>();\n");
80 WritePropertyDocComment(printer, descriptor_);
81 AddPublicMemberAttributes(printer);
82 printer->Print(
83 variables_,
84 "$access_level$ pbc::RepeatedField<$type_name$> $property_name$ {\n"
85 " get { return $name$_; }\n"
86 "}\n");
87 }
88
GenerateMergingCode(io::Printer * printer)89 void RepeatedMessageFieldGenerator::GenerateMergingCode(io::Printer* printer) {
90 printer->Print(
91 variables_,
92 "$name$_.Add(other.$name$_);\n");
93 }
94
GenerateParsingCode(io::Printer * printer)95 void RepeatedMessageFieldGenerator::GenerateParsingCode(io::Printer* printer) {
96 GenerateParsingCode(printer, true);
97 }
98
GenerateParsingCode(io::Printer * printer,bool use_parse_context)99 void RepeatedMessageFieldGenerator::GenerateParsingCode(io::Printer* printer, bool use_parse_context) {
100 printer->Print(
101 variables_,
102 use_parse_context
103 ? "$name$_.AddEntriesFrom(ref input, _repeated_$name$_codec);\n"
104 : "$name$_.AddEntriesFrom(input, _repeated_$name$_codec);\n");
105 }
106
GenerateSerializationCode(io::Printer * printer)107 void RepeatedMessageFieldGenerator::GenerateSerializationCode(io::Printer* printer) {
108 GenerateSerializationCode(printer, true);
109 }
110
GenerateSerializationCode(io::Printer * printer,bool use_write_context)111 void RepeatedMessageFieldGenerator::GenerateSerializationCode(io::Printer* printer, bool use_write_context) {
112 printer->Print(
113 variables_,
114 use_write_context
115 ? "$name$_.WriteTo(ref output, _repeated_$name$_codec);\n"
116 : "$name$_.WriteTo(output, _repeated_$name$_codec);\n");
117 }
118
GenerateSerializedSizeCode(io::Printer * printer)119 void RepeatedMessageFieldGenerator::GenerateSerializedSizeCode(io::Printer* printer) {
120 printer->Print(
121 variables_,
122 "size += $name$_.CalculateSize(_repeated_$name$_codec);\n");
123 }
124
WriteHash(io::Printer * printer)125 void RepeatedMessageFieldGenerator::WriteHash(io::Printer* printer) {
126 printer->Print(
127 variables_,
128 "hash ^= $name$_.GetHashCode();\n");
129 }
130
WriteEquals(io::Printer * printer)131 void RepeatedMessageFieldGenerator::WriteEquals(io::Printer* printer) {
132 printer->Print(
133 variables_,
134 "if(!$name$_.Equals(other.$name$_)) return false;\n");
135 }
136
WriteToString(io::Printer * printer)137 void RepeatedMessageFieldGenerator::WriteToString(io::Printer* printer) {
138 variables_["field_name"] = GetFieldName(descriptor_);
139 printer->Print(
140 variables_,
141 "PrintField(\"$field_name$\", $name$_, writer);\n");
142 }
143
GenerateCloningCode(io::Printer * printer)144 void RepeatedMessageFieldGenerator::GenerateCloningCode(io::Printer* printer) {
145 printer->Print(variables_,
146 "$name$_ = other.$name$_.Clone();\n");
147 }
148
GenerateFreezingCode(io::Printer * printer)149 void RepeatedMessageFieldGenerator::GenerateFreezingCode(io::Printer* printer) {
150 }
151
GenerateExtensionCode(io::Printer * printer)152 void RepeatedMessageFieldGenerator::GenerateExtensionCode(io::Printer* printer) {
153 WritePropertyDocComment(printer, descriptor_);
154 AddDeprecatedFlag(printer);
155 printer->Print(
156 variables_,
157 "$access_level$ static readonly pb::RepeatedExtension<$extended_type$, $type_name$> $property_name$ =\n"
158 " new pb::RepeatedExtension<$extended_type$, $type_name$>($number$, ");
159 if (IsWrapperType(descriptor_)) {
160 std::unique_ptr<FieldGeneratorBase> single_generator(
161 new WrapperFieldGenerator(descriptor_, -1, this->options()));
162 single_generator->GenerateCodecCode(printer);
163 } else {
164 std::unique_ptr<FieldGeneratorBase> single_generator(
165 new MessageFieldGenerator(descriptor_, -1, this->options()));
166 single_generator->GenerateCodecCode(printer);
167 }
168 printer->Print(");\n");
169 }
170
171 } // namespace csharp
172 } // namespace compiler
173 } // namespace protobuf
174 } // namespace google
175