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

Added enum handling

Browse Source
This commit is contained in:
Bartek Kryza
2021-03-24 11:00:17 +01:00
parent 5dc841d9e6
commit a26cfb6d60
2 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/uml/class_diagram_visitor.cc
View File

@@ -23,6 +23,7 @@
#include <cppast/cpp_member_variable.hpp> #include <cppast/cpp_member_variable.hpp>
#include <cppast/cpp_template.hpp> #include <cppast/cpp_template.hpp>
#include <cppast/cpp_variable.hpp> #include <cppast/cpp_variable.hpp>
#include <cppast/cpp_enum.hpp>
#include <spdlog/spdlog.h> #include <spdlog/spdlog.h>
namespace clanguml { namespace clanguml {
@@ -71,16 +72,38 @@ void tu_visitor::operator()(const cppast::cpp_entity &file)
[&, this](const cppast::cpp_entity &e, cppast::visitor_info info) { [&, this](const cppast::cpp_entity &e, cppast::visitor_info info) {
if (e.kind() == cppast::cpp_entity_kind::class_t) { if (e.kind() == cppast::cpp_entity_kind::class_t) {
spdlog::debug("{}'{}' - {}", prefix, spdlog::debug("{}'{}' - {}", prefix,
cx::util::fully_prefixed(e), cppast::to_string(e.kind())); cx::util::full_name(e), cppast::to_string(e.kind()));
auto &cls = static_cast<const cppast::cpp_class &>(e); auto &cls = static_cast<const cppast::cpp_class &>(e);
if (ctx.config.should_include(cx::util::fully_prefixed(cls))) if (ctx.config.should_include(cx::util::fully_prefixed(cls)))
process_class_declaration(cls); process_class_declaration(cls);
} }
else if(e.kind() == cppast::cpp_entity_kind::enum_t) {
spdlog::debug("{}'{}' - {}", prefix,
cx::util::full_name(e), cppast::to_string(e.kind()));
auto &enm = static_cast<const cppast::cpp_enum &>(e);
if (ctx.config.should_include(cx::util::fully_prefixed(enm)))
process_enum_declaration(enm);
}
}); });
} }
void tu_visitor::process_enum_declaration(const cppast::cpp_enum &enm) {
enum_ e;
e.name = cx::util::full_name(enm);
for(const auto& ev : enm) {
if(ev.kind() == cppast::cpp_entity_kind::enum_value_t) {
e.constants.push_back(ev.name());
}
}
ctx.d.add_enum(std::move(e));
}
void tu_visitor::process_class_declaration(const cppast::cpp_class &cls) void tu_visitor::process_class_declaration(const cppast::cpp_class &cls)
{ {
class_ c; class_ c;

2
src/uml/class_diagram_visitor.h
View File

@@ -75,6 +75,8 @@ public:
void process_class_declaration(const cppast::cpp_class &cls); void process_class_declaration(const cppast::cpp_class &cls);
void process_enum_declaration(const cppast::cpp_enum &enm);
void process_field(const cppast::cpp_member_variable &mv, void process_field(const cppast::cpp_member_variable &mv,
clanguml::model::class_diagram::class_ &c, clanguml::model::class_diagram::class_ &c,
cppast::cpp_access_specifier_kind as); cppast::cpp_access_specifier_kind as);