KReplaySystem.cc

#include "KReplaySystem.hh"
 
#include "host/Option.hh"
#include "host/SceneCreatorDynamic.hh"
 
#include <abstract/File.hh>
 
#include <game/system/RaceManager.hh>
 
#include <iomanip>
 
void KReplaySystem::init() {
    ASSERT(m_currentGhostFileName);
    ASSERT(m_currentRawGhost);
    ASSERT(m_currentGhost);
 
    auto *sceneCreator = new Host::SceneCreatorDynamic;
    m_sceneMgr = new EGG::SceneManager(sceneCreator);
 
    System::RaceConfig::RegisterInitCallback(OnInit, nullptr);
    Abstract::File::Remove("results.txt");
 
    m_sceneMgr->changeScene(0);
}
 
void KReplaySystem::calc() {
    m_sceneMgr->calc();
}
 
bool KReplaySystem::run() {
    while (!calcEnd()) {
        calc();
    }
 
    return success();
}
 
void KReplaySystem::parseOptions(int argc, char **argv) {
    if (argc < 2) {
        PANIC("Expected ghost argument!");
    }
 
    for (int i = 0; i < argc; ++i) {
        std::optional<Host::EOption> flag = Host::Option::CheckFlag(argv[i]);
        if (!flag || *flag == Host::EOption::Invalid) {
            WARN("Expected a flag! Got: %s", argv[i]);
            continue;
        }
 
        switch (*flag) {
        case Host::EOption::Ghost: {
            ASSERT(i + 1 < argc);
 
            m_currentGhostFileName = argv[++i];
            m_currentRawGhost = Abstract::File::Load(m_currentGhostFileName, m_currentRawGhostSize);
 
            if (m_currentRawGhostSize < System::RKG_HEADER_SIZE ||
                    m_currentRawGhostSize > sizeof(System::RawGhostFile)) {
                PANIC("File cannot be a ghost! Check the file size.");
            }
 
            // Creating the raw ghost file validates it
            System::RawGhostFile file = System::RawGhostFile(m_currentRawGhost);
 
            m_currentGhost = new System::GhostFile(file);
            ASSERT(m_currentGhost);
        } break;
        case Host::EOption::Invalid:
        default:
            PANIC("Invalid flag!");
            break;
        }
    }
}
 
KReplaySystem *KReplaySystem::CreateInstance() {
    ASSERT(!s_instance);
    s_instance = new KReplaySystem;
    return static_cast<KReplaySystem *>(s_instance);
}
 
void KReplaySystem::DestroyInstance() {
    ASSERT(s_instance);
    auto *instance = s_instance;
    s_instance = nullptr;
    delete instance;
}
 
KReplaySystem::KReplaySystem()
    : m_currentGhostFileName(nullptr), m_currentGhost(nullptr), m_currentRawGhost(nullptr),
      m_currentRawGhostSize(0) {}
 
KReplaySystem::~KReplaySystem() {
    if (s_instance) {
        s_instance = nullptr;
        WARN("KReplaySystem instance not explicitly handled!");
    }
 
    delete m_sceneMgr;
    delete m_currentGhost;
    delete m_currentRawGhost;
}
 
bool KReplaySystem::calcEnd() const {
    constexpr u16 MAX_MINUTE_COUNT = 10;
 
    const auto *raceManager = System::RaceManager::Instance();
    if (raceManager->stage() == System::RaceManager::Stage::FinishGlobal) {
        return true;
    }
 
    if (raceManager->timerManager().currentTimer().min >= MAX_MINUTE_COUNT) {
        return true;
    }
 
    return false;
}
 
void KReplaySystem::reportFail(const std::string &msg) const {
    std::string report(m_currentGhostFileName);
    report += "\n" + std::string(msg);
    Abstract::File::Append("results.txt", report.c_str(), report.size());
}
 
bool KReplaySystem::success() const {
    auto format = [](const System::Timer &timer) {
        std::ostringstream oss;
        oss << std::setw(2) << std::setfill('0') << timer.min << ":" << std::setw(2)
            << std::setfill('0') << timer.sec << "." << std::setw(3) << std::setfill('0')
            << timer.mil;
        return oss.str();
    };
 
    const auto *raceManager = System::RaceManager::Instance();
    if (raceManager->stage() != System::RaceManager::Stage::FinishGlobal) {
        m_sceneMgr->currentScene()->heap()->enableAllocation();
        reportFail("Race didn't finish");
        return false;
    }
 
    s32 desyncingTimerIdx = getDesyncingTimerIdx();
    if (desyncingTimerIdx != -1) {
        m_sceneMgr->currentScene()->heap()->enableAllocation();
        std::string msg;
 
        const auto [correct, incorrect] = getDesyncingTimer(desyncingTimerIdx);
        if (desyncingTimerIdx == 0) {
            msg = "Final timer desync!";
        } else {
            msg = "Lap " + std::to_string(desyncingTimerIdx) + " timer desync!";
        }
 
        msg += " Expected " + format(correct) + ", got " + format(incorrect);
        reportFail(msg);
        return false;
    }
 
    return true;
}
 
s32 KReplaySystem::getDesyncingTimerIdx() const {
    const auto &player = System::RaceManager::Instance()->player();
    if (m_currentGhost->raceTimer() != player.raceTimer()) {
        return 0;
    }
 
    for (size_t i = 0; i < 3; ++i) {
        if (m_currentGhost->lapTimer(i) != player.getLapSplit(i + 1)) {
            return i + 1;
        }
    }
 
    return -1;
}
 
KReplaySystem::DesyncingTimerPair KReplaySystem::getDesyncingTimer(s32 i) const {
    auto cond = i <=> 0;
    ASSERT(cond != std::strong_ordering::less);
 
    if (cond == std::strong_ordering::equal) {
        const auto &correct = m_currentGhost->raceTimer();
        const auto &incorrect = System::RaceManager::Instance()->player().raceTimer();
        ASSERT(correct != incorrect);
        return DesyncingTimerPair(correct, incorrect);
    } else if (cond == std::strong_ordering::greater) {
        const auto &correct = m_currentGhost->lapTimer(i - 1);
        const auto &incorrect = System::RaceManager::Instance()->player().lapTimer(i - 1);
        ASSERT(correct != incorrect);
        return DesyncingTimerPair(correct, incorrect);
    }
 
    // This is unreachable
    return DesyncingTimerPair(System::Timer(), System::Timer());
}
 
void KReplaySystem::OnInit(System::RaceConfig *config, void * /* arg */) {
    config->setGhost(Instance()->m_currentRawGhost);
    config->raceScenario().players[0].type = System::RaceConfig::Player::Type::Ghost;
}