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