examples/cpp/dht-kademlia/generate.py

   1 #!/usr/bin/env python
   2
   3 # Copyright (c) 2012-2023. 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 """
  10 Usage: python generate.py nb_nodes nb_bits end_date > deployment_file.xml
  11 """
  12
  13 import sys
  14 import random
  15
  16 if len(sys.argv) != 4:
  17     print(
  18         "Usage: python generate.py nb_nodes nb_bits end_date > deployment_file.xml")
  19     sys.exit(1)
  20
  21 nb_nodes = int(sys.argv[1])
  22 nb_bits = int(sys.argv[2])
  23 end_date = int(sys.argv[3])
  24
  25 max_id = 2 ** nb_bits - 1
  26 all_ids = [0]
  27
  28 sys.stdout.write("<?xml version='1.0'?>\n"
  29                  "<!DOCTYPE platform SYSTEM \"https://simgrid.org/simgrid.dtd\">\n"
  30                  "<platform version=\"4\">\n  <process host=\"node-0.simgrid.org\" function=\"node\">\n"
  31                  "     <argument value=\"0\"/>\n     <argument value=\"%d\"/>\n  </process>\n" % end_date)
  32
  33 for i in range(1, nb_nodes):
  34     ok = False
  35     while not ok:
  36         my_id = random.randint(0, max_id)
  37         ok = my_id not in all_ids
  38     known_id = all_ids[random.randint(0, len(all_ids) - 1)]
  39     start_date = i * 10
  40     line = "  <process host=\"node-%d.simgrid.org\" function=\"node\">\n    <argument value=\"%s\"/>"\
  41            "\n    <argument value=\"%s\"/>\n    <argument value=\"%d\"/>\n  </process>\n" % (
  42                i, my_id, known_id, end_date)
  43     sys.stdout.write(line)
  44     all_ids.append(my_id)
  45
  46 sys.stdout.write("</platform>")