CTreeCtrl自绘

ネット 2023-10-04 18:24:07 訪問数: null

CSWTreeCtrl.h）

#pragma once
 
#define _OWNER_DRAWN_TREE  // 自绘CTreeCtrl，可支持背景图片显示功能
 
class CSWTreeCtrl : public CTreeCtrl
{
    
    
	DECLARE_DYNAMIC(CSWTreeCtrl)
 
	// 成员私有结构定义
 
	// 构造/析构函数
public:
	CSWTreeCtrl();
	virtual ~CSWTreeCtrl();
 
	// 私有成员变量
private:
 
	// 私有成员函数
private:
	void CommonConstruct(); // 初始化
 
	// 受保护成员变量
protected:
	int m_nFirstColumnWidth; // 第一列宽度
	int m_nOffsetX;      	 // 相对于父窗体X偏移量
	LVBKIMAGE m_bkImage;	 // 背景图片
	CImageList m_imgBtns;	 // 数控件展开的+/-图标符号
 
	// 受保护成员函数
protected:
	BOOL CheckHit(CPoint point);
#ifdef _OWNER_DRAWN_TREE
	LRESULT CustomDrawNotify(LPNMTVCUSTOMDRAW lpNMTVCustomDraw);
	void OwnerDrawBackground(CDC* pDC);
#endif //_OWNER_DRAWN_TREE
 
	// 虚函数
protected:
 
	// 消息函数
protected:
	DECLARE_MESSAGE_MAP()
	afx_msg void OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags);
	afx_msg void OnMouseMove(UINT nFlags, CPoint point);
	afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
	afx_msg void OnLButtonDblClk(UINT nFlags, CPoint point);
	afx_msg void OnPaint();
	afx_msg void OnTimer(UINT_PTR nIDEvent);
	afx_msg BOOL OnEraseBkgnd(CDC* pDC);
#ifdef _OWNER_DRAWN_TREE
	afx_msg BOOL OnMouseWheel(UINT nFlags, short zDelta, CPoint pt);
	afx_msg void OnVScroll(UINT nSBCode, UINT nPos, CScrollBar *pScrollBar);
#endif //_OWNER_DRAWN_TREE
 
	// 共有成员变量
public:
 
	// 共有成员函数
public:
	// 获取背景图片
	BOOL GetBkImage(LVBKIMAGE* plvbkImage) const;
 
	// 设置背景图片
	BOOL SetBkImage(LVBKIMAGE* plvbkImage);
 
#ifdef _OWNER_DRAWN_TREE
	// 设置树控件+/-按钮图标
	void SetTreeBtnBitmap(CBitmap* pBitmap, COLORREF clrMask = RGB(255, 0, 255));
#endif //_OWNER_DRAWN_TREE
 
	// 静态成员变量
public:
 
	// 静态成员函数
public:
 
};

成し遂げる

#include "stdafx.h"
#include <shlwapi.h>
#include "CSWTreeCtrl.h"
 
// IE 5.0 or higher required
#ifndef TVS_NOHSCROLL
#error CSWTreeCtrl requires IE 5.0 or higher; _WIN32_IE should be greater than 0x500.
#endif
 
 
// 绘画树控件横向节点连接虚线
static void _DotHLine(HDC hdc, LONG x, LONG y, LONG w, COLORREF cr)
{
    
    
	for (; w > 0; w -= 2, x += 2)
		SetPixel(hdc, x, y, cr);
}
 
// 绘画树控件竖向节点连接虚线
static void _DotVLine(HDC hdc, LONG x, LONG y, LONG w, COLORREF cr)
{
    
    
	for (; w > 0; w -= 2, y += 2)
		SetPixel(hdc, x, y, cr);
}
 
IMPLEMENT_DYNAMIC(CSWTreeCtrl, CTreeCtrl)
CSWTreeCtrl::CSWTreeCtrl()
{
    
    
	CommonConstruct();
}
 
CSWTreeCtrl::~CSWTreeCtrl()
{
    
    
}
 
BEGIN_MESSAGE_MAP(CSWTreeCtrl, CTreeCtrl)
	ON_WM_MOUSEMOVE()
	ON_WM_LBUTTONDOWN()
	ON_WM_LBUTTONDBLCLK()
	ON_WM_PAINT()
	ON_WM_ERASEBKGND()
	ON_WM_VSCROLL()
	ON_WM_MOUSEWHEEL()
	ON_WM_KEYDOWN()
END_MESSAGE_MAP()
 
void CSWTreeCtrl::CommonConstruct()
{
    
    
#ifdef _OWNER_DRAWN_TREE 
	m_bkImage.hbm = NULL;
	m_bkImage.xOffsetPercent = 0;
	m_bkImage.yOffsetPercent = 0;
#endif //_OWNER_DRAWN_TREE
}
 
#ifdef _OWNER_DRAWN_TREE
// 设置树控件+/-按钮图标
void CSWTreeCtrl::SetTreeBtnBitmap(CBitmap* pBitmap, COLORREF clrMask)
{
    
    
	ASSERT(pBitmap);
	if (pBitmap && pBitmap->GetSafeHandle())
	{
    
    
		m_imgBtns.Add(pBitmap, clrMask);
		BITMAP bmp;
		pBitmap->GetBitmap(&bmp);
		if (m_imgBtns.GetSafeHandle())
			m_imgBtns.DeleteImageList();
		m_imgBtns.Create(bmp.bmWidth / 2, bmp.bmHeight, ILC_COLOR32 | ILC_MASK, 2, 1);
	}
}
#endif //_OWNER_DRAWN_TREE
 
BOOL CSWTreeCtrl::GetBkImage(LVBKIMAGE* plvbkImage) const
{
    
    
	memcpy(plvbkImage, &m_bkImage, sizeof(LVBKIMAGE));
	return TRUE;
}
 
BOOL CSWTreeCtrl::SetBkImage(LVBKIMAGE* plvbkImage)
{
    
    
	memcpy(&m_bkImage, plvbkImage, sizeof(LVBKIMAGE));
	Invalidate();
	return TRUE;
}
 
void CSWTreeCtrl::OnKeyDown(UINT nChar, UINT nRepCnt, UINT nFlags)
{
    
    
	Invalidate();
 
	CTreeCtrl::OnKeyDown(nChar, nRepCnt, nFlags);
}
 
void CSWTreeCtrl::OnLButtonDown(UINT nFlags, CPoint point)
{
    
    
	if (CheckHit(point))
	{
    
    
		HTREEITEM hItem = HitTest(point);
		if (hItem)
		{
    
    
#ifdef _OWNER_DRAWN_TREE
			CRect rcItem, rcClient;
			GetClientRect(&rcClient);
			GetItemRect(hItem, &rcItem, FALSE);
			if (rcItem.bottom > rcClient.bottom)
			{
    
    
				Invalidate();
				EnsureVisible(hItem);
				SelectItem(hItem);
				return;
			}
#endif //_OWNER_DRAWN_TREE
 
			SelectItem(hItem);
		}
 
		CTreeCtrl::OnLButtonDown(nFlags, point);
	}
	else
	{
    
    
		SetFocus();
	}
}
 
void CSWTreeCtrl::OnLButtonDblClk(UINT nFlags, CPoint point)
{
    
    
	if (CheckHit(point))
	{
    
    
		HTREEITEM hItem = HitTest(point);
		if (hItem)
		{
    
    
#ifdef _OWNER_DRAWN_TREE
			CRect rcItem, rcClient;
			GetClientRect(&rcClient);
			GetItemRect(hItem, &rcItem, FALSE);
			if (rcItem.bottom > rcClient.bottom)
			{
    
    
				Invalidate();
				CTreeCtrl::OnLButtonDown(nFlags, point);
				return;
			}
#endif //_OWNER_DRAWN_TREE
 
			SelectItem(hItem);
		}
 
		CTreeCtrl::OnLButtonDblClk(nFlags, point);
	}
	else
	{
    
    
		SetFocus();
	}
}
 
void CSWTreeCtrl::OnMouseMove(UINT nFlags, CPoint point)
{
    
    
	if (CheckHit(point))
	{
    
    
		CTreeCtrl::OnMouseMove(nFlags, point);
	}
}
 
#ifdef _OWNER_DRAWN_TREE
LRESULT CSWTreeCtrl::CustomDrawNotify(LPNMTVCUSTOMDRAW lpNMTVCustomDraw)
{
    
    
	lpNMTVCustomDraw->nmcd.hdr.hwndFrom = GetSafeHwnd();
	lpNMTVCustomDraw->nmcd.hdr.code = NM_CUSTOMDRAW;
	lpNMTVCustomDraw->nmcd.hdr.idFrom = GetWindowLong(m_hWnd, GWL_ID);
	return GetParent()->SendMessage(WM_NOTIFY, (WPARAM)0, (LPARAM)lpNMTVCustomDraw);
}
 
void CSWTreeCtrl::OwnerDrawBackground(CDC* pDC)
{
    
    
	NMTVCUSTOMDRAW stuNMTVCustomDraw;
	DWORD dwFlags;
	DWORD dwRet;
	CRect rcClient;
	GetClientRect(&rcClient);
	memset(&stuNMTVCustomDraw, 0, sizeof(NMTVCUSTOMDRAW));
	stuNMTVCustomDraw.nmcd.dwDrawStage = CDDS_PREPAINT;
	stuNMTVCustomDraw.nmcd.hdc = pDC->m_hDC;
	stuNMTVCustomDraw.nmcd.rc = rcClient;
	dwFlags = (DWORD)CustomDrawNotify(&stuNMTVCustomDraw); // CDDS_PREPAINT
 
	// 填充背景色
	COLORREF crBkgnd = IsWindowEnabled() ? pDC->GetBkColor() : GetSysColor(COLOR_BTNFACE);
	pDC->FillSolidRect(rcClient, crBkgnd);
	if (m_bkImage.hbm && IsWindowEnabled())
	{
    
    
		// 画背景图片
		int xOffset = m_nOffsetX;
		int yOffset = rcClient.left;
		int yHeight = rcClient.Height();
		SCROLLINFO srlInfo;
		if (GetScrollInfo(SB_VERT, &srlInfo, SIF_POS | SIF_RANGE))
		{
    
    
			yOffset = -srlInfo.nPos;
			yHeight = max(srlInfo.nMax + 1, rcClient.Height());
		}
 
		CDC dcMem;
		dcMem.CreateCompatibleDC(pDC);
		BITMAP bmp;
		::GetObject(m_bkImage.hbm, sizeof(BITMAP), &bmp);
		CBitmap* pOldBitmap =
			dcMem.SelectObject(CBitmap::FromHandle(m_bkImage.hbm));
 
		// copy the background bitmap from temporary DC to painting DC
		float x = (float)m_bkImage.xOffsetPercent / 100 * (float)rcClient.Width();
		float y = (float)m_bkImage.yOffsetPercent / 100 * (float)rcClient.Height();
		pDC->BitBlt(/*xOffset*/+(int)x,
			/*yOffset+*/(int)y,
			bmp.bmWidth, bmp.bmHeight, &dcMem, 0, 0, SRCCOPY);
		dcMem.SelectObject(pOldBitmap);
	}
 
	if (dwFlags&CDRF_NOTIFYPOSTERASE)
	{
    
    
		stuNMTVCustomDraw.nmcd.dwDrawStage = CDDS_POSTERASE;
		dwRet = (DWORD)CustomDrawNotify(&stuNMTVCustomDraw); // CDDS_POSTERASE
	}
 
	CFont* pOldFont = pDC->SelectObject(GetFont());
	CImageList* pstateList = GetImageList(TVSIL_STATE);
	CImageList* pimgList = GetImageList(TVSIL_NORMAL);
	CSize szIcon, szStateImg;
	if (pimgList)
	{
    
    
		IMAGEINFO ii;
		if (pimgList->GetImageInfo(0, &ii))
			szIcon = CSize(ii.rcImage.right - ii.rcImage.left, ii.rcImage.bottom - ii.rcImage.top);
	}
 
	if (pstateList)
	{
    
    
		IMAGEINFO ii;
		if (pstateList->GetImageInfo(0, &ii))
			szStateImg = CSize(ii.rcImage.right - ii.rcImage.left, ii.rcImage.bottom - ii.rcImage.top);
	}
 
	// 开始画树节点
	HTREEITEM hItem = GetFirstVisibleItem();
	while (hItem)
	{
    
    
		int nOldBkMode = pDC->SetBkMode(TRANSPARENT);
		DWORD dwStyle = GetStyle();
		DWORD dwState = GetItemState(hItem, 0xFFFF);
		BOOL bEnabled = IsWindowEnabled();
		BOOL bSelected = dwState & TVIS_SELECTED;
		BOOL bHasFocus = (GetFocus() && GetFocus()->m_hWnd == m_hWnd) ? TRUE : FALSE;
 
		// 更新NMCUSTOMDRAW结果内容
		stuNMTVCustomDraw.nmcd.dwItemSpec = (DWORD_PTR)hItem;
		if (bEnabled)
		{
    
    
			if (bHasFocus)
			{
    
    
				stuNMTVCustomDraw.clrTextBk = bSelected ? GetSysColor(COLOR_HIGHLIGHT) : crBkgnd;
				stuNMTVCustomDraw.clrText = ::GetSysColor(bSelected ? COLOR_HIGHLIGHTTEXT : COLOR_MENUTEXT);
				stuNMTVCustomDraw.nmcd.uItemState = dwState | (bSelected ? CDIS_FOCUS : 0);
			}
			else
			{
    
    
				if (GetStyle()&TVS_SHOWSELALWAYS)
				{
    
    
					stuNMTVCustomDraw.clrTextBk = bSelected ? GetSysColor(COLOR_INACTIVEBORDER) : crBkgnd;
					stuNMTVCustomDraw.clrText = ::GetSysColor(COLOR_MENUTEXT);
				}
				else
				{
    
    
					stuNMTVCustomDraw.clrTextBk = crBkgnd;
					stuNMTVCustomDraw.clrText = ::GetSysColor(COLOR_MENUTEXT);
				}
			}
		}
		else
		{
    
    
			stuNMTVCustomDraw.clrTextBk = bSelected ? GetSysColor(COLOR_BTNSHADOW) : crBkgnd;
			stuNMTVCustomDraw.clrText = ::GetSysColor(COLOR_GRAYTEXT);
		}
 
		GetItemRect(hItem, &stuNMTVCustomDraw.nmcd.rc, 0);
		CRgn rgn;
		rgn.CreateRectRgn(stuNMTVCustomDraw.nmcd.rc.left, stuNMTVCustomDraw.nmcd.rc.top, stuNMTVCustomDraw.nmcd.rc.left + m_nFirstColumnWidth, stuNMTVCustomDraw.nmcd.rc.bottom);
		pDC->SelectClipRgn(&rgn);
		dwRet = CDRF_DODEFAULT;
		if (dwFlags&CDRF_NOTIFYITEMDRAW)
		{
    
    
			stuNMTVCustomDraw.nmcd.dwDrawStage = CDDS_ITEMPREPAINT;
			dwRet = (DWORD)CustomDrawNotify(&stuNMTVCustomDraw); // CDDS_ITEMPREPAINT
		}
 
		if (!(dwFlags&CDRF_SKIPDEFAULT))
		{
    
    
			// 画树节点图标和+/-按钮
			CRect rcItem;
			int nImage, nSelImage;
			GetItemRect(hItem, &rcItem, TRUE);
			GetItemImage(hItem, nImage, nSelImage);
			int x = rcItem.left - 3;
			int yCenterItem = rcItem.top + (rcItem.bottom - rcItem.top) / 2;
			if (pimgList)
			{
    
    
				int nImageIndex = dwState & TVIS_SELECTED ? nImage : nSelImage;
				x -= szIcon.cx + 1;
				pimgList->Draw(pDC, nImageIndex, CPoint(x, yCenterItem - szIcon.cy / 2), ILD_TRANSPARENT);
			}
 
			if (GetStyle()&TVS_CHECKBOXES && pstateList != NULL)
			{
    
    
				DWORD dwStateImg = GetItemState(hItem, TVIS_STATEIMAGEMASK) >> 12;
				x -= szStateImg.cx;
				pstateList->Draw(pDC, dwStateImg, CPoint(x, yCenterItem - szStateImg.cy / 2), ILD_TRANSPARENT);
			}
 
			if (dwStyle&TVS_HASLINES)
			{
    
    
				CPen pen;
				pen.CreatePen(PS_DOT, 1, GetLineColor());
				CPen* pOldPen = pDC->SelectObject(&pen);
				HTREEITEM hItemParent = GetParentItem(hItem);
				if (hItemParent != NULL || dwStyle & TVS_LINESATROOT)
				{
    
    
					_DotHLine(pDC->m_hDC, x - szIcon.cx / 2 - 2, yCenterItem, szIcon.cx / 2 + 2, RGB(128, 128, 128));
				}
 
				if (hItemParent != NULL || dwStyle & TVS_LINESATROOT && GetPrevSiblingItem(hItem) != NULL)
				{
    
    
					_DotVLine(pDC->m_hDC, x - szIcon.cx / 2 - 2, rcItem.top,
						yCenterItem - rcItem.top, RGB(128, 128, 128));
				}
 
				if (GetNextSiblingItem(hItem) != NULL)
				{
    
    
					_DotVLine(pDC->m_hDC, x - szIcon.cx / 2 - 2, yCenterItem,
						rcItem.bottom - yCenterItem, RGB(128, 128, 128));
				}
 
				int x1 = x - szIcon.cx / 2 - 2;
				while (hItemParent != NULL)
				{
    
    
					x1 -= szIcon.cx + 3;
					if (GetNextSiblingItem(hItemParent) != NULL)
					{
    
    
						_DotVLine(pDC->m_hDC, x1, rcItem.top, rcItem.Height(), RGB(128, 128, 128));
					}
					hItemParent = GetParentItem(hItemParent);
				}
 
				pDC->SelectObject(pOldPen);
 
			}
 
			if (dwStyle&TVS_HASBUTTONS && ItemHasChildren(hItem))
			{
    
    
				// 画+/-符号按钮
				int nImg = GetItemState(hItem, TVIF_STATE) & TVIS_EXPANDED ? 1 : 0;
				m_imgBtns.Draw(pDC, nImg, CPoint(x - szIcon.cx / 2 - 6, yCenterItem - 4),	ILD_TRANSPARENT);
			}
		}
 
		pDC->SelectClipRgn(NULL);
		if (dwRet&CDRF_NOTIFYPOSTPAINT)
		{
    
    
			stuNMTVCustomDraw.nmcd.dwDrawStage = CDDS_ITEMPOSTPAINT;
			dwRet = (DWORD)CustomDrawNotify(&stuNMTVCustomDraw); // CDDS_ITEMPOSTPAINT
		}
 
		pDC->SetBkMode(nOldBkMode);
		hItem = GetNextVisibleItem(hItem);
	};
 
	pDC->SelectObject(pOldFont);
}
 
BOOL CSWTreeCtrl::OnMouseWheel(UINT nFlags, short zDelta, CPoint pt)
{
    
    
	Invalidate();
	return CTreeCtrl::OnMouseWheel(nFlags, zDelta, pt);
}
 
void CSWTreeCtrl::OnVScroll(UINT nSBCode, UINT nPos, CScrollBar *pScrollBar)
{
    
    
	Invalidate();
	CTreeCtrl::OnVScroll(nSBCode, nPos, pScrollBar);
}
#endif //_OWNER_DRAWN_TREE
 
void CSWTreeCtrl::OnPaint()
{
    
    
	CRect rcClient;
	GetClientRect(&rcClient);
	CPaintDC dc(this);
	CDC dcMem;
	CBitmap bmpMem;
	dcMem.CreateCompatibleDC(&dc);
	if (bmpMem.CreateCompatibleBitmap(&dc, rcClient.Width(), rcClient.Height()))
	{
    
    
		CBitmap* pOldBmp = dcMem.SelectObject(&bmpMem);
#ifdef _OWNER_DRAWN_TREE	// if owner-drawn
		OwnerDrawBackground(&dcMem);	// use our code
#else						// else use standard code
		CWnd::DefWindowProc(WM_PAINT, (WPARAM)dcMem.m_hDC, 0);
#endif //_OWNER_DRAWN_TREE
		dc.BitBlt(0, 0, rcClient.right, rcClient.bottom, &dcMem, 0, 0, SRCCOPY);
		dcMem.SelectObject(pOldBmp);
		bmpMem.DeleteObject();
	}
 
	dcMem.DeleteDC();
}
 
BOOL CSWTreeCtrl::OnEraseBkgnd(CDC* pDC)
{
    
    
	return TRUE;
}
 
BOOL CSWTreeCtrl::CheckHit(CPoint point)
{
    
    
	UINT fFlags;
	HTREEITEM hItem = HitTest(point, &fFlags);
 
	CRect rcItem, rcClient;
	GetClientRect(rcClient);
	GetItemRect(hItem, &rcItem, TRUE);
	if (fFlags & TVHT_ONITEMICON ||
		fFlags & TVHT_ONITEMBUTTON ||
		fFlags & TVHT_ONITEMSTATEICON)
		return TRUE;
 
	if (GetStyle()&TVS_FULLROWSELECT)
	{
    
    
		rcItem.right = rcClient.right;
		if (rcItem.PtInRect(point))
			return TRUE;
 
		return FALSE;
	}
 
	if (fFlags & TVHT_ONITEMLABEL)
	{
    
    
		rcItem.right = m_nFirstColumnWidth;
		if (!rcItem.PtInRect(point))
			return FALSE;
 
		CString strSub;
		AfxExtractSubString(strSub, GetItemText(hItem), 0, '\t');
		CDC* pDC = GetDC();
		pDC->SelectObject(GetFont());
		rcItem.right = rcItem.left + pDC->GetTextExtent(strSub).cx + 6;
		ReleaseDC(pDC);
		if (!rcItem.PtInRect(point))
			return FALSE;
 
		return TRUE;
	}
 
	return FALSE;
}

ノードに関連付けられたデータ値に従って、再帰アルゴリズムを使用して指定されたノードが検索されます。

HTREEITEM findTreeItem(CTreeCtrl& treeCtrl, HTREEITEM hItem, DWORD dwData) {
    
    
	if (treeCtrl.GetItemData(hItem) == dwData)
		return hItem;
 
	if (treeCtrl.ItemHasChildren(hItem)) {
    
    
		HTREEITEM hChild = treeCtrl.GetChildItem(hItem);
		HTREEITEM hFindItem = findTreeItem(treeCtrl, hChild, dwData);
		if (hFindItem != NULL)
			return hFindItem;
	}
 
	HTREEITEM hNext = treeCtrl.GetNextSiblingItem(hItem);
	if (hNext) {
    
    
		HTREEITEM hFindItem = findTreeItem(treeCtrl, hNext, dwData);
		if (hFindItem != NULL)
			return hFindItem;
		else
			return NULL;
	}
	else {
    
    
		return NULL;
	}
};

おすすめ

転載: blog.csdn.net/vv1025/article/details/132818896
おすすめ
「世界のサイバースペースの安全保障と開発に対する米国の脅威と被害」報告書が発表
ランキング
vue + swiper の正しい使用姿勢 (プラグアンドプレイ CSS、動的更新データ、使用方法の提案付き)
ダウンロードapache-maven-3.3.9-bin.zipにMavenの公式サイト
C言語を練習するための演習
ユニティ喜びスフィア
C语言如何用十六进制数按二进制输出
Discussion sur la sécurité des threads de StringBuilder
アニメーション：インタビューが大きいノードKの二分木検索を磨くであろう
Gitの理論的基礎
C＃のどのケース変換
javaコマンドRedisのセットの動作例（Jedis）
アーカイブ
もっと
2024-04-29(5)
2024-04-28(12)
2024-04-27(29)
2024-04-26(22)
2024-04-25(31)
2024-04-24(30)
2024-04-23(31)
2024-04-22(5)
2024-04-21(0)
2024-04-20(6)

コードワールド

© 2018 codetd.com