blob: 6ee97aa47f90434771ba246bcea1ac5f2c121020 [file] [log] [blame]
Austin Schuh272c6132020-11-14 16:37:52 -08001/*
2 * Copyright 2014 Google Inc. All rights reserved.
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17// independent from idl_parser, since this code is not needed for most clients
18
19#include "flatbuffers/code_generators.h"
20#include "flatbuffers/flatbuffers.h"
21#include "flatbuffers/idl.h"
22#include "flatbuffers/util.h"
Austin Schuh2dd86a92022-09-14 21:19:23 -070023#include "idl_namer.h"
Austin Schuh272c6132020-11-14 16:37:52 -080024
Austin Schuh272c6132020-11-14 16:37:52 -080025namespace flatbuffers {
26namespace java {
27
Austin Schuh2dd86a92022-09-14 21:19:23 -070028namespace {
29
30static Namer::Config JavaDefaultConfig() {
31 return {
32 /*types=*/Case::kKeep,
33 /*constants=*/Case::kScreamingSnake,
34 /*methods=*/Case::kLowerCamel,
35 /*functions=*/Case::kLowerCamel,
36 /*fields=*/Case::kLowerCamel,
37 /*variables=*/Case::kLowerCamel,
38 /*variants=*/Case::kKeep,
39 /*enum_variant_seperator=*/".",
40 /*escape_keywords=*/Namer::Config::Escape::AfterConvertingCase,
41 /*namespaces=*/Case::kKeep,
42 /*namespace_seperator=*/".",
43 /*object_prefix=*/"",
44 /*object_suffix=*/"T",
45 /*keyword_prefix=*/"",
46 /*keyword_suffix=*/"_",
47 /*filenames=*/Case::kKeep,
48 /*directories=*/Case::kKeep,
49 /*output_path=*/"",
50 /*filename_suffix=*/"_generated",
51 /*filename_extension=*/".java",
52 };
53}
54
55static std::set<std::string> JavaKeywords() {
56 return {
57 "abstract", "continue", "for", "new", "switch",
58 "assert", "default", "goto", "package", "synchronized",
59 "boolean", "do", "if", "private", "this",
60 "break", "double", "implements", "protected", "throw",
61 "byte", "else", "import", "public", "throws",
62 "case", "enum", "instanceof", "return", "transient",
63 "catch", "extends", "int", "short", "try",
64 "char", "final", "interface", "static", "void",
65 "class", "finally", "long", "strictfp", "volatile",
66 "const", "float", "native", "super", "while",
67 };
68}
69
70static const TypedFloatConstantGenerator JavaFloatGen("Double.", "Float.", "NaN",
Austin Schuh272c6132020-11-14 16:37:52 -080071 "POSITIVE_INFINITY",
72 "NEGATIVE_INFINITY");
73
Austin Schuh2dd86a92022-09-14 21:19:23 -070074static const CommentConfig comment_config = {
Austin Schuh272c6132020-11-14 16:37:52 -080075 "/**",
76 " *",
77 " */",
78};
79
Austin Schuh2dd86a92022-09-14 21:19:23 -070080} // namespace
81
Austin Schuh272c6132020-11-14 16:37:52 -080082class JavaGenerator : public BaseGenerator {
James Kuszmaul8e62b022022-03-22 09:33:25 -070083 struct FieldArrayLength {
84 std::string name;
85 int length;
86 };
87
Austin Schuh272c6132020-11-14 16:37:52 -080088 public:
89 JavaGenerator(const Parser &parser, const std::string &path,
90 const std::string &file_name)
91 : BaseGenerator(parser, path, file_name, "", ".", "java"),
Austin Schuh2dd86a92022-09-14 21:19:23 -070092 cur_name_space_(nullptr),
93 namer_(WithFlagOptions(JavaDefaultConfig(), parser.opts, path),
94 JavaKeywords()) {}
Austin Schuh272c6132020-11-14 16:37:52 -080095
96 JavaGenerator &operator=(const JavaGenerator &);
97 bool generate() {
98 std::string one_file_code;
99 cur_name_space_ = parser_.current_namespace_;
100
101 for (auto it = parser_.enums_.vec.begin(); it != parser_.enums_.vec.end();
102 ++it) {
103 std::string enumcode;
104 auto &enum_def = **it;
105 if (!parser_.opts.one_file) cur_name_space_ = enum_def.defined_namespace;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700106 GenEnum(enum_def, enumcode);
Austin Schuh272c6132020-11-14 16:37:52 -0800107 if (parser_.opts.one_file) {
108 one_file_code += enumcode;
109 } else {
110 if (!SaveType(enum_def.name, *enum_def.defined_namespace, enumcode,
111 /* needs_includes= */ false))
112 return false;
113 }
James Kuszmaul8e62b022022-03-22 09:33:25 -0700114
115 if (parser_.opts.generate_object_based_api && enum_def.is_union) {
116 enumcode = "";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700117 GenEnum_ObjectAPI(enum_def, enumcode);
118 auto class_name = namer_.Type(enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -0700119 if (parser_.opts.one_file) {
120 one_file_code += enumcode;
121 } else {
122 if (!SaveType(class_name, *enum_def.defined_namespace, enumcode,
123 /* needs_includes= */ false))
124 return false;
125 }
126 }
Austin Schuh272c6132020-11-14 16:37:52 -0800127 }
128
129 for (auto it = parser_.structs_.vec.begin();
130 it != parser_.structs_.vec.end(); ++it) {
131 std::string declcode;
132 auto &struct_def = **it;
133 if (!parser_.opts.one_file)
134 cur_name_space_ = struct_def.defined_namespace;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700135 GenStruct(struct_def, declcode, parser_.opts);
Austin Schuh272c6132020-11-14 16:37:52 -0800136 if (parser_.opts.one_file) {
137 one_file_code += declcode;
138 } else {
139 if (!SaveType(struct_def.name, *struct_def.defined_namespace, declcode,
140 /* needs_includes= */ true))
141 return false;
142 }
James Kuszmaul8e62b022022-03-22 09:33:25 -0700143
144 if (parser_.opts.generate_object_based_api) {
145 declcode = "";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700146 GenStruct_ObjectAPI(struct_def, declcode);
147 auto class_name = namer_.ObjectType(struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -0700148 if (parser_.opts.one_file) {
149 one_file_code += declcode;
150 } else {
151 if (!SaveType(class_name, *struct_def.defined_namespace, declcode,
152 /* needs_includes= */ true))
153 return false;
154 }
155 }
Austin Schuh272c6132020-11-14 16:37:52 -0800156 }
157
158 if (parser_.opts.one_file) {
159 return SaveType(file_name_, *parser_.current_namespace_, one_file_code,
160 /* needs_includes= */ true);
161 }
162 return true;
163 }
164
165 // Save out the generated code for a single class while adding
166 // declaration boilerplate.
167 bool SaveType(const std::string &defname, const Namespace &ns,
168 const std::string &classcode, bool needs_includes) const {
169 if (!classcode.length()) return true;
170
171 std::string code;
172 code = "// " + std::string(FlatBuffersGeneratedWarning()) + "\n\n";
173
Austin Schuh2dd86a92022-09-14 21:19:23 -0700174 const std::string namespace_name = FullNamespace(".", ns);
Austin Schuh272c6132020-11-14 16:37:52 -0800175 if (!namespace_name.empty()) {
176 code += "package " + namespace_name + ";";
177 code += "\n\n";
178 }
179 if (needs_includes) {
180 code +=
181 "import java.nio.*;\nimport java.lang.*;\nimport "
182 "java.util.*;\nimport com.google.flatbuffers.*;\n";
183 if (parser_.opts.gen_nullable) {
184 code += "\nimport javax.annotation.Nullable;\n";
185 }
186 if (parser_.opts.java_checkerframework) {
187 code += "\nimport org.checkerframework.dataflow.qual.Pure;\n";
188 }
189 code += "\n";
190 }
191
192 code += classcode;
193 if (!namespace_name.empty()) code += "";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700194 const std::string dirs = namer_.Directories(ns);
195 EnsureDirExists(dirs);
196 const std::string filename =
197 dirs + namer_.File(defname, /*skips=*/SkipFile::Suffix);
Austin Schuh272c6132020-11-14 16:37:52 -0800198 return SaveFile(filename.c_str(), code, false);
199 }
200
201 const Namespace *CurrentNameSpace() const { return cur_name_space_; }
202
203 std::string GenNullableAnnotation(const Type &t) const {
204 return parser_.opts.gen_nullable &&
205 !IsScalar(DestinationType(t, true).base_type) &&
206 t.base_type != BASE_TYPE_VECTOR
207 ? " @Nullable "
208 : "";
209 }
210
211 std::string GenPureAnnotation(const Type &t) const {
212 return parser_.opts.java_checkerframework &&
213 !IsScalar(DestinationType(t, true).base_type)
214 ? " @Pure "
215 : "";
216 }
217
218 std::string GenTypeBasic(const Type &type) const {
219 // clang-format off
220 static const char * const java_typename[] = {
221 #define FLATBUFFERS_TD(ENUM, IDLTYPE, CTYPE, JTYPE, ...) \
222 #JTYPE,
223 FLATBUFFERS_GEN_TYPES(FLATBUFFERS_TD)
224 #undef FLATBUFFERS_TD
225 };
226 // clang-format on
227 return java_typename[type.base_type];
228 }
229
230 std::string GenTypePointer(const Type &type) const {
231 switch (type.base_type) {
232 case BASE_TYPE_STRING: return "String";
233 case BASE_TYPE_VECTOR: return GenTypeGet(type.VectorType());
Austin Schuh2dd86a92022-09-14 21:19:23 -0700234 case BASE_TYPE_STRUCT: return namer_.NamespacedType(*type.struct_def);
Austin Schuh272c6132020-11-14 16:37:52 -0800235 case BASE_TYPE_UNION: FLATBUFFERS_FALLTHROUGH(); // else fall thru
236 default: return "Table";
237 }
238 }
239
240 std::string GenTypeGet(const Type &type) const {
241 return IsScalar(type.base_type)
242 ? GenTypeBasic(type)
243 : (IsArray(type) ? GenTypeGet(type.VectorType())
244 : GenTypePointer(type));
245 }
246
247 // Find the destination type the user wants to receive the value in (e.g.
248 // one size higher signed types for unsigned serialized values in Java).
249 Type DestinationType(const Type &type, bool vectorelem) const {
250 switch (type.base_type) {
251 // We use int for both uchar/ushort, since that generally means less
252 // casting than using short for uchar.
253 case BASE_TYPE_UCHAR: return Type(BASE_TYPE_INT);
254 case BASE_TYPE_USHORT: return Type(BASE_TYPE_INT);
255 case BASE_TYPE_UINT: return Type(BASE_TYPE_LONG);
256 case BASE_TYPE_ARRAY:
257 case BASE_TYPE_VECTOR:
258 if (vectorelem) return DestinationType(type.VectorType(), vectorelem);
259 FLATBUFFERS_FALLTHROUGH(); // else fall thru
260 default: return type;
261 }
262 }
263
264 std::string GenOffsetType() const { return "int"; }
265
266 std::string GenOffsetConstruct(const std::string &variable_name) const {
267 return variable_name;
268 }
269
270 std::string GenVectorOffsetType() const { return "int"; }
271
272 // Generate destination type name
273 std::string GenTypeNameDest(const Type &type) const {
274 return GenTypeGet(DestinationType(type, true));
275 }
276
277 // Mask to turn serialized value into destination type value.
278 std::string DestinationMask(const Type &type, bool vectorelem) const {
279 switch (type.base_type) {
280 case BASE_TYPE_UCHAR: return " & 0xFF";
281 case BASE_TYPE_USHORT: return " & 0xFFFF";
282 case BASE_TYPE_UINT: return " & 0xFFFFFFFFL";
283 case BASE_TYPE_VECTOR:
284 if (vectorelem) return DestinationMask(type.VectorType(), vectorelem);
285 FLATBUFFERS_FALLTHROUGH(); // else fall thru
286 default: return "";
287 }
288 }
289
290 // Casts necessary to correctly read serialized data
291 std::string DestinationCast(const Type &type) const {
292 if (IsSeries(type)) {
293 return DestinationCast(type.VectorType());
294 } else {
295 // Cast necessary to correctly read serialized unsigned values.
296 if (type.base_type == BASE_TYPE_UINT) return "(long)";
297 }
298 return "";
299 }
300
301 // Cast statements for mutator method parameters.
302 // In Java, parameters representing unsigned numbers need to be cast down to
303 // their respective type. For example, a long holding an unsigned int value
304 // would be cast down to int before being put onto the buffer.
305 std::string SourceCast(const Type &type, bool castFromDest) const {
306 if (IsSeries(type)) {
307 return SourceCast(type.VectorType(), castFromDest);
308 } else {
309 if (castFromDest) {
310 if (type.base_type == BASE_TYPE_UINT)
James Kuszmaul8e62b022022-03-22 09:33:25 -0700311 return "(int) ";
Austin Schuh272c6132020-11-14 16:37:52 -0800312 else if (type.base_type == BASE_TYPE_USHORT)
James Kuszmaul8e62b022022-03-22 09:33:25 -0700313 return "(short) ";
Austin Schuh272c6132020-11-14 16:37:52 -0800314 else if (type.base_type == BASE_TYPE_UCHAR)
James Kuszmaul8e62b022022-03-22 09:33:25 -0700315 return "(byte) ";
Austin Schuh272c6132020-11-14 16:37:52 -0800316 }
317 }
318 return "";
319 }
320
321 std::string SourceCast(const Type &type) const {
322 return SourceCast(type, true);
323 }
324
325 std::string SourceCastBasic(const Type &type, bool castFromDest) const {
326 return IsScalar(type.base_type) ? SourceCast(type, castFromDest) : "";
327 }
328
329 std::string SourceCastBasic(const Type &type) const {
330 return SourceCastBasic(type, true);
331 }
332
333 std::string GenEnumDefaultValue(const FieldDef &field) const {
334 auto &value = field.value;
335 FLATBUFFERS_ASSERT(value.type.enum_def);
336 auto &enum_def = *value.type.enum_def;
337 auto enum_val = enum_def.FindByValue(value.constant);
Austin Schuh2dd86a92022-09-14 21:19:23 -0700338 return enum_val ? namer_.NamespacedEnumVariant(enum_def, *enum_val)
Austin Schuh272c6132020-11-14 16:37:52 -0800339 : value.constant;
340 }
341
342 std::string GenDefaultValue(const FieldDef &field) const {
343 auto &value = field.value;
344 auto constant = field.IsScalarOptional() ? "0" : value.constant;
345 auto longSuffix = "L";
346 switch (value.type.base_type) {
347 case BASE_TYPE_BOOL: return constant == "0" ? "false" : "true";
348 case BASE_TYPE_ULONG: {
349 // Converts the ulong into its bits signed equivalent
350 uint64_t defaultValue = StringToUInt(constant.c_str());
351 return NumToString(static_cast<int64_t>(defaultValue)) + longSuffix;
352 }
353 case BASE_TYPE_UINT:
354 case BASE_TYPE_LONG: return constant + longSuffix;
355 default:
356 if (IsFloat(value.type.base_type)) {
357 if (field.IsScalarOptional()) {
358 return value.type.base_type == BASE_TYPE_DOUBLE ? "0.0" : "0f";
359 }
360 return JavaFloatGen.GenFloatConstant(field);
361 } else {
362 return constant;
363 }
364 }
365 }
366
367 std::string GenDefaultValueBasic(const FieldDef &field) const {
368 auto &value = field.value;
369 if (!IsScalar(value.type.base_type)) { return "0"; }
370 return GenDefaultValue(field);
371 }
372
Austin Schuh2dd86a92022-09-14 21:19:23 -0700373 void GenEnum(EnumDef &enum_def, std::string &code) const {
Austin Schuh272c6132020-11-14 16:37:52 -0800374 if (enum_def.generated) return;
375
376 // Generate enum definitions of the form:
377 // public static (final) int name = value;
378 // In Java, we use ints rather than the Enum feature, because we want them
379 // to map directly to how they're used in C/C++ and file formats.
380 // That, and Java Enums are expensive, and not universally liked.
Austin Schuh2dd86a92022-09-14 21:19:23 -0700381 GenComment(enum_def.doc_comment, &code, &comment_config);
Austin Schuh272c6132020-11-14 16:37:52 -0800382
James Kuszmaul8e62b022022-03-22 09:33:25 -0700383 code += "@SuppressWarnings(\"unused\")\n";
Austin Schuh272c6132020-11-14 16:37:52 -0800384 if (enum_def.attributes.Lookup("private")) {
385 // For Java, we leave the enum unmarked to indicate package-private
386 } else {
387 code += "public ";
388 }
Austin Schuh2dd86a92022-09-14 21:19:23 -0700389 code += "final class " + namer_.Type(enum_def);
Austin Schuh272c6132020-11-14 16:37:52 -0800390 code += " {\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700391 code += " private " + namer_.Type(enum_def) + "() { }\n";
Austin Schuh272c6132020-11-14 16:37:52 -0800392 for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
393 auto &ev = **it;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700394 GenComment(ev.doc_comment, &code, &comment_config, " ");
Austin Schuh272c6132020-11-14 16:37:52 -0800395 code += " public static final ";
James Kuszmaul8e62b022022-03-22 09:33:25 -0700396 code += GenTypeBasic(DestinationType(enum_def.underlying_type, false));
Austin Schuh272c6132020-11-14 16:37:52 -0800397 code += " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700398 code += namer_.Variant(ev) + " = ";
Austin Schuh272c6132020-11-14 16:37:52 -0800399 code += enum_def.ToString(ev);
400 code += ";\n";
401 }
402
James Kuszmaul8e62b022022-03-22 09:33:25 -0700403 // Generate a string table for enum values.
Austin Schuh272c6132020-11-14 16:37:52 -0800404 // Problem is, if values are very sparse that could generate really big
405 // tables. Ideally in that case we generate a map lookup instead, but for
406 // the moment we simply don't output a table at all.
407 auto range = enum_def.Distance();
408 // Average distance between values above which we consider a table
409 // "too sparse". Change at will.
410 static const uint64_t kMaxSparseness = 5;
James Kuszmaul8e62b022022-03-22 09:33:25 -0700411 if (range / static_cast<uint64_t>(enum_def.size()) < kMaxSparseness &&
412 GenTypeBasic(DestinationType(enum_def.underlying_type, false)) !=
413 "long") {
Austin Schuh272c6132020-11-14 16:37:52 -0800414 code += "\n public static final String";
415 code += "[] names = { ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700416 const EnumVal *prev = enum_def.Vals().front();
Austin Schuh272c6132020-11-14 16:37:52 -0800417 for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end();
418 ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700419 const EnumVal &ev = **it;
420 for (auto k = enum_def.Distance(prev, &ev); k > 1; --k)
421 code += "\"\", ";
422 prev = &ev;
423 code += "\"" + namer_.Variant(ev) + "\", ";
Austin Schuh272c6132020-11-14 16:37:52 -0800424 }
425 code += "};\n\n";
426 code += " public static ";
427 code += "String";
James Kuszmaul8e62b022022-03-22 09:33:25 -0700428 code += " name";
Austin Schuh272c6132020-11-14 16:37:52 -0800429 code += "(int e) { return names[e";
430 if (enum_def.MinValue()->IsNonZero())
Austin Schuh2dd86a92022-09-14 21:19:23 -0700431 code += " - " + namer_.Variant(enum_def.MinValue()->name);
Austin Schuh272c6132020-11-14 16:37:52 -0800432 code += "]; }\n";
433 }
434
435 // Close the class
436 code += "}\n\n";
437 }
438
439 // Returns the function name that is able to read a value of the given type.
440 std::string GenGetter(const Type &type) const {
441 switch (type.base_type) {
442 case BASE_TYPE_STRING: return "__string";
443 case BASE_TYPE_STRUCT: return "__struct";
444 case BASE_TYPE_UNION: return "__union";
445 case BASE_TYPE_VECTOR: return GenGetter(type.VectorType());
446 case BASE_TYPE_ARRAY: return GenGetter(type.VectorType());
447 default: {
448 std::string getter = "bb.get";
449 if (type.base_type == BASE_TYPE_BOOL) {
450 getter = "0!=" + getter;
451 } else if (GenTypeBasic(type) != "byte") {
James Kuszmaul8e62b022022-03-22 09:33:25 -0700452 getter += ConvertCase(GenTypeBasic(type), Case::kUpperCamel);
Austin Schuh272c6132020-11-14 16:37:52 -0800453 }
454 return getter;
455 }
456 }
457 }
458
459 // Returns the function name that is able to read a value of the given type.
460 std::string GenGetterForLookupByKey(flatbuffers::FieldDef *key_field,
461 const std::string &data_buffer,
462 const char *num = nullptr) const {
463 auto type = key_field->value.type;
464 auto dest_mask = DestinationMask(type, true);
465 auto dest_cast = DestinationCast(type);
466 auto getter = data_buffer + ".get";
467 if (GenTypeBasic(type) != "byte") {
James Kuszmaul8e62b022022-03-22 09:33:25 -0700468 getter += ConvertCase(GenTypeBasic(type), Case::kUpperCamel);
Austin Schuh272c6132020-11-14 16:37:52 -0800469 }
470 getter = dest_cast + getter + "(" + GenOffsetGetter(key_field, num) + ")" +
471 dest_mask;
472 return getter;
473 }
474
475 // Direct mutation is only allowed for scalar fields.
476 // Hence a setter method will only be generated for such fields.
477 std::string GenSetter(const Type &type) const {
478 if (IsScalar(type.base_type)) {
479 std::string setter = "bb.put";
480 if (GenTypeBasic(type) != "byte" && type.base_type != BASE_TYPE_BOOL) {
James Kuszmaul8e62b022022-03-22 09:33:25 -0700481 setter += ConvertCase(GenTypeBasic(type), Case::kUpperCamel);
Austin Schuh272c6132020-11-14 16:37:52 -0800482 }
483 return setter;
484 } else {
485 return "";
486 }
487 }
488
489 // Returns the method name for use with add/put calls.
490 std::string GenMethod(const Type &type) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -0700491 return IsScalar(type.base_type)
492 ? ConvertCase(GenTypeBasic(type), Case::kUpperCamel)
493 : (IsStruct(type) ? "Struct" : "Offset");
Austin Schuh272c6132020-11-14 16:37:52 -0800494 }
495
496 // Recursively generate arguments for a constructor, to deal with nested
497 // structs.
Austin Schuh2dd86a92022-09-14 21:19:23 -0700498 void GenStructArgs(const StructDef &struct_def, std::string &code,
Austin Schuh272c6132020-11-14 16:37:52 -0800499 const char *nameprefix, size_t array_count = 0) const {
Austin Schuh272c6132020-11-14 16:37:52 -0800500 for (auto it = struct_def.fields.vec.begin();
501 it != struct_def.fields.vec.end(); ++it) {
502 auto &field = **it;
503 const auto &field_type = field.value.type;
504 const auto array_field = IsArray(field_type);
505 const auto &type = array_field ? field_type.VectorType()
506 : DestinationType(field_type, false);
507 const auto array_cnt = array_field ? (array_count + 1) : array_count;
508 if (IsStruct(type)) {
509 // Generate arguments for a struct inside a struct. To ensure names
510 // don't clash, and to make it obvious these arguments are constructing
511 // a nested struct, prefix the name with the field name.
Austin Schuh2dd86a92022-09-14 21:19:23 -0700512 GenStructArgs(*field_type.struct_def, code,
Austin Schuh272c6132020-11-14 16:37:52 -0800513 (nameprefix + (field.name + "_")).c_str(), array_cnt);
514 } else {
515 code += ", ";
James Kuszmaul8e62b022022-03-22 09:33:25 -0700516 code += GenTypeNameDest(field.value.type);
Austin Schuh272c6132020-11-14 16:37:52 -0800517 for (size_t i = 0; i < array_cnt; i++) code += "[]";
518 code += " ";
519 code += nameprefix;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700520 code += namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800521 }
522 }
523 }
524
525 // Recusively generate struct construction statements of the form:
526 // builder.putType(name);
527 // and insert manual padding.
Austin Schuh2dd86a92022-09-14 21:19:23 -0700528 void GenStructBody(const StructDef &struct_def, std::string &code,
Austin Schuh272c6132020-11-14 16:37:52 -0800529 const char *nameprefix, size_t index = 0,
530 bool in_array = false) const {
Austin Schuh272c6132020-11-14 16:37:52 -0800531 std::string indent((index + 1) * 2, ' ');
532 code += indent + " builder.prep(";
533 code += NumToString(struct_def.minalign) + ", ";
534 code += NumToString(struct_def.bytesize) + ");\n";
535 for (auto it = struct_def.fields.vec.rbegin();
536 it != struct_def.fields.vec.rend(); ++it) {
537 auto &field = **it;
538 const auto &field_type = field.value.type;
539 if (field.padding) {
540 code += indent + " builder.pad(";
541 code += NumToString(field.padding) + ");\n";
542 }
543 if (IsStruct(field_type)) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700544 GenStructBody(*field_type.struct_def, code,
Austin Schuh272c6132020-11-14 16:37:52 -0800545 (nameprefix + (field.name + "_")).c_str(), index,
546 in_array);
547 } else {
548 const auto &type =
549 IsArray(field_type) ? field_type.VectorType() : field_type;
550 const auto index_var = "_idx" + NumToString(index);
551 if (IsArray(field_type)) {
552 code += indent + " for (int " + index_var + " = ";
553 code += NumToString(field_type.fixed_length);
554 code += "; " + index_var + " > 0; " + index_var + "--) {\n";
555 in_array = true;
556 }
557 if (IsStruct(type)) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700558 GenStructBody(*field_type.struct_def, code,
Austin Schuh272c6132020-11-14 16:37:52 -0800559 (nameprefix + (field.name + "_")).c_str(), index + 1,
560 in_array);
561 } else {
562 code += IsArray(field_type) ? " " : "";
563 code += indent + " builder.put";
564 code += GenMethod(type) + "(";
565 code += SourceCast(type);
Austin Schuh2dd86a92022-09-14 21:19:23 -0700566 auto argname = nameprefix + namer_.Variable(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800567 code += argname;
568 size_t array_cnt = index + (IsArray(field_type) ? 1 : 0);
569 for (size_t i = 0; in_array && i < array_cnt; i++) {
570 code += "[_idx" + NumToString(i) + "-1]";
571 }
572 code += ");\n";
573 }
574 if (IsArray(field_type)) { code += indent + " }\n"; }
575 }
576 }
577 }
578
Austin Schuh272c6132020-11-14 16:37:52 -0800579 std::string GenOffsetGetter(flatbuffers::FieldDef *key_field,
580 const char *num = nullptr) const {
581 std::string key_offset = "";
582 key_offset += "__offset(" + NumToString(key_field->value.offset) + ", ";
583 if (num) {
584 key_offset += num;
585 key_offset += ", _bb)";
586 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700587 key_offset += "bb.capacity()";
Austin Schuh272c6132020-11-14 16:37:52 -0800588 key_offset += " - tableOffset, bb)";
589 }
590 return key_offset;
591 }
592
593 std::string GenLookupKeyGetter(flatbuffers::FieldDef *key_field) const {
594 std::string key_getter = " ";
595 key_getter += "int tableOffset = ";
596 key_getter += "__indirect(vectorLocation + 4 * (start + middle)";
597 key_getter += ", bb);\n ";
598 if (IsString(key_field->value.type)) {
599 key_getter += "int comp = ";
600 key_getter += "compareStrings(";
601 key_getter += GenOffsetGetter(key_field);
602 key_getter += ", byteKey, bb);\n";
603 } else {
604 auto get_val = GenGetterForLookupByKey(key_field, "bb");
605 key_getter += GenTypeNameDest(key_field->value.type) + " val = ";
606 key_getter += get_val + ";\n";
607 key_getter += " int comp = val > key ? 1 : val < key ? -1 : 0;\n";
608 }
609 return key_getter;
610 }
611
612 std::string GenKeyGetter(flatbuffers::FieldDef *key_field) const {
613 std::string key_getter = "";
614 auto data_buffer = "_bb";
615 if (IsString(key_field->value.type)) {
616 key_getter += " return ";
617 key_getter += "";
618 key_getter += "compareStrings(";
619 key_getter += GenOffsetGetter(key_field, "o1") + ", ";
620 key_getter += GenOffsetGetter(key_field, "o2") + ", " + data_buffer + ")";
621 key_getter += ";";
622 } else {
623 auto field_getter = GenGetterForLookupByKey(key_field, data_buffer, "o1");
624 key_getter +=
625 "\n " + GenTypeNameDest(key_field->value.type) + " val_1 = ";
626 key_getter +=
627 field_getter + ";\n " + GenTypeNameDest(key_field->value.type);
628 key_getter += " val_2 = ";
629 field_getter = GenGetterForLookupByKey(key_field, data_buffer, "o2");
630 key_getter += field_getter + ";\n";
631 key_getter += " return val_1 > val_2 ? 1 : val_1 < val_2 ? -1 : 0;\n ";
632 }
633 return key_getter;
634 }
635
Austin Schuh2dd86a92022-09-14 21:19:23 -0700636 void GenStruct(StructDef &struct_def, std::string &code,
James Kuszmaul8e62b022022-03-22 09:33:25 -0700637 const IDLOptions &opts) const {
Austin Schuh272c6132020-11-14 16:37:52 -0800638 if (struct_def.generated) return;
Austin Schuh272c6132020-11-14 16:37:52 -0800639
640 // Generate a struct accessor class, with methods of the form:
641 // public type name() { return bb.getType(i + offset); }
642 // or for tables of the form:
643 // public type name() {
644 // int o = __offset(offset); return o != 0 ? bb.getType(o + i) : default;
645 // }
Austin Schuh2dd86a92022-09-14 21:19:23 -0700646 GenComment(struct_def.doc_comment, &code, &comment_config);
Austin Schuh272c6132020-11-14 16:37:52 -0800647
648 if (parser_.opts.gen_generated) {
649 code += "@javax.annotation.Generated(value=\"flatc\")\n";
650 }
651 code += "@SuppressWarnings(\"unused\")\n";
652 if (struct_def.attributes.Lookup("private")) {
653 // For Java, we leave the struct unmarked to indicate package-private
654 } else {
655 code += "public ";
656 }
Austin Schuh2dd86a92022-09-14 21:19:23 -0700657 const auto struct_class = namer_.Type(struct_def);
658 code += "final class " + struct_class;
Austin Schuh272c6132020-11-14 16:37:52 -0800659 code += " extends ";
660 code += struct_def.fixed ? "Struct" : "Table";
661 code += " {\n";
662
663 if (!struct_def.fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700664 // Generate version check method.
Austin Schuh272c6132020-11-14 16:37:52 -0800665 // Force compile time error if not using the same version runtime.
666 code += " public static void ValidateVersion() {";
667 code += " Constants.";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700668 code += "FLATBUFFERS_2_0_8(); ";
Austin Schuh272c6132020-11-14 16:37:52 -0800669 code += "}\n";
670
671 // Generate a special accessor for the table that when used as the root
672 // of a FlatBuffer
Austin Schuh2dd86a92022-09-14 21:19:23 -0700673 const std::string method_name =
674 namer_.LegacyJavaMethod2("getRootAs", struct_def, "");
675 const std::string method_signature =
676 " public static " + struct_class + " " + method_name;
Austin Schuh272c6132020-11-14 16:37:52 -0800677
678 // create convenience method that doesn't require an existing object
679 code += method_signature + "(ByteBuffer _bb) ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700680 code +=
681 "{ return " + method_name + "(_bb, new " + struct_class + "()); }\n";
Austin Schuh272c6132020-11-14 16:37:52 -0800682
683 // create method that allows object reuse
684 code +=
Austin Schuh2dd86a92022-09-14 21:19:23 -0700685 method_signature + "(ByteBuffer _bb, " + struct_class + " obj) { ";
Austin Schuh272c6132020-11-14 16:37:52 -0800686 code += "_bb.order(ByteOrder.LITTLE_ENDIAN); ";
687 code += "return (obj.__assign(_bb.getInt(_bb.";
688 code += "position()";
689 code += ") + _bb.";
690 code += "position()";
691 code += ", _bb)); }\n";
692 if (parser_.root_struct_def_ == &struct_def) {
693 if (parser_.file_identifier_.length()) {
694 // Check if a buffer has the identifier.
695 code += " public static ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700696 code += "boolean " +
697 namer_.LegacyJavaMethod2(
698 "", struct_def, "BufferHasIdentifier(ByteBuffer _bb)") +
699 " { return ";
Austin Schuh272c6132020-11-14 16:37:52 -0800700 code += "__has_identifier(_bb, \"";
701 code += parser_.file_identifier_;
702 code += "\"); }\n";
703 }
704 }
705 }
706 // Generate the __init method that sets the field in a pre-existing
707 // accessor object. This is to allow object reuse.
708 code += " public void __init(int _i, ByteBuffer _bb) ";
709 code += "{ ";
710 code += "__reset(_i, _bb); ";
711 code += "}\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700712 code += " public " + struct_class + " __assign(int _i, ByteBuffer _bb) ";
Austin Schuh272c6132020-11-14 16:37:52 -0800713 code += "{ __init(_i, _bb); return this; }\n\n";
714 for (auto it = struct_def.fields.vec.begin();
715 it != struct_def.fields.vec.end(); ++it) {
716 auto &field = **it;
717 if (field.deprecated) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700718 GenComment(field.doc_comment, &code, &comment_config, " ");
719 const std::string type_name = GenTypeGet(field.value.type);
720 const std::string type_name_dest = GenTypeNameDest(field.value.type);
721 const std::string dest_mask = DestinationMask(field.value.type, true);
722 const std::string dest_cast = DestinationCast(field.value.type);
723 const std::string src_cast = SourceCast(field.value.type);
724 const std::string method_start =
Austin Schuh272c6132020-11-14 16:37:52 -0800725 " public " +
James Kuszmaul8e62b022022-03-22 09:33:25 -0700726 (field.IsRequired() ? "" : GenNullableAnnotation(field.value.type)) +
Austin Schuh2dd86a92022-09-14 21:19:23 -0700727 GenPureAnnotation(field.value.type) + type_name_dest + " " +
728 namer_.Field(field);
729 const std::string obj = "obj";
Austin Schuh272c6132020-11-14 16:37:52 -0800730
731 // Most field accessors need to retrieve and test the field offset first,
732 // this is the prefix code for that:
733 auto offset_prefix =
734 IsArray(field.value.type)
735 ? " { return "
736 : (" { int o = __offset(" + NumToString(field.value.offset) +
737 "); return o != 0 ? ");
738 // Generate the accessors that don't do object reuse.
739 if (field.value.type.base_type == BASE_TYPE_STRUCT) {
740 // Calls the accessor that takes an accessor object with a new object.
741 code += method_start + "() { return ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700742 code += namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800743 code += "(new ";
744 code += type_name + "()); }\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -0700745 } else if (IsSeries(field.value.type) &&
Austin Schuh272c6132020-11-14 16:37:52 -0800746 field.value.type.element == BASE_TYPE_STRUCT) {
747 // Accessors for vectors of structs also take accessor objects, this
748 // generates a variant without that argument.
749 code += method_start + "(int j) { return ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700750 code += namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800751 code += "(new " + type_name + "(), j); }\n";
752 }
753
754 if (field.IsScalarOptional()) { code += GenOptionalScalarCheck(field); }
755 std::string getter = dest_cast + GenGetter(field.value.type);
756 code += method_start;
Austin Schuh272c6132020-11-14 16:37:52 -0800757 std::string member_suffix = "; ";
758 if (IsScalar(field.value.type.base_type)) {
759 code += "()";
760 member_suffix += "";
761 if (struct_def.fixed) {
762 code += " { return " + getter;
763 code += "(bb_pos + ";
764 code += NumToString(field.value.offset) + ")";
765 code += dest_mask;
766 } else {
767 code += offset_prefix + getter;
768 code += "(o + bb_pos)" + dest_mask;
Austin Schuh2dd86a92022-09-14 21:19:23 -0700769 code += " : ";
Austin Schuh272c6132020-11-14 16:37:52 -0800770 code += GenDefaultValue(field);
771 }
772 } else {
773 switch (field.value.type.base_type) {
774 case BASE_TYPE_STRUCT:
775 code += "(" + type_name + " obj)";
776 if (struct_def.fixed) {
777 code += " { return " + obj + ".__assign(";
778 code += "bb_pos + " + NumToString(field.value.offset) + ", ";
779 code += "bb)";
780 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700781 code += offset_prefix;
Austin Schuh272c6132020-11-14 16:37:52 -0800782 code += obj + ".__assign(";
783 code += field.value.type.struct_def->fixed
784 ? "o + bb_pos"
785 : "__indirect(o + bb_pos)";
786 code += ", bb) : null";
787 }
788 break;
789 case BASE_TYPE_STRING:
790 code += "()";
791 member_suffix += "";
792 code += offset_prefix + getter + "(o + ";
793 code += "bb_pos) : null";
794 break;
795 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
796 case BASE_TYPE_VECTOR: {
797 auto vectortype = field.value.type.VectorType();
798 code += "(";
799 if (vectortype.base_type == BASE_TYPE_STRUCT) {
800 code += type_name + " obj, ";
801 getter = obj + ".__assign";
802 } else if (vectortype.base_type == BASE_TYPE_UNION) {
803 code += type_name + " obj, ";
804 }
805 code += "int j)";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700806 const auto body = offset_prefix + getter + "(";
Austin Schuh272c6132020-11-14 16:37:52 -0800807 if (vectortype.base_type == BASE_TYPE_UNION) {
808 code += body + "obj, ";
809 } else {
810 code += body;
811 }
812 std::string index;
813 if (IsArray(field.value.type)) {
814 index += "bb_pos + " + NumToString(field.value.offset) + " + ";
815 } else {
816 index += "__vector(o) + ";
817 }
818 index += "j * " + NumToString(InlineSize(vectortype));
819 if (vectortype.base_type == BASE_TYPE_STRUCT) {
820 code += vectortype.struct_def->fixed
821 ? index
822 : "__indirect(" + index + ")";
823 code += ", bb";
824 } else {
825 code += index;
826 }
827 code += ")" + dest_mask;
828 if (!IsArray(field.value.type)) {
829 code += " : ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700830 code += field.value.type.element == BASE_TYPE_BOOL
831 ? "false"
832 : (IsScalar(field.value.type.element) ? "0" : "null");
Austin Schuh272c6132020-11-14 16:37:52 -0800833 }
834
835 break;
836 }
837 case BASE_TYPE_UNION:
838 code += "(" + type_name + " obj)" + offset_prefix + getter;
839 code += "(obj, o + bb_pos) : null";
840 break;
841 default: FLATBUFFERS_ASSERT(0);
842 }
843 }
844 code += member_suffix;
845 code += "}\n";
846 if (IsVector(field.value.type)) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700847 code += " public int " + namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800848 code += "Length";
849 code += "()";
850 code += offset_prefix;
851 code += "__vector_len(o) : 0; ";
852 code += "";
853 code += "}\n";
854 // See if we should generate a by-key accessor.
855 if (field.value.type.element == BASE_TYPE_STRUCT &&
856 !field.value.type.struct_def->fixed) {
857 auto &sd = *field.value.type.struct_def;
858 auto &fields = sd.fields.vec;
859 for (auto kit = fields.begin(); kit != fields.end(); ++kit) {
860 auto &key_field = **kit;
861 if (key_field.key) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700862 auto qualified_name = namer_.NamespacedType(sd);
Austin Schuh272c6132020-11-14 16:37:52 -0800863 code += " public " + qualified_name + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700864 code += namer_.Method(field) + "ByKey(";
Austin Schuh272c6132020-11-14 16:37:52 -0800865 code += GenTypeNameDest(key_field.value.type) + " key)";
866 code += offset_prefix;
867 code += qualified_name + ".__lookup_by_key(";
868 code += "null, ";
869 code += "__vector(o), key, ";
870 code += "bb) : null; ";
871 code += "}\n";
872 code += " public " + qualified_name + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700873 code += namer_.Method(field) + "ByKey(";
Austin Schuh272c6132020-11-14 16:37:52 -0800874 code += qualified_name + " obj, ";
875 code += GenTypeNameDest(key_field.value.type) + " key)";
876 code += offset_prefix;
877 code += qualified_name + ".__lookup_by_key(obj, ";
878 code += "__vector(o), key, ";
879 code += "bb) : null; ";
880 code += "}\n";
881 break;
882 }
883 }
884 }
885 }
886 // Generate the accessors for vector of structs with vector access object
887 if (IsVector(field.value.type)) {
888 std::string vector_type_name;
889 const auto &element_base_type = field.value.type.VectorType().base_type;
890 if (IsScalar(element_base_type)) {
James Kuszmaul8e62b022022-03-22 09:33:25 -0700891 vector_type_name =
892 ConvertCase(type_name, Case::kUpperCamel) + "Vector";
Austin Schuh272c6132020-11-14 16:37:52 -0800893 } else if (element_base_type == BASE_TYPE_STRING) {
894 vector_type_name = "StringVector";
895 } else if (element_base_type == BASE_TYPE_UNION) {
896 vector_type_name = "UnionVector";
897 } else {
898 vector_type_name = type_name + ".Vector";
899 }
Austin Schuh2dd86a92022-09-14 21:19:23 -0700900 auto vector_method_start = GenNullableAnnotation(field.value.type) +
901 " public " + vector_type_name + " " +
902 namer_.Field(field, "vector");
Austin Schuh272c6132020-11-14 16:37:52 -0800903 code += vector_method_start + "() { return ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700904 code += namer_.Field(field, "vector");
Austin Schuh272c6132020-11-14 16:37:52 -0800905 code += "(new " + vector_type_name + "()); }\n";
906 code += vector_method_start + "(" + vector_type_name + " obj)";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700907 code += offset_prefix + obj + ".__assign(";
Austin Schuh272c6132020-11-14 16:37:52 -0800908 code += "__vector(o), ";
909 if (!IsScalar(element_base_type)) {
910 auto vectortype = field.value.type.VectorType();
911 code += NumToString(InlineSize(vectortype)) + ", ";
912 }
913 code += "bb) : null" + member_suffix + "}\n";
914 }
915 // Generate a ByteBuffer accessor for strings & vectors of scalars.
916 if ((IsVector(field.value.type) &&
917 IsScalar(field.value.type.VectorType().base_type)) ||
918 IsString(field.value.type)) {
919 code += " public ByteBuffer ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700920 code += namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800921 code += "AsByteBuffer() { return ";
922 code += "__vector_as_bytebuffer(";
923 code += NumToString(field.value.offset) + ", ";
924 code += NumToString(IsString(field.value.type)
925 ? 1
926 : InlineSize(field.value.type.VectorType()));
927 code += "); }\n";
928 code += " public ByteBuffer ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700929 code += namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -0800930 code += "InByteBuffer(ByteBuffer _bb) { return ";
931 code += "__vector_in_bytebuffer(_bb, ";
932 code += NumToString(field.value.offset) + ", ";
933 code += NumToString(IsString(field.value.type)
934 ? 1
935 : InlineSize(field.value.type.VectorType()));
936 code += "); }\n";
937 }
938 // generate object accessors if is nested_flatbuffer
939 if (field.nested_flatbuffer) {
Austin Schuh2dd86a92022-09-14 21:19:23 -0700940 auto nested_type_name = namer_.NamespacedType(*field.nested_flatbuffer);
941 auto nested_method_name =
942 namer_.Field(field) + "As" + field.nested_flatbuffer->name;
Austin Schuh272c6132020-11-14 16:37:52 -0800943 auto get_nested_method_name = nested_method_name;
944 code += " public " + nested_type_name + " ";
945 code += nested_method_name + "() { return ";
946 code +=
947 get_nested_method_name + "(new " + nested_type_name + "()); }\n";
948 code += " public " + nested_type_name + " ";
949 code += get_nested_method_name + "(";
950 code += nested_type_name + " obj";
951 code += ") { int o = __offset(";
952 code += NumToString(field.value.offset) + "); ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700953 code += "return o != 0 ? " + obj + ".__assign(";
Austin Schuh272c6132020-11-14 16:37:52 -0800954 code += "";
955 code += "__indirect(__vector(o)), ";
956 code += "bb) : null; }\n";
957 }
958 // Generate mutators for scalar fields or vectors of scalars.
959 if (parser_.opts.mutable_buffer) {
960 auto is_series = (IsSeries(field.value.type));
961 const auto &underlying_type =
962 is_series ? field.value.type.VectorType() : field.value.type;
963 // Boolean parameters have to be explicitly converted to byte
964 // representation.
965 auto setter_parameter = underlying_type.base_type == BASE_TYPE_BOOL
966 ? "(byte)(" + field.name + " ? 1 : 0)"
967 : field.name;
Austin Schuh272c6132020-11-14 16:37:52 -0800968 // A vector mutator also needs the index of the vector element it should
969 // mutate.
970 auto mutator_params = (is_series ? "(int j, " : "(") +
971 GenTypeNameDest(underlying_type) + " " +
972 field.name + ") { ";
973 auto setter_index =
974 is_series
975 ? (IsArray(field.value.type)
976 ? "bb_pos + " + NumToString(field.value.offset)
977 : "__vector(o)") +
978 +" + j * " + NumToString(InlineSize(underlying_type))
979 : (struct_def.fixed
980 ? "bb_pos + " + NumToString(field.value.offset)
981 : "o + bb_pos");
982 if (IsScalar(underlying_type.base_type) && !IsUnion(field.value.type)) {
983 code += " public ";
984 code += struct_def.fixed ? "void " : "boolean ";
Austin Schuh2dd86a92022-09-14 21:19:23 -0700985 code += namer_.Method("mutate", field);
Austin Schuh272c6132020-11-14 16:37:52 -0800986 code += mutator_params;
987 if (struct_def.fixed) {
988 code += GenSetter(underlying_type) + "(" + setter_index + ", ";
989 code += src_cast + setter_parameter + "); }\n";
990 } else {
991 code += "int o = __offset(";
992 code += NumToString(field.value.offset) + ");";
993 code += " if (o != 0) { " + GenSetter(underlying_type);
994 code += "(" + setter_index + ", " + src_cast + setter_parameter +
995 "); return true; } else { return false; } }\n";
996 }
997 }
998 }
999 if (parser_.opts.java_primitive_has_method &&
1000 IsScalar(field.value.type.base_type) && !struct_def.fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001001 auto vt_offset_constant = " public static final int VT_" +
1002 namer_.Constant(field) + " = " +
1003 NumToString(field.value.offset) + ";";
Austin Schuh272c6132020-11-14 16:37:52 -08001004
1005 code += vt_offset_constant;
1006 code += "\n";
1007 }
1008 }
1009 code += "\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001010 auto struct_has_create = false;
1011 std::set<flatbuffers::FieldDef *> field_has_create_set;
Austin Schuh272c6132020-11-14 16:37:52 -08001012 flatbuffers::FieldDef *key_field = nullptr;
1013 if (struct_def.fixed) {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001014 struct_has_create = true;
Austin Schuh272c6132020-11-14 16:37:52 -08001015 // create a struct constructor function
1016 code += " public static " + GenOffsetType() + " ";
1017 code += "create";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001018 code += struct_class + "(FlatBufferBuilder builder";
1019 GenStructArgs(struct_def, code, "");
Austin Schuh272c6132020-11-14 16:37:52 -08001020 code += ") {\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001021 GenStructBody(struct_def, code, "");
Austin Schuh272c6132020-11-14 16:37:52 -08001022 code += " return ";
1023 code += GenOffsetConstruct("builder." + std::string("offset()"));
1024 code += ";\n }\n";
1025 } else {
1026 // Generate a method that creates a table in one go. This is only possible
1027 // when the table has no struct fields, since those have to be created
1028 // inline, and there's no way to do so in Java.
1029 bool has_no_struct_fields = true;
1030 int num_fields = 0;
1031 for (auto it = struct_def.fields.vec.begin();
1032 it != struct_def.fields.vec.end(); ++it) {
1033 auto &field = **it;
1034 if (field.deprecated) continue;
1035 if (IsStruct(field.value.type)) {
1036 has_no_struct_fields = false;
1037 } else {
1038 num_fields++;
1039 }
1040 }
1041 // JVM specifications restrict default constructor params to be < 255.
1042 // Longs and doubles take up 2 units, so we set the limit to be < 127.
1043 if (has_no_struct_fields && num_fields && num_fields < 127) {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001044 struct_has_create = true;
Austin Schuh272c6132020-11-14 16:37:52 -08001045 // Generate a table constructor of the form:
1046 // public static int createName(FlatBufferBuilder builder, args...)
1047 code += " public static " + GenOffsetType() + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001048 code += namer_.LegacyJavaMethod2("create", struct_def, "");
Austin Schuh272c6132020-11-14 16:37:52 -08001049 code += "(FlatBufferBuilder builder";
1050 for (auto it = struct_def.fields.vec.begin();
1051 it != struct_def.fields.vec.end(); ++it) {
1052 auto &field = **it;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001053 auto field_name = namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -08001054 if (field.deprecated) continue;
1055 code += ",\n ";
1056 code += GenTypeBasic(DestinationType(field.value.type, false));
1057 code += " ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001058 code += field_name;
Austin Schuh272c6132020-11-14 16:37:52 -08001059 if (!IsScalar(field.value.type.base_type)) code += "Offset";
1060 }
1061 code += ") {\n builder.";
1062 code += "startTable(";
1063 code += NumToString(struct_def.fields.vec.size()) + ");\n";
1064 for (size_t size = struct_def.sortbysize ? sizeof(largest_scalar_t) : 1;
1065 size; size /= 2) {
1066 for (auto it = struct_def.fields.vec.rbegin();
1067 it != struct_def.fields.vec.rend(); ++it) {
1068 auto &field = **it;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001069 auto field_name = namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -08001070 if (!field.deprecated &&
1071 (!struct_def.sortbysize ||
1072 size == SizeOf(field.value.type.base_type))) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001073 code += " " + struct_class + ".";
1074 code += namer_.Method("add", field) + "(builder, " + field_name;
Austin Schuh272c6132020-11-14 16:37:52 -08001075 if (!IsScalar(field.value.type.base_type)) code += "Offset";
1076 code += ");\n";
1077 }
1078 }
1079 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001080 code += " return " + struct_class + ".";
1081 code += namer_.LegacyJavaMethod2("end", struct_def, "");
Austin Schuh272c6132020-11-14 16:37:52 -08001082 code += "(builder);\n }\n\n";
1083 }
1084 // Generate a set of static methods that allow table construction,
1085 // of the form:
1086 // public static void addName(FlatBufferBuilder builder, short name)
1087 // { builder.addShort(id, name, default); }
1088 // Unlike the Create function, these always work.
1089 code += " public static void start";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001090 code += struct_class;
Austin Schuh272c6132020-11-14 16:37:52 -08001091 code += "(FlatBufferBuilder builder) { builder.";
1092 code += "startTable(";
1093 code += NumToString(struct_def.fields.vec.size()) + "); }\n";
1094 for (auto it = struct_def.fields.vec.begin();
1095 it != struct_def.fields.vec.end(); ++it) {
1096 auto &field = **it;
1097 if (field.deprecated) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001098
1099 code += " public static void " + namer_.Method("add", field);
Austin Schuh272c6132020-11-14 16:37:52 -08001100 code += "(FlatBufferBuilder builder, ";
1101 code += GenTypeBasic(DestinationType(field.value.type, false));
Austin Schuh2dd86a92022-09-14 21:19:23 -07001102 auto argname = namer_.Field(field);
Austin Schuh272c6132020-11-14 16:37:52 -08001103 if (!IsScalar(field.value.type.base_type)) argname += "Offset";
1104 code += " " + argname + ") { builder.add";
1105 code += GenMethod(field.value.type) + "(";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001106
1107 if (field.key) {
1108 // field has key attribute, so always need to exist
1109 // even if its value is equal to default.
1110 // Generated code will bypass default checking
1111 // resulting in { builder.addShort(name); slot(id); }
1112 key_field = &field;
1113 code += SourceCastBasic(field.value.type);
1114 code += argname;
1115 code += "); builder.slot(" +
1116 NumToString(it - struct_def.fields.vec.begin()) + "); }\n";
1117 } else {
1118 code += NumToString(it - struct_def.fields.vec.begin()) + ", ";
1119 code += SourceCastBasic(field.value.type);
1120 code += argname;
1121 code += ", ";
1122 code += SourceCastBasic(field.value.type);
1123 code += GenDefaultValue(field);
1124 code += "); }\n";
1125 }
Austin Schuh272c6132020-11-14 16:37:52 -08001126 if (IsVector(field.value.type)) {
1127 auto vector_type = field.value.type.VectorType();
1128 auto alignment = InlineAlignment(vector_type);
1129 auto elem_size = InlineSize(vector_type);
1130 if (!IsStruct(vector_type)) {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001131 field_has_create_set.insert(&field);
Austin Schuh272c6132020-11-14 16:37:52 -08001132 // generate a method to create a vector from a java array.
1133 if ((vector_type.base_type == BASE_TYPE_CHAR ||
1134 vector_type.base_type == BASE_TYPE_UCHAR)) {
1135 // Handle byte[] and ByteBuffers separately for Java
1136 code += " public static " + GenVectorOffsetType() + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001137 code += namer_.Method("create", field);
Austin Schuh272c6132020-11-14 16:37:52 -08001138 code += "Vector(FlatBufferBuilder builder, byte[] data) ";
1139 code += "{ return builder.createByteVector(data); }\n";
1140
1141 code += " public static " + GenVectorOffsetType() + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001142 code += namer_.Method("create", field);
Austin Schuh272c6132020-11-14 16:37:52 -08001143 code += "Vector(FlatBufferBuilder builder, ByteBuffer data) ";
1144 code += "{ return builder.createByteVector(data); }\n";
1145 } else {
1146 code += " public static " + GenVectorOffsetType() + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001147 code += namer_.Method("create", field);
Austin Schuh272c6132020-11-14 16:37:52 -08001148 code += "Vector(FlatBufferBuilder builder, ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001149 code += GenTypeBasic(DestinationType(vector_type, false)) +
1150 "[] data) ";
Austin Schuh272c6132020-11-14 16:37:52 -08001151 code += "{ builder.startVector(";
1152 code += NumToString(elem_size);
1153 code += ", data.length, ";
1154 code += NumToString(alignment);
1155 code += "); for (int i = data.";
1156 code += "length - 1; i >= 0; i--) builder.";
1157 code += "add";
1158 code += GenMethod(vector_type);
1159 code += "(";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001160 code += SourceCastBasic(vector_type);
Austin Schuh272c6132020-11-14 16:37:52 -08001161 code += "data[i]";
1162 code += "); return ";
1163 code += "builder.endVector(); }\n";
1164 }
1165 }
1166 // Generate a method to start a vector, data to be added manually
1167 // after.
Austin Schuh2dd86a92022-09-14 21:19:23 -07001168 code += " public static void " + namer_.Method("start", field);
Austin Schuh272c6132020-11-14 16:37:52 -08001169 code += "Vector(FlatBufferBuilder builder, int numElems) ";
1170 code += "{ builder.startVector(";
1171 code += NumToString(elem_size);
1172 code += ", numElems, " + NumToString(alignment);
1173 code += "); }\n";
1174 }
1175 }
1176 code += " public static " + GenOffsetType() + " ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001177 code += namer_.LegacyJavaMethod2("end", struct_def, "");
Austin Schuh272c6132020-11-14 16:37:52 -08001178 code += "(FlatBufferBuilder builder) {\n int o = builder.";
1179 code += "endTable();\n";
1180 for (auto it = struct_def.fields.vec.begin();
1181 it != struct_def.fields.vec.end(); ++it) {
1182 auto &field = **it;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001183 if (!field.deprecated && field.IsRequired()) {
Austin Schuh272c6132020-11-14 16:37:52 -08001184 code += " builder.required(o, ";
1185 code += NumToString(field.value.offset);
1186 code += "); // " + field.name + "\n";
1187 }
1188 }
1189 code += " return " + GenOffsetConstruct("o") + ";\n }\n";
1190 if (parser_.root_struct_def_ == &struct_def) {
1191 std::string size_prefix[] = { "", "SizePrefixed" };
1192 for (int i = 0; i < 2; ++i) {
1193 code += " public static void ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001194 code += namer_.LegacyJavaMethod2("finish" + size_prefix[i],
1195 struct_def, "Buffer");
1196 code += "(FlatBufferBuilder builder, " + GenOffsetType();
Austin Schuh272c6132020-11-14 16:37:52 -08001197 code += " offset) {";
1198 code += " builder.finish" + size_prefix[i] + "(offset";
1199
1200 if (parser_.file_identifier_.length())
1201 code += ", \"" + parser_.file_identifier_ + "\"";
1202 code += "); }\n";
1203 }
1204 }
1205 }
1206 // Only generate key compare function for table,
1207 // because `key_field` is not set for struct
1208 if (struct_def.has_key && !struct_def.fixed) {
1209 FLATBUFFERS_ASSERT(key_field);
1210 code += "\n @Override\n protected int keysCompare(";
1211 code += "Integer o1, Integer o2, ByteBuffer _bb) {";
1212 code += GenKeyGetter(key_field);
1213 code += " }\n";
1214
Austin Schuh2dd86a92022-09-14 21:19:23 -07001215 code += "\n public static " + struct_class;
Austin Schuh272c6132020-11-14 16:37:52 -08001216 code += " __lookup_by_key(";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001217 code += struct_class + " obj, ";
Austin Schuh272c6132020-11-14 16:37:52 -08001218 code += "int vectorLocation, ";
1219 code += GenTypeNameDest(key_field->value.type);
1220 code += " key, ByteBuffer bb) {\n";
1221 if (IsString(key_field->value.type)) {
1222 code += " byte[] byteKey = ";
1223 code += "key.getBytes(java.nio.charset.StandardCharsets.UTF_8);\n";
1224 }
1225 code += " int span = ";
1226 code += "bb.getInt(vectorLocation - 4);\n";
1227 code += " int start = 0;\n";
1228 code += " while (span != 0) {\n";
1229 code += " int middle = span / 2;\n";
1230 code += GenLookupKeyGetter(key_field);
1231 code += " if (comp > 0) {\n";
1232 code += " span = middle;\n";
1233 code += " } else if (comp < 0) {\n";
1234 code += " middle++;\n";
1235 code += " start += middle;\n";
1236 code += " span -= middle;\n";
1237 code += " } else {\n";
1238 code += " return ";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001239 code += "(obj == null ? new " + struct_class + "() : obj)";
Austin Schuh272c6132020-11-14 16:37:52 -08001240 code += ".__assign(tableOffset, bb);\n";
1241 code += " }\n }\n";
1242 code += " return null;\n";
1243 code += " }\n";
1244 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001245 GenVectorAccessObject(struct_def, code);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001246 if (opts.generate_object_based_api) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001247 GenPackUnPack_ObjectAPI(struct_def, code, opts, struct_has_create,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001248 field_has_create_set);
1249 }
1250 code += "}\n\n";
Austin Schuh272c6132020-11-14 16:37:52 -08001251 }
1252
1253 std::string GenOptionalScalarCheck(FieldDef &field) const {
1254 if (!field.IsScalarOptional()) return "";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001255 return " public boolean " + namer_.Method("has", field) +
Austin Schuh272c6132020-11-14 16:37:52 -08001256 "() { return 0 != __offset(" + NumToString(field.value.offset) +
1257 "); }\n";
1258 }
1259
Austin Schuh2dd86a92022-09-14 21:19:23 -07001260 void GenVectorAccessObject(StructDef &struct_def, std::string &code) const {
Austin Schuh272c6132020-11-14 16:37:52 -08001261 // Generate a vector of structs accessor class.
1262 code += "\n";
1263 code += " ";
1264 if (!struct_def.attributes.Lookup("private")) code += "public ";
1265 code += "static ";
1266 code += "final ";
1267 code += "class Vector extends ";
1268 code += "BaseVector {\n";
1269
1270 // Generate the __assign method that sets the field in a pre-existing
1271 // accessor object. This is to allow object reuse.
1272 std::string method_indent = " ";
1273 code += method_indent + "public Vector ";
1274 code += "__assign(int _vector, int _element_size, ByteBuffer _bb) { ";
1275 code += "__reset(_vector, _element_size, _bb); return this; }\n\n";
1276
Austin Schuh2dd86a92022-09-14 21:19:23 -07001277 auto type_name = namer_.Type(struct_def);
Austin Schuh272c6132020-11-14 16:37:52 -08001278 auto method_start = method_indent + "public " + type_name + " get";
1279 // Generate the accessors that don't do object reuse.
1280 code += method_start + "(int j) { return get";
1281 code += "(new " + type_name + "(), j); }\n";
1282 code += method_start + "(" + type_name + " obj, int j) { ";
1283 code += " return obj.__assign(";
1284 std::string index = "__element(j)";
1285 code += struct_def.fixed ? index : "__indirect(" + index + ", bb)";
1286 code += ", bb); }\n";
1287 // See if we should generate a by-key accessor.
1288 if (!struct_def.fixed) {
1289 auto &fields = struct_def.fields.vec;
1290 for (auto kit = fields.begin(); kit != fields.end(); ++kit) {
1291 auto &key_field = **kit;
1292 if (key_field.key) {
1293 auto nullable_annotation =
1294 parser_.opts.gen_nullable ? "@Nullable " : "";
1295 code += method_indent + nullable_annotation;
1296 code += "public " + type_name + " ";
1297 code += "getByKey(";
1298 code += GenTypeNameDest(key_field.value.type) + " key) { ";
1299 code += " return __lookup_by_key(null, ";
1300 code += "__vector(), key, ";
1301 code += "bb); ";
1302 code += "}\n";
1303 code += method_indent + nullable_annotation;
1304 code += "public " + type_name + " ";
1305 code += "getByKey(";
1306 code += type_name + " obj, ";
1307 code += GenTypeNameDest(key_field.value.type) + " key) { ";
1308 code += " return __lookup_by_key(obj, ";
1309 code += "__vector(), key, ";
1310 code += "bb); ";
1311 code += "}\n";
1312 break;
1313 }
1314 }
1315 }
1316 code += " }\n";
1317 }
1318
Austin Schuh2dd86a92022-09-14 21:19:23 -07001319 void GenEnum_ObjectAPI(EnumDef &enum_def, std::string &code) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001320 if (enum_def.generated) return;
1321 code += "import com.google.flatbuffers.FlatBufferBuilder;\n\n";
1322
1323 if (!enum_def.attributes.Lookup("private")) { code += "public "; }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001324 auto union_name = namer_.Type(enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001325 auto union_type =
1326 GenTypeBasic(DestinationType(enum_def.underlying_type, false));
1327 code += "class " + union_name + " {\n";
1328 // Type
1329 code += " private " + union_type + " type;\n";
1330 // Value
1331 code += " private Object value;\n";
1332 code += "\n";
1333 // getters and setters
1334 code += " public " + union_type + " getType() { return type; }\n\n";
1335 code += " public void setType(" + union_type +
1336 " type) { this.type = type; }\n\n";
1337 code += " public Object getValue() { return value; }\n\n";
1338 code += " public void setValue(Object value) { this.value = value; }\n\n";
1339 // Constructor
1340 code += " public " + union_name + "() {\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001341 code +=
1342 " this.type = " + namer_.EnumVariant(enum_def, *enum_def.Vals()[0]) +
1343 ";\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001344 code += " this.value = null;\n";
1345 code += " }\n\n";
1346 // As
1347 for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1348 auto &ev = **it;
1349 if (ev.union_type.base_type == BASE_TYPE_NONE) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001350 auto type_name = GenTypeGet_ObjectAPI(ev.union_type, false, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001351 if (ev.union_type.base_type == BASE_TYPE_STRUCT &&
1352 ev.union_type.struct_def->attributes.Lookup("private")) {
1353 code += " ";
1354 } else {
1355 code += " public ";
1356 }
1357 code += type_name + " as" + ev.name + "() { return (" + type_name +
1358 ") value; }\n";
1359 }
1360 code += "\n";
1361 // pack()
1362 code += " public static int pack(FlatBufferBuilder builder, " +
1363 union_name + " _o) {\n";
1364 code += " switch (_o.type) {\n";
1365 for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1366 auto &ev = **it;
1367 if (ev.union_type.base_type == BASE_TYPE_NONE) {
1368 continue;
1369 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001370 code += " case " + namer_.EnumVariant(enum_def, ev) + ": return ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001371 if (IsString(ev.union_type)) {
1372 code += "builder.createString(_o.as" + ev.name + "());\n";
1373 } else {
1374 code += GenTypeGet(ev.union_type) + ".pack(builder, _o.as" + ev.name +
1375 "());\n";
1376 }
1377 }
1378 }
1379 code += " default: return 0;\n";
1380 code += " }\n";
1381 code += " }\n";
1382 code += "}\n\n";
1383 }
1384
Austin Schuh2dd86a92022-09-14 21:19:23 -07001385 void GenUnionUnPack_ObjectAPI(const EnumDef &enum_def, std::string &code,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001386 const std::string &type_name,
Austin Schuh2dd86a92022-09-14 21:19:23 -07001387 const std::string &field_name,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001388 bool is_vector) const {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001389 const std::string variable_type =
1390 is_vector ? type_name.substr(0, type_name.length() - 2) : type_name;
1391 const std::string variable_name =
1392 "_" + namer_.Variable("o", field_name) + (is_vector ? "Element" : "");
1393 const std::string type_params = is_vector ? "_j" : "";
1394 const std::string value_params = is_vector ? ", _j" : "";
1395 const std::string indent = (is_vector ? " " : " ");
James Kuszmaul8e62b022022-03-22 09:33:25 -07001396
James Kuszmaul8e62b022022-03-22 09:33:25 -07001397 code += indent + variable_type + " " + variable_name + " = new " +
1398 variable_type + "();\n";
1399 code += indent +
1400 GenTypeBasic(DestinationType(enum_def.underlying_type, false)) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001401 " " + variable_name + "Type = " + field_name + "Type(" +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001402 type_params + ");\n";
1403 code += indent + variable_name + ".setType(" + variable_name + "Type);\n";
1404 code += indent + "Table " + variable_name + "Value;\n";
1405 code += indent + "switch (" + variable_name + "Type) {\n";
1406 for (auto eit = enum_def.Vals().begin(); eit != enum_def.Vals().end();
1407 ++eit) {
1408 auto &ev = **eit;
1409 if (ev.union_type.base_type == BASE_TYPE_NONE) {
1410 continue;
1411 } else {
1412 if (ev.union_type.base_type == BASE_TYPE_STRING ||
1413 (ev.union_type.base_type == BASE_TYPE_STRUCT &&
1414 ev.union_type.struct_def->fixed)) {
1415 continue; // This branch is due to bad implemantation of Unions in
1416 // Java which doesn't handle non Table types. Should be
1417 // deleted when issue #6561 is fixed.
1418 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001419 code += indent + " case " +
1420 namer_.NamespacedEnumVariant(enum_def, ev) + ":\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001421 auto actual_type = GenTypeGet(ev.union_type);
Austin Schuh2dd86a92022-09-14 21:19:23 -07001422 code += indent + " " + variable_name + "Value = " + field_name +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001423 "(new " + actual_type + "()" + value_params + ");\n";
1424 code += indent + " " + variable_name + ".setValue(" + variable_name +
1425 "Value != null ? ((" + actual_type + ") " + variable_name +
1426 "Value).unpack() : null);\n";
1427 code += indent + " break;\n";
1428 }
1429 }
1430 code += indent + " default: break;\n";
1431 code += indent + "}\n";
1432 if (is_vector) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001433 code += indent + "_" + namer_.Variable("o", field_name) +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001434 "[_j] = " + variable_name + ";\n";
1435 }
1436 }
1437
1438 void GenPackUnPack_ObjectAPI(
Austin Schuh2dd86a92022-09-14 21:19:23 -07001439 StructDef &struct_def, std::string &code, const IDLOptions &opts,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001440 bool struct_has_create,
1441 const std::set<FieldDef *> &field_has_create) const {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001442 auto struct_name = namer_.ObjectType(struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001443 // unpack()
1444 code += " public " + struct_name + " unpack() {\n";
1445 code += " " + struct_name + " _o = new " + struct_name + "();\n";
1446 code += " unpackTo(_o);\n";
1447 code += " return _o;\n";
1448 code += " }\n";
1449 // unpackTo()
1450 code += " public void unpackTo(" + struct_name + " _o) {\n";
1451 for (auto it = struct_def.fields.vec.begin();
1452 it != struct_def.fields.vec.end(); ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001453 const auto &field = **it;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001454 if (field.deprecated) continue;
1455 if (field.value.type.base_type == BASE_TYPE_UTYPE) continue;
1456 if (field.value.type.element == BASE_TYPE_UTYPE) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001457 const auto accessor = namer_.Method(field);
1458 const auto variable = "_" + namer_.Variable("o", field);
1459 const auto get_field = namer_.Method("get", field);
1460 const auto set_field = namer_.Method("set", field);
1461
1462 auto type_name = GenTypeGet_ObjectAPI(field.value.type, false, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001463 if (field.IsScalarOptional())
1464 type_name = ConvertPrimitiveTypeToObjectWrapper_ObjectAPI(type_name);
Austin Schuh2dd86a92022-09-14 21:19:23 -07001465 auto start = " " + type_name + " " + variable + " = ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001466 auto call_setter = true;
1467 switch (field.value.type.base_type) {
1468 case BASE_TYPE_STRUCT: {
1469 auto fixed = struct_def.fixed && field.value.type.struct_def->fixed;
1470 if (fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001471 code +=
1472 " " + accessor + "().unpackTo(_o." + get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001473 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001474 code += " if (" + accessor + "() != null) ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001475 if (field.value.type.struct_def->fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001476 code += accessor + "().unpackTo(_o." + get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001477 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001478 code += "_o." + set_field + "(" + accessor + "().unpack());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001479 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001480 code += " else _o." + set_field + "(null);\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001481 }
1482 call_setter = false;
1483 break;
1484 }
1485 case BASE_TYPE_ARRAY: {
1486 auto length_str = NumToString(field.value.type.fixed_length);
1487 auto unpack_method =
1488 field.value.type.struct_def == nullptr ? "" : ".unpack()";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001489 code += start + "_o." + get_field + "();\n";
1490 code += " for (int _j = 0; _j < " + length_str + "; ++_j) { " +
1491 variable + "[_j] = " + accessor + "(_j)" + unpack_method +
1492 "; }\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001493 call_setter = false;
1494 break;
1495 }
1496 case BASE_TYPE_VECTOR:
1497 if (field.value.type.element == BASE_TYPE_UNION) {
1498 code += start + "new " +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001499 GenConcreteTypeGet_ObjectAPI(field.value.type)
James Kuszmaul8e62b022022-03-22 09:33:25 -07001500 .substr(0, type_name.length() - 1) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001501 accessor + "Length()];\n";
1502 code +=
1503 " for (int _j = 0; _j < " + accessor + "Length(); ++_j) {\n";
1504 GenUnionUnPack_ObjectAPI(*field.value.type.enum_def, code,
1505 type_name, accessor, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001506 code += " }\n";
1507 } else if (field.value.type.element != BASE_TYPE_UTYPE) {
1508 auto fixed = field.value.type.struct_def == nullptr;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001509 const auto length_accessor = namer_.Method(field, "length");
James Kuszmaul8e62b022022-03-22 09:33:25 -07001510 code += start + "new " +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001511 GenConcreteTypeGet_ObjectAPI(field.value.type)
James Kuszmaul8e62b022022-03-22 09:33:25 -07001512 .substr(0, type_name.length() - 1) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001513 length_accessor + "()];\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001514 code +=
Austin Schuh2dd86a92022-09-14 21:19:23 -07001515 " for (int _j = 0; _j < " + length_accessor + "(); ++_j) {";
1516 code += variable + "[_j] = ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001517 if (fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001518 code += accessor + "(_j)";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001519 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001520 code += "(" + accessor + "(_j) != null ? " + accessor +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001521 "(_j).unpack() : null)";
1522 }
1523 code += ";}\n";
1524 }
1525 break;
1526 case BASE_TYPE_UTYPE: break;
1527 case BASE_TYPE_UNION: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001528 GenUnionUnPack_ObjectAPI(*field.value.type.enum_def, code, type_name,
1529 accessor, false);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001530 break;
1531 }
1532 default: {
1533 if (field.IsScalarOptional()) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001534 code += start + namer_.Method("has", field) + "() ? " + accessor +
1535 "() : null;\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001536 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001537 code += start + accessor + "();\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001538 }
1539 break;
1540 }
1541 }
1542 if (call_setter) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001543 code += " _o." + set_field + "(" + variable + ");\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001544 }
1545 }
1546 code += " }\n";
1547 // pack()
1548 code += " public static " + GenOffsetType() +
1549 " pack(FlatBufferBuilder builder, " + struct_name + " _o) {\n";
1550 code += " if (_o == null) return 0;\n";
1551 for (auto it = struct_def.fields.vec.begin();
1552 it != struct_def.fields.vec.end(); ++it) {
1553 auto &field = **it;
1554 if (field.deprecated) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001555 const auto field_name = namer_.Field(field);
1556 const auto variable = "_" + namer_.Variable("o", field);
1557 const auto get_field = namer_.Method("get", field);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001558 // pre
1559 switch (field.value.type.base_type) {
1560 case BASE_TYPE_STRUCT: {
1561 if (!field.value.type.struct_def->fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001562 code += " " + GenOffsetType() + " _" + namer_.Variable(field) +
1563 " = _o." + get_field +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001564 "() == null ? 0 : " + GenTypeGet(field.value.type) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001565 ".pack(builder, _o." + get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001566 } else if (struct_def.fixed && struct_has_create) {
1567 std::vector<FieldArrayLength> array_lengths;
1568 FieldArrayLength tmp_array_length = {
1569 field.name,
1570 field.value.type.fixed_length,
1571 };
1572 array_lengths.push_back(tmp_array_length);
Austin Schuh2dd86a92022-09-14 21:19:23 -07001573 GenStructPackDecl_ObjectAPI(*field.value.type.struct_def,
1574 array_lengths, code);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001575 }
1576 break;
1577 }
1578 case BASE_TYPE_STRING: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001579 code += " int _" + field_name + " = _o." + get_field +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001580 "() == null ? 0 : "
Austin Schuh2dd86a92022-09-14 21:19:23 -07001581 "builder.createString(_o." +
1582 get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001583 break;
1584 }
1585 case BASE_TYPE_VECTOR: {
1586 if (field_has_create.find(&field) != field_has_create.end()) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001587 auto property_name = field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001588 auto gen_for_loop = true;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001589 std::string array_name = "__" + field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001590 std::string array_type = "";
1591 std::string element_type = "";
1592 std::string to_array = "";
1593 switch (field.value.type.element) {
1594 case BASE_TYPE_STRING: {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001595 array_type = "int";
1596 element_type = "String";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001597 to_array = "builder.createString(_e)";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001598 break;
1599 }
1600 case BASE_TYPE_STRUCT:
1601 array_type = "int";
1602 element_type =
Austin Schuh2dd86a92022-09-14 21:19:23 -07001603 GenTypeGet_ObjectAPI(field.value.type, true, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001604 ;
1605 to_array = GenTypeGet(field.value.type) + ".pack(builder, _e)";
1606 break;
1607 case BASE_TYPE_UTYPE:
Austin Schuh2dd86a92022-09-14 21:19:23 -07001608 property_name = field_name.substr(0, field_name.size() - 4);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001609 array_type = GenTypeBasic(DestinationType(
1610 field.value.type.enum_def->underlying_type, false));
1611 element_type = field.value.type.enum_def->name + "Union";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001612 to_array = "_o." + namer_.Method("get", property_name) +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001613 "()[_j].getType()";
1614 break;
1615 case BASE_TYPE_UNION:
1616 array_type = "int";
1617 element_type =
Austin Schuh2dd86a92022-09-14 21:19:23 -07001618 namer_.NamespacedType(*field.value.type.enum_def) + "Union";
1619 to_array = element_type + ".pack(builder, _o." +
1620 namer_.Method("get", property_name) + "()[_j])";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001621 break;
1622 case BASE_TYPE_UCHAR: // TODO this branch of the switch is due to
1623 // inconsistent behavior in unsigned byte.
1624 // Read further at Issue #6574.
1625 array_type = "byte";
1626 element_type = "int";
1627 to_array = "(byte) _e";
1628 break;
1629 default:
1630 gen_for_loop = false;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001631 array_name = "_o." + namer_.Method("get", property_name) + "()";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001632 array_type = GenTypeNameDest(field.value.type);
1633 element_type = array_type;
1634 to_array = "_e";
1635 break;
1636 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001637 code += " int _" + field_name + " = 0;\n";
1638 code += " if (_o." + namer_.Method("get", property_name) +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001639 "() != null) {\n";
1640 if (gen_for_loop) {
1641 code += " " + array_type + "[] " + array_name + " = new " +
1642 array_type + "[_o." +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001643 namer_.Method("get", property_name) + "().length];\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001644 code += " int _j = 0;\n";
1645 code += " for (" + element_type + " _e : _o." +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001646 namer_.Method("get", property_name) + "()) { ";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001647 code += array_name + "[_j] = " + to_array + "; _j++;}\n";
1648 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001649 code += " _" + field_name + " = " +
1650 namer_.Method("create", field) + "Vector(builder, " +
1651 array_name + ");\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001652 code += " }\n";
1653 } else {
1654 auto type_name = GenTypeGet(field.value.type);
1655 auto element_type_name =
Austin Schuh2dd86a92022-09-14 21:19:23 -07001656 GenTypeGet_ObjectAPI(field.value.type, true, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001657 auto pack_method =
1658 field.value.type.struct_def == nullptr
1659 ? "builder.add" + GenMethod(field.value.type.VectorType()) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001660 "(" + variable + "[_j]);"
1661 : "_unused_offset = " + type_name + ".pack(builder, " +
1662 variable + "[_j]);";
1663 code += " int _" + field_name + " = 0;\n";
1664 code += " " + element_type_name + "[] " + variable + " = _o." +
1665 get_field + "();\n";
1666 code += " if (" + variable + " != null) {\n";
1667 if (field.value.type.struct_def != nullptr) {
1668 code += " int _unused_offset = 0;\n";
1669 }
1670 code += " " + namer_.Method("start", field) +
1671 "Vector(builder, " + variable + ".length);\n";
1672 code += " for (int _j = " + variable +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001673 ".length - 1; _j >=0; _j--) { ";
1674 code += pack_method + "}\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001675 code += " _" + field_name + " = builder.endVector();\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001676 code += " }\n";
1677 }
1678 break;
1679 }
1680 case BASE_TYPE_ARRAY: {
1681 if (field.value.type.struct_def != nullptr) {
1682 std::vector<FieldArrayLength> array_lengths;
1683 FieldArrayLength tmp_array_length = {
1684 field.name,
1685 field.value.type.fixed_length,
1686 };
1687 array_lengths.push_back(tmp_array_length);
Austin Schuh2dd86a92022-09-14 21:19:23 -07001688 GenStructPackDecl_ObjectAPI(*field.value.type.struct_def,
1689 array_lengths, code);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001690 } else {
1691 code += " " +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001692 GenTypeGet_ObjectAPI(field.value.type, false, true) + " _" +
1693 field_name + " = _o." + get_field + "();\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001694 }
1695 break;
1696 }
1697 case BASE_TYPE_UNION: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001698 code += " " +
1699 GenTypeBasic(DestinationType(
1700 field.value.type.enum_def->underlying_type, false)) +
1701 " _" + field_name + "Type = _o." + get_field +
1702 "() == null ? " +
1703 namer_.NamespacedType(*field.value.type.enum_def) +
1704 ".NONE : " + "_o." + get_field + "().getType();\n";
1705 code += " " + GenOffsetType() + " _" + field_name + " = _o." +
1706 get_field + "() == null ? 0 : " +
1707 namer_.NamespacedType(*field.value.type.enum_def) +
1708 "Union.pack(builder, _o." + get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001709 break;
1710 }
1711 default: break;
1712 }
1713 }
1714 if (struct_has_create) {
1715 // Create
Austin Schuh2dd86a92022-09-14 21:19:23 -07001716 code += " return " +
1717 namer_.LegacyJavaMethod2("create", struct_def, "") + "(\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001718 code += " builder";
1719 for (auto it = struct_def.fields.vec.begin();
1720 it != struct_def.fields.vec.end(); ++it) {
1721 auto &field = **it;
1722 if (field.deprecated) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001723 const auto field_name = namer_.Field(field);
1724 const auto get_field = namer_.Method("get", field);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001725 switch (field.value.type.base_type) {
1726 case BASE_TYPE_STRUCT: {
1727 if (struct_def.fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001728 GenStructPackCall_ObjectAPI(*field.value.type.struct_def, code,
1729 " _" + field_name + "_");
James Kuszmaul8e62b022022-03-22 09:33:25 -07001730 } else {
1731 code += ",\n";
1732 if (field.value.type.struct_def->fixed) {
1733 if (opts.generate_object_based_api)
Austin Schuh2dd86a92022-09-14 21:19:23 -07001734 code += " _o." + field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001735 else
1736 // Seems like unreachable code
1737 code += " " + GenTypeGet(field.value.type) +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001738 ".Pack(builder, _o." + field_name + ")";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001739 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001740 code += " _" + field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001741 }
1742 }
1743 break;
1744 }
1745 case BASE_TYPE_ARRAY: {
1746 if (field.value.type.struct_def != nullptr) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001747 GenStructPackCall_ObjectAPI(*field.value.type.struct_def, code,
1748 " _" + field_name + "_");
James Kuszmaul8e62b022022-03-22 09:33:25 -07001749 } else {
1750 code += ",\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001751 code += " _" + field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001752 }
1753 break;
1754 }
1755 case BASE_TYPE_UNION: FLATBUFFERS_FALLTHROUGH(); // fall thru
1756 case BASE_TYPE_UTYPE: FLATBUFFERS_FALLTHROUGH(); // fall thru
1757 case BASE_TYPE_STRING: FLATBUFFERS_FALLTHROUGH(); // fall thru
1758 case BASE_TYPE_VECTOR: {
1759 code += ",\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001760 code += " _" + field_name;
James Kuszmaul8e62b022022-03-22 09:33:25 -07001761 break;
1762 }
1763 default: // scalar
1764 code += ",\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001765 code += " _o." + get_field + "()";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001766 break;
1767 }
1768 }
1769 code += ");\n";
1770 } else {
1771 // Start, End
Austin Schuh2dd86a92022-09-14 21:19:23 -07001772 code += " " + namer_.LegacyJavaMethod2("start", struct_def, "") +
1773 "(builder);\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001774 for (auto it = struct_def.fields.vec.begin();
1775 it != struct_def.fields.vec.end(); ++it) {
1776 auto &field = **it;
1777 if (field.deprecated) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07001778 const auto arg = "_" + namer_.Variable(field);
1779 const auto get_field = namer_.Method("get", field);
1780 const auto add_field = namer_.Method("add", field);
1781
James Kuszmaul8e62b022022-03-22 09:33:25 -07001782 switch (field.value.type.base_type) {
1783 case BASE_TYPE_STRUCT: {
1784 if (field.value.type.struct_def->fixed) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001785 code += " " + add_field + "(builder, " +
James Kuszmaul8e62b022022-03-22 09:33:25 -07001786 GenTypeGet(field.value.type) + ".pack(builder, _o." +
Austin Schuh2dd86a92022-09-14 21:19:23 -07001787 get_field + "()));\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001788 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001789 code += " " + add_field + "(builder, " + arg + ");\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001790 }
1791 break;
1792 }
1793 case BASE_TYPE_STRING: FLATBUFFERS_FALLTHROUGH(); // fall thru
1794 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
1795 case BASE_TYPE_VECTOR: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001796 code += " " + add_field + "(builder, " + arg + ");\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001797 break;
1798 }
1799 case BASE_TYPE_UTYPE: break;
1800 case BASE_TYPE_UNION: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001801 code += " " + add_field + "Type(builder, " + arg + "Type);\n";
1802 code += " " + add_field + "(builder, " + arg + ");\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001803 break;
1804 }
1805 // scalar
1806 default: {
1807 if (field.IsScalarOptional()) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001808 code += " if (_o." + get_field + "() != null) { " + add_field +
1809 "(builder, _o." + get_field + "()); }\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001810 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001811 code +=
1812 " " + add_field + "(builder, _o." + get_field + "());\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001813 }
1814 break;
1815 }
1816 }
1817 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07001818 code += " return " + namer_.LegacyJavaMethod2("end", struct_def, "") +
1819 "(builder);\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001820 }
1821 code += " }\n";
1822 }
1823
Austin Schuh2dd86a92022-09-14 21:19:23 -07001824 void GenStructPackDecl_ObjectAPI(const StructDef &struct_def,
1825 std::vector<FieldArrayLength> &array_lengths,
1826 std::string &code) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001827 for (auto it = struct_def.fields.vec.begin();
1828 it != struct_def.fields.vec.end(); ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001829 const FieldDef &field = **it;
1830 const bool is_array = IsArray(field.value.type);
1831 const Type &field_type =
James Kuszmaul8e62b022022-03-22 09:33:25 -07001832 is_array ? field.value.type.VectorType() : field.value.type;
1833 FieldArrayLength tmp_array_length = {
1834 field.name,
1835 field_type.fixed_length,
1836 };
1837 array_lengths.push_back(tmp_array_length);
1838 if (field_type.struct_def != nullptr) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001839 GenStructPackDecl_ObjectAPI(*field_type.struct_def, array_lengths,
1840 code);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001841 } else {
1842 std::vector<FieldArrayLength> array_only_lengths;
1843 for (size_t i = 0; i < array_lengths.size(); ++i) {
1844 if (array_lengths[i].length > 0) {
1845 array_only_lengths.push_back(array_lengths[i]);
1846 }
1847 }
1848 std::string name;
1849 for (size_t i = 0; i < array_lengths.size(); ++i) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001850 name += "_" + namer_.Variable(array_lengths[i].name);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001851 }
1852 code += " " + GenTypeBasic(field_type);
1853 if (array_only_lengths.size() > 0) {
1854 for (size_t i = 0; i < array_only_lengths.size(); ++i) {
1855 code += "[]";
1856 }
1857 code += " " + name + " = ";
1858 code += "new " + GenTypeBasic(field_type) + "[";
1859 for (size_t i = 0; i < array_only_lengths.size(); ++i) {
1860 if (i != 0) { code += "]["; }
1861 code += NumToString(array_only_lengths[i].length);
1862 }
1863 code += "];\n";
1864 code += " ";
1865 // initialize array
1866 for (size_t i = 0; i < array_only_lengths.size(); ++i) {
1867 auto idx = "idx" + NumToString(i);
1868 code += "for (int " + idx + " = 0; " + idx + " < " +
1869 NumToString(array_only_lengths[i].length) + "; ++" + idx +
1870 ") {";
1871 }
1872 for (size_t i = 0; i < array_only_lengths.size(); ++i) {
1873 auto idx = "idx" + NumToString(i);
1874 if (i == 0) {
1875 code += name + "[" + idx;
1876 } else {
1877 code += "][" + idx;
1878 }
1879 }
1880 code += "] = _o";
1881 for (size_t i = 0, j = 0; i < array_lengths.size(); ++i) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001882 code += "." + namer_.Method("get", array_lengths[i].name) + "()";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001883 if (array_lengths[i].length <= 0) continue;
1884 code += "[idx" + NumToString(j++) + "]";
1885 }
1886 code += ";";
1887 for (size_t i = 0; i < array_only_lengths.size(); ++i) {
1888 code += "}";
1889 }
1890 } else {
1891 code += " " + name + " = ";
1892 code += "_o";
1893 for (size_t i = 0; i < array_lengths.size(); ++i) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001894 code += "." + namer_.Method("get", array_lengths[i].name) + "()";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001895 }
1896 code += ";";
1897 }
1898 code += "\n";
1899 }
1900 array_lengths.pop_back();
1901 }
1902 }
1903
1904 void GenStructPackCall_ObjectAPI(const StructDef &struct_def,
Austin Schuh2dd86a92022-09-14 21:19:23 -07001905 std::string &code,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001906 std::string prefix) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001907 for (auto it = struct_def.fields.vec.begin();
1908 it != struct_def.fields.vec.end(); ++it) {
1909 auto &field = **it;
1910 const auto &field_type = field.value.type;
1911 if (field_type.struct_def != nullptr) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001912 GenStructPackCall_ObjectAPI(*field_type.struct_def, code,
1913 prefix + namer_.Field(field) + "_");
James Kuszmaul8e62b022022-03-22 09:33:25 -07001914 } else {
1915 code += ",\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07001916 code += prefix + namer_.Field(field);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001917 }
1918 }
1919 }
1920
1921 std::string ConvertPrimitiveTypeToObjectWrapper_ObjectAPI(
Austin Schuh2dd86a92022-09-14 21:19:23 -07001922 const std::string &type_name) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001923 if (type_name == "boolean")
1924 return "Boolean";
1925 else if (type_name == "byte")
1926 return "Byte";
1927 else if (type_name == "char")
1928 return "Character";
1929 else if (type_name == "short")
1930 return "Short";
1931 else if (type_name == "int")
1932 return "Integer";
1933 else if (type_name == "long")
1934 return "Long";
1935 else if (type_name == "float")
1936 return "Float";
1937 else if (type_name == "double")
1938 return "Double";
1939 return type_name;
1940 }
1941
Austin Schuh2dd86a92022-09-14 21:19:23 -07001942 std::string GenTypeGet_ObjectAPI(const flatbuffers::Type &type,
1943 bool vectorelem,
James Kuszmaul8e62b022022-03-22 09:33:25 -07001944 bool wrap_in_namespace) const {
1945 auto type_name = GenTypeNameDest(type);
1946 // Replace to ObjectBaseAPI Type Name
1947 switch (type.base_type) {
1948 case BASE_TYPE_STRUCT: FLATBUFFERS_FALLTHROUGH(); // fall thru
1949 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
1950 case BASE_TYPE_VECTOR: {
1951 if (type.struct_def != nullptr) {
1952 auto type_name_length = type.struct_def->name.length();
Austin Schuh2dd86a92022-09-14 21:19:23 -07001953 auto new_type_name = namer_.ObjectType(*type.struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001954 type_name.replace(type_name.length() - type_name_length,
1955 type_name_length, new_type_name);
1956 } else if (type.element == BASE_TYPE_UNION) {
1957 if (wrap_in_namespace) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001958 type_name = namer_.NamespacedType(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001959 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001960 type_name = namer_.Type(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001961 }
1962 }
1963 break;
1964 }
1965
1966 case BASE_TYPE_UNION: {
1967 if (wrap_in_namespace) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001968 type_name = namer_.NamespacedType(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001969 } else {
Austin Schuh2dd86a92022-09-14 21:19:23 -07001970 type_name = namer_.Type(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07001971 }
1972 break;
1973 }
1974 default: break;
1975 }
1976 if (vectorelem) { return type_name; }
1977 switch (type.base_type) {
1978 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
1979 case BASE_TYPE_VECTOR: {
1980 type_name = type_name + "[]";
1981 break;
1982 }
1983 default: break;
1984 }
1985 return type_name;
1986 }
1987
Austin Schuh2dd86a92022-09-14 21:19:23 -07001988 std::string GenConcreteTypeGet_ObjectAPI(
1989 const flatbuffers::Type &type) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07001990 auto type_name = GenTypeNameDest(type);
1991 // Replace to ObjectBaseAPI Type Name
1992 switch (type.base_type) {
1993 case BASE_TYPE_STRUCT: FLATBUFFERS_FALLTHROUGH(); // fall thru
1994 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
1995 case BASE_TYPE_VECTOR: {
1996 if (type.struct_def != nullptr) {
1997 auto type_name_length = type.struct_def->name.length();
Austin Schuh2dd86a92022-09-14 21:19:23 -07001998 auto new_type_name = namer_.ObjectType(*type.struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -07001999 type_name.replace(type_name.length() - type_name_length,
2000 type_name_length, new_type_name);
2001 } else if (type.element == BASE_TYPE_UNION) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002002 type_name = namer_.NamespacedType(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002003 }
2004 break;
2005 }
2006
2007 case BASE_TYPE_UNION: {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002008 type_name = namer_.NamespacedType(*type.enum_def) + "Union";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002009 break;
2010 }
2011 default: break;
2012 }
2013
2014 switch (type.base_type) {
2015 case BASE_TYPE_ARRAY: FLATBUFFERS_FALLTHROUGH(); // fall thru
2016 case BASE_TYPE_VECTOR: {
2017 type_name = type_name + "[]";
2018 break;
2019 }
2020 default: break;
2021 }
2022 return type_name;
2023 }
2024
Austin Schuh2dd86a92022-09-14 21:19:23 -07002025 void GenStruct_ObjectAPI(const StructDef &struct_def,
2026 std::string &code) const {
James Kuszmaul8e62b022022-03-22 09:33:25 -07002027 if (struct_def.generated) return;
James Kuszmaul8e62b022022-03-22 09:33:25 -07002028 if (struct_def.attributes.Lookup("private")) {
2029 // For Java, we leave the enum unmarked to indicate package-private
2030 } else {
2031 code += "public ";
2032 }
2033
Austin Schuh2dd86a92022-09-14 21:19:23 -07002034 const auto class_name = namer_.ObjectType(struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -07002035 code += "class " + class_name;
2036 code += " {\n";
2037 // Generate Properties
2038 for (auto it = struct_def.fields.vec.begin();
2039 it != struct_def.fields.vec.end(); ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002040 const auto &field = **it;
James Kuszmaul8e62b022022-03-22 09:33:25 -07002041 if (field.deprecated) continue;
2042 if (field.value.type.base_type == BASE_TYPE_UTYPE) continue;
2043 if (field.value.type.element == BASE_TYPE_UTYPE) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07002044 auto type_name = GenTypeGet_ObjectAPI(field.value.type, false, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07002045 if (field.IsScalarOptional())
2046 type_name = ConvertPrimitiveTypeToObjectWrapper_ObjectAPI(type_name);
Austin Schuh2dd86a92022-09-14 21:19:23 -07002047 const auto field_name = namer_.Field(field);
2048 code += " private " + type_name + " " + field_name + ";\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002049 }
2050 // Generate Java getters and setters
2051 code += "\n";
2052 for (auto it = struct_def.fields.vec.begin();
2053 it != struct_def.fields.vec.end(); ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002054 const auto &field = **it;
James Kuszmaul8e62b022022-03-22 09:33:25 -07002055 if (field.deprecated) continue;
2056 if (field.value.type.base_type == BASE_TYPE_UTYPE) continue;
2057 if (field.value.type.element == BASE_TYPE_UTYPE) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07002058 const auto field_name = namer_.Field(field);
2059 const auto get_field = namer_.Method("get", field);
2060 auto type_name = GenTypeGet_ObjectAPI(field.value.type, false, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07002061 if (field.IsScalarOptional())
2062 type_name = ConvertPrimitiveTypeToObjectWrapper_ObjectAPI(type_name);
Austin Schuh2dd86a92022-09-14 21:19:23 -07002063
2064 code += " public " + type_name + " " + get_field + "() { return " +
2065 field_name + "; }\n\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002066 std::string array_validation = "";
2067 if (field.value.type.base_type == BASE_TYPE_ARRAY) {
2068 array_validation =
Austin Schuh2dd86a92022-09-14 21:19:23 -07002069 "if (" + field_name + " != null && " + field_name +
James Kuszmaul8e62b022022-03-22 09:33:25 -07002070 ".length == " + NumToString(field.value.type.fixed_length) + ") ";
2071 }
Austin Schuh2dd86a92022-09-14 21:19:23 -07002072 code += " public void " + namer_.Method("set", field) + "(" + type_name +
2073 " " + field_name + ") { " + array_validation + "this." +
2074 field_name + " = " + field_name + "; }\n\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002075 }
2076 // Generate Constructor
2077 code += "\n";
2078 code += " public " + class_name + "() {\n";
2079 for (auto it = struct_def.fields.vec.begin();
2080 it != struct_def.fields.vec.end(); ++it) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002081 const auto &field = **it;
James Kuszmaul8e62b022022-03-22 09:33:25 -07002082 if (field.deprecated) continue;
2083 if (field.value.type.base_type == BASE_TYPE_UTYPE) continue;
2084 if (field.value.type.element == BASE_TYPE_UTYPE) continue;
Austin Schuh2dd86a92022-09-14 21:19:23 -07002085 const auto get_field = namer_.Method("get", field);
2086
2087 code += " this." + namer_.Field(field) + " = ";
2088 const auto type_name =
2089 GenTypeGet_ObjectAPI(field.value.type, false, true);
James Kuszmaul8e62b022022-03-22 09:33:25 -07002090 if (IsScalar(field.value.type.base_type)) {
2091 if (field.IsScalarOptional()) {
2092 code += "null;\n";
2093 } else {
2094 code += GenDefaultValue(field) + ";\n";
2095 }
2096 } else {
2097 switch (field.value.type.base_type) {
2098 case BASE_TYPE_STRUCT: {
2099 if (IsStruct(field.value.type)) {
2100 code += "new " + type_name + "();\n";
2101 } else {
2102 code += "null;\n";
2103 }
2104 break;
2105 }
2106 case BASE_TYPE_ARRAY: {
2107 code += "new " + type_name.substr(0, type_name.length() - 1) +
2108 NumToString(field.value.type.fixed_length) + "];\n";
2109 break;
2110 }
2111 default: {
2112 code += "null;\n";
2113 break;
2114 }
2115 }
2116 }
2117 }
2118 code += " }\n";
2119 if (parser_.root_struct_def_ == &struct_def) {
Austin Schuh2dd86a92022-09-14 21:19:23 -07002120 const std::string struct_type = namer_.Type(struct_def);
James Kuszmaul8e62b022022-03-22 09:33:25 -07002121 code += " public static " + class_name +
2122 " deserializeFromBinary(byte[] fbBuffer) {\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07002123 code += " return " + struct_type + "." +
2124 namer_.LegacyJavaMethod2("getRootAs", struct_def, "") +
James Kuszmaul8e62b022022-03-22 09:33:25 -07002125 "(ByteBuffer.wrap(fbBuffer)).unpack();\n";
2126 code += " }\n";
2127 code += " public byte[] serializeToBinary() {\n";
2128 code += " FlatBufferBuilder fbb = new FlatBufferBuilder();\n";
Austin Schuh2dd86a92022-09-14 21:19:23 -07002129 code += " " + struct_type + "." +
2130 namer_.LegacyJavaMethod2("finish", struct_def, "Buffer") +
2131 "(fbb, " + struct_type + ".pack(fbb, this));\n";
James Kuszmaul8e62b022022-03-22 09:33:25 -07002132 code += " return fbb.sizedByteArray();\n";
2133 code += " }\n";
2134 }
2135 code += "}\n\n";
2136 }
2137
Austin Schuh272c6132020-11-14 16:37:52 -08002138 // This tracks the current namespace used to determine if a type need to be
2139 // prefixed by its namespace
2140 const Namespace *cur_name_space_;
Austin Schuh2dd86a92022-09-14 21:19:23 -07002141 const IdlNamer namer_;
Austin Schuh272c6132020-11-14 16:37:52 -08002142};
2143} // namespace java
2144
2145bool GenerateJava(const Parser &parser, const std::string &path,
2146 const std::string &file_name) {
2147 java::JavaGenerator generator(parser, path, file_name);
2148 return generator.generate();
2149}
2150
2151} // namespace flatbuffers