chenluhua1980
2026-01-08 e878ac7788fef44639145ffd971ab0eb06d692ad
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#pragma once
#include <afxcmn.h>
#include <map>
 
class CApredTreeCtrl2 :
    public CTreeCtrl
{
public:
    typedef struct tagBADGE
    {
        HTREEITEM hTreeItem;
        COLORREF badgeBackground;
        COLORREF badgeForeground;
        int type;                       /* 0: ÎÞ,²»ÏÔʾ*/
        int number;
    } BADGE;
 
 
    DECLARE_DYNAMIC(CApredTreeCtrl2)
 
public:
    CApredTreeCtrl2();
    virtual ~CApredTreeCtrl2();
 
 
public:
    virtual COLORREF SetBkColor(COLORREF crColor);
    void SetItemBadge(HTREEITEM hItem, COLORREF badgeBackground, COLORREF badgeForeground);
    void ShowItemBadgeNumber(HTREEITEM hItem, int number);
    void ShowItemBadgeDotMode(HTREEITEM hItem);
    void HideItemBadge(HTREEITEM hItem);
 
private:
    HBRUSH GetItemBkBrush(HTREEITEM hItem);
    HBRUSH GetItemBtnBrush(HTREEITEM hItem);
    HPEN GetItemBtnPen(HTREEITEM hItem);
    void DrawItemButton(HTREEITEM hItem, HDC hDC, CRect* pRect);
 
 
private:
    std::map<HTREEITEM, BADGE> m_badges;
    std::map<HTREEITEM, int> m_items;
    HBRUSH m_hBrushItem[3];
    COLORREF m_crItemBk[3];
    COLORREF m_crItemBtn[3];
 
public:
    DECLARE_MESSAGE_MAP()
    afx_msg void OnNMCustomdraw(NMHDR* pNMHDR, LRESULT* pResult);
    afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
    virtual void PreSubclassWindow();
};