【MFC】控件自绘之CTreeCtrl

一：

     树形控件是非常常用的控件之一，可能有些刚学自绘的同学会觉得树形控件的自绘很麻烦。其实不然，请看代码。

 

二：

.h头文件

class CResourceTreeCtrl : public CTreeCtrl
{
	DECLARE_DYNAMIC(CResourceTreeCtrl)

public:
	CResourceTreeCtrl();
	virtual ~CResourceTreeCtrl();

	void SetExpandImage(CImageList* pImageList);

protected:
	DECLARE_MESSAGE_MAP()

	afx_msg void OnPaint();
	afx_msg void OnMouseMove(UINT nFlags, CPoint point);
	afx_msg void OnLButtonDown(UINT nFlags, CPoint point);
	afx_msg void OnMouseLeave();
	afx_msg BOOL OnEraseBkgnd(CDC* pDC);

protected:

	virtual void PreSubclassWindow();

private:

	int  GetExpandImageIndex(HTREEITEM hItem, CRect& rectExpand);
	
	void ComputeExpandRect(HTREEITEM hItem, CRect& rectExpand);
	void ComputeImageRect(HTREEITEM hItem, CRect& rectImage);

private:

	void DrawTree(CDC* pDC);
	void DrawItem(CDC* pDC, HTREEITEM hItem);
	void DrawExpand(CDC* pDC, HTREEITEM hItem);
	void DrawImage(CDC* pDC, HTREEITEM hItem);
	void DrawText(CDC* pDC, HTREEITEM hItem);
	void DrawBackGnd(CDC* pDC, HTREEITEM hItem);

private:

	BOOL          m_bMouseOut;                            //鼠标是否出去
	HTREEITEM     m_hCurItem;                             //当前选中项
	HTREEITEM     m_hHoverItem;                           //当前悬停项
	CImageList*   m_pExpandImage;
};

cpp源文件

// CResourceTreeCtrl

IMPLEMENT_DYNAMIC(CResourceTreeCtrl, CTreeCtrl)

CResourceTreeCtrl::CResourceTreeCtrl()
{
	m_bMouseOut = TRUE;
	m_hCurItem  = NULL;
	m_hHoverItem = NULL;
}

CResourceTreeCtrl::~CResourceTreeCtrl()
{
}


BEGIN_MESSAGE_MAP(CResourceTreeCtrl, CTreeCtrl)
	ON_WM_MOUSEMOVE()
	ON_WM_MOUSELEAVE()
	ON_WM_PAINT()
	ON_WM_ERASEBKGND()
	ON_WM_LBUTTONDOWN()
END_MESSAGE_MAP()



// CResourceTreeCtrl message handlers

BOOL CResourceTreeCtrl::OnEraseBkgnd(CDC* pDC)
{
	return TRUE;
}

void CResourceTreeCtrl::OnMouseMove(UINT nFlags, CPoint point)
{
	if(m_bMouseOut)
	{
		TRACKMOUSEEVENT tm;

		tm.cbSize = sizeof(TRACKMOUSEEVENT);
		tm.dwFlags = TME_LEAVE;   
		tm.dwHoverTime = 500;  
		tm.hwndTrack = m_hWnd;

		_TrackMouseEvent(&tm);	
		
		m_bMouseOut = FALSE;
	}

	HTREEITEM hMoveItem = HitTest(point);

	if(hMoveItem != m_hHoverItem)
	{
		m_hHoverItem = hMoveItem;

		Invalidate();
	}

	CTreeCtrl::OnMouseMove(nFlags, point);
}

void CResourceTreeCtrl::OnMouseLeave()
{
	if(!m_bMouseOut)
	{
		m_bMouseOut = TRUE;

		HTREEITEM hItem = m_hHoverItem;
		
		m_hHoverItem =  NULL;

		Invalidate();
	}
}

void CResourceTreeCtrl::OnLButtonDown(UINT nFlags, CPoint point)
{
	HTREEITEM hCurItem = HitTest(point);
	
	if(hCurItem)
	{
		SelectItem(hCurItem);

		SetItemState(hCurItem, TVIS_SELECTED, TVIS_SELECTED);

		m_hCurItem = hCurItem;

		Invalidate();
	}

	CTreeCtrl::OnLButtonDown(nFlags, point);
}

void CResourceTreeCtrl::OnPaint()
{
	CPaintDC dc(this);

	CMemDCC  DCMemory(&dc);
	CRect    rect;
	GetClientRect(&rect);

	//////
	DrawTree(&DCMemory);
}

void CResourceTreeCtrl::PreSubclassWindow()
{
	CTreeCtrl::PreSubclassWindow();
}

void CResourceTreeCtrl::DrawTree(CDC* pDC)
{
	/////
	CRect rect;
	COLORREF bkColor;

	CECMSVisualManager::GetInstance()->GetBKColor(bkColor);
	GetClientRect(&rect);

	Graphics graph(pDC->m_hDC);

	graph.FillRectangle(&SolidBrush(Color(255, GetRValue(bkColor), GetGValue(bkColor), GetBValue(bkColor))),\
		RectF(rect.left, rect.top, rect.Width(), rect.Height()));

	/////

	HTREEITEM hItem;

	hItem = GetFirstVisibleItem();

	while(hItem != NULL)
	{
		DrawItem(pDC, hItem);

		hItem = GetNextVisibleItem(hItem);
	}
}

void CResourceTreeCtrl::DrawItem(CDC* pDC, HTREEITEM hItem)
{
	///先画背景，再画展开,图像，文字
	DrawBackGnd(pDC, hItem);
	DrawExpand(pDC, hItem);
	DrawImage(pDC, hItem);
	DrawText(pDC, hItem);
}

void CResourceTreeCtrl::DrawBackGnd(CDC* pDC, HTREEITEM hItem)
{
	Graphics        graph(pDC->m_hDC);

	graph.SetSmoothingMode(SmoothingModeAntiAlias);
	graph.SetInterpolationMode(InterpolationModeHighQualityBicubic);

	CRect           rectItem;

	GetItemRect(hItem, &rectItem, FALSE);

	RectF rectF(rectItem.left, rectItem.top, rectItem.Width(), rectItem.Height());

	////////////////////////选中
	if(m_hCurItem == hItem)
	{
		COLORREF from, to;
		CECMSVisualManager::GetInstance()->GetSelectGradientColor(from, to);
		LinearGradientBrush lineGrBrush(
			rectF,
			Color(255, GetRValue(from), GetGValue(from), GetBValue(from)),//Color(255, 28, 28, 28),
			Color(255, GetRValue(to), GetGValue(to), GetBValue(to)),//Color(255, 172, 172, 172),
			LinearGradientModeHorizontal
			);

		lineGrBrush.SetBlendTriangularShape(0.5f, 1.0f);

		graph.FillRectangle(&lineGrBrush, rectF);

	} //////////////////////悬停
	else if(m_hHoverItem == hItem)
	{	
		/*GraphicsPath      path;

		path.AddRectangle(rectF);

		PathGradientBrush pathGrBrush(&path);

		int nCount = 1;
		pathGrBrush.SetCenterColor(Color(m_nA, m_nR, m_nG, m_nB));
		pathGrBrush.SetSurroundColors(&Color(100, m_nR, m_nG, m_nB), &nCount);

		graph.FillRectangle(&pathGrBrush, rectF);*/
		COLORREF from, to;
		CECMSVisualManager::GetInstance()->GetSelectGradientColor(from, to);
		LinearGradientBrush lineGrBrush(
			rectF,
			Color(100, GetRValue(from), GetGValue(from), GetBValue(from)),//Color(100, 28, 28, 28),
			Color(100, GetRValue(to), GetGValue(to), GetBValue(to)),//Color(100, 172, 172, 172),
			LinearGradientModeHorizontal
			);

		lineGrBrush.SetBlendTriangularShape(0.5f, 1.0f);
		lineGrBrush.SetGammaCorrection(TRUE);

		graph.FillRectangle(&lineGrBrush, rectF);
	}
}

void CResourceTreeCtrl::DrawExpand(CDC* pDC, HTREEITEM hItem)
{
	BOOL        bHasChild   = ItemHasChildren(hItem);

	if(bHasChild && m_pExpandImage->GetSafeHandle())
	{	
		int nImage;

		CRect rectExpand;

		ComputeExpandRect(hItem, rectExpand);

		nImage = GetExpandImageIndex(hItem, rectExpand);

		m_pExpandImage->Draw(pDC, nImage, rectExpand.TopLeft(), ILD_TRANSPARENT);
		
	}
}

void CResourceTreeCtrl::DrawImage(CDC* pDC, HTREEITEM hItem)
{
	CImageList*  pItemImage  = GetImageList(TVSIL_NORMAL);
	CImageList*  pStateImage = GetImageList(TVSIL_STATE);
	DWORD        itemState   = GetItemState(hItem, TVIF_STATE);

	int nImage, nSelectImage;

	VERIFY(GetItemImage(hItem, nImage, nSelectImage));

	CImageList*       pNormalImage = GetImageList(TVSIL_NORMAL);
	
	if(pNormalImage)
	{
		CRect       rectImage;

		ComputeImageRect(hItem, rectImage);

		pNormalImage->Draw(pDC, nImage, rectImage.TopLeft(), ILD_TRANSPARENT);
	}
}

void CResourceTreeCtrl::DrawText(CDC* pDC, HTREEITEM hItem)
{
	CString     strItemText = GetItemText(hItem);
	CRect       rectText;

	GetItemRect(hItem, &rectText, TRUE);

	RectF       rectF(rectText.left, rectText.top, rectText.Width(), rectText.Height());

	Graphics    graph(pDC->m_hDC);
	CFont*      pFont = GetFont();
	LOGFONT     LogFont;

	if(pFont)
	{
		pFont->GetLogFont(&LogFont);
	}
	else
	{
		ZeroMemory(&LogFont, sizeof(LOGFONT));
		lstrcpy(LogFont.lfFaceName, _T("微软雅黑"));
	}

	Gdiplus::Font       font(LogFont.lfFaceName, 10, FontStyleRegular, UnitPoint);
	Gdiplus::StringFormat stringFormat;
	stringFormat.SetAlignment(StringAlignmentNear);
	stringFormat.SetLineAlignment(StringAlignmentCenter);
	stringFormat.SetFormatFlags(StringFormatFlagsLineLimit );

	RectF               boundBox;

	graph.MeasureString(strItemText, strItemText.GetLength(), &font, PointF(0, 0), &stringFormat, &boundBox);

	rectF.X = rectText.left;
	rectF.Y = rectText.top;
	rectF.Width = boundBox.Width;
	rectF.Height = boundBox.Height;

	COLORREF texColor;
	CECMSVisualManager::GetInstance()->GetTextColor(texColor);

	graph.DrawString(strItemText, strItemText.GetLength(), &font, rectF, 
		&stringFormat, &SolidBrush(Color(255, GetRValue(texColor), GetGValue(texColor), GetBValue(texColor))));
}

void CResourceTreeCtrl::ComputeExpandRect(HTREEITEM hItem, CRect& rectExpand)
{
	ASSERT(m_pExpandImage->GetSafeHandle());

	CRect        rectText;
	CRect        rectItem;
	CRect        rectImage;

	GetItemRect(hItem, &rectText, TRUE);
	GetItemRect(hItem, &rectItem, FALSE);

	ComputeImageRect(hItem, rectImage);

	rectExpand = rectImage;

	IMAGEINFO   imageInfo = {0};
	int         nCount    = 0;

	m_pExpandImage->GetImageInfo(0, &imageInfo);
	nCount = m_pExpandImage->GetImageCount();

	int nExpandImageWidth = CRect(imageInfo.rcImage).Width();
	int nExpandImageHeight = CRect(imageInfo.rcImage).Height();

	rectExpand.left -= 3;
	rectExpand.left -= nExpandImageWidth;
	rectExpand.top  = rectItem.top + (rectItem.Height() - nExpandImageHeight)/2;
	rectExpand.bottom = rectExpand.top + nExpandImageHeight;
	rectExpand.right = rectExpand.left + nExpandImageWidth;
}

void CResourceTreeCtrl::ComputeImageRect(HTREEITEM hItem, CRect& rectImage)
{
	CRect        rectText;
	CRect        rectItem;

	GetItemRect(hItem, &rectText, TRUE);
	GetItemRect(hItem, &rectItem, FALSE);

	CImageList*   pNormalImage = GetImageList(TVSIL_NORMAL);
	//	CImageList*   pStateImage = GetImageList(TVSIL_STATE);

	rectImage = rectText;
	rectImage.left -= 3;

	if(pNormalImage)
	{
		int nImage = 0;
		int nSelectedImage = 0;

		if(GetItemImage(hItem, nImage, nSelectedImage))
		{
			IMAGEINFO   imageInfo = {0};
			int         nCount    = 0;

			pNormalImage->GetImageInfo(0, &imageInfo);
			nCount = pNormalImage->GetImageCount();

			int nItemImageWidth = CRect(imageInfo.rcImage).Width();
			int nItemImageHeight = CRect(imageInfo.rcImage).Height();

			rectImage.left -= nItemImageWidth;
			rectImage.right = rectImage.left + nItemImageWidth;
		}
	}
}

int CResourceTreeCtrl::GetExpandImageIndex(HTREEITEM hItem, CRect& rectExpand)
{
	int          nImage = 0;
	CPoint       ptCursor;

	GetCursorPos(&ptCursor);
	ScreenToClient(&ptCursor);

	DWORD dwState = GetItemState(hItem, TVIF_STATE);

	if(rectExpand.PtInRect(ptCursor))
	{
		if(dwState & TVIS_EXPANDED)
		{
			nImage = 2;
		}
		else
		{
			nImage = 1;
		}
	}
	else
	{
		if(dwState & TVIS_EXPANDED)
		{
			nImage = 3;
		}
		else
		{
			nImage = 0;
		}
	}

	return nImage;
}

void CResourceTreeCtrl::SetExpandImage(CImageList* pImageList)
{
	ASSERT(pImageList != NULL);

	m_pExpandImage = pImageList;
}