src/kernel/activity/ActivityImpl.hpp

   1 /* Copyright (c) 2007-2020. 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_ACTIVITY_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_ACTIVITY_HPP
   8
   9 #include <string>
  10 #include <list>
  11
  12 #include <xbt/base.h>
  13 #include "simgrid/forward.h"
  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 enum class State {
  23   WAITING = 0,
  24   READY,
  25   RUNNING,
  26   DONE,
  27   CANCELED,
  28   FAILED,
  29   SRC_HOST_FAILURE,
  30   DST_HOST_FAILURE,
  31   TIMEOUT,
  32   SRC_TIMEOUT,
  33   DST_TIMEOUT,
  34   LINK_FAILURE
  35 };
  36
  37 class XBT_PUBLIC ActivityImpl {
  38   std::atomic_int_fast32_t refcount_{0};
  39
  40 public:
  41   virtual ~ActivityImpl();
  42   ActivityImpl() = default;
  43   State state_   = State::WAITING;      /* State of the activity */
  44   std::list<smx_simcall_t> simcalls_;   /* List of simcalls waiting for this activity */
  45   resource::Action* surf_action_ = nullptr;
  46
  47   virtual void suspend();
  48   virtual void resume();
  49   virtual void cancel();
  50
  51   virtual void post() = 0; // Called by the main loop when the activity is marked as terminated or failed by its model.
  52                            // Setups the status, clean things up, and call finish()
  53   virtual void finish() = 0; // Unlock all simcalls blocked on that activity, either because it was marked as done by
  54                              // the model or because it terminated without waiting for the model
  55
  56   virtual void register_simcall(smx_simcall_t simcall);
  57   void clean_action();
  58   virtual double get_remaining() const;
  59   // boost::intrusive_ptr<ActivityImpl> support:
  60   friend XBT_PUBLIC void intrusive_ptr_add_ref(ActivityImpl* activity);
  61   friend XBT_PUBLIC void intrusive_ptr_release(ActivityImpl* activity);
  62
  63   static xbt::signal<void(ActivityImpl const&)> on_suspended;
  64   static xbt::signal<void(ActivityImpl const&)> on_resumed;
  65 };
  66
  67 template <class AnyActivityImpl> class ActivityImpl_T : public ActivityImpl {
  68 private:
  69   std::string name_             = "";
  70   std::string tracing_category_ = "";
  71
  72 public:
  73   AnyActivityImpl& set_name(const std::string& name)
  74   {
  75     name_ = name;
  76     return static_cast<AnyActivityImpl&>(*this);
  77   }
  78   const std::string& get_name() { return name_; }
  79   const char* get_cname() { return name_.c_str(); }
  80
  81   AnyActivityImpl& set_tracing_category(const std::string& category)
  82   {
  83     tracing_category_ = category;
  84     return static_cast<AnyActivityImpl&>(*this);
  85   }
  86   const std::string& get_tracing_category() { return tracing_category_; }
  87 };
  88
  89 } // namespace activity
  90 } // namespace kernel
  91 } // namespace simgrid
  92
  93 #endif /* SIMGRID_KERNEL_ACTIVITY_ACTIVITYIMPL_HPP */