toy_case.py

"""
Create the graphs using example_model.txt 
"""

import networkx as nx
# import stellargraph as sg

from collections import Counter

import argparse
import logging

#my imports
import vars_paths as vp
import utils as utils
import saves


def read_file(file):

	windows, monitors, trackers, peer_lists = [], [], [], []

	with open(file, 'r') as file:
		
		for line in file:
			l = line.split()

			windows.append(l[0])
			monitors.append(l[1])
			trackers.append(l[2])
			pl = []
			for x in l[3:]:
				pl.append(x)
			peer_lists.append(pl)

	return windows, monitors, trackers, peer_lists


def main():

	parser = argparse.ArgumentParser(description='Create toy case')

	parser.add_argument('--file', '-f', help='Arquivo de entrada', required=True, type=str)
	
	parser.add_argument('--showpeers', '-p', help='peers in graph', action='store_true')
	parser.add_argument('--showmaster', '-g', help='master in graph', action='store_true')
	parser.add_argument('--showtrackers', '-t', help='trackers in graph', action='store_true')
	parser.add_argument('--showmonitors', '-m', help='monitors in graph', action='store_true')
	
	help_msg = "Logging level (INFO=%d DEBUG=%d)" % (logging.INFO, logging.DEBUG)
	parser.add_argument("--log", "-l", help=help_msg, default=vp.DEFAULT_LOG_LEVEL, type=int)

	args = parser.parse_args()

	if args.log == logging.DEBUG:
		logging.basicConfig(format='%(asctime)s.%(msecs)03d: %(levelname)s {%(module)s} [%(funcName)s] %(message)s', datefmt=TIME_FORMAT, level=args.log)
	else:
		logging.basicConfig(format='%(asctime)s.%(msecs)03d: %(message)s', datefmt=vp.TIME_FORMAT, level=args.log)

	logging.info('init ...')
	utils.init(showmaster = args.showmaster, showpeers = args.showpeers, showtrackers = args.showtrackers, showmonitors = args.showmonitors)

	logging.info('reading file ...')
	windows, monitors, trackers, peer_lists = read_file(args.file)


	logging.info('range windows ...')
	windows_index_range = utils.windows_range(windows)


	ms = 'MS'
	graphs = []
	logging.info('creating graphs ...')
	for wir in windows_index_range:

		m = monitors[wir[0]:wir[1]]
		t = trackers[wir[0]:wir[1]]	
		pl = peer_lists[wir[0]:wir[1]]
		
		graph = utils.create_graph_master_tracker(ms, m, t, pl)
		# graph = utils.create_graph_peer_weights(ms, m, t, pl)
		
		graphs.append(graph)

		saves.show_graph(graph)



	saves.save_graphs_txt(graphs)
	logging.info(str(len(graphs)) + ' graphs in directory:' + vp.PATH_GRAPHS + '/')

	saves.save_graphs_fig(graphs)	
	logging.info(str(len(graphs)) + ' images graphs in directory:' + vp.PATH_FIGS + '/')

	saves.save_graph_adj_csv(graphs)
	saves.save_graph_weigths_csv(graphs)
	logging.info('adjacency and weight matrices are directory:' + vp.PATH_MATRICES + '/')


if __name__ == '__main__':
	main()