src/s4u/s4u_ActivitySet.cpp

   1 /* Copyright (c) 2023-. 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 #include "src/kernel/activity/ActivityImpl.hpp"
   7 #include "src/kernel/actor/ActorImpl.hpp"
   8 #include "src/kernel/actor/CommObserver.hpp"
   9 #include <simgrid/Exception.hpp>
  10 #include <simgrid/s4u/ActivitySet.hpp>
  11 #include <simgrid/s4u/Engine.hpp>
  12
  13 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(s4u_activityset, s4u_activity, "S4U set of activities");
  14
  15 namespace simgrid {
  16
  17 template class xbt::Extendable<s4u::ActivitySet>;
  18
  19 namespace s4u {
  20
  21 void ActivitySet::erase(ActivityPtr a)
  22 {
  23   for (auto it = activities_.begin(); it != activities_.end(); it++)
  24     if (*it == a) {
  25       activities_.erase(it);
  26       return;
  27     }
  28 }
  29
  30 void ActivitySet::wait_all_for(double timeout)
  31 {
  32   if (timeout < 0.0) {
  33     for (const auto& act : activities_)
  34       act->wait();
  35
  36   } else {
  37
  38     double deadline = Engine::get_clock() + timeout;
  39     for (const auto& act : activities_)
  40       act->wait_until(deadline);
  41   }
  42 }
  43
  44 ActivityPtr ActivitySet::test_any()
  45 {
  46   std::vector<kernel::activity::ActivityImpl*> act_impls(activities_.size());
  47   std::transform(begin(activities_), end(activities_), begin(act_impls),
  48                  [](const ActivityPtr& act) { return act->pimpl_.get(); });
  49
  50   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
  51   kernel::actor::ActivityTestanySimcall observer{issuer, act_impls, "test_any"};
  52   ssize_t changed_pos = kernel::actor::simcall_answered(
  53       [&observer] {
  54         return kernel::activity::ActivityImpl::test_any(observer.get_issuer(), observer.get_activities());
  55       },
  56       &observer);
  57   if (changed_pos == -1)
  58     return ActivityPtr(nullptr);
  59
  60   auto ret = activities_.at(changed_pos);
  61   erase(ret);
  62   ret->complete(Activity::State::FINISHED);
  63   return ret;
  64 }
  65
  66 ActivityPtr ActivitySet::wait_any_for(double timeout)
  67 {
  68   std::vector<kernel::activity::ActivityImpl*> act_impls(activities_.size());
  69   std::transform(begin(activities_), end(activities_), begin(act_impls),
  70                  [](const ActivityPtr& activity) { return activity->pimpl_.get(); });
  71
  72   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
  73   kernel::actor::ActivityWaitanySimcall observer{issuer, act_impls, timeout, "wait_any_for"};
  74   ssize_t changed_pos = kernel::actor::simcall_blocking(
  75       [&observer] {
  76         kernel::activity::ActivityImpl::wait_any_for(observer.get_issuer(), observer.get_activities(),
  77                                                      observer.get_timeout());
  78       },
  79       &observer);
  80   if (changed_pos == -1)
  81     throw TimeoutException(XBT_THROW_POINT, "Timeouted");
  82
  83   auto ret = activities_.at(changed_pos);
  84   erase(ret);
  85   ret->complete(Activity::State::FINISHED);
  86   return ret;
  87 }
  88
  89 ActivityPtr ActivitySet::get_failed_activity()
  90 {
  91   if (failed_activities_.empty())
  92     return ActivityPtr(nullptr);
  93   auto ret = failed_activities_.back();
  94   failed_activities_.pop_back();
  95   return ret;
  96 }
  97
  98 } // namespace s4u
  99 } // namespace simgrid