Report control - with category

// MyReportCtrl.cpp : implementation file
//
#include "stdafx.h"
#include "reportdemo.h"
#include "MyReportCtrl.h"

#ifdef _DEBUG
    #define new DEBUG_NEW
    #undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

// CMyReportCtrl
CMyReportCtrl::CMyReportCtrl()
{
}

CMyReportCtrl::~CMyReportCtrl()
{
}

BEGIN_MESSAGE_MAP(CMyReportCtrl, CReportCtrl)
//{{AFX_MSG_MAP(CMyReportCtrl)
// NOTE - the ClassWizard will add and remove mapping macros here.
//}}AFX_MSG_MAP
END_MESSAGE_MAP()

// CMyReportCtrl implementation
INT CMyReportCtrl::DrawText(CDC *pDC, CRect rect, LPRVITEM lprvi)
{
    COLORREF    textColor = pDC->GetTextColor();
    INT         result;
    if (lprvi->nMask & RVIM_CATEGORY)
    {
        pDC->SetTextColor(RGB(5, 193, 254));
    }
    else if (lprvi->iItem == 0)
    {
        pDC->SetTextColor(0x000000FF);
    }

    result = CReportCtrl::DrawText(pDC, rect, lprvi);
    pDC->SetTextColor(textColor);
    return result;
}

void CMyReportCtrl::GetCategoryMenuItem(LPCATITEM lpCatItem)
{
    CReportCtrl::GetCategoryMenuItem(lpCatItem);

    switch (lpCatItem->iSubItem)
    {
    case 0:
        if (GetStyle() & RVS_TREEVIEW)
            lpCatItem->mask = 0;
        break;
    case 2:
        if (!(GetStyle() & RVS_TREEVIEW))
        {
            lpCatItem->iImage = 5;
            lpCatItem->mask |= CATEGORY_IMAGE;
        }
        break;
    case 3:
        if (GetStyle() & RVS_TREEVIEW)
        {
            lpCatItem->iImage = 5;
            lpCatItem->mask |= CATEGORY_IMAGE;
        }
        break;
    case -1:
        lpCatItem->iImage = 7;
        lpCatItem->mask |= CATEGORY_IMAGE;
        break;
    }
}

// CMyReportCtrl message handlers