ListCtrl控件[建议收藏]

#pragma once


// CListCtrlEx
#define IDC_ADDEDIT 1002
class CListCtrlEx : public CListCtrl
{
    DECLARE_DYNAMIC(CListCtrlEx)

public:
    CListCtrlEx();
    virtual ~CListCtrlEx();

protected:
    DECLARE_MESSAGE_MAP()
public:
    afx_msg void OnNMClick(NMHDR *pNMHDR, LRESULT *pResult);
    afx_msg void OnLvnColumnclick(NMHDR *pNMHDR, LRESULT *pResult);
    afx_msg void OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult);
public:
    virtual void DrawItem(LPDRAWITEMSTRUCT /*lpDrawItemStruct*/);
    void DrawRowICon(CDC *pDC, CRect &rc, int nRow);
};

// ListCtrlEx.cpp : 实现文件
//

#include "stdafx.h"
#include "mfcctrlstu.h"
#include "ListCtrlEx.h"


// CListCtrlEx

IMPLEMENT_DYNAMIC(CListCtrlEx, CListCtrl)

CListCtrlEx::CListCtrlEx()
{

}

CListCtrlEx::~CListCtrlEx()
{
}


BEGIN_MESSAGE_MAP(CListCtrlEx, CListCtrl)
    ON_NOTIFY_REFLECT(NM_CLICK, &CListCtrlEx::OnNMClick)
    ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, &CListCtrlEx::OnLvnColumnclick)
    ON_NOTIFY_REFLECT(NM_DBLCLK, &CListCtrlEx::OnNMDblclk)
END_MESSAGE_MAP()

void CListCtrlEx::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{
    // 背景
    CDC dc;
    CRect rc = lpDrawItemStruct->rcItem;
    dc.Attach(lpDrawItemStruct->hDC);
    if (lpDrawItemStruct->itemState & ODS_SELECTED)
    {
        // 选中状态蓝底白字
        dc.FillSolidRect(&rc, GetSysColor(COLOR_HIGHLIGHT));
        dc.SetTextColor(GetSysColor(COLOR_HIGHLIGHTTEXT));
    }
    else
    {
        // 普通状态白底黑字
        //dc.FillSolidRect(&rc, GetSysColor(COLOR_WINDOW));
        dc.FillSolidRect(&rc, RGB(222,33,33));
        dc.SetTextColor(GetSysColor(COLOR_WINDOWTEXT));
    }
    
    int nRow = lpDrawItemStruct->itemID;
    CHeaderCtrl *pHeadCtrl = GetHeaderCtrl();
    int nColCount = pHeadCtrl->GetItemCount();
    CRect rcCell; // 注意单元格rcCell的处理
    for (int nCol = 0;nCol < nColCount; nCol++)
    {
        pHeadCtrl->GetItemRect(nCol, &rcCell);
        rcCell.top = rc.top;
        rcCell.bottom = rc.bottom;                            
        CString strText = GetItemText(nRow, nCol);
    
        // 图标
        if (!nCol)
        {
            DrawRowICon(&dc, rc, nRow);
            rcCell.left += 18;
        }
        
        // 文字
        dc.SetBkMode(TRANSPARENT);
        dc.DrawText(strText, &rcCell, DT_LEFT|DT_VCENTER|DT_SINGLELINE);
        //TRACE("%d-%d-%d-%d-%d-%d-%s\r\n", nRow, nCol, rcCell.left, rcCell.top, rcCell.right,rcCell.bottom,strText);
    }
    
    dc.Detach();
}

void CListCtrlEx::DrawRowICon(CDC *pDC, CRect &rc, int nRow)
{
    /*LV_ITEM lvi = {LVIF_IMAGE};
    GetItem(&lvi);*/
    CImageList *pImageList = GetImageList(LVSIL_SMALL);
    pImageList->Draw(pDC, nRow, rc.TopLeft(), ILD_TRANSPARENT);
}

void CListCtrlEx::OnLvnColumnclick(NMHDR *pNMHDR, LRESULT *pResult)
{
    LPNMLISTVIEW pNMLV = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
    int nCol = pNMLV->iSubItem;
    HDITEM hi = {HDI_IMAGE | HDI_FORMAT};
    
    CHeaderCtrl *pHeadCtrl = GetHeaderCtrl();
    pHeadCtrl->GetItem(nCol, &hi);
    static int k = 0;
    hi.iImage = k;
    k = !k;

    int nColCount = pHeadCtrl->GetItemCount();
    for (int i = 0; i < nColCount; i ++)
    {
        if (i == nCol)
            hi.fmt |= HDF_IMAGE;
        else
            hi.fmt &= ~HDF_IMAGE;

        
        pHeadCtrl->SetItem(i, &hi);
    }

    *pResult = 0;
}

void CListCtrlEx::OnNMClick(NMHDR *pNMHDR, LRESULT *pResult)
{
    // TODO: 在此添加控件通知处理程序代码
    /*if (!pNMHDR)
        return;
    LPNMITEMACTIVATE pNmItemActive = (LPNMITEMACTIVATE)pNMHDR;
    LV_ITEM lvItem;
    lvItem.iItem = pNmItemActive->iItem;
    lvItem.iSubItem = 1;
    lvItem.mask = LVIF_TEXT;
    GetItem(&lvItem);*/
    *pResult = 0;
}
void CListCtrlEx::OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult)
{
    // 在1行1列创建个Edit
    LPNMITEMACTIVATE pNmItemActive = (LPNMITEMACTIVATE)pNMHDR;
    int nRow = pNmItemActive->iItem;
    int nCol = pNmItemActive->iSubItem;
    CRect rc;
    GetItemRect(nRow, &rc, LVIR_BOUNDS);
    CRect rcCell;
    CHeaderCtrl *pHeadCtrl = GetHeaderCtrl();
    pHeadCtrl->GetItemRect(nCol, &rcCell);
    rcCell.top = rc.top;
    rcCell.bottom = rc.bottom;
    if (3==nRow&&1==nCol)
    {
        CEdit *pEdit = new CEdit;
        pEdit->Create(WS_CHILD|WS_VISIBLE|WS_TABSTOP|ES_LEFT, rcCell, this, IDC_ADDEDIT);
    }
    
    *pResult = 0;
}