1// Protocol Buffers - Google's data interchange format
2// Copyright 2015 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#include <google/protobuf/stubs/strutil.h>
39
40#include <google/protobuf/compiler/csharp/csharp_doc_comment.h>
41#include <google/protobuf/compiler/csharp/csharp_helpers.h>
42#include <google/protobuf/compiler/csharp/csharp_map_field.h>
43
44namespace google {
45namespace protobuf {
46namespace compiler {
47namespace csharp {
48
49MapFieldGenerator::MapFieldGenerator(const FieldDescriptor* descriptor,
50 int presenceIndex,
51 const Options* options)
52 : FieldGeneratorBase(descriptor, presenceIndex, options) {
53}
54
55MapFieldGenerator::~MapFieldGenerator() {
56}
57
58void MapFieldGenerator::GenerateMembers(io::Printer* printer) {
59 const FieldDescriptor* key_descriptor =
60 descriptor_->message_type()->map_key();
61 const FieldDescriptor* value_descriptor =
62 descriptor_->message_type()->map_value();
63 variables_["key_type_name"] = type_name(descriptor: key_descriptor);
64 variables_["value_type_name"] = type_name(descriptor: value_descriptor);
65 std::unique_ptr<FieldGeneratorBase> key_generator(
66 CreateFieldGenerator(descriptor: key_descriptor, presenceIndex: 1, options: this->options()));
67 std::unique_ptr<FieldGeneratorBase> value_generator(
68 CreateFieldGenerator(descriptor: value_descriptor, presenceIndex: 2, options: this->options()));
69
70 printer->Print(
71 variables: variables_,
72 text: "private static readonly pbc::MapField<$key_type_name$, $value_type_name$>.Codec _map_$name$_codec\n"
73 " = new pbc::MapField<$key_type_name$, $value_type_name$>.Codec(");
74 key_generator->GenerateCodecCode(printer);
75 printer->Print(text: ", ");
76 value_generator->GenerateCodecCode(printer);
77 printer->Print(
78 variables: variables_,
79 text: ", $tag$);\n"
80 "private readonly pbc::MapField<$key_type_name$, $value_type_name$> $name$_ = new pbc::MapField<$key_type_name$, $value_type_name$>();\n");
81 WritePropertyDocComment(printer, field: descriptor_);
82 AddPublicMemberAttributes(printer);
83 printer->Print(
84 variables: variables_,
85 text: "$access_level$ pbc::MapField<$key_type_name$, $value_type_name$> $property_name$ {\n"
86 " get { return $name$_; }\n"
87 "}\n");
88}
89
90void MapFieldGenerator::GenerateMergingCode(io::Printer* printer) {
91 printer->Print(
92 variables: variables_,
93 text: "$name$_.Add(other.$name$_);\n");
94}
95
96void MapFieldGenerator::GenerateParsingCode(io::Printer* printer) {
97 GenerateParsingCode(printer, use_parse_context: true);
98}
99
100void MapFieldGenerator::GenerateParsingCode(io::Printer* printer, bool use_parse_context) {
101 printer->Print(
102 variables: variables_,
103 text: use_parse_context
104 ? "$name$_.AddEntriesFrom(ref input, _map_$name$_codec);\n"
105 : "$name$_.AddEntriesFrom(input, _map_$name$_codec);\n");
106}
107
108void MapFieldGenerator::GenerateSerializationCode(io::Printer* printer) {
109 GenerateSerializationCode(printer, use_write_context: true);
110}
111
112void MapFieldGenerator::GenerateSerializationCode(io::Printer* printer, bool use_write_context) {
113 printer->Print(
114 variables: variables_,
115 text: use_write_context
116 ? "$name$_.WriteTo(ref output, _map_$name$_codec);\n"
117 : "$name$_.WriteTo(output, _map_$name$_codec);\n");
118}
119
120void MapFieldGenerator::GenerateSerializedSizeCode(io::Printer* printer) {
121 printer->Print(
122 variables: variables_,
123 text: "size += $name$_.CalculateSize(_map_$name$_codec);\n");
124}
125
126void MapFieldGenerator::WriteHash(io::Printer* printer) {
127 printer->Print(
128 variables: variables_,
129 text: "hash ^= $property_name$.GetHashCode();\n");
130}
131void MapFieldGenerator::WriteEquals(io::Printer* printer) {
132 printer->Print(
133 variables: variables_,
134 text: "if (!$property_name$.Equals(other.$property_name$)) return false;\n");
135}
136
137void MapFieldGenerator::WriteToString(io::Printer* printer) {
138 // TODO: If we ever actually use ToString, we'll need to impleme this...
139}
140
141void MapFieldGenerator::GenerateCloningCode(io::Printer* printer) {
142 printer->Print(variables: variables_,
143 text: "$name$_ = other.$name$_.Clone();\n");
144}
145
146void MapFieldGenerator::GenerateFreezingCode(io::Printer* printer) {
147}
148
149} // namespace csharp
150} // namespace compiler
151} // namespace protobuf
152} // namespace google
153