[S4U] Engine: Add Engine::get_filtered_links

diff --git a/include/simgrid/s4u/Engine.hpp b/include/simgrid/s4u/Engine.hpp
index eabfef2..8d9477c 100644 (file)
--- a/include/simgrid/s4u/Engine.hpp
+++ b/include/simgrid/s4u/Engine.hpp
@@ -104,6 +104,7 @@ public:
 
   size_t get_link_count();
   std::vector<Link*> get_all_links();
+  std::vector<Link*> get_filtered_links(std::function<bool(Link*)> filter);
 
   size_t get_storage_count();
   std::vector<Storage*> get_all_storages();

diff --git a/src/s4u/s4u_Engine.cpp b/src/s4u/s4u_Engine.cpp
index d522abf..c311f1d 100644 (file)
--- a/src/s4u/s4u_Engine.cpp
+++ b/src/s4u/s4u_Engine.cpp
@@ -190,6 +190,20 @@ std::vector<Link*> Engine::get_all_links()
   return res;
 }
 
+std::vector<Link*> Engine::get_filtered_links(std::function<bool(Link*)> filter)
+{
+  // FIXME: This is a terrible implementation and should be done
+  // without getting all links first.
+  std::vector<Link*> res;
+  kernel::resource::LinkImpl::linksList(&res);
+  std::vector<Link*> filtered_list;
+  for (auto& link : res) {
+    if (filter(link))
+      filtered_list.push_back(link);
+  }
+  return filtered_list;
+}
+
 void Engine::run()
 {
   if (MC_is_active()) {