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;
  32   simgrid::s4u::Engine::getInstance()->getHostList(&list);
  33   for (auto const& host : list)
  34     host->extension_set(new HostChord(host));
  35 }
  36
  37 static void chord_exit()
  38 {
  39   delete[] powers2;
  40 }
  41
  42 int main(int argc, char* argv[])
  43 {
  44   simgrid::s4u::Engine* e = new simgrid::s4u::Engine(&argc, argv);
  45   xbt_assert(argc > 2, "Usage: %s [-nb_bits=n] [-timeout=t] platform_file deployment_file\n"
  46                        "\tExample: %s ../msg_platform.xml chord.xml\n",
  47              argv[0], argv[0]);
  48   char** options = &argv[1];
  49   while (not strncmp(options[0], "-", 1)) {
  50     unsigned int length = strlen("-nb_bits=");
  51     if (not strncmp(options[0], "-nb_bits=", length) && strlen(options[0]) > length) {
  52       nb_bits = xbt_str_parse_int(options[0] + length, "Invalid nb_bits parameter: %s");
  53       XBT_DEBUG("Set nb_bits to %d", nb_bits);
  54     } else {
  55       length = strlen("-timeout=");
  56       if (not strncmp(options[0], "-timeout=", length) && strlen(options[0]) > length) {
  57         timeout = xbt_str_parse_int(options[0] + length, "Invalid timeout parameter: %s");
  58         XBT_DEBUG("Set timeout to %d", timeout);
  59       } else {
  60         xbt_die("Invalid chord option '%s'", options[0]);
  61       }
  62     }
  63     options++;
  64   }
  65
  66   e->loadPlatform(options[0]);
  67
  68   chord_init();
  69
  70   e->registerFunction<Node>("node");
  71   e->loadDeployment(options[1]);
  72
  73   e->run();
  74
  75   XBT_INFO("Simulated time: %g", e->getClock());
  76
  77   chord_exit();
  78
  79   delete e;
  80   return 0;
  81 }