TnbMfcAbstractCustomDraw.h

[詳解]

#pragma once
#include "TnbMfcCommon.h"
 
 
 
//TNB Library
namespace TNB {
namespace MFC {
 
 
 
class CAbstractCustomDraw
{
protected:
 
    struct TDrawParam
    {
        HDC         dc;         
        CRect       rect;       
        DWORD_PTR   itemNo;     
        DWORD       subItemNo;  
        UINT        state;  
        COLORREF    textColor;  
        COLORREF    backColor;  
    };
 
    virtual void OnBeginItemPaint(HDC dc) = 0;
 
    virtual void OnEndItemPaint(HDC dc) = 0;
 
    virtual LRESULT OnItemPainting(TDrawParam* pParam) = 0;
 
    virtual void OnItemPainted(const TDrawParam* pParam) = 0;
 
    virtual bool HasSubItem(void) const = 0;
 
    virtual void OnCalcItemRect(TDrawParam& _pa, bool isSub) = 0;
 
    BOOL OnChildNotify(UINT message, WPARAM wParam, LPARAM lParam, LRESULT* _pResult)
    {
        if ( message == WM_NOTIFY )
        {
            LPNMHDR lpNmHdr = reinterpret_cast<LPNMHDR>(lParam);
            if ( lpNmHdr->code == NM_CUSTOMDRAW ) 
            {
                LPNMLVCUSTOMDRAW lpLvCd = reinterpret_cast<LPNMLVCUSTOMDRAW>(lParam);
                *_pResult = m_OnCustomDraw(lpLvCd);
                return TRUE;
            }
        } 
        return FALSE;
    }
 
private:
 
    LRESULT m_OnItemPaintSub(LPNMLVCUSTOMDRAW P, int subItemNo)
    {
        TDrawParam pa;
        pa.dc           = P->nmcd.hdc;
        pa.itemNo       = P->nmcd.dwItemSpec;
        pa.subItemNo    = subItemNo;
        pa.state        = P->nmcd.uItemState;
        pa.textColor    = P->clrText;
        pa.backColor    = P->clrTextBk;
        pa.rect         = P->nmcd.rc;
        OnCalcItemRect(pa, true);
        LRESULT r = OnItemPainting(&pa);
        P->clrText = pa.textColor;
        P->clrTextBk = pa.backColor;
        return r;
    }
 
    LRESULT m_OnCustomDraw(LPNMLVCUSTOMDRAW P)
    {
        switch ( P->nmcd.dwDrawStage )
        {
        case CDDS_PREPAINT:
            OnBeginItemPaint(P->nmcd.hdc);
            return CDRF_NOTIFYITEMDRAW | CDRF_NOTIFYPOSTPAINT;
        case CDDS_POSTPAINT:
            OnEndItemPaint(P->nmcd.hdc) ;
            return CDRF_DODEFAULT;
        case CDDS_ITEMPOSTPAINT:
            {
                TDrawParam pa;
                pa.dc           = P->nmcd.hdc;
                pa.itemNo       = P->nmcd.dwItemSpec;
                pa.subItemNo    = 0;
                pa.state        = P->nmcd.uItemState;
                pa.textColor    = P->clrText;
                pa.backColor    = P->clrTextBk;
                pa.rect         = P->nmcd.rc;
                OnCalcItemRect(pa, false);
                OnItemPainted(&pa);
            }
            break;
        case CDDS_ITEMPREPAINT:
            if ( ! HasSubItem() )
            {
                return m_OnItemPaintSub(P, 0) | CDRF_NOTIFYPOSTPAINT;
            }
            return CDRF_NOTIFYSUBITEMDRAW | CDRF_NOTIFYPOSTPAINT;
        case CDDS_ITEMPREPAINT | CDDS_SUBITEM:
            return m_OnItemPaintSub(P, P->iSubItem);
        }
        return CDRF_DODEFAULT;
    }
 
public:
 
    CAbstractCustomDraw(void)
    {
    }
};
 
 
 
}; // MFC
}; // TNB