# Time and energy on x86_64 host A, DET1's procedure (linux-a-energy.sh): package energy (powercap
# package-0) around each of 7 rounds of the 8 pictures, 5 s idle brackets, at q2 and q6.
# Variants: int-before (DET1's binary), int-after (this change, kernel set auto, or KSET), float (again), int-gpu-after.
set -euo pipefail
R=$HOME/wai-build/idf; W=$R/work; export CARGO_TARGET_DIR=$R/target   # the toolchain: rustup 1.98.0 on PATH
TAG=${TAG:?binary tag}; PY=$R/venv/bin/python; O=$R/energy; mkdir -p $O
cd $R/src/wai
IDS=$(python3 -c "import json;print(','.join(p['id'] for p in json.load(open('$W/pictures.json'))['pictures'][:8]))")
E="python3 -I tools/wai_det_energy.py --rounds ${ROUNDS:-7} --k 1 --idle-s 5 --counter rapl"
HW='"system": "Linux 7.0", "hardware": "x86_64 SoC, 16 cores, 32 threads, AVX-512, integrated GPU PCI vendor 0x1002", "pictures": 8'
D=$(date +%F)
for q in ${QS:-2 6}; do
  for v in ${VARIANTS:-int-before int-after float}; do
    case $v in
      int-before) L=linux-int-cpu-q$q-before; CH="$R/bin/base serve $W/jobs.energy.q$q.tsv"; M='"path": "integer", "fabric": "CPU", "threads": "all (32)", "kernels": "the record'"'"'s (DET1 build, kernel set wide)"' ;;
      int-after) K=${KSET:-auto}; L=linux-int-cpu-q$q-after${KSET:+-$KSET}; CH="env WAI_INT_ISA=$K $R/bin/native.$TAG serve $W/jobs.energy.q$q.tsv"; M='"path": "integer", "fabric": "CPU", "threads": "all (32)", "kernels": "narrow, kernel set '$K'"' ;;
      int-gpu-after) L=linux-int-gpu-q$q-after; CH="env WAI_DET_GPU_KEY=gpu-vulkan-1002 $R/bin/gpu.$TAG --gpu serve $W/jobs.energy.q$q.tsv"; M='"path": "integer", "fabric": "GPU (Vulkan) for the graphs it takes, CPU for the rest", "threads": "all (32)", "kernels": "narrow, kernel set auto"' ;;
      float) L=linux-float-cpu-q$q-again; CH="$PY -I tools/wai_determinism_cost.py fserve $W $q --device cpu --ids $IDS"; M='"path": "float", "fabric": "CPU", "threads": "the library default"' ;;
    esac
    $E --label $L --meta "{$HW, $M, \"quality\": $q, \"nice\": 10, \"measured\": \"$D\"}" -- nice -n 10 $CH > $O/$L.json
    python3 - $O/$L.json <<'PY'
import json, sys, statistics as st
r = json.load(open(sys.argv[1]))
ms = [x["child_wall_ns"] / x["decodes"] / 1e6 for x in r["rounds"]]
j = [x["rapl"]["marginal_nj"] / x["decodes"] / 1e9 for x in r["rounds"]]
v = [x["rapl"].get("verdict") for x in r["rounds"]]
print(f'{r["label"]}: {st.median(ms):.1f} ms ({min(ms):.1f}-{max(ms):.1f}), {st.median(j):.2f} J ({min(j):.2f}-{max(j):.2f}), verdicts {v}')
PY
  done
done