src/surf/xml/surfxml_parseplatf.cpp

   1 /* Copyright (c) 2006-2019. 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 "surf/surf.hpp"
   7 #include "src/surf/cpu_interface.hpp"
   8 #include "src/surf/network_interface.hpp"
   9 #include "src/surf/surf_interface.hpp"
  10 #include "src/surf/xml/platf_private.hpp"
  11
  12 #include <vector>
  13
  14
  15 #if SIMGRID_HAVE_LUA
  16 #include "src/bindings/lua/simgrid_lua.hpp"
  17
  18 #include <lua.h>                /* Always include this when calling Lua */
  19 #include <lauxlib.h>            /* Always include this when calling Lua */
  20 #include <lualib.h>             /* Prototype for luaL_openlibs(), */
  21 #endif
  22
  23 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(surf_parse);
  24
  25 /* Trace related stuff */
  26 XBT_PRIVATE std::unordered_map<std::string, simgrid::kernel::profile::Profile*> traces_set_list;
  27 XBT_PRIVATE std::unordered_map<std::string, std::string> trace_connect_list_host_avail;
  28 XBT_PRIVATE std::unordered_map<std::string, std::string> trace_connect_list_host_speed;
  29 XBT_PRIVATE std::unordered_map<std::string, std::string> trace_connect_list_link_avail;
  30 XBT_PRIVATE std::unordered_map<std::string, std::string> trace_connect_list_link_bw;
  31 XBT_PRIVATE std::unordered_map<std::string, std::string> trace_connect_list_link_lat;
  32
  33 void sg_platf_trace_connect(simgrid::kernel::routing::TraceConnectCreationArgs* trace_connect)
  34 {
  35   xbt_assert(traces_set_list.find(trace_connect->trace) != traces_set_list.end(),
  36              "Cannot connect trace %s to %s: trace unknown", trace_connect->trace.c_str(),
  37              trace_connect->element.c_str());
  38
  39   switch (trace_connect->kind) {
  40     case simgrid::kernel::routing::TraceConnectKind::HOST_AVAIL:
  41       trace_connect_list_host_avail.insert({trace_connect->trace, trace_connect->element});
  42       break;
  43     case simgrid::kernel::routing::TraceConnectKind::SPEED:
  44       trace_connect_list_host_speed.insert({trace_connect->trace, trace_connect->element});
  45       break;
  46     case simgrid::kernel::routing::TraceConnectKind::LINK_AVAIL:
  47       trace_connect_list_link_avail.insert({trace_connect->trace, trace_connect->element});
  48       break;
  49     case simgrid::kernel::routing::TraceConnectKind::BANDWIDTH:
  50       trace_connect_list_link_bw.insert({trace_connect->trace, trace_connect->element});
  51       break;
  52     case simgrid::kernel::routing::TraceConnectKind::LATENCY:
  53       trace_connect_list_link_lat.insert({trace_connect->trace, trace_connect->element});
  54       break;
  55     default:
  56       surf_parse_error(std::string("Cannot connect trace ") + trace_connect->trace + " to " + trace_connect->element +
  57                        ": unknown kind of trace");
  58       break;
  59   }
  60 }
  61
  62 /* This function acts as a main in the parsing area. */
  63 void parse_platform_file(const std::string& file)
  64 {
  65   const char* cfile = file.c_str();
  66   int len           = strlen(cfile);
  67   int is_lua        = len > 3 && file[len - 3] == 'l' && file[len - 2] == 'u' && file[len - 1] == 'a';
  68
  69   sg_platf_init();
  70
  71   /* Check if file extension is "lua". If so, we will use
  72    * the lua bindings to parse the platform file (since it is
  73    * written in lua). If not, we will use the (old?) XML parser
  74    */
  75   if (is_lua) {
  76 #if SIMGRID_HAVE_LUA
  77     lua_State* L = luaL_newstate();
  78     luaL_openlibs(L);
  79
  80     luaL_loadfile(L, cfile); // This loads the file without executing it.
  81
  82     /* Run the script */
  83     if (lua_pcall(L, 0, 0, 0)) {
  84       XBT_ERROR("FATAL ERROR:\n  %s: %s\n\n", "Lua call failed. Error message:", lua_tostring(L, -1));
  85       xbt_die("Lua call failed. See Log");
  86     }
  87     lua_close(L);
  88     return;
  89 #else
  90     XBT_WARN("This looks like a lua platform file, but your SimGrid was not compiled with lua. Loading it as XML.");
  91 #endif
  92   }
  93
  94   // Use XML parser
  95
  96   int parse_status;
  97
  98   /* init the flex parser */
  99   surf_parse_open(file);
 100
 101   /* Do the actual parsing */
 102   parse_status = surf_parse();
 103
 104   /* connect all profiles relative to hosts */
 105   for (auto const& elm : trace_connect_list_host_avail) {
 106     xbt_assert(traces_set_list.find(elm.first) != traces_set_list.end(), "Trace %s undefined", elm.first.c_str());
 107     simgrid::kernel::profile::Profile* profile = traces_set_list.at(elm.first);
 108
 109     simgrid::s4u::Host* host = sg_host_by_name(elm.second.c_str());
 110     xbt_assert(host, "Host %s undefined", elm.second.c_str());
 111     simgrid::surf::Cpu* cpu = host->pimpl_cpu;
 112
 113     cpu->set_state_profile(profile);
 114   }
 115
 116   for (auto const& elm : trace_connect_list_host_speed) {
 117     xbt_assert(traces_set_list.find(elm.first) != traces_set_list.end(), "Trace %s undefined", elm.first.c_str());
 118     simgrid::kernel::profile::Profile* profile = traces_set_list.at(elm.first);
 119
 120     simgrid::s4u::Host* host = sg_host_by_name(elm.second.c_str());
 121     xbt_assert(host, "Host %s undefined", elm.second.c_str());
 122     simgrid::surf::Cpu* cpu = host->pimpl_cpu;
 123
 124     cpu->set_speed_profile(profile);
 125   }
 126
 127   for (auto const& elm : trace_connect_list_link_avail) {
 128     xbt_assert(traces_set_list.find(elm.first) != traces_set_list.end(), "Trace %s undefined", elm.first.c_str());
 129     simgrid::kernel::profile::Profile* profile = traces_set_list.at(elm.first);
 130
 131     sg_link_t link = simgrid::s4u::Link::by_name(elm.second.c_str());
 132     xbt_assert(link, "Link %s undefined", elm.second.c_str());
 133     link->set_state_profile(profile);
 134   }
 135
 136   for (auto const& elm : trace_connect_list_link_bw) {
 137     xbt_assert(traces_set_list.find(elm.first) != traces_set_list.end(), "Trace %s undefined", elm.first.c_str());
 138     simgrid::kernel::profile::Profile* profile = traces_set_list.at(elm.first);
 139     sg_link_t link     = simgrid::s4u::Link::by_name(elm.second.c_str());
 140     xbt_assert(link, "Link %s undefined", elm.second.c_str());
 141     link->set_bandwidth_profile(profile);
 142   }
 143
 144   for (auto const& elm : trace_connect_list_link_lat) {
 145     xbt_assert(traces_set_list.find(elm.first) != traces_set_list.end(), "Trace %s undefined", elm.first.c_str());
 146     simgrid::kernel::profile::Profile* profile = traces_set_list.at(elm.first);
 147     sg_link_t link     = simgrid::s4u::Link::by_name(elm.second.c_str());
 148     xbt_assert(link, "Link %s undefined", elm.second.c_str());
 149     link->set_latency_profile(profile);
 150   }
 151
 152   surf_parse_close();
 153
 154   if (parse_status)
 155     surf_parse_error(std::string("Parse error in ") + file);
 156 }