xref: /aosp_15_r20/external/flatbuffers/src/idl_gen_go.cpp (revision 890232f25432b36107d06881e0a25aaa6b473652)
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