#include "egg/core/eggDisplay.h"

#include "egg/core/eggXfbManager.h"
#include "nw4r/db/db_directPrint.h"

#include "rvl/GX.h" // IWYU pragma: export
#include "rvl/OS.h" // IWYU pragma: export

extern "C" {
void PreRetraceCallback(u32 retraceCount) {
    EGG::BaseSystem::getDisplay()->preVRetrace();
}
}
namespace EGG {

u32 Display::sTickPeriod = ((OS_BUS_CLOCK_SPEED >> 2) / 125000) * 300 >> 3;

Display::Display(u8 maxRetrace)
    : mMaxRetraces(maxRetrace),
      mScreenStateFlag(0),
      mRetraceCount(0),
      mFrameCount(0),
      mClearColor(0x808080ff),
      mClearZ(0xFFFFFF),
      mBeginTick(0),
      mFlag(0) {
    mFlag.setBit(mFlag_SetClear);
    mFlag.setBit(mFlag_WaitForRetrace);
    nw4r::db::DirectPrint_Init();
    VISetPreRetraceCallback(PreRetraceCallback);
}

u32 Display::getTickPerFrame() {
    return Video::getTickPerVRetrace();
}

void Display::beginFrame() {
    GXDrawDone();
    if (mFlag.onBit(mFlag_WaitForRetrace) ||
        (BaseSystem::getXfbMgr()->mNumXfbs == 1 && OSGetTick() - mLastTick > sTickPeriod)) {
        VIWaitForRetrace();
    }
    // Holds while the next XFB gets to its copy state
    while (BaseSystem::getXfbMgr()->mToCopyXfb == 0) {
        VIWaitForRetrace();
    }
    copyEFBtoXFB();
    calcFrequency();
    mFrameCount++;
}

void Display::beginRender() {
    return;
}

void Display::endRender() {
    return;
}

void Display::endFrame() {
    return;
}

void Display::preVRetrace() {
    mLastTick = OSGetTick();
    if (VIGetRetraceCount() - mRetraceCount < mMaxRetraces) {
        return;
    }
    if (mScreenStateFlag.onBit(mScreenStateFlag_SetBlack)) {
        BaseSystem::getVideo()->setBlack(1);
        mScreenStateFlag.resetBit(mScreenStateFlag_SetBlack);
    }
    BaseSystem::getXfbMgr()->setNextFrameBuffer();
    mRetraceCount = VIGetRetraceCount();
}

void Display::copyEFBtoXFB() {
    if (mFlag.onBit(mFlag_SetClear)) {
        GXSetCopyClear(mClearColor, mClearZ);
    }
    GXRenderModeObj const *renderObj = BaseSystem::getVideo()->pRenderMode;
    GXSetCopyFilter(renderObj->aa, renderObj->sample_pattern, renderObj->aa == 0, renderObj->vfilter);
    BaseSystem::getXfbMgr()->copyEFB(mFlag.onBit(mFlag_SetClear));
}

void Display::calcFrequency() {
    s32 endTick = OSGetTick();
    mDeltaTick = endTick - mBeginTick;
    mFrequency = 1000000.0f / ((mDeltaTick * 8) / ((OS_BUS_CLOCK_SPEED >> 2) / 125000));
    mBeginTick = endTick;
}

} // namespace EGG