src/kernel/activity/ActivityImpl.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_ACTIVITYIMPL_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_ACTIVITYIMPL_HPP
   8
   9 #include <string>
  10 #include <list>
  11
  12 #include "simgrid/forward.h"
  13 #include <xbt/utility.hpp>
  14
  15 #include <atomic>
  16 #include <simgrid/kernel/resource/Action.hpp>
  17 #include <simgrid/simix.hpp>
  18
  19 namespace simgrid {
  20 namespace kernel {
  21 namespace activity {
  22
  23 XBT_DECLARE_ENUM_CLASS(State, WAITING, READY, RUNNING, DONE, CANCELED, FAILED, SRC_HOST_FAILURE, DST_HOST_FAILURE,
  24                        TIMEOUT, SRC_TIMEOUT, DST_TIMEOUT, LINK_FAILURE);
  25
  26 class XBT_PUBLIC ActivityImpl {
  27   std::atomic_int_fast32_t refcount_{0};
  28   std::string name_ = "";
  29
  30 public:
  31   virtual ~ActivityImpl();
  32   ActivityImpl() = default;
  33   State state_   = State::WAITING;      /* State of the activity */
  34   std::list<smx_simcall_t> simcalls_;   /* List of simcalls waiting for this activity */
  35   resource::Action* surf_action_ = nullptr;
  36   actor::ActorImpl* actor_       = nullptr;
  37   s4u::Activity* piface_         = nullptr;
  38
  39 protected:
  40   void inline set_name(const std::string& name)
  41   {
  42     // This is to keep name_ private while allowing ActivityImpl_T<??> to set it and then return a Ptr to qualified
  43     // child type
  44     name_ = name;
  45   }
  46
  47 public:
  48   const std::string& get_name() const { return name_; }
  49   const char* get_cname() const { return name_.c_str(); }
  50
  51   void set_actor(actor::ActorImpl* actor) { actor_ = actor; }
  52   actor::ActorImpl* get_actor() const { return actor_; }
  53
  54   void set_iface(s4u::Activity* iface) { piface_ = iface; }
  55   s4u::Activity* get_iface() { return piface_; }
  56
  57   virtual bool test();
  58   virtual void wait_for(actor::ActorImpl* issuer, double timeout);
  59   virtual ActivityImpl& set_timeout(double) { THROW_UNIMPLEMENTED; }
  60
  61   virtual void suspend();
  62   virtual void resume();
  63   virtual void cancel();
  64
  65   virtual void post() = 0; // Called by the main loop when the activity is marked as terminated or failed by its model.
  66                            // Setups the status, clean things up, and call finish()
  67   virtual void set_exception(actor::ActorImpl* issuer) = 0; // Raising exceptions and stuff
  68   virtual void finish() = 0; // Unlock all simcalls blocked on that activity, either because it was marked as done by
  69                              // the model or because it terminated without waiting for the model
  70
  71   void register_simcall(smx_simcall_t simcall);
  72   void unregister_simcall(smx_simcall_t simcall);
  73   void clean_action();
  74   virtual double get_remaining() const;
  75   const char* get_state_str() const;
  76   // Support for the boost::intrusive_ptr<ActivityImpl> datatype
  77   friend XBT_PUBLIC void intrusive_ptr_add_ref(ActivityImpl* activity);
  78   friend XBT_PUBLIC void intrusive_ptr_release(ActivityImpl* activity);
  79
  80   static xbt::signal<void(ActivityImpl const&)> on_suspended;
  81   static xbt::signal<void(ActivityImpl const&)> on_resumed;
  82 };
  83
  84 /* This class exists to allow chained setters as in exec->set_name()->set_priority()->set_blah()
  85  * The difficulty is that set_name() must return a qualified child class, not the generic ancestor
  86  * But the getter is still in the ancestor to be usable on generic activities with no downcast */
  87 template <class AnyActivityImpl> class ActivityImpl_T : public ActivityImpl {
  88 private:
  89   std::string tracing_category_ = "";
  90
  91 public:
  92   AnyActivityImpl& set_name(const std::string& name) /* Hides the function in the ancestor class */
  93   {
  94     ActivityImpl::set_name(name);
  95     return static_cast<AnyActivityImpl&>(*this);
  96   }
  97
  98   AnyActivityImpl& set_tracing_category(const std::string& category)
  99   {
 100     tracing_category_ = category;
 101     return static_cast<AnyActivityImpl&>(*this);
 102   }
 103   const std::string& get_tracing_category() const { return tracing_category_; }
 104 };
 105
 106 } // namespace activity
 107 } // namespace kernel
 108 } // namespace simgrid
 109
 110 #endif /* SIMGRID_KERNEL_ACTIVITY_ACTIVITYIMPL_HPP */