文件视图中的CTreeCtrl按要求改变颜色

最近开发一个测试工具，各种测试内容在左侧的CTreeCtrl显示由MFC的MDI提供

需求如下图：

 网上有不少网友提供了通过改写CViewTree类的响应NM_CUSTOMDRAW消息，自绘CTreeCtrl控件。但是在操作中还是遇到一些问题现总结如下：

void CViewTree::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
{
    //    LPNMCUSTOMDRAW pNMCD = reinterpret_cast<LPNMCUSTOMDRAW>(pNMHDR);
        // TODO: 在此添加控件通知处理程序代码
    *pResult = 0;
    NMTVCUSTOMDRAW* pNMCD = (NMTVCUSTOMDRAW*)(pNMHDR);
    if (CDDS_PREPAINT == pNMCD->nmcd.dwDrawStage)
    {
        *pResult = CDRF_NOTIFYITEMDRAW;
    }
    else if ((CDDS_ITEMPREPAINT) == pNMCD->nmcd.dwDrawStage)
    {
        COLORREF clrOldColor;
        COLORREF clrLoseColor, clrSuccessColor;
        clrOldColor = pNMCD->clrText;//原始 正常
        clrLoseColor = RGB(255, 0, 0);//红色失败
        clrSuccessColor = RGB(0, 255, 0);//绿色 通过
 //        clrLostNewColor = RGB(0, 0, 255);//蓝色 黄色255，255，0
//        pNMCD->iLevel  分层显示        
        if (m_HostRaw_Count)
        {
            m_HOSTRAW_t_draw = m_HOSTRAW_h;
            while (m_HOSTRAW_t_draw != NULL)
            {
                HTREEITEM hot = (HTREEITEM)pNMCD->nmcd.dwItemSpec;
                if (m_HOSTRAW_t_draw->m_hot == hot)
                {
                    switch (m_HOSTRAW_t_draw->m_State)
                    {
                    case 0:
                    {
                        pNMCD->clrText = clrOldColor;
                    }
                    break;
                    case 1:
                    {
                        pNMCD->clrText = clrSuccessColor;
                    }
                    break;
                    case 2:
                    {
                        pNMCD->clrText = clrLoseColor;
                    }
                    break;
                    default:
                        pNMCD->clrText = clrOldColor;
                        break;
                    }
                }
                m_HOSTRAW_t_draw = m_HOSTRAW_t_draw->next;
            }
        }
//        else
        {
            if (pNMCD->iLevel == 0)
                pNMCD->clrText = clrOldColor;
            if (pNMCD->iLevel == 1)
                pNMCD->clrText = clrOldColor;

        }
    }
}

NMTVCUSTOMDRAW代表的是树的结构体。

因在变色是只是需要将某一个变色，而不是将所有的都变色所以在文件视图生成初期分别记录每一项的编号。本文中采取的是结构体的方式。定义如下：

struct HOSTRAW {//目录结构体
        HTREEITEM m_hot;
        int m_State;//状态0是正常1是成功2是失败
        int m_RootOrChild_Judge;//根结点和子结点的标识0是总根结点1是根结点2是子结点
        HOSTRAW *next;
    }*m_HOSTRAW_h, *m_HOSTRAW_t, *m_HOSTRAW_o, *m_HOSTRAW_t_draw;//
    int m_HostRaw_Count;//目录数量

在每次的重画时轮巡结构体方便确定颜色。后期要改色也只需要改变结构体的状态标识。

详细代码：文件视图中动态改变CTreeCtrl的节点颜色-C++文档类资源-CSDN下载