/*
===========================================================================

Doom 3 BFG Edition GPL Source Code
Copyright (C) 1993-2012 id Software LLC, a ZeniMax Media company.

This file is part of the Doom 3 BFG Edition GPL Source Code ("Doom 3 BFG Edition Source Code").

Doom 3 BFG Edition Source Code 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.

Doom 3 BFG Edition Source Code 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 Doom 3 BFG Edition Source Code.  If not, see <http://www.gnu.org/licenses/>.

In addition, the Doom 3 BFG Edition Source Code is also subject to certain additional terms. You should have received a copy of these additional terms immediately following the terms and conditions of the GNU General Public License which accompanied the Doom 3 BFG Edition Source Code.  If not, please request a copy in writing from id Software at the address below.

If you have questions concerning this license or the applicable additional terms, you may contact in writing id Software LLC, c/o ZeniMax Media Inc., Suite 120, Rockville, Maryland 20850 USA.

===========================================================================
*/
#pragma hdrstop
#include "precompiled.h"
#include "../Game_local.h"

const static int NUM_SETTING_OPTIONS = 8;
//extern idCVar g_nightmare;
//extern idCVar g_roeNightmare;
//extern idCVar g_leNightmare;
//extern idCVar g_skill;
/*
========================
idMenuScreen_Shell_Difficulty::Initialize
========================
*/
void idMenuScreen_Shell_Difficulty::Initialize( idMenuHandler* data )
{
	common->Printf("Initializing Difficulty\n");
	idMenuScreen::Initialize( data );
	
	if( data != NULL )
	{
		menuGUI = data->GetGUI();
	}
	
	SetSpritePath( "menuDifficulty" );
	
	options = new( TAG_SWF ) idMenuWidget_DynamicList();
	idList< idList< idStr, TAG_IDLIB_LIST_MENU >, TAG_IDLIB_LIST_MENU > menuOptions;
	idList< idStr > option;
	option.Append( "#str_04089" );	// Easy
	menuOptions.Append( option );
	option.Clear();
	option.Append( "#str_04091" );	// Medium
	menuOptions.Append( option );
	option.Clear();
	option.Append( "#str_04093" );	// Hard
	menuOptions.Append( option );
	option.Clear();
	option.Append( "#str_02357" );	// Nightmare
	menuOptions.Append( option );
#if 0
	if (game->GetCVarBool("g_nightmare") && game->GetCVarBool("g_roeNightmare") && game->GetCVarBool("g_leNightmare")) {
		option.Append("Difficulty Placeholder");	// Masochism
		menuOptions.Append(option);
	}
#endif
	
	options->SetListData( menuOptions );
	options->SetNumVisibleOptions( NUM_SETTING_OPTIONS );
	options->SetSpritePath( GetSpritePath(), "info", "options" );
	options->SetWrappingAllowed( true );
	AddChild( options );
	
	idMenuWidget_Help* const helpWidget = new( TAG_SWF ) idMenuWidget_Help();
	helpWidget->SetSpritePath( GetSpritePath(), "info",  "helpTooltip" );
	AddChild( helpWidget );
	
	const char* tips[] = { "#str_02358", "#str_02360", "#str_02362", "#str_02364" };
	
	while( options->GetChildren().Num() < NUM_SETTING_OPTIONS )
	{
		idMenuWidget_Button* const buttonWidget = new( TAG_SWF ) idMenuWidget_Button();
		buttonWidget->AddEventAction( WIDGET_EVENT_PRESS ).Set( WIDGET_ACTION_PRESS_FOCUSED, options->GetChildren().Num() );
		buttonWidget->Initialize( data );
		
		if( options->GetChildren().Num() < menuOptions.Num() )
		{
			buttonWidget->SetDescription( tips[options->GetChildren().Num()] );
		}
		
		buttonWidget->RegisterEventObserver( helpWidget );
		options->AddChild( buttonWidget );
	}
	options->Initialize( data );
	
	btnBack = new( TAG_SWF ) idMenuWidget_Button();
	btnBack->Initialize( data );
	btnBack->SetLabel( "#str_02207" );
	btnBack->SetSpritePath( GetSpritePath(), "info", "btnBack" );
	btnBack->AddEventAction( WIDGET_EVENT_PRESS ).Set( WIDGET_ACTION_GO_BACK );
	
	AddChild( btnBack );
	
	options->AddEventAction( WIDGET_EVENT_SCROLL_DOWN ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_SCROLL_DOWN_START_REPEATER_VARIABLE, WIDGET_EVENT_SCROLL_DOWN ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_UP ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_SCROLL_UP_START_REPEATER_VARIABLE, WIDGET_EVENT_SCROLL_UP ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_DOWN_RELEASE ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_STOP_REPEATER, WIDGET_EVENT_SCROLL_DOWN_RELEASE ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_UP_RELEASE ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_STOP_REPEATER, WIDGET_EVENT_SCROLL_UP_RELEASE ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_DOWN_LSTICK ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_SCROLL_DOWN_START_REPEATER_VARIABLE, WIDGET_EVENT_SCROLL_DOWN_LSTICK ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_UP_LSTICK ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_SCROLL_UP_START_REPEATER_VARIABLE, WIDGET_EVENT_SCROLL_UP_LSTICK ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_DOWN_LSTICK_RELEASE ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_STOP_REPEATER, WIDGET_EVENT_SCROLL_DOWN_LSTICK_RELEASE ) );
	options->AddEventAction( WIDGET_EVENT_SCROLL_UP_LSTICK_RELEASE ).Set( new( TAG_SWF ) idWidgetActionHandler( options, WIDGET_ACTION_EVENT_STOP_REPEATER, WIDGET_EVENT_SCROLL_UP_LSTICK_RELEASE ) );
}

/*
========================
idMenuScreen_Shell_Difficulty::Update
========================
*/
void idMenuScreen_Shell_Difficulty::Update()
{

	if( menuData != NULL )
	{
		idMenuWidget_CommandBar* cmdBar = menuData->GetCmdBar();
		if( cmdBar != NULL )
		{
			cmdBar->ClearAllButtons();
			idMenuWidget_CommandBar::buttonInfo_t* buttonInfo;
			buttonInfo = cmdBar->GetButton( idMenuWidget_CommandBar::BUTTON_JOY2 );
			if((!common->IsNewDOOM3() && menuData->GetPlatform() != 2) || (common->IsNewDOOM3() && menuData->GetPlatform() != 5))
			{
				buttonInfo->label = "#str_00395";
			}
			buttonInfo->action.Set( WIDGET_ACTION_GO_BACK );
			
			buttonInfo = cmdBar->GetButton( idMenuWidget_CommandBar::BUTTON_JOY1 );
			if((!common->IsNewDOOM3() && menuData->GetPlatform() != 2) || (common->IsNewDOOM3() && menuData->GetPlatform() != 5))
			{
				buttonInfo->label = "#str_SWF_SELECT";
			}
			buttonInfo->action.Set( WIDGET_ACTION_PRESS_FOCUSED );
		}
	}
	
	idSWFScriptObject& root = GetSWFObject()->GetRootObject();
	if( BindSprite( root ) )
	{
		idSWFTextInstance* heading = GetSprite()->GetScriptObject()->GetNestedText( "info", "txtHeading" );
		if( heading != NULL )
		{
			heading->SetText( "#str_04088" );
			heading->SetStrokeInfo( true, 0.75f, 1.75f );
		}
		
		idSWFSpriteInstance* gradient = GetSprite()->GetScriptObject()->GetNestedSprite( "info", "gradient" );
		if( gradient != NULL && heading != NULL )
		{
			gradient->SetXPos( heading->GetTextLength() );
		}
	}
	
	if( btnBack != NULL )
	{
		btnBack->BindSprite( root );
	}
	
	idMenuScreen::Update();

	options->GetChildByIndex(3).SetState(nightmareUnlocked ? WIDGET_STATE_NORMAL : WIDGET_STATE_DISABLED);
}

/*
========================
idMenuScreen_Shell_Difficulty::ShowScreen
========================
*/
void idMenuScreen_Shell_Difficulty::ShowScreen( const mainMenuTransition_t transitionType )
{
	idMenuScreen::ShowScreen( transitionType );
	
	nightmareUnlocked = false;
	
	idMenuHandler_Shell* shell = dynamic_cast< idMenuHandler_ShellLocal* >( menuData );
	int type = 0;
	if( shell != NULL )
	{
		type = shell->GetNewGameType();
	}
	
	if( type == 0 )
	{
		nightmareUnlocked = game->GetCVarBool("g_nightmare");
	}
	else if( type == 1 )
	{
		nightmareUnlocked = game->GetCVarBool("g_roeNightmare");
	}
	else if( type == 2 )
	{
		nightmareUnlocked = game->GetCVarBool("g_leNightmare");
	}
	//GK: Always have available the Nightmare difficulty on custom expansions
	else if (type == 3)
	{
		nightmareUnlocked = true;
	}
	//GK: End

	int skill = Max( 0, game->GetCVarInteger("g_skill") );
	if( !nightmareUnlocked )
	{
		options->GetChildByIndex( 3 ).SetState( WIDGET_STATE_DISABLED );
		skill = Min( skill, 2 );
	}
	else
	{
		options->GetChildByIndex( 3 ).SetState( WIDGET_STATE_NORMAL );
		skill = Min( skill, 3 );
	}
	options->SetFocusIndex( skill );
	options->SetViewIndex( options->GetViewOffset() + skill );
}

/*
========================
idMenuScreen_Shell_Difficulty::HideScreen
========================
*/
void idMenuScreen_Shell_Difficulty::HideScreen( const mainMenuTransition_t transitionType )
{
	idMenuScreen::HideScreen( transitionType );
}

/*
========================
idMenuScreen_Shell_Difficulty::HandleAction h
========================
*/
bool idMenuScreen_Shell_Difficulty::HandleAction( idWidgetAction& action, const idWidgetEvent& event, idMenuWidget* widget, bool forceHandled )
{

	if( menuData == NULL )
	{
		return true;
	}
	
	if( menuData->ActiveScreen() != SHELL_AREA_DIFFICULTY )
	{
		return false;
	}

	
	
	widgetAction_t actionType = action.GetType();
	const idSWFParmList& parms = action.GetParms();
	
	switch( actionType )
	{
		case WIDGET_ACTION_SCROLL_VERTICAL_VARIABLE:
		{
		
			if( parms.Num() == 0 )
			{
				return true;
			}
			
			if( options == NULL )
			{
				return true;
			}
			
			int dir = parms[ 0 ].ToInteger();
			int scroll = dir;
			
			bool nightmareEnabled = nightmareUnlocked;
			int curIndex = options->GetFocusIndex();
			
			if( ( curIndex + scroll < 0 && !nightmareEnabled ) || ( curIndex + scroll == 3 && !nightmareEnabled ) )
			{
				scroll *= 2;
				options->Scroll( scroll, true );
			}
			else
			{
				options->Scroll( scroll );
			}
			
			return true;
		}
		case WIDGET_ACTION_GO_BACK:
		{
			menuData->SetNextScreen( SHELL_AREA_NEW_GAME, MENU_TRANSITION_SIMPLE );
			return true;
		}
		case WIDGET_ACTION_PRESS_FOCUSED:
		{
			if( options == NULL )
			{
				return true;
			}
			
			int selectionIndex = options->GetViewIndex();
			if( parms.Num() == 1 )
			{
				selectionIndex = parms[0].ToInteger();
			}
			
			if( selectionIndex == 3 && !nightmareUnlocked )
			{
				return true;
			}
			
			if( options->GetFocusIndex() != selectionIndex )
			{
				options->SetFocusIndex( selectionIndex );
				options->SetViewIndex( options->GetViewOffset() + selectionIndex );
			}
			
			idMenuHandler_Shell* shell = dynamic_cast< idMenuHandler_ShellLocal* >( menuData );
			int type = 0;
			if( shell != NULL )
			{
				type = shell->GetNewGameType();
			}
			
			game->SetCVarInteger("g_skill", selectionIndex );
			
			idMenuHandler_Shell* shellMgr = dynamic_cast< idMenuHandler_ShellLocal* >( menuData );
			if( shellMgr )
			{
				if( type == 0 )
				{
					shellMgr->ShowDoomIntro();
				}
				else if( type == 1 )
				{
					shellMgr->ShowROEIntro();
				}
				else if( type == 2 )
				{
					shellMgr->ShowLEIntro();
				}
				//GK: Just start a new game without intro
				else if (type == 3)
				{
					shellMgr->StartGame(3);
				}
				//GK: End
			}
			
			return true;
		}
	}
	this->Update();
	return idMenuWidget::HandleAction( action, event, widget, forceHandled );
}