import argparse
import visualizing as vis
import os
def parse_cmdline():
parser = argparse.ArgumentParser(
description='Visualize results from balancer-binary.')
help_msg = 'Directory where results are laying.'
parser.add_argument(
'--results-dir',
metavar=('RESULTS_DIR'),
required=True,
help=help_msg
)
help_msg = 'Dark or light style'
parser.add_argument(
'--style',
metavar=('STYLE'),
choices=['dark', 'light'],
default='light',
required=False,
help=help_msg
)
args = parser.parse_args()
cwd = os.path.join(os.getcwd(), os.path.dirname(__file__))
results_dir = os.path.join(cwd, '..', '..', '..')
return {
'results_dir': os.path.join(results_dir, args.results_dir),
'style': args.style
}
if __name__ == '__main__':
params = parse_cmdline()
sim = vis.Simulation(
results_dir=params['results_dir'],
)
if params['style'] == 'dark':
plotting_machine = vis.plotting.dark()
if params['style'] == 'light':
plotting_machine = vis.plotting.light()
vis.run(sim=sim, vis=plotting_machine)