Type consistency.

[simgrid.git] / src / mc / api / State.hpp

diff --git a/src/mc/api/State.hpp b/src/mc/api/State.hpp
index ddca326..92164cf 100644 (file)
--- a/src/mc/api/State.hpp
+++ b/src/mc/api/State.hpp
@@ -7,6 +7,7 @@
 #define SIMGRID_MC_STATE_HPP
 
 #include "src/mc/api/ActorState.hpp"
+#include "src/mc/api/RemoteApp.hpp"
 #include "src/mc/sosp/Snapshot.hpp"
 #include "src/mc/transition/Transition.hpp"
 
@@ -29,7 +30,7 @@ class XBT_PRIVATE State : public xbt::Extendable<State> {
   std::shared_ptr<Snapshot> system_state_;
 
 public:
-  explicit State(Session& session);
+  explicit State(const RemoteApp& remote_app);
 
   /* Returns a positive number if there is another transition to pick, or -1 if not */
   aid_t next_transition() const;
@@ -43,10 +44,10 @@ public:
   bool is_done(aid_t actor) const { return actors_to_run_.at(actor).is_done(); }
   Transition* get_transition() const;
   void set_transition(Transition* t) { transition_.reset(t); }
-  std::map<aid_t, ActorState> const& get_actors_list() { return actors_to_run_; }
+  std::map<aid_t, ActorState> const& get_actors_list() const { return actors_to_run_; }
 
-  int get_actor_count() { return actors_to_run_.size(); }
-  bool is_actor_enabled(int actor) { return actors_to_run_.at(actor).is_enabled(); }
+  unsigned long get_actor_count() const { return actors_to_run_.size(); }
+  bool is_actor_enabled(aid_t actor) { return actors_to_run_.at(actor).is_enabled(); }
 
   Snapshot* get_system_state() const { return system_state_.get(); }
   void set_system_state(std::shared_ptr<Snapshot> state) { system_state_ = std::move(state); }