在开发MFC应用程序时，经常需要使用树形控件来展示层次结构的数据。有时，为了满足特定的需求，可能需要对树形控件进行自定义。例如，添加复选框功能，允许用户通过勾选或取消勾选来选择或取消选择树形控件中的节点。本文将介绍如何实现一个带有复选框的树形控件，并展示其基本用法。
要实现一个带有复选框的树形控件，需要遵循以下步骤：
首先，在资源编辑器中为树形控件设置CheckBox样式。这可以通过选择树形控件，然后在属性窗口中设置样式来完成。
接下来，需要在项目中添加一个新的位图资源，命名为IDB_STATE，其中包含三种图像：空白图像、未选中的复选框和选中的复选框。
然后，从CTreeControl派生出一个新的类。在这个示例中，使用了之前文章中相同的类。首先，使用以下语句加载状态图像：
m_imageState.Create(IDB_STATE, 16, 1, RGB(255, 255, 255));
SetImageList(&m_imageState, TVSIL_STATE);
这个图像将与树形控件中的其他图像一起使用。
为了处理用户点击复选框以选中或取消选中节点的事件，需要响应Windows消息WM_LBUTTONDOWN。以下是处理该事件的示例代码：
void CXMLTree::OnLButtonDown(UINT nFlags, CPoint point) {
UINT uFlags = 0;
HTREEITEM hti = HitTest(point, &uFlags);
if (uFlags & TVHT_ONITEMSTATEICON) {
ToggleItemState(hti);
return;
}
CTreeCtrl::OnLButtonDown(nFlags, point);
}
此外，还需要响应WM_KEYDOWN消息，以便用户可以使用空格键来切换复选框的状态。
if (nChar == VK_SPACE) {
HTREEITEM hti = GetSelectedItem();
ToggleItemState(hti);
return;
}
CTreeCtrl::OnKeyDown(nChar, nRepCnt, nFlags);
为了支持复选框的全选和全不选功能，需要添加一些辅助函数来获取或设置树形控件中复选框的状态。以下是一些示例函数：
BOOL CXMLTree::IsItemChecked(HTREEITEM hItem) {
return GetItemState(hItem, TVIS_STATEIMAGEMASK) >> 12 == 2;
}
HTREEITEM CXMLTree::GetFirstCheckedItem() {
for (HTREEITEM hItem = GetRootItem(); hItem != NULL; hItem = GetNextItem(hItem)) {
if (IsItemChecked(hItem)) return hItem;
}
return NULL;
}
HTREEITEM CXMLTree::GetNextCheckedItem(HTREEITEM hItem) {
for (hItem = GetNextItem(hItem); hItem != NULL; hItem = GetNextItem(hItem)) {
if (IsItemChecked(hItem)) return hItem;
}
return NULL;
}
为了实现全选和全不选功能，需要在插入节点时添加一个新的函数。以下是实现该功能的示例代码：
void CXMLTree::ToggleItemState(HTREEITEM hti) {
int iImage = GetItemState(hti, TVIS_STATEIMAGEMASK) >> 12;
SetItemState(hti, INDEXTOSTATEIMAGEMASK(iImage == 1 ? 2 : 1), TVIS_STATEIMAGEMASK);
if (ItemHasChildren(hti)) {
HTREEITEM htiChild = GetChildItem(hti);
if (htiChild) {
ToggleItemState(htiChild);
HTREEITEM htiSibling = GetNextSiblingItem(htiChild);
while (htiSibling) {
ToggleItemState(htiSibling);
htiSibling = GetNextSiblingItem(htiSibling);
}
}
}
}