MFC CTreeCtrl 使用复选框、添加节点图标，使用位图的图片列表

 // MFCApplication7Dlg.cpp: 实现文件 // #include "pch.h" #include "framework.h" #include "MFCApplication7.h" #include "MFCApplication7Dlg.h" #include "afxdialogex.h" #ifdef _DEBUG #define new DEBUG_NEW #endif // 用于应用程序“关于”菜单项的 CAboutDlg 对话框 class CAboutDlg : public CDialogEx { public: CAboutDlg(); // 对话框数据 #ifdef AFX_DESIGN_TIME enum { IDD = IDD_ABOUTBOX }; #endif protected: virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV 支持 // 实现 protected: DECLARE_MESSAGE_MAP() }; CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX) { } void CAboutDlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); } BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx) END_MESSAGE_MAP() // CMFCApplication7Dlg 对话框 CMFCApplication7Dlg::CMFCApplication7Dlg(CWnd* pParent /*=nullptr*/) : CDialogEx(IDD_MFCAPPLICATION7_DIALOG, pParent) { m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); } void CMFCApplication7Dlg::DoDataExchange(CDataExchange* pDX) { CDialogEx::DoDataExchange(pDX); DDX_Control(pDX, IDC_TREE1, m_Tree); } BEGIN_MESSAGE_MAP(CMFCApplication7Dlg, CDialogEx) ON_WM_SYSCOMMAND() ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_BUTTON1, &CMFCApplication7Dlg::OnBnClickedButton1) ON_BN_CLICKED(IDC_BUTTON2, &CMFCApplication7Dlg::OnBnClickedButton2) ON_BN_CLICKED(IDC_BUTTON3, &CMFCApplication7Dlg::OnBnClickedButton3) ON_BN_CLICKED(IDC_BUTTON4, &CMFCApplication7Dlg::OnBnClickedButton4) END_MESSAGE_MAP() // CMFCApplication7Dlg 消息处理程序 BOOL CMFCApplication7Dlg::OnInitDialog() { CDialogEx::OnInitDialog(); // 将“关于...”菜单项添加到系统菜单中。 // IDM_ABOUTBOX 必须在系统命令范围内。 ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX); ASSERT(IDM_ABOUTBOX < 0xF000); CMenu* pSysMenu = GetSystemMenu(FALSE); if (pSysMenu != nullptr) { BOOL bNameValid; CString strAboutMenu; bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX); ASSERT(bNameValid); if (!strAboutMenu.IsEmpty()) { pSysMenu->AppendMenu(MF_SEPARATOR); pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu); } } // 设置此对话框的图标。 当应用程序主窗口不是对话框时，框架将自动 // 执行此操作 SetIcon(m_hIcon, TRUE); // 设置大图标 SetIcon(m_hIcon, FALSE); // 设置小图标 // TODO: 在此添加额外的初始化代码 // HTREEITEM hItem1A = m_wndTree.InsertItem(_T("LAYER1_A")/*,0,1*/);//添加节点 InsertItem(节点名,未选中时图标下标，节点选中时图标下标,父节点) // m_wndTree.InsertItem(_T("LAYER2_1A")/*,0,1*/, hItem1A, TVI_LAST); // HTREEITEM hItem2_1B = m_wndTree.InsertItem(_T("LAYER2_1B")/*,0,1*/, hItem1A, TVI_LAST); // m_wndTree.InsertItem(_T("LAYER3_1A")/*,0,1*/, hItem2_1B); // m_wndTree.InsertItem(_T("LAYER3_1B")/*,0,1*/, hItem2_1B); // m_wndTree.InsertItem(_T("LAYER3_1C")/*,0,1*/, hItem2_1B); // m_wndTree.InsertItem(_T("LAYER2_1C")/*,0,1*/, hItem1A, TVI_LAST); // HTREEITEM hItem1B = m_wndTree.InsertItem(_T("LAYER1_B")/*,0,1*/); // m_wndTree.InsertItem(_T("LAYER2_2A")/*,0,1*/, hItem1B, TVI_LAST); // m_wndTree.InsertItem(_T("LAYER2_2B")/*,0,1*/, hItem1B, TVI_LAST); // m_wndTree.InsertItem(_T("LAYER2_2C")/*,0,1*/, hItem1B, TVI_LAST); return TRUE; // 除非将焦点设置到控件，否则返回 TRUE } void CMFCApplication7Dlg::OnSysCommand(UINT nID, LPARAM lParam) { if ((nID & 0xFFF0) == IDM_ABOUTBOX) { CAboutDlg dlgAbout; dlgAbout.DoModal(); } else { CDialogEx::OnSysCommand(nID, lParam); } } // 如果向对话框添加最小化按钮，则需要下面的代码 // 来绘制该图标。 对于使用文档/视图模型的 MFC 应用程序， // 这将由框架自动完成。 void CMFCApplication7Dlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // 用于绘制的设备上下文 SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0); // 使图标在工作区矩形中居中 int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // 绘制图标 dc.DrawIcon(x, y, m_hIcon); } else { CDialogEx::OnPaint(); } } //当用户拖动最小化窗口时系统调用此函数取得光标 //显示。 HCURSOR CMFCApplication7Dlg::OnQueryDragIcon() { return static_cast<HCURSOR>(m_hIcon); } void CMFCApplication7Dlg::OnBnClickedButton1() { // TODO: 在此添加控件通知处理程序代码 m_imageListTest.Create(IDB_BITMAP_TEST, 16, 5, RGB(255, 255, 255)); m_Tree.SetImageList(&m_imageListTest, TVSIL_NORMAL); // 添加图片链表，所有不设置图标的节点都使用第一块图标 HTREEITEM hroot = m_Tree.InsertItem(_T("父亲"), 2, 3); HTREEITEM hroot_sub1 = m_Tree.InsertItem(_T("儿子1"), 1, 1, hroot, TVI_LAST); HTREEITEM hroot_sub2 = m_Tree.InsertItem(_T("儿子2"), hroot, TVI_LAST); HTREEITEM hroot_sub3 = m_Tree.InsertItem(_T("儿子3"), hroot); HTREEITEM hroot_sub1_sub1 = m_Tree.InsertItem(_T("孙子1-1"), hroot_sub1); HTREEITEM hroot_sub1_sub2 = m_Tree.InsertItem(_T("孙子1-2"), hroot_sub1); HTREEITEM hroot_sub3_sub1 = m_Tree.InsertItem(_T("孙子3-1"), hroot_sub3); HTREEITEM hroot_sub3_sub2 = m_Tree.InsertItem(_T("孙子3-2"), hroot_sub3); } void CMFCApplication7Dlg::OnBnClickedButton2() { // TODO: 在此添加控件通知处理程序代码 CString str; HTREEITEM hRoot = m_Tree.GetRootItem(); str = GetAllSelectedItem(hRoot); SetWindowText(str); } void CMFCApplication7Dlg::OnBnClickedButton3() { // TODO: 在此添加控件通知处理程序代码 HTREEITEM hRoot = m_Tree.GetRootItem(); m_Tree.SetCheck(hRoot); if (TRUE == m_Tree.ItemHasChildren(hRoot)) { BOOL bCheck = TRUE; // 查询所有节点，递归 HTREEITEM hChild = m_Tree.GetChildItem(hRoot); while (NULL != hChild) { m_Tree.SetCheck(hChild, bCheck); bCheck = !bCheck; hChild = m_Tree.GetNextSiblingItem(hChild); } } } CString CMFCApplication7Dlg::GetAllSelectedItem(HTREEITEM hRoot) { CString str; if (TRUE == m_Tree.GetCheck(hRoot)) { // 插入所有页节点 str += m_Tree.GetItemText(hRoot); str += _T(" "); } if (TRUE == m_Tree.ItemHasChildren(hRoot)) { // 查询所有节点，递归 HTREEITEM hChild = m_Tree.GetChildItem(hRoot); while (NULL != hChild) { str += GetAllSelectedItem(hChild); hChild = m_Tree.GetNextSiblingItem(hChild); } } return str; } void CMFCApplication7Dlg::SetState(HTREEITEM item) { BOOL bCheck = m_Tree.GetCheck(item); if (m_Tree.ItemHasChildren(item)) { HTREEITEM itemTemp = m_Tree.GetChildItem(item); while (itemTemp) { m_Tree.SetCheck(itemTemp, bCheck); SetState(itemTemp); itemTemp = m_Tree.GetNextSiblingItem(itemTemp); } } } void CMFCApplication7Dlg::OnBnClickedButton4() { // TODO: 在此添加控件通知处理程序代码 HTREEITEM hRoot = m_Tree.GetRootItem(); SetState(hRoot); }