Refactoring getID to a unique global ID for entities

This commit is contained in:
Bartek Kryza
2022-08-11 17:04:16 +02:00
parent d63426e574
commit e4289c4cab
15 changed files with 264 additions and 22 deletions

View File

@@ -87,8 +87,9 @@ std::string class_::base_template() const { return base_template_full_name_; }
bool operator==(const class_ &l, const class_ &r)
{
return (l.name_and_ns() == r.name_and_ns()) &&
(l.templates_ == r.templates_);
return l.id() == r.id();
//(l.name_and_ns() == r.name_and_ns()) &&
// (l.templates_ == r.templates_);
}
void class_::add_type_alias(type_alias &&ta)

View File

@@ -165,8 +165,11 @@ bool diagram::add_class(std::unique_ptr<class_> &&c)
auto name_with_ns = c->name_and_ns();
auto name_and_ns = ns | name;
auto &cc = *c;
auto id = cc.id();
if (!has_class(cc)) {
if (base_name == "cpp_function_parameter")
LOG_DBG("AAAAAAAAAAAAAAAAAAAAAAA");
if (add_element(ns, std::move(c)))
classes_.push_back(std::ref(cc));
@@ -176,10 +179,13 @@ bool diagram::add_class(std::unique_ptr<class_> &&c)
throw std::runtime_error(
"Invalid element stored in the diagram tree");
LOG_DBG("Added class {} ({} - [{}])", base_name, full_name, id);
return true;
}
LOG_DBG("Class {} ({}) already in the model", base_name, full_name);
LOG_DBG(
"Class {} ({} - [{}]) already in the model", base_name, full_name, id);
return false;
}

View File

@@ -124,8 +124,6 @@ std::string get_source_text(
{
clang::LangOptions lo;
// NOTE: sm.getSpellingLoc() used in case the range corresponds to a
// macro/preprocessed source.
auto start_loc = sm.getSpellingLoc(range.getBegin());
auto last_token_loc = sm.getSpellingLoc(range.getEnd());
auto end_loc = clang::Lexer::getLocForEndOfToken(last_token_loc, 0, sm, lo);
@@ -149,6 +147,10 @@ bool translation_unit_visitor::VisitNamespaceDecl(clang::NamespaceDecl *ns)
if (ns->isAnonymousNamespace() || ns->isInline())
return true;
LOG_DBG("= Visiting namespace declaration {} at {}",
ns->getQualifiedNameAsString(),
ns->getLocation().printToString(source_manager_));
auto package_path = namespace_{common::get_qualified_name(*ns)};
auto package_parent = package_path;
@@ -198,6 +200,10 @@ bool translation_unit_visitor::VisitEnumDecl(clang::EnumDecl *enm)
if (enm->getNameAsString().empty())
return true;
LOG_DBG("= Visiting enum declaration {} at {}",
enm->getQualifiedNameAsString(),
enm->getLocation().printToString(source_manager_));
auto e_ptr = std::make_unique<enum_>(config_.using_namespace());
auto &e = *e_ptr;
@@ -241,6 +247,10 @@ bool translation_unit_visitor::VisitClassTemplateSpecializationDecl(
if (source_manager_.isInSystemHeader(cls->getSourceRange().getBegin()))
return true;
LOG_DBG("= Visiting template specialization declaration {} at {}",
cls->getQualifiedNameAsString(),
cls->getLocation().printToString(source_manager_));
// Skip forward declarations
if (!cls->isCompleteDefinition()) {
// Register this forward declaration in case there is no complete
@@ -251,7 +261,7 @@ bool translation_unit_visitor::VisitClassTemplateSpecializationDecl(
// Check if the class was already processed within
// VisitClassTemplateDecl()
if (diagram_.has_element(cls->getID()))
return true;
return true;
// TODO: Add support for classes defined in function/method bodies
if (cls->isLocalClass())
@@ -289,6 +299,10 @@ bool translation_unit_visitor::VisitTypeAliasTemplateDecl(
if (source_manager_.isInSystemHeader(cls->getSourceRange().getBegin()))
return true;
LOG_DBG("= Visiting template type alias declaration {} at {}",
cls->getQualifiedNameAsString(),
cls->getLocation().printToString(source_manager_));
auto *template_type_specialization_ptr =
cls->getTemplatedDecl()
->getUnderlyingType()
@@ -320,9 +334,9 @@ bool translation_unit_visitor::VisitClassTemplateDecl(
if (source_manager_.isInSystemHeader(cls->getSourceRange().getBegin()))
return true;
// Skip forward declarations
// if (!cls->getTemplatedDecl()->isCompleteDefinition())
// return true;
LOG_DBG("= Visiting class template declaration {} at {}",
cls->getQualifiedNameAsString(),
cls->getLocation().printToString(source_manager_));
auto c_ptr = create_class_declaration(cls->getTemplatedDecl());
@@ -362,9 +376,27 @@ bool translation_unit_visitor::VisitCXXRecordDecl(clang::CXXRecordDecl *cls)
if (source_manager_.isInSystemHeader(cls->getSourceRange().getBegin()))
return true;
LOG_DBG("= Visiting class declaration {} at {}",
cls->getQualifiedNameAsString(),
cls->getLocation().printToString(source_manager_));
// Skip forward declarations
if (!cls->isCompleteDefinition())
return true;
// if (!cls->isCompleteDefinition())
// return true;
if (cls->getQualifiedNameAsString() == "cppast::cpp_function_parameter" &&
cls->getLocation().printToString(source_manager_) ==
"/home/bartek/devel/clang-uml-showcases/cppast/src/../include/"
"cppast/cpp_function.hpp:16:7") {
LOG_DBG("##############################################################"
"##########################");
for (const auto &c : diagram().classes()) {
LOG_DBG(" >> {} [{}]", c.get().full_name(false), c.get().id());
}
const auto &ccc = diagram().get_class(cls->getID());
if (ccc.has_value())
LOG_DBG("---------- {}", ccc.get()->full_name(false));
}
// Templated records are handled by VisitClassTemplateDecl()
if (cls->isTemplated() || cls->isTemplateDecl() ||
@@ -373,8 +405,9 @@ bool translation_unit_visitor::VisitCXXRecordDecl(clang::CXXRecordDecl *cls)
return true;
// Check if the class was already processed within VisitClassTemplateDecl()
if (diagram_.has_element(cls->getID()))
return true;
// auto cls_id = cls->getID();
// if (diagram_.has_element(cls_id))
// return true;
// TODO: Add support for classes defined in function/method bodies
if (cls->isLocalClass())
@@ -385,9 +418,12 @@ bool translation_unit_visitor::VisitCXXRecordDecl(clang::CXXRecordDecl *cls)
if (!c_ptr)
return true;
auto &class_model = *c_ptr;
auto &class_model = diagram().get_class(cls->getID()).has_value()
? *diagram().get_class(cls->getID()).get()
: *c_ptr;
process_class_declaration(*cls, class_model);
if (cls->isCompleteDefinition())
process_class_declaration(*cls, class_model);
auto id = class_model.id();
if (!cls->isCompleteDefinition()) {
@@ -403,6 +439,10 @@ bool translation_unit_visitor::VisitCXXRecordDecl(clang::CXXRecordDecl *cls)
diagram_.add_class(std::move(c_ptr));
}
else {
LOG_DBG("Skipping class {} with id {}", class_model.full_name(),
class_model.id());
}
return true;
}
@@ -901,7 +941,7 @@ void translation_unit_visitor::process_function_parameter(
const auto *default_arg = p.getDefaultArg();
if (default_arg != nullptr) {
auto default_arg_str =
default_arg->getSourceRange().printToString(source_manager_);
get_source_text(default_arg->getSourceRange(), source_manager_);
parameter.set_default_value(default_arg_str);
}
}
@@ -1864,7 +1904,7 @@ void translation_unit_visitor::add_incomplete_forward_declarations()
void translation_unit_visitor::finalize()
{
add_incomplete_forward_declarations();
// add_incomplete_forward_declarations();
}
bool translation_unit_visitor::simplify_system_template(

View File

@@ -262,6 +262,9 @@ public:
{
visitor_.TraverseDecl(ast_context.getTranslationUnitDecl());
visitor_.finalize();
// LOG_DBG("= Finalized translation unit: {}",
// ast_context.getTranslationUnitDecl());
}
};
@@ -287,6 +290,8 @@ public:
protected:
bool BeginSourceFileAction(clang::CompilerInstance &ci) override
{
LOG_DBG("Visiting source file: {}", getCurrentFile().str());
if constexpr (std::is_same_v<DiagramModel,
clanguml::include_diagram::model::diagram>) {
auto find_includes_callback =
@@ -351,11 +356,10 @@ std::unique_ptr<DiagramModel> generate(
const auto matches = glob::rglob(g);
std::copy(matches.begin(), matches.end(),
std::back_inserter(translation_units));
LOG_DBG(
"Found translation units: {}", fmt::join(translation_units, ", "));
}
LOG_DBG("Found translation units: {}", fmt::join(translation_units, ", "));
clang::tooling::ClangTool clang_tool(db, translation_units);
auto action_factory =
std::make_unique<diagram_action_visitor_factory<DiagramModel,