1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
|
import matplotlib.pyplot as plt
import numpy as np
import json
import sys
BAR_WIDTH=0.25
def load_data(filename):
data = {}
with open(filename) as f:
line = f.readline()
while line:
for l in json.loads(line):
gbps = l["totalSize"] * 8 / l["totalElapsedSecond"] / 1024 / 1024 / 1024
file = l["url"].split("/")[3]
if file not in data:
data[file] = 0.0
data[file] += gbps
line = f.readline()
return data
labels=['blk-1k', 'blk-32k', 'blk-512k', 'blk-1m', 'blk-32m', 'blk-128m', 'blk-512m', 'blk-1g']
plt.ylabel("Gbps")
data_num = len(sys.argv)-2
factor = (data_num+1) * BAR_WIDTH
for i in range(0, data_num):
filename = sys.argv[1+i]
data = load_data(filename)
value = []
for l in labels:
value.append(data[l])
plt.bar([x*factor+(BAR_WIDTH*i) for x in range(0, len(labels))], value, align="edge", edgecolor="black", linewidth=1, width=BAR_WIDTH, label=filename)
plt.legend()
plt.xlim(0, (len(labels)-1)*factor+BAR_WIDTH*data_num)
plt.xticks([x*factor+BAR_WIDTH*data_num/2 for x in range(0, len(labels))], labels)
plt.savefig(sys.argv[1+data_num])
|