calculateMetrics.py

import sys
from ecosystemDataManager.ecosystemDataManager import EcosystemDataManager

def isValidArguments(arguments):
	options = ["evaluate", "globalrate", "globalmean", "contextsize", "proportion"]
	for argument in arguments:
		if argument not in options:
			return False
	return True

if __name__ == '__main__':
	if len(sys.argv) < 2 or not isValidArguments(sys.argv[2:]):
		print("Usage:", sys.argv[0], "<ecosystem> [evaluate] [globalrate] [globalmean] [contextsize] [proportion]")
		sys.exit(1)
	if len(sys.argv) == 2:
		print("no options provided. calculating all metrics")
		options = ["evaluate", "globalrate", "globalmean", "contextsize", "proportion"]
	else:
		options = [argument for argument in sys.argv[2:]]
	ecosystem = sys.argv[1]
	ecosystemDataManager = EcosystemDataManager(ecosystem)
	if "evaluate" in options:
		print("evaluating edges")
		ecosystemDataManager.evaluateEdges()
		print("saving")
		ecosystemDataManager.save()
		print("done")
	else:
		print("assuming that edges are already evaluated")
	if "globalrate" in options or "globalmean" in options:
		if "globalrate" in options and "globalmean" in options:
			print("calculating both globalrate and globalmean metrics")
			ecosystemDataManager.calculateGlobalRegularityMetrics()
		else:
			print("calculate globalrate and globalmean together is faster")
			if "globalrate" in options:
				print("calculating globalrate")
				ecosystemDataManager.calculateGlobalRegularityRate()
			elif "globalmean" in options:
				print("calculating globalmean")
				ecosystemDataManager.calculateGlobalRegularityMean()
		print("saving")
		ecosystemDataManager.save()
		print("done")
	if "contextsize" in options:
		print("calculating contextsize")
		ecosystemDataManager.calculateContextSize()
		print("saving")
		ecosystemDataManager.save()
		print("done")
	if "proportion" in options:
		print("calculating proportion")
		ecosystemDataManager.proportion()
		print("in case rates or means has to been calculated: saving")
		ecosystemDataManager.save()
		print("done")