mobility-Centrality

import getopt

import code  # code.interact(local=dict(globals(), **locals()))

from pprint import pprint

import sys

import os

import shutil

import datetime

from simulator import Simulator

if __name__ == '__main__':

    settingsFile = ''

    sectionExp = ''

    outPath = ''

    cdir = ''

    num_workers = 1

    gui = False

    options, remainder = getopt.getopt(

        sys.argv[1:], '', ['sf=', 'en=', 'out=', 'thr=', 'gui'])

    mand = set(['--sf', '--en', '--out'])

    opts = [x[0] for x in options]

    # print 'OPTIONS   :', opts

    if (not mand.issubset(opts)):

        print 'Mandatory args are: --sf (settings file) --en (experiment name)'\

            ' --out (Output path, overwrite or create folder)'

        sys.exit(1)

    for opt, arg in options:

        if opt in ('--sf'):

            settingsFile = arg

        if opt in ('--en'):

            sectionExp = arg

        if opt in ('--thr'):

            num_workers = int(arg)

        if opt in ('--out'):

            outPath = arg

            outPath.rstrip("/")

            if not os.path.exists(outPath):

                sys.stderr.write("Cannot find the outpath you provided!")

                sys.exit(1)

            else:

                start = datetime.datetime.now().strftime("%Hh%Mm%Ss_%d-%m-%Y")

                cdir = outPath + '/' + sectionExp + '_'+ start

                if os.path.exists(cdir):

                    shutil.rmtree(cdir)

                os.makedirs(cdir)

        if opt in ('--gui'):

            gui = True

    sim = Simulator(settingsFile, sectionExp, cdir, num_workers=num_workers, gui=gui)

    print "Simulation Parameters\n"

    pprint(sim.params)

    #code.interact(local=dict(globals(), **locals()))

    sim.runSimulation()