import json
import logging
from typing import List

import analyzers
from analyzers import get_renderer, render
from analyzers.analyzer import ResultStore
from analyzers.analyzer.default import write_logentry_count_csv, write_simulation_flag_csv
from analyzers.render import wip
from analyzers.render.default import LogEntryCountCSV
from analyzers.render.wip import time_distribution, plot_data
from analyzers.settings import LogSettings, load_settings
from loaders import LOADERS
from util.processing import grep, run_analysis, src_file

logging.basicConfig(format='%(levelname)s %(name)s:%(message)s', level=logging.DEBUG)
log: logging.Logger = logging.getLogger(__name__)

logging.getLogger('requests').setLevel(logging.WARN)
logging.getLogger("urllib3").setLevel(logging.WARNING)


def urach_logs(log_ids, settings):
	# return ["data/inst_{id}.{format}".format(id=log_id, format=settings.log_format) for log_id in log_ids]
	return ["data/{id}.{format}".format(id=log_id, format=settings.log_format) for log_id in log_ids]


if __name__ == '__main__':
	settings: LogSettings = load_settings("biogames2.json")
	log_ids_urach: List[str] = urach_logs([
		# "34fecf49dbaca3401d745fb467",
		#	"44ea194de594cd8d63ac0314be",
		#	"57c444470dbf88605433ca935c",
		#	"78e0c545b594e82edfad55bd7f",
		#	"91abfd4b31a5562b1c66be37d9",
		# "597b704fe9ace475316c345903",
		# "e01a684aa29dff9ddd9705edf8",
		"597b704fe9ace475316c345903",
		"e01a684aa29dff9ddd9705edf8",
		"fbf9d64ae0bdad0de7efa3eec6",
		# "fbf9d64ae0bdad0de7efa3eec6",
		"fe1331481f85560681f86827ec",  # urach
		# "fe1331481f85560681f86827ec"]
		"fec57041458e6cef98652df625",
	]
		, settings)
	log_ids_gf = grep(["9d11b749c78a57e786bf5c8d28",  # filderstadt
					   "a192ff420b8bdd899fd28573e2",  # eichstätt
					   "3a3d994c04b1b1d87168422309",  # stadtökologie
					   "fe1331481f85560681f86827ec",  # urach
					   "96f6d9cc556b42f3b2fec0a2cb7ed36e"  # oberelsbach
					   ],
					  "/home/clemens/git/ma/test/src",
					  settings)
	log_ids = src_file("/home/clemens/git/ma/test/filtered_5_actions")

	#store: ResultStore = run_analysis(log_ids_gf, settings, LOADERS)
	#store: ResultStore = run_analysis(log_ids, settings, LOADERS)

	if False:
		for r in get_renderer(analyzers.LocomotionActionAnalyzer):
			r().render(store.get_all())
	if False:
		render(analyzers.LocationAnalyzer, store.get_all())
	# print(json.dumps(store.serializable(), indent=1))
	if False:
		for cat in store.get_categories():
			render(analyzers.ActivityMapper, store.get_category(cat), name=cat)
		# render(analyzers.ProgressAnalyzer, store.get_all())

	if False:
		from analyzers.postprocessing import graph

		g = graph.Cache(settings)
		g.run(store)
	if False:
		# render(analyzers.SimulationOrderAnalyzer, store.get_all())
		for cat in store.get_categories():
			data = store.get_category(cat)
			render(analyzers.SimulationOrderAnalyzer, data, name=cat)
	if False:
		write_logentry_count_csv(LogEntryCountCSV, store, render, analyzers)
	if False:
		write_simulation_flag_csv(store)
	if False:
		time_distribution(store)

	if True:
		# spatial_data = get_data_distance(store,relative_values=False)
		# temporal_data = get_data(store,relative_values=False)
		# spatial_data_rel = get_data_distance(store,relative_values=True)
		# temporal_data_rel = get_data(store,relative_values=True)
		# temporal_data_rel = json.load(open("temporal_rel.json"))
		# spatial_data_rel = json.load(open("spatial_rel.json"))
		# import IPython
		# IPython.embed()

		# print(json.dumps(get_all_data(store)))
		# json.dump(get_all_data(store), open("combined.json", "w"))
		# combined = get_all_data(store, sort=True, relative=True)
		# json.dump(combined, open("combined_rel.json", "w"))
		# combined = json.load(open("combined_rel.json"))
		combined = json.load(open("combined_total.json"))
		# plot_time_space_rel(combined, keys)
		plot_data(combined, wip.keys)


# for analyzers in analyzers:
#	if analyzers.name() in ["LogEntryCount", "ActionSequenceAnalyzer"]:
#		print(json.dumps(analyzers.result(), indent=2))

# for analyzers in analyzers:
#	if analyzers.name() in ["BoardDuration"]:
#		print(json.dumps(analyzers.result(), indent=2))
#		print(analyzers.render())

# coords = analyzers[1].render()
# with open("test.js", "w") as out:
#	out.write("coords = "+coords)