Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Implement SemaphoreImpl::acquire_async (timeouts are allowed)
[simgrid.git] / src / s4u / s4u_Activity.cpp
1 /* Copyright (c) 2006-2022. 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 <simgrid/Exception.hpp>
7 #include <simgrid/s4u/Activity.hpp>
8 #include <simgrid/s4u/Comm.hpp>
9 #include <simgrid/s4u/Engine.hpp>
10 #include <simgrid/s4u/Exec.hpp>
11 #include <simgrid/s4u/Io.hpp>
12 #include <xbt/log.h>
13
14 #include "src/kernel/activity/ActivityImpl.hpp"
15 #include "src/kernel/actor/ActorImpl.hpp"
16 #include "src/kernel/actor/CommObserver.hpp"
17
18 XBT_LOG_EXTERNAL_CATEGORY(s4u);
19 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(s4u_activity, s4u, "S4U activities");
20
21 namespace simgrid {
22
23 template class xbt::Extendable<s4u::Activity>;
24
25 namespace s4u {
26
27 xbt::signal<void(Activity&)> Activity::on_veto;
28 xbt::signal<void(Activity&)> Activity::on_completion;
29
30 std::set<Activity*>* Activity::vetoed_activities_ = nullptr;
31
32 void Activity::destroy()
33 {
34   /* First Remove all dependencies */
35   while (not dependencies_.empty())
36     (*(dependencies_.begin()))->remove_successor(this);
37   while (not successors_.empty())
38     this->remove_successor(successors_.front());
39
40   cancel();
41 }
42
43 void Activity::wait_until(double time_limit)
44 {
45   double now = Engine::get_clock();
46   if (time_limit > now)
47     wait_for(time_limit - now);
48 }
49
50 Activity* Activity::wait_for(double timeout)
51 {
52   if (state_ == State::INITED)
53     vetoable_start();
54
55   if (state_ == State::FAILED) {
56     if (dynamic_cast<Comm*>(this))
57       throw NetworkFailureException(XBT_THROW_POINT, "Cannot wait for a failed comm");
58     if (dynamic_cast<Exec*>(this))
59       throw HostFailureException(XBT_THROW_POINT, "Cannot wait for a failed exec");
60     if (dynamic_cast<Io*>(this))
61       throw StorageFailureException(XBT_THROW_POINT, "Cannot wait for a failed I/O");
62     THROW_IMPOSSIBLE;
63   }
64
65   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
66   kernel::actor::ActivityWaitSimcall observer{issuer, pimpl_.get(), timeout};
67   if (kernel::actor::simcall_blocking(
68           [&observer] { observer.get_activity()->wait_for(observer.get_issuer(), observer.get_timeout()); }, &observer))
69     throw TimeoutException(XBT_THROW_POINT, "Timeouted");
70   complete(State::FINISHED);
71   return this;
72 }
73
74 bool Activity::test()
75 {
76   xbt_assert(state_ == State::INITED || state_ == State::STARTED || state_ == State::STARTING ||
77              state_ == State::CANCELED || state_ == State::FINISHED);
78
79   if (state_ == State::CANCELED || state_ == State::FINISHED)
80     return true;
81
82   if (state_ == State::INITED || state_ == State::STARTING)
83     this->vetoable_start();
84
85   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
86   kernel::actor::ActivityTestSimcall observer{issuer, pimpl_.get()};
87   if (kernel::actor::simcall([&observer] { return observer.get_activity()->test(observer.get_issuer()); }, &observer)) {
88     complete(State::FINISHED);
89     return true;
90   }
91   return false;
92 }
93
94 ssize_t Activity::test_any(const std::vector<ActivityPtr>& activities)
95 {
96   std::vector<kernel::activity::ActivityImpl*> ractivities(activities.size());
97   std::transform(begin(activities), end(activities), begin(ractivities),
98                  [](const ActivityPtr& act) { return act->pimpl_.get(); });
99
100   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
101   kernel::actor::ActivityTestanySimcall observer{issuer, ractivities};
102   ssize_t changed_pos = kernel::actor::simcall(
103       [&observer] {
104         return kernel::activity::ActivityImpl::test_any(observer.get_issuer(), observer.get_activities());
105       },
106       &observer);
107   if (changed_pos != -1)
108     activities.at(changed_pos)->complete(State::FINISHED);
109   return changed_pos;
110 }
111
112 ssize_t Activity::wait_any_for(const std::vector<ActivityPtr>& activities, double timeout)
113 {
114   std::vector<kernel::activity::ActivityImpl*> ractivities(activities.size());
115   std::transform(begin(activities), end(activities), begin(ractivities),
116                  [](const ActivityPtr& activity) { return activity->pimpl_.get(); });
117
118   kernel::actor::ActorImpl* issuer = kernel::actor::ActorImpl::self();
119   kernel::actor::ActivityWaitanySimcall observer{issuer, ractivities, timeout};
120   ssize_t changed_pos = kernel::actor::simcall_blocking(
121       [&observer] {
122         kernel::activity::ActivityImpl::wait_any_for(observer.get_issuer(), observer.get_activities(),
123                                                      observer.get_timeout());
124       },
125       &observer);
126   if (changed_pos != -1)
127     activities.at(changed_pos)->complete(State::FINISHED);
128   return changed_pos;
129 }
130
131 Activity* Activity::cancel()
132 {
133   kernel::actor::simcall([this] {
134     XBT_HERE();
135     if (pimpl_)
136       pimpl_->cancel();
137   });
138   complete(State::CANCELED);
139   return this;
140 }
141
142 Activity* Activity::suspend()
143 {
144   if (suspended_)
145     return this; // Already suspended
146   suspended_ = true;
147
148   if (state_ == State::STARTED)
149     pimpl_->suspend();
150
151   return this;
152 }
153
154 Activity* Activity::resume()
155 {
156   if (not suspended_)
157     return this; // nothing to restore when it's not suspended
158
159   if (state_ == State::STARTED)
160     pimpl_->resume();
161
162   return this;
163 }
164
165 const char* Activity::get_state_str() const
166 {
167   return to_c_str(state_);
168 }
169
170 double Activity::get_remaining() const
171 {
172   if (state_ == State::INITED || state_ == State::STARTING)
173     return remains_;
174   else
175     return pimpl_->get_remaining();
176 }
177 double Activity::get_start_time() const
178 {
179   return pimpl_->get_start_time();
180 }
181 double Activity::get_finish_time() const
182 {
183   return pimpl_->get_finish_time();
184 }
185
186 Activity* Activity::set_remaining(double remains)
187 {
188   xbt_assert(state_ == State::INITED || state_ == State::STARTING,
189              "Cannot change the remaining amount of work once the Activity is started");
190   remains_ = remains;
191   return this;
192 }
193
194 } // namespace s4u
195 } // namespace simgrid