Do the same for IoImpl as for ExecImpl when timeout on completion

[simgrid.git] / src / kernel / EngineImpl.cpp

diff --git a/src/kernel/EngineImpl.cpp b/src/kernel/EngineImpl.cpp
index 1956f49336817bd772381586f7bc8324f0df2119..e45b5e027395a0af3f2a808c41842adac26e25e8 100644 (file)
--- a/src/kernel/EngineImpl.cpp
+++ b/src/kernel/EngineImpl.cpp
@@ -53,5 +53,29 @@ void EngineImpl::register_default(const actor::ActorCodeFactory& code)
   default_function = code;
 }
 
+void EngineImpl::add_model_ptask(simgrid::kernel::resource::Model::Type type, simgrid::kernel::resource::Model* model,
+                                 bool is_default)
+{
+  if (is_default)
+    models_by_type_[type].insert(models_by_type_[type].begin(), model);
+  else
+    models_by_type_[type].push_back(model);
+}
+
+void EngineImpl::add_model(simgrid::kernel::resource::Model::Type type,
+                           std::unique_ptr<simgrid::kernel::resource::Model> model, bool is_default)
+{
+  add_model_ptask(type, model.get(), is_default);
+  models_.push_back(std::move(model));
+}
+
+simgrid::kernel::resource::Model* EngineImpl::get_default_model(simgrid::kernel::resource::Model::Type type)
+{
+  if (models_by_type_[type].size() > 0)
+    return models_by_type_[type][0];
+  else
+    return nullptr;
+}
+
 } // namespace kernel
 } // namespace simgrid