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