src/kernel/routing/VivaldiZone.cpp

   1 /* Copyright (c) 2013-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/kernel/routing/VivaldiZone.hpp"
   7 #include "simgrid/kernel/routing/NetPoint.hpp"
   8 #include "simgrid/s4u/Engine.hpp"
   9 #include "simgrid/s4u/Host.hpp"
  10 #include "src/surf/network_interface.hpp"
  11 #include "surf/surf.hpp"
  12
  13 #include <boost/algorithm/string.hpp>
  14
  15 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_route_vivaldi, surf, "Routing part of surf");
  16
  17 namespace simgrid {
  18 namespace kernel {
  19 namespace routing {
  20
  21 namespace vivaldi {
  22
  23 xbt::Extension<NetPoint, Coords> Coords::EXTENSION_ID;
  24
  25 Coords::Coords(NetPoint* netpoint, const std::string& coordStr)
  26 {
  27   if (not Coords::EXTENSION_ID.valid())
  28     Coords::EXTENSION_ID = NetPoint::extension_create<Coords>();
  29
  30   std::vector<std::string> string_values;
  31   boost::split(string_values, coordStr, boost::is_any_of(" "));
  32   xbt_assert(string_values.size() == 3, "Coordinates of %s must have 3 dimensions", netpoint->get_cname());
  33
  34   for (auto const& str : string_values)
  35     try {
  36       coords.push_back(std::stod(str));
  37     } catch (std::invalid_argument const& ia) {
  38       throw std::invalid_argument(std::string("Invalid coordinate: ") + ia.what());
  39     }
  40   coords.shrink_to_fit();
  41
  42   netpoint->extension_set<Coords>(this);
  43   XBT_DEBUG("Coords of %s %p: %s", netpoint->get_cname(), netpoint, coordStr.c_str());
  44 }
  45 } // namespace vivaldi
  46
  47 static inline double euclidean_dist_comp(double src_coord, double dst_coord)
  48 {
  49   return (src_coord - dst_coord) * (src_coord - dst_coord);
  50 }
  51
  52 static const std::vector<double>& netpoint_get_coords(const NetPoint* np)
  53 {
  54   const auto* coords = np->extension<vivaldi::Coords>();
  55   xbt_assert(coords, "Please specify the Vivaldi coordinates of %s %s (%p)",
  56              (np->is_netzone() ? "Netzone" : (np->is_host() ? "Host" : "Router")), np->get_cname(), np);
  57   return coords->coords;
  58 }
  59
  60 void VivaldiZone::set_peer_link(NetPoint* netpoint, double bw_in, double bw_out)
  61 {
  62   xbt_assert(netpoint->get_englobing_zone() == this,
  63              "Cannot add a peer link to a netpoint that is not in this netzone");
  64
  65   std::string link_up        = "link_" + netpoint->get_name() + "_UP";
  66   std::string link_down      = "link_" + netpoint->get_name() + "_DOWN";
  67   const auto* linkUp         = create_link(link_up, {bw_out})->seal();
  68   const auto* linkDown       = create_link(link_down, {bw_in})->seal();
  69   add_route(netpoint, nullptr, nullptr, nullptr, {s4u::LinkInRoute(linkUp)}, false);
  70   add_route(nullptr, netpoint, nullptr, nullptr, {s4u::LinkInRoute(linkDown)}, false);
  71 }
  72
  73 void VivaldiZone::get_local_route(const NetPoint* src, const NetPoint* dst, Route* route, double* lat)
  74 {
  75   XBT_DEBUG("vivaldi getLocalRoute from '%s'[%lu] '%s'[%lu]", src->get_cname(), src->id(), dst->get_cname(), dst->id());
  76   auto* engine = s4u::Engine::get_instance();
  77   if (src->is_netzone()) {
  78     std::string srcName = "router_" + src->get_name();
  79     std::string dstName = "router_" + dst->get_name();
  80     route->gw_src_      = engine->netpoint_by_name_or_null(srcName);
  81     route->gw_dst_      = engine->netpoint_by_name_or_null(dstName);
  82   }
  83
  84   StarZone::get_local_route(src, dst, route, lat);
  85   /* Compute the extra latency due to the euclidean distance if needed */
  86   if (lat) {
  87     std::vector<double> srcCoords = netpoint_get_coords(src);
  88     std::vector<double> dstCoords = netpoint_get_coords(dst);
  89
  90     double euclidean_dist =
  91         sqrt(euclidean_dist_comp(srcCoords[0], dstCoords[0]) + euclidean_dist_comp(srcCoords[1], dstCoords[1])) +
  92         fabs(srcCoords[2]) + fabs(dstCoords[2]);
  93
  94     XBT_DEBUG("Updating latency %f += %f", *lat, euclidean_dist);
  95     *lat += euclidean_dist / 1000.0; // From .ms to .s
  96   }
  97 }
  98
  99 } // namespace routing
 100 } // namespace kernel
 101
 102 namespace s4u {
 103 NetZone* create_vivaldi_zone(const std::string& name)
 104 {
 105   return (new kernel::routing::VivaldiZone(name))->get_iface();
 106 }
 107 } // namespace s4u
 108
 109 } // namespace simgrid