在VC++中改变CTreeCtrl子项Item字体和颜色

在VC++中改变CTreeCtrl子项Item字体和颜色

定义一个继承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 );

}