94 lines
2.4 KiB
Python
Executable File
94 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- encoding: utf-8 -*-
|
|
|
|
import csv
|
|
import json
|
|
from csv2md import table_to_md as tbl2md
|
|
from io import StringIO
|
|
from pathlib import Path
|
|
|
|
|
|
def tbl2csv(tbl):
|
|
conv = StringIO()
|
|
csvw = csv.writer(conv)
|
|
for row in tbl:
|
|
csvw.writerow(list(map(str, row)))
|
|
return conv.getvalue()
|
|
|
|
|
|
def float_fmt(f):
|
|
return '%0.4f' % f
|
|
|
|
|
|
def kbps_fmt(f):
|
|
return '%0.1f kbps' % (f/1000,)
|
|
|
|
|
|
def avg(itr):
|
|
if len(itr) <= 0:
|
|
return 0
|
|
return sum(itr)/len(itr)
|
|
|
|
|
|
class Phrases:
|
|
def __init__(self, d=dict()):
|
|
self._d = d
|
|
|
|
def __call__(self, val):
|
|
return self[val]
|
|
|
|
def __getitem__(self, val):
|
|
return self._d.get(val, val)
|
|
|
|
|
|
def build_table(d, clmns, rws, fld, fmt, phrase_dict=dict()):
|
|
phrases = Phrases(phrase_dict)
|
|
tbl = [['' for x in range(len(clmns)+1)] for y in range(len(rws)+1)]
|
|
for i, clmn in enumerate(clmns):
|
|
tbl[0][i+1] = phrases[str(clmn)]
|
|
for i, rw in enumerate(rws):
|
|
tbl[i+1][0] = phrases[str(rw)]
|
|
for x, clmn in enumerate(clmns):
|
|
for y, rw in enumerate(rws):
|
|
tbl[y+1][x+1] = fmt(d[clmn][rw][fld])
|
|
tbl[0][0] = phrases[fld]
|
|
return tbl
|
|
|
|
|
|
def main(target='.'):
|
|
d = dict()
|
|
for stat in Path(target).glob('*.autotest.json'):
|
|
topo, algo = stat.name.split('.')[:-2]
|
|
jsdata = json.loads(stat.read_text())
|
|
spng = jsdata['sequential']['ping']['avg']
|
|
ppng = jsdata['parallel']['ping']['avg']
|
|
sprf = avg(jsdata['sequential']['iperfs'])
|
|
pprf = avg(jsdata['parallel']['iperfs'])
|
|
if topo not in d:
|
|
d[topo] = dict()
|
|
d[topo][algo] = {
|
|
'sequential_ping': spng,
|
|
'sequential_iperf': sprf,
|
|
'parallel_ping': ppng,
|
|
'parallel_iperf': pprf,
|
|
}
|
|
columns = sorted(list(d.keys()))
|
|
rows = ['ospf', 'lowlat']
|
|
fields = [
|
|
'sequential_ping',
|
|
'sequential_iperf',
|
|
'parallel_ping',
|
|
'parallel_iperf',
|
|
]
|
|
for field in fields:
|
|
formatter = float_fmt if 'ping' in field else kbps_fmt
|
|
tbl = build_table(d, columns, rows, field, formatter)
|
|
outcsvfile = Path(target).joinpath(f'autotest_table.{field}.csv')
|
|
outmdfile = Path(target).joinpath(f'autotest_table.{field}.md')
|
|
outcsvfile.write_text(tbl2csv(tbl))
|
|
outmdfile.write_text(tbl2md(tbl))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|