src/kernel/routing/ClusterZone.cpp

   1 /* Copyright (c) 2009-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/ClusterZone.hpp"
   7 #include "simgrid/kernel/routing/NetPoint.hpp"
   8 #include "simgrid/kernel/routing/RoutedZone.hpp"
   9 #include "src/surf/network_interface.hpp"
  10
  11 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(surf_route_cluster, surf, "Routing part of surf");
  12
  13 /* This routing is specifically setup to represent clusters, aka homogeneous sets of machines
  14  * Note that a router is created, easing the interconnection with the rest of the world. */
  15
  16 namespace simgrid {
  17 namespace kernel {
  18 namespace routing {
  19
  20 void ClusterBase::set_loopback()
  21 {
  22   if (not has_loopback_) {
  23     num_links_per_node_++;
  24     has_loopback_ = true;
  25   }
  26 }
  27
  28 void ClusterBase::set_limiter()
  29 {
  30   if (not has_limiter_) {
  31     num_links_per_node_++;
  32     has_limiter_ = true;
  33   }
  34 }
  35
  36 void ClusterBase::set_link_characteristics(double bw, double lat, s4u::Link::SharingPolicy sharing_policy)
  37 {
  38   link_sharing_policy_ = sharing_policy;
  39   link_bw_             = bw;
  40   link_lat_            = lat;
  41 }
  42
  43 void ClusterBase::add_private_link_at(unsigned int position, std::pair<resource::LinkImpl*, resource::LinkImpl*> link)
  44 {
  45   private_links_.insert({position, link});
  46 }
  47
  48 void ClusterBase::set_gateway(unsigned int position, NetPoint* gateway)
  49 {
  50   xbt_assert(not gateway || not gateway->is_netzone(), "ClusterBase: gateway cannot be another netzone %s",
  51              gateway->get_cname());
  52   gateways_[position] = gateway;
  53 }
  54
  55 NetPoint* ClusterBase::get_gateway(unsigned int position)
  56 {
  57   NetPoint* res = nullptr;
  58   auto it       = gateways_.find(position);
  59   if (it != gateways_.end()) {
  60     res = it->second;
  61   }
  62   return res;
  63 }
  64
  65 void ClusterBase::fill_leaf_from_cb(unsigned int position, const std::vector<unsigned int>& dimensions,
  66                                     const s4u::ClusterCallbacks& set_callbacks, NetPoint** node_netpoint,
  67                                     s4u::Link** lb_link, s4u::Link** limiter_link)
  68 {
  69   xbt_assert(node_netpoint, "Invalid node_netpoint parameter");
  70   xbt_assert(lb_link, "Invalid lb_link parameter");
  71   xbt_assert(limiter_link, "Invalid limiter_link paramater");
  72   *lb_link      = nullptr;
  73   *limiter_link = nullptr;
  74
  75   // auxiliary function to get dims from index
  76   auto index_to_dims = [&dimensions](int index) {
  77     std::vector<unsigned int> dims_array(dimensions.size());
  78     for (auto i = static_cast<int>(dimensions.size() - 1); i >= 0; --i) {
  79       if (index <= 0) {
  80         break;
  81       }
  82       unsigned int value = index % dimensions[i];
  83       dims_array[i]      = value;
  84       index              = (index / dimensions[i]);
  85     }
  86     return dims_array;
  87   };
  88
  89   kernel::routing::NetPoint* netpoint = nullptr;
  90   kernel::routing::NetPoint* gw       = nullptr;
  91   auto dims                           = index_to_dims(position);
  92   std::tie(netpoint, gw)              = set_callbacks.netpoint(get_iface(), dims, position);
  93   xbt_assert(netpoint, "set_netpoint(elem=%u): Invalid netpoint (nullptr)", position);
  94   if (netpoint->is_netzone()) {
  95     xbt_assert(gw && not gw->is_netzone(),
  96                "set_netpoint(elem=%u): Netpoint (%s) is a netzone, but gateway (%s) is invalid", position,
  97                netpoint->get_cname(), gw ? gw->get_cname() : "nullptr");
  98   } else {
  99     xbt_assert(not gw, "set_netpoint: Netpoint (%s) isn't netzone, gateway must be nullptr", netpoint->get_cname());
 100   }
 101   // setting gateway
 102   set_gateway(position, gw);
 103
 104   if (set_callbacks.loopback) {
 105     s4u::Link* loopback = set_callbacks.loopback(get_iface(), dims, position);
 106     xbt_assert(loopback, "set_loopback: Invalid loopback link (nullptr) for element %u", position);
 107     set_loopback();
 108     add_private_link_at(node_pos(netpoint->id()), {loopback->get_impl(), loopback->get_impl()});
 109     *lb_link = loopback;
 110   }
 111
 112   if (set_callbacks.limiter) {
 113     s4u::Link* limiter = set_callbacks.limiter(get_iface(), dims, position);
 114     xbt_assert(limiter, "set_limiter: Invalid limiter link (nullptr) for element %u", position);
 115     set_limiter();
 116     add_private_link_at(node_pos_with_loopback(netpoint->id()), {limiter->get_impl(), limiter->get_impl()});
 117     *limiter_link = limiter;
 118   }
 119   *node_netpoint = netpoint;
 120 }
 121
 122 } // namespace routing
 123 } // namespace kernel
 124 } // namespace simgrid