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_options.h>
42#include <google/protobuf/compiler/csharp/csharp_enum_field.h>
43
44namespace google {
45namespace protobuf {
46namespace compiler {
47namespace csharp {
48
49EnumFieldGenerator::EnumFieldGenerator(const FieldDescriptor* descriptor,
50 int presenceIndex, const Options *options)
51 : PrimitiveFieldGenerator(descriptor, presenceIndex, options) {
52}
53
54EnumFieldGenerator::~EnumFieldGenerator() {
55}
56
57void EnumFieldGenerator::GenerateParsingCode(io::Printer* printer) {
58 printer->Print(variables: variables_,
59 text: "$property_name$ = ($type_name$) input.ReadEnum();\n");
60}
61
62void EnumFieldGenerator::GenerateSerializationCode(io::Printer* printer) {
63 printer->Print(variables: variables_,
64 text: "if ($has_property_check$) {\n"
65 " output.WriteRawTag($tag_bytes$);\n"
66 " output.WriteEnum((int) $property_name$);\n"
67 "}\n");
68}
69
70void EnumFieldGenerator::GenerateSerializedSizeCode(io::Printer* printer) {
71 printer->Print(
72 variables: variables_,
73 text: "if ($has_property_check$) {\n"
74 " size += $tag_size$ + pb::CodedOutputStream.ComputeEnumSize((int) $property_name$);\n"
75 "}\n");
76}
77
78void EnumFieldGenerator::GenerateCodecCode(io::Printer* printer) {
79 printer->Print(
80 variables: variables_,
81 text: "pb::FieldCodec.ForEnum($tag$, x => (int) x, x => ($type_name$) x, $default_value$)");
82}
83
84void EnumFieldGenerator::GenerateExtensionCode(io::Printer* printer) {
85 WritePropertyDocComment(printer, field: descriptor_);
86 AddDeprecatedFlag(printer);
87 printer->Print(
88 variables: variables_,
89 text: "$access_level$ static readonly pb::Extension<$extended_type$, $type_name$> $property_name$ =\n"
90 " new pb::Extension<$extended_type$, $type_name$>($number$, ");
91 GenerateCodecCode(printer);
92 printer->Print(text: ");\n");
93}
94
95EnumOneofFieldGenerator::EnumOneofFieldGenerator(
96 const FieldDescriptor* descriptor, int presenceIndex, const Options *options)
97 : PrimitiveOneofFieldGenerator(descriptor, presenceIndex, options) {
98}
99
100EnumOneofFieldGenerator::~EnumOneofFieldGenerator() {
101}
102
103void EnumOneofFieldGenerator::GenerateMergingCode(io::Printer* printer) {
104 printer->Print(variables: variables_, text: "$property_name$ = other.$property_name$;\n");
105}
106
107void EnumOneofFieldGenerator::GenerateParsingCode(io::Printer* printer) {
108 // TODO(jonskeet): What about if we read the default value?
109 printer->Print(
110 variables: variables_,
111 text: "$oneof_name$_ = input.ReadEnum();\n"
112 "$oneof_name$Case_ = $oneof_property_name$OneofCase.$oneof_case_name$;\n");
113}
114
115void EnumOneofFieldGenerator::GenerateSerializationCode(io::Printer* printer) {
116 printer->Print(
117 variables: variables_,
118 text: "if ($has_property_check$) {\n"
119 " output.WriteRawTag($tag_bytes$);\n"
120 " output.WriteEnum((int) $property_name$);\n"
121 "}\n");
122}
123
124void EnumOneofFieldGenerator::GenerateSerializedSizeCode(io::Printer* printer) {
125 printer->Print(
126 variables: variables_,
127 text: "if ($has_property_check$) {\n"
128 " size += $tag_size$ + pb::CodedOutputStream.ComputeEnumSize((int) $property_name$);\n"
129 "}\n");
130}
131
132} // namespace csharp
133} // namespace compiler
134} // namespace protobuf
135} // namespace google
136