examples/cpp/dag-from-dot/s4u-dag-from-dot.cpp

   1 /* simple test trying to load a DOT file.                                   */
   2
   3 /* Copyright (c) 2010-2022. The SimGrid Team. All rights reserved.          */
   4
   5 /* This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the license (GNU LGPL) which comes with this package. */
   7
   8 #include "simgrid/s4u.hpp"
   9 #include <stdio.h>
  10
  11 XBT_LOG_NEW_DEFAULT_CATEGORY(dag_from_dot, "Logging specific to this example");
  12 namespace sg4 = simgrid::s4u;
  13
  14 int main(int argc, char** argv)
  15 {
  16   sg4::Engine e(&argc, argv);
  17   e.load_platform(argv[1]);
  18
  19   std::vector<sg4::ActivityPtr> dag = sg4::create_DAG_from_dot(argv[2]);
  20
  21   if (dag.empty()) {
  22     XBT_CRITICAL("No dot loaded. Do you have a cycle in your graph?");
  23     exit(2);
  24   }
  25
  26   XBT_INFO("--------- Display all activities of the loaded DAG -----------");
  27   for (const auto& a : dag) {
  28     std::string type = "an Exec";
  29     std::string task = "flops to execute";
  30     if (dynamic_cast<sg4::Comm*>(a.get()) != nullptr) {
  31       type = "a Comm";
  32       task = "bytes to transfer";
  33     }
  34     XBT_INFO("'%s' is %s: %.0f %s. Dependencies: %s; Ressources: %s", a->get_cname(), type.c_str(), a->get_remaining(),
  35              task.c_str(), (a->dependencies_solved() ? "solved" : "NOT solved"),
  36              (a->is_assigned() ? "assigned" : "NOT assigned"));
  37   }
  38
  39   XBT_INFO("------------------- Schedule tasks ---------------------------");
  40   auto hosts = e.get_all_hosts();
  41   auto count = e.get_host_count();
  42   int cursor = 0;
  43   // Schedule end first
  44   static_cast<sg4::Exec*>(dag.back().get())->set_host(hosts[0]);
  45
  46   for (const auto& a : dag) {
  47     auto* exec = dynamic_cast<sg4::Exec*>(a.get());
  48     if (exec != nullptr && exec->get_name() != "end") {
  49       exec->set_host(hosts[cursor % count]);
  50       cursor++;
  51     }
  52     if (auto* comm = dynamic_cast<sg4::Comm*>(a.get())) {
  53       auto pred = dynamic_cast<sg4::Exec*>((*comm->get_dependencies().begin()).get());
  54       auto succ = dynamic_cast<sg4::Exec*>(comm->get_successors().front().get());
  55       comm->set_source(pred->get_host())->set_destination(succ->get_host());
  56     }
  57   }
  58
  59   XBT_INFO("------------------- Run the schedule -------------------------");
  60   e.run();
  61
  62   XBT_INFO("-------------- Summary of executed schedule ------------------");
  63   for (const auto& a : dag) {
  64     if (const auto* exec = dynamic_cast<sg4::Exec*>(a.get())) {
  65       XBT_INFO("[%f->%f] '%s' executed on %s", exec->get_start_time(), exec->get_finish_time(), exec->get_cname(),
  66                exec->get_host()->get_cname());
  67     }
  68     if (const auto* comm = dynamic_cast<sg4::Comm*>(a.get())) {
  69       XBT_INFO("[%f->%f] '%s' transferred from %s to %s", comm->get_start_time(), comm->get_finish_time(),
  70                comm->get_cname(), comm->get_source()->get_cname(), comm->get_destination()->get_cname());
  71     }
  72   }
  73   return 0;
  74 }