#include <random>
#include <thread>

#include "data_structures/bloom/bloom_filter.hpp"
#include "data_structures/concurrent/concurrent_bloom_map.hpp"
#include "logging/default.hpp"
#include "logging/streams/stdout.hpp"
#include "utils/command_line/arguments.hpp"
#include "utils/hashing/fnv64.hpp"
#include "utils/random/generator.h"

#include "benchmark/benchmark_api.h"

/*
  ConcurrentMap Benchmark Test:
    - tests time of Insertion, Contain and Delete operations

    - benchmarking time per operation

    - test run ConcurrentMap with the following keys and values:
        - <int,int>
        - <int, string>
        - <string, int>
        - <string, string>
*/

using utils::random::NumberGenerator;
using utils::random::PairGenerator;
using utils::random::StringGenerator;
using StringHashFunction = std::function<uint64_t(const std::string &)>;

using IntegerGenerator = NumberGenerator<std::uniform_int_distribution<int>,
                                         std::default_random_engine, int>;

// Global arguments
int MAX_ELEMENTS = 1 << 18, MULTIPLIER = 2;
int THREADS, RANGE_START, RANGE_END, STRING_LENGTH;

/*
  ConcurrentMap Insertion Benchmark Test
*/
template <class K, class V, class F>
static void InsertValue(benchmark::State &state,
                        ConcurrentBloomMap<K, V, F> *map,
                        const std::vector<std::pair<K, V>> &elements)
{
    while (state.KeepRunning())
    {
        for (int start = 0; start < state.range(0); start++)
        {
            map->insert(elements[start].first, elements[start].second);
        }
    }
    state.SetComplexityN(state.range(0));
}

/*
  ConcurrentMap Contains Benchmark Test
*/
template <class K, class V, class F>
static void ContainsValue(benchmark::State &state,
                          ConcurrentBloomMap<K, V, F> *map,
                          const std::vector<std::pair<K, V>> elements)
{
    while (state.KeepRunning())
    {
        for (int start = 0; start < state.range(0); start++)
        {
            map->contains(elements[start].first);
        }
    }
    state.SetComplexityN(state.range(0));
}

auto BM_InsertValue = [](benchmark::State &state, auto *map, auto &elements) {
    InsertValue(state, map, elements);
};

auto BM_ContainsValue = [](benchmark::State &state, auto *map, auto elements) {
    ContainsValue(state, map, elements);
};

/*
  Commandline Argument Parsing

  Arguments:
   * Integer Range Minimum
      -start number

   *  Integer Range Maximum
      - end number

   * Number of threads
      - threads number

   * Random String lenght
      -string-length number
*/
void parse_arguments(int argc, char **argv)
{
    REGISTER_ARGS(argc, argv);

    RANGE_START = GET_ARG("-start", "0").get_int();
    RANGE_END   = GET_ARG("-end", "1000000000").get_int();

    THREADS = std::min(GET_ARG("-threads", "1").get_int(),
                       (int)std::thread::hardware_concurrency());

    STRING_LENGTH =
        ProgramArguments::instance().get_arg("-string-length", "128").get_int();
}

int main(int argc, char **argv)
{
    logging::init_async();
    logging::log->pipe(std::make_unique<Stdout>());

    parse_arguments(argc, argv);

    StringGenerator sg(STRING_LENGTH);
    IntegerGenerator ig(RANGE_START, RANGE_END);

    /*
      Creates RandomGenerators, ConcurentMaps and Random Element Vectors for the
      following use cases:

        Map elements contain keys and value for:
          <int, int>,
          <int, string>
          <string, int>
          <string, string>
    */

    // random generators for tests
    PairGenerator<IntegerGenerator, IntegerGenerator> piig(&ig, &ig);
    PairGenerator<StringGenerator, StringGenerator> pssg(&sg, &sg);
    PairGenerator<StringGenerator, IntegerGenerator> psig(&sg, &ig);
    PairGenerator<IntegerGenerator, StringGenerator> pisg(&ig, &sg);

    StringHashFunction hash1              = fnv64<std::string>;
    StringHashFunction hash2              = fnv1a64<std::string>;
    std::vector<StringHashFunction> funcs = {hash1, hash2};

    BloomFilter<std::string, 128> bloom_filter_(funcs);

    // maps used for testing
    // ConcurrentBloomMap<int, int> ii_map;
    // ConcurrentBloomMap<int, std::string> is_map;
    using Filter = BloomFilter<std::string, 128>;
    ConcurrentBloomMap<std::string, int, Filter> si_map(bloom_filter_);
    ConcurrentBloomMap<std::string, std::string, Filter> ss_map(bloom_filter_);

    // random elements for testing
    // auto ii_elems = utils::random::generate_vector(piig, MAX_ELEMENTS);
    // auto is_elems = utils::random::generate_vector(pisg, MAX_ELEMENTS);
    auto si_elems = utils::random::generate_vector(psig, MAX_ELEMENTS);
    auto ss_elems = utils::random::generate_vector(pssg, MAX_ELEMENTS);

    /* insertion Tests */
    benchmark::RegisterBenchmark("InsertValue[String, Int]", BM_InsertValue,
                                 &si_map, si_elems)
        ->RangeMultiplier(MULTIPLIER)
        ->Range(1, MAX_ELEMENTS)
        ->Complexity(benchmark::oN)
        ->Threads(THREADS);

    benchmark::RegisterBenchmark("InsertValue[String, String]", BM_InsertValue,
                                 &ss_map, ss_elems)
        ->RangeMultiplier(MULTIPLIER)
        ->Range(1, MAX_ELEMENTS)
        ->Complexity(benchmark::oN)
        ->Threads(THREADS);

    // Contains Benchmark Tests
    benchmark::RegisterBenchmark("ContainsValue[String, Int]", BM_ContainsValue,
                                 &si_map, si_elems)
        ->RangeMultiplier(MULTIPLIER)
        ->Range(1, MAX_ELEMENTS)
        ->Complexity(benchmark::oN)
        ->Threads(THREADS);

    benchmark::RegisterBenchmark("ContainsValue[String, String]",
                                 BM_ContainsValue, &ss_map, ss_elems)
        ->RangeMultiplier(MULTIPLIER)
        ->Range(1, MAX_ELEMENTS)
        ->Complexity(benchmark::oN)
        ->Threads(THREADS);

    benchmark::Initialize(&argc, argv);
    benchmark::RunSpecifiedBenchmarks();

    return 0;
}