MFC 自定义CListCtrl

全栈程序员站长

发布于 2022-09-09 21:00:24

6070

发布于 2022-09-09 21:00:24

文章被收录于专栏：全栈程序员必看全栈程序员必看

大家好，又见面了，我是你们的朋友全栈君。

本文的代码也是根据网上现有的改编而来

首先是.h

#pragma once
#include "HeaderCtrlCl.h"
// CListCtrlCl

class CListCtrlCl : public CListCtrl
{
    DECLARE_DYNAMIC(CListCtrlCl)

public:
    CHeaderCtrlCl m_Header;
    CListCtrlCl();
    virtual ~CListCtrlCl();

protected:
    DECLARE_MESSAGE_MAP()
    virtual void PreSubclassWindow();
public:
    virtual void DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct);
public:
    afx_msg void OnMeasureItem(int nIDCtl, LPMEASUREITEMSTRUCT lpMeasureItemStruct);
    void MeasureItem(LPMEASUREITEMSTRUCT lpMeasureItemStruct);
    afx_msg void OnLButtonDown(UINT nFlags, CPoint point){}
    afx_msg void OnRButtonDown(UINT nFlags, CPoint point){}
public:
    // 行高
    int m_nRowHeight;
    int InsertColumn(int nCol, LPCTSTR lpszColumnHeading,
        int nFormat = LVCFMT_LEFT, int nWidth = -1, int nSubItem = -1);
public:
    // Gradient - 渐变系数，立体背景用,不用渐变设为0
    void SetHeaderBKColor(int R, int G, int B, int Gradient);
public:
    // 设置表头高度
    void SetHeaderHeight(float Height);
    CPtrList m_ptrListCol;  //保存列颜色
    CPtrList m_ptrListItem; //保存Item颜色表
    CPtrList m_colTextColor; //保存列字体颜色
    CPtrList m_ItemTextColor; //保存单元格字体颜色
    CPtrList m_ItemFontSize;//保存单元格字体

    bool FindColColor(int col ,COLORREF &color); //查找列颜色
    bool FindItemColor(int col,int row,COLORREF &color);
    bool FindColTextColor(int col,COLORREF &color); //查找列字体颜色
    bool FindItemTextColor(int col,int row,COLORREF &color);
    bool FindItemFontSize(int col,int row,int &height,int &width);//查找字体大小

    void SetColColor(int col,COLORREF color);  //设置列颜色
    void SetItemColor(int col,int row,COLORREF color); //设置Item颜色
    void SetColTextColor(int col,COLORREF color);   //设置列文本颜色
    void SetItemTextColor(int col,int row,COLORREF color);
    void SetRowHeigt(int nHeight); //设置行高
    void SetHeaderFontHW(int nHeight,int nWith); //设置表头字体大小
    void SetHeaderTextColor(COLORREF color);
    COLORREF m_color;
    BOOL SetTextColor(COLORREF cr);
    void SetFontHW(int nHeight,int nWith);  //设置字体的高和宽

    void SetItemFontHW(int col,int row,int nHeight,int nWidth);
public:
    // 字体高度
    int m_fontHeight;
public:
    // 字体宽度
    int m_fontWith;
};

分割线

.cpp如下

#include "stdafx.h"
#include "ListCtrlCl.h"

struct stColor
{
    int nRow;
    int nCol;
    COLORREF rgb;
};

struct stFont
{
    int nRow;
    int nCol;
    int nFontWidth;
    int nFontHeight;
};
// CListCtrlCl

IMPLEMENT_DYNAMIC(CListCtrlCl, CListCtrl)

CListCtrlCl::CListCtrlCl()
: m_nRowHeight(0)
, m_fontHeight(12)
, m_fontWith(0)
{
    m_color = RGB(0,0,0);
}

CListCtrlCl::~CListCtrlCl()
{
}


BEGIN_MESSAGE_MAP(CListCtrlCl, CListCtrl)
    ON_WM_MEASUREITEM()
    ON_WM_MEASUREITEM_REFLECT()
    ON_WM_LBUTTONDOWN()
    ON_WM_RBUTTONDOWN()
END_MESSAGE_MAP()



// CListCtrlCl 消息处理程序



void CListCtrlCl::PreSubclassWindow()
{
    // TODO: 在此添加专用代码和/或调用基类
    ModifyStyle(0,LVS_OWNERDRAWFIXED);
    CListCtrl::PreSubclassWindow();
    CHeaderCtrl *pHeader = GetHeaderCtrl();
    m_Header.SubclassWindow(pHeader->GetSafeHwnd());
}

void CListCtrlCl::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{

    // TODO: 添加您的代码以绘制指定项
    TCHAR lpBuffer[256];

    LV_ITEM lvi;

    lvi.mask = LVIF_TEXT | LVIF_PARAM ;
    lvi.iItem = lpDrawItemStruct->itemID ; 
    lvi.iSubItem = 0;
    lvi.pszText = lpBuffer ;
    lvi.cchTextMax = sizeof(lpBuffer);
    VERIFY(GetItem(&lvi));

    LV_COLUMN lvc, lvcprev ;
    ::ZeroMemory(&lvc, sizeof(lvc));
    ::ZeroMemory(&lvcprev, sizeof(lvcprev));
    lvc.mask = LVCF_WIDTH | LVCF_FMT;
    lvcprev.mask = LVCF_WIDTH | LVCF_FMT;

    CDC* pDC;
    pDC = CDC::FromHandle(lpDrawItemStruct->hDC);
    CRect rtClient;
    GetClientRect(&rtClient);
    for ( int nCol=0; GetColumn(nCol, &lvc); nCol++)
    {
        if ( nCol > 0 ) 
        {
            // Get Previous Column Width in order to move the next display item
            GetColumn(nCol-1, &lvcprev) ;
            lpDrawItemStruct->rcItem.left += lvcprev.cx ;
            lpDrawItemStruct->rcItem.right += lpDrawItemStruct->rcItem.left; 
        }

        CRect rcItem;   
        if (!GetSubItemRect(lpDrawItemStruct->itemID,nCol,LVIR_LABEL,rcItem))   
            continue;   

        ::ZeroMemory(&lvi, sizeof(lvi));
        lvi.iItem = lpDrawItemStruct->itemID;
        lvi.mask = LVIF_TEXT | LVIF_PARAM;
        lvi.iSubItem = nCol;
        lvi.pszText = lpBuffer;
        lvi.cchTextMax = sizeof(lpBuffer);
        VERIFY(GetItem(&lvi));
        CRect rcTemp;
        rcTemp = rcItem;

        if (nCol==0)
        {
            rcTemp.left -=2;
        }

        if ( lpDrawItemStruct->itemState & ODS_SELECTED )
        {
            pDC->FillSolidRect(&rcTemp, GetSysColor(COLOR_HIGHLIGHT)) ;
            pDC->SetTextColor(GetSysColor(COLOR_HIGHLIGHTTEXT)) ;
        }
        else
        {
            COLORREF color;
            color = GetBkColor();
            pDC->FillSolidRect(rcTemp,color);

            if (FindColColor(nCol,color))
            {
                pDC->FillSolidRect(rcTemp,color);
            }
            if (FindItemColor(nCol,lpDrawItemStruct->itemID,color))
            {
                pDC->FillSolidRect(rcTemp,color);
            }

            //pDC->SetTextColor(m_color);
        }

        pDC->SelectObject(GetStockObject(DEFAULT_GUI_FONT));

        UINT   uFormat    = DT_CENTER ;
        if (m_Header.m_Format[nCol]=='0')
        {
            uFormat = DT_LEFT;
        }
        else if (m_Header.m_Format[nCol]=='1')
        {
            uFormat = DT_CENTER;
        }
        else if (m_Header.m_Format[nCol]=='2')
        {
            uFormat = DT_RIGHT;
        }

        pDC->SetTextColor(m_color);
        COLORREF color;
        if (FindColTextColor(nCol,color))
        {
            pDC->SetTextColor(color);
        }
        if (FindItemTextColor(nCol,lpDrawItemStruct->itemID,color))
        {
            pDC->SetTextColor(color);
        }

        CFont nFont ,* nOldFont; 
        int fontheight, fontwidth;
        if(FindItemFontSize(nCol,lpDrawItemStruct->itemID,fontheight,fontwidth))
        {
            nFont.CreateFont(fontheight,fontwidth,0,0,0,FALSE,FALSE,0,0,0,0,0,0,_TEXT("宋体"));//创建字体 
            nOldFont = pDC->SelectObject(&nFont);

            int ofst;
            ofst = rcItem.Height() - fontheight;
            rcItem.OffsetRect(0,ofst/2);//居中显示
        }
        else
        {
            nFont.CreateFont(m_fontHeight,m_fontWith,0,0,0,FALSE,FALSE,0,0,0,0,0,0,_TEXT("宋体"));//创建字体 
            nOldFont = pDC->SelectObject(&nFont);

            TEXTMETRIC metric;
            metric.tmHeight;
            pDC->GetTextMetrics(&metric);
            int ofst;
            ofst = rcItem.Height() - metric.tmHeight;
            rcItem.OffsetRect(0,ofst/2);//居中显示
        }
        DrawText(lpDrawItemStruct->hDC, lpBuffer, strlen(lpBuffer), &rcItem, uFormat) ;
        DeleteObject(&nFont);
        pDC->SelectStockObject(SYSTEM_FONT) ;
    }

}

void CListCtrlCl::OnMeasureItem(int nIDCtl, LPMEASUREITEMSTRUCT lpMeasureItemStruct)
{
    // TODO: 在此添加消息处理程序代码和/或调用默认值

    CListCtrl::OnMeasureItem(nIDCtl, lpMeasureItemStruct);
}
void CListCtrlCl::MeasureItem(LPMEASUREITEMSTRUCT lpMeasureItemStruct)
{
    if (m_nRowHeight>0)
    {
        lpMeasureItemStruct->itemHeight = m_nRowHeight;
    }
}
int CListCtrlCl::InsertColumn(int nCol, LPCTSTR lpszColumnHeading, int nFormat /* = LVCFMT_LEFT */, int nWidth /* = -1 */, int nSubItem /* = -1 */)
{
    m_Header.m_HChar.Add(lpszColumnHeading);
    if (nFormat==LVCFMT_LEFT)
    {
        m_Header.m_Format = m_Header.m_Format + "0";
    }
    else if (nFormat==LVCFMT_CENTER)
    {
        m_Header.m_Format = m_Header.m_Format + "1";
    }
    else if (nFormat==LVCFMT_RIGHT)
    {
        m_Header.m_Format = m_Header.m_Format + "2";
    }
    else
    {
        m_Header.m_Format = m_Header.m_Format + "1";
    }
    return CListCtrl::InsertColumn(nCol,lpszColumnHeading,nFormat,nWidth,nSubItem);
}
// Gradient - 渐变系数，立体背景用,不用渐变设为0
void CListCtrlCl::SetHeaderBKColor(int R, int G, int B, int Gradient) //设置表头背景色
{
    m_Header.m_R = R;
    m_Header.m_G = G;
    m_Header.m_B = B;
    m_Header.m_Gradient = Gradient;
}

// 设置表头高度
void CListCtrlCl::SetHeaderHeight(float Height) //设置表头高度
{
    m_Header.m_Height = Height;
}
bool CListCtrlCl::FindColColor(int col,COLORREF &color) //查找列颜色
{
    int flag = 0;
    for (POSITION pos = m_ptrListCol.GetHeadPosition();pos!=NULL;)
    {
        stColor *pColor = (stColor*)m_ptrListCol.GetNext(pos);
        if (pColor->nCol==col)
        {
            flag = 1;
            color = pColor->rgb;
            break;
        }
    }
    if (1==flag)
    {
        return true;
    }
    return false;
}
bool CListCtrlCl::FindItemColor(int col,int row,COLORREF &color) //查找颜色
{
    int flag = 0;
    for (POSITION pos = m_ptrListItem.GetHeadPosition();pos!=NULL;)
    {
        stColor *pColor = (stColor*)m_ptrListItem.GetNext(pos);
        if (pColor->nCol==col&&pColor->nRow==row)
        {
            flag = 1;
            color = pColor->rgb;
            break;
        }
    }
    if (1==flag)
    {
        return true;
    }
    return false;
}
void CListCtrlCl::SetColColor(int col,COLORREF color) //设置列颜色
{
    stColor *pColor  = new stColor;
    pColor->nCol = col;
    pColor->rgb = color;
    m_ptrListCol.AddTail(pColor);
}
void CListCtrlCl::SetItemColor(int col,int row,COLORREF color) //设置格子颜色
{
    stColor *pColor  = new stColor;
    pColor->nCol = col;
    pColor->nRow = row;
    pColor->rgb = color;
    m_ptrListItem.AddTail(pColor);
}
void CListCtrlCl::SetRowHeigt(int nHeight) //高置行高
{
    m_nRowHeight = nHeight;

    CRect rcWin;
    GetWindowRect(&rcWin);
    WINDOWPOS wp;
    wp.hwnd = m_hWnd;
    wp.cx = rcWin.Width();
    wp.cy = rcWin.Height();
    wp.flags = SWP_NOACTIVATE | SWP_NOMOVE | SWP_NOOWNERZORDER | SWP_NOZORDER;
    SendMessage(WM_WINDOWPOSCHANGED, 0, (LPARAM)&wp);
}
void CListCtrlCl::SetHeaderFontHW(int nHeight,int nWith) //设置头部字体宽和高
{
    m_Header.m_fontHeight = nHeight;
    m_Header.m_fontWith = nWith;
}
void CListCtrlCl::SetHeaderTextColor(COLORREF color) //设置头部字体颜色
{
    m_Header.m_color = color;
}
BOOL CListCtrlCl::SetTextColor(COLORREF cr)  //设置字体颜色
{
    m_color = cr;
    return TRUE;
}
void CListCtrlCl::SetFontHW(int nHeight,int nWith) //设置字体高和宽
{
    m_fontHeight = nHeight;
    m_fontWith = nWith;
}
void CListCtrlCl::SetColTextColor(int col,COLORREF color)
{
    stColor *pColor = new stColor;
    pColor->nCol = col;
    pColor->rgb = color;
    m_colTextColor.AddTail(pColor);
}

bool CListCtrlCl::FindColTextColor(int col,COLORREF &color)
{
    int flag = 0;
    for (POSITION pos = m_colTextColor.GetHeadPosition();pos!=NULL;)
    {
        stColor *pColor = (stColor*)m_colTextColor.GetNext(pos);
        if (pColor->nCol==col)
        {
            flag = 1;
            color = pColor->rgb;
            break;
        }
    }
    if (1==flag)
    {
        return true;
    }
    return false;
}

bool CListCtrlCl::FindItemTextColor(int col,int row,COLORREF &color)
{
    int flag = 0;
    for (POSITION pos = m_ItemTextColor.GetHeadPosition();pos!=NULL;)
    {
        stColor *pColor = (stColor*)m_ItemTextColor.GetNext(pos);
        if (pColor->nCol==col&&pColor->nRow==row)
        {
            flag = 1;
            color = pColor->rgb;
            break;
        }
    }
    if (1==flag)
    {
        return true;
    }
    return false;
}

void CListCtrlCl::SetItemTextColor(int col,int row,COLORREF color)
{
    stColor *pColor = new stColor;
    pColor->nCol = col;
    pColor->nRow = row;
    pColor->rgb = color;
    m_ItemTextColor.AddTail(pColor);
}

void CListCtrlCl::SetItemFontHW(int col,int row,int nHeight,int nWidth)
{
    stFont *pFont=new stFont;
    pFont->nCol=col;
    pFont->nRow=row;
    pFont->nFontHeight=nHeight;
    pFont->nFontWidth=nWidth;
    m_ItemFontSize.AddTail(pFont);
}
bool CListCtrlCl::FindItemFontSize(int col,int row,int &height,int &width)
{
    int flag = 0;
    for (POSITION pos = m_ItemFontSize.GetHeadPosition();pos!=NULL;)
    {
        stFont *pFont = (stFont*)m_ItemFontSize.GetNext(pos);
        if (pFont->nCol==col&&pFont->nRow==row)
        {
            flag = 1;
            height=pFont->nFontHeight;
            width = pFont->nFontWidth;
            break;
        }
    }
    if (1==flag)
    {
        return true;
    }
    return false;
}

原理：要明白DrawItem是关键，即绘每一个项，我们就在这里面搞事！CPtrList 看名字就知道叫指针列表，用来保存要修改的项，如果那项要修改，stFont与stColor就是我们要添加进去的指针，哪里要修改哪里就new一个，再addtail（添加到尾部），再写一个Find方法，某项有无被设置了颜色或者字体等属性，并在DrawItem里调用，找到就设置某项，基本原理即清楚了！附一张效果图：