main.py

"""
Massive MIMO Network slicing on MAC layer simulation

"""

import json
import time
import sys
import os
import numpy as np
from utilities.trace import Trace
from simulation import Simulation
import argparse
sys.path.append(os.path.abspath('../'))

with open('default_config.json') as config_file:
    config = json.load(config_file)

def isprime(N):
    if N<=1 or N==4:
        return False
    else:
        for i in range(2, N//2):
            if N%i == 0:
                return False
        return True


if __name__ == '__main__':
    # Load simulation parameters

    parser = argparse.ArgumentParser()

    parser.add_argument('--variance', action="store", type=float, default=None)
    parser.add_argument('--period', action="store", type=float, default=None)
    parser.add_argument('--d_on', action="store", type=float, default=None)
    parser.add_argument('--d_off', action="store", type=float, default=None)
    parser.add_argument('--no_periodic', action="store", type=int, default=None)
    parser.add_argument('--no_file', action="store", type=int, default=None)
    parser.add_argument('--seed', action="store", type=int, default=None)
    parser.add_argument('--distribution', action="store", default="constant")

    args = parser.parse_args()
    # print(args.scheduler)

    seed = args.seed

    inner_periods = args.period

    inner_variance = args.variance
    t_on = args.d_on
    t_off = args.d_off
    np.random.seed(seed)
    distribution = args.distribution
    no_file = args.no_file
    no_periodic = args.no_periodic

    trace_file_path = 'trace/' + 'Trace_' + str(inner_periods) + '_' + str(inner_variance) + '-' + str(no_file) + '_' + str(no_periodic) + '_' + str(round(time.time())) + '_event_trace.csv'

    trace = Trace(trace_file_path, log=True)
    simulation = Simulation(config, no_file, no_periodic, inner_periods, inner_variance, t_on, t_off, distribution, trace, seed)

    simulation.run()
    # Close files
    trace.close()
    trace.plot_arrivals()
    trace.show_plot()