Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
3b707ac48a1220c832f17b19a2d3232ff613b5a2
[simgrid.git] / examples / msg / chord / generate.py
1 #!/usr/bin/env python
2
3 # Copyright (c) 2011-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 # This script generates a specific deployment file for the Chord example.
10 # It assumes that the platform will be a cluster.
11 # Usage: python generate.py nb_nodes nb_bits end_date
12 # Example: python generate.py 100000 32 1000
13
14 import sys, random
15
16 if len(sys.argv) != 4:
17         print("Usage: python generate.py nb_nodes nb_bits end_date > deployment_file.xml")
18         sys.exit(1)
19
20 nb_nodes = int(sys.argv[1])
21 nb_bits = int(sys.argv[2])
22 end_date = int(sys.argv[3])
23
24 max_id = 2 ** nb_bits - 1
25 all_ids = [42]
26
27 sys.stdout.write("<?xml version='1.0'?>\n"
28 "<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid.dtd\">\n"
29 "<platform version=\"3\">\n"
30 "  <process host=\"node-0.acme.org\" function=\"node\"><argument value=\"42\"/><argument value=\"%d\"/></process>\n" % end_date)
31
32 for i in range(1, nb_nodes):
33
34         ok = False
35         while not ok:
36                 my_id = random.randint(0, max_id)
37                 ok = not my_id in all_ids
38
39         known_id = all_ids[random.randint(0, len(all_ids) - 1)]
40         start_date = i * 10
41         line = "  <process host=\"node-%d.acme.org\" function=\"node\"><argument value=\"%d\" /><argument value=\"%d\" /><argument value=\"%d\" /><argument value=\"%d\" /></process>\n" % (i, my_id, known_id, start_date, end_date)
42         sys.stdout.write(line)
43         all_ids.append(my_id)
44
45 sys.stdout.write("</platform>")
46