EightBit/M6502/HarteTest_6502/TestRunner.h

#pragma once

#include <cstdint>
#include <sstream>
#include <string>
#include <vector>

#include <Bus.h>
#include <Ram.h>
#include <mos6502.h>

#include "test_t.h"

class TestRunner final : public EightBit::Bus {
private:
    EightBit::Ram m_ram = 0x10000;
    EightBit::MOS6502 m_cpu = { *this };
    const test_t& m_test;

    std::ostringstream m_os;
    std::vector<std::string> m_messages;

    cycles_t m_actualCycles;
    bool m_cycle_count_mismatch = false;

    void initialiseState();
    [[nodiscard]] bool checkState();

    void raise(std::string what, uint16_t expected, uint16_t actual);
    void raise(std::string what, uint8_t expected, uint8_t actual);
    void raise(std::string what, cycle_t::action_t expected, cycle_t::action_t actual);

    template<class T>
    bool check(std::string what, T expected, T actual) {
        const auto success = actual == expected;
        if (!success)
            raise(what, expected, actual);
        return success;
    }

    bool check(std::string what, uint16_t address, uint8_t expected, uint8_t actual);

    void addActualCycle(const cycle_t& value);
    void addActualCycle(uint16_t address, uint8_t value, cycle_t::action_t action);
    void addActualCycle(EightBit::register16_t address, uint8_t value, cycle_t::action_t action);

    void addActualReadCycle(EightBit::register16_t address, uint8_t value);
    void addActualWriteCycle(EightBit::register16_t address, uint8_t value);

    void dumpCycles(std::string which, const cycles_t& cycles);
    void dumpCycles(const cycles_t& cycles);
    void dumpCycle(const cycle_t& cycle);

    [[nodiscard]] auto& os() { return m_os; }

protected:
    virtual EightBit::MemoryMapping mapping(uint16_t address) noexcept final;

public:
    TestRunner(const test_t& test);

    virtual void raisePOWER() final;
    virtual void lowerPOWER() final;

    virtual void initialise() final;

    [[nodiscard]] constexpr auto& RAM() noexcept { return m_ram; }
    [[nodiscard]] constexpr auto& CPU() noexcept { return m_cpu; }
    [[nodiscard]] constexpr const auto& test() const noexcept { return m_test; }
    [[nodiscard]] constexpr const auto& messages() const noexcept { return m_messages; }

    [[nodiscard]] bool check();
};