//========= Copyright © 1996-2009, Valve Corporation, All rights reserved. ====
//
//=============================================================================

#ifndef TREELIST_H
#define TREELIST_H
#ifdef _WIN32
#pragma once
#endif

#include "UtlVector.h"


#define TREELIST_MSG_TOGGLE_STATE		"TreeList_ToggleState"
#define TREELIST_MSG_LEFT_DRAG_DROP		"TreeList_LeftDragDrop"
#define TREELIST_MSG_RIGHT_DRAG_DROP	"TreeList_RightDragDrop"
#define TREELIST_MSG_SEL_CHANGE			"TreeList_SelChange"
#define TREELIST_MSG_KEY_DOWN			"TreeList_KeyDown"


struct TreeListItemState_t
{
	void *pItem;
	bool bExpanded;
};


class CTreeList : public CTreeCtrl
{
public:

	CTreeList();
	virtual ~CTreeList();

	void DeleteAllItems();
	void AddItem(void *pItem, void *pParent, const char *pszText, bool bHasCheckBox );

	inline bool SubclassDlgItem(int nCtrlID, CWnd *pwndParent);
	inline void SetRedraw(bool bRedraw);
	inline void Invalidate(bool bErase = true);

	void SelectItem(void *pItem);
	void EnsureVisible(void *pItem);

	void SelectNearestItem( int nItem );
	void EnsureVisible( int nItem );

	void ExpandAll();
	void EnableChecks();

	void UpdateItem(void *pItem, const char *pszText);
	
	void *GetSelectedItem();
	int GetSelectedIndex();
	
	int GetItemCount();
	void *GetItem(int nIndex);
	void SetCheck(void *pItem, int nCheckState);
	int GetCheck(void *pItem);
	
	void ExpandItem(void *pItem);
	void CollapseItem(void *pItem);

	void SaveTreeListExpandStates();
	void RestoreTreeListExpandStates();

	// Virtuals that can be overridden by the derived class
	virtual void OnRenameItem(void *pItem, const char *pszText) {}

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CGroupList)
	//}}AFX_VIRTUAL

protected:

	// Generated message map functions
	//{{AFX_MSG(CGroupList)
	afx_msg void OnBegindrag(NMHDR* pNMHDR, LRESULT* pResult);
	afx_msg void OnEndLabelEdit(NMHDR* pNMHDR, LRESULT* pResult);
	afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
	afx_msg void OnLButtonUp(UINT nFlags, CPoint point);
	afx_msg void OnLButtonDblClk(UINT nFlags, CPoint point);
	afx_msg void OnMouseMove(UINT nFlags, CPoint point);
	afx_msg void OnRButtonDown(UINT nFlags, CPoint point);
	afx_msg void OnRButtonUp(UINT nFlags, CPoint point);
	afx_msg void OnTimer(UINT nIDEvent);
	afx_msg void OnContextMenu(CWnd *, CPoint);
	afx_msg void OnSelChange(NMHDR *pNMHDR, LRESULT *pResult);
	afx_msg void OnKeyDown( NMHDR *pNMHDR, LRESULT *pResult );
	//}}AFX_MSG

	DECLARE_MESSAGE_MAP()

private:

	enum DropType_t
	{
		DROP_LEFT = 0,
		DROP_RIGHT,
	};

	void ExpandRecursive(HTREEITEM hItem);

	int GetCheck(HTREEITEM hItem);

	HTREEITEM FindHTreeItem(void *pItem);
	HTREEITEM FindHTreeItemRecursive(HTREEITEM hItem, void *pItem);

	void BeginDrag(CPoint pt, HTREEITEM hItem);
	void Drop(DropType_t eDropType, UINT nFlags, CPoint point);

	bool m_bRButtonDown;

	CPoint m_ptRButtonDown;

	CPoint m_ptLDown;

	CImageList m_cNormalImageList;
	CImageList *m_pDragImageList;

	HTREEITEM m_hDragItem;

	CUtlVector<void *> m_Items;
	CUtlVector<TreeListItemState_t> m_ItemState;
};


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
void CTreeList::Invalidate(bool bErase)
{
	CTreeCtrl::Invalidate(bErase ? TRUE : FALSE);
}


//-----------------------------------------------------------------------------
// Enables or disables updates. Useful for populating the list and only
// updating at the end.
//-----------------------------------------------------------------------------
void CTreeList::SetRedraw(bool bRedraw)
{
	CTreeCtrl::SetRedraw(bRedraw ? TRUE : FALSE);
}


//-----------------------------------------------------------------------------
//-----------------------------------------------------------------------------
bool CTreeList::SubclassDlgItem(int nCtrlID, CWnd *pwndParent)
{
	return (CTreeCtrl::SubclassDlgItem(nCtrlID, pwndParent) == TRUE);
}


#endif // TREELIST_H