get_enabled_transitions() implemented

diff --git a/src/mc/api.cpp b/src/mc/api.cpp
index c4c6895..3c3d488 100644 (file)
--- a/src/mc/api.cpp
+++ b/src/mc/api.cpp
@@ -580,6 +580,42 @@ smx_simcall_t Api::mc_state_choose_request(simgrid::mc::State* state) const
   return nullptr;
 }
 
+std::list<udpor_transition_t> Api::get_enabled_transitions(simgrid::mc::State* state)
+{
+  std::list<udpor_transition_t> tr_list{};
+
+  for (auto& actor : mc_model_checker->get_remote_simulation().actors()) {
+    auto actor_pid  = actor.copy.get_buffer()->get_pid();
+    auto actor_impl = actor.copy.get_buffer();
+
+    // Only consider the actors that were marked as interleaving by the checker algorithm
+    if (not state->actor_states_[actor_pid].is_todo())
+      continue;
+    // Not executable in the application
+    if (not simgrid::mc::actor_is_enabled(actor_impl))
+      continue;
+
+    udpor_transition_t udpor_transition = std::unique_ptr<s_udpor_transition>(new s_udpor_transition());
+    Simcall simcall_call                = actor_impl->simcall_.call_;
+    smx_simcall_t simcall               = &actor_impl->simcall_;
+    udpor_transition->call_             = simcall_call;
+    switch (simcall_call) {
+      case Simcall::COMM_ISEND:
+      case Simcall::COMM_IRECV: {
+        udpor_transition->mbox_remote_addr = get_mbox_remote_addr(simcall);
+        udpor_transition->comm_remote_addr = get_comm_remote_addr(simcall);
+        break;
+      }
+
+      default:
+        break;
+    }
+    tr_list.emplace_back(std::move(udpor_transition));
+  }
+  
+  return tr_list;
+}
+
 bool Api::simcall_check_dependency(smx_simcall_t const req1, smx_simcall_t const req2) const
 {
   if (req1->issuer_ == req2->issuer_)

diff --git a/src/mc/api.hpp b/src/mc/api.hpp
index 1331079..55ecf6e 100644 (file)
--- a/src/mc/api.hpp
+++ b/src/mc/api.hpp
@@ -15,6 +15,18 @@
 namespace simgrid {
 namespace mc {
 
+/**
+ * @brief Maintains the transition's information.
+ */
+struct s_udpor_transition {
+  simgrid::simix::Simcall call_ = simgrid::simix::Simcall::NONE;
+  long issuer_id                = -1;
+  RemotePtr<kernel::activity::MailboxImpl> mbox_remote_addr {}; // used to represent mailbox remote address for isend and ireceive transitions
+  RemotePtr<kernel::activity::ActivityImpl> comm_remote_addr {}; // the communication this transition concerns (to be used only for isend, ireceive, wait and test)
+};
+
+typedef std::unique_ptr<s_udpor_transition> udpor_transition_t;
+
 /*
 ** This class aimes to implement FACADE APIs for simgrid. The FACADE layer sits between the CheckerSide
 ** (Unfolding_Checker, DPOR, ...) layer and the
@@ -92,6 +104,9 @@ public:
   void dump_record_path() const;
   smx_simcall_t mc_state_choose_request(simgrid::mc::State* state) const;
 
+  // UDPOR APIs
+  std::list<udpor_transition_t> get_enabled_transitions(simgrid::mc::State* state);
+
   // SIMCALL APIs
   std::string request_to_string(smx_simcall_t req, int value, RequestType request_type) const;
   std::string request_get_dot_output(smx_simcall_t req, int value) const;