src/kernel/activity/ExecImpl.hpp

   1 /* Copyright (c) 2007-2022. 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_EXEC_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_EXEC_HPP
   8
   9 #include <simgrid/s4u/Exec.hpp>
  10
  11 #include "src/kernel/activity/ActivityImpl.hpp"
  12 #include "src/kernel/context/Context.hpp"
  13
  14 namespace simgrid::kernel::activity {
  15
  16 class XBT_PUBLIC ExecImpl : public ActivityImpl_T<ExecImpl> {
  17   std::unique_ptr<resource::Action, std::function<void(resource::Action*)>> timeout_detector_{
  18       nullptr, [](resource::Action* a) { a->unref(); }};
  19   double sharing_penalty_             = 1.0;
  20   double bound_                       = 0.0;
  21   std::vector<s4u::Host*> hosts_;
  22   std::vector<double> flops_amounts_;
  23   std::vector<double> bytes_amounts_;
  24   int thread_count_ = 1;
  25   int cb_id_ = -1; // callback id from Host::on_state_change.connect()
  26
  27 public:
  28   ExecImpl();
  29
  30   ExecImpl& set_timeout(double timeout) override;
  31   ExecImpl& set_bound(double bound);
  32   ExecImpl& set_sharing_penalty(double sharing_penalty);
  33   ExecImpl& update_sharing_penalty(double sharing_penalty);
  34
  35   void set_cb_id(unsigned int cb_id) { cb_id_ = cb_id; }
  36
  37   ExecImpl& set_flops_amount(double flop_amount);
  38   ExecImpl& set_host(s4u::Host* host);
  39   s4u::Host* get_host() const { return hosts_.front(); }
  40   const std::vector<s4u::Host*>& get_hosts() const { return hosts_; }
  41
  42   ExecImpl& set_flops_amounts(const std::vector<double>& flops_amounts);
  43   ExecImpl& set_bytes_amounts(const std::vector<double>& bytes_amounts);
  44   ExecImpl& set_thread_count(int thread_count);
  45   ExecImpl& set_hosts(const std::vector<s4u::Host*>& hosts);
  46
  47   unsigned int get_host_number() const { return static_cast<unsigned>(hosts_.size()); }
  48   double get_seq_remaining_ratio();
  49   double get_par_remaining_ratio();
  50   double get_remaining() const override;
  51   virtual ActivityImpl* migrate(s4u::Host* to);
  52
  53   ExecImpl* start();
  54   void post() override;
  55   void set_exception(actor::ActorImpl* issuer) override;
  56   void finish() override;
  57
  58   void reset();
  59
  60   static xbt::signal<void(ExecImpl const&, s4u::Host*)> on_migration;
  61 };
  62 } // namespace simgrid::kernel::activity
  63 #endif