自身支持排序的CSortListCtrl

自身支持排序的CSortListCtrl

VC里的CListCtrl是个很不好用的控件，特别是排序，实现起来很麻烦。

关于排序的基本用法，有一篇很好的文章：
http://www.vchelp.net/vchelp/zart/sortl.asp?type_id=9&class_id=1&cata_id=1&article_id=73&search_term=

但是，在这个例子里也存在个问题：排序的实现是和数据源相关的，如果有若干个表需要排序的话，每个表都要写相应的代码，这是一件非常痛苦的事。

所以，自己重新写了一个自身支持排序的CSoftListCtrl。并且实现了按不同数据类型排序，比较有实用价值！
下面说说几个主要的地方。

需要用到的一些函数：
int CListCtrl::InsertColumn( int nCol, const LVCOLUMN* pColumn ); - 创建Column
BOOL CListCtrl::SetItem( const LVITEM* pItem ); - 设置ListCtrl的Item
int CListCtrl::FindItem( LVFINDINFO* pFindInfo, int nStart = -1 ) const; - 查找相应的Item
BOOL CListCtrl::SortItems( PFNLVCOMPARE pfnCompare, DWORD dwData ); - 调用回调函数进行排序
int CALLBACK CListCtrl::ListCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort); - 负责排序的回调函数
CHeaderCtrl* CListCtrl::GetHeaderCtrl( ); - 得到CHeaderCtrl
BOOL CHeaderCtrl::SetItem( int nPos, HDITEM* pHeaderItem ); - 设置CHeaderCtrl的Item

函数的详细细节请参考MSDN.

1. CSortListCtrl是CListCtrl的派生类

2. 要让CSortList自行排序，需要让CSortList自己处理LVN_COLUMNCLICK消息
ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, OnLvnColumnclick)
OnLvnColumnclick的作用就是设置排序列，排序方式，最后调用SortItems()。

3. 记录数据类型的信息
为了支持不同的数据类型，需要在Header中保存必要的类型信息
在SortListCtrl.h定义数据类型: enum { INT_TYPE = 0, STRING_TYPE, DOUBLE_TYPE };
目前只定义了三种，可自行添加。
在创建Column时，用CHeaderCtrl::SetItem()在HeaderCtrl中设置数据类型信息。

4. 排序方式和排序列
在CSortListCtrl中增加两个成员：
BOOL m_bAsc;//是否顺序排序
int m_nSortedCol;//当前排序的列
在OnLvnColumnclick()中设置这两个值

5. BOOL CListCtrl::SortItems( PFNLVCOMPARE pfnCompare, DWORD dwData );
pfnCompare - 回调函数
dwData - 传递到回调函数的参数，实际传递的是CListCtrl的this的指针

6. int CALLBACK CSortListCtrl::ListCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort);
lParam1,lParam2 - CListCtrl::SetItem()所设置的LVITEM::lParam, 参阅LVITEM结构和CListCtrl::SetItemData()定义
lParamSort - lParamSort 即 CListCtrl::SortItems()的第二个参数 DWORD dwData
实现：
根据lParam1,lParam2得到Item及ItemText
根据m_nSortedCol从Header中得到数据类型
根据数据类型和m_bAsc对ItemText进行排序

下面是示例代码和源文件：

只有两个文件：CSortListCtrl.H, CSortListCtrl.cpp
直接把这两个文件放到项目里就可以了。
唯一要注意的是：IDB_HDRUP, IDB_HDRDOWN 是两个位图资源，分别表示顺序和倒序，这个就需要自己弄了。

A. SoftListCtrl.h

#if !defined(AFX_SORTLIST_H__6ACE2F6F_AEFE_11D3_BDE9_F4145AA4F676__INCLUDED_)
#define AFX_SORTLIST_H__6ACE2F6F_AEFE_11D3_BDE9_F4145AA4F676__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
// SortList.h : header file
//

/////////////////////////////////////////////////////////////////////////////
// CSortListCtrl window

class CSortListCtrl : public CListCtrl
{
// Construction
public:
CSortListCtrl();
// Attributes
public:

// Operations
public:
enum { INT_TYPE = 0, STRING_TYPE, DOUBLE_TYPE };
// Overrides
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CSortListCtrl)
//}}AFX_VIRTUAL

// Implementation
public:
virtual ~CSortListCtrl();

// Generated message map functions
protected:
//{{AFX_MSG(CSortListCtrl)
afx_msg bool OnColumnclick(NMHDR* pNMHDR, LRESULT* pResult);
//}}AFX_MSG

DECLARE_MESSAGE_MAP()
public:
afx_msg void OnLvnColumnclick(NMHDR *pNMHDR, LRESULT *pResult);

void CreateSortIcons();
void SetSortIcon();

bool GetFullRowSelect();
// 设置为行选中
void SetFullRowSelect( bool bFullRowSelect );

bool GetGridLines();
// 设置绘制表格
void SetGridLines( bool bGridLines );

void UpdateControlTitle();

static CALLBACK ListCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort);
public:
BOOL m_bAsc;//是否顺序排序
int m_nSortedCol;//当前排序的列

private:
CBitmap m_bmpUpArrow;
CBitmap m_bmpDownArrow;
int m_nUpArrow;
int m_nDownArrow;
CImageList m_imglstSortIcons;
};

/////////////////////////////////////////////////////////////////////////////

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_SORTLIST_H__6ACE2F6F_AEFE_11D3_BDE9_F4145AA4F676__INCLUDED_)

B. SoftListCtrl.cpp

// SortList.cpp : implementation file
//

#include "stdafx.h"
#include "../DbpsGUI.h"
#include "SortListCtrl.h"

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

/////////////////////////////////////////////////////////////////////////////
// CSortListCtrl

CSortListCtrl::CSortListCtrl()
{
m_bAsc=TRUE;
m_nSortedCol = 0;
CreateSortIcons();
//GetHeaderCtrl()->SetImageList(&m_imglstSortIcons);
}

CSortListCtrl::~CSortListCtrl()
{
m_imglstSortIcons.DeleteImageList();
m_bmpUpArrow.DeleteObject();
m_bmpDownArrow.DeleteObject();
}

BEGIN_MESSAGE_MAP(CSortListCtrl, CListCtrl)
//{{AFX_MSG_MAP(CSortListCtrl)
//}}AFX_MSG_MAP
ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, OnLvnColumnclick)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CSortListCtrl message handlers

void CSortListCtrl::OnLvnColumnclick(NMHDR *pNMHDR, LRESULT *pResult)
{
LPNMLISTVIEW pNMListView = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
// TODO: Add your control notification handler code here
//NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
if( pNMListView->iSubItem == m_nSortedCol )
{
m_bAsc = !m_bAsc;
}
else
{
m_bAsc = TRUE;
m_nSortedCol = pNMListView->iSubItem;
}
SortItems( ListCompare, (DWORD)this );
SetSortIcon();
*pResult = 0;
}

void CSortListCtrl::CreateSortIcons()
{
if (!m_imglstSortIcons.m_hImageList)
{
COLORMAP cm = {RGB(0, 0, 0), GetSysColor(COLOR_GRAYTEXT)};
m_imglstSortIcons.Create (9, 5, ILC_COLOR24 | ILC_MASK, 2, 0);
m_bmpUpArrow.LoadMappedBitmap(IDB_HDRUP, 0, &cm, 1);
m_nUpArrow = m_imglstSortIcons.Add(&m_bmpUpArrow, RGB(255, 255, 255));
m_bmpDownArrow.LoadMappedBitmap(IDB_HDRDOWN, 0, &cm, 1);
m_nDownArrow = m_imglstSortIcons.Add(&m_bmpDownArrow, RGB(255, 255, 255));
}
}

void CSortListCtrl::SetSortIcon()
{
CHeaderCtrl* pHeaderCtrl = this->GetHeaderCtrl();
ASSERT(pHeaderCtrl);

pHeaderCtrl->SetImageList(&m_imglstSortIcons);
for( int col = 0; col< GetHeaderCtrl()->GetItemCount(); col++ )
{
HDITEM hdrItem = { 0,};

hdrItem.mask = HDI_FORMAT | HDI_IMAGE;

BOOL ret = pHeaderCtrl->GetItem(col-1, &hdrItem);
ret = pHeaderCtrl->GetItem(col+1, &hdrItem);
ret = pHeaderCtrl->GetItem(col, &hdrItem);
if ( m_nSortedCol == col)
{
hdrItem.fmt = hdrItem.fmt & HDF_JUSTIFYMASK | HDF_IMAGE | HDF_STRING | HDF_BITMAP_ON_RIGHT;
if( m_bAsc )
hdrItem.iImage = m_nUpArrow;
else
hdrItem.iImage = m_nDownArrow;
}
else
{
hdrItem.fmt = hdrItem.fmt & HDF_JUSTIFYMASK | HDF_STRING;
}
pHeaderCtrl->SetItem(col, &hdrItem);
}
}

bool CSortListCtrl::GetFullRowSelect()
{
return ( GetExtendedStyle()&LVS_EX_FULLROWSELECT) == LVS_EX_FULLROWSELECT;
}

void CSortListCtrl::SetFullRowSelect( bool bFullRowSelect )
{
if( bFullRowSelect )
SetExtendedStyle( GetExtendedStyle()|LVS_EX_FULLROWSELECT );
else
SetExtendedStyle( GetExtendedStyle()&(~LVS_EX_FULLROWSELECT) );
}

bool CSortListCtrl::GetGridLines()
{
return ( GetExtendedStyle() & LVS_EX_GRIDLINES ) == LVS_EX_GRIDLINES;
}

void CSortListCtrl::SetGridLines( bool bGridLines )
{
if( bGridLines )
SetExtendedStyle( GetExtendedStyle()|LVS_EX_GRIDLINES );
else
SetExtendedStyle( GetExtendedStyle()&(~LVS_EX_GRIDLINES) );
}

void CSortListCtrl::UpdateControlTitle()
{
CHeaderCtrl* pHeaderCtrl = GetHeaderCtrl( );
if( pHeaderCtrl != NULL && pHeaderCtrl->GetItemCount()>0 )
{
HDITEM headerItem = {0};
for(int i=0;i<pHeaderCtrl->GetItemCount();i++)
{
headerItem.mask = HDI_LPARAM;
pHeaderCtrl->GetItem( i, &headerItem );
UINT nID = (UINT)(headerItem.lParam) >> 16;
CString sHead;
sHead.Format( nID );
headerItem.mask = HDI_TEXT ;
headerItem.pszText = (LPSTR)(LPCSTR)sHead;
pHeaderCtrl->SetItem( i, &headerItem );
}
}
}

int CALLBACK CSortListCtrl::ListCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
{
CSortListCtrl* pList=(CSortListCtrl*)lParamSort;
int nItem1, nItem2;

LVFINDINFO FindInfo;
FindInfo.flags = LVFI_PARAM; // 指定查找方式
FindInfo.lParam = lParam1;
nItem1 = pList->FindItem(&FindInfo, -1); // 得到对应Item索引
FindInfo.lParam = lParam2;
nItem2 = pList->FindItem(&FindInfo, -1);

if((nItem1 == -1) || (nItem2 == -1))
{
TRACE("无法找到!/n");
return 0;
}

CString str1,str2;
str1 = pList->GetItemText(nItem1, pList->m_nSortedCol); // 得到排序列的Text
str2 = pList->GetItemText(nItem2, pList->m_nSortedCol);

HDITEM headerItem;
headerItem.mask = HDI_LPARAM;
CHeaderCtrl* pHeaderCtrl = pList->GetHeaderCtrl( );
pHeaderCtrl->GetItem( pList->m_nSortedCol, &headerItem );
UINT nType = (UINT)(headerItem.lParam);

int iCompRes = 0;
switch( nType )
{
case INT_TYPE:
{
int i1,i2;
i1 = atoi( str1 );
i2 = atoi( str2 );
if( i1 > i2)
iCompRes = 1;
else if( i1 == i2 )
iCompRes = 0;
else
iCompRes = -1;
}
break;
case DOUBLE_TYPE:
{
double i1,i2;
i1 = atof( str1 );
i2 = atof( str2 );
if( i1 > i2)
iCompRes = 1;
else if( i1 == i2 )
iCompRes = 0;
else
iCompRes = -1;
}
break;
case STRING_TYPE:
default:
{
if( str1 > str2)
iCompRes = 1;
else if( str1 == str2 )
iCompRes = 0;
else
iCompRes = -1;
}
break;

}
if(pList->m_bAsc)
return iCompRes;
else
return iCompRes*-1;
return 0;
}