PyDraw

Draw.py

import re
import matplotlib.pyplot as plt
import numpy as np

class ResultReader:

    def __init__(self):
        self.lines = list()
        self.op = ""
        self.dataName = ""
        self.algoData = {"defaultTree": dict(), "fbsTree": dict(),
                         "starTree": dict(), "strTree": dict()}
        self.algoFormat = {"defaultTree": "ro-", "fbsTree": "b^-",
                           "starTree": "bs-", "strTree": "yo-"}

    def read(self, path):
        with open(path, 'r') as f:
            self.lines = f.readlines()

    def get_match(self):
        for l in self.lines:
            for k in self.algoData:
                patM = "(?<=" + k + self.dataName + "M).*(?=" + self.op + ")"
                if not (self.dataName in l and k in l and self.op in l):
                    continue

                m = int(re.search(patM, l).group(0))
                s = float(str(re.search("[0-9]+\.[0-9]+(?= \$)", l).group(0)))  # please change the '+-' sign into '$' first

                self.algoData[k][m] = s
                break

    def draw(self, save, filename):
        fig, ax = plt.subplots()
        for algo in self.algoData:
            xaxis = list()
            yaxis = list()
            for k in self.algoData[algo]:
                xaxis.append(k)
                yaxis.append(self.algoData[algo][k])

            ax.plot(np.array(xaxis), np.array(yaxis), self.algoFormat[algo], label=algo)

        legend = ax.legend(loc='upper center', shadow=True, fontsize='x-large')
        legend.get_frame().set_facecolor('#00FFCC')

        if save:
            plt.savefig(filename)
        else:
            plt.show()


if __name__ == "__main__":

    opList = ["Construction", "SearchOne", "SearchNearest", "InsertOne", "InsertBatch", "DeleteOne", "DeleteBatch", ]
    dataList = ["1k", "Greek"]

    rr = ResultReader()
    rr.read("rtreeexp.txt")

    for _op in opList:
        for _ds in dataList:
            rr.op = _op
            rr.dataName = _ds
            rr.get_match()
            rr.draw(True, _op + "." + _ds + ".png")