#!/usr/bin/env/python

from rumba.model import *

# import testbed plugins
import rumba.testbeds.emulab as emulab
import rumba.testbeds.jfed as jfed
import rumba.testbeds.faketestbed as fake
import rumba.testbeds.qemu as qemu

# import prototype plugins
import rumba.prototypes.ouroboros as our
import rumba.prototypes.rlite as rl
import rumba.prototypes.irati as irati

shim500 = ShimEthDIF("shim500", link_speed=0))
shim600 = ShimEthDIF("shim600", link_speed=0))
shim700 = ShimEthDIF("shim700", link_speed=0))
shim400 = ShimEthDIF("shim400", link_speed=0))
shim300 = ShimEthDIF("shim300", link_speed=0))
shim800 = ShimEthDIF("shim800", link_speed=0))

n1 = NormalDIF("n1")

b = Node("b",
    difs=[300, 400, n1],
    dif_registrations={n1: [300, 400]})

g = Node("g",
    difs=[800, n1],
    dif_registrations={n1: [800]})

c = Node("c",
    difs=[400, 500, 600, n1],
    dif_registrations={n1: [400, 500, 600]})

f = Node("f",
    difs=[500, n1],
    dif_registrations={n1: [500]})

a = Node("a",
    difs=[300, n1],
    dif_registrations={n1: [300]})

d = Node("d",
    difs=[600, 700, n1],
    dif_registrations={n1: [600, 700]})

e = Node("e",
    difs=[700, 800, n1],
    dif_registrations={n1: [700, 800]})

tb = qemu.Testbed(exp_name = "seven",
    bzimage = "/home/mcapitani/RINA/checkouts/rlite/demo/buildroot/bzImage",
    initramfs = "/home/mcapitani/RINA/checkouts/rlite/demo/buildroot/rootfs.cpio")

exp = rl.Experiment(tb, nodes = [a, b, c, d])

print(exp)

exp.run()