123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272 |
- /*
- * Copyright 2014 Google Inc. All rights reserved.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- // independent from idl_parser, since this code is not needed for most clients
- #include "flatbuffers/flatbuffers.h"
- #include "flatbuffers/idl.h"
- #include "flatbuffers/util.h"
- namespace flatbuffers {
- static void GenStruct(const StructDef &struct_def, const Table *table,
- int indent, const GeneratorOptions &opts,
- std::string *_text);
- // If indentation is less than 0, that indicates we don't want any newlines
- // either.
- const char *NewLine(const GeneratorOptions &opts) {
- return opts.indent_step >= 0 ? "\n" : "";
- }
- int Indent(const GeneratorOptions &opts) {
- return std::max(opts.indent_step, 0);
- }
- // Output an identifier with or without quotes depending on strictness.
- void OutputIdentifier(const std::string &name, const GeneratorOptions &opts,
- std::string *_text) {
- std::string &text = *_text;
- if (opts.strict_json) text += "\"";
- text += name;
- if (opts.strict_json) text += "\"";
- }
- // Print (and its template specialization below for pointers) generate text
- // for a single FlatBuffer value into JSON format.
- // The general case for scalars:
- template<typename T> void Print(T val, Type type, int /*indent*/,
- StructDef * /*union_sd*/,
- const GeneratorOptions &opts,
- std::string *_text) {
- std::string &text = *_text;
- if (type.enum_def && opts.output_enum_identifiers) {
- auto enum_val = type.enum_def->ReverseLookup(static_cast<int>(val));
- if (enum_val) {
- OutputIdentifier(enum_val->name, opts, _text);
- return;
- }
- }
- text += NumToString(val);
- }
- // Print a vector a sequence of JSON values, comma separated, wrapped in "[]".
- template<typename T> void PrintVector(const Vector<T> &v, Type type,
- int indent, const GeneratorOptions &opts,
- std::string *_text) {
- std::string &text = *_text;
- text += "[";
- text += NewLine(opts);
- for (uoffset_t i = 0; i < v.size(); i++) {
- if (i) {
- text += ",";
- text += NewLine(opts);
- }
- text.append(indent + Indent(opts), ' ');
- if (IsStruct(type))
- Print(v.GetStructFromOffset(i * type.struct_def->bytesize), type,
- indent + Indent(opts), nullptr, opts, _text);
- else
- Print(v.Get(i), type, indent + Indent(opts), nullptr,
- opts, _text);
- }
- text += NewLine(opts);
- text.append(indent, ' ');
- text += "]";
- }
- static void EscapeString(const String &s, std::string *_text) {
- std::string &text = *_text;
- text += "\"";
- for (uoffset_t i = 0; i < s.size(); i++) {
- char c = s.Get(i);
- switch (c) {
- case '\n': text += "\\n"; break;
- case '\t': text += "\\t"; break;
- case '\r': text += "\\r"; break;
- case '\b': text += "\\b"; break;
- case '\f': text += "\\f"; break;
- case '\"': text += "\\\""; break;
- case '\\': text += "\\\\"; break;
- default:
- if (c >= ' ' && c <= '~') {
- text += c;
- } else {
- // Not printable ASCII data. Let's see if it's valid UTF-8 first:
- const char *utf8 = s.c_str() + i;
- int ucc = FromUTF8(&utf8);
- if (ucc >= 0x80 && ucc <= 0xFFFF) {
- // Parses as Unicode within JSON's \uXXXX range, so use that.
- text += "\\u";
- text += IntToStringHex(ucc, 4);
- // Skip past characters recognized.
- i = static_cast<uoffset_t>(utf8 - s.c_str() - 1);
- } else {
- // It's either unprintable ASCII, arbitrary binary, or Unicode data
- // that doesn't fit \uXXXX, so use \xXX escape code instead.
- text += "\\x";
- text += IntToStringHex(static_cast<uint8_t>(c), 2);
- }
- }
- break;
- }
- }
- text += "\"";
- }
- // Specialization of Print above for pointer types.
- template<> void Print<const void *>(const void *val,
- Type type, int indent,
- StructDef *union_sd,
- const GeneratorOptions &opts,
- std::string *_text) {
- switch (type.base_type) {
- case BASE_TYPE_UNION:
- // If this assert hits, you have an corrupt buffer, a union type field
- // was not present or was out of range.
- assert(union_sd);
- GenStruct(*union_sd,
- reinterpret_cast<const Table *>(val),
- indent,
- opts,
- _text);
- break;
- case BASE_TYPE_STRUCT:
- GenStruct(*type.struct_def,
- reinterpret_cast<const Table *>(val),
- indent,
- opts,
- _text);
- break;
- case BASE_TYPE_STRING: {
- EscapeString(*reinterpret_cast<const String *>(val), _text);
- break;
- }
- case BASE_TYPE_VECTOR:
- type = type.VectorType();
- // Call PrintVector above specifically for each element type:
- switch (type.base_type) {
- #define FLATBUFFERS_TD(ENUM, IDLTYPE, CTYPE, JTYPE, GTYPE, NTYPE) \
- case BASE_TYPE_ ## ENUM: \
- PrintVector<CTYPE>( \
- *reinterpret_cast<const Vector<CTYPE> *>(val), \
- type, indent, opts, _text); break;
- FLATBUFFERS_GEN_TYPES(FLATBUFFERS_TD)
- #undef FLATBUFFERS_TD
- }
- break;
- default: assert(0);
- }
- }
- // Generate text for a scalar field.
- template<typename T> static void GenField(const FieldDef &fd,
- const Table *table, bool fixed,
- const GeneratorOptions &opts,
- int indent,
- std::string *_text) {
- Print(fixed ?
- reinterpret_cast<const Struct *>(table)->GetField<T>(fd.value.offset) :
- table->GetField<T>(fd.value.offset, 0), fd.value.type, indent, nullptr,
- opts, _text);
- }
- // Generate text for non-scalar field.
- static void GenFieldOffset(const FieldDef &fd, const Table *table, bool fixed,
- int indent, StructDef *union_sd,
- const GeneratorOptions &opts, std::string *_text) {
- const void *val = nullptr;
- if (fixed) {
- // The only non-scalar fields in structs are structs.
- assert(IsStruct(fd.value.type));
- val = reinterpret_cast<const Struct *>(table)->
- GetStruct<const void *>(fd.value.offset);
- } else {
- val = IsStruct(fd.value.type)
- ? table->GetStruct<const void *>(fd.value.offset)
- : table->GetPointer<const void *>(fd.value.offset);
- }
- Print(val, fd.value.type, indent, union_sd, opts, _text);
- }
- // Generate text for a struct or table, values separated by commas, indented,
- // and bracketed by "{}"
- static void GenStruct(const StructDef &struct_def, const Table *table,
- int indent, const GeneratorOptions &opts,
- std::string *_text) {
- std::string &text = *_text;
- text += "{";
- int fieldout = 0;
- StructDef *union_sd = nullptr;
- for (auto it = struct_def.fields.vec.begin();
- it != struct_def.fields.vec.end();
- ++it) {
- FieldDef &fd = **it;
- if (struct_def.fixed || table->CheckField(fd.value.offset)) {
- // The field is present.
- if (fieldout++) {
- text += ",";
- }
- text += NewLine(opts);
- text.append(indent + Indent(opts), ' ');
- OutputIdentifier(fd.name, opts, _text);
- text += ": ";
- switch (fd.value.type.base_type) {
- #define FLATBUFFERS_TD(ENUM, IDLTYPE, CTYPE, JTYPE, GTYPE, NTYPE) \
- case BASE_TYPE_ ## ENUM: \
- GenField<CTYPE>(fd, table, struct_def.fixed, \
- opts, indent + Indent(opts), _text); \
- break;
- FLATBUFFERS_GEN_TYPES_SCALAR(FLATBUFFERS_TD)
- #undef FLATBUFFERS_TD
- // Generate drop-thru case statements for all pointer types:
- #define FLATBUFFERS_TD(ENUM, IDLTYPE, CTYPE, JTYPE, GTYPE, NTYPE) \
- case BASE_TYPE_ ## ENUM:
- FLATBUFFERS_GEN_TYPES_POINTER(FLATBUFFERS_TD)
- #undef FLATBUFFERS_TD
- GenFieldOffset(fd, table, struct_def.fixed, indent + Indent(opts),
- union_sd, opts, _text);
- break;
- }
- if (fd.value.type.base_type == BASE_TYPE_UTYPE) {
- auto enum_val = fd.value.type.enum_def->ReverseLookup(
- table->GetField<uint8_t>(fd.value.offset, 0));
- assert(enum_val);
- union_sd = enum_val->struct_def;
- }
- }
- }
- text += NewLine(opts);
- text.append(indent, ' ');
- text += "}";
- }
- // Generate a text representation of a flatbuffer in JSON format.
- void GenerateText(const Parser &parser, const void *flatbuffer,
- const GeneratorOptions &opts, std::string *_text) {
- std::string &text = *_text;
- assert(parser.root_struct_def); // call SetRootType()
- text.reserve(1024); // Reduce amount of inevitable reallocs.
- GenStruct(*parser.root_struct_def,
- GetRoot<Table>(flatbuffer),
- 0,
- opts,
- _text);
- text += NewLine(opts);
- }
- } // namespace flatbuffers
|