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 // Author: kenton@google.com (Kenton Varda)
32 // Based on original Protocol Buffers design by
33 // Sanjay Ghemawat, Jeff Dean, and others.
34
35 #ifndef GOOGLE_PROTOBUF_COMPILER_JAVA_FIELD_H__
36 #define GOOGLE_PROTOBUF_COMPILER_JAVA_FIELD_H__
37
38 #include <cstdint>
39 #include <map>
40 #include <memory>
41 #include <string>
42
43 #include <google/protobuf/stubs/logging.h>
44 #include <google/protobuf/stubs/common.h>
45 #include <google/protobuf/descriptor.h>
46
47 namespace google {
48 namespace protobuf {
49 namespace compiler {
50 namespace java {
51 class Context; // context.h
52 class ClassNameResolver; // name_resolver.h
53 } // namespace java
54 } // namespace compiler
55 namespace io {
56 class Printer; // printer.h
57 }
58 } // namespace protobuf
59 } // namespace google
60
61 namespace google {
62 namespace protobuf {
63 namespace compiler {
64 namespace java {
65
66 class ImmutableFieldGenerator {
67 public:
ImmutableFieldGenerator()68 ImmutableFieldGenerator() {}
69 virtual ~ImmutableFieldGenerator();
70
71 virtual int GetNumBitsForMessage() const = 0;
72 virtual int GetNumBitsForBuilder() const = 0;
73 virtual void GenerateInterfaceMembers(io::Printer* printer) const = 0;
74 virtual void GenerateMembers(io::Printer* printer) const = 0;
75 virtual void GenerateBuilderMembers(io::Printer* printer) const = 0;
76 virtual void GenerateInitializationCode(io::Printer* printer) const = 0;
77 virtual void GenerateBuilderClearCode(io::Printer* printer) const = 0;
78 virtual void GenerateMergingCode(io::Printer* printer) const = 0;
79 virtual void GenerateBuildingCode(io::Printer* printer) const = 0;
80 virtual void GenerateBuilderParsingCode(io::Printer* printer) const = 0;
81 virtual void GenerateBuilderParsingCodeFromPacked(io::Printer* printer) const;
82 virtual void GenerateSerializationCode(io::Printer* printer) const = 0;
83 virtual void GenerateSerializedSizeCode(io::Printer* printer) const = 0;
84 virtual void GenerateFieldBuilderInitializationCode(
85 io::Printer* printer) const = 0;
86 virtual void GenerateKotlinDslMembers(io::Printer* printer) const = 0;
87
88 virtual void GenerateEqualsCode(io::Printer* printer) const = 0;
89 virtual void GenerateHashCode(io::Printer* printer) const = 0;
90
91 virtual std::string GetBoxedType() const = 0;
92
93 private:
94 GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ImmutableFieldGenerator);
95 };
96
97 class ImmutableFieldLiteGenerator {
98 public:
ImmutableFieldLiteGenerator()99 ImmutableFieldLiteGenerator() {}
100 virtual ~ImmutableFieldLiteGenerator();
101
102 virtual int GetNumBitsForMessage() const = 0;
103 virtual void GenerateInterfaceMembers(io::Printer* printer) const = 0;
104 virtual void GenerateMembers(io::Printer* printer) const = 0;
105 virtual void GenerateBuilderMembers(io::Printer* printer) const = 0;
106 virtual void GenerateInitializationCode(io::Printer* printer) const = 0;
107 virtual void GenerateFieldInfo(io::Printer* printer,
108 std::vector<uint16_t>* output) const = 0;
109 virtual void GenerateKotlinDslMembers(io::Printer* printer) const = 0;
110
111 virtual std::string GetBoxedType() const = 0;
112
113 private:
114 GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(ImmutableFieldLiteGenerator);
115 };
116
117
118 // Convenience class which constructs FieldGenerators for a Descriptor.
119 template <typename FieldGeneratorType>
120 class FieldGeneratorMap {
121 public:
122 explicit FieldGeneratorMap(const Descriptor* descriptor, Context* context);
123 ~FieldGeneratorMap();
124
125 const FieldGeneratorType& get(const FieldDescriptor* field) const;
126
127 private:
128 const Descriptor* descriptor_;
129 std::vector<std::unique_ptr<FieldGeneratorType>> field_generators_;
130
131 GOOGLE_DISALLOW_EVIL_CONSTRUCTORS(FieldGeneratorMap);
132 };
133
134 template <typename FieldGeneratorType>
get(const FieldDescriptor * field)135 inline const FieldGeneratorType& FieldGeneratorMap<FieldGeneratorType>::get(
136 const FieldDescriptor* field) const {
137 GOOGLE_CHECK_EQ(field->containing_type(), descriptor_);
138 return *field_generators_[field->index()];
139 }
140
141 // Instantiate template for mutable and immutable maps.
142 template <>
143 FieldGeneratorMap<ImmutableFieldGenerator>::FieldGeneratorMap(
144 const Descriptor* descriptor, Context* context);
145
146 template <>
147 FieldGeneratorMap<ImmutableFieldGenerator>::~FieldGeneratorMap();
148
149
150 template <>
151 FieldGeneratorMap<ImmutableFieldLiteGenerator>::FieldGeneratorMap(
152 const Descriptor* descriptor, Context* context);
153
154 template <>
155 FieldGeneratorMap<ImmutableFieldLiteGenerator>::~FieldGeneratorMap();
156
157
158 // Field information used in FieldGenerators.
159 struct FieldGeneratorInfo {
160 std::string name;
161 std::string capitalized_name;
162 std::string disambiguated_reason;
163 };
164
165 // Oneof information used in OneofFieldGenerators.
166 struct OneofGeneratorInfo {
167 std::string name;
168 std::string capitalized_name;
169 };
170
171 // Set some common variables used in variable FieldGenerators.
172 void SetCommonFieldVariables(const FieldDescriptor* descriptor,
173 const FieldGeneratorInfo* info,
174 std::map<std::string, std::string>* variables);
175
176 // Set some common oneof variables used in OneofFieldGenerators.
177 void SetCommonOneofVariables(const FieldDescriptor* descriptor,
178 const OneofGeneratorInfo* info,
179 std::map<std::string, std::string>* variables);
180
181 // Print useful comments before a field's accessors.
182 void PrintExtraFieldInfo(const std::map<std::string, std::string>& variables,
183 io::Printer* printer);
184
185 } // namespace java
186 } // namespace compiler
187 } // namespace protobuf
188 } // namespace google
189
190 #endif // GOOGLE_PROTOBUF_COMPILER_JAVA_FIELD_H__
191