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

# Simple runner for larson contention benchmark without LD_PRELOAD.
# Builds larson variants if missing and runs with given presets.

dur=2
threads_csv="1,4"
min_sz=8
max_sz=128
chunks=1024
rounds=1
seed=12345

while getopts ":d:t:m:M:c:r:s:" opt; do
  case $opt in
    d) dur="$OPTARG";;
    t) threads_csv="$OPTARG";;
    m) min_sz="$OPTARG";;
    M) max_sz="$OPTARG";;
    c) chunks="$OPTARG";;
    r) rounds="$OPTARG";;
    s) seed="$OPTARG";;
    *) echo "Usage: $0 [-d sec] [-t threads_csv] [-m min] [-M max] [-c chunks] [-r rounds] [-s seed]"; exit 1;;
  esac
done

if [[ ! -x ./larson_system ]]; then
  make -s larson_system >/dev/null
fi
if [[ ! -x ./larson_hakmem ]]; then
  make -s larson_hakmem >/dev/null
fi

MI_SO="${MIMALLOC_SO:-mimalloc-bench/extern/mi/out/release/libmimalloc.so}"
have_mi=0
if [[ -f "$MI_SO" ]]; then
  have_mi=1
  if [[ ! -x ./larson_mi ]]; then
    make -s larson_mi >/dev/null || have_mi=0
  fi
fi

IFS=',' read -ra tlist <<<"$threads_csv"
printf "\nLarson benchmark (min=%s, max=%s, chunks=%s, rounds=%s, seed=%s)\n" "$min_sz" "$max_sz" "$chunks" "$rounds" "$seed"
printf "Duration: %ss\n\n" "$dur"

printf "%-10s %-5s %s\n" "Allocator" "Thr" "Output"
printf "%-10s %-5s %s\n" "---------" "----" "------"
for t in "${tlist[@]}"; do
  out=$(./larson_system "$dur" "$min_sz" "$max_sz" "$chunks" "$rounds" "$seed" "$t" 2>/dev/null | rg "Throughput" -n || true)
  printf "%-10s %-5s %s\n" "system" "$t" "$out"

  if (( have_mi == 1 )); then
    out=$(./larson_mi "$dur" "$min_sz" "$max_sz" "$chunks" "$rounds" "$seed" "$t" 2>/dev/null | rg "Throughput" -n || true)
    printf "%-10s %-5s %s\n" "mimalloc" "$t" "$out"
  fi

  # Gate fast paths to isolate correctness first; toggle via env if needed
  HAKMEM_TINY_META_FREE=${HAKMEM_TINY_META_FREE:-0} \
  HAKMEM_TINY_META_ALLOC=${HAKMEM_TINY_META_ALLOC:-0} \
  HAKMEM_DISABLE_BATCH=${HAKMEM_DISABLE_BATCH:-1} \
  # Enable partial adopt for multi-thread runs to improve reuse
  if [[ "$t" -gt 1 ]]; then
    HAKMEM_TINY_SS_ADOPT=1 HAKMEM_TINY_SS_ADOPT_COOLDOWN=${HAKMEM_TINY_SS_ADOPT_COOLDOWN:-4} \
      out=$(./larson_hakmem "$dur" "$min_sz" "$max_sz" "$chunks" "$rounds" "$seed" "$t" 2>/dev/null | rg "Throughput" -n || true)
  else
    out=$(./larson_hakmem "$dur" "$min_sz" "$max_sz" "$chunks" "$rounds" "$seed" "$t" 2>/dev/null | rg "Throughput" -n || true)
  fi
  printf "%-10s %-5s %s\n" "hakmem" "$t" "$out"
done