#include "global.h"

UI_Menu::UI_Menu()
{
	cancelCode = MENU_CODE_NONE;
	fModifyingItem = false;
	headControl = NULL;
	current = NULL;
}

UI_Menu::~UI_Menu()
{
	std::list<UI_Control*>::iterator iterateAll = controls.begin();
	
	while (iterateAll != controls.end())
	{
		delete (*iterateAll);
		iterateAll++;
	}

	controls.clear();
}


void UI_Menu::AddControl(UI_Control * control, UI_Control * up, UI_Control * down, UI_Control * left, UI_Control * right)
{
	control->SetMenuParent(this);
	controls.push_back(control);
	control->SetNeighbor(MENU_ITEM_NEIGHBOR_UP, up);
	control->SetNeighbor(MENU_ITEM_NEIGHBOR_DOWN, down);
	control->SetNeighbor(MENU_ITEM_NEIGHBOR_LEFT, left);
	control->SetNeighbor(MENU_ITEM_NEIGHBOR_RIGHT, right);
}

void UI_Menu::AddNonControl(UI_Control * control)
{
	control->SetMenuParent(this);
	controls.push_back(control);
}

void UI_Menu::SetHeadControl(UI_Control * control)
{
	headControl = control;
	ResetMenu();
}

void UI_Menu::ResetMenu()
{
	if(current)
	{
		current->Modify(false);
		current->Select(false);
	}

	current = headControl;

	if(current)
		fModifyingItem = current->Select(true);

	eyeCandy.clean();
}

void UI_Menu::Update()
{
	std::list<UI_Control*>::iterator iterateAll = controls.begin();
	
	while (iterateAll != controls.end())
	{
		(*iterateAll)->Update();
		iterateAll++;
	}

	eyeCandy.cleandeadobjects();
	eyeCandy.update();
}

void UI_Menu::Draw()
{
	std::list<UI_Control*>::iterator iterateAll = controls.begin();
	
	while (iterateAll != controls.end())
	{
		(*iterateAll)->Draw();
		iterateAll++;
	}

	eyeCandy.draw();
}

MenuCodeEnum UI_Menu::SendInput(CPlayerInput * playerInput)
{
	if(fModifyingItem)
	{
		MenuCodeEnum ret = current->SendInput(playerInput);
		
		if(MENU_CODE_UNSELECT_ITEM == ret)
		{
			fModifyingItem = false;
			
			if(current->IsAutoModify())
				return cancelCode;

			return MENU_CODE_NONE;
		}
		
		return ret;
	}

	for(short iPlayer = 0; iPlayer < 4; iPlayer++)
	{
		if(iPlayer != 0 && game_values.playerInput.inputControls[iPlayer]->iDevice == DEVICE_KEYBOARD)
			continue;

		if(playerInput->outputControls[iPlayer].menu_up.fPressed)
		{
			MoveNextControl(MENU_ITEM_NEIGHBOR_UP);
		}

		if(playerInput->outputControls[iPlayer].menu_down.fPressed)
		{
			MoveNextControl(MENU_ITEM_NEIGHBOR_DOWN);
		}
		
		if(playerInput->outputControls[iPlayer].menu_left.fPressed)
		{
			MoveNextControl(MENU_ITEM_NEIGHBOR_LEFT);
		}

		if(playerInput->outputControls[iPlayer].menu_right.fPressed)
		{
			MoveNextControl(MENU_ITEM_NEIGHBOR_RIGHT);
		}

		if(playerInput->outputControls[iPlayer].menu_select.fPressed)
		{
			MenuCodeEnum ret = current->Modify(true);
			
			if(MENU_CODE_MODIFY_ACCEPTED == ret)
			{
				fModifyingItem = true;
				return MENU_CODE_NONE;
			}
			
			if(MENU_CODE_UNSELECT_ITEM == ret)
			{
				fModifyingItem = false;
				return MENU_CODE_NONE;
			}
			
			return ret;
		}

		if(playerInput->outputControls[iPlayer].menu_cancel.fPressed)
		{
			return cancelCode;
		}
	}
	
	return MENU_CODE_NONE;
}

void UI_Menu::MoveNextControl(short iDirection)
{
	UI_Control * neighbor = current->GetNeighbor(iDirection);

	while(neighbor && !neighbor->IsVisible())
	{
		neighbor = neighbor->GetNeighbor(iDirection);
	}

	if(neighbor)
	{
		current->Select(false);
		current = neighbor;
		fModifyingItem = current->Select(true);
	}
}

void UI_Menu::RememberCurrent()
{
	savedCurrent = current;
}

void UI_Menu::RestoreCurrent()
{
	if(current)
	{
		current->Modify(false);
		current->Select(false);
	}

	current = savedCurrent;

	if(current)
		fModifyingItem = current->Select(true);

	eyeCandy.clean();
}