src/kernel/routing/AsVivaldi.cpp

   1 /* Copyright (c) 2013-2016. 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/host.hpp>
   9
  10 #include "src/kernel/routing/AsVivaldi.hpp"
  11 #include "src/kernel/routing/NetCard.hpp"
  12 #include "src/surf/network_interface.hpp"
  13
  14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_route_vivaldi, surf, "Routing part of surf");
  15
  16 namespace simgrid {
  17 namespace kernel {
  18 namespace routing {
  19 namespace vivaldi {
  20 simgrid::xbt::Extension<NetCard, Coords> Coords::EXTENSION_ID;
  21
  22 Coords::Coords(NetCard* netcard, const char* coordStr)
  23 {
  24   if (!Coords::EXTENSION_ID.valid())
  25     Coords::EXTENSION_ID = NetCard::extension_create<Coords>();
  26
  27   std::vector<std::string> string_values;
  28   boost::split(string_values, coordStr, boost::is_any_of(" "));
  29   xbt_assert(string_values.size() == 3, "Coordinates of %s must have 3 dimensions", netcard->cname());
  30
  31   for (auto str : string_values)
  32     coords.push_back(xbt_str_parse_double(str.c_str(), "Invalid coordinate: %s"));
  33   coords.shrink_to_fit();
  34
  35   netcard->extension_set<Coords>(this);
  36   XBT_DEBUG("Coords of %s %p: %s", netcard->cname(), netcard, coordStr);
  37 }
  38 Coords::~Coords() = default;
  39 }; // namespace vivaldi
  40
  41 static inline double euclidean_dist_comp(int index, std::vector<double>* src, std::vector<double>* dst)
  42 {
  43   double src_coord = src->at(index);
  44   double dst_coord = dst->at(index);
  45
  46   return (src_coord - dst_coord) * (src_coord - dst_coord);
  47 }
  48
  49 static std::vector<double>* getCoordsFromNetcard(NetCard* nc)
  50 {
  51   simgrid::kernel::routing::vivaldi::Coords* coords = nc->extension<simgrid::kernel::routing::vivaldi::Coords>();
  52   xbt_assert(coords, "Please specify the Vivaldi coordinates of %s %s (%p)",
  53              (nc->isAS() ? "AS" : (nc->isHost() ? "Host" : "Router")), nc->cname(), nc);
  54   return &coords->coords;
  55 }
  56 AsVivaldi::AsVivaldi(As* father, const char* name) : AsCluster(father, name)
  57 {
  58 }
  59
  60 void AsVivaldi::getLocalRoute(NetCard* src, NetCard* dst, sg_platf_route_cbarg_t route, double* lat)
  61 {
  62   XBT_DEBUG("vivaldi getLocalRoute from '%s'[%d] '%s'[%d]", src->cname(), src->id(), dst->cname(), dst->id());
  63
  64   if (src->isAS()) {
  65     char* srcName = bprintf("router_%s", src->cname());
  66     char* dstName = bprintf("router_%s", dst->cname());
  67     route->gw_src = (sg_netcard_t)xbt_lib_get_or_null(as_router_lib, srcName, ROUTING_ASR_LEVEL);
  68     route->gw_dst = (sg_netcard_t)xbt_lib_get_or_null(as_router_lib, dstName, ROUTING_ASR_LEVEL);
  69     xbt_free(srcName);
  70     xbt_free(dstName);
  71   }
  72
  73   /* Retrieve the private links */
  74   if (privateLinks_.size() > src->id()) {
  75     s_surf_parsing_link_up_down_t info = privateLinks_.at(src->id());
  76     if (info.linkUp) {
  77       route->link_list->push_back(info.linkUp);
  78       if (lat)
  79         *lat += info.linkUp->latency();
  80     }
  81   }
  82   if (privateLinks_.size() > dst->id()) {
  83     s_surf_parsing_link_up_down_t info = privateLinks_.at(dst->id());
  84     if (info.linkDown) {
  85       route->link_list->push_back(info.linkDown);
  86       if (lat)
  87         *lat += info.linkDown->latency();
  88     }
  89   }
  90
  91   /* Compute the extra latency due to the euclidean distance if needed */
  92   if (lat) {
  93     std::vector<double>* srcCoords = getCoordsFromNetcard(src);
  94     std::vector<double>* dstCoords = getCoordsFromNetcard(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 }
 106 }