[mc] Get rid of the global simgrid::mc::visited_states, use a class instead

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

diff --git a/src/mc/VisitedState.hpp b/src/mc/VisitedState.hpp
index 4e0c4fb..3e06ecc 100644 (file)
--- a/src/mc/VisitedState.hpp
+++ b/src/mc/VisitedState.hpp
@@ -27,8 +27,14 @@ struct XBT_PRIVATE VisitedState {
   ~VisitedState();
 };
 
-extern XBT_PRIVATE std::vector<std::unique_ptr<simgrid::mc::VisitedState>> visited_states;
-XBT_PRIVATE std::unique_ptr<simgrid::mc::VisitedState> is_visited_state(simgrid::mc::State* graph_state, bool compare_snpashots);
+class VisitedStates {
+  std::vector<std::unique_ptr<simgrid::mc::VisitedState>> states_;
+public:
+  void clear() { states_.clear(); }
+  std::unique_ptr<simgrid::mc::VisitedState> addVisitedState(simgrid::mc::State* graph_state, bool compare_snpashots);
+private:
+  void prune();
+};
 
 }
 }