from migen.flow.hooks import *

class EndpointReporter(EndpointSimHook):
	def __init__(self, endpoint):
		super().__init__(endpoint)
		self.reset()
	
	def reset(self):
		self.inactive = 0
		self.ack = 0
		self.nack = 0
	
	# Total number of cycles per token (inverse token rate)
	def cpt(self):
		return (self.inactive + self.nack + self.ack)/self.ack
	
	# Inactivity cycles per token (slack)
	def ipt(self):
		return self.inactive/self.ack
	
	# NAK cycles per token (backpressure)
	def npt(self):
		return self.nack/self.ack
	
	def report_str(self):
		if self.ack:
			return "C/T={:.2f}\nI/T={:.2f}\nN/T={:.2f}".format(self.cpt(), self.ipt(), self.npt())
		else:
			return "N/A"
	
	def on_ack(self):
		self.ack += 1
	
	def on_nack(self):
		self.nack += 1
		
	def on_inactive(self):
		self.inactive += 1

class DFGReporter(DFGHook):
	def __init__(self, dfg):
		super().__init__(dfg, lambda u, ep, v: EndpointReporter(u.actor.endpoints[ep]))

	def get_edge_labels(self):
		d = dict()
		for (u, v), eps in self.nodepair_to_ep.items():
			if len(eps) == 1:
				d[(u, v)] = list(eps.values())[0].report_str()
			else:
				d[(u, v)] = "\n".join(ep + ":\n" + reporter.report_str()
					for ep, reporter in eps)
		return d