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_ACTIVITYIMPL_HPP
   7 #define SIMGRID_KERNEL_ACTIVITY_ACTIVITYIMPL_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   bool test();
  48
  49   virtual void suspend();
  50   virtual void resume();
  51   virtual void cancel();
  52
  53   virtual void post() = 0; // Called by the main loop when the activity is marked as terminated or failed by its model.
  54                            // Setups the status, clean things up, and call finish()
  55   virtual void finish() = 0; // Unlock all simcalls blocked on that activity, either because it was marked as done by
  56                              // the model or because it terminated without waiting for the model
  57
  58   virtual void register_simcall(smx_simcall_t simcall);
  59   void clean_action();
  60   virtual double get_remaining() const;
  61   // Support for the boost::intrusive_ptr<ActivityImpl> datatype
  62   friend XBT_PUBLIC void intrusive_ptr_add_ref(ActivityImpl* activity);
  63   friend XBT_PUBLIC void intrusive_ptr_release(ActivityImpl* activity);
  64
  65   static xbt::signal<void(ActivityImpl const&)> on_suspended;
  66   static xbt::signal<void(ActivityImpl const&)> on_resumed;
  67 };
  68
  69 template <class AnyActivityImpl> class ActivityImpl_T : public ActivityImpl {
  70 private:
  71   std::string name_             = "";
  72   std::string tracing_category_ = "";
  73
  74 public:
  75   AnyActivityImpl& set_name(const std::string& name)
  76   {
  77     name_ = name;
  78     return static_cast<AnyActivityImpl&>(*this);
  79   }
  80   const std::string& get_name() { return name_; }
  81   const char* get_cname() { return name_.c_str(); }
  82
  83   AnyActivityImpl& set_tracing_category(const std::string& category)
  84   {
  85     tracing_category_ = category;
  86     return static_cast<AnyActivityImpl&>(*this);
  87   }
  88   const std::string& get_tracing_category() { return tracing_category_; }
  89 };
  90
  91 } // namespace activity
  92 } // namespace kernel
  93 } // namespace simgrid
  94
  95 #endif /* SIMGRID_KERNEL_ACTIVITY_ACTIVITYIMPL_HPP */