teshsuite/simdag/flatifier/flatifier.cpp

   1 /* Copyright (c) 2008-2017. 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 <xbt/xbt_os_time.h>
   7
   8 #include "simgrid/s4u/Engine.hpp"
   9 #include "simgrid/s4u/Host.hpp"
  10
  11 #include "simgrid/simdag.h"
  12
  13 #include "src/kernel/routing/NetPoint.hpp"
  14 #include "src/surf/network_interface.hpp"
  15
  16
  17 XBT_LOG_NEW_DEFAULT_CATEGORY(flatifier, "Logging specific to this platform parsing tool");
  18
  19 static int name_compare_hosts(const void *n1, const void *n2)
  20 {
  21   return std::strcmp(sg_host_get_name(*(sg_host_t *) n1), sg_host_get_name(*(sg_host_t *) n2));
  22 }
  23
  24 static int name_compare_links(const void *n1, const void *n2)
  25 {
  26   return std::strcmp(sg_link_name(*(SD_link_t *) n1),sg_link_name(*(SD_link_t *) n2));
  27 }
  28
  29 static bool parse_cmdline(int* timings, char** platformFile, int argc, char** argv)
  30 {
  31   bool parse_ok = true;
  32   for (int i = 1; i < argc; i++) {
  33     if (std::strlen(argv[i]) > 1 && argv[i][0] == '-' && argv[i][1] == '-') {
  34       if (not std::strcmp(argv[i], "--timings")) {
  35         *timings = 1;
  36       } else {
  37         parse_ok = false;
  38         break;
  39       }
  40     } else {
  41       *platformFile = argv[i];
  42     }
  43   }
  44   return parse_ok;
  45 }
  46
  47 static void create_environment(xbt_os_timer_t parse_time, const char *platformFile)
  48 {
  49   try {
  50     xbt_os_cputimer_start(parse_time);
  51     SD_create_environment(platformFile);
  52     xbt_os_cputimer_stop(parse_time);
  53   }
  54   catch (std::exception& e) {
  55     xbt_die("Error while loading %s: %s", platformFile, e.what());
  56   }
  57 }
  58
  59 static void dump_platform()
  60 {
  61   int version = 4;
  62   xbt_dict_t props = nullptr;
  63   xbt_dict_cursor_t cursor = nullptr;
  64   char* key;
  65   char* data;
  66
  67   std::printf("<?xml version='1.0'?>\n");
  68   std::printf("<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid/simgrid.dtd\">\n");
  69   std::printf("<platform version=\"%d\">\n", version);
  70   std::printf("<AS id=\"AS0\" routing=\"Full\">\n");
  71
  72   // Hosts
  73   unsigned int totalHosts = sg_host_count();
  74   sg_host_t* hosts        = sg_host_list();
  75   std::qsort((void*)hosts, totalHosts, sizeof(sg_host_t), name_compare_hosts);
  76
  77   for (unsigned int i = 0; i < totalHosts; i++) {
  78     std::printf("  <host id=\"%s\" speed=\"%.0f\"", hosts[i]->cname(), sg_host_speed(hosts[i]));
  79     props = sg_host_get_properties(hosts[i]);
  80     if (hosts[i]->coreCount() > 1) {
  81       std::printf(" core=\"%d\"", hosts[i]->coreCount());
  82     }
  83     if (props && not xbt_dict_is_empty(props)) {
  84       std::printf(">\n");
  85       xbt_dict_foreach (props, cursor, key, data) {
  86         std::printf("    <prop id=\"%s\" value=\"%s\"/>\n", key, data);
  87       }
  88       std::printf("  </host>\n");
  89     } else {
  90       std::printf("/>\n");
  91     }
  92   }
  93
  94   // Routers
  95   std::vector<simgrid::kernel::routing::NetPoint*> netcardList;
  96   simgrid::s4u::Engine::instance()->netpointList(&netcardList);
  97   std::sort(netcardList.begin(), netcardList.end(),
  98             [](simgrid::kernel::routing::NetPoint* a, simgrid::kernel::routing::NetPoint* b) {
  99               return a->name() < b->name();
 100             });
 101
 102   for (auto srcCard : netcardList)
 103     if (srcCard->isRouter())
 104       std::printf("  <router id=\"%s\"/>\n", srcCard->cname());
 105
 106   // Links
 107   unsigned int totalLinks    = sg_link_count();
 108   simgrid::s4u::Link** links = sg_link_list();
 109
 110   std::qsort((void*)links, totalLinks, sizeof(SD_link_t), name_compare_links);
 111
 112   for (unsigned int i = 0; i < totalLinks; i++) {
 113     simgrid::s4u::Link* link = links[i];
 114     std::printf("  <link id=\"");
 115
 116     std::printf("%s\" bandwidth=\"%.0f\" latency=\"%.9f\"", link->name(), link->bandwidth(), link->latency());
 117     if (sg_link_is_shared(link)) {
 118       std::printf("/>\n");
 119     } else {
 120       std::printf(" sharing_policy=\"FATPIPE\"/>\n");
 121     }
 122   }
 123
 124   for (unsigned int it_src = 0; it_src < totalHosts; it_src++) { // Routes from host
 125     simgrid::s4u::Host* host1                      = hosts[it_src];
 126     simgrid::kernel::routing::NetPoint* netcardSrc = host1->pimpl_netpoint;
 127     for (unsigned int it_dst = 0; it_dst < totalHosts; it_dst++) { // Routes to host
 128       simgrid::s4u::Host* host2 = hosts[it_dst];
 129       std::vector<simgrid::surf::LinkImpl*> route;
 130       simgrid::kernel::routing::NetPoint* netcardDst = host2->pimpl_netpoint;
 131       simgrid::kernel::routing::NetZoneImpl::getGlobalRoute(netcardSrc, netcardDst, &route, nullptr);
 132       if (not route.empty()) {
 133         std::printf("  <route src=\"%s\" dst=\"%s\">\n  ", host1->cname(), host2->cname());
 134         for (auto link : route)
 135           std::printf("<link_ctn id=\"%s\"/>", link->cname());
 136         std::printf("\n  </route>\n");
 137       }
 138     }
 139     for (auto netcardDst : netcardList) { // to router
 140       if (netcardDst->isRouter()) {
 141         std::printf("  <route src=\"%s\" dst=\"%s\">\n  ", host1->cname(), netcardDst->cname());
 142         std::vector<simgrid::surf::LinkImpl*> route;
 143         simgrid::kernel::routing::NetZoneImpl::getGlobalRoute(netcardSrc, netcardDst, &route, nullptr);
 144         for (auto link : route)
 145           std::printf("<link_ctn id=\"%s\"/>", link->cname());
 146         std::printf("\n  </route>\n");
 147       }
 148     }
 149   }
 150
 151   for (auto value1 : netcardList) { // Routes from router
 152     if (value1->isRouter()) {
 153       for (auto value2 : netcardList) { // to router
 154         if (value2->isRouter()) {
 155           std::printf("  <route src=\"%s\" dst=\"%s\">\n  ", value1->cname(), value2->cname());
 156           std::vector<simgrid::surf::LinkImpl*> route;
 157           simgrid::kernel::routing::NetZoneImpl::getGlobalRoute(value1, value2, &route, nullptr);
 158           for (auto link : route)
 159             std::printf("<link_ctn id=\"%s\"/>", link->cname());
 160           std::printf("\n  </route>\n");
 161         }
 162       }
 163       for (unsigned int it_dst = 0; it_dst < totalHosts; it_dst++) { // Routes to host
 164         simgrid::s4u::Host* host2 = hosts[it_dst];
 165         std::printf("  <route src=\"%s\" dst=\"%s\">\n  ", value1->cname(), host2->cname());
 166         std::vector<simgrid::surf::LinkImpl*> route;
 167         simgrid::kernel::routing::NetPoint* netcardDst = host2->pimpl_netpoint;
 168         simgrid::kernel::routing::NetZoneImpl::getGlobalRoute(value1, netcardDst, &route, nullptr);
 169         for (auto link : route)
 170           std::printf("<link_ctn id=\"%s\"/>", link->cname());
 171         std::printf("\n  </route>\n");
 172       }
 173     }
 174   }
 175
 176   std::printf("</AS>\n");
 177   std::printf("</platform>\n");
 178   std::free(hosts);
 179   std::free(links);
 180 }
 181
 182 int main(int argc, char** argv)
 183 {
 184   char* platformFile = nullptr;
 185   int timings        = 0;
 186
 187   xbt_os_timer_t parse_time = xbt_os_timer_new();
 188
 189   SD_init(&argc, argv);
 190
 191   xbt_assert(parse_cmdline(&timings, &platformFile, argc, argv) && platformFile,
 192              "Invalid command line arguments: expected [--timings] platformFile");
 193
 194   XBT_DEBUG("%d,%s", timings, platformFile);
 195
 196   create_environment(parse_time, platformFile);
 197
 198   if (timings) {
 199     XBT_INFO("Parsing time: %fs (%zu hosts, %d links)", xbt_os_timer_elapsed(parse_time), sg_host_count(),
 200              sg_link_count());
 201   } else {
 202     dump_platform();
 203   }
 204
 205   SD_exit();
 206   xbt_os_timer_free(parse_time);
 207
 208   return 0;
 209 }