Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
fix some newly introduced smells
[simgrid.git] / examples / s4u / dht-chord / s4u_dht-chord.cpp
1 /* Copyright (c) 2010-2016. 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::instance()->hostList(&list);
33   for (auto 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 (!strncmp(options[0], "-", 1)) {
50     unsigned int length = strlen("-nb_bits=");
51     if (!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 (!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   return 0;
80 }