/*
	GWEN
	Copyright (c) 2010 Facepunch Studios
	See license in Gwen.h
*/


#include "Gwen/Controls/TreeControl.h"
#include "Gwen/Controls/ScrollControl.h"
#include "Gwen/Utility.h"

using namespace Gwen;
using namespace Gwen::Controls;

GWEN_CONTROL_CONSTRUCTOR( TreeControl )
{
	m_TreeControl = this;

	m_ToggleButton->DelayedDelete();
	m_ToggleButton = NULL;
	m_Title->DelayedDelete();
	m_Title = NULL;
	m_InnerPanel->DelayedDelete();
	m_InnerPanel = NULL;

	m_bAllowMultipleSelection = false;

	m_ScrollControl = new ScrollControl( this );
	m_ScrollControl->Dock( Pos::Fill );
	m_ScrollControl->SetScroll( false, true );
	m_ScrollControl->SetAutoHideBars( true );
	m_ScrollControl->SetMargin( Margin( 1, 1, 1, 1 ) );

	m_InnerPanel = m_ScrollControl;

	m_ScrollControl->SetInnerSize( 1000, 1000 );
}

void TreeControl::Render( Skin::Base* skin )
{
	if ( ShouldDrawBackground() )
		skin->DrawTreeControl( this );
}

void TreeControl::OnChildBoundsChanged( Gwen::Rect /*oldChildBounds*/, Base* /*pChild*/ )
{
	m_ScrollControl->UpdateScrollBars();
}

void TreeControl::Clear()
{
	m_ScrollControl->Clear();
}

void TreeControl::Layout( Skin::Base* skin )
{
	BaseClass::BaseClass::Layout( skin );
}

void TreeControl::PostLayout( Skin::Base* skin )
{
	BaseClass::BaseClass::PostLayout( skin );
}

void TreeControl::OnNodeAdded( TreeNode* pNode )
{
	pNode->onNamePress.Add( this, &TreeControl::OnNodeSelection );
}

void TreeControl::OnNodeSelection( Controls::Base* /*control*/ )
{
	if ( !m_bAllowMultipleSelection || !Gwen::Input::IsKeyDown( Key::Control ) )
		DeselectAll();
}

				
void TreeControl::iterate(int action, int* curIndex, int* targetIndex)
{

	Base::List& children = m_InnerPanel->GetChildren();
	for ( Base::List::iterator iter = children.begin(); iter != children.end(); ++iter )
	{
		TreeNode* pChild = (*iter)->DynamicCastTreeNode();
		if ( !pChild ) 
			continue;
		pChild->iterate(action ,curIndex, targetIndex);
	}
	
}


bool TreeControl::OnKeyUp( bool bDown )
{
	if (bDown)
	{
		int maxIndex = 0;
		int newIndex = 0;
		int curIndex=0;
		int targetIndex=-1;
		iterate(ITERATE_ACTION_FIND_SELECTED_INDEX,&curIndex,&targetIndex);
		maxIndex = curIndex;
		if (targetIndex>0)
		{
			curIndex=0;
			int deselectIndex = targetIndex;
			targetIndex--;
			newIndex = targetIndex;
			iterate(ITERATE_ACTION_SELECT,&curIndex,&targetIndex);
			if (targetIndex<0)
			{
				curIndex=0;
				iterate(ITERATE_ACTION_DESELECT_INDEX,&curIndex,&deselectIndex);
			}
			float amount = float(newIndex)/float(maxIndex);
			m_ScrollControl->m_VerticalScrollBar->SetScrolledAmount(amount,true);
		}
	}
	return true;
}


bool TreeControl::OnKeyDown( bool bDown )
{
	if (bDown)
	{
		int maxIndex = 0;
		int newIndex = 0;
		int curIndex=0;
		int targetIndex=-1;
		iterate(ITERATE_ACTION_FIND_SELECTED_INDEX,&curIndex,&targetIndex);
		maxIndex = curIndex;
		if (targetIndex>=0)
		{
			curIndex=0;
			int deselectIndex = targetIndex;
			targetIndex++;
			newIndex = targetIndex;
			iterate(ITERATE_ACTION_SELECT,&curIndex,&targetIndex);
			if (targetIndex<0)
			{
				curIndex=0;
				iterate(ITERATE_ACTION_DESELECT_INDEX,&curIndex,&deselectIndex);
			}
			float amount = float(newIndex)/float(maxIndex);
			m_ScrollControl->m_VerticalScrollBar->SetScrolledAmount(amount,true);
		}
	}
	return true;
}


bool TreeControl::OnKeyRight( bool bDown )
{
	if (bDown)
	{
		iterate(ITERATE_ACTION_OPEN,0,0);
		int curIndex=0;
		int targetIndex=0;
		iterate(ITERATE_ACTION_FIND_SELECTED_INDEX,&curIndex,&targetIndex);
		float amount = float(targetIndex)/float(curIndex);
		m_ScrollControl->m_VerticalScrollBar->SetScrolledAmount(amount,true);
	}
	return true;
}
bool TreeControl::OnKeyLeft( bool bDown )
{
	if (bDown)
	{
		iterate(ITERATE_ACTION_CLOSE,0,0);

		int curIndex=0;
		int targetIndex=0;
		iterate(ITERATE_ACTION_FIND_SELECTED_INDEX,&curIndex,&targetIndex);
		float amount = float(targetIndex)/float(curIndex);
		m_ScrollControl->m_VerticalScrollBar->SetScrolledAmount(amount,true);

	}
	return true;
}