#include "ViewManager.hpp"
#include "specter/Control.hpp"
#include "specter/Space.hpp"
#include "specter/Menu.hpp"
#include "SplashScreen.hpp"
#include "locale/locale.hpp"
#include "ResourceBrowser.hpp"
#include "icons/icons.hpp"
#include "badging/Badging.hpp"
#include "Runtime/Particle/CGenDescription.hpp"
#include "Runtime/Particle/CElectricDescription.hpp"
#include "Runtime/Particle/CSwooshDescription.hpp"
#include "Runtime/Graphics/CModel.hpp"
#include "Runtime/Graphics/CGraphics.hpp"
#include "Runtime/Character/CSkinRules.hpp"
#include "Graphics/CMetroidModelInstance.hpp"
#include "World/CWorldTransManager.hpp"
#include "Graphics/Shaders/CColoredQuadFilter.hpp"
#include "Graphics/Shaders/CTexturedQuadFilter.hpp"
#include "Audio/CStreamAudioManager.hpp"
#include "Runtime/CStateManager.hpp"
#include "Runtime/World/CPlayer.hpp"
#include "hecl/Pipeline.hpp"
#include <cstdio>

using YAMLNode = athena::io::YAMLNode;

extern hecl::SystemString ExeDir;

namespace urde
{

void ViewManager::InitMP1(MP1::CMain& main)
{
    main.Init(m_fileStoreManager, &m_cvarManager, m_mainWindow.get(), m_voiceEngine.get(), *m_amuseAllocWrapper);
    if (!m_noShaderWarmup)
        main.WarmupShaders();

    m_testGameView.reset(new TestGameView(*this, m_viewResources, *m_rootView));

    m_rootView->accessContentViews().clear();
    m_rootView->accessContentViews().push_back(m_testGameView.get());
    m_rootView->updateSize();
}

void ViewManager::TestGameView::resized(const boo::SWindowRect& root, const boo::SWindowRect& sub)
{
    specter::View::resized(root, sub);
    urde::CGraphics::SetViewportResolution({sub.size[0], sub.size[1]});
    if (m_debugText)
    {
        boo::SWindowRect newSub = sub;
        newSub.location[1] = 5 * m_vm.m_viewResources.pixelFactor();
        m_debugText->resized(root, newSub);
    }
}

void ViewManager::TestGameView::draw(boo::IGraphicsCommandQueue* gfxQ)
{
    m_vm.m_projManager.mainDraw();
    if (m_debugText && g_StateManager && g_StateManager->Player())
        m_debugText->draw(gfxQ);
}

void ViewManager::TestGameView::think()
{
    if (!m_debugText)
    {
        m_debugText.reset(
                    new specter::MultiLineTextView(m_vm.m_viewResources, *this, m_vm.m_viewResources.m_monoFont18));
        boo::SWindowRect sub = subRect();
        sub.location[1] = 5 * m_vm.m_viewResources.pixelFactor();
        m_debugText->resized(rootView().subRect(), sub);
    }

    if (m_debugText && g_StateManager)
    {
        std::string overlayText;
        const hecl::CVar* showFrameIdx = hecl::CVarManager::instance()->findCVar("debugOverlay.showFrameCounter");
        const hecl::CVar* playerInfo = hecl::CVarManager::instance()->findCVar("debugOverlay.playerInfo");
        const hecl::CVar* worldInfo = hecl::CVarManager::instance()->findCVar("debugOverlay.worldInfo");
        const hecl::CVar* areaInfo = hecl::CVarManager::instance()->findCVar("debugOverlay.areaInfo");
        if (showFrameIdx && showFrameIdx->toBoolean())
            overlayText += hecl::Format("Frame: %d\n",
                                        g_StateManager->GetUpdateFrameIndex());

        if (g_StateManager->Player() && playerInfo &&  playerInfo->toBoolean())
        {
            const CPlayer& pl = g_StateManager->GetPlayer();
            zeus::CQuaternion plQ = zeus::CQuaternion(pl.GetTransform().getRotation().buildMatrix3f());
            zeus::CTransform camXf = g_StateManager->GetCameraManager()->GetCurrentCameraTransform(*g_StateManager);
            zeus::CQuaternion camQ = zeus::CQuaternion(camXf.getRotation().buildMatrix3f());
            overlayText += hecl::Format("Player Position: x %f, y %f, z %f\n"
                                        "       Quaternion: w %f, x %f, y %f, z %f\n"
                                        "       Roll: %f, Pitch: %f, Yaw: %f\n"
                                        "Camera Position: x %f, y %f, z %f\n"
                                        "       Quaternion: w %f, x %f, y %f, z %f\n"
                                        "       Roll: %f, Pitch: %f, Yaw: %f\n",
                                        pl.GetTranslation().x, pl.GetTranslation().y, pl.GetTranslation().z,
                                        plQ.w, plQ.x, plQ.y, plQ.z,
                                        plQ.roll(), plQ.pitch(), plQ.yaw(),
                                        camXf.origin.x, camXf.origin.y, camXf.origin.z,
                                        camQ.w, camQ.x, camQ.y, camQ.z,
                                        camQ.roll(), camQ.pitch(), camQ.yaw());
        }
        if (worldInfo && worldInfo->toBoolean())
        {
            TLockedToken<CStringTable> tbl =
                    g_SimplePool->GetObj({FOURCC('STRG'), g_StateManager->GetWorld()->IGetStringTableAssetId()});
            const urde::TAreaId aId = g_GameState->CurrentWorldState().GetCurrentAreaId();
            overlayText += hecl::Format("World: 0x%08X%s, Area: %i\n",
                                        u32(g_GameState->CurrentWorldAssetId().Value()),
                                        (tbl.IsLoaded() ? (" " + hecl::Char16ToUTF8(tbl->GetString(0))).c_str() : ""), aId);
        }

        const urde::TAreaId aId = g_GameState->CurrentWorldState().GetCurrentAreaId();
        if (areaInfo && areaInfo->toBoolean() && g_StateManager->WorldNC() && g_StateManager->WorldNC()->DoesAreaExist(aId))
        {
            const auto& layerStates = g_GameState->CurrentWorldState().GetLayerState();
            std::string layerBits;
            u32 totalActive = 0;
            for (u32 i = 0; i < layerStates->GetAreaLayerCount(aId); ++i)
            {
                if (layerStates->IsLayerActive(aId, i))
                {
                    ++totalActive;
                    layerBits += "1";
                }
                else
                    layerBits += "0";
            }
            overlayText += hecl::Format("Area AssetId: 0x%08X, Total Objects: %i, Total Layers: %i, Total Active Layers: %i\n"
                                        "Active Layer bits: %s\n",
                                        (unsigned int)g_StateManager->WorldNC()->GetArea(aId)->GetAreaAssetId().Value(),
                                        g_StateManager->GetAllObjectList().size(), layerStates->GetAreaLayerCount(aId),
                                        totalActive, layerBits.c_str());
        }

        if (!overlayText.empty())
            m_debugText->typesetGlyphs(overlayText);
    }
}

specter::View* ViewManager::BuildSpaceViews()
{
    m_rootSpaceView = m_rootSpace->buildSpaceView(m_viewResources);
    return m_rootSpaceView;
}

specter::RootView* ViewManager::SetupRootView()
{
    m_rootView.reset(new specter::RootView(*this, m_viewResources, m_mainWindow.get()));
    m_rootView->setBackground(zeus::CColor::skBlack);
    return m_rootView.get();
}

SplashScreen* ViewManager::SetupSplashView()
{
    m_splash.reset(new SplashScreen(*this, m_viewResources));
    if (!m_showSplash)
        m_splash->close(true);
    return m_splash.get();
}

void ViewManager::RootSpaceViewBuilt(specter::View* view)
{
    std::vector<specter::View*>& cViews = m_rootView->accessContentViews();
    cViews.clear();
    cViews.push_back(view);
    cViews.push_back(m_splash.get());
    m_rootView->updateSize();
}

void ViewManager::ProjectChanged(hecl::Database::Project& proj)
{
    CDvdFile::Shutdown();
    CDvdFile::Initialize(hecl::ProjectPath(proj.getProjectWorkingPath(), _SYS_STR("out/files")));
}

void ViewManager::SetupEditorView()
{
    m_rootSpace.reset(new RootSpace(*this));

    SplitSpace* split = new SplitSpace(*this, nullptr, specter::SplitView::Axis::Horizontal);
    m_rootSpace->setChild(std::unique_ptr<Space>(split));
    split->setChildSlot(0, std::make_unique<ResourceBrowser>(*this, split));
    split->setChildSlot(1, std::make_unique<ResourceBrowser>(*this, split));

    BuildSpaceViews();
}

void ViewManager::SetupEditorView(ConfigReader& r)
{
    m_rootSpace.reset(Space::NewRootSpaceFromConfigStream(*this, r));
    BuildSpaceViews();
}

void ViewManager::SaveEditorView(ConfigWriter& w)
{
    if (!m_rootSpace)
        return;
    m_rootSpace->saveState(w);
}

void ViewManager::DismissSplash()
{
    if (!m_showSplash)
        return;
    m_showSplash = false;
    m_splash->close();
}

ViewManager::ViewManager(hecl::Runtime::FileStoreManager& fileMgr, hecl::CVarManager& cvarMgr)
: m_fileStoreManager(fileMgr)
, m_cvarManager(cvarMgr)
, m_projManager(*this)
, m_fontCache(fileMgr)
, m_translator(urde::SystemLocaleOrEnglish())
, m_recentProjectsPath(hecl::SysFormat(_SYS_STR("%s/recent_projects.txt"), fileMgr.getStoreRoot().data()))
, m_recentFilesPath(hecl::SysFormat(_SYS_STR("%s/recent_files.txt"), fileMgr.getStoreRoot().data()))
{
    Space::SpaceMenuNode::InitializeStrings(*this);
    char path[2048];
    hecl::Sstat theStat;

    FILE* fp = hecl::Fopen(m_recentProjectsPath.c_str(), _SYS_STR("r"), hecl::FileLockType::Read);
    if (fp)
    {
        while (fgets(path, 2048, fp))
        {
            std::string pathStr(path);
            pathStr.pop_back();
            hecl::SystemStringConv pathStrView(pathStr);
            if (!hecl::Stat(pathStrView.c_str(), &theStat) && S_ISDIR(theStat.st_mode))
                m_recentProjects.emplace_back(pathStrView.sys_str());
        }
        fclose(fp);
    }

    fp = hecl::Fopen(m_recentFilesPath.c_str(), _SYS_STR("r"), hecl::FileLockType::Read);
    if (fp)
    {
        while (fgets(path, 2048, fp))
        {
            std::string pathStr(path);
            pathStr.pop_back();
            hecl::SystemStringConv pathStrView(pathStr);
            if (!hecl::Stat(pathStrView.c_str(), &theStat) && S_ISDIR(theStat.st_mode))
                m_recentFiles.emplace_back(pathStrView.sys_str());
        }
        fclose(fp);
    }
}

ViewManager::~ViewManager() {}

void ViewManager::pushRecentProject(hecl::SystemStringView path)
{
    for (hecl::SystemString& testPath : m_recentProjects)
    {
        if (path == testPath)
            return;
    }
    m_recentProjects.emplace_back(path);
    FILE* fp = hecl::Fopen(m_recentProjectsPath.c_str(), _SYS_STR("w"), hecl::FileLockType::Write);
    if (fp)
    {
        for (hecl::SystemString& pPath : m_recentProjects)
            fprintf(fp, "%s\n", hecl::SystemUTF8Conv(pPath).c_str());
        fclose(fp);
    }
}

void ViewManager::pushRecentFile(hecl::SystemStringView path)
{
    for (hecl::SystemString& testPath : m_recentFiles)
    {
        if (path == testPath)
            return;
    }
    m_recentFiles.emplace_back(path);
    FILE* fp = hecl::Fopen(m_recentFilesPath.c_str(), _SYS_STR("w"), hecl::FileLockType::Write);
    if (fp)
    {
        for (hecl::SystemString& pPath : m_recentFiles)
            fprintf(fp, "%s\n", hecl::SystemUTF8Conv(pPath).c_str());
        fclose(fp);
    }
}

void ViewManager::init(boo::IApplication* app)
{
    m_mainWindow = app->newWindow(_SYS_STR("URDE"));
    m_mainWindow->showWindow();
    m_mainWindow->setWaitCursor(true);

    float pixelFactor = m_mainWindow->getVirtualPixelFactor();

    m_mainBooFactory = m_mainWindow->getMainContextDataFactory();
    m_pipelineConv = hecl::NewPipelineConverter(m_mainBooFactory);
    hecl::conv = m_pipelineConv.get();
    m_mainPlatformName = m_mainBooFactory->platformName();
    m_mainWindow->setTitle(_SYS_STR("URDE [") + hecl::SystemString(m_mainPlatformName) + _SYS_STR("]"));
    m_mainCommandQueue = m_mainWindow->getCommandQueue();
    m_viewResources.init(m_mainBooFactory, &m_fontCache, &m_themeData, pixelFactor);
    InitializeIcons(m_viewResources);
    InitializeBadging(m_viewResources);
    m_viewResources.prepFontCacheAsync(m_mainWindow.get());
    specter::RootView* root = SetupRootView();
    m_showSplash = true;
    root->accessContentViews().push_back(SetupSplashView());
    root->updateSize();
    m_renderTex = root->renderTex();
    m_mainWindow->setWaitCursor(false);
    m_voiceEngine = boo::NewAudioVoiceEngine();
    m_voiceEngine->setVolume(0.7f);
    m_amuseAllocWrapper.emplace(*m_voiceEngine);

    for (const auto& arg : app->getArgs())
    {
        if (m_deferedProject.empty() && hecl::SearchForProject(arg))
            m_deferedProject = arg;
        if (arg == _SYS_STR("--no-shader-warmup"))
            m_noShaderWarmup = true;
        else if (arg == _SYS_STR("--no-sound"))
            m_voiceEngine->setVolume(0.f);
    }

    if (m_deferedProject.empty())
    {
        /* Default behavior - search upwards for packaged project containing the program */
        if (hecl::ProjectRootPath root = hecl::SearchForProject(ExeDir))
        {
            hecl::SystemString rootPath(root.getAbsolutePath());
            hecl::Sstat theStat;
            if (!hecl::Stat((rootPath + _SYS_STR("/out/files/Metroid1.upak")).c_str(), &theStat) && S_ISREG(theStat.st_mode))
                m_deferedProject = rootPath + _SYS_STR("/out");
        }
    }
}

bool ViewManager::proc()
{
    if (!m_deferedProject.empty() && m_viewResources.fontCacheReady())
    {
        m_projManager.openProject(m_deferedProject);
        m_deferedProject.clear();
    }

    boo::IGraphicsCommandQueue* gfxQ = m_mainWindow->getCommandQueue();
    if (m_rootView->isDestroyed())
        return false;

    if (m_updatePf)
    {
        m_viewResources.resetPixelFactor(m_reqPf);
        specter::RootView* root = SetupRootView();
        if (m_rootSpace)
            BuildSpaceViews();
        else
        {
            std::vector<specter::View*>& cViews = m_rootView->accessContentViews();
            cViews.push_back(SetupSplashView());
        }
        root->updateSize();
        m_updatePf = false;
    }

    m_rootView->dispatchEvents();
    m_rootView->internalThink();
    if (m_rootSpace)
        m_rootSpace->think();
    if (m_splash)
        m_splash->think();

    if (m_deferSplit)
    {
        SplitSpace* ss = static_cast<SplitSpace*>(m_deferSplit->spaceSplit(m_deferSplitAxis, m_deferSplitThisSlot));
        m_rootView->startSplitDrag(ss->splitView(), m_deferSplitCoord);
        m_deferSplit = nullptr;
    }

    ++m_editorFrames;
    if (m_rootSpaceView && m_editorFrames <= 30)
        m_rootSpaceView->setMultiplyColor(zeus::CColor::lerp({1, 1, 1, 0}, {1, 1, 1, 1}, m_editorFrames / 30.0));

    m_projManager.mainUpdate();

    if (m_testGameView)
        m_testGameView->think();

    if (g_Renderer)
        g_Renderer->BeginScene();
    m_rootView->draw(gfxQ);
    if (g_Renderer)
        g_Renderer->EndScene();
    gfxQ->execute();
    if (g_ResFactory)
        g_ResFactory->AsyncIdle();
    m_voiceEngine->pumpAndMixVoices();
    if (!m_skipWait || !hecl::com_developer->toBoolean())
        m_mainWindow->waitForRetrace();
    CBooModel::ClearModelUniformCounters();
    CGraphics::TickRenderTimings();
    ++logvisor::FrameIndex;
    return true;
}

void ViewManager::stop()
{
    m_videoVoice.reset();
    m_projManager.shutdown();
    CDvdFile::Shutdown();
    DestroyIcons();
    DestroyBadging();
    m_viewResources.destroyResData();
    m_fontCache.destroyAtlases();
    m_mainWindow->getCommandQueue()->stopRenderer();
}

} // namespace urde