examples/cpp/dag-from-dax-simple/s4u-dag-from-dax-simple.cpp

   1 /* Copyright (c) 2003-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 "simgrid/s4u.hpp"
   7
   8 XBT_LOG_NEW_DEFAULT_CATEGORY(dag_from_dax_simple, "Messages specific for this s4u example");
   9
  10 int main(int argc, char* argv[])
  11 {
  12   simgrid::s4u::Engine e(&argc, argv);
  13   e.load_platform(argv[1]);
  14
  15   std::vector<simgrid::s4u::ActivityPtr> dag = simgrid::s4u::create_DAG_from_DAX(argv[2]);
  16
  17   simgrid::s4u::Host* tremblay = e.host_by_name("Tremblay");
  18   simgrid::s4u::Host* jupiter  = e.host_by_name("Jupiter");
  19   simgrid::s4u::Host* fafard   = e.host_by_name("Fafard");
  20
  21   dynamic_cast<simgrid::s4u::Exec*>(dag[0].get())->set_host(fafard);
  22   dynamic_cast<simgrid::s4u::Exec*>(dag[1].get())->set_host(tremblay);
  23   dynamic_cast<simgrid::s4u::Exec*>(dag[2].get())->set_host(jupiter);
  24   dynamic_cast<simgrid::s4u::Exec*>(dag[3].get())->set_host(jupiter);
  25   dynamic_cast<simgrid::s4u::Exec*>(dag[8].get())->set_host(jupiter);
  26
  27   for (const auto& a : dag) {
  28     if (auto* comm = dynamic_cast<simgrid::s4u::Comm*>(a.get())) {
  29       const auto* pred = dynamic_cast<simgrid::s4u::Exec*>((*comm->get_dependencies().begin()).get());
  30       const auto* succ = dynamic_cast<simgrid::s4u::Exec*>(comm->get_successors().front().get());
  31       comm->set_source(pred->get_host())->set_destination(succ->get_host());
  32     }
  33   }
  34
  35   simgrid::s4u::Exec::on_completion_cb([](simgrid::s4u::Exec const& exec) {
  36     XBT_INFO("Exec '%s' is complete (start time: %f, finish time: %f)", exec.get_cname(),
  37              exec.get_start_time(), exec.get_finish_time());
  38   });
  39
  40   simgrid::s4u::Comm::on_completion_cb([](simgrid::s4u::Comm const& comm) {
  41     XBT_INFO("Comm '%s' is complete (start time: %f, finish time: %f)", comm.get_cname(),
  42              comm.get_start_time(), comm.get_finish_time());
  43   });
  44
  45   e.run();
  46   return 0;
  47 }