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/compiler/plugin.h>
35 #include <google/protobuf/descriptor.h>
36 #include <google/protobuf/descriptor.pb.h>
37 #include <google/protobuf/io/printer.h>
38 #include <google/protobuf/io/zero_copy_stream.h>
39
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 fieldOrdinal,const Options * options)49 EnumFieldGenerator::EnumFieldGenerator(const FieldDescriptor* descriptor,
50 int fieldOrdinal, const Options *options)
51 : PrimitiveFieldGenerator(descriptor, fieldOrdinal, 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 "$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)");
82 }
83
EnumOneofFieldGenerator(const FieldDescriptor * descriptor,int fieldOrdinal,const Options * options)84 EnumOneofFieldGenerator::EnumOneofFieldGenerator(
85 const FieldDescriptor* descriptor, int fieldOrdinal, const Options *options)
86 : PrimitiveOneofFieldGenerator(descriptor, fieldOrdinal, options) {
87 }
88
~EnumOneofFieldGenerator()89 EnumOneofFieldGenerator::~EnumOneofFieldGenerator() {
90 }
91
GenerateParsingCode(io::Printer * printer)92 void EnumOneofFieldGenerator::GenerateParsingCode(io::Printer* printer) {
93 // TODO(jonskeet): What about if we read the default value?
94 printer->Print(
95 variables_,
96 "$oneof_name$_ = input.ReadEnum();\n"
97 "$oneof_name$Case_ = $oneof_property_name$OneofCase.$property_name$;\n");
98 }
99
GenerateSerializationCode(io::Printer * printer)100 void EnumOneofFieldGenerator::GenerateSerializationCode(io::Printer* printer) {
101 printer->Print(
102 variables_,
103 "if ($has_property_check$) {\n"
104 " output.WriteRawTag($tag_bytes$);\n"
105 " output.WriteEnum((int) $property_name$);\n"
106 "}\n");
107 }
108
GenerateSerializedSizeCode(io::Printer * printer)109 void EnumOneofFieldGenerator::GenerateSerializedSizeCode(io::Printer* printer) {
110 printer->Print(
111 variables_,
112 "if ($has_property_check$) {\n"
113 " size += $tag_size$ + pb::CodedOutputStream.ComputeEnumSize((int) $property_name$);\n"
114 "}\n");
115 }
116
117 } // namespace csharp
118 } // namespace compiler
119 } // namespace protobuf
120 } // namespace google
121