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