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

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

BENCH_FILE="${1:-benchmark_full_result.txt}"
if [[ ! -f "$BENCH_FILE" ]]; then
  echo "benchmark file not found: $BENCH_FILE" >&2
  exit 1
fi

TARGETS=(finder_full default_full)
MODES=(ystripes noindex)
RUNS=5

find /tmp -maxdepth 1 -type f -name "tzf_mem_finder_full_*" -delete || true
find /tmp -maxdepth 1 -type f -name "tzf_mem_default_full_*" -delete || true

for target in "${TARGETS[@]}"; do
  for mode in "${MODES[@]}"; do
    for ((i=1; i<=RUNS; i++)); do
      out_file="/tmp/tzf_mem_${target}_${mode}_${i}.out"
      time_file="/tmp/tzf_mem_${target}_${mode}_${i}.time"

      if [[ "$(uname)" == "Darwin" ]]; then
        /usr/bin/time -l cargo run --release --no-default-features --features full \
          --example index_memory_probe_full -- "$target" "$mode" >"$out_file" 2>"$time_file"
      else
        /usr/bin/time -v cargo run --release --no-default-features --features full \
          --example index_memory_probe_full -- "$target" "$mode" >"$out_file" 2>"$time_file"
      fi
    done
  done
done

python3 "$ROOT_DIR/scripts/bench_memory_full_report.py" "$BENCH_FILE"