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

# Larson triad (system/mimalloc/HAKMEM), CSV保存
# Usage: benchmarks/scripts/run_larson_matrix.sh [dur_csv] [threads_csv] [reps]
#        dur_csv default: 2,10  threads_csv default: 1,4  reps default: 5

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")"/../.. && pwd)"
cd "$ROOT_DIR"

dur_csv=${1:-"2,10"}
thr_csv=${2:-"1,4"}
reps=${3:-5}

MIN=8; MAX=128; CHUNKS=1024; ROUNDS=1; SEED=12345
MI_LIB_DEFAULT="mimalloc-bench/extern/mi/out/release/libmimalloc.so"
MI_LIB="${MIMALLOC_SO:-$MI_LIB_DEFAULT}"

[[ -x ./larson_system ]] || make -s larson_system >/dev/null
if [[ -f "$MI_LIB" ]]; then
  [[ -x ./larson_mi ]] || make -s larson_mi >/dev/null
  HAVE_MI=1
else
  HAVE_MI=0
fi
[[ -x ./larson_hakmem ]] || make -s larson_hakmem >/dev/null

TS=$(date +%Y%m%d_%H%M%S)
OUTDIR="bench_results/auto/larson_${TS}"
mkdir -p "$OUTDIR"
CSV="$OUTDIR/results.csv"
echo "ts,scenario,dur_s,threads,allocator,env,rep,throughput_ops_s" >"$CSV"

IFS=',' read -ra DLIST <<<"$dur_csv"
IFS=',' read -ra TLIST <<<"$thr_csv"

extract_ops_s() {
  awk '/Throughput =/{print $3}' | tail -n1
}

run_case() {
  local dur="$1"; shift
  local thr="$1"; shift
  local alloc="$1"; shift
  local envstr="$1"; shift
  local rep="$2"; shift
  local ts=$(date +%H%M%S)
  local out
  out=$($envstr ./larson_${alloc} "$dur" "$MIN" "$MAX" "$CHUNKS" "$ROUNDS" "$SEED" "$thr" 2>/dev/null || true)
  local tput=$(echo "$out" | extract_ops_s)
  if [[ -n "${tput:-}" ]]; then
    echo "$ts,larson,$dur,$thr,$alloc,$(echo "$envstr" | sed 's/,/;/g'),$rep,$tput" >>"$CSV"
  fi
}

echo "[info] writing CSV to $CSV"
for d in "${DLIST[@]}"; do
  for t in "${TLIST[@]}"; do
    for ((i=1;i<=reps;i++)); do run_case "$d" "$t" system "env -i" "$i"; done
    if (( HAVE_MI == 1 )); then
      for ((i=1;i<=reps;i++)); do run_case "$d" "$t" mi "env -i LD_LIBRARY_PATH=$(dirname "$MI_LIB")" "$i"; done
    fi
    for ((i=1;i<=reps;i++)); do run_case "$d" "$t" hakmem "env -i HAKMEM_WRAP_TINY=1" "$i"; done
  done
done

echo "[done] $CSV"