src/kernel/activity/CommImpl.hpp

   1 /* Copyright (c) 2007-2021. The SimGrid Team. All rights reserved.          */
   2
   3 /* This program is free software; you can redistribute it and/or modify it
   4  * under the terms of the license (GNU LGPL) which comes with this package. */
   5
   6 #ifndef SIMGRID_KERNEL_ACTIVITY_COMM_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_COMM_HPP
   8
   9 #include "src/kernel/activity/ActivityImpl.hpp"
  10 #include "src/kernel/actor/ActorImpl.hpp"
  11 #include "surf/surf.hpp"
  12
  13
  14 namespace simgrid {
  15 namespace kernel {
  16 namespace activity {
  17
  18 class XBT_PUBLIC CommImpl : public ActivityImpl_T<CommImpl> {
  19   ~CommImpl() override;
  20   void cleanupSurf();
  21
  22   double rate_       = 0.0;
  23   double size_       = 0.0;
  24   bool detached_     = false;   /* If detached or not */
  25   bool copied_       = false;   /* whether the data were already copied */
  26   MailboxImpl* mbox_ = nullptr; /* Rendez-vous where the comm is queued */
  27
  28 public:
  29   CommImpl() = default;
  30   CommImpl(s4u::Host* from, s4u::Host* to, double bytes);
  31
  32   enum class Type { SEND = 0, RECEIVE, READY, DONE };
  33
  34   CommImpl& set_type(CommImpl::Type type);
  35   CommImpl& set_size(double size);
  36   CommImpl& set_src_buff(unsigned char* buff, size_t size);
  37   CommImpl& set_dst_buff(unsigned char* buff, size_t* size);
  38   CommImpl& set_rate(double rate);
  39   CommImpl& set_mailbox(MailboxImpl* mbox);
  40   CommImpl& detach();
  41
  42   double get_rate() const { return rate_; }
  43   MailboxImpl* get_mailbox() const { return mbox_; }
  44   bool detached() const { return detached_; }
  45
  46   void copy_data();
  47
  48   CommImpl* start();
  49   void suspend() override;
  50   void resume() override;
  51   void cancel() override;
  52   void post() override;
  53   void finish() override;
  54
  55   CommImpl::Type type_; /* Type of the communication (SEND or RECEIVE) */
  56
  57 #if SIMGRID_HAVE_MC
  58   MailboxImpl* mbox_cpy = nullptr; /* Copy of the rendez-vous where the comm is queued, MC needs it for DPOR
  59                                      (comm.mbox set to nullptr when the communication is removed from the mailbox
  60                                      (used as garbage collector)) */
  61 #endif
  62
  63   void (*clean_fun)(void*) = nullptr; /* Function to clean the detached src_buf if something goes wrong */
  64   bool (*match_fun)(void*, void*, CommImpl*) = nullptr; /* Filter function used by the other side. It is used when
  65 looking if a given communication matches my needs. For that, myself must match the
  66 expectations of the other side, too. See  */
  67   void (*copy_data_fun)(CommImpl*, void*, size_t) = nullptr;
  68
  69   /* Surf action data */
  70   resource::Action* src_timeout_ = nullptr; /* Surf's actions to instrument the timeouts */
  71   resource::Action* dst_timeout_ = nullptr; /* Surf's actions to instrument the timeouts */
  72   actor::ActorImplPtr src_actor_ = nullptr;
  73   actor::ActorImplPtr dst_actor_ = nullptr;
  74   s4u::Host* from_               = nullptr; /* Pre-determined only for direct host-to-host communications */
  75   s4u::Host* to_                 = nullptr; /* Otherwise, computed at start() time from the actors */
  76
  77   /* Data to be transferred */
  78   unsigned char* src_buff_ = nullptr;
  79   unsigned char* dst_buff_ = nullptr;
  80   size_t src_buff_size_    = 0;
  81   size_t* dst_buff_size_   = nullptr;
  82
  83   void* src_data_ = nullptr; /* User data associated to the communication */
  84   void* dst_data_ = nullptr;
  85 };
  86 } // namespace activity
  87 } // namespace kernel
  88 } // namespace simgrid
  89
  90 #endif