X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/936d213bbf44dbbcdfdadd3f0a777cb5af9407b8..4b90ce2abff77ca7898fd9292a122440df739916:/src/mc/VisitedState.hpp

diff --git a/src/mc/VisitedState.hpp b/src/mc/VisitedState.hpp
index 7028f4d644..f2eae0c9dc 100644
--- a/src/mc/VisitedState.hpp
+++ b/src/mc/VisitedState.hpp
@@ -1,5 +1,4 @@
-/* Copyright (c) 2007-2016. The SimGrid Team.
- * All rights reserved.                                                     */
+/* Copyright (c) 2007-2020. The SimGrid Team. All rights reserved.          */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -7,22 +6,22 @@
 #ifndef SIMGRID_MC_VISITED_STATE_HPP
 #define SIMGRID_MC_VISITED_STATE_HPP
 
-#include <cstddef>
+#include "src/mc/mc_state.hpp"
+#include "src/mc/sosp/Snapshot.hpp"
 
+#include <cstddef>
 #include <memory>
 
-#include "src/mc/mc_snapshot.h"
-#include "src/mc/mc_state.h"
-
 namespace simgrid {
 namespace mc {
 
-struct XBT_PRIVATE VisitedState {
+class XBT_PRIVATE VisitedState {
+public:
   std::shared_ptr<simgrid::mc::Snapshot> system_state = nullptr;
   std::size_t heap_bytes_used = 0;
   int actors_count            = 0;
   int num          = 0; // unique id of that state in the storage of all stored IDs
-  int original_num = 0; // num field of the VisitedState to which I was declared equal to (used for dot_output)
+  int original_num = -1; // num field of the VisitedState to which I was declared equal to (used for dot_output)
 
   explicit VisitedState(unsigned long state_number);
   ~VisitedState() = default;
@@ -32,7 +31,9 @@ class XBT_PRIVATE VisitedStates {
   std::vector<std::unique_ptr<simgrid::mc::VisitedState>> states_;
 public:
   void clear() { states_.clear(); }
-  std::unique_ptr<simgrid::mc::VisitedState> addVisitedState(unsigned long state_number, simgrid::mc::State* graph_state, bool compare_snpashots);
+  std::unique_ptr<simgrid::mc::VisitedState> addVisitedState(unsigned long state_number,
+                                                             simgrid::mc::State* graph_state, bool compare_snapshots);
+
 private:
   void prune();
 };