examples/s4u/routing-get-clusters/s4u-routing-get-clusters.cpp

   1 /* Copyright (c) 2009-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/ClusterZone.hpp"
   7 #include "simgrid/kernel/routing/DragonflyZone.hpp"
   8 #include "simgrid/s4u.hpp"
   9
  10 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this s4u example");
  11
  12 int main(int argc, char* argv[])
  13 {
  14   simgrid::s4u::Engine e(&argc, argv);
  15   e.load_platform(argv[1]);
  16
  17   std::vector<simgrid::kernel::routing::ClusterZone*> clusters =
  18       e.filter_netzones_by_type<simgrid::kernel::routing::ClusterZone>();
  19
  20   for (auto c : clusters) {
  21     XBT_INFO("%s", c->get_cname());
  22     std::vector<simgrid::s4u::Host*>* hosts = new std::vector<simgrid::s4u::Host*>;
  23     c->getHosts(hosts);
  24     for (auto h : *hosts)
  25       XBT_INFO("   %s", h->get_cname());
  26     delete hosts;
  27   }
  28
  29   std::vector<simgrid::kernel::routing::DragonflyZone*> dragonfly_clusters =
  30       e.filter_netzones_by_type<simgrid::kernel::routing::DragonflyZone>();
  31
  32   if (not dragonfly_clusters.empty()) {
  33     for (auto d : dragonfly_clusters) {
  34       XBT_INFO("%s' dragonfly topology:", d->get_cname());
  35       for (int i = 0; i < d->getHostCount(); i++) {
  36         unsigned int coords[4];
  37         d->rankId_to_coords(i, coords);
  38         XBT_INFO("   %d: (%u, %u, %u, %u)", i, coords[0], coords[1], coords[2], coords[3]);
  39       }
  40     }
  41   }
  42
  43   return 0;
  44 }