teshsuite/simdag/flatifier/flatifier.cpp

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