Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
all DHT examples are now called dht-<protocol>
[simgrid.git] / examples / msg / dht-kademlia / generate.py
1 #!/usr/bin/env python
2
3 # Copyright (c) 2012, 2014. The SimGrid Team.
4 # All rights reserved.
5
6 # This program is free software; you can redistribute it and/or modify it
7 # under the terms of the license (GNU LGPL) which comes with this package.
8
9 import sys
10 import random
11
12 if len(sys.argv) != 4:
13     print(
14         "Usage: python generate.py nb_nodes nb_bits end_date > deployment_file.xml")
15     sys.exit(1)
16
17 nb_nodes = int(sys.argv[1])
18 nb_bits = int(sys.argv[2])
19 end_date = int(sys.argv[3])
20
21 max_id = 2 ** nb_bits - 1
22 all_ids = [0]
23
24 sys.stdout.write("<?xml version='1.0'?>\n"
25                  "<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid/simgrid.dtd\">\n"
26                  "<platform version=\"4\">\n  <process host=\"node-0.acme.org\" function=\"node\">\n"
27                  "     <argument value=\"0\"/>\n     <argument value=\"%d\"/>\n  </process>\n" % end_date)
28
29 for i in range(1, nb_nodes):
30     ok = False
31     while not ok:
32         my_id = random.randint(0, max_id)
33         ok = not my_id in all_ids
34     known_id = all_ids[random.randint(0, len(all_ids) - 1)]
35     start_date = i * 10
36     line = "  <process host=\"node-%d.acme.org\" function=\"node\">\n    <argument value=\"%s\"/>"\
37            "\n    <argument value=\"%s\"/>\n    <argument value=\"%d\"/>\n  </process>\n" % (
38                i, my_id, known_id, end_date)
39     sys.stdout.write(line)
40     all_ids.append(my_id)
41
42 sys.stdout.write("</platform>")