Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
d7613fb87b5c7f2de31375c0fe64d1899f8691c1
[simgrid.git] / examples / s4u / dht-chord / s4u-dht-chord.cpp
1 /* Copyright (c) 2010-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 "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()->get_all_hosts();
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.load_platform(options[0]);
66
67   chord_init(); // FIXME: inline me
68
69   e.register_actor<Node>("node");
70   e.load_deployment(options[1]);
71
72   e.run();
73
74   XBT_INFO("Simulated time: %g", e.get_clock());
75
76   chord_exit();
77
78   return 0;
79 }