#include "benchmark/benchmark.h"

#include <cstdlib>
#include "hb.h"

void RandomSet(unsigned size, unsigned max_value, hb_set_t* out) {
  hb_set_clear(out);

  srand(size * max_value);
  for (unsigned i = 0; i < size; i++) {
    while (true) {
      unsigned next = rand() % max_value;
      if (hb_set_has (out, next)) continue;

      hb_set_add(out, next);
      break;
    }
  }
}

static void BM_SetInsert(benchmark::State& state) {
  unsigned set_size = state.range(0);
  unsigned max_value = state.range(0) * state.range(1);

  hb_set_t* original = hb_set_create ();
  RandomSet(set_size, max_value, original);
  assert(hb_set_get_population(original) == set_size);

  for (auto _ : state) {
    state.PauseTiming();
    hb_set_t* data = hb_set_copy(original);
    state.ResumeTiming();

    hb_set_add(data, rand() % max_value);

    state.PauseTiming();
    hb_set_destroy(data);
    state.ResumeTiming();
  }

  hb_set_destroy(original);
}
BENCHMARK(BM_SetInsert)
    ->ArgsProduct({
        benchmark::CreateRange(1 << 10, 1 << 16, 8), // Set size
        benchmark::CreateDenseRange(2, 8, /*step=*/2) // Density
      });

BENCHMARK_MAIN();