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