Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
[codefactor] Cosmetics.
[simgrid.git] / src / kernel / activity / CommImpl.hpp
index e9485d664a25bf41681726d9aa9e3a6518fe83b2..6f8770a19983ebee14dd4843da2534640040f2bc 100644 (file)
@@ -19,16 +19,22 @@ class XBT_PUBLIC CommImpl : public ActivityImpl_T<CommImpl> {
   ~CommImpl() override;
   void cleanup_surf();
 
+  static void (*copy_data_callback_)(CommImpl*, void*, size_t);
+
   double rate_       = 0.0;
   double size_       = 0.0;
   bool detached_     = false;   /* If detached or not */
   bool copied_       = false;   /* whether the data were already copied */
   MailboxImpl* mbox_ = nullptr; /* Rendez-vous where the comm is queued */
+  s4u::Host* from_   = nullptr; /* Pre-determined only for direct host-to-host communications */
+  s4u::Host* to_     = nullptr; /* Otherwise, computed at start() time from the actors */
 
 public:
   enum class Type { SEND, RECEIVE };
 
-  CommImpl(Type type) : type_(type) {}
+  static void set_copy_data_callback(void (*callback)(CommImpl*, void*, size_t));
+
+  explicit CommImpl(Type type) : type_(type) {}
   CommImpl(s4u::Host* from, s4u::Host* to, double bytes);
 
   CommImpl& set_size(double size);
@@ -44,6 +50,11 @@ public:
 
   void copy_data();
 
+  bool test() override;
+  void wait_for(actor::ActorImpl* issuer, double timeout) override;
+  static int test_any(const actor::ActorImpl* issuer, const std::vector<CommImpl*>& comms);
+  static void wait_any_for(actor::ActorImpl* issuer, const std::vector<CommImpl*>& comms, double timeout);
+
   CommImpl* start();
   void suspend() override;
   void resume() override;
@@ -51,7 +62,7 @@ public:
   void post() override;
   void finish() override;
 
-  const CommImpl::Type type_; /* Type of the communication (SEND or RECEIVE) */
+  const Type type_ = Type::SEND; /* Type of the communication (SEND or RECEIVE) */
 
 #if SIMGRID_HAVE_MC
   MailboxImpl* mbox_cpy = nullptr; /* Copy of the rendez-vous where the comm is queued, MC needs it for DPOR
@@ -70,8 +81,6 @@ expectations of the other side, too. See  */
   resource::Action* dst_timeout_ = nullptr; /* Surf's actions to instrument the timeouts */
   actor::ActorImplPtr src_actor_ = nullptr;
   actor::ActorImplPtr dst_actor_ = nullptr;
-  s4u::Host* from_               = nullptr; /* Pre-determined only for direct host-to-host communications */
-  s4u::Host* to_                 = nullptr; /* Otherwise, computed at start() time from the actors */
 
   /* Data to be transferred */
   unsigned char* src_buff_ = nullptr;