visualize.py

#!/usr/bin/env python3
import sys
from matplotlib import patches
from matplotlib import pyplot
from random import shuffle
from itertools import dropwhile, takewhile

from LogReader import LogReader

if len(sys.argv) != 2 and len(sys.argv) != 4:
	print("Usage: {} <simulation.log> [from to]".format(sys.argv[0]))
	sys.exit(1)

LINEWIDTH = 2
ROW = {"ALG" : 1, "ADV": -1}
COLORS = ["brown", "orange", "pink", "magenta", "yellow", "purple", "blue", "green", "red"]
# shuffle(COLORS)

pyplot.rcParams["figure.figsize"] = [5.75, 3.0]

reader = LogReader(sys.argv[1])
xlim = (float(sys.argv[2]), float(sys.argv[3])) if len(sys.argv) == 4 else None

colormap = {}
colorlegend = []
def chooseColor(length):
	if length not in colormap:
		color = COLORS.pop()
		colormap[length] = color
		colorPatch = patches.Patch(color=color, label=str(length))
		colorlegend.append(colorPatch)
	return colormap[length]

def plotPacket(algorithm, packetEntry):
	y = ROW[algorithm]
	length = packetEntry.end - packetEntry.start
	color = chooseColor(packetEntry.packet)
	eb = pyplot.errorbar(packetEntry.start, y, xerr=[[0], [length]], capsize=10, capthick=LINEWIDTH, linewidth=LINEWIDTH, ecolor=color)
	eb[-1][0].set_linestyle('-' if packetEntry.successful else '--')

def plotError(time):
	pyplot.plot([time, time], [-2, 2], ':', linewidth=LINEWIDTH, color='black')

def readPackets(packets):
	return takewhile(lambda pe: pe.end <= xlim[1], dropwhile(lambda pe: pe.end < xlim[0], packets)) if xlim else packets

def readErrors():
	return takewhile(lambda t: t <= xlim[1], dropwhile(lambda t: t < xlim[0], reader.errors)) if xlim else reader.errors

for packetEntry in readPackets(reader.algPackets):
	plotPacket("ALG", packetEntry)

for packetEntry in readPackets(reader.advPackets):
	plotPacket("ADV", packetEntry)

for time in readErrors():
	plotError(time)

pyplot.yticks([-1, 1], ['ADV', 'ALG'])
pyplot.ylim(-2, 3)
if xlim:
	pyplot.xlim(*xlim)
pyplot.legend(handles=colorlegend, ncol=2)

pyplot.show()