examples/s4u/dht-chord/s4u-dht-chord.cpp

   1 /* Copyright (c) 2010-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 "s4u-dht-chord.hpp"
   7
   8 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_chord, "Messages specific for this s4u example");
   9 simgrid::xbt::Extension<simgrid::s4u::Host, HostChord> HostChord::EXTENSION_ID;
  10
  11 int nb_bits  = 24;
  12 int nb_keys  = 0;
  13 int timeout  = 50;
  14 int* powers2 = nullptr;
  15
  16 /* Global initialization of the Chord simulation. */
  17 static void chord_init()
  18 {
  19   // compute the powers of 2 once for all
  20   powers2 = new int[nb_bits];
  21   int pow = 1;
  22   for (int i = 0; i < nb_bits; i++) {
  23     powers2[i] = pow;
  24     pow        = pow << 1;
  25   }
  26   nb_keys = pow;
  27   XBT_DEBUG("Sets nb_keys to %d", nb_keys);
  28
  29   HostChord::EXTENSION_ID = simgrid::s4u::Host::extension_create<HostChord>();
  30
  31   std::vector<simgrid::s4u::Host*> list = simgrid::s4u::Engine::getInstance()->getAllHosts();
  32   for (auto const& host : list)
  33     host->extension_set(new HostChord(host));
  34 }
  35
  36 static void chord_exit()
  37 {
  38   delete[] powers2;
  39 }
  40
  41 int main(int argc, char* argv[])
  42 {
  43   simgrid::s4u::Engine e(&argc, argv);
  44   xbt_assert(argc > 2, "Usage: %s [-nb_bits=n] [-timeout=t] platform_file deployment_file\n"
  45                        "\tExample: %s ../msg_platform.xml chord.xml\n",
  46              argv[0], argv[0]);
  47   char** options = &argv[1];
  48   while (not strncmp(options[0], "-", 1)) {
  49     unsigned int length = strlen("-nb_bits=");
  50     if (not strncmp(options[0], "-nb_bits=", length) && strlen(options[0]) > length) {
  51       nb_bits = xbt_str_parse_int(options[0] + length, "Invalid nb_bits parameter: %s");
  52       XBT_DEBUG("Set nb_bits to %d", nb_bits);
  53     } else {
  54       length = strlen("-timeout=");
  55       if (not strncmp(options[0], "-timeout=", length) && strlen(options[0]) > length) {
  56         timeout = xbt_str_parse_int(options[0] + length, "Invalid timeout parameter: %s");
  57         XBT_DEBUG("Set timeout to %d", timeout);
  58       } else {
  59         xbt_die("Invalid chord option '%s'", options[0]);
  60       }
  61     }
  62     options++;
  63   }
  64
  65   e.loadPlatform(options[0]);
  66
  67   chord_init();
  68
  69   e.registerFunction<Node>("node");
  70   e.loadDeployment(options[1]);
  71
  72   e.run();
  73
  74   XBT_INFO("Simulated time: %g", e.getClock());
  75
  76   chord_exit();
  77
  78   return 0;
  79 }