/*
* All or portions of this file Copyright (c) Amazon.com, Inc. or its affiliates or
* its licensors.
*
* For complete copyright and license terms please see the LICENSE at the root of this
* distribution (the "License"). All use of this software is governed by the License,
* or, if provided, by the license below or the license accompanying this file. Do not
* remove or modify any license notices. This file is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*
*/
// Original file Copyright Crytek GMBH or its affiliates, used under license.

#include "StdAfx.h"
#include "SelectionTree_RefBrowser.h"
#include "CustomFileDialog.h"

IMPLEMENT_DYNAMIC(CSelectionTree_RefBrowser, CDialog)

BEGIN_MESSAGE_MAP(CSelectionTree_RefBrowser, CDialog)
	ON_NOTIFY(TVN_SELCHANGED, IDC_BST_REFTREE, OnSelChanged)
	ON_BN_CLICKED(IDC_BST_SHOWCONTENTS, OnShowContents)
	ON_WM_SIZE()
	ON_WM_HSCROLL()
END_MESSAGE_MAP()

CSelectionTree_RefBrowser::CSelectionTree_RefBrowser(CWnd* pParent /*=NULL*/)
	:	CDialog( CSelectionTree_RefBrowser::IDD, pParent )
	, m_refType( eSTTI_Undefined )
{
}

CSelectionTree_RefBrowser::~CSelectionTree_RefBrowser()
{
}

void CSelectionTree_RefBrowser::Init( ESelectionTreeTypeId type )
{
	m_refType = type;
}

const char* CSelectionTree_RefBrowser::GetSelectedRef()
{
	return m_SelectedRef.length() > 0 ? m_SelectedRef.c_str() : NULL;
}

void CSelectionTree_RefBrowser::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_BST_REFTREE, m_RefTree);
}


BOOL CSelectionTree_RefBrowser::OnInitDialog()
{
	CDialog::OnInitDialog();
	CButton* rb = ( CButton* )GetDlgItem( IDC_BST_SHOWCONTENTS );
	rb->ShowWindow(false);
	LoadReferences();
	return TRUE;
}

void CSelectionTree_RefBrowser::OnSelChanged( NMHDR *pNMHDR, LRESULT *pResult )
{
	HTREEITEM item = m_RefTree.GetSelectedItem();
	if( item )
	{
		m_SelectedRef = m_RefNames[ item ];
	}
	else
	{
		m_SelectedRef = "";
	}
}

void CSelectionTree_RefBrowser::OnShowContents()
{
	CButton* rb = ( CButton* )GetDlgItem( IDC_BST_SHOWCONTENTS );
	LoadReferences();
}

void CSelectionTree_RefBrowser::LoadReferences()
{
	assert( m_refType != eSTTI_Undefined );
	m_RefTree.DeleteAllItems();

	TSelectionTreeInfoList infoList;
	GetIEditor()->GetSelectionTreeManager()->GetInfoList( infoList );

	for ( TSelectionTreeInfoList::const_iterator it = infoList.begin(); it != infoList.end(); ++it )
	{
		const SSelectionTreeInfo& info = *it;

		if ( !info.IsTree )
		{
			const int blockCount = info.GetBlockCountById( m_refType );
			for ( int i = 0; i < blockCount; ++i )
			{
				SSelectionTreeBlockInfo blockInfo;
				bool ok = info.GetBlockById( blockInfo, m_refType, i);
				assert(ok);
				if ( ok )
				{
					HTREEITEM item = m_RefTree.InsertItem( blockInfo.Name , 0, 0, TVI_ROOT, TVI_SORT );
					m_RefNames[ item ] = blockInfo.Name;
				}
			}
		}
	}
}