#!/usr/bin/env bash
set -euo pipefail

# Sweep Tiny env knobs quickly to tune small-size hot path.
# Knobs:
#  - HAKMEM_SLL_MULTIPLIER ∈ {1,2,3}
#  - HAKMEM_TINY_REFILL_MAX ∈ {64,96,128}
#  - HAKMEM_TINY_REFILL_MAX_HOT ∈ {160,192,224}
#  - HAKMEM_TINY_MAG_CAP (global) ∈ {128,256}
#  - Optional: per-class MAG_CAP_C3=512 for 64B（フラグ --mag64-512）
#
# Usage: scripts/sweep_tiny_advanced.sh [cycles] [--mag64-512]

ROOT_DIR=$(cd "$(dirname "$0")/.." && pwd)
cd "$ROOT_DIR"

cycles=${1:-80000}
shift || true
MAG64=0
if [[ "${1:-}" == "--mag64-512" ]]; then MAG64=1; fi

make -s bench_fast >/dev/null

TS=$(date +%Y%m%d_%H%M%S)
OUTDIR="bench_results/sweep_tiny_adv_${TS}"
mkdir -p "$OUTDIR"
CSV="$OUTDIR/results.csv"
echo "size,sllmul,rmax,rmaxh,mag_cap,mag_cap_c3,throughput_mops" > "$CSV"

sizes=(16 32 64)
sllm=(1 2 3)
rmax=(64 96 128)
rmaxh=(160 192 224)
mags=(128 256)

run_case() {
  local size="$1"; shift
  local smul="$1"; shift
  local r1="$1"; shift
  local r2="$1"; shift
  local mcap="$1"; shift
  local mag64="$1"; shift
  local out
  if [[ "$size" == "64" && "$mag64" == "1" ]]; then
    HAKMEM_WRAP_TINY=1 \
    HAKMEM_TINY_TLS_SLL=1 \
    HAKMEM_SLL_MULTIPLIER="$smul" \
    HAKMEM_TINY_REFILL_MAX="$r1" \
    HAKMEM_TINY_REFILL_MAX_HOT="$r2" \
    HAKMEM_TINY_MAG_CAP="$mcap" \
    HAKMEM_TINY_MAG_CAP_C3=512 \
    ./bench_tiny_hot_hakmem "$size" 100 "$cycles" | sed -n 's/^Throughput: \([0-9.][0-9.]*\) M ops.*/\1/p' >"$OUTDIR/tmp.txt" || true
  else
    HAKMEM_WRAP_TINY=1 \
    HAKMEM_TINY_TLS_SLL=1 \
    HAKMEM_SLL_MULTIPLIER="$smul" \
    HAKMEM_TINY_REFILL_MAX="$r1" \
    HAKMEM_TINY_REFILL_MAX_HOT="$r2" \
    HAKMEM_TINY_MAG_CAP="$mcap" \
    ./bench_tiny_hot_hakmem "$size" 100 "$cycles" | sed -n 's/^Throughput: \([0-9.][0-9.]*\) M ops.*/\1/p' >"$OUTDIR/tmp.txt" || true
  fi
  out=$(cat "$OUTDIR/tmp.txt" || true)
  if [[ -n "$out" ]]; then
    echo "$size,$smul,$r1,$r2,$mcap,$([[ "$size" == "64" && "$mag64" == "1" ]] && echo 512 || echo -) ,$out" >> "$CSV"
  fi
}

for sz in "${sizes[@]}"; do
  for sm in "${sllm[@]}"; do
    for r1 in "${rmax[@]}"; do
      for r2 in "${rmaxh[@]}"; do
        for mc in "${mags[@]}"; do
          echo "[sweep-adv] size=$sz mul=$sm rmax=$r1 hot=$r2 mag=$mc mag64=$( [[ "$MAG64" == "1" ]] && echo 512 || echo - ) cycles=$cycles"
          run_case "$sz" "$sm" "$r1" "$r2" "$mc" "$MAG64"
        done
      done
    done
  done
done

echo "[done] CSV: $CSV"
sed -n '1,40p' "$CSV" || true