examples/s4u/engine-filtering/s4u-engine-filtering.cpp

   1 /* Copyright (c) 2017-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 "xbt/log.h"
   7 #include <simgrid/s4u.hpp>
   8
   9 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_engine_filtering, "Messages specific for this s4u example");
  10
  11 namespace filter {
  12 class FrequencyChanged {
  13   std::map<simgrid::s4u::Host*, int> host_list;
  14
  15 public:
  16   FrequencyChanged(simgrid::s4u::Engine& e) {
  17     std::vector<simgrid::s4u::Host*> list = e.get_all_hosts();
  18     for (auto& host : list) {
  19       host_list.insert({host, host->get_pstate()});
  20     }
  21   }
  22
  23   bool operator()(simgrid::s4u::Host* host) { return host->get_pstate() != host_list.at(host); }
  24
  25   double get_old_speed(simgrid::s4u::Host* host) { return host_list.at(host); }
  26 };
  27 class SingleCore {
  28 public:
  29   bool operator()(simgrid::s4u::Host* host) { return host->get_core_count() == 1; }
  30 };
  31
  32 bool filter_speed_more_than_50Mf(simgrid::s4u::Host* host);
  33 bool filter_speed_more_than_50Mf(simgrid::s4u::Host* host)
  34 {
  35   return host->getSpeed() > 50E6;
  36 }
  37 }
  38 int main(int argc, char* argv[])
  39 {
  40   simgrid::s4u::Engine e(&argc, argv);
  41   e.load_platform(argv[1]);
  42
  43   /**
  44    * Use a lambda function to filter hosts: We only want multicore hosts
  45    */
  46   XBT_INFO("Hosts currently registered with this engine: %lu", e.get_host_count());
  47   std::vector<simgrid::s4u::Host*> list =
  48       e.get_filtered_hosts([](simgrid::s4u::Host* host) { return host->get_core_count() > 1; });
  49
  50   for (auto& host : list)
  51     XBT_INFO("The following hosts have more than one core: %s", host->get_cname());
  52
  53   xbt_assert(list.size() == 1);
  54
  55   /*
  56    * Use a function object (functor) without memory
  57    */
  58   list = e.get_filtered_hosts(filter::SingleCore());
  59
  60   for (auto& host : list)
  61     XBT_INFO("The following hosts are SingleCore: %s", host->get_cname());
  62
  63   /**
  64    * This just shows how to use a function object that uses
  65    * memory to filter.
  66    */
  67   XBT_INFO("A simple example: Let's retrieve all hosts that changed their frequency");
  68   filter::FrequencyChanged filter(e);
  69   e.host_by_name("MyHost2")->set_pstate(2);
  70   list = e.get_filtered_hosts(filter);
  71
  72   for (auto& host : list)
  73     XBT_INFO("The following hosts changed their frequency: %s (from %.1ff to %.1ff)", host->get_cname(), host->getPstateSpeed(filter.get_old_speed(host)), host->getSpeed());
  74
  75   /*
  76    * You can also just use any normal function (namespaced as well, if you want) to filter
  77    */
  78   list = e.get_filtered_hosts(filter::filter_speed_more_than_50Mf);
  79
  80   for (auto& host : list)
  81     XBT_INFO("The following hosts have a frequency > 50Mf: %s", host->get_cname());
  82
  83   return 0;
  84 }