+ s4u::Comm::on_send_cb([](s4u::Comm const& c) {
+ std::string pid = instr_pid(*s4u::Actor::self());
+ if (pid == "-0") //Comm is launched directly by Maestro, use the host as container
+ Container::by_name(c.get_source()->get_name())->get_state("HOST_STATE")->push_event("send");
+ else
+ Container::by_name(pid)->get_state("ACTOR_STATE")->push_event("send");
+ });
+ s4u::Comm::on_recv_cb([](s4u::Comm const& c) {
+ std::string pid = instr_pid(*s4u::Actor::self());
+ if (pid == "-0") //Comm is launched directly by Maestro, use the host as container
+ Container::by_name(c.get_destination()->get_name())->get_state("HOST_STATE")->push_event("receive");
+ else
+ Container::by_name(pid)->get_state("ACTOR_STATE")->push_event("receive");
+ });
+ s4u::Actor::on_host_change_cb(on_actor_host_change);
+ }
+
+ if (TRACE_smpi_is_enabled() && TRACE_smpi_is_computing()) {
+ s4u::Exec::on_start_cb([](s4u::Exec const& exec) {
+ Container::by_name("rank-" + std::to_string(s4u::Actor::self()->get_pid()))
+ ->get_state("MPI_STATE")
+ ->push_event("computing", new CpuTIData("compute", exec.get_cost()));
+ });
+ s4u::Exec::on_completion_cb([](const s4u::Exec&) {
+ Container::by_name("rank-" + std::to_string(s4u::Actor::self()->get_pid()))->get_state("MPI_STATE")->pop_event();