#!/usr/bin/env python3
import os, csv
import matplotlib.pyplot as plt
from collections import defaultdict

ROOT="/home/carlos/projects/gem5"
OUT_DATA=os.path.join(ROOT,"gem5-data","SmartEdgeAI","results")
OUT_IOT =os.path.join(ROOT,"iot","results")
src=os.path.join(OUT_DATA,"summary_energy.csv")
out_data=os.path.join(OUT_DATA,"fig_epi_across_workloads.png")
out_iot =os.path.join(OUT_IOT ,"fig_epi_across_workloads.png")

epi_by_core=defaultdict(list)
with open(src) as f:
    r=csv.DictReader(f)
    for row in r:
        insts=float(row['insts']); energy=float(row['energy_J'])
        epi = 1e12*energy/insts if insts>0 else 0.0
        epi_by_core[row['core']].append(epi)

cores=['big','little','hybrid']
vals=[sum(epi_by_core[c])/max(1,len(epi_by_core[c])) for c in cores]

plt.figure()
plt.bar(cores, vals)
plt.ylabel('EPI (pJ/inst)')
plt.title('Energy per Instruction across workloads (avg by core mode)')
plt.tight_layout(); plt.savefig(out_data); plt.savefig(out_iot)
print(f"[plot] wrote {out_data} and mirrored to {out_iot}")