aboutsummaryrefslogtreecommitdiff
path: root/examples
diff options
context:
space:
mode:
authorDimitri Staessens <dimitri@ouroboros.rocks>2022-11-19 21:44:33 +0100
committerDimitri Staessens <dimitri@ouroboros.rocks>2022-11-19 21:44:33 +0100
commit9d4efe67c01c8c5a3f3bd1ee71af8b7c03021a18 (patch)
treeb2c767fb238614943831b962e2322238d3ef88be /examples
parentc05125b35934d25b44f3bf932420bc0f7f950093 (diff)
downloadrumba-9d4efe67c01c8c5a3f3bd1ee71af8b7c03021a18.tar.gz
rumba-9d4efe67c01c8c5a3f3bd1ee71af8b7c03021a18.zip
rumba: Add visualisation tool
The draw_experiment(exp) function will draw a 3D representation of the network. Depends on igraph and plotly packages. Signed-off-by: Dimitri Staessens <dimitri@ouroboros.rocks>
Diffstat (limited to 'examples')
-rwxr-xr-xexamples/scalingtime.py94
1 files changed, 48 insertions, 46 deletions
diff --git a/examples/scalingtime.py b/examples/scalingtime.py
index bf4a739..ff3c2ed 100755
--- a/examples/scalingtime.py
+++ b/examples/scalingtime.py
@@ -16,20 +16,20 @@ import rumba.prototypes.rlite as rl
import rumba.prototypes.irati as irati
import argparse
-import sys
+from rumba.visualizer import draw_experiment
description = "Script to create the demo from the Rumba paper"
-argparser = argparse.ArgumentParser(description = description)
+argparser = argparse.ArgumentParser(description=description)
# Parameters to set the experiment size
-argparser.add_argument('--metro-networks', type = int, default = '2',
- help = "Number of metro networks")
-argparser.add_argument('--metro-nodes', type = int, default = '5',
- help = "Number of nodes per metro")
-argparser.add_argument('--end-nodes', type = int, default = '2',
- help = "Number of nodes connected to each metro")
+argparser.add_argument('--metro-networks', type=int, default='2',
+ help="Number of metro networks")
+argparser.add_argument('--metro-nodes', type=int, default='5',
+ help="Number of nodes per metro")
+argparser.add_argument('--end-nodes', type=int, default='2',
+ help="Number of nodes connected to each metro")
# Other parameters
argparser.add_argument('--verbosity', metavar='VERBOSITY', type=str,
@@ -57,8 +57,8 @@ qemu_p.add_argument('--use_vhost', action='store_true',
default=False, help='Use vhost')
qemu_p.add_argument('--qemu_logs_dir', metavar='QEMU_LOGS', type=str,
default=None, help='path to the folder for qemu logs')
-qemu_p.add_argument('--username', type = str, help = "Username")
-qemu_p.add_argument('--password', type = str, help = "Password")
+qemu_p.add_argument('--username', type=str, help="Username")
+qemu_p.add_argument('--password', type=str, help="Password")
emulab_p.add_argument('--url', metavar='URL', type=str,
default="wall2.ilabt.iminds.be",
@@ -66,27 +66,27 @@ emulab_p.add_argument('--url', metavar='URL', type=str,
emulab_p.add_argument('--image', metavar='IMG', type=str,
default="UBUNTU14-64-STD",
help='Ubuntu image')
-emulab_p.add_argument('--username', type = str, required=True,
- help = "Username")
-emulab_p.add_argument('--password', type = str, help = "Password")
-emulab_p.add_argument('--exp-name', type = str, required=True,
- help = "Name of the experiment")
+emulab_p.add_argument('--username', type=str, required=True,
+ help="Username")
+emulab_p.add_argument('--password', type=str, help="Password")
+emulab_p.add_argument('--exp-name', type=str, required=True,
+ help="Name of the experiment")
-jfed_p.add_argument('--cert-file', type = str,
+jfed_p.add_argument('--cert-file', type=str,
required=True,
- help = "Absolute path to certificate (.pem) file"
+ help="Absolute path to certificate (.pem) file"
" to be used with jFed")
-jfed_p.add_argument('--authority', type = str, default = 'exogeni.net',
- help = "jFed testbed to use")
+jfed_p.add_argument('--authority', type=str, default='exogeni.net',
+ help="jFed testbed to use")
jfed_p.add_argument('--image', metavar='IMAGE', type=str,
default=None,
help='Image to be used')
jfed_p.add_argument('--exp-hours', metavar='HOURS', type=str,
default="2", help='Experiment hours')
-jfed_p.add_argument('--username', type = str, required=True,
- help = "Username")
-jfed_p.add_argument('--exp-name', type = str, required=True,
- help = "Name of the experiment")
+jfed_p.add_argument('--username', type=str, required=True,
+ help="Username")
+jfed_p.add_argument('--exp-name', type=str, required=True,
+ help="Name of the experiment")
docker_p.add_argument('--base-image', metavar='DOCKIMAGE', type=str,
default=None, required=True,
@@ -117,19 +117,19 @@ for i in range(0, args.metro_networks):
e = ShimEthDIF("e-" + str(i) + "-" + str(j))
else:
e = None
- if e_prev == None and e != None:
+ if e_prev is None and e is not None:
node = Node("metro-" + str(i) + "-" + str(j),
- difs = [m, pi, e],
- dif_registrations = {pi : [m], m : [e]})
+ difs=[m, pi, e],
+ dif_registrations={pi: [m], m: [e]})
m1 = node
- elif e_prev != None and e != None:
+ elif e_prev is not None and e is not None:
node = Node("metro-" + str(i) + "-" + str(j),
- difs = [m, pi, e, e_prev],
- dif_registrations = {pi : [m], m : [e, e_prev]})
+ difs=[m, pi, e, e_prev],
+ dif_registrations={pi: [m], m: [e, e_prev]})
else:
node = Node("metro-" + str(i) + "-" + str(j),
- difs = [m, pi, e_prev],
- dif_registrations = {pi : [m], m : [e_prev]})
+ difs=[m, pi, e_prev],
+ dif_registrations={pi: [m], m: [e_prev]})
mn = node
nodes.append(node)
e_prev = e
@@ -137,8 +137,8 @@ for i in range(0, args.metro_networks):
for k in range(0, args.end_nodes):
ee = ShimEthDIF("e-" + str(i) + "-" + str(j) + "-" + str(k))
end_node = Node("end-" + str(i) + "-" + str(j) + "-" + str(k),
- difs = [pi, ee],
- dif_registrations = {pi : [ee]})
+ difs=[pi, ee],
+ dif_registrations={pi: [ee]})
node.add_dif(ee)
node.add_dif_registration(pi, ee)
nodes.append(end_node)
@@ -155,12 +155,12 @@ for i in range(0, args.metro_networks):
e2 = ShimEthDIF("c-e-" + str(i) + "-2")
c1 = Node("c-" + str(i) + "-0",
- difs = [c, m, pi, e0, e2],
- dif_registrations = {pi : [m, c], m : [e0], c : [e2]})
+ difs=[c, m, pi, e0, e2],
+ dif_registrations={pi: [m, c], m: [e0], c: [e2]})
c2 = Node("c-" + str(i) + "-1",
- difs = [c, m, pi, e1, e2],
- dif_registrations = {pi : [m, c], m : [e1], c : [e2]})
+ difs=[c, m, pi, e1, e2],
+ dif_registrations={pi: [m, c], m: [e1], c: [e2]})
nodes.append(c1)
nodes.append(c2)
@@ -235,12 +235,14 @@ else:
print(argparser.format_help())
exit(1)
-exp = exp_class(testbed, nodes = nodes)
-
-with ExperimentManager(exp, swap_out_strategy=PAUSE_SWAPOUT):
- exp.swap_in()
- if not isinstance(testbed, docker.Testbed) \
- and not isinstance(testbed, qemu.Testbed) \
- and not isinstance(testbed, local.Testbed):
- exp.install_prototype()
- exp.bootstrap_prototype()
+exp = exp_class(testbed, nodes=nodes)
+
+if __name__ == "__main__":
+ draw_experiment(exp)
+ with ExperimentManager(exp, swap_out_strategy=PROMPT_SWAPOUT):
+ exp.swap_in()
+ if not isinstance(testbed, docker.Testbed) \
+ and not isinstance(testbed, qemu.Testbed) \
+ and not isinstance(testbed, local.Testbed):
+ exp.install_prototype()
+ exp.bootstrap_prototype()