plotwdl.py

"""
   Script to turn .wdl raw data into .png plots.
"""

import matplotlib.pyplot as plt


class wdldata:
    def __init__(self, prefix, dir=""):
        # Load d data from the file prefix.wdl
        self.prefix = prefix
        self.wl = []
        with open(dir + prefix + ".wdl") as f:
            for line in f:
                self.wl.append(1000 - int(line.split()[1]))
                _, _, self.bench = line.partition("bench ")
                self.bench, _, _ = self.bench.partition(" .epd")

    def create_graph(self, dir=""):
        dweek = self.wl[-168:]
        rangeMin, rangeMax = 0, 1000
        if min(self.wl) >= 800:
            rangeMin = 800
        elif min(self.wl) >= 500:
            rangeMin = 500
        if max(self.wl) <= 200:
            rangeMax = 200
        elif max(self.wl) <= 500:
            rangeMax = 500
        fig, ax = plt.subplots()
        ax.hist(
            self.wl,
            range=(rangeMin, rangeMax),
            bins=(rangeMax - rangeMin) // 10,
            density=True,
            alpha=0.5,
            color="red",
            edgecolor="yellow",
            label="all time",
        )
        ax.hist(
            dweek,
            range=(rangeMin, rangeMax),
            bins=(rangeMax - rangeMin) // 10,
            density=True,
            alpha=0.5,
            color="blue",
            edgecolor="black",
            label="last 7 days",
        )
        ax.legend()
        move, _, ply = self.prefix.partition("m")
        fig.suptitle(f"Distribution of W+L likelihood for 1. {move}")
        ax.set_title(
            f'(WDL data obtained by "stockfish bench {self.bench} {self.prefix}.epd depth" on {"leaf positions" if ply == "" else f"positions {ply} plies from leafs"} in {move}.poll)',
            fontsize=6,
            family="monospace",
        )
        xt = list(range(rangeMin, rangeMax + 1, (rangeMax - rangeMin) // 5))
        xtl = [str(x // 10) for x in xt]
        mmmm = [min(self.wl), max(self.wl), min(dweek), max(dweek)]
        for m in mmmm:
            if m not in xt:
                xt.append(m)
                xtl.append(f"{m/10:.1f}")
        ax.set_xticks(xt, xtl)
        plt.setp(
            ax.get_xticklabels()[6:],
            color="darkgray",
            position=(0, -0.03),
            rotation=45,
            ha="right",
            rotation_mode="anchor",
            fontsize=6,
            weight="bold",
        )
        plt.savefig(dir + self.prefix + "wdl.png", dpi=300)


for move in ["g4", "h4", "Na3", "Nh3", "f3"]:
    data = wdldata(move, dir="wdl/")
    data.create_graph(dir="images/")
    data = wdldata(move + "m6", dir="wdl/")
    data.create_graph(dir="images/")
    data = wdldata(move + "m12", dir="wdl/")
    data.create_graph(dir="images/")