/****************************************************************************
 * Copyright (C) 2015 Dimok
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ****************************************************************************/
#include <coreinit/core.h>
#include <coreinit/foreground.h>
#include <coreinit/title.h>
#include <proc_ui/procui.h>
#include <sysapp/launch.h>
#include "Application.h"
#include "common/common.h"
#include <gui/FreeTypeGX.h>
#include <gui/VPadController.h>
#include <gui/WPadController.h>
#include "resources/Resources.h"
#include <gui/sounds/SoundHandler.hpp>
#include <gui/memory.h>
#include "utils/logger.h"
#include "utils/AsyncExecutor.h"
#include <thread>

#define HBL_TITLE_ID (0x0005000013374842)
#define MII_MAKER_JPN_TITLE_ID (0x000500101004A000)
#define MII_MAKER_USA_TITLE_ID (0x000500101004A100)
#define MII_MAKER_EUR_TITLE_ID (0x000500101004A200)

Application *Application::applicationInstance = NULL;
bool Application::exitApplication = false;
bool Application::quitRequest = false;

Application::Application()
    : CThread(CThread::eAttributeAffCore1 | CThread::eAttributePinnedAff, 0, 0x20000)
    , bgMusic(NULL)
    , video(NULL)
    , mainWindow(NULL)
    , fontSystem(NULL)
    , exitCode(0) {
    controller[0] = new VPadController(GuiTrigger::CHANNEL_1);
    controller[1] = new WPadController(GuiTrigger::CHANNEL_2);
    controller[2] = new WPadController(GuiTrigger::CHANNEL_3);
    controller[3] = new WPadController(GuiTrigger::CHANNEL_4);
    controller[4] = new WPadController(GuiTrigger::CHANNEL_5);

    //! create bgMusic
    bgMusic = new GuiSound(Resources::GetFile("bgMusic.ogg"), Resources::GetFileSize("bgMusic.ogg"));
    bgMusic->SetLoop(true);
    bgMusic->Play();
    bgMusic->SetVolume(50);

    exitApplication = false;

    uint64_t titleID = OSGetTitleID();

    // Homebrew Launcher does not like the standard ProcUI application loop,
    // so instead we disable the home buttom menu and use the home button
    // to trigger an exit.
    if (titleID == HBL_TITLE_ID ||
            titleID == MII_MAKER_JPN_TITLE_ID ||
            titleID == MII_MAKER_USA_TITLE_ID ||
            titleID == MII_MAKER_EUR_TITLE_ID) {
        // Important: OSEnableHomeButtonMenu must come before ProcUIInitEx.
        OSEnableHomeButtonMenu(FALSE);
        sFromHBL = TRUE;
    }

    ProcUIInit(OSSavesDone_ReadyToRelease);
}

Application::~Application() {
    DEBUG_FUNCTION_LINE("Destroy music\n");
    delete bgMusic;

    DEBUG_FUNCTION_LINE("Destroy controller\n");

    for(int32_t i = 0; i < 5; i++)
        delete controller[i];

    DEBUG_FUNCTION_LINE("Clear resources\n");
    Resources::Clear();

    DEBUG_FUNCTION_LINE("Stop sound handler\n");
    SoundHandler::DestroyInstance();

    DEBUG_FUNCTION_LINE("Clear AsyncExecutor\n");
    AsyncExecutor::destroyInstance();

    ProcUIShutdown();
}

int32_t Application::exec() {
    //! start main GX2 thread
    resumeThread();
    //! now wait for thread to finish
    shutdownThread();

    return exitCode;
}

void Application::quit(int32_t code) {
    exitCode = code;
    exitApplication = true;
    quitRequest = true;
}

void Application::fadeOut() {
    GuiImage fadeOut(video->getTvWidth(), video->getTvHeight(), (GX2Color) {
        0, 0, 0, 255
    });

    for(int32_t i = 0; i < 255; i += 10) {
        if(i > 255)
            i = 255;

        fadeOut.setAlpha(i / 255.0f);

        //! start rendering DRC
        video->prepareDrcRendering();
        mainWindow->drawDrc(video);

        GX2SetDepthOnlyControl(GX2_DISABLE, GX2_DISABLE, GX2_COMPARE_FUNC_ALWAYS);
        fadeOut.draw(video);
        GX2SetDepthOnlyControl(GX2_ENABLE, GX2_ENABLE, GX2_COMPARE_FUNC_LEQUAL);

        video->drcDrawDone();

        //! start rendering TV
        video->prepareTvRendering();

        mainWindow->drawTv(video);

        GX2SetDepthOnlyControl(GX2_DISABLE, GX2_DISABLE, GX2_COMPARE_FUNC_ALWAYS);
        fadeOut.draw(video);
        GX2SetDepthOnlyControl(GX2_ENABLE, GX2_ENABLE, GX2_COMPARE_FUNC_LEQUAL);

        video->tvDrawDone();

        //! as last point update the effects as it can drop elements
        mainWindow->updateEffects();

        video->waitForVSync();
    }

    //! one last cleared black screen
    video->prepareDrcRendering();
    video->drcDrawDone();
    video->prepareTvRendering();
    video->tvDrawDone();
    video->waitForVSync();
    video->tvEnable(false);
    video->drcEnable(false);
}

bool Application::procUI(void) {
    bool executeProcess = false;

    switch(ProcUIProcessMessages(true)) {
    case PROCUI_STATUS_EXITING: {
        DEBUG_FUNCTION_LINE("PROCUI_STATUS_EXITING\n");
        exitCode = EXIT_SUCCESS;
        exitApplication = true;
        break;
    }
    case PROCUI_STATUS_RELEASE_FOREGROUND: {
        DEBUG_FUNCTION_LINE("PROCUI_STATUS_RELEASE_FOREGROUND\n");
        if(video != NULL) {
            // we can turn of the screen but we don't need to and it will display the last image
            video->tvEnable(true);
            video->drcEnable(true);

            DEBUG_FUNCTION_LINE("delete fontSystem\n");
            delete fontSystem;
            fontSystem = NULL;

            DEBUG_FUNCTION_LINE("delete video\n");
            delete video;
            video = NULL;

            DEBUG_FUNCTION_LINE("deinitialze memory\n");
            libgui_memoryRelease();
            ProcUIDrawDoneRelease();
        } else {
            ProcUIDrawDoneRelease();
        }
        break;
    }
    case PROCUI_STATUS_IN_FOREGROUND: {
        if(!quitRequest) {
            if(video == NULL) {
                DEBUG_FUNCTION_LINE("PROCUI_STATUS_IN_FOREGROUND\n");
                DEBUG_FUNCTION_LINE("initialze memory\n");
                libgui_memoryInitialize();

                DEBUG_FUNCTION_LINE("Initialize video\n");
                video = new CVideo(GX2_TV_SCAN_MODE_720P, GX2_DRC_RENDER_MODE_SINGLE);
                DEBUG_FUNCTION_LINE("Video size %i x %i\n", video->getTvWidth(), video->getTvHeight());

                //! setup default Font
                DEBUG_FUNCTION_LINE("Initialize main font system\n");
                FreeTypeGX *fontSystem = new FreeTypeGX(Resources::GetFile("font.ttf"), Resources::GetFileSize("font.ttf"), true);
                GuiText::setPresetFont(fontSystem);

                if(mainWindow == NULL) {
                    DEBUG_FUNCTION_LINE("Initialize main window\n");
                    mainWindow = new MainWindow(video->getTvWidth(), video->getTvHeight());
                }

            }
            executeProcess = true;
        }
        break;
    }
    case PROCUI_STATUS_IN_BACKGROUND:
    default:
        break;
    }

    return executeProcess;
}

void Application::executeThread(void) {
    DEBUG_FUNCTION_LINE("Entering main loop\n");

    //! main GX2 loop (60 Hz cycle with max priority on core 1)
    while(!exitApplication) {
        if(procUI() == false) {
            continue;
        }

        mainWindow->lockGUI();
        mainWindow->process();

        //! Read out inputs
        for(int32_t i = 0; i < 5; i++) {
            if(controller[i]->update(video->getTvWidth(), video->getTvHeight()) == false)
                continue;

            //! update controller states
            mainWindow->update(controller[i]);

            if(controller[i]->data.buttons_d & VPAD_BUTTON_HOME) {
                if (sFromHBL) {
                    SYSRelaunchTitle(0, NULL);
                }
            }
        }

        //! start rendering DRC
        video->prepareDrcRendering();
        mainWindow->drawDrc(video);
        video->drcDrawDone();

        //! start rendering TV
        video->prepareTvRendering();
        mainWindow->drawTv(video);
        video->tvDrawDone();

        //! enable screen after first frame render
        if(video->getFrameCount() == 0) {
            video->tvEnable(true);
            video->drcEnable(true);
        }

        //! as last point update the effects as it can drop elements
        mainWindow->updateEffects();
        mainWindow->unlockGUI();

        video->waitForVSync();
    }

    //! in case we exit to a homebrew let's smoothly fade out
    if(video) {
        fadeOut();
    }

    DEBUG_FUNCTION_LINE("delete mainWindow\n");
    delete mainWindow;
    mainWindow = NULL;

    DEBUG_FUNCTION_LINE("delete fontSystem\n");
    delete fontSystem;
    fontSystem = NULL;

    DEBUG_FUNCTION_LINE("delete video\n");
    delete video;
    video = NULL;

    DEBUG_FUNCTION_LINE("deinitialize memory\n");
    libgui_memoryRelease();
}