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