Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
984b68ab66b9376f8d43c27a492a68e09b0e3f92
[simgrid.git] / examples / deprecated / java / app / bittorrent / generate.py
1 #!/usr/bin/env python
2
3 # Copyright (c) 2013-2019. 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 Bittorrent example.
10 # It assumes that the platform will be a cluster.
11 # Usage: python generate.py nb_nodes nb_bits end_date percentage
12 # Example: python generate.py 10000 5000
13
14 import sys
15 import random
16
17 if len(sys.argv) != 4:
18     print(
19         "Usage: python generate.py nb_nodes end_date seed_percentage > deployment_file.xml")
20     sys.exit(1)
21
22 nb_nodes = int(sys.argv[1])
23 end_date = int(sys.argv[2])
24 seed_percentage = int(sys.argv[3])
25
26 nb_bits = 24
27 max_id = 2 ** nb_bits - 1
28 all_ids = [42]
29
30 sys.stdout.write("<?xml version='1.0'?>\n"
31                  "<!DOCTYPE platform SYSTEM \"http://simgrid.gforge.inria.fr/simgrid.dtd\">\n"
32                  "<platform version=\"3\">\n"
33                  "  <process host=\"c-0.me\" function=\"bittorrent.Tracker\"><argument value=\"%d\"/></process>\n" % end_date)
34
35 for i in range(1, nb_nodes):
36
37     ok = False
38     while not ok:
39         my_id = random.randint(0, max_id)
40         ok = not my_id in all_ids
41     start_date = i * 10
42     line = "  <process host=\"c-%d.me\" function=\"bittorrent.Peer\"><argument value=\"%d\" /><argument value=\"%d\" />" % (
43         i, my_id, end_date)
44     if random.randint(0, 100) < seed_percentage:
45         line += "<argument value=\"1\" />"
46     line += "</process>\n"
47     sys.stdout.write(line)
48     all_ids.append(my_id)
49 sys.stdout.write("</platform>")