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

ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
ARTIFACTS_DIR="${ARTIFACTS_DIR:-$ROOT_DIR/artifacts}"
REPORT_PREFIX="${REPORT_PREFIX:-vector_ann_bench}"
CASE="${CASE:-all}"
OUT_LOG="${OUT_LOG:-$ARTIFACTS_DIR/${REPORT_PREFIX}_${CASE}.log}"

mkdir -p "$ARTIFACTS_DIR"
cd "$ROOT_DIR"

if [[ "$CASE" == "all" ]]; then
  cargo bench --bench vector_ann | tee "$OUT_LOG"
else
  cargo bench --bench vector_ann -- "$CASE" | tee "$OUT_LOG"
fi

echo "vector_ann_bench_log: $OUT_LOG"