Make CommDet a plugin on top of Safety

[simgrid.git] / src / mc / checker / SafetyChecker.hpp

diff --git a/src/mc/checker/SafetyChecker.hpp b/src/mc/checker/SafetyChecker.hpp
index 4cf5f25..3851d39 100644 (file)
--- a/src/mc/checker/SafetyChecker.hpp
+++ b/src/mc/checker/SafetyChecker.hpp
@@ -23,6 +23,18 @@ class XBT_PRIVATE SafetyChecker : public Checker {
   ReductionMode reductionMode_ = ReductionMode::unset;
   long backtrack_count_        = 0;
 
+  static xbt::signal<void()> on_exploration_start_signal;
+  static xbt::signal<void()> on_backtracking_signal;
+
+  static xbt::signal<void(State*)> on_state_creation_signal;
+
+  static xbt::signal<void(State*)> on_restore_system_state_signal;
+  static xbt::signal<void()> on_restore_initial_state_signal;
+  static xbt::signal<void(Transition*)> on_transition_replay_signal;
+  static xbt::signal<void(Transition*)> on_transition_execute_signal;
+
+  static xbt::signal<void()> on_log_state_signal;
+
 public:
   explicit SafetyChecker(Session* session);
   void run() override;
@@ -30,6 +42,24 @@ public:
   std::vector<std::string> get_textual_trace() override;
   void log_state() override;
 
+  /** Called once when the exploration starts */
+  static void on_exploration_start(std::function<void()> f) { on_exploration_start_signal.connect(f); }
+  /** Called each time that the exploration backtracks from a exploration end */
+  static void on_backtracking(std::function<void()> f) { on_backtracking_signal.connect(f); }
+  /** Called each time that a new state is create */
+  static void on_state_creation(std::function<void(State*)> f) { on_state_creation_signal.connect(f); }
+  /** Called when rollbacking directly onto a previously checkpointed state */
+  static void on_restore_system_state(std::function<void(State*)> f) { on_restore_system_state_signal.connect(f); }
+  /** Called when the state to which we backtrack was not checkpointed state, forcing us to restore the initial state
+   * before replaying some transitions */
+  static void on_restore_initial_state(std::function<void()> f) { on_restore_initial_state_signal.connect(f); }
+  /** Called when replaying a transition that was previously executed, to reach a backtracked state */
+  static void on_transition_replay(std::function<void(Transition*)> f) { on_transition_replay_signal.connect(f); }
+  /** Called when executing a new transition */
+  static void on_transition_execute(std::function<void(Transition*)> f) { on_transition_execute_signal.connect(f); }
+  /** Called when displaying the statistics at the end of the exploration */
+  static void on_log_state(std::function<void()> f) { on_log_state_signal.connect(f); }
+
 private:
   void check_non_termination(const State* current_state);
   void backtrack();