#include <src/protocol/generation.h>
#include <iostream>
#include <array>
#include <vector>

using namespace ts::protocol;


typedef std::vector<std::pair<size_t, size_t>> test_vector_t;

test_vector_t generate_test_vector(size_t size, int loss) {
    test_vector_t result{};
    result.reserve(size);


    for(size_t i = 0; i < size; i++) {
        if ((rand() % 100) < loss) continue;
        result.emplace_back(i & 0xFFFFU, i >> 16U);
    }

    return result;
}

test_vector_t swap_elements(test_vector_t vector, int per, int max_distance) {
    for(size_t index = 0; index < vector.size() - max_distance; index++) {
        if ((rand() % 100) < per) {
            //lets switch
            auto offset = rand() % max_distance;
            if(!offset) offset = 1;

            std::swap(vector[index], vector[index + offset]);
        }
    }

    return vector;
}

bool test_vector(const std::string_view& name, const test_vector_t& vector) {
    generation_estimator gen{};

    size_t last_value{0};
    for(auto [id, exp] : vector) {
        if(auto val = gen.visit_packet(id); val != exp) {
            std::cout << "[" << name << "] failed for " << id << " -> " << exp << " | " << val << ". Last value: " << last_value << "\n";
            return false;
        }
        last_value = id;
    }
    return true;
}

template <size_t N>
bool test_vector(generation_estimator& generator, const std::array<uint16_t, N>& packet_ids, const std::array<uint16_t, N>& expected) {
    for(size_t index = 0; index < N; index++) {
        auto result = generator.visit_packet(packet_ids[index]);
        if(result != expected[index]) {
            std::cout << "failed to packet id " << packet_ids[index] << " (" << index << "). Result: " << result << " Expected: " << expected[index] << "\n";
            std::cout << "----- fail\n";
            return false;
        }

        std::cout << "PacketID: " << packet_ids[index] << " -> " << result << "\n";
    }
    std::cout << "----- pass\n";

    return true;
}

int main() {
    generation_estimator gen{};

    {
        test_vector("00 loss", generate_test_vector(0x3000, 0));
        test_vector("10 loss", generate_test_vector(0x3000, 10));
        test_vector("25 loss", generate_test_vector(0x3000, 25));
        test_vector("50 loss", generate_test_vector(0x3000, 50));
        test_vector("80 loss", generate_test_vector(0x3000, 80));
    }

    {
        auto base = generate_test_vector(0x3000, 0);
        test_vector("swap 30:20", swap_elements(base, 30, 20));
        test_vector("swap 30:1000", swap_elements(base, 30, 200));
        test_vector("swap 80:1000", swap_elements(base, 80, 200));
    }

    if(false) {
        test_vector("10 loss", generate_test_vector(0x3000, 10));
        test_vector("25 loss", generate_test_vector(0x3000, 25));
        test_vector("50 loss", generate_test_vector(0x3000, 50));
        test_vector("80 loss", generate_test_vector(0x3000, 80));
    }

    gen.set_last_state(0, 0);
    test_vector<6>(gen, {0, 1, 2, 4, 3, 5}, {0, 0, 0, 0, 0, 0});

    gen.set_last_state(0xFF00, 0);
    test_vector<6>(gen, {0, 1, 2, 4, 3, 5}, {1, 1, 1, 1, 1, 1});

    gen.set_last_state(0xFF00, 0);
    test_vector<6>(gen, {0, 1, 2, 0xFF00, 3, 5}, {1, 1, 1, 0, 1, 1});

    gen.set_last_state(0xFF00, 0);
    test_vector<6>(gen, {0xFFFE, 0xFFFF, 0, 1, 0xFFFC, 2}, {0, 0, 1, 1, 0, 1});
}