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 }
  36
  37 ActivityPtr ActivitySet::test_any()
  38 {
  39   std::vector<kernel::activity::ActivityImpl*> act_impls(activities_.size());
  40   std::transform(begin(activities_), end(activities_), begin(act_impls),
  41                  [](const ActivityPtr& act) { return act->pimpl_.get(); });
  42
  43   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
  44   kernel::actor::ActivityTestanySimcall observer{issuer, act_impls, "test_any"};
  45   ssize_t changed_pos = kernel::actor::simcall_answered(
  46       [&observer] {
  47         return kernel::activity::ActivityImpl::test_any(observer.get_issuer(), observer.get_activities());
  48       },
  49       &observer);
  50   if (changed_pos == -1)
  51     return ActivityPtr(nullptr);
  52
  53   auto ret = activities_.at(changed_pos);
  54   erase(ret);
  55   ret->complete(Activity::State::FINISHED);
  56   return ret;
  57 }
  58
  59 ActivityPtr ActivitySet::wait_any_for(double timeout)
  60 {
  61   std::vector<kernel::activity::ActivityImpl*> act_impls(activities_.size());
  62   std::transform(begin(activities_), end(activities_), begin(act_impls),
  63                  [](const ActivityPtr& activity) { return activity->pimpl_.get(); });
  64
  65   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
  66   kernel::actor::ActivityWaitanySimcall observer{issuer, act_impls, timeout, "wait_any_for"};
  67   ssize_t changed_pos = kernel::actor::simcall_blocking(
  68       [&observer] {
  69         kernel::activity::ActivityImpl::wait_any_for(observer.get_issuer(), observer.get_activities(),
  70                                                      observer.get_timeout());
  71       },
  72       &observer);
  73   xbt_assert(changed_pos != -1,
  74              "ActivityImpl::wait_any_for is not supposed to return -1 but instead to raise exceptions");
  75
  76   auto ret = activities_.at(changed_pos);
  77   erase(ret);
  78   ret->complete(Activity::State::FINISHED);
  79   return ret;
  80 }
  81
  82 ActivityPtr ActivitySet::get_failed_activity()
  83 {
  84   if (failed_activities_.empty())
  85     return ActivityPtr(nullptr);
  86   auto ret = failed_activities_.back();
  87   failed_activities_.pop_back();
  88   return ret;
  89 }
  90
  91 }; // namespace simgrid::s4u