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 {
  15 namespace kernel {
  16 namespace activity {
  17
  18 class XBT_PUBLIC ExecImpl : public ActivityImpl_T<ExecImpl> {
  19   std::unique_ptr<resource::Action, std::function<void(resource::Action*)>> timeout_detector_{
  20       nullptr, [](resource::Action* a) { a->unref(); }};
  21   double sharing_penalty_             = 1.0;
  22   double bound_                       = 0.0;
  23   std::vector<s4u::Host*> hosts_;
  24   std::vector<double> flops_amounts_;
  25   std::vector<double> bytes_amounts_;
  26   int cb_id_ = -1; // callback id from Host::on_state_change.connect()
  27
  28 public:
  29   ExecImpl();
  30
  31   ExecImpl& set_timeout(double timeout) override;
  32   ExecImpl& set_bound(double bound);
  33   ExecImpl& set_sharing_penalty(double sharing_penalty);
  34   ExecImpl& update_sharing_penalty(double sharing_penalty);
  35
  36   void set_cb_id(unsigned int cb_id) { cb_id_ = cb_id; }
  37
  38   ExecImpl& set_flops_amount(double flop_amount);
  39   ExecImpl& set_host(s4u::Host* host);
  40   s4u::Host* get_host() const { return hosts_.front(); }
  41   const std::vector<s4u::Host*>& get_hosts() const { return hosts_; }
  42
  43   ExecImpl& set_flops_amounts(const std::vector<double>& flops_amounts);
  44   ExecImpl& set_bytes_amounts(const std::vector<double>& bytes_amounts);
  45   ExecImpl& set_hosts(const std::vector<s4u::Host*>& hosts);
  46
  47   unsigned int get_host_number() const { return 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 activity
  63 } // namespace kernel
  64 } // namespace simgrid
  65 #endif