src/kernel/routing/VivaldiZone.cpp

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