collector.py

# Module to collect statistics for different simulation runs

class Collector:
	"Collects statistics about the simulation for visualization or aggregation"

	def __init__(self, statTable, verbose=False):
		self.stats = { }
		self.statTable = statTable
		self.debug = verbose

	def startRow(self, name):
		"This is called at the start of a row"
		self.stats[name] = { }
		self.tempList = [ ]

	def collect(self, x):
		self.tempList.append( x )

	def doneRow(self, name):
		"This is called when you're done with a row"	
		for (statName, func) in self.statTable.items():
			statValue = func(self.tempList)
			self.stats[ name ][ statName ] = statValue	

	def getRows(self):
		"Return all the rows as a list"
		return self.stats.keys()
	
	def __str__(self):
		res = "--------------\n"
		for (name, stat) in self.stats.items():
			res += str(name) + "\n"
			for (statName, statValue) in stat.items():
				res += "\t" + str(statName) +  " = " + str(statValue) + "\n"
		res += "-------------\n"
		return res 

	def extract(self,stat_name):
		# Extract all the statistics corresponding to statname
		res = []
		# print("Extracting ", stat_name)
		for (name, stat) in self.stats.items():
			for (statName, statValue) in stat.items():
				# print(statName)
				if (statName==stat_name):
					res.append(statValue)
		return res