1*890232f2SAndroid Build Coastguard Worker /*
2*890232f2SAndroid Build Coastguard Worker * Copyright 2014 Google Inc. All rights reserved.
3*890232f2SAndroid Build Coastguard Worker *
4*890232f2SAndroid Build Coastguard Worker * Licensed under the Apache License, Version 2.0 (the "License");
5*890232f2SAndroid Build Coastguard Worker * you may not use this file except in compliance with the License.
6*890232f2SAndroid Build Coastguard Worker * You may obtain a copy of the License at
7*890232f2SAndroid Build Coastguard Worker *
8*890232f2SAndroid Build Coastguard Worker * http://www.apache.org/licenses/LICENSE-2.0
9*890232f2SAndroid Build Coastguard Worker *
10*890232f2SAndroid Build Coastguard Worker * Unless required by applicable law or agreed to in writing, software
11*890232f2SAndroid Build Coastguard Worker * distributed under the License is distributed on an "AS IS" BASIS,
12*890232f2SAndroid Build Coastguard Worker * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*890232f2SAndroid Build Coastguard Worker * See the License for the specific language governing permissions and
14*890232f2SAndroid Build Coastguard Worker * limitations under the License.
15*890232f2SAndroid Build Coastguard Worker */
16*890232f2SAndroid Build Coastguard Worker
17*890232f2SAndroid Build Coastguard Worker // independent from idl_parser, since this code is not needed for most clients
18*890232f2SAndroid Build Coastguard Worker
19*890232f2SAndroid Build Coastguard Worker #include <sstream>
20*890232f2SAndroid Build Coastguard Worker #include <string>
21*890232f2SAndroid Build Coastguard Worker
22*890232f2SAndroid Build Coastguard Worker #include "flatbuffers/code_generators.h"
23*890232f2SAndroid Build Coastguard Worker #include "flatbuffers/flatbuffers.h"
24*890232f2SAndroid Build Coastguard Worker #include "flatbuffers/idl.h"
25*890232f2SAndroid Build Coastguard Worker #include "flatbuffers/util.h"
26*890232f2SAndroid Build Coastguard Worker #include "idl_namer.h"
27*890232f2SAndroid Build Coastguard Worker
28*890232f2SAndroid Build Coastguard Worker #ifdef _WIN32
29*890232f2SAndroid Build Coastguard Worker # include <direct.h>
30*890232f2SAndroid Build Coastguard Worker # define PATH_SEPARATOR "\\"
31*890232f2SAndroid Build Coastguard Worker # define mkdir(n, m) _mkdir(n)
32*890232f2SAndroid Build Coastguard Worker #else
33*890232f2SAndroid Build Coastguard Worker # include <sys/stat.h>
34*890232f2SAndroid Build Coastguard Worker # define PATH_SEPARATOR "/"
35*890232f2SAndroid Build Coastguard Worker #endif
36*890232f2SAndroid Build Coastguard Worker
37*890232f2SAndroid Build Coastguard Worker namespace flatbuffers {
38*890232f2SAndroid Build Coastguard Worker
39*890232f2SAndroid Build Coastguard Worker namespace go {
40*890232f2SAndroid Build Coastguard Worker
41*890232f2SAndroid Build Coastguard Worker namespace {
42*890232f2SAndroid Build Coastguard Worker
43*890232f2SAndroid Build Coastguard Worker // see https://golang.org/ref/spec#Keywords
GoKeywords()44*890232f2SAndroid Build Coastguard Worker static std::set<std::string> GoKeywords() {
45*890232f2SAndroid Build Coastguard Worker return {
46*890232f2SAndroid Build Coastguard Worker "break", "default", "func", "interface", "select",
47*890232f2SAndroid Build Coastguard Worker "case", "defer", "go", "map", "struct",
48*890232f2SAndroid Build Coastguard Worker "chan", "else", "goto", "package", "switch",
49*890232f2SAndroid Build Coastguard Worker "const", "fallthrough", "if", "range", "type",
50*890232f2SAndroid Build Coastguard Worker "continue", "for", "import", "return", "var",
51*890232f2SAndroid Build Coastguard Worker };
52*890232f2SAndroid Build Coastguard Worker }
53*890232f2SAndroid Build Coastguard Worker
GoDefaultConfig()54*890232f2SAndroid Build Coastguard Worker static Namer::Config GoDefaultConfig() {
55*890232f2SAndroid Build Coastguard Worker // Note that the functions with user defined types in the name use
56*890232f2SAndroid Build Coastguard Worker // upper camel case for all but the user defined type itself, which is keep
57*890232f2SAndroid Build Coastguard Worker // cased. Despite being a function, we interpret it as a Type.
58*890232f2SAndroid Build Coastguard Worker return { /*types=*/Case::kKeep,
59*890232f2SAndroid Build Coastguard Worker /*constants=*/Case::kUnknown,
60*890232f2SAndroid Build Coastguard Worker /*methods=*/Case::kUpperCamel,
61*890232f2SAndroid Build Coastguard Worker /*functions=*/Case::kUpperCamel,
62*890232f2SAndroid Build Coastguard Worker /*fields=*/Case::kUpperCamel,
63*890232f2SAndroid Build Coastguard Worker /*variables=*/Case::kLowerCamel,
64*890232f2SAndroid Build Coastguard Worker /*variants=*/Case::kKeep,
65*890232f2SAndroid Build Coastguard Worker /*enum_variant_seperator=*/"", // I.e. Concatenate.
66*890232f2SAndroid Build Coastguard Worker /*escape_keywords=*/Namer::Config::Escape::AfterConvertingCase,
67*890232f2SAndroid Build Coastguard Worker /*namespaces=*/Case::kKeep,
68*890232f2SAndroid Build Coastguard Worker /*namespace_seperator=*/"__",
69*890232f2SAndroid Build Coastguard Worker /*object_prefix=*/"",
70*890232f2SAndroid Build Coastguard Worker /*object_suffix=*/"T",
71*890232f2SAndroid Build Coastguard Worker /*keyword_prefix=*/"",
72*890232f2SAndroid Build Coastguard Worker /*keyword_suffix=*/"_",
73*890232f2SAndroid Build Coastguard Worker /*filenames=*/Case::kKeep,
74*890232f2SAndroid Build Coastguard Worker /*directories=*/Case::kKeep,
75*890232f2SAndroid Build Coastguard Worker /*output_path=*/"",
76*890232f2SAndroid Build Coastguard Worker /*filename_suffix=*/"",
77*890232f2SAndroid Build Coastguard Worker /*filename_extension=*/".go" };
78*890232f2SAndroid Build Coastguard Worker }
79*890232f2SAndroid Build Coastguard Worker
80*890232f2SAndroid Build Coastguard Worker } // namespace
81*890232f2SAndroid Build Coastguard Worker
82*890232f2SAndroid Build Coastguard Worker class GoGenerator : public BaseGenerator {
83*890232f2SAndroid Build Coastguard Worker public:
GoGenerator(const Parser & parser,const std::string & path,const std::string & file_name,const std::string & go_namespace)84*890232f2SAndroid Build Coastguard Worker GoGenerator(const Parser &parser, const std::string &path,
85*890232f2SAndroid Build Coastguard Worker const std::string &file_name, const std::string &go_namespace)
86*890232f2SAndroid Build Coastguard Worker : BaseGenerator(parser, path, file_name, "" /* not used*/,
87*890232f2SAndroid Build Coastguard Worker "" /* not used */, "go"),
88*890232f2SAndroid Build Coastguard Worker cur_name_space_(nullptr),
89*890232f2SAndroid Build Coastguard Worker namer_(WithFlagOptions(GoDefaultConfig(), parser.opts, path),
90*890232f2SAndroid Build Coastguard Worker GoKeywords()) {
91*890232f2SAndroid Build Coastguard Worker std::istringstream iss(go_namespace);
92*890232f2SAndroid Build Coastguard Worker std::string component;
93*890232f2SAndroid Build Coastguard Worker while (std::getline(iss, component, '.')) {
94*890232f2SAndroid Build Coastguard Worker go_namespace_.components.push_back(component);
95*890232f2SAndroid Build Coastguard Worker }
96*890232f2SAndroid Build Coastguard Worker }
97*890232f2SAndroid Build Coastguard Worker
generate()98*890232f2SAndroid Build Coastguard Worker bool generate() {
99*890232f2SAndroid Build Coastguard Worker std::string one_file_code;
100*890232f2SAndroid Build Coastguard Worker bool needs_imports = false;
101*890232f2SAndroid Build Coastguard Worker for (auto it = parser_.enums_.vec.begin(); it != parser_.enums_.vec.end();
102*890232f2SAndroid Build Coastguard Worker ++it) {
103*890232f2SAndroid Build Coastguard Worker tracked_imported_namespaces_.clear();
104*890232f2SAndroid Build Coastguard Worker needs_imports = false;
105*890232f2SAndroid Build Coastguard Worker std::string enumcode;
106*890232f2SAndroid Build Coastguard Worker GenEnum(**it, &enumcode);
107*890232f2SAndroid Build Coastguard Worker if ((*it)->is_union && parser_.opts.generate_object_based_api) {
108*890232f2SAndroid Build Coastguard Worker GenNativeUnion(**it, &enumcode);
109*890232f2SAndroid Build Coastguard Worker GenNativeUnionPack(**it, &enumcode);
110*890232f2SAndroid Build Coastguard Worker GenNativeUnionUnPack(**it, &enumcode);
111*890232f2SAndroid Build Coastguard Worker needs_imports = true;
112*890232f2SAndroid Build Coastguard Worker }
113*890232f2SAndroid Build Coastguard Worker if (parser_.opts.one_file) {
114*890232f2SAndroid Build Coastguard Worker one_file_code += enumcode;
115*890232f2SAndroid Build Coastguard Worker } else {
116*890232f2SAndroid Build Coastguard Worker if (!SaveType(**it, enumcode, needs_imports, true)) return false;
117*890232f2SAndroid Build Coastguard Worker }
118*890232f2SAndroid Build Coastguard Worker }
119*890232f2SAndroid Build Coastguard Worker
120*890232f2SAndroid Build Coastguard Worker for (auto it = parser_.structs_.vec.begin();
121*890232f2SAndroid Build Coastguard Worker it != parser_.structs_.vec.end(); ++it) {
122*890232f2SAndroid Build Coastguard Worker tracked_imported_namespaces_.clear();
123*890232f2SAndroid Build Coastguard Worker std::string declcode;
124*890232f2SAndroid Build Coastguard Worker GenStruct(**it, &declcode);
125*890232f2SAndroid Build Coastguard Worker if (parser_.opts.one_file) {
126*890232f2SAndroid Build Coastguard Worker one_file_code += declcode;
127*890232f2SAndroid Build Coastguard Worker } else {
128*890232f2SAndroid Build Coastguard Worker if (!SaveType(**it, declcode, true, false)) return false;
129*890232f2SAndroid Build Coastguard Worker }
130*890232f2SAndroid Build Coastguard Worker }
131*890232f2SAndroid Build Coastguard Worker
132*890232f2SAndroid Build Coastguard Worker if (parser_.opts.one_file) {
133*890232f2SAndroid Build Coastguard Worker std::string code = "";
134*890232f2SAndroid Build Coastguard Worker const bool is_enum = !parser_.enums_.vec.empty();
135*890232f2SAndroid Build Coastguard Worker BeginFile(LastNamespacePart(go_namespace_), true, is_enum, &code);
136*890232f2SAndroid Build Coastguard Worker code += one_file_code;
137*890232f2SAndroid Build Coastguard Worker const std::string filename =
138*890232f2SAndroid Build Coastguard Worker GeneratedFileName(path_, file_name_, parser_.opts);
139*890232f2SAndroid Build Coastguard Worker return SaveFile(filename.c_str(), code, false);
140*890232f2SAndroid Build Coastguard Worker }
141*890232f2SAndroid Build Coastguard Worker
142*890232f2SAndroid Build Coastguard Worker return true;
143*890232f2SAndroid Build Coastguard Worker }
144*890232f2SAndroid Build Coastguard Worker
145*890232f2SAndroid Build Coastguard Worker private:
146*890232f2SAndroid Build Coastguard Worker Namespace go_namespace_;
147*890232f2SAndroid Build Coastguard Worker Namespace *cur_name_space_;
148*890232f2SAndroid Build Coastguard Worker const IdlNamer namer_;
149*890232f2SAndroid Build Coastguard Worker
150*890232f2SAndroid Build Coastguard Worker struct NamespacePtrLess {
operator ()flatbuffers::go::GoGenerator::NamespacePtrLess151*890232f2SAndroid Build Coastguard Worker bool operator()(const Namespace *a, const Namespace *b) const {
152*890232f2SAndroid Build Coastguard Worker return *a < *b;
153*890232f2SAndroid Build Coastguard Worker }
154*890232f2SAndroid Build Coastguard Worker };
155*890232f2SAndroid Build Coastguard Worker std::set<const Namespace *, NamespacePtrLess> tracked_imported_namespaces_;
156*890232f2SAndroid Build Coastguard Worker
157*890232f2SAndroid Build Coastguard Worker // Most field accessors need to retrieve and test the field offset first,
158*890232f2SAndroid Build Coastguard Worker // this is the prefix code for that.
OffsetPrefix(const FieldDef & field)159*890232f2SAndroid Build Coastguard Worker std::string OffsetPrefix(const FieldDef &field) {
160*890232f2SAndroid Build Coastguard Worker return "{\n\to := flatbuffers.UOffsetT(rcv._tab.Offset(" +
161*890232f2SAndroid Build Coastguard Worker NumToString(field.value.offset) + "))\n\tif o != 0 {\n";
162*890232f2SAndroid Build Coastguard Worker }
163*890232f2SAndroid Build Coastguard Worker
164*890232f2SAndroid Build Coastguard Worker // Begin a class declaration.
BeginClass(const StructDef & struct_def,std::string * code_ptr)165*890232f2SAndroid Build Coastguard Worker void BeginClass(const StructDef &struct_def, std::string *code_ptr) {
166*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
167*890232f2SAndroid Build Coastguard Worker
168*890232f2SAndroid Build Coastguard Worker code += "type " + namer_.Type(struct_def) + " struct {\n\t";
169*890232f2SAndroid Build Coastguard Worker
170*890232f2SAndroid Build Coastguard Worker // _ is reserved in flatbuffers field names, so no chance of name conflict:
171*890232f2SAndroid Build Coastguard Worker code += "_tab ";
172*890232f2SAndroid Build Coastguard Worker code += struct_def.fixed ? "flatbuffers.Struct" : "flatbuffers.Table";
173*890232f2SAndroid Build Coastguard Worker code += "\n}\n\n";
174*890232f2SAndroid Build Coastguard Worker }
175*890232f2SAndroid Build Coastguard Worker
176*890232f2SAndroid Build Coastguard Worker // Construct the name of the type for this enum.
GetEnumTypeName(const EnumDef & enum_def)177*890232f2SAndroid Build Coastguard Worker std::string GetEnumTypeName(const EnumDef &enum_def) {
178*890232f2SAndroid Build Coastguard Worker return WrapInNameSpaceAndTrack(enum_def.defined_namespace,
179*890232f2SAndroid Build Coastguard Worker namer_.Type(enum_def));
180*890232f2SAndroid Build Coastguard Worker }
181*890232f2SAndroid Build Coastguard Worker
182*890232f2SAndroid Build Coastguard Worker // Create a type for the enum values.
GenEnumType(const EnumDef & enum_def,std::string * code_ptr)183*890232f2SAndroid Build Coastguard Worker void GenEnumType(const EnumDef &enum_def, std::string *code_ptr) {
184*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
185*890232f2SAndroid Build Coastguard Worker code += "type " + GetEnumTypeName(enum_def) + " ";
186*890232f2SAndroid Build Coastguard Worker code += GenTypeBasic(enum_def.underlying_type) + "\n\n";
187*890232f2SAndroid Build Coastguard Worker }
188*890232f2SAndroid Build Coastguard Worker
189*890232f2SAndroid Build Coastguard Worker // Begin enum code with a class declaration.
BeginEnum(std::string * code_ptr)190*890232f2SAndroid Build Coastguard Worker void BeginEnum(std::string *code_ptr) {
191*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
192*890232f2SAndroid Build Coastguard Worker code += "const (\n";
193*890232f2SAndroid Build Coastguard Worker }
194*890232f2SAndroid Build Coastguard Worker
195*890232f2SAndroid Build Coastguard Worker // A single enum member.
EnumMember(const EnumDef & enum_def,const EnumVal & ev,size_t max_name_length,std::string * code_ptr)196*890232f2SAndroid Build Coastguard Worker void EnumMember(const EnumDef &enum_def, const EnumVal &ev,
197*890232f2SAndroid Build Coastguard Worker size_t max_name_length, std::string *code_ptr) {
198*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
199*890232f2SAndroid Build Coastguard Worker code += "\t";
200*890232f2SAndroid Build Coastguard Worker code += namer_.EnumVariant(enum_def, ev);
201*890232f2SAndroid Build Coastguard Worker code += " ";
202*890232f2SAndroid Build Coastguard Worker code += std::string(max_name_length - ev.name.length(), ' ');
203*890232f2SAndroid Build Coastguard Worker code += GetEnumTypeName(enum_def);
204*890232f2SAndroid Build Coastguard Worker code += " = ";
205*890232f2SAndroid Build Coastguard Worker code += enum_def.ToString(ev) + "\n";
206*890232f2SAndroid Build Coastguard Worker }
207*890232f2SAndroid Build Coastguard Worker
208*890232f2SAndroid Build Coastguard Worker // End enum code.
EndEnum(std::string * code_ptr)209*890232f2SAndroid Build Coastguard Worker void EndEnum(std::string *code_ptr) {
210*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
211*890232f2SAndroid Build Coastguard Worker code += ")\n\n";
212*890232f2SAndroid Build Coastguard Worker }
213*890232f2SAndroid Build Coastguard Worker
214*890232f2SAndroid Build Coastguard Worker // Begin enum name map.
BeginEnumNames(const EnumDef & enum_def,std::string * code_ptr)215*890232f2SAndroid Build Coastguard Worker void BeginEnumNames(const EnumDef &enum_def, std::string *code_ptr) {
216*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
217*890232f2SAndroid Build Coastguard Worker code += "var EnumNames";
218*890232f2SAndroid Build Coastguard Worker code += enum_def.name;
219*890232f2SAndroid Build Coastguard Worker code += " = map[" + GetEnumTypeName(enum_def) + "]string{\n";
220*890232f2SAndroid Build Coastguard Worker }
221*890232f2SAndroid Build Coastguard Worker
222*890232f2SAndroid Build Coastguard Worker // A single enum name member.
EnumNameMember(const EnumDef & enum_def,const EnumVal & ev,size_t max_name_length,std::string * code_ptr)223*890232f2SAndroid Build Coastguard Worker void EnumNameMember(const EnumDef &enum_def, const EnumVal &ev,
224*890232f2SAndroid Build Coastguard Worker size_t max_name_length, std::string *code_ptr) {
225*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
226*890232f2SAndroid Build Coastguard Worker code += "\t";
227*890232f2SAndroid Build Coastguard Worker code += namer_.EnumVariant(enum_def, ev);
228*890232f2SAndroid Build Coastguard Worker code += ": ";
229*890232f2SAndroid Build Coastguard Worker code += std::string(max_name_length - ev.name.length(), ' ');
230*890232f2SAndroid Build Coastguard Worker code += "\"";
231*890232f2SAndroid Build Coastguard Worker code += ev.name;
232*890232f2SAndroid Build Coastguard Worker code += "\",\n";
233*890232f2SAndroid Build Coastguard Worker }
234*890232f2SAndroid Build Coastguard Worker
235*890232f2SAndroid Build Coastguard Worker // End enum name map.
EndEnumNames(std::string * code_ptr)236*890232f2SAndroid Build Coastguard Worker void EndEnumNames(std::string *code_ptr) {
237*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
238*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
239*890232f2SAndroid Build Coastguard Worker }
240*890232f2SAndroid Build Coastguard Worker
241*890232f2SAndroid Build Coastguard Worker // Generate String() method on enum type.
EnumStringer(const EnumDef & enum_def,std::string * code_ptr)242*890232f2SAndroid Build Coastguard Worker void EnumStringer(const EnumDef &enum_def, std::string *code_ptr) {
243*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
244*890232f2SAndroid Build Coastguard Worker const std::string enum_type = namer_.Type(enum_def);
245*890232f2SAndroid Build Coastguard Worker code += "func (v " + enum_type + ") String() string {\n";
246*890232f2SAndroid Build Coastguard Worker code += "\tif s, ok := EnumNames" + enum_type + "[v]; ok {\n";
247*890232f2SAndroid Build Coastguard Worker code += "\t\treturn s\n";
248*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
249*890232f2SAndroid Build Coastguard Worker code += "\treturn \"" + enum_def.name;
250*890232f2SAndroid Build Coastguard Worker code += "(\" + strconv.FormatInt(int64(v), 10) + \")\"\n";
251*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
252*890232f2SAndroid Build Coastguard Worker }
253*890232f2SAndroid Build Coastguard Worker
254*890232f2SAndroid Build Coastguard Worker // Begin enum value map.
BeginEnumValues(const EnumDef & enum_def,std::string * code_ptr)255*890232f2SAndroid Build Coastguard Worker void BeginEnumValues(const EnumDef &enum_def, std::string *code_ptr) {
256*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
257*890232f2SAndroid Build Coastguard Worker code += "var EnumValues";
258*890232f2SAndroid Build Coastguard Worker code += namer_.Type(enum_def);
259*890232f2SAndroid Build Coastguard Worker code += " = map[string]" + GetEnumTypeName(enum_def) + "{\n";
260*890232f2SAndroid Build Coastguard Worker }
261*890232f2SAndroid Build Coastguard Worker
262*890232f2SAndroid Build Coastguard Worker // A single enum value member.
EnumValueMember(const EnumDef & enum_def,const EnumVal & ev,size_t max_name_length,std::string * code_ptr)263*890232f2SAndroid Build Coastguard Worker void EnumValueMember(const EnumDef &enum_def, const EnumVal &ev,
264*890232f2SAndroid Build Coastguard Worker size_t max_name_length, std::string *code_ptr) {
265*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
266*890232f2SAndroid Build Coastguard Worker code += "\t\"";
267*890232f2SAndroid Build Coastguard Worker code += ev.name;
268*890232f2SAndroid Build Coastguard Worker code += "\": ";
269*890232f2SAndroid Build Coastguard Worker code += std::string(max_name_length - ev.name.length(), ' ');
270*890232f2SAndroid Build Coastguard Worker code += namer_.EnumVariant(enum_def, ev);
271*890232f2SAndroid Build Coastguard Worker code += ",\n";
272*890232f2SAndroid Build Coastguard Worker }
273*890232f2SAndroid Build Coastguard Worker
274*890232f2SAndroid Build Coastguard Worker // End enum value map.
EndEnumValues(std::string * code_ptr)275*890232f2SAndroid Build Coastguard Worker void EndEnumValues(std::string *code_ptr) {
276*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
277*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
278*890232f2SAndroid Build Coastguard Worker }
279*890232f2SAndroid Build Coastguard Worker
280*890232f2SAndroid Build Coastguard Worker // Initialize a new struct or table from existing data.
NewRootTypeFromBuffer(const StructDef & struct_def,std::string * code_ptr)281*890232f2SAndroid Build Coastguard Worker void NewRootTypeFromBuffer(const StructDef &struct_def,
282*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
283*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
284*890232f2SAndroid Build Coastguard Worker const std::string size_prefix[] = { "", "SizePrefixed" };
285*890232f2SAndroid Build Coastguard Worker const std::string struct_type = namer_.Type(struct_def);
286*890232f2SAndroid Build Coastguard Worker
287*890232f2SAndroid Build Coastguard Worker for (int i = 0; i < 2; i++) {
288*890232f2SAndroid Build Coastguard Worker code += "func Get" + size_prefix[i] + "RootAs" + struct_type;
289*890232f2SAndroid Build Coastguard Worker code += "(buf []byte, offset flatbuffers.UOffsetT) ";
290*890232f2SAndroid Build Coastguard Worker code += "*" + struct_type + "";
291*890232f2SAndroid Build Coastguard Worker code += " {\n";
292*890232f2SAndroid Build Coastguard Worker if (i == 0) {
293*890232f2SAndroid Build Coastguard Worker code += "\tn := flatbuffers.GetUOffsetT(buf[offset:])\n";
294*890232f2SAndroid Build Coastguard Worker } else {
295*890232f2SAndroid Build Coastguard Worker code +=
296*890232f2SAndroid Build Coastguard Worker "\tn := "
297*890232f2SAndroid Build Coastguard Worker "flatbuffers.GetUOffsetT(buf[offset+flatbuffers.SizeUint32:])\n";
298*890232f2SAndroid Build Coastguard Worker }
299*890232f2SAndroid Build Coastguard Worker code += "\tx := &" + struct_type + "{}\n";
300*890232f2SAndroid Build Coastguard Worker if (i == 0) {
301*890232f2SAndroid Build Coastguard Worker code += "\tx.Init(buf, n+offset)\n";
302*890232f2SAndroid Build Coastguard Worker } else {
303*890232f2SAndroid Build Coastguard Worker code += "\tx.Init(buf, n+offset+flatbuffers.SizeUint32)\n";
304*890232f2SAndroid Build Coastguard Worker }
305*890232f2SAndroid Build Coastguard Worker code += "\treturn x\n";
306*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
307*890232f2SAndroid Build Coastguard Worker }
308*890232f2SAndroid Build Coastguard Worker }
309*890232f2SAndroid Build Coastguard Worker
310*890232f2SAndroid Build Coastguard Worker // Initialize an existing object with other data, to avoid an allocation.
InitializeExisting(const StructDef & struct_def,std::string * code_ptr)311*890232f2SAndroid Build Coastguard Worker void InitializeExisting(const StructDef &struct_def, std::string *code_ptr) {
312*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
313*890232f2SAndroid Build Coastguard Worker
314*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
315*890232f2SAndroid Build Coastguard Worker code += " Init(buf []byte, i flatbuffers.UOffsetT) ";
316*890232f2SAndroid Build Coastguard Worker code += "{\n";
317*890232f2SAndroid Build Coastguard Worker code += "\trcv._tab.Bytes = buf\n";
318*890232f2SAndroid Build Coastguard Worker code += "\trcv._tab.Pos = i\n";
319*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
320*890232f2SAndroid Build Coastguard Worker }
321*890232f2SAndroid Build Coastguard Worker
322*890232f2SAndroid Build Coastguard Worker // Implement the table accessor
GenTableAccessor(const StructDef & struct_def,std::string * code_ptr)323*890232f2SAndroid Build Coastguard Worker void GenTableAccessor(const StructDef &struct_def, std::string *code_ptr) {
324*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
325*890232f2SAndroid Build Coastguard Worker
326*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
327*890232f2SAndroid Build Coastguard Worker code += " Table() flatbuffers.Table ";
328*890232f2SAndroid Build Coastguard Worker code += "{\n";
329*890232f2SAndroid Build Coastguard Worker
330*890232f2SAndroid Build Coastguard Worker if (struct_def.fixed) {
331*890232f2SAndroid Build Coastguard Worker code += "\treturn rcv._tab.Table\n";
332*890232f2SAndroid Build Coastguard Worker } else {
333*890232f2SAndroid Build Coastguard Worker code += "\treturn rcv._tab\n";
334*890232f2SAndroid Build Coastguard Worker }
335*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
336*890232f2SAndroid Build Coastguard Worker }
337*890232f2SAndroid Build Coastguard Worker
338*890232f2SAndroid Build Coastguard Worker // Get the length of a vector.
GetVectorLen(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)339*890232f2SAndroid Build Coastguard Worker void GetVectorLen(const StructDef &struct_def, const FieldDef &field,
340*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
341*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
342*890232f2SAndroid Build Coastguard Worker
343*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
344*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field) + "Length(";
345*890232f2SAndroid Build Coastguard Worker code += ") int " + OffsetPrefix(field);
346*890232f2SAndroid Build Coastguard Worker code += "\t\treturn rcv._tab.VectorLen(o)\n\t}\n";
347*890232f2SAndroid Build Coastguard Worker code += "\treturn 0\n}\n\n";
348*890232f2SAndroid Build Coastguard Worker }
349*890232f2SAndroid Build Coastguard Worker
350*890232f2SAndroid Build Coastguard Worker // Get a [ubyte] vector as a byte slice.
GetUByteSlice(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)351*890232f2SAndroid Build Coastguard Worker void GetUByteSlice(const StructDef &struct_def, const FieldDef &field,
352*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
353*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
354*890232f2SAndroid Build Coastguard Worker
355*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
356*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field) + "Bytes(";
357*890232f2SAndroid Build Coastguard Worker code += ") []byte " + OffsetPrefix(field);
358*890232f2SAndroid Build Coastguard Worker code += "\t\treturn rcv._tab.ByteVector(o + rcv._tab.Pos)\n\t}\n";
359*890232f2SAndroid Build Coastguard Worker code += "\treturn nil\n}\n\n";
360*890232f2SAndroid Build Coastguard Worker }
361*890232f2SAndroid Build Coastguard Worker
362*890232f2SAndroid Build Coastguard Worker // Get the value of a struct's scalar.
GetScalarFieldOfStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)363*890232f2SAndroid Build Coastguard Worker void GetScalarFieldOfStruct(const StructDef &struct_def,
364*890232f2SAndroid Build Coastguard Worker const FieldDef &field, std::string *code_ptr) {
365*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
366*890232f2SAndroid Build Coastguard Worker std::string getter = GenGetter(field.value.type);
367*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
368*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
369*890232f2SAndroid Build Coastguard Worker code += "() " + TypeName(field) + " {\n";
370*890232f2SAndroid Build Coastguard Worker code += "\treturn " +
371*890232f2SAndroid Build Coastguard Worker CastToEnum(field.value.type,
372*890232f2SAndroid Build Coastguard Worker getter + "(rcv._tab.Pos + flatbuffers.UOffsetT(" +
373*890232f2SAndroid Build Coastguard Worker NumToString(field.value.offset) + "))");
374*890232f2SAndroid Build Coastguard Worker code += "\n}\n";
375*890232f2SAndroid Build Coastguard Worker }
376*890232f2SAndroid Build Coastguard Worker
377*890232f2SAndroid Build Coastguard Worker // Get the value of a table's scalar.
GetScalarFieldOfTable(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)378*890232f2SAndroid Build Coastguard Worker void GetScalarFieldOfTable(const StructDef &struct_def, const FieldDef &field,
379*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
380*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
381*890232f2SAndroid Build Coastguard Worker std::string getter = GenGetter(field.value.type);
382*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
383*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
384*890232f2SAndroid Build Coastguard Worker code += "() " + TypeName(field) + " ";
385*890232f2SAndroid Build Coastguard Worker code += OffsetPrefix(field);
386*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) {
387*890232f2SAndroid Build Coastguard Worker code += "\t\tv := ";
388*890232f2SAndroid Build Coastguard Worker } else {
389*890232f2SAndroid Build Coastguard Worker code += "\t\treturn ";
390*890232f2SAndroid Build Coastguard Worker }
391*890232f2SAndroid Build Coastguard Worker code += CastToEnum(field.value.type, getter + "(o + rcv._tab.Pos)");
392*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) { code += "\n\t\treturn &v"; }
393*890232f2SAndroid Build Coastguard Worker code += "\n\t}\n";
394*890232f2SAndroid Build Coastguard Worker code += "\treturn " + GenConstant(field) + "\n";
395*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
396*890232f2SAndroid Build Coastguard Worker }
397*890232f2SAndroid Build Coastguard Worker
398*890232f2SAndroid Build Coastguard Worker // Get a struct by initializing an existing struct.
399*890232f2SAndroid Build Coastguard Worker // Specific to Struct.
GetStructFieldOfStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)400*890232f2SAndroid Build Coastguard Worker void GetStructFieldOfStruct(const StructDef &struct_def,
401*890232f2SAndroid Build Coastguard Worker const FieldDef &field, std::string *code_ptr) {
402*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
403*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
404*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
405*890232f2SAndroid Build Coastguard Worker code += "(obj *" + TypeName(field);
406*890232f2SAndroid Build Coastguard Worker code += ") *" + TypeName(field);
407*890232f2SAndroid Build Coastguard Worker code += " {\n";
408*890232f2SAndroid Build Coastguard Worker code += "\tif obj == nil {\n";
409*890232f2SAndroid Build Coastguard Worker code += "\t\tobj = new(" + TypeName(field) + ")\n";
410*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
411*890232f2SAndroid Build Coastguard Worker code += "\tobj.Init(rcv._tab.Bytes, rcv._tab.Pos+";
412*890232f2SAndroid Build Coastguard Worker code += NumToString(field.value.offset) + ")";
413*890232f2SAndroid Build Coastguard Worker code += "\n\treturn obj\n";
414*890232f2SAndroid Build Coastguard Worker code += "}\n";
415*890232f2SAndroid Build Coastguard Worker }
416*890232f2SAndroid Build Coastguard Worker
417*890232f2SAndroid Build Coastguard Worker // Get a struct by initializing an existing struct.
418*890232f2SAndroid Build Coastguard Worker // Specific to Table.
GetStructFieldOfTable(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)419*890232f2SAndroid Build Coastguard Worker void GetStructFieldOfTable(const StructDef &struct_def, const FieldDef &field,
420*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
421*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
422*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
423*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
424*890232f2SAndroid Build Coastguard Worker code += "(obj *";
425*890232f2SAndroid Build Coastguard Worker code += TypeName(field);
426*890232f2SAndroid Build Coastguard Worker code += ") *" + TypeName(field) + " " + OffsetPrefix(field);
427*890232f2SAndroid Build Coastguard Worker if (field.value.type.struct_def->fixed) {
428*890232f2SAndroid Build Coastguard Worker code += "\t\tx := o + rcv._tab.Pos\n";
429*890232f2SAndroid Build Coastguard Worker } else {
430*890232f2SAndroid Build Coastguard Worker code += "\t\tx := rcv._tab.Indirect(o + rcv._tab.Pos)\n";
431*890232f2SAndroid Build Coastguard Worker }
432*890232f2SAndroid Build Coastguard Worker code += "\t\tif obj == nil {\n";
433*890232f2SAndroid Build Coastguard Worker code += "\t\t\tobj = new(" + TypeName(field) + ")\n";
434*890232f2SAndroid Build Coastguard Worker code += "\t\t}\n";
435*890232f2SAndroid Build Coastguard Worker code += "\t\tobj.Init(rcv._tab.Bytes, x)\n";
436*890232f2SAndroid Build Coastguard Worker code += "\t\treturn obj\n\t}\n\treturn nil\n";
437*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
438*890232f2SAndroid Build Coastguard Worker }
439*890232f2SAndroid Build Coastguard Worker
440*890232f2SAndroid Build Coastguard Worker // Get the value of a string.
GetStringField(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)441*890232f2SAndroid Build Coastguard Worker void GetStringField(const StructDef &struct_def, const FieldDef &field,
442*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
443*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
444*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
445*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
446*890232f2SAndroid Build Coastguard Worker code += "() " + TypeName(field) + " ";
447*890232f2SAndroid Build Coastguard Worker code += OffsetPrefix(field) + "\t\treturn " + GenGetter(field.value.type);
448*890232f2SAndroid Build Coastguard Worker code += "(o + rcv._tab.Pos)\n\t}\n\treturn nil\n";
449*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
450*890232f2SAndroid Build Coastguard Worker }
451*890232f2SAndroid Build Coastguard Worker
452*890232f2SAndroid Build Coastguard Worker // Get the value of a union from an object.
GetUnionField(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)453*890232f2SAndroid Build Coastguard Worker void GetUnionField(const StructDef &struct_def, const FieldDef &field,
454*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
455*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
456*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
457*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field) + "(";
458*890232f2SAndroid Build Coastguard Worker code += "obj " + GenTypePointer(field.value.type) + ") bool ";
459*890232f2SAndroid Build Coastguard Worker code += OffsetPrefix(field);
460*890232f2SAndroid Build Coastguard Worker code += "\t\t" + GenGetter(field.value.type);
461*890232f2SAndroid Build Coastguard Worker code += "(obj, o)\n\t\treturn true\n\t}\n";
462*890232f2SAndroid Build Coastguard Worker code += "\treturn false\n";
463*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
464*890232f2SAndroid Build Coastguard Worker }
465*890232f2SAndroid Build Coastguard Worker
466*890232f2SAndroid Build Coastguard Worker // Get the value of a vector's struct member.
GetMemberOfVectorOfStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)467*890232f2SAndroid Build Coastguard Worker void GetMemberOfVectorOfStruct(const StructDef &struct_def,
468*890232f2SAndroid Build Coastguard Worker const FieldDef &field, std::string *code_ptr) {
469*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
470*890232f2SAndroid Build Coastguard Worker auto vectortype = field.value.type.VectorType();
471*890232f2SAndroid Build Coastguard Worker
472*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
473*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
474*890232f2SAndroid Build Coastguard Worker code += "(obj *" + TypeName(field);
475*890232f2SAndroid Build Coastguard Worker code += ", j int) bool " + OffsetPrefix(field);
476*890232f2SAndroid Build Coastguard Worker code += "\t\tx := rcv._tab.Vector(o)\n";
477*890232f2SAndroid Build Coastguard Worker code += "\t\tx += flatbuffers.UOffsetT(j) * ";
478*890232f2SAndroid Build Coastguard Worker code += NumToString(InlineSize(vectortype)) + "\n";
479*890232f2SAndroid Build Coastguard Worker if (!(vectortype.struct_def->fixed)) {
480*890232f2SAndroid Build Coastguard Worker code += "\t\tx = rcv._tab.Indirect(x)\n";
481*890232f2SAndroid Build Coastguard Worker }
482*890232f2SAndroid Build Coastguard Worker code += "\t\tobj.Init(rcv._tab.Bytes, x)\n";
483*890232f2SAndroid Build Coastguard Worker code += "\t\treturn true\n\t}\n";
484*890232f2SAndroid Build Coastguard Worker code += "\treturn false\n";
485*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
486*890232f2SAndroid Build Coastguard Worker }
487*890232f2SAndroid Build Coastguard Worker
488*890232f2SAndroid Build Coastguard Worker // Get the value of a vector's non-struct member.
GetMemberOfVectorOfNonStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)489*890232f2SAndroid Build Coastguard Worker void GetMemberOfVectorOfNonStruct(const StructDef &struct_def,
490*890232f2SAndroid Build Coastguard Worker const FieldDef &field,
491*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
492*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
493*890232f2SAndroid Build Coastguard Worker auto vectortype = field.value.type.VectorType();
494*890232f2SAndroid Build Coastguard Worker
495*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
496*890232f2SAndroid Build Coastguard Worker code += " " + namer_.Function(field);
497*890232f2SAndroid Build Coastguard Worker code += "(j int) " + TypeName(field) + " ";
498*890232f2SAndroid Build Coastguard Worker code += OffsetPrefix(field);
499*890232f2SAndroid Build Coastguard Worker code += "\t\ta := rcv._tab.Vector(o)\n";
500*890232f2SAndroid Build Coastguard Worker code += "\t\treturn " +
501*890232f2SAndroid Build Coastguard Worker CastToEnum(field.value.type,
502*890232f2SAndroid Build Coastguard Worker GenGetter(field.value.type) +
503*890232f2SAndroid Build Coastguard Worker "(a + flatbuffers.UOffsetT(j*" +
504*890232f2SAndroid Build Coastguard Worker NumToString(InlineSize(vectortype)) + "))");
505*890232f2SAndroid Build Coastguard Worker code += "\n\t}\n";
506*890232f2SAndroid Build Coastguard Worker if (IsString(vectortype)) {
507*890232f2SAndroid Build Coastguard Worker code += "\treturn nil\n";
508*890232f2SAndroid Build Coastguard Worker } else if (vectortype.base_type == BASE_TYPE_BOOL) {
509*890232f2SAndroid Build Coastguard Worker code += "\treturn false\n";
510*890232f2SAndroid Build Coastguard Worker } else {
511*890232f2SAndroid Build Coastguard Worker code += "\treturn 0\n";
512*890232f2SAndroid Build Coastguard Worker }
513*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
514*890232f2SAndroid Build Coastguard Worker }
515*890232f2SAndroid Build Coastguard Worker
516*890232f2SAndroid Build Coastguard Worker // Begin the creator function signature.
BeginBuilderArgs(const StructDef & struct_def,std::string * code_ptr)517*890232f2SAndroid Build Coastguard Worker void BeginBuilderArgs(const StructDef &struct_def, std::string *code_ptr) {
518*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
519*890232f2SAndroid Build Coastguard Worker
520*890232f2SAndroid Build Coastguard Worker if (code.substr(code.length() - 2) != "\n\n") {
521*890232f2SAndroid Build Coastguard Worker // a previous mutate has not put an extra new line
522*890232f2SAndroid Build Coastguard Worker code += "\n";
523*890232f2SAndroid Build Coastguard Worker }
524*890232f2SAndroid Build Coastguard Worker code += "func Create" + struct_def.name;
525*890232f2SAndroid Build Coastguard Worker code += "(builder *flatbuffers.Builder";
526*890232f2SAndroid Build Coastguard Worker }
527*890232f2SAndroid Build Coastguard Worker
528*890232f2SAndroid Build Coastguard Worker // Recursively generate arguments for a constructor, to deal with nested
529*890232f2SAndroid Build Coastguard Worker // structs.
StructBuilderArgs(const StructDef & struct_def,const char * nameprefix,std::string * code_ptr)530*890232f2SAndroid Build Coastguard Worker void StructBuilderArgs(const StructDef &struct_def, const char *nameprefix,
531*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
532*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
533*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
534*890232f2SAndroid Build Coastguard Worker auto &field = **it;
535*890232f2SAndroid Build Coastguard Worker if (IsStruct(field.value.type)) {
536*890232f2SAndroid Build Coastguard Worker // Generate arguments for a struct inside a struct. To ensure names
537*890232f2SAndroid Build Coastguard Worker // don't clash, and to make it obvious these arguments are constructing
538*890232f2SAndroid Build Coastguard Worker // a nested struct, prefix the name with the field name.
539*890232f2SAndroid Build Coastguard Worker StructBuilderArgs(*field.value.type.struct_def,
540*890232f2SAndroid Build Coastguard Worker (nameprefix + (field.name + "_")).c_str(), code_ptr);
541*890232f2SAndroid Build Coastguard Worker } else {
542*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
543*890232f2SAndroid Build Coastguard Worker code += std::string(", ") + nameprefix;
544*890232f2SAndroid Build Coastguard Worker code += namer_.Variable(field);
545*890232f2SAndroid Build Coastguard Worker code += " " + TypeName(field);
546*890232f2SAndroid Build Coastguard Worker }
547*890232f2SAndroid Build Coastguard Worker }
548*890232f2SAndroid Build Coastguard Worker }
549*890232f2SAndroid Build Coastguard Worker
550*890232f2SAndroid Build Coastguard Worker // End the creator function signature.
EndBuilderArgs(std::string * code_ptr)551*890232f2SAndroid Build Coastguard Worker void EndBuilderArgs(std::string *code_ptr) {
552*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
553*890232f2SAndroid Build Coastguard Worker code += ") flatbuffers.UOffsetT {\n";
554*890232f2SAndroid Build Coastguard Worker }
555*890232f2SAndroid Build Coastguard Worker
556*890232f2SAndroid Build Coastguard Worker // Recursively generate struct construction statements and instert manual
557*890232f2SAndroid Build Coastguard Worker // padding.
StructBuilderBody(const StructDef & struct_def,const char * nameprefix,std::string * code_ptr)558*890232f2SAndroid Build Coastguard Worker void StructBuilderBody(const StructDef &struct_def, const char *nameprefix,
559*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
560*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
561*890232f2SAndroid Build Coastguard Worker code += "\tbuilder.Prep(" + NumToString(struct_def.minalign) + ", ";
562*890232f2SAndroid Build Coastguard Worker code += NumToString(struct_def.bytesize) + ")\n";
563*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.rbegin();
564*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.rend(); ++it) {
565*890232f2SAndroid Build Coastguard Worker auto &field = **it;
566*890232f2SAndroid Build Coastguard Worker if (field.padding)
567*890232f2SAndroid Build Coastguard Worker code += "\tbuilder.Pad(" + NumToString(field.padding) + ")\n";
568*890232f2SAndroid Build Coastguard Worker if (IsStruct(field.value.type)) {
569*890232f2SAndroid Build Coastguard Worker StructBuilderBody(*field.value.type.struct_def,
570*890232f2SAndroid Build Coastguard Worker (nameprefix + (field.name + "_")).c_str(), code_ptr);
571*890232f2SAndroid Build Coastguard Worker } else {
572*890232f2SAndroid Build Coastguard Worker code += "\tbuilder.Prepend" + GenMethod(field) + "(";
573*890232f2SAndroid Build Coastguard Worker code += CastToBaseType(field.value.type,
574*890232f2SAndroid Build Coastguard Worker nameprefix + namer_.Variable(field)) +
575*890232f2SAndroid Build Coastguard Worker ")\n";
576*890232f2SAndroid Build Coastguard Worker }
577*890232f2SAndroid Build Coastguard Worker }
578*890232f2SAndroid Build Coastguard Worker }
579*890232f2SAndroid Build Coastguard Worker
EndBuilderBody(std::string * code_ptr)580*890232f2SAndroid Build Coastguard Worker void EndBuilderBody(std::string *code_ptr) {
581*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
582*890232f2SAndroid Build Coastguard Worker code += "\treturn builder.Offset()\n";
583*890232f2SAndroid Build Coastguard Worker code += "}\n";
584*890232f2SAndroid Build Coastguard Worker }
585*890232f2SAndroid Build Coastguard Worker
586*890232f2SAndroid Build Coastguard Worker // Get the value of a table's starting offset.
GetStartOfTable(const StructDef & struct_def,std::string * code_ptr)587*890232f2SAndroid Build Coastguard Worker void GetStartOfTable(const StructDef &struct_def, std::string *code_ptr) {
588*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
589*890232f2SAndroid Build Coastguard Worker code += "func " + namer_.Type(struct_def) + "Start";
590*890232f2SAndroid Build Coastguard Worker code += "(builder *flatbuffers.Builder) {\n";
591*890232f2SAndroid Build Coastguard Worker code += "\tbuilder.StartObject(";
592*890232f2SAndroid Build Coastguard Worker code += NumToString(struct_def.fields.vec.size());
593*890232f2SAndroid Build Coastguard Worker code += ")\n}\n";
594*890232f2SAndroid Build Coastguard Worker }
595*890232f2SAndroid Build Coastguard Worker
596*890232f2SAndroid Build Coastguard Worker // Set the value of a table's field.
BuildFieldOfTable(const StructDef & struct_def,const FieldDef & field,const size_t offset,std::string * code_ptr)597*890232f2SAndroid Build Coastguard Worker void BuildFieldOfTable(const StructDef &struct_def, const FieldDef &field,
598*890232f2SAndroid Build Coastguard Worker const size_t offset, std::string *code_ptr) {
599*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
600*890232f2SAndroid Build Coastguard Worker const std::string field_var = namer_.Variable(field);
601*890232f2SAndroid Build Coastguard Worker code += "func " + namer_.Type(struct_def) + "Add" + namer_.Function(field);
602*890232f2SAndroid Build Coastguard Worker code += "(builder *flatbuffers.Builder, ";
603*890232f2SAndroid Build Coastguard Worker code += field_var + " ";
604*890232f2SAndroid Build Coastguard Worker if (!IsScalar(field.value.type.base_type) && (!struct_def.fixed)) {
605*890232f2SAndroid Build Coastguard Worker code += "flatbuffers.UOffsetT";
606*890232f2SAndroid Build Coastguard Worker } else {
607*890232f2SAndroid Build Coastguard Worker code += GenTypeGet(field.value.type);
608*890232f2SAndroid Build Coastguard Worker }
609*890232f2SAndroid Build Coastguard Worker code += ") {\n\t";
610*890232f2SAndroid Build Coastguard Worker code += "builder.Prepend";
611*890232f2SAndroid Build Coastguard Worker code += GenMethod(field);
612*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) {
613*890232f2SAndroid Build Coastguard Worker code += "(";
614*890232f2SAndroid Build Coastguard Worker } else {
615*890232f2SAndroid Build Coastguard Worker code += "Slot(" + NumToString(offset) + ", ";
616*890232f2SAndroid Build Coastguard Worker }
617*890232f2SAndroid Build Coastguard Worker if (!IsScalar(field.value.type.base_type) && (!struct_def.fixed)) {
618*890232f2SAndroid Build Coastguard Worker code += "flatbuffers.UOffsetT";
619*890232f2SAndroid Build Coastguard Worker code += "(" + field_var + ")";
620*890232f2SAndroid Build Coastguard Worker } else {
621*890232f2SAndroid Build Coastguard Worker code += CastToBaseType(field.value.type, field_var);
622*890232f2SAndroid Build Coastguard Worker }
623*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) {
624*890232f2SAndroid Build Coastguard Worker code += ")\n";
625*890232f2SAndroid Build Coastguard Worker code += "\tbuilder.Slot(" + NumToString(offset);
626*890232f2SAndroid Build Coastguard Worker } else {
627*890232f2SAndroid Build Coastguard Worker code += ", " + GenConstant(field);
628*890232f2SAndroid Build Coastguard Worker }
629*890232f2SAndroid Build Coastguard Worker code += ")\n";
630*890232f2SAndroid Build Coastguard Worker code += "}\n";
631*890232f2SAndroid Build Coastguard Worker }
632*890232f2SAndroid Build Coastguard Worker
633*890232f2SAndroid Build Coastguard Worker // Set the value of one of the members of a table's vector.
BuildVectorOfTable(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)634*890232f2SAndroid Build Coastguard Worker void BuildVectorOfTable(const StructDef &struct_def, const FieldDef &field,
635*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
636*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
637*890232f2SAndroid Build Coastguard Worker code += "func " + namer_.Type(struct_def) + "Start";
638*890232f2SAndroid Build Coastguard Worker code += namer_.Function(field);
639*890232f2SAndroid Build Coastguard Worker code += "Vector(builder *flatbuffers.Builder, numElems int) ";
640*890232f2SAndroid Build Coastguard Worker code += "flatbuffers.UOffsetT {\n\treturn builder.StartVector(";
641*890232f2SAndroid Build Coastguard Worker auto vector_type = field.value.type.VectorType();
642*890232f2SAndroid Build Coastguard Worker auto alignment = InlineAlignment(vector_type);
643*890232f2SAndroid Build Coastguard Worker auto elem_size = InlineSize(vector_type);
644*890232f2SAndroid Build Coastguard Worker code += NumToString(elem_size);
645*890232f2SAndroid Build Coastguard Worker code += ", numElems, " + NumToString(alignment);
646*890232f2SAndroid Build Coastguard Worker code += ")\n}\n";
647*890232f2SAndroid Build Coastguard Worker }
648*890232f2SAndroid Build Coastguard Worker
649*890232f2SAndroid Build Coastguard Worker // Get the offset of the end of a table.
GetEndOffsetOnTable(const StructDef & struct_def,std::string * code_ptr)650*890232f2SAndroid Build Coastguard Worker void GetEndOffsetOnTable(const StructDef &struct_def, std::string *code_ptr) {
651*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
652*890232f2SAndroid Build Coastguard Worker code += "func " + namer_.Type(struct_def) + "End";
653*890232f2SAndroid Build Coastguard Worker code += "(builder *flatbuffers.Builder) flatbuffers.UOffsetT ";
654*890232f2SAndroid Build Coastguard Worker code += "{\n\treturn builder.EndObject()\n}\n";
655*890232f2SAndroid Build Coastguard Worker }
656*890232f2SAndroid Build Coastguard Worker
657*890232f2SAndroid Build Coastguard Worker // Generate the receiver for function signatures.
GenReceiver(const StructDef & struct_def,std::string * code_ptr)658*890232f2SAndroid Build Coastguard Worker void GenReceiver(const StructDef &struct_def, std::string *code_ptr) {
659*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
660*890232f2SAndroid Build Coastguard Worker code += "func (rcv *" + namer_.Type(struct_def) + ")";
661*890232f2SAndroid Build Coastguard Worker }
662*890232f2SAndroid Build Coastguard Worker
663*890232f2SAndroid Build Coastguard Worker // Generate a struct field getter, conditioned on its child type(s).
GenStructAccessor(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)664*890232f2SAndroid Build Coastguard Worker void GenStructAccessor(const StructDef &struct_def, const FieldDef &field,
665*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
666*890232f2SAndroid Build Coastguard Worker GenComment(field.doc_comment, code_ptr, nullptr, "");
667*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type)) {
668*890232f2SAndroid Build Coastguard Worker if (struct_def.fixed) {
669*890232f2SAndroid Build Coastguard Worker GetScalarFieldOfStruct(struct_def, field, code_ptr);
670*890232f2SAndroid Build Coastguard Worker } else {
671*890232f2SAndroid Build Coastguard Worker GetScalarFieldOfTable(struct_def, field, code_ptr);
672*890232f2SAndroid Build Coastguard Worker }
673*890232f2SAndroid Build Coastguard Worker } else {
674*890232f2SAndroid Build Coastguard Worker switch (field.value.type.base_type) {
675*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_STRUCT:
676*890232f2SAndroid Build Coastguard Worker if (struct_def.fixed) {
677*890232f2SAndroid Build Coastguard Worker GetStructFieldOfStruct(struct_def, field, code_ptr);
678*890232f2SAndroid Build Coastguard Worker } else {
679*890232f2SAndroid Build Coastguard Worker GetStructFieldOfTable(struct_def, field, code_ptr);
680*890232f2SAndroid Build Coastguard Worker }
681*890232f2SAndroid Build Coastguard Worker break;
682*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_STRING:
683*890232f2SAndroid Build Coastguard Worker GetStringField(struct_def, field, code_ptr);
684*890232f2SAndroid Build Coastguard Worker break;
685*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_VECTOR: {
686*890232f2SAndroid Build Coastguard Worker auto vectortype = field.value.type.VectorType();
687*890232f2SAndroid Build Coastguard Worker if (vectortype.base_type == BASE_TYPE_STRUCT) {
688*890232f2SAndroid Build Coastguard Worker GetMemberOfVectorOfStruct(struct_def, field, code_ptr);
689*890232f2SAndroid Build Coastguard Worker } else {
690*890232f2SAndroid Build Coastguard Worker GetMemberOfVectorOfNonStruct(struct_def, field, code_ptr);
691*890232f2SAndroid Build Coastguard Worker }
692*890232f2SAndroid Build Coastguard Worker break;
693*890232f2SAndroid Build Coastguard Worker }
694*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_UNION: GetUnionField(struct_def, field, code_ptr); break;
695*890232f2SAndroid Build Coastguard Worker default: FLATBUFFERS_ASSERT(0);
696*890232f2SAndroid Build Coastguard Worker }
697*890232f2SAndroid Build Coastguard Worker }
698*890232f2SAndroid Build Coastguard Worker if (IsVector(field.value.type)) {
699*890232f2SAndroid Build Coastguard Worker GetVectorLen(struct_def, field, code_ptr);
700*890232f2SAndroid Build Coastguard Worker if (field.value.type.element == BASE_TYPE_UCHAR) {
701*890232f2SAndroid Build Coastguard Worker GetUByteSlice(struct_def, field, code_ptr);
702*890232f2SAndroid Build Coastguard Worker }
703*890232f2SAndroid Build Coastguard Worker }
704*890232f2SAndroid Build Coastguard Worker }
705*890232f2SAndroid Build Coastguard Worker
706*890232f2SAndroid Build Coastguard Worker // Mutate the value of a struct's scalar.
MutateScalarFieldOfStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)707*890232f2SAndroid Build Coastguard Worker void MutateScalarFieldOfStruct(const StructDef &struct_def,
708*890232f2SAndroid Build Coastguard Worker const FieldDef &field, std::string *code_ptr) {
709*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
710*890232f2SAndroid Build Coastguard Worker std::string setter =
711*890232f2SAndroid Build Coastguard Worker "rcv._tab.Mutate" + namer_.Method(GenTypeBasic(field.value.type));
712*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
713*890232f2SAndroid Build Coastguard Worker code += " Mutate" + namer_.Function(field);
714*890232f2SAndroid Build Coastguard Worker code +=
715*890232f2SAndroid Build Coastguard Worker "(n " + GenTypeGet(field.value.type) + ") bool {\n\treturn " + setter;
716*890232f2SAndroid Build Coastguard Worker code += "(rcv._tab.Pos+flatbuffers.UOffsetT(";
717*890232f2SAndroid Build Coastguard Worker code += NumToString(field.value.offset) + "), ";
718*890232f2SAndroid Build Coastguard Worker code += CastToBaseType(field.value.type, "n") + ")\n}\n\n";
719*890232f2SAndroid Build Coastguard Worker }
720*890232f2SAndroid Build Coastguard Worker
721*890232f2SAndroid Build Coastguard Worker // Mutate the value of a table's scalar.
MutateScalarFieldOfTable(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)722*890232f2SAndroid Build Coastguard Worker void MutateScalarFieldOfTable(const StructDef &struct_def,
723*890232f2SAndroid Build Coastguard Worker const FieldDef &field, std::string *code_ptr) {
724*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
725*890232f2SAndroid Build Coastguard Worker std::string setter = "rcv._tab.Mutate" +
726*890232f2SAndroid Build Coastguard Worker namer_.Method(GenTypeBasic(field.value.type)) + "Slot";
727*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
728*890232f2SAndroid Build Coastguard Worker code += " Mutate" + namer_.Function(field);
729*890232f2SAndroid Build Coastguard Worker code += "(n " + GenTypeGet(field.value.type) + ") bool {\n\treturn ";
730*890232f2SAndroid Build Coastguard Worker code += setter + "(" + NumToString(field.value.offset) + ", ";
731*890232f2SAndroid Build Coastguard Worker code += CastToBaseType(field.value.type, "n") + ")\n";
732*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
733*890232f2SAndroid Build Coastguard Worker }
734*890232f2SAndroid Build Coastguard Worker
735*890232f2SAndroid Build Coastguard Worker // Mutate an element of a vector of scalars.
MutateElementOfVectorOfNonStruct(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)736*890232f2SAndroid Build Coastguard Worker void MutateElementOfVectorOfNonStruct(const StructDef &struct_def,
737*890232f2SAndroid Build Coastguard Worker const FieldDef &field,
738*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
739*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
740*890232f2SAndroid Build Coastguard Worker auto vectortype = field.value.type.VectorType();
741*890232f2SAndroid Build Coastguard Worker std::string setter =
742*890232f2SAndroid Build Coastguard Worker "rcv._tab.Mutate" + namer_.Method(GenTypeBasic(vectortype));
743*890232f2SAndroid Build Coastguard Worker GenReceiver(struct_def, code_ptr);
744*890232f2SAndroid Build Coastguard Worker code += " Mutate" + namer_.Function(field);
745*890232f2SAndroid Build Coastguard Worker code += "(j int, n " + TypeName(field) + ") bool ";
746*890232f2SAndroid Build Coastguard Worker code += OffsetPrefix(field);
747*890232f2SAndroid Build Coastguard Worker code += "\t\ta := rcv._tab.Vector(o)\n";
748*890232f2SAndroid Build Coastguard Worker code += "\t\treturn " + setter + "(";
749*890232f2SAndroid Build Coastguard Worker code += "a+flatbuffers.UOffsetT(j*";
750*890232f2SAndroid Build Coastguard Worker code += NumToString(InlineSize(vectortype)) + "), ";
751*890232f2SAndroid Build Coastguard Worker code += CastToBaseType(vectortype, "n") + ")\n";
752*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
753*890232f2SAndroid Build Coastguard Worker code += "\treturn false\n";
754*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
755*890232f2SAndroid Build Coastguard Worker }
756*890232f2SAndroid Build Coastguard Worker
757*890232f2SAndroid Build Coastguard Worker // Generate a struct field setter, conditioned on its child type(s).
GenStructMutator(const StructDef & struct_def,const FieldDef & field,std::string * code_ptr)758*890232f2SAndroid Build Coastguard Worker void GenStructMutator(const StructDef &struct_def, const FieldDef &field,
759*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
760*890232f2SAndroid Build Coastguard Worker GenComment(field.doc_comment, code_ptr, nullptr, "");
761*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type)) {
762*890232f2SAndroid Build Coastguard Worker if (struct_def.fixed) {
763*890232f2SAndroid Build Coastguard Worker MutateScalarFieldOfStruct(struct_def, field, code_ptr);
764*890232f2SAndroid Build Coastguard Worker } else {
765*890232f2SAndroid Build Coastguard Worker MutateScalarFieldOfTable(struct_def, field, code_ptr);
766*890232f2SAndroid Build Coastguard Worker }
767*890232f2SAndroid Build Coastguard Worker } else if (IsVector(field.value.type)) {
768*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.element)) {
769*890232f2SAndroid Build Coastguard Worker MutateElementOfVectorOfNonStruct(struct_def, field, code_ptr);
770*890232f2SAndroid Build Coastguard Worker }
771*890232f2SAndroid Build Coastguard Worker }
772*890232f2SAndroid Build Coastguard Worker }
773*890232f2SAndroid Build Coastguard Worker
774*890232f2SAndroid Build Coastguard Worker // Generate table constructors, conditioned on its members' types.
GenTableBuilders(const StructDef & struct_def,std::string * code_ptr)775*890232f2SAndroid Build Coastguard Worker void GenTableBuilders(const StructDef &struct_def, std::string *code_ptr) {
776*890232f2SAndroid Build Coastguard Worker GetStartOfTable(struct_def, code_ptr);
777*890232f2SAndroid Build Coastguard Worker
778*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
779*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
780*890232f2SAndroid Build Coastguard Worker auto &field = **it;
781*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
782*890232f2SAndroid Build Coastguard Worker
783*890232f2SAndroid Build Coastguard Worker auto offset = it - struct_def.fields.vec.begin();
784*890232f2SAndroid Build Coastguard Worker BuildFieldOfTable(struct_def, field, offset, code_ptr);
785*890232f2SAndroid Build Coastguard Worker if (IsVector(field.value.type)) {
786*890232f2SAndroid Build Coastguard Worker BuildVectorOfTable(struct_def, field, code_ptr);
787*890232f2SAndroid Build Coastguard Worker }
788*890232f2SAndroid Build Coastguard Worker }
789*890232f2SAndroid Build Coastguard Worker
790*890232f2SAndroid Build Coastguard Worker GetEndOffsetOnTable(struct_def, code_ptr);
791*890232f2SAndroid Build Coastguard Worker }
792*890232f2SAndroid Build Coastguard Worker
793*890232f2SAndroid Build Coastguard Worker // Generate struct or table methods.
GenStruct(const StructDef & struct_def,std::string * code_ptr)794*890232f2SAndroid Build Coastguard Worker void GenStruct(const StructDef &struct_def, std::string *code_ptr) {
795*890232f2SAndroid Build Coastguard Worker if (struct_def.generated) return;
796*890232f2SAndroid Build Coastguard Worker
797*890232f2SAndroid Build Coastguard Worker cur_name_space_ = struct_def.defined_namespace;
798*890232f2SAndroid Build Coastguard Worker
799*890232f2SAndroid Build Coastguard Worker GenComment(struct_def.doc_comment, code_ptr, nullptr);
800*890232f2SAndroid Build Coastguard Worker if (parser_.opts.generate_object_based_api) {
801*890232f2SAndroid Build Coastguard Worker GenNativeStruct(struct_def, code_ptr);
802*890232f2SAndroid Build Coastguard Worker }
803*890232f2SAndroid Build Coastguard Worker BeginClass(struct_def, code_ptr);
804*890232f2SAndroid Build Coastguard Worker if (!struct_def.fixed) {
805*890232f2SAndroid Build Coastguard Worker // Generate a special accessor for the table that has been declared as
806*890232f2SAndroid Build Coastguard Worker // the root type.
807*890232f2SAndroid Build Coastguard Worker NewRootTypeFromBuffer(struct_def, code_ptr);
808*890232f2SAndroid Build Coastguard Worker }
809*890232f2SAndroid Build Coastguard Worker // Generate the Init method that sets the field in a pre-existing
810*890232f2SAndroid Build Coastguard Worker // accessor object. This is to allow object reuse.
811*890232f2SAndroid Build Coastguard Worker InitializeExisting(struct_def, code_ptr);
812*890232f2SAndroid Build Coastguard Worker // Generate _tab accessor
813*890232f2SAndroid Build Coastguard Worker GenTableAccessor(struct_def, code_ptr);
814*890232f2SAndroid Build Coastguard Worker
815*890232f2SAndroid Build Coastguard Worker // Generate struct fields accessors
816*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
817*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
818*890232f2SAndroid Build Coastguard Worker auto &field = **it;
819*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
820*890232f2SAndroid Build Coastguard Worker
821*890232f2SAndroid Build Coastguard Worker GenStructAccessor(struct_def, field, code_ptr);
822*890232f2SAndroid Build Coastguard Worker GenStructMutator(struct_def, field, code_ptr);
823*890232f2SAndroid Build Coastguard Worker }
824*890232f2SAndroid Build Coastguard Worker
825*890232f2SAndroid Build Coastguard Worker // Generate builders
826*890232f2SAndroid Build Coastguard Worker if (struct_def.fixed) {
827*890232f2SAndroid Build Coastguard Worker // create a struct constructor function
828*890232f2SAndroid Build Coastguard Worker GenStructBuilder(struct_def, code_ptr);
829*890232f2SAndroid Build Coastguard Worker } else {
830*890232f2SAndroid Build Coastguard Worker // Create a set of functions that allow table construction.
831*890232f2SAndroid Build Coastguard Worker GenTableBuilders(struct_def, code_ptr);
832*890232f2SAndroid Build Coastguard Worker }
833*890232f2SAndroid Build Coastguard Worker }
834*890232f2SAndroid Build Coastguard Worker
GenNativeStruct(const StructDef & struct_def,std::string * code_ptr)835*890232f2SAndroid Build Coastguard Worker void GenNativeStruct(const StructDef &struct_def, std::string *code_ptr) {
836*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
837*890232f2SAndroid Build Coastguard Worker
838*890232f2SAndroid Build Coastguard Worker code += "type " + NativeName(struct_def) + " struct {\n";
839*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
840*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
841*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
842*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
843*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type) &&
844*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def != nullptr &&
845*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def->is_union)
846*890232f2SAndroid Build Coastguard Worker continue;
847*890232f2SAndroid Build Coastguard Worker code += "\t" + namer_.Field(field) + " ";
848*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) { code += "*"; }
849*890232f2SAndroid Build Coastguard Worker code += NativeType(field.value.type) + " `json:\"" + field.name + "\"`" +
850*890232f2SAndroid Build Coastguard Worker "\n";
851*890232f2SAndroid Build Coastguard Worker }
852*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
853*890232f2SAndroid Build Coastguard Worker
854*890232f2SAndroid Build Coastguard Worker if (!struct_def.fixed) {
855*890232f2SAndroid Build Coastguard Worker GenNativeTablePack(struct_def, code_ptr);
856*890232f2SAndroid Build Coastguard Worker GenNativeTableUnPack(struct_def, code_ptr);
857*890232f2SAndroid Build Coastguard Worker } else {
858*890232f2SAndroid Build Coastguard Worker GenNativeStructPack(struct_def, code_ptr);
859*890232f2SAndroid Build Coastguard Worker GenNativeStructUnPack(struct_def, code_ptr);
860*890232f2SAndroid Build Coastguard Worker }
861*890232f2SAndroid Build Coastguard Worker }
862*890232f2SAndroid Build Coastguard Worker
GenNativeUnion(const EnumDef & enum_def,std::string * code_ptr)863*890232f2SAndroid Build Coastguard Worker void GenNativeUnion(const EnumDef &enum_def, std::string *code_ptr) {
864*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
865*890232f2SAndroid Build Coastguard Worker code += "type " + NativeName(enum_def) + " struct {\n";
866*890232f2SAndroid Build Coastguard Worker code += "\tType " + namer_.Type(enum_def) + "\n";
867*890232f2SAndroid Build Coastguard Worker code += "\tValue interface{}\n";
868*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
869*890232f2SAndroid Build Coastguard Worker }
870*890232f2SAndroid Build Coastguard Worker
GenNativeUnionPack(const EnumDef & enum_def,std::string * code_ptr)871*890232f2SAndroid Build Coastguard Worker void GenNativeUnionPack(const EnumDef &enum_def, std::string *code_ptr) {
872*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
873*890232f2SAndroid Build Coastguard Worker code += "func (t *" + NativeName(enum_def) +
874*890232f2SAndroid Build Coastguard Worker ") Pack(builder *flatbuffers.Builder) flatbuffers.UOffsetT {\n";
875*890232f2SAndroid Build Coastguard Worker code += "\tif t == nil {\n\t\treturn 0\n\t}\n";
876*890232f2SAndroid Build Coastguard Worker
877*890232f2SAndroid Build Coastguard Worker code += "\tswitch t.Type {\n";
878*890232f2SAndroid Build Coastguard Worker for (auto it2 = enum_def.Vals().begin(); it2 != enum_def.Vals().end();
879*890232f2SAndroid Build Coastguard Worker ++it2) {
880*890232f2SAndroid Build Coastguard Worker const EnumVal &ev = **it2;
881*890232f2SAndroid Build Coastguard Worker if (ev.IsZero()) continue;
882*890232f2SAndroid Build Coastguard Worker code += "\tcase " + namer_.EnumVariant(enum_def, ev) + ":\n";
883*890232f2SAndroid Build Coastguard Worker code += "\t\treturn t.Value.(" + NativeType(ev.union_type) +
884*890232f2SAndroid Build Coastguard Worker ").Pack(builder)\n";
885*890232f2SAndroid Build Coastguard Worker }
886*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
887*890232f2SAndroid Build Coastguard Worker code += "\treturn 0\n";
888*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
889*890232f2SAndroid Build Coastguard Worker }
890*890232f2SAndroid Build Coastguard Worker
GenNativeUnionUnPack(const EnumDef & enum_def,std::string * code_ptr)891*890232f2SAndroid Build Coastguard Worker void GenNativeUnionUnPack(const EnumDef &enum_def, std::string *code_ptr) {
892*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
893*890232f2SAndroid Build Coastguard Worker
894*890232f2SAndroid Build Coastguard Worker code += "func (rcv " + namer_.Type(enum_def) +
895*890232f2SAndroid Build Coastguard Worker ") UnPack(table flatbuffers.Table) *" + NativeName(enum_def) +
896*890232f2SAndroid Build Coastguard Worker " {\n";
897*890232f2SAndroid Build Coastguard Worker code += "\tswitch rcv {\n";
898*890232f2SAndroid Build Coastguard Worker
899*890232f2SAndroid Build Coastguard Worker for (auto it2 = enum_def.Vals().begin(); it2 != enum_def.Vals().end();
900*890232f2SAndroid Build Coastguard Worker ++it2) {
901*890232f2SAndroid Build Coastguard Worker const EnumVal &ev = **it2;
902*890232f2SAndroid Build Coastguard Worker if (ev.IsZero()) continue;
903*890232f2SAndroid Build Coastguard Worker code += "\tcase " + namer_.EnumVariant(enum_def, ev) + ":\n";
904*890232f2SAndroid Build Coastguard Worker code += "\t\tx := " + ev.union_type.struct_def->name + "{_tab: table}\n";
905*890232f2SAndroid Build Coastguard Worker
906*890232f2SAndroid Build Coastguard Worker code += "\t\treturn &" +
907*890232f2SAndroid Build Coastguard Worker WrapInNameSpaceAndTrack(enum_def.defined_namespace,
908*890232f2SAndroid Build Coastguard Worker NativeName(enum_def)) +
909*890232f2SAndroid Build Coastguard Worker "{ Type: " + namer_.EnumVariant(enum_def, ev) +
910*890232f2SAndroid Build Coastguard Worker ", Value: x.UnPack() }\n";
911*890232f2SAndroid Build Coastguard Worker }
912*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
913*890232f2SAndroid Build Coastguard Worker code += "\treturn nil\n";
914*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
915*890232f2SAndroid Build Coastguard Worker }
916*890232f2SAndroid Build Coastguard Worker
GenNativeTablePack(const StructDef & struct_def,std::string * code_ptr)917*890232f2SAndroid Build Coastguard Worker void GenNativeTablePack(const StructDef &struct_def, std::string *code_ptr) {
918*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
919*890232f2SAndroid Build Coastguard Worker const std::string struct_type = namer_.Type(struct_def);
920*890232f2SAndroid Build Coastguard Worker
921*890232f2SAndroid Build Coastguard Worker code += "func (t *" + NativeName(struct_def) +
922*890232f2SAndroid Build Coastguard Worker ") Pack(builder *flatbuffers.Builder) flatbuffers.UOffsetT {\n";
923*890232f2SAndroid Build Coastguard Worker code += "\tif t == nil { return 0 }\n";
924*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
925*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
926*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
927*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
928*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type)) continue;
929*890232f2SAndroid Build Coastguard Worker
930*890232f2SAndroid Build Coastguard Worker const std::string field_field = namer_.Field(field);
931*890232f2SAndroid Build Coastguard Worker const std::string field_var = namer_.Variable(field);
932*890232f2SAndroid Build Coastguard Worker const std::string offset = field_var + "Offset";
933*890232f2SAndroid Build Coastguard Worker
934*890232f2SAndroid Build Coastguard Worker if (IsString(field.value.type)) {
935*890232f2SAndroid Build Coastguard Worker code +=
936*890232f2SAndroid Build Coastguard Worker "\t" + offset + " := builder.CreateString(t." + field_field + ")\n";
937*890232f2SAndroid Build Coastguard Worker } else if (IsVector(field.value.type) &&
938*890232f2SAndroid Build Coastguard Worker field.value.type.element == BASE_TYPE_UCHAR &&
939*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def == nullptr) {
940*890232f2SAndroid Build Coastguard Worker code += "\t" + offset + " := flatbuffers.UOffsetT(0)\n";
941*890232f2SAndroid Build Coastguard Worker code += "\tif t." + field_field + " != nil {\n";
942*890232f2SAndroid Build Coastguard Worker code += "\t\t" + offset + " = builder.CreateByteString(t." +
943*890232f2SAndroid Build Coastguard Worker field_field + ")\n";
944*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
945*890232f2SAndroid Build Coastguard Worker } else if (IsVector(field.value.type)) {
946*890232f2SAndroid Build Coastguard Worker code += "\t" + offset + " := flatbuffers.UOffsetT(0)\n";
947*890232f2SAndroid Build Coastguard Worker code += "\tif t." + field_field + " != nil {\n";
948*890232f2SAndroid Build Coastguard Worker std::string length = field_var + "Length";
949*890232f2SAndroid Build Coastguard Worker std::string offsets = field_var + "Offsets";
950*890232f2SAndroid Build Coastguard Worker code += "\t\t" + length + " := len(t." + field_field + ")\n";
951*890232f2SAndroid Build Coastguard Worker if (field.value.type.element == BASE_TYPE_STRING) {
952*890232f2SAndroid Build Coastguard Worker code += "\t\t" + offsets + " := make([]flatbuffers.UOffsetT, " +
953*890232f2SAndroid Build Coastguard Worker length + ")\n";
954*890232f2SAndroid Build Coastguard Worker code += "\t\tfor j := 0; j < " + length + "; j++ {\n";
955*890232f2SAndroid Build Coastguard Worker code += "\t\t\t" + offsets + "[j] = builder.CreateString(t." +
956*890232f2SAndroid Build Coastguard Worker field_field + "[j])\n";
957*890232f2SAndroid Build Coastguard Worker code += "\t\t}\n";
958*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.element == BASE_TYPE_STRUCT &&
959*890232f2SAndroid Build Coastguard Worker !field.value.type.struct_def->fixed) {
960*890232f2SAndroid Build Coastguard Worker code += "\t\t" + offsets + " := make([]flatbuffers.UOffsetT, " +
961*890232f2SAndroid Build Coastguard Worker length + ")\n";
962*890232f2SAndroid Build Coastguard Worker code += "\t\tfor j := 0; j < " + length + "; j++ {\n";
963*890232f2SAndroid Build Coastguard Worker code += "\t\t\t" + offsets + "[j] = t." + field_field +
964*890232f2SAndroid Build Coastguard Worker "[j].Pack(builder)\n";
965*890232f2SAndroid Build Coastguard Worker code += "\t\t}\n";
966*890232f2SAndroid Build Coastguard Worker }
967*890232f2SAndroid Build Coastguard Worker code += "\t\t" + struct_type + "Start" + namer_.Function(field) +
968*890232f2SAndroid Build Coastguard Worker "Vector(builder, " + length + ")\n";
969*890232f2SAndroid Build Coastguard Worker code += "\t\tfor j := " + length + " - 1; j >= 0; j-- {\n";
970*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.element)) {
971*890232f2SAndroid Build Coastguard Worker code += "\t\t\tbuilder.Prepend" +
972*890232f2SAndroid Build Coastguard Worker namer_.Method(GenTypeBasic(field.value.type.VectorType())) +
973*890232f2SAndroid Build Coastguard Worker "(" +
974*890232f2SAndroid Build Coastguard Worker CastToBaseType(field.value.type.VectorType(),
975*890232f2SAndroid Build Coastguard Worker "t." + field_field + "[j]") +
976*890232f2SAndroid Build Coastguard Worker ")\n";
977*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.element == BASE_TYPE_STRUCT &&
978*890232f2SAndroid Build Coastguard Worker field.value.type.struct_def->fixed) {
979*890232f2SAndroid Build Coastguard Worker code += "\t\t\tt." + field_field + "[j].Pack(builder)\n";
980*890232f2SAndroid Build Coastguard Worker } else {
981*890232f2SAndroid Build Coastguard Worker code += "\t\t\tbuilder.PrependUOffsetT(" + offsets + "[j])\n";
982*890232f2SAndroid Build Coastguard Worker }
983*890232f2SAndroid Build Coastguard Worker code += "\t\t}\n";
984*890232f2SAndroid Build Coastguard Worker code += "\t\t" + offset + " = builder.EndVector(" + length + ")\n";
985*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
986*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.base_type == BASE_TYPE_STRUCT) {
987*890232f2SAndroid Build Coastguard Worker if (field.value.type.struct_def->fixed) continue;
988*890232f2SAndroid Build Coastguard Worker code += "\t" + offset + " := t." + field_field + ".Pack(builder)\n";
989*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.base_type == BASE_TYPE_UNION) {
990*890232f2SAndroid Build Coastguard Worker code += "\t" + offset + " := t." + field_field + ".Pack(builder)\n";
991*890232f2SAndroid Build Coastguard Worker code += "\t\n";
992*890232f2SAndroid Build Coastguard Worker } else {
993*890232f2SAndroid Build Coastguard Worker FLATBUFFERS_ASSERT(0);
994*890232f2SAndroid Build Coastguard Worker }
995*890232f2SAndroid Build Coastguard Worker }
996*890232f2SAndroid Build Coastguard Worker code += "\t" + struct_type + "Start(builder)\n";
997*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
998*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
999*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
1000*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
1001*890232f2SAndroid Build Coastguard Worker const std::string field_field = namer_.Field(field);
1002*890232f2SAndroid Build Coastguard Worker const std::string field_fn = namer_.Function(field);
1003*890232f2SAndroid Build Coastguard Worker const std::string offset = namer_.Variable(field) + "Offset";
1004*890232f2SAndroid Build Coastguard Worker
1005*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type)) {
1006*890232f2SAndroid Build Coastguard Worker std::string prefix;
1007*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) {
1008*890232f2SAndroid Build Coastguard Worker code += "\tif t." + field_field + " != nil {\n\t";
1009*890232f2SAndroid Build Coastguard Worker prefix = "*";
1010*890232f2SAndroid Build Coastguard Worker }
1011*890232f2SAndroid Build Coastguard Worker if (field.value.type.enum_def == nullptr ||
1012*890232f2SAndroid Build Coastguard Worker !field.value.type.enum_def->is_union) {
1013*890232f2SAndroid Build Coastguard Worker code += "\t" + struct_type + "Add" + field_fn + "(builder, " +
1014*890232f2SAndroid Build Coastguard Worker prefix + "t." + field_field + ")\n";
1015*890232f2SAndroid Build Coastguard Worker }
1016*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) { code += "\t}\n"; }
1017*890232f2SAndroid Build Coastguard Worker } else {
1018*890232f2SAndroid Build Coastguard Worker if (field.value.type.base_type == BASE_TYPE_STRUCT &&
1019*890232f2SAndroid Build Coastguard Worker field.value.type.struct_def->fixed) {
1020*890232f2SAndroid Build Coastguard Worker code += "\t" + offset + " := t." + field_field + ".Pack(builder)\n";
1021*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.enum_def != nullptr &&
1022*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def->is_union) {
1023*890232f2SAndroid Build Coastguard Worker code += "\tif t." + field_field + " != nil {\n";
1024*890232f2SAndroid Build Coastguard Worker code += "\t\t" + struct_type + "Add" +
1025*890232f2SAndroid Build Coastguard Worker namer_.Method(field.name + UnionTypeFieldSuffix()) +
1026*890232f2SAndroid Build Coastguard Worker "(builder, t." + field_field + ".Type)\n";
1027*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
1028*890232f2SAndroid Build Coastguard Worker }
1029*890232f2SAndroid Build Coastguard Worker code += "\t" + struct_type + "Add" + field_fn + "(builder, " + offset +
1030*890232f2SAndroid Build Coastguard Worker ")\n";
1031*890232f2SAndroid Build Coastguard Worker }
1032*890232f2SAndroid Build Coastguard Worker }
1033*890232f2SAndroid Build Coastguard Worker code += "\treturn " + struct_type + "End(builder)\n";
1034*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
1035*890232f2SAndroid Build Coastguard Worker }
1036*890232f2SAndroid Build Coastguard Worker
GenNativeTableUnPack(const StructDef & struct_def,std::string * code_ptr)1037*890232f2SAndroid Build Coastguard Worker void GenNativeTableUnPack(const StructDef &struct_def,
1038*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
1039*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
1040*890232f2SAndroid Build Coastguard Worker const std::string struct_type = namer_.Type(struct_def);
1041*890232f2SAndroid Build Coastguard Worker
1042*890232f2SAndroid Build Coastguard Worker code += "func (rcv *" + struct_type + ") UnPackTo(t *" +
1043*890232f2SAndroid Build Coastguard Worker NativeName(struct_def) + ") {\n";
1044*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
1045*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
1046*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
1047*890232f2SAndroid Build Coastguard Worker if (field.deprecated) continue;
1048*890232f2SAndroid Build Coastguard Worker const std::string field_field = namer_.Field(field);
1049*890232f2SAndroid Build Coastguard Worker const std::string field_var = namer_.Variable(field);
1050*890232f2SAndroid Build Coastguard Worker const std::string length = field_var + "Length";
1051*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.base_type)) {
1052*890232f2SAndroid Build Coastguard Worker if (field.value.type.enum_def != nullptr &&
1053*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def->is_union)
1054*890232f2SAndroid Build Coastguard Worker continue;
1055*890232f2SAndroid Build Coastguard Worker code += "\tt." + field_field + " = rcv." + field_field + "()\n";
1056*890232f2SAndroid Build Coastguard Worker } else if (IsString(field.value.type)) {
1057*890232f2SAndroid Build Coastguard Worker code += "\tt." + field_field + " = string(rcv." + field_field + "())\n";
1058*890232f2SAndroid Build Coastguard Worker } else if (IsVector(field.value.type) &&
1059*890232f2SAndroid Build Coastguard Worker field.value.type.element == BASE_TYPE_UCHAR &&
1060*890232f2SAndroid Build Coastguard Worker field.value.type.enum_def == nullptr) {
1061*890232f2SAndroid Build Coastguard Worker code += "\tt." + field_field + " = rcv." + field_field + "Bytes()\n";
1062*890232f2SAndroid Build Coastguard Worker } else if (IsVector(field.value.type)) {
1063*890232f2SAndroid Build Coastguard Worker code += "\t" + length + " := rcv." + field_field + "Length()\n";
1064*890232f2SAndroid Build Coastguard Worker code += "\tt." + field_field + " = make(" +
1065*890232f2SAndroid Build Coastguard Worker NativeType(field.value.type) + ", " + length + ")\n";
1066*890232f2SAndroid Build Coastguard Worker code += "\tfor j := 0; j < " + length + "; j++ {\n";
1067*890232f2SAndroid Build Coastguard Worker if (field.value.type.element == BASE_TYPE_STRUCT) {
1068*890232f2SAndroid Build Coastguard Worker code += "\t\tx := " +
1069*890232f2SAndroid Build Coastguard Worker WrapInNameSpaceAndTrack(*field.value.type.struct_def) +
1070*890232f2SAndroid Build Coastguard Worker "{}\n";
1071*890232f2SAndroid Build Coastguard Worker code += "\t\trcv." + field_field + "(&x, j)\n";
1072*890232f2SAndroid Build Coastguard Worker }
1073*890232f2SAndroid Build Coastguard Worker code += "\t\tt." + field_field + "[j] = ";
1074*890232f2SAndroid Build Coastguard Worker if (IsScalar(field.value.type.element)) {
1075*890232f2SAndroid Build Coastguard Worker code += "rcv." + field_field + "(j)";
1076*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.element == BASE_TYPE_STRING) {
1077*890232f2SAndroid Build Coastguard Worker code += "string(rcv." + field_field + "(j))";
1078*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.element == BASE_TYPE_STRUCT) {
1079*890232f2SAndroid Build Coastguard Worker code += "x.UnPack()";
1080*890232f2SAndroid Build Coastguard Worker } else {
1081*890232f2SAndroid Build Coastguard Worker // TODO(iceboy): Support vector of unions.
1082*890232f2SAndroid Build Coastguard Worker FLATBUFFERS_ASSERT(0);
1083*890232f2SAndroid Build Coastguard Worker }
1084*890232f2SAndroid Build Coastguard Worker code += "\n";
1085*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
1086*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.base_type == BASE_TYPE_STRUCT) {
1087*890232f2SAndroid Build Coastguard Worker code +=
1088*890232f2SAndroid Build Coastguard Worker "\tt." + field_field + " = rcv." + field_field + "(nil).UnPack()\n";
1089*890232f2SAndroid Build Coastguard Worker } else if (field.value.type.base_type == BASE_TYPE_UNION) {
1090*890232f2SAndroid Build Coastguard Worker const std::string field_table = field_var + "Table";
1091*890232f2SAndroid Build Coastguard Worker code += "\t" + field_table + " := flatbuffers.Table{}\n";
1092*890232f2SAndroid Build Coastguard Worker code +=
1093*890232f2SAndroid Build Coastguard Worker "\tif rcv." + namer_.Method(field) + "(&" + field_table + ") {\n";
1094*890232f2SAndroid Build Coastguard Worker code += "\t\tt." + field_field + " = rcv." +
1095*890232f2SAndroid Build Coastguard Worker namer_.Method(field.name + UnionTypeFieldSuffix()) +
1096*890232f2SAndroid Build Coastguard Worker "().UnPack(" + field_table + ")\n";
1097*890232f2SAndroid Build Coastguard Worker code += "\t}\n";
1098*890232f2SAndroid Build Coastguard Worker } else {
1099*890232f2SAndroid Build Coastguard Worker FLATBUFFERS_ASSERT(0);
1100*890232f2SAndroid Build Coastguard Worker }
1101*890232f2SAndroid Build Coastguard Worker }
1102*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
1103*890232f2SAndroid Build Coastguard Worker
1104*890232f2SAndroid Build Coastguard Worker code += "func (rcv *" + struct_type + ") UnPack() *" +
1105*890232f2SAndroid Build Coastguard Worker NativeName(struct_def) + " {\n";
1106*890232f2SAndroid Build Coastguard Worker code += "\tif rcv == nil { return nil }\n";
1107*890232f2SAndroid Build Coastguard Worker code += "\tt := &" + NativeName(struct_def) + "{}\n";
1108*890232f2SAndroid Build Coastguard Worker code += "\trcv.UnPackTo(t)\n";
1109*890232f2SAndroid Build Coastguard Worker code += "\treturn t\n";
1110*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
1111*890232f2SAndroid Build Coastguard Worker }
1112*890232f2SAndroid Build Coastguard Worker
GenNativeStructPack(const StructDef & struct_def,std::string * code_ptr)1113*890232f2SAndroid Build Coastguard Worker void GenNativeStructPack(const StructDef &struct_def, std::string *code_ptr) {
1114*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
1115*890232f2SAndroid Build Coastguard Worker
1116*890232f2SAndroid Build Coastguard Worker code += "func (t *" + NativeName(struct_def) +
1117*890232f2SAndroid Build Coastguard Worker ") Pack(builder *flatbuffers.Builder) flatbuffers.UOffsetT {\n";
1118*890232f2SAndroid Build Coastguard Worker code += "\tif t == nil { return 0 }\n";
1119*890232f2SAndroid Build Coastguard Worker code += "\treturn Create" + namer_.Type(struct_def) + "(builder";
1120*890232f2SAndroid Build Coastguard Worker StructPackArgs(struct_def, "", code_ptr);
1121*890232f2SAndroid Build Coastguard Worker code += ")\n";
1122*890232f2SAndroid Build Coastguard Worker code += "}\n";
1123*890232f2SAndroid Build Coastguard Worker }
1124*890232f2SAndroid Build Coastguard Worker
StructPackArgs(const StructDef & struct_def,const char * nameprefix,std::string * code_ptr)1125*890232f2SAndroid Build Coastguard Worker void StructPackArgs(const StructDef &struct_def, const char *nameprefix,
1126*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
1127*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
1128*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
1129*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
1130*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
1131*890232f2SAndroid Build Coastguard Worker if (field.value.type.base_type == BASE_TYPE_STRUCT) {
1132*890232f2SAndroid Build Coastguard Worker StructPackArgs(*field.value.type.struct_def,
1133*890232f2SAndroid Build Coastguard Worker (nameprefix + namer_.Field(field) + ".").c_str(),
1134*890232f2SAndroid Build Coastguard Worker code_ptr);
1135*890232f2SAndroid Build Coastguard Worker } else {
1136*890232f2SAndroid Build Coastguard Worker code += std::string(", t.") + nameprefix + namer_.Field(field);
1137*890232f2SAndroid Build Coastguard Worker }
1138*890232f2SAndroid Build Coastguard Worker }
1139*890232f2SAndroid Build Coastguard Worker }
1140*890232f2SAndroid Build Coastguard Worker
GenNativeStructUnPack(const StructDef & struct_def,std::string * code_ptr)1141*890232f2SAndroid Build Coastguard Worker void GenNativeStructUnPack(const StructDef &struct_def,
1142*890232f2SAndroid Build Coastguard Worker std::string *code_ptr) {
1143*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
1144*890232f2SAndroid Build Coastguard Worker
1145*890232f2SAndroid Build Coastguard Worker code += "func (rcv *" + namer_.Type(struct_def) + ") UnPackTo(t *" +
1146*890232f2SAndroid Build Coastguard Worker NativeName(struct_def) + ") {\n";
1147*890232f2SAndroid Build Coastguard Worker for (auto it = struct_def.fields.vec.begin();
1148*890232f2SAndroid Build Coastguard Worker it != struct_def.fields.vec.end(); ++it) {
1149*890232f2SAndroid Build Coastguard Worker const FieldDef &field = **it;
1150*890232f2SAndroid Build Coastguard Worker if (field.value.type.base_type == BASE_TYPE_STRUCT) {
1151*890232f2SAndroid Build Coastguard Worker code += "\tt." + namer_.Field(field) + " = rcv." +
1152*890232f2SAndroid Build Coastguard Worker namer_.Method(field) + "(nil).UnPack()\n";
1153*890232f2SAndroid Build Coastguard Worker } else {
1154*890232f2SAndroid Build Coastguard Worker code += "\tt." + namer_.Field(field) + " = rcv." +
1155*890232f2SAndroid Build Coastguard Worker namer_.Method(field) + "()\n";
1156*890232f2SAndroid Build Coastguard Worker }
1157*890232f2SAndroid Build Coastguard Worker }
1158*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
1159*890232f2SAndroid Build Coastguard Worker
1160*890232f2SAndroid Build Coastguard Worker code += "func (rcv *" + namer_.Type(struct_def) + ") UnPack() *" +
1161*890232f2SAndroid Build Coastguard Worker NativeName(struct_def) + " {\n";
1162*890232f2SAndroid Build Coastguard Worker code += "\tif rcv == nil { return nil }\n";
1163*890232f2SAndroid Build Coastguard Worker code += "\tt := &" + NativeName(struct_def) + "{}\n";
1164*890232f2SAndroid Build Coastguard Worker code += "\trcv.UnPackTo(t)\n";
1165*890232f2SAndroid Build Coastguard Worker code += "\treturn t\n";
1166*890232f2SAndroid Build Coastguard Worker code += "}\n\n";
1167*890232f2SAndroid Build Coastguard Worker }
1168*890232f2SAndroid Build Coastguard Worker
1169*890232f2SAndroid Build Coastguard Worker // Generate enum declarations.
GenEnum(const EnumDef & enum_def,std::string * code_ptr)1170*890232f2SAndroid Build Coastguard Worker void GenEnum(const EnumDef &enum_def, std::string *code_ptr) {
1171*890232f2SAndroid Build Coastguard Worker if (enum_def.generated) return;
1172*890232f2SAndroid Build Coastguard Worker
1173*890232f2SAndroid Build Coastguard Worker auto max_name_length = MaxNameLength(enum_def);
1174*890232f2SAndroid Build Coastguard Worker cur_name_space_ = enum_def.defined_namespace;
1175*890232f2SAndroid Build Coastguard Worker
1176*890232f2SAndroid Build Coastguard Worker GenComment(enum_def.doc_comment, code_ptr, nullptr);
1177*890232f2SAndroid Build Coastguard Worker GenEnumType(enum_def, code_ptr);
1178*890232f2SAndroid Build Coastguard Worker BeginEnum(code_ptr);
1179*890232f2SAndroid Build Coastguard Worker for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1180*890232f2SAndroid Build Coastguard Worker const EnumVal &ev = **it;
1181*890232f2SAndroid Build Coastguard Worker GenComment(ev.doc_comment, code_ptr, nullptr, "\t");
1182*890232f2SAndroid Build Coastguard Worker EnumMember(enum_def, ev, max_name_length, code_ptr);
1183*890232f2SAndroid Build Coastguard Worker }
1184*890232f2SAndroid Build Coastguard Worker EndEnum(code_ptr);
1185*890232f2SAndroid Build Coastguard Worker
1186*890232f2SAndroid Build Coastguard Worker BeginEnumNames(enum_def, code_ptr);
1187*890232f2SAndroid Build Coastguard Worker for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1188*890232f2SAndroid Build Coastguard Worker const EnumVal &ev = **it;
1189*890232f2SAndroid Build Coastguard Worker EnumNameMember(enum_def, ev, max_name_length, code_ptr);
1190*890232f2SAndroid Build Coastguard Worker }
1191*890232f2SAndroid Build Coastguard Worker EndEnumNames(code_ptr);
1192*890232f2SAndroid Build Coastguard Worker
1193*890232f2SAndroid Build Coastguard Worker BeginEnumValues(enum_def, code_ptr);
1194*890232f2SAndroid Build Coastguard Worker for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1195*890232f2SAndroid Build Coastguard Worker auto &ev = **it;
1196*890232f2SAndroid Build Coastguard Worker EnumValueMember(enum_def, ev, max_name_length, code_ptr);
1197*890232f2SAndroid Build Coastguard Worker }
1198*890232f2SAndroid Build Coastguard Worker EndEnumValues(code_ptr);
1199*890232f2SAndroid Build Coastguard Worker
1200*890232f2SAndroid Build Coastguard Worker EnumStringer(enum_def, code_ptr);
1201*890232f2SAndroid Build Coastguard Worker }
1202*890232f2SAndroid Build Coastguard Worker
1203*890232f2SAndroid Build Coastguard Worker // Returns the function name that is able to read a value of the given type.
GenGetter(const Type & type)1204*890232f2SAndroid Build Coastguard Worker std::string GenGetter(const Type &type) {
1205*890232f2SAndroid Build Coastguard Worker switch (type.base_type) {
1206*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_STRING: return "rcv._tab.ByteVector";
1207*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_UNION: return "rcv._tab.Union";
1208*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_VECTOR: return GenGetter(type.VectorType());
1209*890232f2SAndroid Build Coastguard Worker default: return "rcv._tab.Get" + namer_.Function(GenTypeBasic(type));
1210*890232f2SAndroid Build Coastguard Worker }
1211*890232f2SAndroid Build Coastguard Worker }
1212*890232f2SAndroid Build Coastguard Worker
1213*890232f2SAndroid Build Coastguard Worker // Returns the method name for use with add/put calls.
GenMethod(const FieldDef & field)1214*890232f2SAndroid Build Coastguard Worker std::string GenMethod(const FieldDef &field) {
1215*890232f2SAndroid Build Coastguard Worker return IsScalar(field.value.type.base_type)
1216*890232f2SAndroid Build Coastguard Worker ? namer_.Method(GenTypeBasic(field.value.type))
1217*890232f2SAndroid Build Coastguard Worker : (IsStruct(field.value.type) ? "Struct" : "UOffsetT");
1218*890232f2SAndroid Build Coastguard Worker }
1219*890232f2SAndroid Build Coastguard Worker
GenTypeBasic(const Type & type)1220*890232f2SAndroid Build Coastguard Worker std::string GenTypeBasic(const Type &type) {
1221*890232f2SAndroid Build Coastguard Worker // clang-format off
1222*890232f2SAndroid Build Coastguard Worker static const char *ctypename[] = {
1223*890232f2SAndroid Build Coastguard Worker #define FLATBUFFERS_TD(ENUM, IDLTYPE, CTYPE, JTYPE, GTYPE, ...) \
1224*890232f2SAndroid Build Coastguard Worker #GTYPE,
1225*890232f2SAndroid Build Coastguard Worker FLATBUFFERS_GEN_TYPES(FLATBUFFERS_TD)
1226*890232f2SAndroid Build Coastguard Worker #undef FLATBUFFERS_TD
1227*890232f2SAndroid Build Coastguard Worker };
1228*890232f2SAndroid Build Coastguard Worker // clang-format on
1229*890232f2SAndroid Build Coastguard Worker return ctypename[type.base_type];
1230*890232f2SAndroid Build Coastguard Worker }
1231*890232f2SAndroid Build Coastguard Worker
GenTypePointer(const Type & type)1232*890232f2SAndroid Build Coastguard Worker std::string GenTypePointer(const Type &type) {
1233*890232f2SAndroid Build Coastguard Worker switch (type.base_type) {
1234*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_STRING: return "[]byte";
1235*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_VECTOR: return GenTypeGet(type.VectorType());
1236*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_STRUCT: return WrapInNameSpaceAndTrack(*type.struct_def);
1237*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_UNION:
1238*890232f2SAndroid Build Coastguard Worker // fall through
1239*890232f2SAndroid Build Coastguard Worker default: return "*flatbuffers.Table";
1240*890232f2SAndroid Build Coastguard Worker }
1241*890232f2SAndroid Build Coastguard Worker }
1242*890232f2SAndroid Build Coastguard Worker
GenTypeGet(const Type & type)1243*890232f2SAndroid Build Coastguard Worker std::string GenTypeGet(const Type &type) {
1244*890232f2SAndroid Build Coastguard Worker if (type.enum_def != nullptr) { return GetEnumTypeName(*type.enum_def); }
1245*890232f2SAndroid Build Coastguard Worker return IsScalar(type.base_type) ? GenTypeBasic(type) : GenTypePointer(type);
1246*890232f2SAndroid Build Coastguard Worker }
1247*890232f2SAndroid Build Coastguard Worker
TypeName(const FieldDef & field)1248*890232f2SAndroid Build Coastguard Worker std::string TypeName(const FieldDef &field) {
1249*890232f2SAndroid Build Coastguard Worker std::string prefix;
1250*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) { prefix = "*"; }
1251*890232f2SAndroid Build Coastguard Worker return prefix + GenTypeGet(field.value.type);
1252*890232f2SAndroid Build Coastguard Worker }
1253*890232f2SAndroid Build Coastguard Worker
1254*890232f2SAndroid Build Coastguard Worker // If type is an enum, returns value with a cast to the enum type, otherwise
1255*890232f2SAndroid Build Coastguard Worker // returns value as-is.
CastToEnum(const Type & type,std::string value)1256*890232f2SAndroid Build Coastguard Worker std::string CastToEnum(const Type &type, std::string value) {
1257*890232f2SAndroid Build Coastguard Worker if (type.enum_def == nullptr) {
1258*890232f2SAndroid Build Coastguard Worker return value;
1259*890232f2SAndroid Build Coastguard Worker } else {
1260*890232f2SAndroid Build Coastguard Worker return GenTypeGet(type) + "(" + value + ")";
1261*890232f2SAndroid Build Coastguard Worker }
1262*890232f2SAndroid Build Coastguard Worker }
1263*890232f2SAndroid Build Coastguard Worker
1264*890232f2SAndroid Build Coastguard Worker // If type is an enum, returns value with a cast to the enum base type,
1265*890232f2SAndroid Build Coastguard Worker // otherwise returns value as-is.
CastToBaseType(const Type & type,std::string value)1266*890232f2SAndroid Build Coastguard Worker std::string CastToBaseType(const Type &type, std::string value) {
1267*890232f2SAndroid Build Coastguard Worker if (type.enum_def == nullptr) {
1268*890232f2SAndroid Build Coastguard Worker return value;
1269*890232f2SAndroid Build Coastguard Worker } else {
1270*890232f2SAndroid Build Coastguard Worker return GenTypeBasic(type) + "(" + value + ")";
1271*890232f2SAndroid Build Coastguard Worker }
1272*890232f2SAndroid Build Coastguard Worker }
1273*890232f2SAndroid Build Coastguard Worker
GenConstant(const FieldDef & field)1274*890232f2SAndroid Build Coastguard Worker std::string GenConstant(const FieldDef &field) {
1275*890232f2SAndroid Build Coastguard Worker if (field.IsScalarOptional()) { return "nil"; }
1276*890232f2SAndroid Build Coastguard Worker switch (field.value.type.base_type) {
1277*890232f2SAndroid Build Coastguard Worker case BASE_TYPE_BOOL:
1278*890232f2SAndroid Build Coastguard Worker return field.value.constant == "0" ? "false" : "true";
1279*890232f2SAndroid Build Coastguard Worker default: return field.value.constant;
1280*890232f2SAndroid Build Coastguard Worker }
1281*890232f2SAndroid Build Coastguard Worker }
1282*890232f2SAndroid Build Coastguard Worker
NativeName(const StructDef & struct_def) const1283*890232f2SAndroid Build Coastguard Worker std::string NativeName(const StructDef &struct_def) const {
1284*890232f2SAndroid Build Coastguard Worker return namer_.ObjectType(struct_def);
1285*890232f2SAndroid Build Coastguard Worker }
1286*890232f2SAndroid Build Coastguard Worker
NativeName(const EnumDef & enum_def) const1287*890232f2SAndroid Build Coastguard Worker std::string NativeName(const EnumDef &enum_def) const {
1288*890232f2SAndroid Build Coastguard Worker return namer_.ObjectType(enum_def);
1289*890232f2SAndroid Build Coastguard Worker }
1290*890232f2SAndroid Build Coastguard Worker
NativeType(const Type & type)1291*890232f2SAndroid Build Coastguard Worker std::string NativeType(const Type &type) {
1292*890232f2SAndroid Build Coastguard Worker if (IsScalar(type.base_type)) {
1293*890232f2SAndroid Build Coastguard Worker if (type.enum_def == nullptr) {
1294*890232f2SAndroid Build Coastguard Worker return GenTypeBasic(type);
1295*890232f2SAndroid Build Coastguard Worker } else {
1296*890232f2SAndroid Build Coastguard Worker return GetEnumTypeName(*type.enum_def);
1297*890232f2SAndroid Build Coastguard Worker }
1298*890232f2SAndroid Build Coastguard Worker } else if (IsString(type)) {
1299*890232f2SAndroid Build Coastguard Worker return "string";
1300*890232f2SAndroid Build Coastguard Worker } else if (IsVector(type)) {
1301*890232f2SAndroid Build Coastguard Worker return "[]" + NativeType(type.VectorType());
1302*890232f2SAndroid Build Coastguard Worker } else if (type.base_type == BASE_TYPE_STRUCT) {
1303*890232f2SAndroid Build Coastguard Worker return "*" + WrapInNameSpaceAndTrack(type.struct_def->defined_namespace,
1304*890232f2SAndroid Build Coastguard Worker NativeName(*type.struct_def));
1305*890232f2SAndroid Build Coastguard Worker } else if (type.base_type == BASE_TYPE_UNION) {
1306*890232f2SAndroid Build Coastguard Worker return "*" + WrapInNameSpaceAndTrack(type.enum_def->defined_namespace,
1307*890232f2SAndroid Build Coastguard Worker NativeName(*type.enum_def));
1308*890232f2SAndroid Build Coastguard Worker }
1309*890232f2SAndroid Build Coastguard Worker FLATBUFFERS_ASSERT(0);
1310*890232f2SAndroid Build Coastguard Worker return std::string();
1311*890232f2SAndroid Build Coastguard Worker }
1312*890232f2SAndroid Build Coastguard Worker
1313*890232f2SAndroid Build Coastguard Worker // Create a struct with a builder and the struct's arguments.
GenStructBuilder(const StructDef & struct_def,std::string * code_ptr)1314*890232f2SAndroid Build Coastguard Worker void GenStructBuilder(const StructDef &struct_def, std::string *code_ptr) {
1315*890232f2SAndroid Build Coastguard Worker BeginBuilderArgs(struct_def, code_ptr);
1316*890232f2SAndroid Build Coastguard Worker StructBuilderArgs(struct_def, "", code_ptr);
1317*890232f2SAndroid Build Coastguard Worker EndBuilderArgs(code_ptr);
1318*890232f2SAndroid Build Coastguard Worker
1319*890232f2SAndroid Build Coastguard Worker StructBuilderBody(struct_def, "", code_ptr);
1320*890232f2SAndroid Build Coastguard Worker EndBuilderBody(code_ptr);
1321*890232f2SAndroid Build Coastguard Worker }
1322*890232f2SAndroid Build Coastguard Worker // Begin by declaring namespace and imports.
BeginFile(const std::string & name_space_name,const bool needs_imports,const bool is_enum,std::string * code_ptr)1323*890232f2SAndroid Build Coastguard Worker void BeginFile(const std::string &name_space_name, const bool needs_imports,
1324*890232f2SAndroid Build Coastguard Worker const bool is_enum, std::string *code_ptr) {
1325*890232f2SAndroid Build Coastguard Worker std::string &code = *code_ptr;
1326*890232f2SAndroid Build Coastguard Worker code = code +
1327*890232f2SAndroid Build Coastguard Worker "// Code generated by the FlatBuffers compiler. DO NOT EDIT.\n\n";
1328*890232f2SAndroid Build Coastguard Worker code += "package " + name_space_name + "\n\n";
1329*890232f2SAndroid Build Coastguard Worker if (needs_imports) {
1330*890232f2SAndroid Build Coastguard Worker code += "import (\n";
1331*890232f2SAndroid Build Coastguard Worker if (is_enum) { code += "\t\"strconv\"\n\n"; }
1332*890232f2SAndroid Build Coastguard Worker if (!parser_.opts.go_import.empty()) {
1333*890232f2SAndroid Build Coastguard Worker code += "\tflatbuffers \"" + parser_.opts.go_import + "\"\n";
1334*890232f2SAndroid Build Coastguard Worker } else {
1335*890232f2SAndroid Build Coastguard Worker code += "\tflatbuffers \"github.com/google/flatbuffers/go\"\n";
1336*890232f2SAndroid Build Coastguard Worker }
1337*890232f2SAndroid Build Coastguard Worker if (tracked_imported_namespaces_.size() > 0) {
1338*890232f2SAndroid Build Coastguard Worker code += "\n";
1339*890232f2SAndroid Build Coastguard Worker for (auto it = tracked_imported_namespaces_.begin();
1340*890232f2SAndroid Build Coastguard Worker it != tracked_imported_namespaces_.end(); ++it) {
1341*890232f2SAndroid Build Coastguard Worker code += "\t" + NamespaceImportName(*it) + " \"" +
1342*890232f2SAndroid Build Coastguard Worker NamespaceImportPath(*it) + "\"\n";
1343*890232f2SAndroid Build Coastguard Worker }
1344*890232f2SAndroid Build Coastguard Worker }
1345*890232f2SAndroid Build Coastguard Worker code += ")\n\n";
1346*890232f2SAndroid Build Coastguard Worker } else {
1347*890232f2SAndroid Build Coastguard Worker if (is_enum) { code += "import \"strconv\"\n\n"; }
1348*890232f2SAndroid Build Coastguard Worker }
1349*890232f2SAndroid Build Coastguard Worker }
1350*890232f2SAndroid Build Coastguard Worker
1351*890232f2SAndroid Build Coastguard Worker // Save out the generated code for a Go Table type.
SaveType(const Definition & def,const std::string & classcode,const bool needs_imports,const bool is_enum)1352*890232f2SAndroid Build Coastguard Worker bool SaveType(const Definition &def, const std::string &classcode,
1353*890232f2SAndroid Build Coastguard Worker const bool needs_imports, const bool is_enum) {
1354*890232f2SAndroid Build Coastguard Worker if (!classcode.length()) return true;
1355*890232f2SAndroid Build Coastguard Worker
1356*890232f2SAndroid Build Coastguard Worker Namespace &ns = go_namespace_.components.empty() ? *def.defined_namespace
1357*890232f2SAndroid Build Coastguard Worker : go_namespace_;
1358*890232f2SAndroid Build Coastguard Worker std::string code = "";
1359*890232f2SAndroid Build Coastguard Worker BeginFile(LastNamespacePart(ns), needs_imports, is_enum, &code);
1360*890232f2SAndroid Build Coastguard Worker code += classcode;
1361*890232f2SAndroid Build Coastguard Worker // Strip extra newlines at end of file to make it gofmt-clean.
1362*890232f2SAndroid Build Coastguard Worker while (code.length() > 2 && code.substr(code.length() - 2) == "\n\n") {
1363*890232f2SAndroid Build Coastguard Worker code.pop_back();
1364*890232f2SAndroid Build Coastguard Worker }
1365*890232f2SAndroid Build Coastguard Worker std::string directory = namer_.Directories(ns);
1366*890232f2SAndroid Build Coastguard Worker std::string file = namer_.File(def, SkipFile::Suffix);
1367*890232f2SAndroid Build Coastguard Worker EnsureDirExists(directory);
1368*890232f2SAndroid Build Coastguard Worker std::string filename = directory + file;
1369*890232f2SAndroid Build Coastguard Worker return SaveFile(filename.c_str(), code, false);
1370*890232f2SAndroid Build Coastguard Worker }
1371*890232f2SAndroid Build Coastguard Worker
1372*890232f2SAndroid Build Coastguard Worker // Create the full name of the imported namespace (format: A__B__C).
NamespaceImportName(const Namespace * ns) const1373*890232f2SAndroid Build Coastguard Worker std::string NamespaceImportName(const Namespace *ns) const {
1374*890232f2SAndroid Build Coastguard Worker return namer_.Namespace(*ns);
1375*890232f2SAndroid Build Coastguard Worker }
1376*890232f2SAndroid Build Coastguard Worker
1377*890232f2SAndroid Build Coastguard Worker // Create the full path for the imported namespace (format: A/B/C).
NamespaceImportPath(const Namespace * ns) const1378*890232f2SAndroid Build Coastguard Worker std::string NamespaceImportPath(const Namespace *ns) const {
1379*890232f2SAndroid Build Coastguard Worker return namer_.Directories(*ns, SkipDir::OutputPathAndTrailingPathSeparator);
1380*890232f2SAndroid Build Coastguard Worker }
1381*890232f2SAndroid Build Coastguard Worker
1382*890232f2SAndroid Build Coastguard Worker // Ensure that a type is prefixed with its go package import name if it is
1383*890232f2SAndroid Build Coastguard Worker // used outside of its namespace.
WrapInNameSpaceAndTrack(const Namespace * ns,const std::string & name)1384*890232f2SAndroid Build Coastguard Worker std::string WrapInNameSpaceAndTrack(const Namespace *ns,
1385*890232f2SAndroid Build Coastguard Worker const std::string &name) {
1386*890232f2SAndroid Build Coastguard Worker if (CurrentNameSpace() == ns) return name;
1387*890232f2SAndroid Build Coastguard Worker
1388*890232f2SAndroid Build Coastguard Worker tracked_imported_namespaces_.insert(ns);
1389*890232f2SAndroid Build Coastguard Worker return NamespaceImportName(ns) + "." + name;
1390*890232f2SAndroid Build Coastguard Worker }
1391*890232f2SAndroid Build Coastguard Worker
WrapInNameSpaceAndTrack(const Definition & def)1392*890232f2SAndroid Build Coastguard Worker std::string WrapInNameSpaceAndTrack(const Definition &def) {
1393*890232f2SAndroid Build Coastguard Worker return WrapInNameSpaceAndTrack(def.defined_namespace, def.name);
1394*890232f2SAndroid Build Coastguard Worker }
1395*890232f2SAndroid Build Coastguard Worker
CurrentNameSpace() const1396*890232f2SAndroid Build Coastguard Worker const Namespace *CurrentNameSpace() const { return cur_name_space_; }
1397*890232f2SAndroid Build Coastguard Worker
MaxNameLength(const EnumDef & enum_def)1398*890232f2SAndroid Build Coastguard Worker static size_t MaxNameLength(const EnumDef &enum_def) {
1399*890232f2SAndroid Build Coastguard Worker size_t max = 0;
1400*890232f2SAndroid Build Coastguard Worker for (auto it = enum_def.Vals().begin(); it != enum_def.Vals().end(); ++it) {
1401*890232f2SAndroid Build Coastguard Worker max = std::max((*it)->name.length(), max);
1402*890232f2SAndroid Build Coastguard Worker }
1403*890232f2SAndroid Build Coastguard Worker return max;
1404*890232f2SAndroid Build Coastguard Worker }
1405*890232f2SAndroid Build Coastguard Worker };
1406*890232f2SAndroid Build Coastguard Worker } // namespace go
1407*890232f2SAndroid Build Coastguard Worker
GenerateGo(const Parser & parser,const std::string & path,const std::string & file_name)1408*890232f2SAndroid Build Coastguard Worker bool GenerateGo(const Parser &parser, const std::string &path,
1409*890232f2SAndroid Build Coastguard Worker const std::string &file_name) {
1410*890232f2SAndroid Build Coastguard Worker go::GoGenerator generator(parser, path, file_name, parser.opts.go_namespace);
1411*890232f2SAndroid Build Coastguard Worker return generator.generate();
1412*890232f2SAndroid Build Coastguard Worker }
1413*890232f2SAndroid Build Coastguard Worker
1414*890232f2SAndroid Build Coastguard Worker } // namespace flatbuffers
1415