#include "gamepadui_interface.h"
#include "gamepadui_basepanel.h"
#include "gamepadui_mainmenu.h"

#include "vgui/ILocalize.h"

// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"

static CDllDemandLoader s_GameUI( "GameUI" );

EXPOSE_SINGLE_INTERFACE_GLOBALVAR( GamepadUI, IGamepadUI, GAMEPADUI_INTERFACE_VERSION, GamepadUI::GetInstance() );

GamepadUI *GamepadUI::s_pGamepadUI = NULL;

GamepadUI& GamepadUI::GetInstance()
{
    if ( !s_pGamepadUI )
        s_pGamepadUI = new GamepadUI;

    return *s_pGamepadUI;
}

void GamepadUI::Initialize( CreateInterfaceFn factory )
{
    ConnectTier1Libraries( &factory, 1 );
    ConnectTier2Libraries( &factory, 1 );
    ConVar_Register( FCVAR_CLIENTDLL );
    ConnectTier3Libraries( &factory, 1 );

    m_pEngineClient           = (IVEngineClient*)         factory( VENGINE_CLIENT_INTERFACE_VERSION, NULL );
    m_pEngineSound            = (IEngineSound*)           factory( IENGINESOUND_CLIENT_INTERFACE_VERSION, NULL );
    m_pEngineVGui             = (IEngineVGui*)            factory( VENGINE_VGUI_VERSION, NULL );
    m_pGameUIFuncs            = (IGameUIFuncs*)           factory( VENGINE_GAMEUIFUNCS_VERSION, NULL );
    m_pMaterialSystem         = (IMaterialSystem*)        factory( MATERIAL_SYSTEM_INTERFACE_VERSION, NULL );
    m_pMaterialSystemSurface  = (IMatSystemSurface*)      factory( MAT_SYSTEM_SURFACE_INTERFACE_VERSION, NULL );
    m_pRenderView             = (IVRenderView*)           factory( VENGINE_RENDERVIEW_INTERFACE_VERSION, NULL );
    m_pSoundEmitterSystemBase = (ISoundEmitterSystemBase*)factory( SOUNDEMITTERSYSTEM_INTERFACE_VERSION, NULL );

    CreateInterfaceFn gameuiFactory = s_GameUI.GetFactory();
    if ( gameuiFactory )
        m_pGameUI = (IGameUI*) gameuiFactory( GAMEUI_INTERFACE_VERSION, NULL );

    m_pAchievementMgr = (IAchievementMgr*) m_pEngineClient->GetAchievementMgr();

    bool bFailed = !m_pEngineClient           ||
                   !m_pEngineSound            ||
                   !m_pEngineVGui             ||
                   !m_pGameUIFuncs            ||
                   !m_pMaterialSystem         ||
                   !m_pMaterialSystemSurface  ||
                   !m_pRenderView             ||
                   !m_pSoundEmitterSystemBase ||
                   !m_pGameUI                 ||
                   !m_pAchievementMgr;
    if ( bFailed )
    {
        GamepadUI_Log( "GamepadUI::Initialize() failed to get necessary interfaces.\n" );
        return;
    }

    g_pVGuiLocalize->AddFile( "resource/gameui_%language%.txt", "GAME", true );
    g_pVGuiLocalize->AddFile( "resource/deck_%language%.txt", "GAME", true );

#if defined(HL2_RETAIL) || defined(STEAM_INPUT)
    SteamAPI_InitSafe();
    SteamAPI_SetTryCatchCallbacks( false );
    m_SteamAPIContext.Init();
#endif // HL2_RETAIL

    m_pBasePanel = new GamepadUIBasePanel( GetRootVPanel() );
    if ( !m_pBasePanel )
    {
        GamepadUI_Log( "Failed to create BasePanel.\n" );
        return;
    }

    GamepadUI_Log( "Overriding menu.\n" );

    m_pGameUI->SetMainMenuOverride( GetBaseVPanel() );

    m_pAnimationController = new vgui::AnimationController( m_pBasePanel );
    m_pAnimationController->SetProportional( false );

    //GetMainMenu()->Activate();
}

void GamepadUI::Shutdown()
{
    if ( m_pGameUI )
        m_pGameUI->SetMainMenuOverride( NULL );

    if ( m_pBasePanel )
        m_pBasePanel->DeletePanel();

#if defined(HL2_RETAIL) || defined(STEAM_INPUT)
    m_SteamAPIContext.Clear();
#endif

    ConVar_Unregister();
    DisconnectTier3Libraries();
    DisconnectTier2Libraries();
    DisconnectTier1Libraries();
}


void GamepadUI::OnUpdate( float flFrametime )
{
    if ( m_pAnimationController )
        m_pAnimationController->UpdateAnimations( GetTime() );

    static bool tempNumDraw = true;

    if (ConVarRef("cl_gamepadui_mainmenu_draw").GetBool() && tempNumDraw)
    {
        GetMainMenu()->Activate();
        tempNumDraw=false;
    }
}

void GamepadUI::OnLevelInitializePreEntity()
{
}

void GamepadUI::OnLevelInitializePostEntity()
{
    m_pBasePanel->OnMenuStateChanged();
    GetMainMenu()->OnMenuStateChanged();
}

void GamepadUI::OnLevelShutdown()
{
    if ( m_pAnimationController )
    {
        m_pAnimationController->UpdateAnimations( GetTime() );
        m_pAnimationController->RunAllAnimationsToCompletion();
    }

    m_pBasePanel->OnMenuStateChanged();
    GetMainMenu()->OnMenuStateChanged();
}

void GamepadUI::VidInit()
{
    int w, h;
    vgui::surface()->GetScreenSize( w, h );

    Assert( w != 0 && h != 0 );

    // Scale elements proportional to the aspect ratio's distance from 16:10
    const float flDefaultInvAspect = 0.625f;
    float flInvAspectRatio = ((float)h) / ((float)w);

    if (flInvAspectRatio != flDefaultInvAspect)
    {
        m_flScreenXRatio = 1.0f - (flInvAspectRatio - flDefaultInvAspect);
    }
    else
    {
        m_flScreenXRatio = 1.0f;
    }

    m_flScreenYRatio = 1.0f;

    m_pBasePanel->InvalidateLayout( false, true );
}


bool GamepadUI::IsInLevel() const
{
	const char *pLevelName = m_pEngineClient->GetLevelName();
    return pLevelName && *pLevelName && !m_pEngineClient->IsLevelMainMenuBackground();
}

bool GamepadUI::IsInBackgroundLevel() const
{
	const char *pLevelName = m_pEngineClient->GetLevelName();
    return pLevelName && *pLevelName && m_pEngineClient->IsLevelMainMenuBackground();
}

bool GamepadUI::IsInMultiplayer() const
{
    return IsInLevel() && m_pEngineClient->GetMaxClients() > 1;
}

bool GamepadUI::IsGamepadUIVisible() const
{
    return !IsInLevel() || IsInBackgroundLevel();
}


void GamepadUI::ResetToMainMenuGradients()
{
    GetMainMenu()->UpdateGradients();
}

vgui::VPANEL GamepadUI::GetRootVPanel() const
{
    return m_pEngineVGui->GetPanel( PANEL_GAMEUIDLL );
}

vgui::Panel *GamepadUI::GetBasePanel() const
{
    return m_pBasePanel;
}

vgui::VPANEL GamepadUI::GetBaseVPanel() const
{
    return m_pBasePanel ? m_pBasePanel->GetVPanel() : 0;
}

vgui::Panel *GamepadUI::GetSizingPanel() const
{
    return m_pBasePanel ? m_pBasePanel->GetSizingPanel() : NULL;
}

vgui::VPANEL GamepadUI::GetSizingVPanel() const
{
    return GetSizingPanel() ? GetSizingPanel()->GetVPanel() : 0;
}

vgui::Panel *GamepadUI::GetMainMenuPanel() const
{
    return m_pBasePanel ? m_pBasePanel->GetMainMenuPanel() : NULL;
}

vgui::VPANEL GamepadUI::GetMainMenuVPanel() const
{
    return GetMainMenuPanel() ? GetMainMenuPanel()->GetVPanel() : 0;
}

GamepadUIMainMenu* GamepadUI::GetMainMenu() const
{
    return static_cast<GamepadUIMainMenu*>( GetMainMenuPanel() );
}

void GamepadUI::GetSizingPanelScale( float &flX, float &flY ) const
{
    vgui::Panel *pPanel = GetSizingPanel();
    if (!pPanel)
        return;
    static_cast<GamepadUISizingPanel*>(pPanel)->GetScale( flX, flY );
}

void GamepadUI::GetSizingPanelOffset( int &nX, int &nY ) const
{
    vgui::Panel *pPanel = GetSizingPanel();
    if (!pPanel)
        return;
    pPanel->GetPos( nX, nY );
}

#ifdef MAPBASE
void GamepadUI::BonusMapChallengeNames( char *pchFileName, char *pchMapName, char *pchChallengeName )
{
    Q_strncpy( pchFileName, m_szChallengeFileName, sizeof( m_szChallengeFileName ) );
    Q_strncpy( pchMapName, m_szChallengeMapName, sizeof( m_szChallengeMapName ) );
    Q_strncpy( pchChallengeName, m_szChallengeName, sizeof( m_szChallengeName ) );
}

void GamepadUI::BonusMapChallengeObjectives( int &iBronze, int &iSilver, int &iGold )
{
    iBronze = m_iBronze; iSilver = m_iSilver; iGold = m_iGold;
}

void GamepadUI::SetCurrentChallengeObjectives( int iBronze, int iSilver, int iGold )
{
    m_iBronze = iBronze; m_iSilver = iSilver; m_iGold = iGold;
}

void GamepadUI::SetCurrentChallengeNames( const char *pszFileName, const char *pszMapName, const char *pszChallengeName )
{
    Q_strncpy( m_szChallengeFileName, pszFileName, sizeof( m_szChallengeFileName ) );
    Q_strncpy( m_szChallengeMapName, pszMapName, sizeof( m_szChallengeMapName ) );
    Q_strncpy( m_szChallengeName, pszChallengeName, sizeof( m_szChallengeName ) );
}
#endif