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