//-----------------------------------------------------------------------------

#include "prime.h"

#include <benchmark/benchmark.h>
#include <iostream>
#include <random>

//-----------------------------------------------------------------------------

const std::size_t N = 10'000;

const auto seed = std::random_device()();

//-----------------------------------------------------------------------------

template<auto f> void test(benchmark::State &state)
{
    using T = typename decltype(std::function{f})::argument_type;

    std::mt19937 generator;
    std::uniform_int_distribution<T> distribution;
    auto rng = [&] { return distribution(generator); };

    for (auto _ : state)
    {
        std::size_t count = 0;
        generator.seed(seed);
        for (std::size_t i = 0; i < N; ++i)
            if (f(rng()))
                ++count;

        state.counters["count"] = count;
    }
}

//-----------------------------------------------------------------------------

BENCHMARK(test<is_prime1<int>>);
BENCHMARK(test<is_prime2<int>>);
BENCHMARK(test<is_prime3<int>>);

BENCHMARK_MAIN();

//-----------------------------------------------------------------------------