#!/bin/bash
# SPDX-License-Identifier: GPL-2.0

source ./benchs/run_common.sh

set -eufo pipefail

htab_mem()
{
 echo -n "per-prod-op: "
 echo -n "$*" | sed -E "s/.* per-prod-op\s+([0-9]+\.[0-9]+ ± [0-9]+\.[0-9]+k\/s).*/\1/"
 echo -n -e ", avg mem: "
 echo -n "$*" | sed -E "s/.* memory usage\s+([0-9]+\.[0-9]+ ± [0-9]+\.[0-9]+MiB).*/\1/"
 echo -n ", peak mem: "
 echo "$*" | sed -E "s/.* peak memory usage\s+([0-9]+\.[0-9]+MiB).*/\1/"
}

summarize_htab_mem()
{
 local bench="$1"
 local summary=$(echo $2 | tail -n1)

 printf "%-20s %s\n" "$bench" "$(htab_mem $summary)"
}

htab_mem_bench()
{
 local name

 for name in overwrite batch_add_batch_del add_del_on_diff_cpu
 do
  summarize_htab_mem "$name" "$($RUN_BENCH htab-mem --use-case $name -p8 "$@")"
 done
}

header "preallocated"
htab_mem_bench "--preallocated"

header "normal bpf ma"
htab_mem_bench