src/instr/jedule/jedule_sd_binding.cpp

   1 /* Copyright (c) 2010-2021. 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/jedule/jedule.hpp"
   7 #include "simgrid/s4u/Engine.hpp"
   8 #include "simgrid/s4u/NetZone.hpp"
   9 #include "src/simdag/simdag_private.hpp"
  10 #include "xbt/virtu.h"
  11
  12 #if SIMGRID_HAVE_JEDULE
  13
  14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(jed_sd, sd, "Jedule SimDag binding");
  15
  16 jedule_t my_jedule;
  17
  18 void jedule_log_sd_event(const_SD_task_t task)
  19 {
  20   xbt_assert(task != nullptr);
  21
  22   simgrid::jedule::Event event(task->get_name(), task->get_start_time(), task->get_finish_time(), "SD");
  23   event.add_resources(*task->get_allocation());
  24   my_jedule->add_event(event);
  25 }
  26
  27 void jedule_sd_init()
  28 {
  29   const_sg_netzone_t root_comp = simgrid::s4u::Engine::get_instance()->get_netzone_root();
  30   XBT_DEBUG("root name %s\n", root_comp->get_cname());
  31
  32   my_jedule = new simgrid::jedule::Jedule(root_comp->get_name());
  33 }
  34
  35 void jedule_sd_exit()
  36 {
  37   delete my_jedule;
  38 }
  39
  40 void jedule_sd_dump(const char * filename)
  41 {
  42   if (my_jedule) {
  43     std::string fname;
  44     if (not filename) {
  45       fname = simgrid::xbt::binary_name + ".jed";
  46     } else {
  47       fname = filename;
  48     }
  49
  50     FILE* fh = fopen(fname.c_str(), "w");
  51     xbt_assert(fh != nullptr, "Failed to open file: %s", fname.c_str());
  52
  53     my_jedule->write_output(fh);
  54
  55     fclose(fh);
  56   }
  57 }
  58 #endif