add two more utility functions to the Operation plugin

diff --git a/include/simgrid/plugins/operation.hpp b/include/simgrid/plugins/operation.hpp
index 6725466..7dbbb33 100644 (file)
--- a/include/simgrid/plugins/operation.hpp
+++ b/include/simgrid/plugins/operation.hpp
@@ -67,6 +67,8 @@ public:
   double get_amount() const { return amount_; }
   void add_successor(OperationPtr op);
   void remove_successor(OperationPtr op);
+  void remove_all_successors();
+  const std::set<Operation*>& get_successors() const { return successors_ ;}
   void on_this_start(const std::function<void(Operation*)>& func);
   void on_this_end(const std::function<void(Operation*)>& func);
   int get_count() const;

diff --git a/src/plugins/operation.cpp b/src/plugins/operation.cpp
index ce34fe6..32ed866 100644 (file)
--- a/src/plugins/operation.cpp
+++ b/src/plugins/operation.cpp
@@ -170,6 +170,17 @@ void Operation::remove_successor(OperationPtr successor)
   successor->remove_predecessor(this);
 }
 
+void Operation::remove_all_successors()
+{
+  simgrid::kernel::actor::simcall_answered([this] {
+    while (not successors_.empty()) {
+      auto* successor = *(successors_.begin());
+      successor->predecessors_.erase(this);
+      successors_.erase(successor);
+    }
+  });
+}
+
 /** @ingroup plugin_operation
  *  @param func The function to set.
  *  @brief Set a function to be called before each execution.