在VC++中改变树状控件子项Item字体和颜色

在VC++中改变树状控件子项Item字体和颜色
Setting color and font attribute of the item for CTreeCtrl in VC++.

翻译Zafir Anjum 的文章Setting color and font attribute for individual items
原文http://www.codeguru.com/Cpp/controls/treeview/misc-advanced/article.php/c633

通过定义一个继承CTreeCtrl的类来实现改变子项字体和颜色：
首先定义一个字体、颜色的结构体：
protected:
struct Color_Font
{
COLORREF color;
LOGFONT logfont;
};
使用CMap来保存各个Item对应的字体、颜色信息
CMap< void*, void*, Color_Font, Color_Font& > m_mapColorFont ;

添加设置字体、颜色、粗体的函数
void CTreeCtrlX::SetItemFont(HTREEITEM hItem, LOGFONT& logfont)
{
Color_Font cf;
if( !m_mapColorFont.Lookup( hItem, cf ) )
cf.color = (COLORREF)-1;
cf.logfont = logfont;
m_mapColorFont[hItem] = cf;
}

void CTreeCtrlX::SetItemBold(HTREEITEM hItem, BOOL bBold)
{
SetItemState( hItem, bBold ? TVIS_BOLD: 0, TVIS_BOLD );
}

void CTreeCtrlX::SetItemColor(HTREEITEM hItem, COLORREF color)
{
Color_Font cf;
if( !m_mapColorFont.Lookup( hItem, cf ) )
cf.logfont.lfFaceName[0] = '/0';
cf.color = color;
m_mapColorFont[hItem] = cf;
}

BOOL CTreeCtrlX::GetItemFont(HTREEITEM hItem, LOGFONT * plogfont)
{
Color_Font cf;
if( !m_mapColorFont.Lookup( hItem, cf ) )
return FALSE;
if( cf.logfont.lfFaceName[0] == '/0' )
return FALSE;
*plogfont = cf.logfont;
return TRUE;

}

BOOL CTreeCtrlX::GetItemBold(HTREEITEM hItem)
{
return GetItemState( hItem, TVIS_BOLD ) & TVIS_BOLD;
}

COLORREF CTreeCtrlX::GetItemColor(HTREEITEM hItem)
{
// Returns (COLORREF)-1 if color was not set
Color_Font cf;
if( !m_mapColorFont.Lookup( hItem, cf ) )
return (COLORREF)-1;
return cf.color;

}

添加WM_PAINT 信息监控，重载OnPaint()函数来实现绘制

afx_msg void CMyTreeCtrl::OnPaint()
{
//获取当前绘制对象的DC
CPaintDC dc(this);

// 使用将要绘制的对象的DC创建一个memory DC
//memory device context的概念：是在内存中创建一个结构来反映一个显示（屏幕区域、窗口、
//打印机等）的表面。可以用来先在内存中准备好要显示的图像，从而实现双缓存，提高刷新
//速度减少刷新时产生的闪烁。
CDC memDC;
//从当前DC创建内存对象
memDC.CreateCompatibleDC( &dc );

//定义CRect对象，用来确定区域
CRect rcClip, rcClient;
//获取当前对象的边界区域
dc.GetClipBox( &rcClip );
//获取当前对象的用户区域
GetClientRect(&rcClient);

// Select a compatible bitmap into the memory DC
//创建一个bmp文件，作为memDC的内容
//该文件的大小与用于区域相同
CBitmap bitmap;
bitmap.CreateCompatibleBitmap( &dc, rcClient.Width(), rcClient.Height() );
memDC.SelectObject( &bitmap );

// Set clip region to be same as that in paint DC
//通过对象的边界区域创建CRgn对象
CRgn rgn;
rgn.CreateRectRgnIndirect( &rcClip );
memDC.SelectClipRgn(&rgn);
rgn.DeleteObject();

// First let the control do its default drawing.
//首先让控件自己进行默认的绘制，绘制到内存中
CWnd::DefWindowProc( WM_PAINT, (WPARAM)memDC.m_hDC, 0 );

//获取树状控件的第一个节点
HTREEITEM hItem = GetFirstVisibleItem();

//遍历这棵树
int n = GetVisibleCount()+1;
while( hItem && n--)
{
CRect rect;

// Do not meddle with selected items or drop highlighted items
//不对选中的节点和实行拖放功能的节点进行操作
UINT selflag = TVIS_DROPHILITED | TVIS_SELECTED;

//定义字体、颜色
Color_Font cf;

//设置字体
if ( !(GetItemState( hItem, selflag ) & selflag )
&& m_mapColorFont.Lookup( hItem, cf ))
{
CFont *pFontDC;
CFont fontDC;
LOGFONT logfont;

if( cf.logfont.lfFaceName[0] != '/0' )
{
//用户定义了字体
logfont = cf.logfont;
}
else
{
// 用户没有定义，使用系统字体
CFont *pFont = GetFont();
pFont->GetLogFont( &logfont );
}

//用户是否设定节点为加粗
if( GetItemBold( hItem ) )
logfont.lfWeight = 700;
//创建字体
fontDC.CreateFontIndirect( &logfont );
pFontDC = memDC.SelectObject( &fontDC );

//设置字体颜色
if( cf.color != (COLORREF)-1 )
memDC.SetTextColor( cf.color );

//获取节点文字
CString sItem = GetItemText( hItem );

//获取节点区域
GetItemRect( hItem, &rect, TRUE );
//设置背景色为系统色
memDC.SetBkColor( GetSysColor( COLOR_WINDOW ) );
//向内存中的图片写入内容,为该节点的内容
memDC.TextOut( rect.left+2, rect.top+1, sItem );

memDC.SelectObject( pFontDC );
}
hItem = GetNextVisibleItem( hItem );
}

dc.BitBlt( rcClip.left, rcClip.top, rcClip.Width(), rcClip.Height(), &memDC,
rcClip.left, rcClip.top, SRCCOPY );

}