#include "StdAfx.h"
#include "AnimationGraphStateEditor.h"
#include "AnimationGraphDialog.h"
#include "resource.h"
#include "StringDlg.h"
#include "Controls/PropertyItem.h"

namespace
{

	int ShowQuickPopup( const std::vector<CString>& elems, CWnd * pWnd )
	{
		CMenu menu;
		menu.CreatePopupMenu();
		for (std::vector<CString>::const_iterator iter = elems.begin(); iter != elems.end(); ++iter)
			menu.AppendMenu( MF_STRING, iter - elems.begin() + 1, *iter );

		CPoint p;
		::GetCursorPos(&p);
		return menu.TrackPopupMenuEx( TPM_RETURNCMD|TPM_LEFTBUTTON|TPM_TOPALIGN|TPM_LEFTALIGN, p.x, p.y, pWnd, NULL ) - 1;
	}

}

/*
 * CPropGroup
 */

CAnimationGraphStateEditor::CPropGroup::CPropGroup( CAnimationGraphStateEditor * pParent, const CString& name, UINT id, const TAGPropMap& props, const Verb * pVerbs )
{
	CommonInit( pParent, name, id, pVerbs );

	// create the property editor
	CRect rc(0,0,220,500);
	m_pProps = new CPropertyCtrl();
	m_pProps->Create( WS_CHILD|WS_VISIBLE|WS_CLIPCHILDREN|WS_CLIPSIBLINGS, rc, &m_pParent->m_panel );
	m_pProps->ModifyStyle( 0, WS_CLIPCHILDREN|WS_CLIPSIBLINGS );
	m_pProps->SetItemHeight(16);
	m_pProps->ExpandAll();

	m_pItem = m_pGroup->AddControlItem( m_pProps->GetSafeHwnd() );
	m_pItem->SetCaption("Properties");

	Reset(props);
}

CAnimationGraphStateEditor::CPropGroup::CPropGroup( CAnimationGraphStateEditor * pParent, const CString& name, UINT id, const std::vector<CString>& props, const Verb * pVerbs )
{
	CommonInit( pParent, name, id, pVerbs );

	CRect rc(0,0,220,500);
	m_pList = new CListBox();
	m_pList->Create( WS_CHILD|WS_VISIBLE|WS_CLIPCHILDREN|WS_CLIPSIBLINGS|WS_VSCROLL|WS_HSCROLL, rc, &m_pParent->m_panel, 400 );
	m_pList->SetFont( CFont::FromHandle((HFONT)gSettings.gui.hSystemFont) );

	m_pItem = m_pGroup->AddControlItem( m_pList->GetSafeHwnd() );
	m_pItem->SetCaption("Properties");
	
	Reset(props);
}

void CAnimationGraphStateEditor::CPropGroup::CommonInit( CAnimationGraphStateEditor * pParent, const CString& name, UINT id, const Verb * pVerbs )
{
	m_pParent = pParent;
	m_id = id;

	m_pItem = 0;
	m_pGroup = 0;
	m_pProps = 0;
	m_pList = 0;

	// create a group for us
	m_pGroup = m_pParent->m_panel.AddGroup(id);
	m_pGroup->SetCaption(name);
	m_pGroup->SetItemLayout(xtpTaskItemLayoutDefault);

	if (pVerbs)
	{
		for (int i=0; !pVerbs[i].first.IsEmpty(); i++)
		{
			CXTPTaskPanelGroupItem * pGI = m_pGroup->AddLinkItem( 100*id + i );
			pGI->SetType( xtpTaskItemTypeLink );
			pGI->SetCaption( pVerbs[i].first );
			m_callbacks.push_back( pVerbs[i].second );
		}
	}
}

void CAnimationGraphStateEditor::CPropGroup::Reset( const TAGPropMap& props )
{
	UpdateLock updateLock(m_pParent);

	m_pProps->RemoveAllItems();
	for (TAGPropMap::const_iterator iter = props.begin(); iter != props.end(); ++iter)
	{
		if (iter->first == "")
			m_pProps->AddVarBlock( iter->second );
		else
			m_pProps->AddVarBlock( iter->second, iter->first );
	}
	m_pProps->ExpandAll();

	// get things positioned correctly
	CRect rc;
	m_pParent->m_panel.ExpandGroup( m_pGroup, FALSE );
	m_pParent->m_panel.GetClientRect(rc);
	int h = m_pProps->GetVisibleHeight();
	m_pProps->SetWindowPos( NULL, 0, 0, rc.right, h+1, SWP_NOMOVE );
	m_pItem->SetControlHandle( m_pProps->GetSafeHwnd() );
	m_pParent->m_panel.Reposition(FALSE);
	m_pParent->m_panel.ExpandGroup( m_pGroup, TRUE );
}

void CAnimationGraphStateEditor::CPropGroup::Reset( const std::vector<CString>& props )
{
	UpdateLock updateLock(m_pParent);

	m_pList->ResetContent();
	for (int i=0; i<props.size(); i++)
	{
		m_pList->AddString( props[i] );
	}

	// get things positioned correctly
	CRect rc;
	m_pParent->m_panel.ExpandGroup( m_pGroup, FALSE );
	m_pParent->m_panel.GetClientRect(rc);
	m_pList->SetWindowPos( NULL, 0, 0, rc.right, 500, SWP_NOMOVE );
	m_pItem->SetControlHandle( m_pList->GetSafeHwnd() );
	m_pParent->m_panel.Reposition(FALSE);
	m_pParent->m_panel.ExpandGroup( m_pGroup, TRUE );
}

void CAnimationGraphStateEditor::CPropGroup::OnCommand( UINT id )
{
	if (id > m_callbacks.size())
		return;
	m_pParent->StateEvent(m_callbacks[id]);
}

void CAnimationGraphStateEditor::CPropGroup::Disable()
{
	if ( m_pProps )
	{
		m_pProps->ClearSelection();
		m_pProps->SetFlags( m_pProps->GetFlags() | CPropertyCtrl::F_READ_ONLY );
	}

	int i = m_callbacks.size();
	while ( i-- )
		m_pGroup->FindItem( 100*m_id+i )->SetEnabled( FALSE );
}

CAnimationGraphStateEditor::CPropGroup::~CPropGroup()
{
	m_pItem->Remove();
	m_pGroup->Remove();
	delete m_pProps;
}

/*
 * CPropEditorImpl<CAGStatePtr>
 */

CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::CPropEditorImpl( CAnimationGraphStateEditor * pSE, CAGStatePtr pCurState )
: CPropEditor(pSE, ePET_State)
, m_pCurState(pCurState)
{
	m_pCurState->GetGraph()->AddListener(this);
}

CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::~CPropEditorImpl()
{
	m_pCurState->GetGraph()->RemoveListener(this);
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::OnStateEvent( EAGStateEvent evt, CAGStatePtr pState )
{
	if (pState != m_pCurState)
		return;

	switch (evt)
	{
	case eAGSE_ChangeParent:
		{
			UpdateLock updateLock(m_pSE);
			OnEvent(eSEE_RebuildCriteria);
			OnEvent(eSEE_RebuildOverridable);
			OnEvent(eSEE_RebuildExtra);
		}
		break;
	case eAGSE_CanNotChangeName:
		if ( m_pGeneralPanel )
		{
			CPropertyCtrl* pCtrl = m_pGeneralPanel->GetCtrl();
			CPropertyItem* pItem = pCtrl->GetRootItem();
			pItem = pItem->FindItemByFullName( "::State name" );
			pItem->GetVariable()->Set( pState->GetName() );
		}
		break;
	}
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::BuildExtras()
{
	UpdateLock updateLock(m_pSE);

	m_pOverridablePanel = m_pExtrasPanel = m_pCriteriaPanel = m_pTemplatePanel = NULL;

	static const Verb verbsExtras[] = 
	{
		Verb("Add", eSEE_AddExtra),
		Verb("Remove", eSEE_RemoveExtra),
		Verb()
	};

	if (!m_pCurState->IsTemplateState())
	{
		m_pOverridablePanel = m_pSE->AddPropertyGroup("Core Properties", m_pCurState->GetOverridableProperties(m_pSE));		
	}
	else
	{
		m_pTemplatePanel = m_pSE->AddPropertyGroup("Template Properties", m_pCurState->GetTemplateProperties(m_pSE));
	}
  m_pExtrasPanel = m_pSE->AddPropertyGroup("Extra Properties", m_pCurState->GetExtraProperties(), verbsExtras);
	m_pCriteriaPanel = m_pSE->AddPropertyGroup("Selection Criteria", m_pCurState->GetCriteriaProperties(m_pSE));
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::OnEvent( EStateEditorEvent evt )
{
	switch (evt)
	{
	case eSEE_InitContainer:
		{
			static const Verb verbsGeneral[] =
			{
				Verb("Delete...", eSEE_Delete),
				Verb()
			};
			m_pGeneralPanel = m_pSE->AddPropertyGroup("General", m_pCurState->GetGeneralProperties(), verbsGeneral);
			BuildExtras();
		}
		break;
	case eSEE_AddExtra:
		{
			std::vector<CString> items;
			CAnimationGraphPtr pGraph = m_pSE->m_pParent->GetAnimationGraph();
			for (CAnimationGraph::state_factory_iterator iter = pGraph->StateFactoryBegin(); iter != pGraph->StateFactoryEnd(); ++iter)
      {        
        CAGCategoryPtr pCat = (*iter)->GetCategory();
				if (!pCat)
					continue;
        if (!pCat->IsOverridable() && (!m_pCurState->IsTemplateState() || pCat->IsUsableWithTemplate())) 
          items.push_back( (*iter)->GetHumanName() );
      }
			std::sort(items.begin(), items.end());
			int item = ShowQuickPopup( items, &m_pSE->m_panel );
			if (item < 0)
				return;
			UpdateLock updateLock(m_pSE);

			m_pCurState->AddExtraState( pGraph->FindStateFactory(items[item]) );
			OnEvent(eSEE_RebuildExtra);
		}
		break;
	case eSEE_RemoveExtra:
		{
			std::vector<CString> items;
			m_pCurState->GetExtraStateStrings( items );
			int item = ShowQuickPopup( items, &m_pSE->m_panel );
			if (item < 0)
				return;
			UpdateLock updateLock(m_pSE);

			m_pCurState->RemoveExtraState( item );
			if (m_pExtrasPanel)
				m_pExtrasPanel->Reset( m_pCurState->GetExtraProperties() );
		}
		break;
	case eSEE_RebuildExtra:
		{
			UpdateLock updateLock(m_pSE);
			if (m_pExtrasPanel)
				m_pExtrasPanel->Reset( m_pCurState->GetExtraProperties() );
		}
		break;
	case eSEE_RebuildCriteria:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pCriteriaPanel)
				m_pCriteriaPanel->Reset( m_pCurState->GetCriteriaProperties(m_pSE) );
		}
		break;
	case eSEE_RebuildOverridable:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pOverridablePanel)
				m_pOverridablePanel->Reset( m_pCurState->GetOverridableProperties(m_pSE) );
		}
		break;
	case eSEE_RebuildTemplate:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pTemplatePanel)
				m_pTemplatePanel->Reset( m_pCurState->GetTemplateProperties(m_pSE) );
			OnEvent(eSEE_RebuildCriteria);
		}
		break;
	case eSEE_Delete:
		{
			IAGCheckedOperationPtr pOp = m_pCurState->CreateDeleteOp();
			if (IDYES == AfxMessageBox( "Delete " + m_pCurState->GetName() + "?\n\n" + pOp->GetExplanation(), MB_YESNO ))
				pOp->Perform();
		}
		break;
	case eSEE_RebuildExtraPanels:
		m_pSE->m_propGroups.resize(1);
		BuildExtras();
		break;
	}
}

bool CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::CheckSameEditor( CPropEditor* pEditor )
{
	return m_pCurState == static_cast<CPropEditorImpl*>(pEditor)->m_pCurState;
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGStatePtr>::ActivateParameterization( const TParameterizationId* pParamId, const CString& viewName )
{
	m_pCurState->ActivateParameterization( pParamId, viewName );
}

/*
* CPropEditorImpl< std::vector< CAGStatePtr > >
*/

CAnimationGraphStateEditor::CPropEditorImpl< std::vector< CAGStatePtr > >::CPropEditorImpl( CAnimationGraphStateEditor * pSE, std::vector< CAGStatePtr >& vCurStates )
: CPropEditor( pSE, ePET_States )
{
	m_vCurStates = vCurStates;
}

CAnimationGraphStateEditor::CPropEditorImpl< std::vector< CAGStatePtr > >::~CPropEditorImpl()
{
}

/*
void CAnimationGraphStateEditor::CPropEditorImpl< std::vector< CAGStatePtr > >::BuildExtras()
{
	UpdateLock updateLock(m_pSE);

	m_pOverridablePanel = m_pExtrasPanel = m_pCriteriaPanel = m_pTemplatePanel = NULL;

	static const Verb verbsExtras[] = 
	{
		Verb("Add", eSEE_AddExtra),
		Verb("Remove", eSEE_RemoveExtra),
		Verb()
	};

	if (!m_pCurState->IsTemplateState())
	{
		m_pOverridablePanel = m_pSE->AddPropertyGroup("Core Properties", m_pCurState->GetOverridableProperties(m_pSE));		
	}
	else
	{
		m_pTemplatePanel = m_pSE->AddPropertyGroup("Template Properties", m_pCurState->GetTemplateProperties(m_pSE));
	}
	m_pExtrasPanel = m_pSE->AddPropertyGroup("Extra Properties", m_pCurState->GetExtraProperties(), verbsExtras);
	m_pCriteriaPanel = m_pSE->AddPropertyGroup("Selection Criteria", m_pCurState->GetCriteriaProperties(m_pSE));
}
*/

void CAnimationGraphStateEditor::CPropEditorImpl< std::vector< CAGStatePtr > >::OnEvent( EStateEditorEvent evt )
{
	switch (evt)
	{
	case eSEE_InitContainer:
		{
			static const Verb verbsGeneral[] =
			{
				Verb("Delete All...", eSEE_Delete),
				Verb()
			};
			TAGPropMap props;
			m_pGeneralPanel = m_pSE->AddPropertyGroup("General", props, verbsGeneral);
//			BuildExtras();
		}
		break;
/*
	case eSEE_AddExtra:
		{
			std::vector<CString> items;
			CAnimationGraphPtr pGraph = m_pSE->m_pParent->GetAnimationGraph();
			for (CAnimationGraph::state_factory_iterator iter = pGraph->StateFactoryBegin(); iter != pGraph->StateFactoryEnd(); ++iter)
			{        
				CAGCategoryPtr pCat = (*iter)->GetCategory();
				if (!pCat)
					continue;
				if (!pCat->IsOverridable() && (!m_pCurState->IsTemplateState() || pCat->IsUsableWithTemplate())) 
					items.push_back( (*iter)->GetHumanName() );
			}
			std::sort(items.begin(), items.end());
			int item = ShowQuickPopup( items, &m_pSE->m_panel );
			if (item < 0)
				return;
			UpdateLock updateLock(m_pSE);

			m_pCurState->AddExtraState( pGraph->FindStateFactory(items[item]) );
			OnEvent(eSEE_RebuildExtra);
		}
		break;
	case eSEE_RemoveExtra:
		{
			std::vector<CString> items;
			m_pCurState->GetExtraStateStrings( items );
			int item = ShowQuickPopup( items, &m_pSE->m_panel );
			if (item < 0)
				return;
			UpdateLock updateLock(m_pSE);

			m_pCurState->RemoveExtraState( item );
			if (m_pExtrasPanel)
				m_pExtrasPanel->Reset( m_pCurState->GetExtraProperties() );
		}
		break;
	case eSEE_RebuildExtra:
		{
			UpdateLock updateLock(m_pSE);
			if (m_pExtrasPanel)
				m_pExtrasPanel->Reset( m_pCurState->GetExtraProperties() );
		}
		break;
	case eSEE_RebuildCriteria:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pCriteriaPanel)
				m_pCriteriaPanel->Reset( m_pCurState->GetCriteriaProperties(m_pSE) );
		}
		break;
	case eSEE_RebuildOverridable:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pOverridablePanel)
				m_pOverridablePanel->Reset( m_pCurState->GetOverridableProperties(m_pSE) );
		}
		break;
	case eSEE_RebuildTemplate:
		{
			UpdateLock updateLock(m_pSE);

			if (m_pTemplatePanel)
				m_pTemplatePanel->Reset( m_pCurState->GetTemplateProperties(m_pSE) );
			OnEvent(eSEE_RebuildCriteria);
		}
		break;
	case eSEE_RebuildExtraPanels:
		m_pSE->m_propGroups.resize(1);
		BuildExtras();
		break;
*/
	case eSEE_Delete:
		{
			CString explanation;
			explanation.Format( "Delete %d states?\n", m_vCurStates.size() );
			std::vector< CAGStatePtr >::iterator it, itEnd = m_vCurStates.end();
			std::vector< IAGCheckedOperationPtr > operations;
			int i = 0;
			for ( it = m_vCurStates.begin(); it != itEnd; ++it )
			{
				CAGState* pState = *it;
				IAGCheckedOperationPtr pOp = pState->CreateDeleteOp();
				operations.push_back( pOp );
				if ( ++i == 11 )
				{
					explanation += "\n<...>";
				}
				else if ( i < 11 )
				{
					explanation += '\n';
					explanation += pState->GetName();
					explanation += ":\n";
					explanation += pOp->GetExplanation();
					explanation += '\n';
				}
			}
			if (IDYES == AfxMessageBox( explanation, MB_YESNO ))
			{
				for ( int i = 0; i < operations.size(); ++i )
					operations[i]->Perform();
			}
		}
		break;
	}
}

bool CAnimationGraphStateEditor::CPropEditorImpl< std::vector< CAGStatePtr > >::CheckSameEditor( CPropEditor* pEditor )
{
	return m_vCurStates == static_cast<CPropEditorImpl*>(pEditor)->m_vCurStates;
}

/*
 * CPropEditorImpl<CAGViewPtr>
 */

CAnimationGraphStateEditor::CPropEditorImpl<CAGViewPtr>::CPropEditorImpl( CAnimationGraphStateEditor * pSE, CAGViewPtr pCurView )
: CPropEditor(pSE, ePET_View)
, m_pCurView(pCurView)
{
}

CAnimationGraphStateEditor::CPropEditorImpl<CAGViewPtr>::~CPropEditorImpl()
{
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGViewPtr>::OnEvent( EStateEditorEvent evt )
{
	switch (evt)
	{
	case eSEE_InitContainer:
		{
			static const Verb verbsGeneral[] =
			{
				Verb("Delete...", eSEE_Delete),
				Verb()
			};
			m_pSE->AddPropertyGroup("General", m_pCurView->GetGeneralProperties(), verbsGeneral);
		}
		break;
	case eSEE_Delete:
		{
			if ( IDYES == AfxMessageBox( "Delete " + m_pCurView->GetName() + "?", MB_YESNO|MB_DEFBUTTON2|MB_ICONEXCLAMATION ))
				m_pCurView->GetGraph()->RemoveView( m_pCurView );
		}
		break;
	}
}

bool CAnimationGraphStateEditor::CPropEditorImpl<CAGViewPtr>::CheckSameEditor( CPropEditor* pEditor )
{
	return m_pCurView == static_cast<CPropEditorImpl*>(pEditor)->m_pCurView;
}

/*
 * CPropEditorImpl<CAGInputPtr>
 */

CAnimationGraphStateEditor::CPropEditorImpl<CAGInputPtr>::CPropEditorImpl( CAnimationGraphStateEditor * pSE, CAGInputPtr pCurInput )
: CPropEditor(pSE, ePET_Input)
, m_pCurInput(pCurInput)
{
	m_pCurInput->GetGraph()->AddListener( this );
}

CAnimationGraphStateEditor::CPropEditorImpl<CAGInputPtr>::~CPropEditorImpl()
{
	m_pCurInput->GetGraph()->RemoveListener( this );
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGInputPtr>::OnEvent( EStateEditorEvent evt )
{
	switch (evt)
	{
	case eSEE_InitContainer:
		{
			static const Verb verbsGeneral[] =
			{
				Verb("Delete", eSEE_Delete),
				Verb()
			};
			m_pSE->AddPropertyGroup("General", m_pCurInput->GetGeneralProperties(), verbsGeneral);
			OnEvent( eSEE_RebuildInputTypedProperties );
		}
		break;
	case eSEE_RebuildInputTypedProperties:
		{
			UpdateLock updateLock( m_pSE );
			if ( m_pSE->m_propGroups.size() >= 2 )
				m_pSE->m_propGroups.erase( m_pSE->m_propGroups.begin()+1 );

			static const Verb verbs[] = 
			{
				Verb("Add", eSEE_AddValue),
				Verb("Remove", eSEE_RemoveValue),
				Verb("Rename", eSEE_RenameValue),
				Verb()
			};

			switch (m_pCurInput->GetType())
			{
			case eAGIT_Float:
				m_pSE->AddPropertyGroup("Float Properties", m_pCurInput->GetFloatProperties());
				break;
			case eAGIT_Integer:
				m_pSE->AddPropertyGroup("Integer Properties", m_pCurInput->GetIntegerProperties());
				break;
			case eAGIT_String:
				m_pSE->AddPropertyGroup("Key Properties", m_pCurInput->GetKeyProperties(), verbs);
				break;
			default:
				assert(false);
			}
		}
		break;
	case eSEE_RebuildInputPanel:
		{
			UpdateLock updateLock( m_pSE );
			m_pSE->m_propGroups.clear();
			OnEvent(eSEE_InitContainer);
		}
		break;
	case eSEE_AddValue:
		{
			CString name;
			for (int i=0; ; i++)
			{
				name.Format("input%d", i);
				if (!m_pCurInput->HasKey(name))
					break;
			}
			CStringDlg dlg;
			dlg.m_strString = name;
			if (dlg.DoModal() == IDOK)
				m_pCurInput->AddKey(dlg.m_strString);
		}
		OnEvent(eSEE_RebuildInputPanel);
		break;
	case eSEE_RemoveValue:
		{
			CListBox * pListBox = m_pSE->m_propGroups.back()->GetList();
			int sel = pListBox->GetCurSel();
			if (sel != LB_ERR)
			{
				CString value;
				pListBox->GetText( sel, value );
				m_pCurInput->RemoveKey( value );
			}
		}
		OnEvent(eSEE_RebuildInputPanel);
		break;
	case eSEE_RenameValue:
		{
			CListBox * pListBox = m_pSE->m_propGroups.back()->GetList();
			int sel = pListBox->GetCurSel();
			if (sel != LB_ERR)
			{
				CString value;
				pListBox->GetText( sel, value );
				CStringDlg dlg;
				dlg.m_strString = value;
				if (dlg.DoModal() == IDOK)
				{
					m_pCurInput->RemoveKey(value);
					m_pCurInput->AddKey(dlg.m_strString);
				}
			}
		}
		OnEvent(eSEE_RebuildInputPanel);
		break;
	}
}

bool CAnimationGraphStateEditor::CPropEditorImpl<CAGInputPtr>::CheckSameEditor( CPropEditor* pEditor )
{
	return m_pCurInput == static_cast<CPropEditorImpl*>(pEditor)->m_pCurInput;
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGInputPtr>::OnInputEvent( EAGInputEvent event, CAGInputPtr pInput )
{
	if (m_pCurInput != pInput)
		return;

	switch (event)
	{
	case eAGIE_ChangeType:
		OnEvent( eSEE_RebuildInputTypedProperties );
		break;
	}
}

/*
 * CPropEditorImpl<CAGLinkPtr>
 */

CAnimationGraphStateEditor::CPropEditorImpl<CAGLinkPtr>::CPropEditorImpl( CAnimationGraphStateEditor * pSE, CAGLinkPtr pCurLink )
: CPropEditor(pSE, ePET_Link)
, m_pCurLink(pCurLink)
{
	m_pCurLink->GetGraph()->AddListener( this );
}

CAnimationGraphStateEditor::CPropEditorImpl<CAGLinkPtr>::~CPropEditorImpl()
{
	m_pCurLink->GetGraph()->RemoveListener( this );
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGLinkPtr>::OnLinkEvent( EAGLinkEvent event, CAGLinkPtr pLink )
{
	if (m_pCurLink != pLink)
		return;

	switch (event)
	{
	case eAGLE_ChangeMapping:
		m_pSE->m_pParent->PostMessage( WM_COMMAND, ID_AGLINK_MAPPINGCHANGED, 0 );
		break;
	}
}

void CAnimationGraphStateEditor::CPropEditorImpl<CAGLinkPtr>::OnEvent( EStateEditorEvent evt )
{
	switch (evt)
	{
	case eSEE_InitContainer:
		{
			UpdateLock updateLock( m_pSE );

			m_pSE->m_propGroups.resize(0);

			static const Verb verbsGeneral[] =
			{
				Verb("Delete", eSEE_Delete),
				Verb()
			};
			m_pSE->AddPropertyGroup("General", m_pCurLink->GetGeneralProperties(), verbsGeneral);

			CAGStatePtr from, to;
			m_pCurLink->GetLinkedStates( from, to );
			assert( from && to );
			if ( from->GetParamsDeclaration()->empty() && to->GetParamsDeclaration()->empty() )
				break;

			CParamsDeclaration* fromParams = from->GetParamsDeclaration();
			CParamsDeclaration* toParams = to->GetParamsDeclaration();

			m_pSE->AddPropertyGroup("Parameters Mapping", m_pCurLink->GetMappingProperties());
		}
		break;
	}
}

bool CAnimationGraphStateEditor::CPropEditorImpl<CAGLinkPtr>::CheckSameEditor( CPropEditor* pEditor )
{
	return m_pCurLink == static_cast<CPropEditorImpl*>(pEditor)->m_pCurLink;
}

/*
 * class proper
 */

CXTPDockingPane * CAnimationGraphStateEditor::Init( CAnimationGraphDialog * pParent, UINT id )
{
	m_pParent = pParent;

	CRect rc(0,0,250,500);
	CXTPDockingPane * pDockPane = pParent->GetDockingPaneManager()->CreatePane( id, rc, dockRightOf );
	pDockPane->SetTitle( _T("State Editor") );

	m_panel.Create( WS_CHILD|/*WS_VISIBLE|*/WS_CLIPSIBLINGS|WS_CLIPCHILDREN, rc, m_pParent, 0 );
	m_panel.SetBehaviour(xtpTaskPanelBehaviourExplorer);
	m_panel.SetTheme(xtpTaskPanelThemeNativeWinXP);
	m_panel.SetSelectItemOnFocus(TRUE);
	m_panel.AllowDrag(FALSE);
	m_panel.SetAnimation( xtpTaskPanelAnimationNo );
	m_panel.GetPaintManager()->m_rcGroupOuterMargins.SetRect(0,0,0,0);
	m_panel.GetPaintManager()->m_rcGroupInnerMargins.SetRect(0,0,0,0);
	m_panel.GetPaintManager()->m_rcItemOuterMargins.SetRect(0,0,0,0);
	m_panel.GetPaintManager()->m_rcItemInnerMargins.SetRect(1,1,1,1);
	m_panel.GetPaintManager()->m_rcControlMargins.SetRect(2,0,2,0);
	m_panel.GetPaintManager()->m_nGroupSpacing = 0;

	return pDockPane;
}

void CAnimationGraphStateEditor::ClearActiveItem()
{
	UpdateLock updateLock(this);
	if ( m_pCurEdit )
		m_pCurEdit->ActivateParameterization( NULL, "" );
	m_pCurEdit = NULL;
	m_propGroups.resize(0);
	m_pParent->SetParamsDeclaration( NULL );
}

void CAnimationGraphStateEditor::SetEditor( _smart_ptr<CPropEditor> pImpl )
{
	assert( !!pImpl );
	if (m_pCurEdit)
	{
		if (pImpl->IsSameEditor(m_pCurEdit))
			return;
		m_pCurEdit->ActivateParameterization( NULL, "" );
	}
	UpdateLock updateLock(this);
	m_pCurEdit = pImpl;
	m_propGroups.resize(0);
	m_pCurEdit->OnEvent( eSEE_InitContainer );
//	m_panel.Invalidate();

	m_pParent->SetParamsDeclaration( m_pCurEdit->GetParamsDeclaration() );
}

CAnimationGraphStateEditor::CPropGroupPtr CAnimationGraphStateEditor::AddPropertyGroup( const CString& name, const TAGPropMap& props, const Verb * pVerbs)
{
	UpdateLock updateLock(this);

	CPropGroupPtr pPropGroup = new CPropGroup( this, name, m_propGroups.size()+1, props, pVerbs );
	m_propGroups.push_back(pPropGroup);
	return pPropGroup;
}

CAnimationGraphStateEditor::CPropGroupPtr CAnimationGraphStateEditor::AddPropertyGroup( const CString& name, const std::vector<CString>& props, const Verb * pVerbs)
{
	UpdateLock updateLock(this);

	CPropGroupPtr pPropGroup = new CPropGroup( this, name, m_propGroups.size()+1, props, pVerbs );
	m_propGroups.push_back(pPropGroup);
	return pPropGroup;
}

void CAnimationGraphStateEditor::OnCommand( UINT id )
{
	UINT grp = id / 100;
	if (!grp || grp > m_propGroups.size())
		return;
	m_propGroups[grp-1]->OnCommand( id - grp*100 );
}

void CAnimationGraphStateEditor::StateEvent( EStateEditorEvent event )
{
	if (m_pCurEdit)
		m_pCurEdit->OnEvent( event );
}

void CAnimationGraphStateEditor::OnStateParamSelChanged( const TParameterizationId* pParamsId, const CString& viewName )
{
	if ( m_pCurEdit )
	{
		m_pCurEdit->ActivateParameterization( pParamsId, viewName );
		
		UpdateLock updateLock(this);

		// rebuild panels
		if ( !pParamsId )
		{
			m_propGroups.resize(0);
			m_pCurEdit->OnEvent( eSEE_InitContainer );
		}
		else
		{
			m_pCurEdit->OnEvent( eSEE_RebuildCriteria );
			m_pCurEdit->OnEvent( eSEE_RebuildTemplate );
			if ( m_propGroups.size() >= 1 && m_propGroups[0] )
				m_propGroups[0]->Disable();
			if ( m_propGroups.size() >= 3 && m_propGroups[2] )
				m_propGroups[2]->Disable();
		}
	}
}
