1 files changed, 341 insertions, 1 deletions

diff --git a/src/egg/gfx/eggScreen.cpp b/src/egg/gfx/eggScreen.cpp
index 66b80c4e..af26dc25 100644
--- a/src/egg/gfx/eggScreen.cpp
+++ b/src/egg/gfx/eggScreen.cpp
@@ -1,3 +1,343 @@
 #include "egg/gfx/eggScreen.h"
 
-namespace EGG {} // namespace EGG
+#include "common.h"
+#include "egg/gfx/eggDrawGX.h"
+#include "egg/gfx/eggStateGX.h"
+#include "rvl/SC/scapi.h"
+
+
+// ported from OGWS, though this class has evolved a fair bit
+
+using namespace nw4r;
+
+namespace EGG {
+
+Screen::TVMode Screen::sTVMode;
+Screen::TVModeInfo Screen::sTVModeInfo[TV_MODE_MAX] = {
+    {
+     608, 456,
+     nw4r::math::VEC2(1.0526316f, 1.0f),
+     },
+    {
+     812, 456,
+     nw4r::math::VEC2(0.7881773, 1.0f),
+     },
+    {
+     640, 456,
+     nw4r::math::VEC2(1.0f, 1.0f),
+     }
+};
+
+Screen *Screen::spRoot = nullptr;
+
+Screen::ChangeTVModeFunc Screen::sChangeTVModeFunc = nullptr;
+void *Screen::spChangeTVModeFuncInfo = nullptr;
+
+math::VEC2 Screen::sCanvasScale(1.0f, 1.0f);
+math::VEC2 Screen::sCanvasOffset(0.0f, 0.0f);
+
+void Screen::Initialize(const u16 *maxX, const u16 *maxY, Screen *userRoot) {
+
+    sTVModeInfo[TV_MODE_4_3].width = maxX[0];
+    sTVModeInfo[TV_MODE_4_3].height = maxY[0];
+    
+    sTVModeInfo[TV_MODE_4_3].ratios.x = (f32)StateGX::s_widthEfb / (f32)sTVModeInfo[TV_MODE_4_3].width;
+    sTVModeInfo[TV_MODE_4_3].ratios.y = (f32)StateGX::s_heightEfb / (f32)sTVModeInfo[TV_MODE_4_3].height;
+    
+    sTVModeInfo[TV_MODE_16_9].width = maxX[1];
+    sTVModeInfo[TV_MODE_16_9].height = maxY[1];
+
+    sTVModeInfo[TV_MODE_16_9].ratios.x = (f32)StateGX::s_widthEfb / (f32)sTVModeInfo[TV_MODE_16_9].width;
+    sTVModeInfo[TV_MODE_16_9].ratios.y = (f32)StateGX::s_heightEfb / (f32)sTVModeInfo[TV_MODE_16_9].height;
+    
+    sTVModeInfo[TV_MODE_UNK_3].width = StateGX::s_widthEfb;
+    sTVModeInfo[TV_MODE_UNK_3].height = StateGX::s_heightEfb;
+    
+    sTVModeInfo[TV_MODE_UNK_3].ratios.x = 1.0f;
+    sTVModeInfo[TV_MODE_UNK_3].ratios.y = 1.0f;
+
+    static Screen defaultRoot;
+    defaultRoot.SetProjectionType(PROJ_ORTHO);
+    defaultRoot.SetCanvasMode(CANVASMODE_1);
+    defaultRoot.SetNearZ(0.0f);
+    defaultRoot.SetFarZ(1.0f);
+    defaultRoot.SetSize(math::VEC2(sTVModeInfo[sTVMode].width, sTVModeInfo[sTVMode].height));
+
+    userRoot = (userRoot == nullptr) ? &defaultRoot : userRoot;
+    spRoot = userRoot;
+    userRoot->mParent = nullptr;
+    SetTVMode(sTVMode);
+}
+
+Screen::Screen()
+    : Frustum(
+          PROJ_PERSP, math::VEC2(sTVModeInfo[sTVMode].width, sTVModeInfo[sTVMode].height), 10.0f, 100000.0f,
+          CANVASMODE_1
+      ) {
+    mParent = nullptr;
+    mPosition.x = 0.0f;
+    mPosition.y = 0.0f;
+    field_0x44.x = 0.0f;
+    field_0x44.y = 0.0f;
+    field_0x48.x = 0.0f;
+    field_0x48.y = 0.0f;
+    SetParent(nullptr);
+}
+
+Screen::Screen(f32 x, f32 y, f32 w, f32 h, const Screen *userRoot, CanvasMode canvasMode)
+    : Frustum(PROJ_PERSP, math::VEC2(w, h), 10.0f, 100000.0f, canvasMode) {
+    mParent = nullptr;
+    mPosition.x = x;
+    mPosition.y = y;
+    field_0x44.x = 0.0f;
+    field_0x44.y = 0.0f;
+    field_0x48.x = 0.0f;
+    field_0x48.y = 0.0f;
+    SetParent(userRoot);
+}
+
+Screen::Screen(const Screen &other)
+    : Frustum(other),
+      mPosition(other.mPosition),
+      mDataEfb(other.mDataEfb),
+      field_0x44(other.field_0x44),
+      field_0x48(other.field_0x48) {
+    SetParent(other.mParent);
+}
+
+void Screen::SetProjectionGX() const {
+    if (!(mFlags & 0x40)) {
+        f32 sx, sy, ox, oy;
+        GetGlobalScaleOffset(&sx, &sy, &ox, &oy);
+
+        SetGlobalScaleOffset(sCanvasScale.x, sCanvasScale.y, sCanvasOffset.x, sCanvasOffset.y);
+        Frustum::SetProjectionGX();
+
+        SetGlobalScaleOffset(sx, sy, ox, oy);
+    } else {
+        Frustum::SetProjectionGX();
+    }
+
+    const DataEfb &efb = GetDataEfb();
+
+    StateGX::GXSetViewport_(efb.vp.x1, efb.vp.y1, efb.vp.x2, efb.vp.y2, efb.vp.z1, efb.vp.z2);
+    StateGX::GXSetScissor_(efb.sc_x, efb.sc_y, efb.sc_w, efb.sc_h);
+    StateGX::GXSetScissorBoxOffset_(efb.sc_ox, efb.sc_oy);
+}
+
+void Screen::CopyToG3D(g3d::Camera cam) const {
+    if (!(mFlags & 0x40)) {
+        f32 sx, sy, ox, oy;
+        GetGlobalScaleOffset(&sx, &sy, &ox, &oy);
+
+        SetGlobalScaleOffset(sCanvasScale.x, sCanvasScale.y, sCanvasOffset.x, sCanvasOffset.y);
+        Frustum::CopyToG3D(cam);
+
+        SetGlobalScaleOffset(sx, sy, ox, oy);
+    } else {
+        Frustum::CopyToG3D(cam);
+    }
+
+    const DataEfb &efb = GetDataEfb();
+
+    f32 x1, x2, y1, y2;
+    f32 z1, z2;
+
+    y2 = efb.vp.y2;
+    x2 = efb.vp.x2;
+
+    y1 = efb.vp.y1;
+    x1 = efb.vp.x1;
+
+    z2 = efb.vp.z2;
+    z1 = efb.vp.z1;
+
+    cam.SetViewport(x1, y1, x2, y2);
+    cam.SetViewportZRange(z1, z2);
+    cam.SetScissor(efb.sc_x, efb.sc_y, efb.sc_w, efb.sc_h);
+    cam.SetScissorBoxOffset(efb.sc_ox, efb.sc_oy);
+}
+
+void Screen::CopyFromAnother(const Screen &other) {
+    Frustum::CopyFromAnother(other);
+
+    mPosition = other.mPosition;
+    field_0x44 = other.field_0x44;
+    field_0x48 = other.field_0x48;
+    mDataEfb = other.mDataEfb;
+    mParent = other.mParent;
+}
+
+void Screen::Reset(f32 x, f32 y, f32 w, f32 h, CanvasMode mode) {
+    mPosition.x = x;
+    mPosition.y = y;
+    mSize.x = w;
+    SetDirty(true);
+    mSize.y = h;
+    SetCanvasMode(mode);
+}
+
+void Screen::SetParent(const Screen *parent) {
+    const Screen *newParent = parent != nullptr ? parent : spRoot;
+
+    if (mParent != newParent) {
+        SetDirty(true);
+        mParent = newParent;
+    }
+}
+
+void Screen::SetUnkFlag8() {
+    if ((mFlags & 0x8) == 0) {
+        SetDirty(true);
+    }
+    mFlags |= 0x8;
+}
+
+void Screen::GetPosSizeInEfb() const {
+    // TODO
+    const TVMode tvMode = (!(mFlags & 0x20)) ? sTVMode : TV_MODE_4_3;
+    const TVModeInfo &tvRatio = sTVModeInfo[tvMode];
+
+    f32 *px = &mDataEfb.vp.x1;
+    f32 *py = &mDataEfb.vp.y1;
+    GetGlobalPos(px, py);
+
+    *px = sCanvasScale.x * *px;
+    *py = sCanvasScale.y * *py;
+
+    *px *= tvRatio.ratios.x;
+    *py *= tvRatio.ratios.y;
+
+    mDataEfb.sc_oy = 0;
+    mDataEfb.sc_ox = 0;
+
+    const f32 x = *px;
+    if (x < 0.0f) {
+        *px = 0.0f;
+
+        const s32 lx = static_cast<s32>(-x);
+        mDataEfb.sc_ox = lx - (lx & 0x1);
+    }
+
+    const f32 y = *py;
+    if (y < 0.0f) {
+        *py = 0.0f;
+
+        const s32 ly = static_cast<s32>(-y);
+        mDataEfb.sc_oy = ly - (ly & 0x1);
+    }
+
+    if (mFlags & 0x2) {
+        *px -= static_cast<s32>(*px) & 0x1;
+        *py -= static_cast<s32>(*py) & 0x1;
+    }
+
+    *px = static_cast<s32>(*px);
+    *py = static_cast<s32>(*py);
+
+    mDataEfb.vp.x2 = mSize.x;
+    mDataEfb.vp.y2 = mSize.y;
+
+    if (!(mFlags & 0x8)) {
+        mDataEfb.vp.x2 = mSize.x * tvRatio.ratios.x;
+        mDataEfb.vp.y2 = mSize.y * tvRatio.ratios.y;
+    }
+
+    if (mFlags & 0x4) {
+        mDataEfb.vp.x2 -= static_cast<s32>(mDataEfb.vp.x2) & 0x3;
+        mDataEfb.vp.y2 -= static_cast<s32>(mDataEfb.vp.y2) & 0x3;
+    }
+
+    mDataEfb.vp.x2 = (mDataEfb.vp.x2 >= 0.0f) ? mDataEfb.vp.x2 : 0.0f;
+    mDataEfb.vp.y2 = (mDataEfb.vp.y2 >= 0.0f) ? mDataEfb.vp.y2 : 0.0f;
+
+    mDataEfb.vp.x2 = static_cast<s32>(mDataEfb.vp.x2);
+    mDataEfb.vp.y2 = static_cast<s32>(mDataEfb.vp.y2);
+}
+
+const Screen::DataEfb &Screen::GetDataEfb() const {
+    if (IsChangeEfb()) {
+        GetPosSizeInEfb();
+        mDataEfb.vp.z1 = 0.0f;
+        mDataEfb.vp.z2 = 1.0f;
+        SetDirty(false);
+    }
+
+    return mDataEfb;
+}
+
+bool Screen::IsChangeEfb() const {
+    if (mFlags & FLAG_DIRTY) {
+        return true;
+    }
+
+    if (mParent != nullptr) {
+        return mParent->IsChangeEfb();
+    }
+
+    return false;
+}
+
+void Screen::CalcMatrixForDrawQuad(math::MTX34 *mtx, f32 x, f32 y, f32 sx, f32 sy) const {
+    PSMTXScale(*mtx, sx, sy, 1.0f);
+
+    mtx->m[0][3] = x;
+    mtx->m[1][3] = mCanvasMode == CANVASMODE_0 ? y - sy : y;
+    mtx->m[2][3] = 0.0f;
+}
+
+void Screen::FillBufferGX(u32 flags, GXColor color, u32 r6) const {
+    if (flags != 0) {
+        math::MTX34 drawMtx;
+        Screen clone(mPosition.x, mPosition.y, mSize.x, mSize.y, mParent, CANVASMODE_1);
+
+        if ((mFlags & 0x8) != 0) {
+            clone.SetUnkFlag8();
+        }
+
+        clone.SetNearZ(0.0f);
+        clone.SetFarZ(1.0f);
+        clone.SetFlag(0x40);
+        clone.SetProjectionType(PROJ_ORTHO);
+
+        clone.SetProjectionGX();
+        clone.CalcMatrixForDrawQuad(&drawMtx, 0.0f, 0.0f, mSize.x, mSize.y);
+        // Ugh
+        DrawGX::ClearEfb(drawMtx, (flags & 1) != 0, (flags & 2) != 0, (flags & 4) != 0, color, true);
+    }
+}
+
+void Screen::GetGlobalPos(f32 *ox, f32 *oy) const {
+    f32 px, py;
+    const Screen *parent = GetParent();
+
+    if (parent != nullptr) {
+        parent->GetGlobalPos(&px, &py);
+        parent->ConvertToCanvasLU(mPosition.x, mPosition.y, ox, oy);
+
+        *ox += px;
+        *oy += py;
+    } else {
+
+        *ox = mPosition.x;
+        *oy = mPosition.y;
+    }
+}
+
+void Screen::SetTVMode(TVMode tvMode) {
+    sTVMode = tvMode;
+    if (spRoot != nullptr) {
+        spRoot->SetSizeX(GetSizeXMax());
+        spRoot->SetSizeY(GetSizeYMax());
+    }
+
+    if (sChangeTVModeFunc != nullptr) {
+        sChangeTVModeFunc(spChangeTVModeFuncInfo);
+    }
+}
+
+void Screen::SetTVModeDefault() {
+    SetTVMode(SCGetAspectRatio() == SC_ASPECT_STD ? TV_MODE_4_3 : TV_MODE_16_9);
+}
+
+} // namespace EGG