class Stats:
	def __init__(self):
		self.received=0
		self.sent=0
		self.exchangedNodes=0
		self.transferredBlocks=0

stats=Stats()


def logReceived(data):
	stats.received+=len(data)


def logSent(data):
	stats.sent+=len(data)


def logExchangedNode(k=1):
	stats.exchangedNodes+=k


def logTransferredBlock(k=1):
	stats.transferredBlocks+=k


def reset():
	global stats
	stats=Stats()


def report():
	return """received {rf} ({r:,} B)
sent {sf} ({s:,} B)
exchanged {nodes:,} hash tree nodes
transferred {blocks:,} blocks""".format(rf=formatBytes(stats.received), r=stats.received, sf=formatBytes(stats.sent), s=stats.sent, nodes=stats.exchangedNodes, blocks=stats.transferredBlocks)


def formatBytes(x):
	exts=["B","kiB","MiB","GiB","TiB","PiB"]
	i=0
	while x>1024:
		x/=1024
		i+=1
	if x>=100: x=round(x)
	elif x>=10: x=round(x,1)
	else: x=round(x,2)
	return "{0} {1}".format(x,exts[i])