#include "stdafx.h"
#include "TestRunner.h"

std::set<uint8_t> TestRunner::m_undocumented_opcodes;
bool TestRunner::m_undocumented_opcodes_initialised = false;

TestRunner::TestRunner() {}

EightBit::MemoryMapping TestRunner::mapping(const uint16_t address) noexcept {
    return { RAM(), 0x0000, 0xffff, EightBit::MemoryMapping::AccessLevel::ReadWrite };
}

void TestRunner::raisePOWER() {
    EightBit::Bus::raisePOWER();
    CPU().raisePOWER();
    CPU().raiseRESET();
    CPU().raiseINT();
    CPU().raiseNMI();
    CPU().raiseSO();
    CPU().raiseRDY();
}

void TestRunner::lowerPOWER() {
    CPU().lowerPOWER();
    EightBit::Bus::lowerPOWER();
}

void TestRunner::addActualCycle(const uint16_t address, const uint8_t value, std::string_view action) {
    m_actualCycles.push_back({ address, value, std::string(action) });
}

void TestRunner::addActualCycle(const EightBit::register16_t address, const uint8_t value, std::string_view action) {
    addActualCycle(address.word, value, action);
}

void TestRunner::addActualReadCycle(const EightBit::register16_t address, const uint8_t value) {
    addActualCycle(address, value, "read");
}

void TestRunner::addActualWriteCycle(const EightBit::register16_t address, const uint8_t value) {
    addActualCycle(address, value, "write");
}

void TestRunner::dumpCycles(std::string_view which, const actual_cycles_t& events) {
    m_messages.push_back(std::string(which));
    dumpCycles(events);
}

void TestRunner::dumpCycles(const actual_cycles_t& cycles) {
    for (const auto& cycle: cycles)
        dumpCycle(cycle);
}

void TestRunner::dumpCycle(const actual_cycle_t& cycle) {
    dumpCycle(std::get<0>(cycle), std::get<1>(cycle), std::get<2>(cycle));
}

void TestRunner::dumpCycles(std::string_view which, const cycles_t events) {
    m_messages.push_back(std::string(which));
    dumpCycles(events);
}

void TestRunner::dumpCycles(const cycles_t cycles) {
    for (const auto cycle: cycles)
        dumpCycle(cycle_t(cycle));
}

void TestRunner::dumpCycle(const cycle_t cycle) {
    dumpCycle(cycle.address(), cycle.value(), cycle.action());
}

void TestRunner::initialise() {

    seedUndocumentedOpcodes();

    ReadByte.connect([this](EightBit::EventArgs&) {
        addActualReadCycle(ADDRESS(), DATA());
    });

    WrittenByte.connect([this](EightBit::EventArgs&) {
        addActualWriteCycle(ADDRESS(), DATA());
    });

    os() << std::hex << std::uppercase;
}

void TestRunner::raise(std::string_view what, const uint16_t expected, const uint16_t actual) {
    os()
        << std::setw(2) << std::setfill(' ')
        << what
        << std::setw(4) << std::setfill('0')
        << ": expected: " << (int)expected
        << ", actual: " << (int)actual;
    pushCurrentMessage();
}

void TestRunner::raise(std::string_view what, const uint8_t expected, const uint8_t actual) {
    os()
        << std::setw(2) << std::setfill(' ')
        << what
        << std::setfill('0')
        << ": expected: " << (int)expected
            << " (" << EightBit::Disassembly::dump_Flags(expected) << ")"
        << ", actual: " << (int)actual
            << " (" << EightBit::Disassembly::dump_Flags(actual) << ")";
    pushCurrentMessage();
}

void TestRunner::raise(std::string_view what, std::string_view expected, std::string_view actual) {
    os()
        << std::setw(0) << std::setfill(' ')
        << what
        << ": expected: " << expected
        << ", actual: " << actual;
    pushCurrentMessage();
}

bool TestRunner::check(std::string_view what, const uint16_t address, const uint8_t expected, const uint8_t actual) {
    const auto success = actual == expected;
    if (!success) {
        os() << what << ": " << std::setw(4) << std::setfill('0') << (int)address;
        raise(os().str(), expected, actual);
    }
    return success;
}

void TestRunner::initialiseState() {

    const auto initial = test().initial();

    CPU().PC().word = initial.pc();
    CPU().S() = initial.s();
    CPU().A() = initial.a();
    CPU().X() = initial.x();
    CPU().Y() = initial.y();
    CPU().P() = initial.p();
    for (const auto entry : initial.ram()) {
        const byte_t byte(entry);
        RAM().poke(byte.address(), byte.value());
    }
}

bool TestRunner::checkState() {

    const auto expected_cycles = test().cycles();
    const auto actual_cycles = m_actualCycles;
    m_cycle_count_mismatch = expected_cycles.size() != actual_cycles.size();
    if (m_cycle_count_mismatch)
        return false;

    size_t actual_idx = 0;
    for (const auto expected_cycle : expected_cycles) {
        const auto expected = cycle_t(expected_cycle);
        const auto actual = actual_cycles.at(actual_idx++);   // actual could be less than expected
        check("Cycle address", expected.address(), std::get<0>(actual));
        check("Cycle value", expected.value(), std::get<1>(actual));
        check("Cycle action", expected.action(), std::string_view(std::get<2>(actual)));
    }

    const auto final = test().final();
    const auto pc_good = check("PC", final.pc(), CPU().PC().word);
    const auto s_good = check("S", final.s(), CPU().S());
    const auto a_good = check("A", final.a(), CPU().A());
    const auto x_good = check("X", final.x(), CPU().X());
    const auto y_good = check("Y", final.y(), CPU().Y());
    const auto p_good = check("P", final.p(), CPU().P());

    bool ram_problem = false;
    for (const auto entry : final.ram()) {
        const byte_t byte(entry);
        const auto ram_good = check("RAM", byte.address(), byte.value(), RAM().peek(byte.address()));
        if (!ram_good && !ram_problem)
            ram_problem = true;
    }

    return pc_good && s_good && a_good && x_good && y_good && p_good && !ram_problem;
}

void TestRunner::pushCurrentMessage() {
    m_messages.push_back(os().str());
    os().str("");
}

void TestRunner::disassemble(uint16_t address) {
    try {
        os() << m_disassembler.disassemble(address);
    } catch (const std::domain_error& error) {
        os() << "Disassembly problem: " << error.what();
    }
    pushCurrentMessage();
}

void TestRunner::check(const test_t updated) {

    m_test = updated;

    m_messages.clear();
    m_actualCycles.clear();
    m_cycles = 0;
    m_valid = true;
    m_undocumented = false;

    raisePOWER();
    initialiseState();
    const auto pc = CPU().PC().word;
    const auto start_opcode = peek(pc);
    m_cycles = CPU().step();
    lowerPOWER();

    m_valid = checkState();

    m_undocumented = m_undocumented_opcodes.find(start_opcode) != m_undocumented_opcodes.end();
    if (undocumented()) {
        m_messages.push_back("Undocumented");
        return;
    }

    if (unimplemented()) {
        m_messages.push_back("Unimplemented");
        return;
    }

    if (invalid() && implemented()) {

        disassemble(pc);

        const auto final = test().final();
        raise("PC", final.pc(), CPU().PC().word);
        raise("S", final.s(), CPU().S());
        raise("A", final.a(), CPU().A());
        raise("X", final.x(), CPU().X());
        raise("Y", final.y(), CPU().Y());
        raise("P", final.p(), CPU().P());

        os()
            << std::dec << std::setfill(' ')
            << "Stepped cycles: " << cycles()
            << ", expected events: " << test().cycles().size()
            << ", actual events: " << m_actualCycles.size();
        pushCurrentMessage();

        dumpCycles("-- Expected cycles", test().cycles());
        dumpCycles("-- Actual cycles", m_actualCycles);
    }
}

void TestRunner::seedUndocumentedOpcodes() {
    if (m_undocumented_opcodes_initialised) return;
    m_undocumented_opcodes = {
        0x02, 0x03, 0x04, 0x07, 0x0b, 0x0c, 0x0f,
        0x12, 0x13, 0x14, 0x17, 0x1a, 0x1b, 0x1c, 0x1f,
        0x22, 0x23, 0x27, 0x2b, 0x2f,
        0x32, 0x33, 0x34, 0x37, 0x3a, 0x3b, 0x3c, 0x3f,
        0x42, 0x43, 0x44, 0x47, 0x4b, 0x4f,
        0x52, 0x53, 0x54, 0x57, 0x5a, 0x5b, 0x5c, 0x5f,
        0x62, 0x63, 0x64, 0x67, 0x6b, 0x6f,
        0x72, 0x73, 0x74, 0x77, 0x7a, 0x7b, 0x7c, 0x7f,
        0x80, 0x82, 0x83, 0x87, 0x89, 0x8b, 0x8f,
        0x92, 0x93, 0x97, 0x9b, 0x9c, 0x9e, 0x9f,
        0xa3, 0xa7, 0xab, 0xaf,
        0xb2, 0xb3, 0xb7, 0xbb, 0xbf,
        0xc2, 0xc3, 0xc7, 0xcb, 0xcf,
        0xd2, 0xd3, 0xd4, 0xd7, 0xda, 0xdb, 0xdc, 0xdf,
        0xe2, 0xe3, 0xe7, 0xeb, 0xef,
        0xf1, 0xf2, 0xf3, 0xf4, 0xf7, 0xfa, 0xfb, 0xfc, 0xff,
    };
    m_undocumented_opcodes_initialised = true;
}