matthias/clang-uml
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Generate enums as aliases

Browse Source
This commit is contained in:
Bartek Kryza
2021-05-22 14:10:50 +02:00
parent 6cd1abcd38
commit e1c6da5f2e
4 changed files with 60 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/puml/class_diagram_generator.h
View File

@@ -124,7 +124,7 @@ public:
}
}
void generate_aliases(const class_ &c, std::ostream &ostr) const
void generate_alias(const class_ &c, std::ostream &ostr) const
{
std::string class_type{"class"};
if (c.is_abstract())
@@ -135,6 +135,13 @@ public:
ostr << "\" as " << c.alias() << std::endl;
}
void generate_alias(const enum_ &e, std::ostream &ostr) const
{
ostr << "enum" << " \"" << e.full_name(m_config.using_namespace);
ostr << "\" as " << e.alias() << std::endl;
}
void generate(const class_ &c, std::ostream &ostr) const
{
std::string class_type{"class"};
@@ -266,7 +273,7 @@ public:
void generate(const enum_ &e, std::ostream &ostr) const
{
ostr << "enum " << ns_relative(m_config.using_namespace, e.name) << " {"
ostr << "enum " << e.alias() << " {"
<< std::endl;
for (const auto &enum_constant : e.constants) {
@@ -325,7 +332,12 @@ public:
if (m_config.should_include_entities("classes")) {
for (const auto &c : m_model.classes) {
generate_aliases(c, ostr);
generate_alias(c, ostr);
ostr << std::endl;
}
for (const auto &e : m_model.enums) {
generate_alias(e, ostr);
ostr << std::endl;
}