Initial implementation of radius parameter in context filter

This commit is contained in:
Bartek Kryza
2023-11-08 23:42:22 +01:00
parent 205d6de060
commit 237ef26389
2 changed files with 135 additions and 52 deletions

View File

@@ -534,13 +534,118 @@ tvl::value_t access_filter::match(
[&a](const auto &access) { return a == access; });
}
context_filter::context_filter(
filter_t type, std::vector<common::string_or_regex> context)
context_filter::context_filter(filter_t type,
std::vector<common::string_or_regex> context, unsigned radius)
: filter_visitor{type}
, radius_{radius}
, context_{std::move(context)}
{
}
void context_filter::initialize(const diagram &d) const
{
if (initialized_)
return;
initialized_ = true;
bool effective_context_extended{true};
// First add to effective context all elements matching context_
for (const auto &c : context_) {
const auto &context_matches =
static_cast<const class_diagram::model::diagram &>(d)
.find<class_diagram::model::class_>(c);
for (const auto &maybe_match : context_matches) {
if (maybe_match)
effective_context_.emplace(maybe_match.value().id());
}
}
// Now repeat radius times - extend the effective context with elements
// matching in direct relationship to what is in context
auto radius_counter = radius_;
decltype(effective_context_) current_iteration_context;
while (radius_counter > 0 && effective_context_extended) {
// If at any iteration the effective context was not extended - we
// don't to need to continue
radius_counter--;
effective_context_extended = false;
current_iteration_context.clear();
// For each class in the model
for (const auto &c :
static_cast<const class_diagram::model::diagram &>(d).classes()) {
// Return a positive match if the element e is in a direct
// relationship with any of the effective context elements ...
for (const relationship &rel : c.get().relationships()) {
for (const auto &ec : effective_context_) {
if (d.should_include(rel.type()) && rel.destination() == ec)
current_iteration_context.emplace(c.get().id());
}
}
// ... or vice-versa
for (const auto &ec : effective_context_) {
const auto &maybe_class =
static_cast<const class_diagram::model::diagram &>(d)
.find<class_diagram::model::class_>(ec);
if (!maybe_class)
continue;
for (const relationship &rel :
maybe_class.value().relationships()) {
if (d.should_include(rel.type()) &&
rel.destination() == c.get().id())
current_iteration_context.emplace(c.get().id());
}
}
// Check if any of the elements parents are already in the
// effective context...
for (const class_diagram::model::class_parent &p :
c.get().parents()) {
for (const auto &ec : effective_context_) {
const auto &maybe_parent =
static_cast<const class_diagram::model::diagram &>(d)
.find<class_diagram::model::class_>(ec);
if (!maybe_parent)
continue;
if (d.should_include(relationship_t::kExtension) &&
maybe_parent.value().full_name(false) == p.name())
current_iteration_context.emplace(c.get().id());
}
}
// .. or vice-versa
for (const auto &ec : effective_context_) {
const auto &maybe_child =
static_cast<const class_diagram::model::diagram &>(d)
.find<class_diagram::model::class_>(ec);
for (const class_diagram::model::class_parent &p :
maybe_child.value().parents()) {
if (p.name() == c.get().full_name(false)) {
current_iteration_context.emplace(c.get().id());
}
}
}
}
for (auto id : current_iteration_context) {
if (effective_context_.count(id) == 0) {
// Found new element to add to context
effective_context_.emplace(id);
effective_context_extended = true;
}
}
}
}
tvl::value_t context_filter::match(const diagram &d, const element &e) const
{
if (d.type() != diagram_t::kClass)
@@ -551,53 +656,15 @@ tvl::value_t context_filter::match(const diagram &d, const element &e) const
if (!d.complete())
return {};
return tvl::any_of(context_.begin(), context_.end(),
[&e, &d](const auto &context_root_pattern) {
const auto &context_roots =
static_cast<const class_diagram::model::diagram &>(d)
.find<class_diagram::model::class_>(context_root_pattern);
initialize(d);
for (auto &context_root : context_roots) {
if (context_root.has_value()) {
// This is a direct match to the context root
if (context_root.value().id() == e.id())
return true;
if (effective_context_.empty())
return {};
// Return a positive match if the element e is in a direct
// relationship with any of the context_root's
for (const relationship &rel :
context_root.value().relationships()) {
if (d.should_include(rel.type()) &&
rel.destination() == e.id())
return true;
}
for (const relationship &rel : e.relationships()) {
if (d.should_include(rel.type()) &&
rel.destination() == context_root.value().id())
return true;
}
if (effective_context_.count(e.id()) > 0)
return true;
// Return a positive match if the context_root is a parent
// of the element
for (const class_diagram::model::class_parent &p :
context_root.value().parents()) {
if (p.name() == e.full_name(false))
return true;
}
if (dynamic_cast<const class_diagram::model::class_ *>(
&e) != nullptr) {
for (const class_diagram::model::class_parent &p :
static_cast<const class_diagram::model::class_ &>(e)
.parents()) {
if (p.name() ==
context_root.value().full_name(false))
return true;
}
}
}
}
return false;
});
return false;
}
paths_filter::paths_filter(filter_t type, const std::filesystem::path &root,
@@ -630,15 +697,17 @@ paths_filter::paths_filter(filter_t type, const std::filesystem::path &root,
match_successful = true;
}
catch (std::filesystem::filesystem_error &e) {
LOG_WARN("Cannot add non-existent path {} to paths filter",
LOG_WARN("Cannot add non-existent path {} to "
"paths filter",
absolute_path.string());
continue;
}
}
if (!match_successful)
LOG_WARN(
"Paths filter pattern '{}' did not match any files...", path);
LOG_WARN("Paths filter pattern '{}' did not match "
"any files...",
path);
}
}

View File

@@ -443,18 +443,32 @@ private:
};
/**
* Match diagram elements which are in direct relationship to any of the
* elements specified in context.
* Match diagram elements which are in within a 'radius' distance relationship
* to any of the elements specified in context.
*/
struct context_filter : public filter_visitor {
context_filter(filter_t type, std::vector<common::string_or_regex> context);
context_filter(filter_t type, std::vector<common::string_or_regex> context,
unsigned radius = 1);
~context_filter() override = default;
tvl::value_t match(const diagram &d, const element &r) const override;
private:
void initialize(const diagram &d) const;
const unsigned radius_;
std::vector<common::string_or_regex> context_;
/*!
* Represents all elements which should belong to the diagram based
* on this filter. It is populated by the initialize() method.
*/
mutable std::set<clanguml::common::id_t> effective_context_;
/*! Flag to mark whether the filter context has been computed */
mutable bool initialized_{false};
};
/**