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