script.py

# import os
# import matplotlib.pyplot as plt
# import numpy as np
#
# os.system("gcc -O0 tlb-size.c")
# page_size = 10
# for i in range(1000):
#     page_size+=100
#     # print(f"curr page size: {page_size}")
#     os.system(f"taskset -c 0 ./a.out {page_size} >> output.txt")
#


# import os
import re
import subprocess
import matplotlib

matplotlib.use("TkAgg")  # you're on i3wm
import matplotlib.pyplot as plt

C_FILE = "tlb-size.c"
OUTPUT = "output.txt"

# 1) Compile C
subprocess.run(["gcc", "-O0", C_FILE], check=True)

# 2) Run experiments (clear previous output first)
open(OUTPUT, "w").close()

page_size = 1
for i in range(200):
    page_size += 1
    # Run pinned to CPU 0, capture stdout, append to file
    res = subprocess.run(
        ["taskset", "-c", "0", "./a.out", str(page_size)],
        check=True,
        capture_output=True,
        text=True,
    )
    with open(OUTPUT, "a") as f:
        f.write(res.stdout)
        if not res.stdout.endswith("\n"):
            f.write("\n")  # ensure newline between runs

# 3) Parse output.txt
iters, pages, times = [], [], []
pat = re.compile(
    r"iter:\s*(\d+)\s+pages:\s*(\d+)\s+total_avg_access_time_for_10_runs:\s*([0-9]*\.?[0-9]+)"
)

with open(OUTPUT, "r") as f:
    for line in f:
        m = pat.search(line)
        if m:
            iters.append(int(m.group(1)))
            pages.append(int(m.group(2)))
            times.append(float(m.group(3)))

print(f"Parsed {len(iters)} lines from {OUTPUT}.")

if not iters:
    raise RuntimeError(
        "No data parsed. Check that your C program prints lines like:\n"
        "iter: <n> pages: <n> total_avg_access_time_for_10_runs: <time>"
    )

# 4) Plot: Iteration vs Time
plt.figure(figsize=(10, 6))
plt.plot(iters, times, marker="o", linestyle="-")
plt.xlabel("Iteration")
plt.ylabel("Average Access Time (ns)")
plt.title("TLB Experiment: Iteration vs Access Time")
plt.grid(True)
plt.tight_layout()
plt.show()

# Optional: if you also want Page Size vs Time, uncomment below:
# plt.figure(figsize=(10, 6))
# plt.plot(pages, times, marker="o", linestyle="-")
# plt.xlabel("Page Size (whatever unit your a.out expects)")
# plt.ylabel("Average Access Time (ns)")
# plt.title("TLB Experiment: Page Size vs Access Time")
# plt.grid(True)
# plt.tight_layout()
# plt.show()