Merge branch 'master' of scm.gforge.inria.fr:/gitroot/simgrid/simgrid

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

diff --git a/src/mc/SafetyChecker.hpp b/src/mc/SafetyChecker.hpp
index 0bc502d..9694a93 100644 (file)
--- a/src/mc/SafetyChecker.hpp
+++ b/src/mc/SafetyChecker.hpp
@@ -8,14 +8,18 @@
 #define SIMGRID_MC_SAFETY_CHECKER_HPP
 
 #include <list>
+#include <memory>
+#include <string>
+#include <vector>
 
 #include "src/mc/mc_forward.hpp"
 #include "src/mc/Checker.hpp"
+#include "src/mc/VisitedState.hpp"
 
 namespace simgrid {
 namespace mc {
 
-class SafetyChecker : public Checker {
+class XBT_PRIVATE SafetyChecker : public Checker {
   simgrid::mc::ReductionMode reductionMode_ = simgrid::mc::ReductionMode::unset;
 public:
   SafetyChecker(Session& session);
@@ -29,7 +33,8 @@ private:
   bool checkNonDeterminism(simgrid::mc::State* current_state);
 private:
   /** Stack representing the position in the exploration graph */
-  std::list<simgrid::mc::State*> stack_;
+  std::list<std::unique_ptr<simgrid::mc::State>> stack_;
+  simgrid::mc::VisitedStates visitedStates_;
 };
 
 }