#include <gtest/gtest.h>
#include <BS_thread_pool.hpp>

#include "utils/common.h"
#include "helper/codec.h"
#include "helper/sample.h"
#include "raw_code/raw_code.h"
#include "all_cases/all_cases.h"
#include "common_code/common_code.h"

using klotski::range_reverse;

using klotski::codec::RawCode;
using klotski::codec::CommonCode;

using klotski::cases::AllCases;
using klotski::cases::ALL_CASES_NUM_;

TEST(RawCode, validity) {
    EXPECT_FALSE(RawCode::check(0x0A34'182B'3810'2D21)); // invalid code
    EXPECT_FALSE(RawCode::check(0x8603'EDF5'CAFF'F5E2)); // high 4-bits not zero

    EXPECT_FALSE(RawCode::create(0x0000'0000'0000'0000).has_value()); // invalid code
}

TEST(RawCode, operators) {
    auto raw_code = RawCode::unsafe_create(TEST_R_CODE);

    std::ostringstream tmp;
    tmp << raw_code; // ostream capture
    EXPECT_TRUE(tmp.str().starts_with("603EDF5CAFFF5E2\n"));
    EXPECT_EQ((uint64_t)raw_code, TEST_R_CODE); // convert as uint64_t

    EXPECT_EQ(TEST_R_CODE, raw_code); // uint64_t == RawCode
    EXPECT_EQ(raw_code, TEST_R_CODE); // RawCode == uint64_t
    EXPECT_EQ(raw_code, raw_code); // RawCode == RawCode

    EXPECT_NE(TEST_R_CODE + 1, raw_code); // uint64_t != RawCode
    EXPECT_NE(raw_code, TEST_R_CODE + 1); // RawCode != uint64_t
    EXPECT_NE(raw_code, RawCode::unsafe_create(TEST_R_CODE + 1)); // RawCode != RawCode
}

TEST(RawCode, exporter) {
    auto raw_code = RawCode::unsafe_create(TEST_R_CODE);
    EXPECT_EQ(raw_code.unwrap(), TEST_R_CODE);
    EXPECT_EQ(raw_code.to_common_code(), TEST_C_CODE);
}

TEST(RawCode, initializate) {
    auto raw_code = RawCode::unsafe_create(TEST_R_CODE);
    auto common_code = CommonCode::unsafe_create(TEST_C_CODE);

    // RawCode(...)
    EXPECT_EQ(RawCode(common_code), TEST_R_CODE);
    EXPECT_EQ(RawCode(raw_code), TEST_R_CODE); // l-value
    EXPECT_EQ(RawCode(RawCode(raw_code)), TEST_R_CODE); // r-value

    // RawCode::create(uint64_t)
    EXPECT_TRUE(RawCode::create(TEST_R_CODE).has_value());
    EXPECT_FALSE(RawCode::create(TEST_R_CODE_ERR).has_value());
    EXPECT_EQ(RawCode::create(TEST_R_CODE), TEST_R_CODE);

    // RawCode::unsafe_create(uint64_t)
    EXPECT_EQ(RawCode::unsafe_create(TEST_R_CODE), TEST_R_CODE);

    // RawCode::from_common_code(CommonCode)
    EXPECT_EQ(RawCode::from_common_code(common_code), TEST_R_CODE);

    // RawCode::from_common_code(uint64_t)
    EXPECT_TRUE(RawCode::from_common_code(TEST_C_CODE).has_value());
    EXPECT_FALSE(RawCode::from_common_code(TEST_C_CODE_ERR).has_value());
    EXPECT_EQ(RawCode::from_common_code(TEST_C_CODE), TEST_R_CODE);

    // RawCode::from_common_code(const std::string &)
    EXPECT_TRUE(RawCode::from_common_code(TEST_C_CODE_STR).has_value());
    EXPECT_FALSE(RawCode::from_common_code(TEST_C_CODE_STR_ERR).has_value());
    EXPECT_EQ(RawCode::from_common_code(TEST_C_CODE_STR), TEST_R_CODE);

    // RawCode::from_common_code(std::string &&)
    EXPECT_TRUE(RawCode::from_common_code(TEST_C_CODE_STR_RV).has_value());
    EXPECT_FALSE(RawCode::from_common_code(TEST_C_CODE_STR_ERR_RV).has_value());
    EXPECT_EQ(RawCode::from_common_code(TEST_C_CODE_STR_RV), TEST_R_CODE);
}

TEST(RawCode, code_verify) {
    BS::thread_pool pool;
    pool.detach_sequence(0, 16, [](const uint64_t head) {
        for (const auto range : AllCases::instance().fetch()[head]) {
            const auto code = RawCode::from_common_code(head << 32 | range);
            EXPECT_TRUE(code.has_value());
            EXPECT_TRUE(RawCode::check(code->unwrap()));
            EXPECT_EQ(code->to_common_code(), head << 32 | range);
        }
    });
    pool.wait();
}

TEST(RawCode, DISABLED_global_verify) {
    auto force_convert = [](uint64_t common_code) -> uint64_t {
        auto range = range_reverse((uint32_t)common_code);
        auto raw_code = K_MASK_2x2 << (common_code >> 32) * 3;
        for (int addr = 0; range; range >>= 2) {
            while ((raw_code >> addr) & 0b111 && addr < 60) // found next space
                addr += 3;
            if (addr >= 60) // invalid address
                return 0;
            switch (range & 0b11) {
                case 0b01: raw_code |= K_MASK_1x2 << addr; break;
                case 0b10: raw_code |= K_MASK_2x1 << addr; break;
                case 0b11: raw_code |= K_MASK_1x1 << addr; break;
                case 0b00: addr += 3;
            }
        }
        return raw_code;
    };

    BS::thread_pool pool;
    auto futures = pool.submit_blocks(0ULL, 0x10'0000'0000ULL, [&force_convert](uint64_t start, uint64_t end) {
        std::vector<uint64_t> archive;
        for (uint64_t common_code = start; common_code < end; ++common_code) {
            if (RawCode::check(force_convert(common_code))) {
                archive.emplace_back(common_code); // store valid raw code
            }
        }
        return archive;
    }, 0x1000); // split as 4096 pieces

    std::vector<uint64_t> result;
    result.reserve(ALL_CASES_NUM_);
    for (auto &future : futures) {
        const auto data = future.get();
        result.insert(result.end(), data.begin(), data.end()); // combine sections
    }
    EXPECT_EQ(result, all_common_codes());
}