allow to call s4u::Exec->setHost() after its start, to migrate it

[simgrid.git] / src / kernel / activity / ExecImpl.cpp

diff --git a/src/kernel/activity/ExecImpl.cpp b/src/kernel/activity/ExecImpl.cpp
index 0484cc7..d233924 100644 (file)
--- a/src/kernel/activity/ExecImpl.cpp
+++ b/src/kernel/activity/ExecImpl.cpp
@@ -3,11 +3,16 @@
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
 
-#include "simgrid/s4u/Host.hpp"
+#include "simgrid/modelchecker.h"
+#include "src/mc/mc_replay.hpp"
 
 #include "src/kernel/activity/ExecImpl.hpp"
 #include "src/simix/smx_host_private.hpp"
 #include "src/surf/surf_interface.hpp"
+#include "src/surf/cpu_interface.hpp"
+
+
+#include "simgrid/s4u/Host.hpp"
 
 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(simix_process);
 
@@ -87,8 +92,34 @@ void simgrid::kernel::activity::ExecImpl::post()
     SIMIX_execution_finish(this);
 }
 
+simgrid::kernel::activity::ActivityImpl*
+simgrid::kernel::activity::ExecImpl::migrate(simgrid::s4u::Host* to)
+{
+
+  if (not MC_is_active() && not MC_record_replay_is_active()) {
+    surf_action_t oldAction = this->surfAction_;
+    surf_action_t newAction = to->pimpl_cpu->execution_start(oldAction->getCost());
+    newAction->setRemains(oldAction->getRemains());
+    newAction->setData(this);
+    newAction->setSharingWeight(oldAction->getPriority());
+
+    // FIXME: the user-defined bound seem to not be kept by LMM, that seem to overwrite it for the multi-core modeling.
+    // I hope that the user did not provide any.
+
+    oldAction->setData(nullptr);
+    oldAction->cancel();
+    oldAction->unref();
+    this->surfAction_ = newAction;
+  }
+
+  onMigration(this, to);
+  return this;
+}
+
+
 /*************
  * Callbacks *
  *************/
 simgrid::xbt::signal<void(simgrid::kernel::activity::ExecImplPtr)> simgrid::kernel::activity::ExecImpl::onCreation;
 simgrid::xbt::signal<void(simgrid::kernel::activity::ExecImplPtr)> simgrid::kernel::activity::ExecImpl::onCompletion;
+simgrid::xbt::signal<void(simgrid::kernel::activity::ExecImplPtr, simgrid::s4u::Host*)> simgrid::kernel::activity::ExecImpl::onMigration;