#include "CMessageScreen.hpp"
#include "GameGlobalObjects.hpp"
#include "CSimplePool.hpp"
#include "GuiSys/CGuiModel.hpp"
#include "GuiSys/CGuiTextPane.hpp"
#include "CGameState.hpp"
#include "Input/CFinalInput.hpp"
#include "Audio/CSfxManager.hpp"
#include "GuiSys/CGuiWidgetDrawParms.hpp"

namespace urde::MP1
{

CMessageScreen::CMessageScreen(CAssetId msg, float delayTime)
: x74_delayTime(delayTime)
{
    x78_24_exit = false;
    x0_msg = g_SimplePool->GetObj({FOURCC('STRG'), msg});
    xc_msgScreen = g_SimplePool->GetObj("FRME_MsgScreen");
}

void CMessageScreen::ProcessControllerInput(const CFinalInput& input)
{
    if (!x18_loadedMsgScreen || x74_delayTime > 0.f || !input.PA())
        return;

    if (x1c_textpane_message->TextSupport().GetCurTime() <
        x1c_textpane_message->TextSupport().GetTotalAnimationTime())
    {
        x1c_textpane_message->TextSupport().SetCurTime(
            x1c_textpane_message->TextSupport().GetTotalAnimationTime());
        return;
    }

    x6c_page += 1;

    if (x6c_page >= x0_msg->GetStringCount())
    {
        x78_24_exit = true;
        return;
    }

    x1c_textpane_message->TextSupport().SetTypeWriteEffectOptions(false, 0.1f, 30.f);
    x1c_textpane_message->TextSupport().SetText(x0_msg->GetString(x6c_page));

    CSfxManager::SfxStart(SFXui_message_screen_key, 1.f, 0.f, false, 0x7f, false, kInvalidAreaId);

    x74_delayTime = 0.8f;
}

bool CMessageScreen::Update(float dt, float blurAmt)
{
    x70_blurAmt = blurAmt;
    if (!x18_loadedMsgScreen && xc_msgScreen.IsLoaded() && x0_msg.IsLoaded())
    {
        x18_loadedMsgScreen = xc_msgScreen.GetObj();
        x1c_textpane_message = static_cast<CGuiTextPane*>(x18_loadedMsgScreen->FindWidget("textpane_message"));
        x20_basewidget_top = x18_loadedMsgScreen->FindWidget("basewidget_top");
        x24_basewidget_center = x18_loadedMsgScreen->FindWidget("basewidget_center");
        x28_basewidget_bottom = x18_loadedMsgScreen->FindWidget("basewidget_bottom");
        x2c_model_abutton = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_abutton"));
        x30_model_top = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_top"));
        x38_model_bottom = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_bottom"));
        x34_model_center = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_center"));
        x3c_model_bg = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_bg"));
        x40_model_videoband = static_cast<CGuiModel*>(x18_loadedMsgScreen->FindWidget("model_videoband"));
        x44_topPos = x20_basewidget_top->GetLocalPosition();
        x50_bottomPos = x28_basewidget_bottom->GetLocalPosition();
        x5c_videoBandPos = x40_model_videoband->GetLocalPosition();

        if (CGuiWidget* w = x18_loadedMsgScreen->FindWidget("basewidget_top"))
            w->SetColor(g_tweakGuiColors->GetHudFrameColor());
        if (CGuiWidget* w = x18_loadedMsgScreen->FindWidget("basewidget_centerdeco"))
            w->SetColor(g_tweakGuiColors->GetHudFrameColor());
        if (CGuiWidget* w = x18_loadedMsgScreen->FindWidget("model_bottom"))
            w->SetColor(g_tweakGuiColors->GetHudFrameColor());

        x40_model_videoband->SetDepthGreater(true);
        x30_model_top->SetDepthWrite(true);
        x38_model_bottom->SetDepthWrite(true);
        x34_model_center->SetDepthWrite(true);
        x3c_model_bg->SetDepthWrite(true);

        if (x0_msg->GetStringCount())
        {
            x1c_textpane_message->TextSupport().SetTypeWriteEffectOptions(false, 0.1f, 30.f);
            x1c_textpane_message->TextSupport().SetText(x0_msg->GetString(0));
            x1c_textpane_message->TextSupport().SetFontColor(g_tweakGuiColors->GetHudMessageFill());
            x1c_textpane_message->TextSupport().SetControlTXTRMap(&g_GameState->GameOptions().GetControlTXTRMap());
        }
    }

    if (x18_loadedMsgScreen)
    {
        if (x74_delayTime > 0.f)
            x74_delayTime -= dt;

        float xT = std::max(0.f, (x70_blurAmt - 0.5f) / 0.5f);
        float scaleX;
        if (xT < 0.7f)
            scaleX = xT / 0.7f;
        else if (xT < 0.85f)
            scaleX = 0.1f * (1.f - (xT - 0.7f) / 0.15f) + 0.9f;
        else
            scaleX = 0.1f * ((xT - 0.7f - 0.15f) / 0.3f) + 0.9f;

        x24_basewidget_center->SetLocalTransform(zeus::CTransform::Scale(scaleX, 1.f, 1.f));
        x20_basewidget_top->SetLocalTransform(zeus::CTransform::Translate(0.f, 0.f, 12.f * (1.f - xT)));
        x28_basewidget_bottom->SetLocalTransform(zeus::CTransform::Translate(0.f, 0.f, -12.f * (1.f - xT)));

        float alpha = std::max(0.f, (x70_blurAmt - 0.7f) / 0.3f);
        zeus::CColor color = g_tweakGuiColors->GetHudFrameColor();
        color.a *= alpha;
        x20_basewidget_top->SetColor(color);
        x28_basewidget_bottom->SetColor(zeus::CColor(1.f, alpha));

        float pulse = 0.f;
        if (x74_delayTime <= 0.f)
            pulse = zeus::clamp(0.f, 0.5f * (1.f + std::sin(5.f * CGraphics::GetSecondsMod900() - M_PIF / 2.f)), 1.f);
        x2c_model_abutton->SetColor(zeus::CColor(1.f, pulse));

        x68_videoBandOffset += 12.f * dt;
        if (x68_videoBandOffset > 10.f)
            x68_videoBandOffset -= 20.f;

        x40_model_videoband->SetColor(zeus::CColor(1.f, 0.04f * (rand() / float(RAND_MAX)) + 0.08f));
        x40_model_videoband->SetLocalTransform(
            zeus::CTransform::Translate(x5c_videoBandPos + zeus::CVector3f(0.f, 0.f, x68_videoBandOffset)));

        x18_loadedMsgScreen->Update(dt);
    }

    return !x78_24_exit;
}

void CMessageScreen::Draw() const
{
    if (!x18_loadedMsgScreen)
        return;

    x18_loadedMsgScreen->Draw(CGuiWidgetDrawParms(x70_blurAmt, zeus::CVector3f::skZero));
}

}