plot-lat2.py

#!/usr/bin/env python3

import argparse
import statistics
import matplotlib.pyplot as plt
import matplotlib
from myplotlib import get_marker, get_color, change_aspect_ratio

ratio = 1.5
fontsize = 14
lfontsize = 13.5
markersize = 5
linewidth = 2.4
lines = { "linewidth" : linewidth,
          "markersize" : markersize,
          "markeredgewidth" : linewidth, }
markers = { "fillstyle" : "none", }
plt.rc("lines", **lines)
plt.rc("markers", **markers)
matplotlib.rcParams['pdf.fonttype'] = 42
matplotlib.rcParams['ps.fonttype'] = 42


def parse(filename):
    
    rtts = []

    with open(filename, "r") as f:
        for line in f:
            rtts.append(float(line.strip().split(",")[3]) / 1000)

    print("%s: %d lines" % (filename, len(rtts)))

    #return statistics.median(rtts)
    return min(rtts)


def main():
    
    parser = argparse.ArgumentParser()
    parser.add_argument("-p", "--pattern", help = "access pattern",
                        choices = ["fix", "random"], default = "fix")
    args = parser.parse_args()

    pdffile = "graph/graph_lat2_{}.pdf".format(args.pattern)

    fig, ax = plt.subplots()

    yaxis = []
    xticks = []

    for x in range(1, 129):

        dma_len = x * 16
        xticks.append(dma_len)

        f = ("output/pciebench_lat2_" +
             "read_ptr-{}_len-{}.txt".format(args.pattern, dma_len))
                                             
        yaxis.append(parse(f))

    ax.plot(xticks, yaxis, color = get_color())
    
    plt.yticks(list(map(lambda x: x * 5, range(6))))
    plt.xticks([0, 512, 1024, 1536, 2048])

    ax.tick_params(labelsize = fontsize)
    ax.set_ylabel("latency (usec)", fontsize = fontsize)
    ax.set_xlabel("transfer size (byte)", fontsize = fontsize)
    
    ax.grid(True, linestyle = "--", linewidth = 0.5)

    change_aspect_ratio(ax, 2)

    print("save '{}'".format(pdffile))
    plt.savefig(pdffile, bbox_inches = "tight", pad_inches = 0.005)

if __name__ == "__main__":
    main()