MFC TreeCtrl 控件设置双击选项的背景颜色

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

前言

MFC TreeCtrl 控件设置双击选项的背景颜色

一、实现思路

1. 自定义一个类继承自CTreeCtrl，然后添加NM_CUSTOMDRAW消息事件处理
2. 在主程序中添加双击事件的处理函数，设置tree控件的item类的data内容
3. 在自定义类的NM_CUSTOMDRAW消息处理函数中根据item项的data数据设置其背景颜色

二、实现步骤

1.创建类继承自CTreeCtrl

代码如下（示例）：

//头文件
class WorksSceneTreeCtrl : public CTreeCtrl
{
	DECLARE_DYNAMIC(WorksSceneTreeCtrl )
public:
	WorksSceneTreeCtrl ();
	virtual ~WorksSceneTreeCtrl ();
protected:
	DECLARE_MESSAGE_MAP()
public:
	afx_msg void OnNMCustomdraw(NMHDR* pNMHDR, LRESULT* pResult);
};
//实现文件
IMPLEMENT_DYNAMIC(WorksSceneTreeCtrl , CTreeCtrl)
BEGIN_MESSAGE_MAP(WorksSceneTreeCtrl , CTreeCtrl)
	ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, &WorksSceneTreeCtrl ::OnNMCustomdraw)
END_MESSAGE_MAP()

void WorksSceneTreeCtrl::OnNMCustomdraw(NMHDR* pNMHDR, LRESULT* pResult)
{
	LPNMCUSTOMDRAW pNMCD = reinterpret_cast<LPNMCUSTOMDRAW>(pNMHDR);
	*pResult = CDRF_DODEFAULT;
	
	if (this->m_hWnd != pNMHDR->hwndFrom)
		return;

	LPNMTVCUSTOMDRAW pCustomDraw = (LPNMTVCUSTOMDRAW)pNMHDR;
	switch (pCustomDraw->nmcd.dwDrawStage)
	{
	case CDDS_PREPAINT:
		*pResult = CDRF_NOTIFYITEMDRAW;
		return;

	case CDDS_ITEMPREPAINT:
		if (this->IsWindowEnabled() == 1)
		{
			info* pSec = (info*)(pCustomDraw->nmcd.lItemlParam);
			if (pSec && pSec->is_select) {
				pCustomDraw->clrText = RGB(255, 0, 0);//设置文字颜色
				pCustomDraw->clrTextBk = RGB(255, 0, 0);//设置背景颜色
			}
			*pResult = CDRF_NOTIFYPOSTPAINT;
			return;
		}
		else {
			*pResult = CDRF_DODEFAULT;
			return;
		}
	}
}

2.主文件代码

代码如下（示例）：

//定义item项的data数据
typedef struct {
	string name;
	bool is_select;
	vector<info> children;
}info;

WorksSceneTreeCtrl m_tree;//树形控件

//初始化树形控件
info section_ptr;
append_leafs(TVI_ROOT, section_ptr);
//添加叶子节点设置data值
void append_leafs(HTREEITEM parent, info* pSec) {	
	HTREEITEM ht = m_tree.InsertItem(pSec->name.c_str(), parent);
	m_tree.SetItemData(ht, (DWORD_PTR)pSec);

	for (auto it = pSec->children.begin(); it != pSec->children.end(); it++){
		append_leafs(ht, (*it));
	}
}

//实现双击效果
void OnDblclkTreeExplorer(NMHDR* pNMHDR, LRESULT* pResult)
{
	// TODO: 在此添加控件通知处理程序代码
	UINT uFlags;
	CPoint pt, ptClient;
	::GetCursorPos(&pt);
	ptClient = pt;
	m_tree.ScreenToClient(&ptClient);

	HTREEITEM htNew = m_tree.HitTest(ptClient, &uFlags);
	if (htNew != NULL && (TVHT_ONITEMLABEL & uFlags)) {
		info* pSec = ((info*)m_tree.GetItemData(htNew));
		if (pSec) {
			pSec->is_select = true;
		}
	}
	*pResult = 0;
}

总结

本例子是利用树形控件的data数据指针来指示item的状态，利用重写的树形控件的OnNMCustomdraw消息的方法来判断item的状态以处理item的背景颜色和文字颜色