include/simgrid/s4u/ConditionVariable.hpp

   1 /* Copyright (c) 2006-2018. 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_S4U_COND_VARIABLE_HPP
   7 #define SIMGRID_S4U_COND_VARIABLE_HPP
   8
   9 #include <simgrid/chrono.hpp>
  10 #include <simgrid/s4u/Mutex.hpp>
  11
  12 #include <future>
  13
  14 namespace simgrid {
  15 namespace s4u {
  16
  17 /** @brief A condition variable
  18  *  @ingroup s4u_api
  19  *
  20  *  This is a drop-in replacement of `std::condition_variable` and should respect the same
  21  *  semantic. But we currently use (only) double for both durations and
  22  *  timestamp timeouts.
  23  */
  24 class XBT_PUBLIC ConditionVariable {
  25 private:
  26   friend s_smx_cond_t;
  27   smx_cond_t cond_;
  28   explicit ConditionVariable(smx_cond_t cond) : cond_(cond) {}
  29 public:
  30   ConditionVariable(ConditionVariable const&) = delete;
  31   ConditionVariable& operator=(ConditionVariable const&) = delete;
  32
  33   friend XBT_PUBLIC void intrusive_ptr_add_ref(ConditionVariable * cond);
  34   friend XBT_PUBLIC void intrusive_ptr_release(ConditionVariable * cond);
  35   using Ptr = boost::intrusive_ptr<ConditionVariable>;
  36
  37   static Ptr createConditionVariable();
  38
  39   //  Wait functions without time:
  40
  41   void wait(MutexPtr lock);
  42   void wait(std::unique_lock<Mutex> & lock);
  43   template <class P> void wait(std::unique_lock<Mutex> & lock, P pred)
  44   {
  45     while (not pred())
  46       wait(lock);
  47   }
  48
  49   // Wait function taking a plain double as time:
  50
  51   std::cv_status wait_until(std::unique_lock<Mutex> & lock, double timeout_time);
  52   std::cv_status wait_for(std::unique_lock<Mutex> & lock, double duration);
  53   template <class P> bool wait_until(std::unique_lock<Mutex> & lock, double timeout_time, P pred)
  54   {
  55     while (not pred())
  56       if (this->wait_until(lock, timeout_time) == std::cv_status::timeout)
  57         return pred();
  58     return true;
  59   }
  60   template <class P> bool wait_for(std::unique_lock<Mutex> & lock, double duration, P pred)
  61   {
  62     return this->wait_until(lock, SIMIX_get_clock() + duration, std::move(pred));
  63   }
  64
  65   // Wait function taking a C++ style time:
  66
  67   template <class Rep, class Period, class P>
  68   bool wait_for(std::unique_lock<Mutex> & lock, std::chrono::duration<Rep, Period> duration, P pred)
  69   {
  70     auto seconds = std::chrono::duration_cast<SimulationClockDuration>(duration);
  71     return this->wait_for(lock, seconds.count(), pred);
  72   }
  73   template <class Rep, class Period>
  74   std::cv_status wait_for(std::unique_lock<Mutex> & lock, std::chrono::duration<Rep, Period> duration)
  75   {
  76     auto seconds = std::chrono::duration_cast<SimulationClockDuration>(duration);
  77     return this->wait_for(lock, seconds.count());
  78   }
  79   template <class Duration>
  80   std::cv_status wait_until(std::unique_lock<Mutex> & lock, const SimulationTimePoint<Duration>& timeout_time)
  81   {
  82     auto timeout_native = std::chrono::time_point_cast<SimulationClockDuration>(timeout_time);
  83     return this->wait_until(lock, timeout_native.time_since_epoch().count());
  84   }
  85   template <class Duration, class P>
  86   bool wait_until(std::unique_lock<Mutex> & lock, const SimulationTimePoint<Duration>& timeout_time, P pred)
  87   {
  88     auto timeout_native = std::chrono::time_point_cast<SimulationClockDuration>(timeout_time);
  89     return this->wait_until(lock, timeout_native.time_since_epoch().count(), std::move(pred));
  90   }
  91
  92   // Notify functions
  93
  94   void notify_one();
  95   void notify_all();
  96 };
  97
  98 using ConditionVariablePtr = ConditionVariable::Ptr;
  99 }
 100 } // namespace simgrid::s4u
 101
 102 #endif