SDNLoadBalancingRL/sim.py at master · WilliamDormer/SDNLoadBalancingRL · 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
from fake_sim_class import FakeSim
import argparse
from ast import literal_eval

if __name__ == "__main__":

    parser = argparse.ArgumentParser()
    parser.add_argument('-g', "--generation_mode", type=str, required=True)
    parser.add_argument('-s', "--num_switches", type=int, required=True)
    parser.add_argument('-c', "--num_controllers", type=int, required=True)
    parser.add_argument('-ca', "--capacities", type=literal_eval, required=True)
    parser.add_argument('-is', "--initial_sbc", type=literal_eval, required=True)
    parser.add_argument('-br', "--base_rates", type=literal_eval, required=True)
    parser.add_argument('-f', "--fast_mode", type=bool, required=True)
    parser.add_argument('-sd', "--simulated_delay", type=float, required=True)
    parser.add_argument( "-fa", "--fluctuation_amplitudes", default=None, type=literal_eval)
    parser.add_argument("--period", default=None, type=float)
    parser.add_argument("--random_init", type=bool, default=False)

    args = parser.parse_args()

    generation_mode = args.generation_mode
    num_switches = args.num_switches
    num_controllers = args.num_controllers
    capacities = args.capacities
    initial_sbc = args.initial_sbc
    base_rates = args.base_rates
    fast_mode = args.fast_mode
    simulated_delay = args.simulated_delay

    print(args)

    print("running simulator")

    sim = FakeSim(
        generation_mode=generation_mode,
        num_switches=num_switches,
        num_controllers=num_controllers,
        capacities=capacities,
        initial_sbc=initial_sbc,
        base_rates = base_rates,
        fast_mode=fast_mode,
        simulated_delay=simulated_delay,
        fluctuation_amplitudes=args.fluctuation_amplitudes,
        period = args.period,
        random_init=args.random_init
    )

    # sim = FakeSim(
    #     generation_mode="base_rate",
    #     num_switches=2,
    #     num_controllers=2,
    #     capacities=[1000,1000],
    #     initial_sbc=[[1,2],[]],
    #     base_rates = [10,20],
    #     fast_mode=True,
    #     simulated_delay=5
    # )