VS File Finder

// ColourListCtrl.cpp : implementation file

#include "stdafx.h"
#include "ColourListCtrl.h"
#include "VSFileFinderCtrl.h"

ColourListCtrl::ColourListCtrl()
	: m_bHideText(false)
{
}

ColourListCtrl::~ColourListCtrl()
{
}

void ColourListCtrl::SetHideText(bool b)
{
	m_bHideText = b;
}

BEGIN_MESSAGE_MAP(ColourListCtrl, CListCtrl)
    ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, OnCustomDraw)
END_MESSAGE_MAP()

void ColourListCtrl::OnCustomDraw(NMHDR* pNMHDR, LRESULT* pResult)
{
    NMLVCUSTOMDRAW* pCustomDraw = reinterpret_cast<NMLVCUSTOMDRAW*>(pNMHDR);
    *pResult = OnCustomDraw(pCustomDraw);
}

LRESULT ColourListCtrl::OnCustomDraw(NMLVCUSTOMDRAW* pCustomDraw)
{
	DWORD dwStyle = GetStyle() & LVS_TYPEMASK;

    if (dwStyle == LVS_REPORT || dwStyle == LVS_LIST)
    {
        switch (pCustomDraw->nmcd.dwDrawStage)
        {
        case CDDS_PREPAINT:
            return CDRF_NOTIFYITEMDRAW;

        case CDDS_ITEMPREPAINT:
			{
				COLORREF c = GetItemColour(pCustomDraw);
				if (c == 0)
					return CDRF_DODEFAULT;
				if (m_bHideText)
					pCustomDraw->clrText = c;
				pCustomDraw->clrTextBk = c;
				return CDRF_NEWFONT;
			}
        }
    }

    return CDRF_DODEFAULT;
}

COLORREF ColourListCtrl::GetItemColour(NMLVCUSTOMDRAW* pCustomDraw) const
{
	return static_cast<COLORREF>(pCustomDraw->nmcd.lItemlParam);
}

FileFinderListCtrl::FileFinderListCtrl(const CVSFileFinderCtrl* pOwner)
	: m_pOwner(pOwner)
{
}

COLORREF FileFinderListCtrl::GetItemColour(NMLVCUSTOMDRAW* pCustomDraw) const
{
	return m_pOwner->GetColour(static_cast<int>(pCustomDraw->nmcd.dwItemSpec));
}