def generate(self, out_path):
log(0, "Generating C++ code for " + self.path)
fname = get_cpp_file_paths(self, out_path)
self.generate_header(fname.h)
self.generate_source(fname.cc)
file = open_file(args.cpp_out + "/infra.h")
writeln(file, Templates.infra)
def generate_extend_definition(self, file):
if self.is_extend:
for id, field in self.fields.items():
writeln(file,
"const ::" + field.parent.cpp_extend_namespace() + "::" + field.name + "_t " +
field.parent.cpp_extend_namespace() + "::" +
field.name + "{" + str(id) + "};")
for _, extend in self.extends.items():
extend.generate_extend_definition(file)
for _, msg in self.messages.items():
msg.generate_extend_definition(file)
def generate_extend_declarations(self, file, indent):
assert(self.is_extend)
for id, field in self.fields.items():
# Deal with C++ scopes - namespace-based things are "extern" while class-based
# things are static.
if self.is_file_scope():
prefix = "extern"
else:
prefix = "static"
writeln(file,
prefix + " const struct " + field.name + "_t { int id; } " +
field.name + ";",
indent)
def generate_shortcut_declarations(self, file, indent):
writeln(file, "// Enum: " + self.fq_name, indent)
writeln(file, "using " + self.name() + " = " + self.impl_cpp_type + ";", indent)
for id, value in self.values.items():
writeln(file,
"const static " + self.impl_cpp_type + " " + value + " = "
+ self.impl_cpp_type + "_" + value + ";",
indent)
writeln(file, "")
def generate_implementation_definition(self, file):
if self.is_algebraic and not self.is_container():
writeln(file, cpp_impl_type(self.raw_type) + " " + self.name + " = 0;", 1)
elif self.is_builtin and not self.is_container():
writeln(file, cpp_impl_type(self.raw_type) + " " + self.name + ";", 1)
elif self.is_enum and not self.is_container():
writeln(file, self.cpp_type_ref() + " " + self.name + " = " + \
self.initializer() + ";", 1)
else:
writeln(file, self.cpp_type_ref() + " " + self.name + ";", 1)
def generate_less_check(self, file, indent):
writeln(file, "// [" + str(self.id) + "] " + self.name, indent)
'''
# This may be used to implement equality that takes presence into account
if not self.is_container() and self.is_enum:
writeln(file,
"if (rep_->_Presence.test(" + str(self.id) +
") != arg.rep_->_Presence.test(" + str(self.id) + "))",
indent)
writeln(file, "return arg.rep_->_Presence.test(" + str(self.id) + ");",
indent + 1)
'''
writeln(file,
"if (rep_->" + self.name + " < arg.rep_->" + self.name + ")",
indent)
writeln(file, "return true;", indent + 1)
writeln(file, "")
def generate_forward_declarations(self, file):
# Forward declarations for sub-messages and enums.
forwards = 0
for _, sub_msg in self.messages.items():
writeln(file, "class " + sub_msg.impl_cpp_type + ";")
forwards += sub_msg.generate_forward_declarations(file)
# Forward declarations for the implicitly declared (forward-declared) local messages.
for _, field in self.fields.items():
if field.is_forward_decl:
writeln(file, "class " + "_".join(field.raw_type.split(".")) + ";")
forwards += 1
# Enums
for _, enum in self.enums.items():
writeln(file, "enum " + enum.impl_cpp_type.split("::")[-1] + " : int;")
forwards += 1
return forwards
def generate_header(self, fname):
file = open_file(fname)
writeln(file, "#pragma once\n")
writeln(file, "#include <cstdint>")
writeln(file, "#include <map>")
writeln(file, "#include <memory>")
writeln(file, "#include <string>")
writeln(file, "#include <vector>")
writeln(file, "#include <infra.h>")
writeln(file, "")
# Generate and print forward type declarations bunched by their namespace
self.generate_forward_imported_declarations_header(file)
# Top-level enums.
for _, enum in self.enums.items():
enum.generate_declaration(file)
if len(self.enums.keys()) > 0:
writeln(file, "")
extend_count = self.count_extends()
# Messages.
for _, msg in self.messages.items():
msg.generate_header(file, self.namespace)
# Extension declarations
if len(self.extends) > 0:
writeln(file, "// Extensions")
for _, extend in self.extends.items():
extend.generate_extend_declarations(file, 0)
writeln(file, "")
for ns in reversed(self.namespace.split(".")):
writeln(file, "} // " + ns)
# Extension helpers
if extend_count > 0:
writeln(file, "")
writeln(file, "namespace proto_ng { namespace detail {")
for _, extend in self.extends.items():
extend.generate_extend_helpers(file)
for _, msg in self.messages.items():
msg.generate_extend_helpers(file)
writeln(file, "} }")
writeln(file, "")
# Hashing and equivalence support
for _, msg in self.messages.items():
msg.generate_hasher(file)
msg.generate_equivalence(file)
def generate_accessor_definitions(self, file):
writeln(file, "// [" + str(self.id) + "] " + self.name)
if self.is_builtin and not self.is_container():
writeln(file,
self.cpp_type_ref() + " " \
+ self.parent.impl_cpp_type + "::" + self.name + "() const {")
writeln(file, "return rep_->" + self.name + ";", 1)
writeln(file, "}")
writeln(file,
"void " + self.parent.impl_cpp_type + "::set_" + self.name + \
"(" + self.cpp_type_ref() + " val) {")
writeln(file, "rep_->" + self.name + " = val;", 1)
writeln(file, "rep_->_Presence.set(" + str(self.id) + ");", 1)
writeln(file, "}")
elif self.is_enum and not self.is_container():
writeln(file,
self.cpp_type_ref() + " " \
+ self.parent.impl_cpp_type + "::" + self.name + "() const {")
writeln(file, "return rep_->" + self.name + ";", 1)
writeln(file, "}")
writeln(file,
"void " + self.parent.impl_cpp_type + "::set_" + self.name + \
"(" + self.cpp_type_ref() + " val) {")
writeln(file, "rep_->" + self.name + " = val;", 1)
writeln(file, "rep_->_Presence.set(" + str(self.id) + ");", 1)
writeln(file, "}")
else:
writeln(file,
"const " + self.cpp_type_ref() + "& " + \
self.parent.impl_cpp_type + "::" + self.name + "() const {")
writeln(file, "return rep_->" + self.name + ";", 1)
writeln(file, "}")
writeln(file,
self.cpp_type_ref() + "& " + \
self.parent.impl_cpp_type + "::" + self.name + "() {")
writeln(file, "rep_->_Presence.set(" + str(self.id) + ");", 1)
writeln(file, "return rep_->" + self.name + ";", 1)
writeln(file, "}")
if not self.is_container():
writeln(file,
"void " + self.parent.impl_cpp_type + "::clear_" + self.name + "() {")
if self.is_algebraic:
writeln(file, "rep_->" + self.name + " = 0;", 1)
elif self.is_builtin:
writeln(file, "rep_->" + self.name + ".clear();", 1)
elif self.is_enum:
writeln(file, "rep_->" + self.name + " = " + self.initializer() + ";", 1)
else:
writeln(file, "rep_->" + self.name + ".Clear();", 1)
writeln(file, "rep_->_Presence.reset(" + str(self.id) + ");", 1)
writeln(file, "}")
writeln(file,
"bool " + self.parent.impl_cpp_type + "::has_" + self.name + "() const {")
writeln(file, "return rep_->_Presence.test(" + str(self.id) + ");", 1)
writeln(file, "}")
if self.is_repeated and not args.omit_deprecated:
if self.is_builtin or self.is_enum:
writeln(file,
"/* deprecated */ void " + self.parent.impl_cpp_type + "::add_" + self.name + "(" + \
self.base_cpp_type_ref() + " value) {")
writeln(file, self.name + "().push_back(std::move(value));", 1)
writeln(file, "}")
writeln(file,
"/* deprecated */ " + \
self.base_cpp_type_ref() + " " + self.parent.impl_cpp_type + "::" + \
self.name + "(int idx) const {")
writeln(file, "return " + self.name + "().at(idx);", 1)
writeln(file, "}")
else:
writeln(file,
"/* deprecated */ " + self.base_cpp_type_ref() + "* " + \
self.parent.impl_cpp_type + "::add_" + self.name + "() {")
writeln(file, self.name + "().push_back({});", 1)
writeln(file, "return &" + self.name + "().back();", 1)
writeln(file, "}")
writeln(file,
"/* deprecated */ " + \
"const " + self.base_cpp_type_ref() + "& " + \
self.parent.impl_cpp_type + "::" + self.name + "(int idx) const {")
writeln(file, "return " + self.name + "().at(idx);", 1)
writeln(file, "}")
writeln(file,
"/* deprecated */ " + \
self.base_cpp_type_ref() + "* " + \
self.parent.impl_cpp_type + "::mutable_" + self.name + "(int idx) {")
writeln(file, "return &" + self.name + "().at(idx);", 1)
writeln(file, "}")
writeln(file, "")
def generate_accessor_declarations(self, file, indent):
writeln(file, "// [" + str(self.id) + "] " + self.name, indent)
if self.is_builtin and not self.is_container():
# These accessors take built-in args by value.
writeln(file,
self.cpp_type_ref() + " " + self.name + "() const;",
indent)
writeln(file,
"void set_" + self.name + "(" + self.cpp_type_ref() + ");",
indent)
elif self.is_enum and not self.is_container():
# This one must deal with scopes, but the accessors work as built-ins.
writeln(file,
self.cpp_type_ref() + " " + self.name + "() const;",
indent)
writeln(file,
"void set_" + self.name + "(" + self.cpp_type_ref() + ");",
indent)
else:
# These are sub-messages/containers and, thus, have reference-based accessors.
writeln(file,
"const " + self.cpp_type_ref() + "& " + self.name + "() const;",
indent)
writeln(file,
self.cpp_type_ref() + "& " + self.name + "();",
indent)
if not args.omit_deprecated:
writeln(file,
"/* deprecated */ " + self.cpp_type_ref() + "* mutable_" + self.name + "();",
indent)
if not self.is_container():
writeln(file, "bool has_" + self.name + "() const;", indent)
writeln(file, "void clear_" + self.name + "();", indent)
if self.is_container() and not args.omit_deprecated:
writeln(file,
"/* deprecated */ " + "void clear_" + self.name + "();",
indent)
if self.is_repeated and not args.omit_deprecated:
if self.is_builtin or self.is_enum:
writeln(file,
"/* deprecated */ " + \
"void add_" + self.name + "(" + self.base_cpp_type_ref() + ");",
indent)
writeln(file,
"/* deprecated */ " + \
self.base_cpp_type_ref() + " " + self.name + "(int idx) const;",
indent)
else:
writeln(file,
"/* deprecated */ " + \
self.base_cpp_type_ref() + "* add_" + self.name + "();",
indent)
writeln(file,
"/* deprecated */ " + \
self.base_cpp_type_ref() + "* mutable_" + self.name + "(int idx);",
indent)
writeln(file,
"/* deprecated */ " + \
"const " + self.base_cpp_type_ref() + "& " + self.name + "(int idx) const;",
indent)
writeln(file,
"/* deprecated */ " + "int32_t " + self.name + "_size() const;",
indent)
writeln(file, "")
def generate_extend_helpers(self, file):
# TODO(Oleg): get base types working here.
if not self.resolved_type:
return
writeln(file, "// [" + str(self.id) + "] " + self.name + " : " + self.resolved_type.fq_name)
writeln(file, "template<>")
writeln(file,
"inline int ResolveField(::" + self.parent.cpp_extend_namespace() + "::" +
self.name + "_t) { return " + str(self.id) + "; }")
writeln(file, "template<>")
writeln(file,
"struct Helper<::" + self.parent.cpp_extend_namespace() + "::" + self.name + "_t> {")
writeln(file, "using ref = const ::" + self.resolved_type.fq_cpp_ref() + "&;", 1)
writeln(file, "using ptr = const ::" + self.resolved_type.fq_cpp_ref() + "*;", 1)
writeln(file, "using mutable_ptr = ::" + self.resolved_type.fq_cpp_ref() + "*;", 1)
writeln(file, "};")
writeln(file, "")
def generate_declaration(self, file):
writeln(file, "enum " + self.impl_cpp_type + " : int {")
for id, value in self.values.items():
writeln(file, self.decorate(value) + " = " + str(id) + ",", 1)
writeln(file, "};")
writeln(file, "std::ostream& operator<<(std::ostream&, " + self.impl_cpp_type + ");")
def generate_source(self, file, ns):
# Implementation
writeln(file, "//")
writeln(file, "// " + self.fq_name)
writeln(file, "//")
writeln(file, "struct " + self.impl_cpp_type + "::Representation {")
for id, field in self.fields.items():
field.generate_implementation_definition(file)
writeln(file, "")
if len(self.fields) > 0:
writeln(file,
"std::bitset<" + str(sorted(self.fields.keys())[-1] + 1) + "> _Presence;",
1)
writeln(file, "};\n")
# Construction, copying and assigment
writeln(file, self.impl_cpp_type + "::" + self.impl_cpp_type +
"() : rep_(std::make_unique<Representation>()) {}")
writeln(file,
self.impl_cpp_type + "::" + self.impl_cpp_type +
"(const " + self.impl_cpp_type + "& arg) : rep_(new Representation(*arg.rep_)) {}")
writeln(file, self.impl_cpp_type + "::" + self.impl_cpp_type +
"(" + self.impl_cpp_type + "&&) = default;")
writeln(file,
self.impl_cpp_type + "& " + self.impl_cpp_type + "::operator=(" +
"const " + self.impl_cpp_type + "& arg) { ")
writeln(file, "if (this != &arg) *rep_ = *arg.rep_;", 1)
writeln(file, "return *this;", 1)
writeln(file, "}")
writeln(file, self.impl_cpp_type + "& " + self.impl_cpp_type + "::operator=(" +
self.impl_cpp_type + "&&) = default;")
writeln(file, self.impl_cpp_type + "::~" + self.impl_cpp_type + "() = default;")
writeln(file, "")
writeln(file, "const " + self.impl_cpp_type + "& " + self.impl_cpp_type + "::default_instance() {")
writeln(file, "static " + self.impl_cpp_type + " obj;", 1)
writeln(file, "return obj;", 1)
writeln(file, "}")
writeln(file, "")
writeln(file, "void " + self.impl_cpp_type + "::Clear() {")
writeln(file, "*this = default_instance();", 1)
writeln(file, "rep_->_Presence.reset();", 1)
writeln(file, "}")
writeln(file, "")
# The key comparison operator on which Regular semantics are built
writeln(file,
"bool " + self.impl_cpp_type + "::operator<(const " + self.impl_cpp_type +
"& arg) const {")
for _, field in self.fields.items():
field.generate_less_check(file, 1)
writeln(file, "return false;", 1)
writeln(file, "}")
writeln(file, "")
# Debug helper functions
writeln(file,
"std::string " + self.impl_cpp_type + "::DebugString(std::string prefix) const {")
writeln(file, "std::stringstream ss;", 1)
writeln(file, "")
for _, field in self.fields.items():
field.generate_debug_output(file, 1)
writeln(file, "return ss.str();", 1)
writeln(file, "}")
writeln(file, "")
# Field accessors for the given message
for id, field in self.fields.items():
field.generate_accessor_definitions(file)
# Enums
for _, enum in self.enums.items():
enum.generate_definition(file)
# Sub-messages
#
# The header cannot be generated in the normal/native DFS style as C++ does not
# allow forward declarations Outer::Inner. So, just pre-order DFS to flatten out
# the tree.
for _, sub_msg in self.messages.items():
sub_msg.generate_source(file, ns)
def generate_source(self, fname):
file = open_file(fname)
writeln(file, Templates.impl)
# Include directives. At this point we need every generated type that comes from
# every "import" statement.
writeln(file, "#include <" + self.cpp_include_path() + ">")
for _, file_ast in self.imports.items():
writeln(file, "#include <" + file_ast.cpp_include_path() + ">")
writeln(file, "")
for ns in self.namespace.split("."):
writeln(file, "namespace " + ns + " {")
writeln(file, "")
# Messages
for _, msg in self.messages.items():
msg.generate_source(file, self.namespace)
# Enums
for _, enum in self.enums.items():
enum.generate_definition(file)
for ns in reversed(self.namespace.split(".")):
writeln(file, "} // " + ns)
# Finally generate extension objects. It's easier to declare these in a FQ fashion.
for _, extend in self.extends.items():
extend.generate_extend_definition(file)
for _, msg in self.messages.items():
msg.generate_extend_definition(file)
def generate_debug_output(self, file, indent):
writeln(file, "// " + self.as_string("ns"), indent)
if self.is_repeated:
# This is a vector of something
writeln(file, "for (const auto& entry : rep_->" + self.name + ") {", indent)
if self.is_builtin or self.is_enum:
if self.is_algebraic:
entry = 'entry'
else:
entry = 'Escape(entry)'
writeln(file,
'ss << "' + self.name + ': " << ' + entry + ' << "\\n";',
indent + 1)
else:
writeln(file, 'ss << prefix << "' + self.name + ' {\\n";', indent + 1)
writeln(file, 'ss << entry.DebugString(prefix + " ");', indent + 1)
writeln(file, 'ss << prefix << "}\\n";', indent + 1)
writeln(file, '}', indent)
elif self.is_map:
# This is a map of something
writeln(file, "for (const auto& entry : rep_->" + self.name + ") {", indent)
writeln(file, 'ss << prefix << "' + self.name + ' {\\n";', indent + 1)
writeln(file, 'ss << prefix << " key: " << entry.first << "\\n";', indent + 1)
writeln(file, 'ss << prefix << " value {\\n";', indent + 1)
if self.resolved_type:
writeln(file, 'ss << entry.second.DebugString(prefix + " ");', indent + 1)
else:
writeln(file, "ss << entry.second;", indent + 1)
writeln(file, 'ss << "}\\n";', indent + 1)
writeln(file, "}", indent)
elif self.is_builtin or self.is_enum:
# This is a singular built-in
writeln(file, "if (rep_->_Presence.test(" + str(self.id) + "))",
indent)
if self.is_algebraic or self.is_enum:
value = 'rep_->' + self.name
else:
value = 'Escape(rep_->' + self.name + ')'
writeln(file,
'ss << prefix << "' + self.name + ': " << ' + value + ' << "\\n";',
indent + 1)
else:
# This is a singular sub-message
writeln(file, "if (rep_->_Presence.test(" + str(self.id) + "))",
indent)
writeln(file,
"ss << prefix << \"" + self.name +
": \" << rep_->" + self.name + '.DebugString(prefix + " ");',
indent + 1)
writeln(file, "")
def generate_equivalence(self, file, indent = 0):
with_hashing = False
for _, field in self.fields.items():
for name, value in field.options.items():
if name.find("include_in_equivalence") >= 0:
with_hashing = True
if not with_hashing: return
writeln(file, "namespace std {", indent)
writeln(file, "// Equivalence for " + self.fq_name)
writeln(file, "template<>", indent)
writeln(file, "struct equal_to<" + self.fq_cpp_ref() + "> {", indent)
writeln(file, "using result_type = bool;", indent + 1)
writeln(file, "using first_argument_type = " + self.fq_cpp_ref() + ";", indent + 1)
writeln(file, "using second_argument_type = " + self.fq_cpp_ref() + ";", indent + 1)
writeln(file,
"size_t operator()(const %s& a, const %s& b) const noexcept {" %
(self.fq_cpp_ref(), self.fq_cpp_ref()),
indent + 1)
for _, field in self.fields.items():
for name, value in field.options.items():
if name.find("include_in_equivalence") >= 0:
writeln(file, "if (a." + field.name + "() != b." + field.name + "()) return false;",
indent + 2)
break
writeln(file, "return true;", indent + 2)
writeln(file, "}", indent + 1)
writeln(file, "};", indent)
writeln(file, "} // std", indent)
writeln(file, "")
def generate_hasher(self, file, indent = 0):
with_hashing = False
for _, field in self.fields.items():
for name, value in field.options.items():
if name.find("include_in_hash") >= 0:
with_hashing = True
if not with_hashing: return
writeln(file, "namespace std {", indent)
writeln(file, "// Hashing for " + self.fq_name)
writeln(file, "template<>", indent)
writeln(file, "struct hash<" + self.fq_cpp_ref() + "> {", indent)
writeln(file, "using argument_type = " + self.fq_cpp_ref() + ";", indent + 1)
writeln(file, "using result_type = std::size_t;", indent + 1)
writeln(file, "size_t operator()(const argument_type& arg) const noexcept {", indent + 1)
writeln(file, "size_t hash = 0;", indent + 2)
for _, field in self.fields.items():
for name, value in field.options.items():
if name.find("include_in_hash") >= 0:
writeln(file, "::proto_ng::hash_combine(hash, arg." + field.name + "());",
indent + 2)
break
writeln(file, "return hash;", indent + 2)
writeln(file, "}", indent + 1)
writeln(file, "};", indent)
writeln(file, "} // std", indent)
writeln(file, "")
def generate_header(self, file, ns, indent = 0):
forwards = self.generate_forward_declarations(file)
if forwards > 0:
writeln(file, "")
# Enums
if len(self.enums) > 0:
writeln(file, "// Enums")
for _, enum in self.enums.items():
enum.generate_declaration(file)
if len(self.enums) > 0:
writeln(file, "")
# Start the C++ class.
writeln(file, "class " + self.impl_cpp_type + " {", indent)
writeln(file, "public:", indent)
# Construction and assignment
writeln(file, "// Construction and assignment", indent + 1)
writeln(file, self.impl_cpp_type + "();", indent + 1)
writeln(file, self.impl_cpp_type + "(const " + self.impl_cpp_type + "&);",
indent + 1)
writeln(file, self.impl_cpp_type + "(" + self.impl_cpp_type + "&&);",
indent + 1)
writeln(file, self.impl_cpp_type + "& " + "operator=(const " + \
self.impl_cpp_type + "&);",
indent + 1)
writeln(file, self.impl_cpp_type + "& " + "operator=(" + \
self.impl_cpp_type + "&&);",
indent + 1)
writeln(file, "~" + self.impl_cpp_type + "();", indent + 1)
writeln(file, "")
# Generally accessible, common API
writeln(file, "// Common API", indent + 1)
writeln(file,
"static const " + self.impl_cpp_type + "& " + "default_instance();",
indent + 1)
writeln(file, "void Clear();", indent + 1)
writeln(file, "bool ParseFromString(const std::string& input_data);", indent + 1)
writeln(file, "std::string SerializeAsString() const;", indent + 1)
writeln(file, 'std::string DebugString(std::string prefix = "") const;', indent + 1)
writeln(file, "std::string ShortDebugString() const;", indent + 1)
writeln(file, "")
# Equality
writeln(file, "// This type is Regular and totally ordered.", indent + 1)
writeln(file, "bool operator<(const " + self.impl_cpp_type + "&) const;", indent + 1)
writeln(file,
"friend bool operator>=(const " + self.impl_cpp_type + "& a, " +
"const " + self.impl_cpp_type + "& b) { return !(a < b); }",
indent + 1)
writeln(file,
"friend bool operator>(const " + self.impl_cpp_type + "& a, " +
"const " + self.impl_cpp_type + "& b) { return b < a; }",
indent + 1)
writeln(file,
"friend bool operator<=(const " + self.impl_cpp_type + "& a, " +
"const " + self.impl_cpp_type + "& b) { return !(b > a); }",
indent + 1)
writeln(file,
"friend bool operator==(const " + self.impl_cpp_type + "& a, " +
"const " + self.impl_cpp_type + "& b) { return !(a > b) && !(a < b); }",
indent + 1)
writeln(file,
"friend bool operator!=(const " + self.impl_cpp_type + "& a, " +
"const " + self.impl_cpp_type + "& b) { return !(a == b); }",
indent + 1)
writeln(file, "")
# Extension support
if self.min_extension_id:
writeln(file, "// Extension API (the base type's part)", indent + 1)
writeln(file, "template<class Extension>", indent + 1)
writeln(file, "bool HasExtension(Extension ext) const {", indent + 1)
writeln(file, "return _HasField(::proto_ng::detail::ResolveField(ext));", indent + 2)
writeln(file, "}", indent + 1)
writeln(file, "template<class Extension>", indent + 1)
writeln(file,
"typename ::proto_ng::detail::Helper<Extension>::mutable_ptr " +
"MutableExtension(Extension ext) {",
indent + 1)
writeln(file, "if (!HasExtension(ext)) return nullptr;", indent + 2)
writeln(file,
"return reinterpret_cast<typename ::proto_ng::detail::Helper<Extension>::mutable_ptr>(",
indent + 2)
writeln(file, "_GetField(::proto_ng::detail::ResolveField(ext)));", indent + 3)
writeln(file, "}", indent + 1)
writeln(file, "template<class Extension>", indent + 1)
writeln(file,
"typename ::proto_ng::detail::Helper<Extension>::ref " +
"GetExtension(Extension ext) const {",
indent + 1)
writeln(file, "assert(HasExtension(ext));", indent + 2)
writeln(file,
"auto ptr = reinterpret_cast<typename ::proto_ng::detail::Helper<Extension>::ptr>(",
indent + 2)
writeln(file, "_GetField(::proto_ng::detail::ResolveField(ext)));", indent + 3)
writeln(file,
"return *ptr;", indent + 2)
writeln(file, "}", indent + 1)
writeln(file, "bool _HasField(int id) const { return false; }", indent + 1)
writeln(file, "void* _GetField(int id) { return nullptr; }", indent + 1)
writeln(file, "const void* _GetField(int id) const { return nullptr; }", indent + 1)
writeln(file, "")
# Extensions
if len(self.extends) > 0:
writeln(file, "// Extensions", indent + 1)
for _, extend in self.extends.items():
extend.generate_extend_declarations(file, indent + 1)
writeln(file, "")
# Aliases for sub-messages
if len(self.messages) > 0:
writeln(file, "// Sub-messages", 1)
for _, sub_msg in self.messages.items():
writeln(file,
"using " + sub_msg.name() + " = " + sub_msg.impl_cpp_type + ";",
1)
write_blank_if(file, self.messages)
# Aliases for sub-enums.
for _, enum in self.enums.items():
enum.generate_shortcut_declarations(file, 1)
# Fields
for id, field in self.fields.items():
field.generate_accessor_declarations(file, indent + 1)
# Implementation
writeln(file, " private:", indent)
writeln(file, "struct Representation;", indent + 1)
writeln(file, "std::unique_ptr<Representation> rep_;", indent + 1)
writeln(file, "};\n", indent)
# Sub-messages
#
# The header cannot be generated in the normal/native DFS style as C++ does not
# allow forward declarations Outer::Inner. So, just pre-order DFS to flatten out
# the tree.
for _, sub_msg in self.messages.items():
sub_msg.generate_header(file, ns)
def generate_definition(self, file):
writeln(file, "// Enum: " + self.fq_name)
writeln(file, "std::ostream& operator<<(std::ostream& st, " + self.impl_cpp_type + " val) {")
writeln(file, "switch (val) {", 1)
for id, value in self.values.items():
writeln(file, "case " + self.cpp_value_prefix() + value + ': // [' + str(id) + ']' , 1)
writeln(file, 'st << "' + value + '";', 2)
writeln(file, 'break;', 2)
writeln(file, "default:", 1)
writeln(file, 'st << "<unrecignized>";', 2)
writeln(file, "}", 1)
writeln(file, "return st;", 1)
writeln(file, "}")
writeln(file, "")
#!/usr/bin/env python3
# Import functions that should be used when communicating
# with the device.
# Note that pyserial is needed to use the library.
# See utils.py for more details about this and how to use the functions.
from utils import establish_connection, readln, writeln, user_input
from time import sleep
# Establish a connection
# Edit name of device v v v
serial_device = establish_connection(device="COM15")
sleep(0.5)
# Reset the device to start from the beginning
writeln(serial_device, 'reset')
# Read and print until no further input
while True:
line = readln(serial_device)
if not line:
break
# Write one line of user input to device
writeln(serial_device, user_input())
# Read and print until no further input
while True:
line = readln(serial_device)
if not line:
break
def generate_forward_imported_declarations_header(self, file):
# First build sets of the enums/messages used by this File.
fdecls = {}
for name, file_ast in self.imports.items():
if not file_ast.namespace in fdecls:
fdecls[file_ast.namespace] = set()
file_ast.generate_forward_imported_declarations(
self.imported_type_names,
fdecls[file_ast.namespace])
# Now print them in "imported" batches.
for ns in sorted(fdecls.keys()):
decl_set = fdecls[ns]
writeln(file, "// Forward declarations from " + ns)
for part in ns.split("."):
writeln(file, "namespace " + part + " {")
for decl in sorted(decl_set):
writeln(file, decl)
for part in reversed(ns.split(".")):
writeln(file, "} // " + part)
writeln(file, "")
for ns in self.namespace.split("."):
writeln(file, "namespace " + ns + " {")
writeln(file, "")