BPR-Scripts/rand_network.py at master · danposch/BPR-Scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
from igraph import *
import random
import importlib

NODES = 20
CLIENTS = 12
SERVERS = 4
EDGE_PROBABILITY = 0.15
MIN_BANDWIDTH = 3000
MAX_BANDWIDTH = 4000
CONTENT_POPULARITY = "uniform"
MIN_DELAY = 5
MAX_DELAY = 20
# "" -> no loss, "random_loss_model", "markov_loss_model", "gilbert_elliot_loss_model"
LOSS_MODEL = ""

def genRandomNetwork(EMU_RUN_NUMBER):

	print "Generating Random Network:"

	random.seed(EMU_RUN_NUMBER)

	if CLIENTS + SERVERS > NODES:
		print "To much Clients/Servers for this Network! Exiting..."
		exit(-1)

	is_connected = False
	while not is_connected:
		g = Graph.Erdos_Renyi(n=NODES, p=EDGE_PROBABILITY, directed=False, loops=False)
		is_connected = g.is_connected()

	#g.to_directed() # not needed will be done by the parser...

	# load selected loss model if specified
	if LOSS_MODEL:
		lossModel = importlib.import_module(LOSS_MODEL)

	fname = "generated_network_top.txt"

	with open(fname ,"w") as f:

		f.write("#number of nodes\n")
		f.write(str(NODES)+"\n")

		f.write("#nodes setting (n1,n2,bandwidth in kbits a -> b, bandwidth in kbits a <- b, delay a -> b in ms, delay b -> a in ms"
			+ (", loss a -> b, loss a <- b" if LOSS_MODEL else "") + ")\n")

		for edge in g.es:
			f.write( str(edge.source) + "," + str(edge.target) + ","
				 + str(int(random.uniform(MIN_BANDWIDTH, MAX_BANDWIDTH))) + ","
				 + str(int(random.uniform(MIN_BANDWIDTH, MAX_BANDWIDTH))) + ","
				 + str(int(random.uniform(MIN_DELAY, MAX_DELAY))) + ","
				 + str(int(random.uniform(MIN_DELAY, MAX_DELAY))))

			if not LOSS_MODEL:
				f.write("\n")
			else:
				f.write("," + lossModel.getLossCmd() + ","
				+ lossModel.getLossCmd() + "\n")

		f.write("#properties (Client, Server)\n")

		nodes = range(0,20)
		servers = []
		clients = []

		for i in range(0,SERVERS):
			index = random.randint(0, len(nodes)-1)
			servers.append(nodes.pop(index))

		for i in range(0,CLIENTS):
			index = random.randint(0, len(nodes)-1)
			clients.append(nodes.pop(index))

		for client in clients:

			if CONTENT_POPULARITY == "uniform":
				f.write(str(client) + "," + str(servers[int(round(random.uniform(0,len(servers)-1)))]) +"\n")
			elif CONTENT_POPULARITY == "zipf":
				f.write(str(client) + "," + str(servers[int(round(random.zipf(0,len(servers)-1)))]) +"\n")
			else:
				print "Invalid Content Popularity! Exiting.."
				exit(-1)

		f.write("#eof //do not delete this")
		f.close()

	#plot(g)

	return fname

#main test
#genRandomNetwork(0)