src/kernel/routing/VivaldiZone.cpp

   1 /* Copyright (c) 2013-2018. 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 namespace vivaldi {
  22 simgrid::xbt::Extension<NetPoint, Coords> Coords::EXTENSION_ID;
  23
  24 Coords::Coords(NetPoint* netpoint, std::string coordStr)
  25 {
  26   if (not Coords::EXTENSION_ID.valid())
  27     Coords::EXTENSION_ID = NetPoint::extension_create<Coords>();
  28
  29   std::vector<std::string> string_values;
  30   boost::split(string_values, coordStr, boost::is_any_of(" "));
  31   xbt_assert(string_values.size() == 3, "Coordinates of %s must have 3 dimensions", netpoint->get_cname());
  32
  33   for (auto const& str : string_values)
  34     try {
  35       coords.push_back(std::stod(str));
  36     } catch (std::invalid_argument const& ia) {
  37       throw std::invalid_argument(std::string("Invalid coordinate: ") + ia.what());
  38     }
  39   coords.shrink_to_fit();
  40
  41   netpoint->extension_set<Coords>(this);
  42   XBT_DEBUG("Coords of %s %p: %s", netpoint->get_cname(), netpoint, coordStr.c_str());
  43 }
  44 }; // namespace vivaldi
  45
  46 static inline double euclidean_dist_comp(int index, std::vector<double>* src, std::vector<double>* dst)
  47 {
  48   double src_coord = src->at(index);
  49   double dst_coord = dst->at(index);
  50
  51   return (src_coord - dst_coord) * (src_coord - dst_coord);
  52 }
  53
  54 static std::vector<double>* netpoint_get_coords(NetPoint* np)
  55 {
  56   simgrid::kernel::routing::vivaldi::Coords* coords = np->extension<simgrid::kernel::routing::vivaldi::Coords>();
  57   xbt_assert(coords, "Please specify the Vivaldi coordinates of %s %s (%p)",
  58              (np->is_netzone() ? "Netzone" : (np->is_host() ? "Host" : "Router")), np->get_cname(), np);
  59   return &coords->coords;
  60 }
  61
  62 VivaldiZone::VivaldiZone(NetZoneImpl* father, std::string name, resource::NetworkModel* netmodel)
  63     : ClusterZone(father, name, netmodel)
  64 {
  65 }
  66
  67 void VivaldiZone::set_peer_link(NetPoint* netpoint, double bw_in, double bw_out, std::string coord)
  68 {
  69   xbt_assert(netpoint->get_englobing_zone() == this,
  70              "Cannot add a peer link to a netpoint that is not in this netzone");
  71
  72   new simgrid::kernel::routing::vivaldi::Coords(netpoint, coord);
  73
  74   std::string link_up      = "link_" + netpoint->get_name() + "_UP";
  75   std::string link_down    = "link_" + netpoint->get_name() + "_DOWN";
  76   resource::LinkImpl* linkUp   = network_model_->create_link(link_up, bw_out, 0, s4u::Link::SharingPolicy::SHARED);
  77   resource::LinkImpl* linkDown = network_model_->create_link(link_down, bw_in, 0, s4u::Link::SharingPolicy::SHARED);
  78   private_links_.insert({netpoint->id(), {linkUp, linkDown}});
  79 }
  80
  81 void VivaldiZone::get_local_route(NetPoint* src, NetPoint* dst, RouteCreationArgs* route, double* lat)
  82 {
  83   XBT_DEBUG("vivaldi getLocalRoute from '%s'[%u] '%s'[%u]", src->get_cname(), src->id(), dst->get_cname(), dst->id());
  84
  85   if (src->is_netzone()) {
  86     std::string srcName = "router_" + src->get_name();
  87     std::string dstName = "router_" + dst->get_name();
  88     route->gw_src       = simgrid::s4u::Engine::get_instance()->netpoint_by_name_or_null(srcName.c_str());
  89     route->gw_dst       = simgrid::s4u::Engine::get_instance()->netpoint_by_name_or_null(dstName.c_str());
  90   }
  91
  92   /* Retrieve the private links */
  93   auto src_link = private_links_.find(src->id());
  94   if (src_link != private_links_.end()) {
  95     std::pair<resource::LinkImpl*, resource::LinkImpl*> info = src_link->second;
  96     if (info.first) {
  97       route->link_list.push_back(info.first);
  98       if (lat)
  99         *lat += info.first->get_latency();
 100     }
 101   } else {
 102     XBT_DEBUG("Source of private link (%u) doesn't exist", src->id());
 103   }
 104
 105   auto dst_link = private_links_.find(dst->id());
 106   if (dst_link != private_links_.end()) {
 107     std::pair<resource::LinkImpl*, resource::LinkImpl*> info = dst_link->second;
 108     if (info.second) {
 109       route->link_list.push_back(info.second);
 110       if (lat)
 111         *lat += info.second->get_latency();
 112     }
 113   } else {
 114     XBT_DEBUG("Destination of private link (%u) doesn't exist", dst->id());
 115   }
 116
 117   /* Compute the extra latency due to the euclidean distance if needed */
 118   if (lat) {
 119     std::vector<double>* srcCoords = netpoint_get_coords(src);
 120     std::vector<double>* dstCoords = netpoint_get_coords(dst);
 121
 122     double euclidean_dist =
 123         sqrt(euclidean_dist_comp(0, srcCoords, dstCoords) + euclidean_dist_comp(1, srcCoords, dstCoords)) +
 124         fabs(srcCoords->at(2)) + fabs(dstCoords->at(2));
 125
 126     XBT_DEBUG("Updating latency %f += %f", *lat, euclidean_dist);
 127     *lat += euclidean_dist / 1000.0; // From .ms to .s
 128   }
 129 }
 130 }
 131 }
 132 }