#include "StdAfx.h"
#include "ListControl.h"

//////////////////////////////////////////////////////////////////////////////////
//宏定义

//列表属性
#define LIST_STYTE LVS_EX_FULLROWSELECT|LVS_EX_HEADERDRAGDROP|LVS_EX_FLATSB

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

//排序信息
struct tagSortParameter
{
	bool							bAscendSort;						//升序标志
	WORD							wColumnIndex;						//位置索引
	CListControl *					pListControl;						//列表控件
};

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

BEGIN_MESSAGE_MAP(CListHeaderCtrl, CHeaderCtrl)
	ON_WM_PAINT()
END_MESSAGE_MAP()

BEGIN_MESSAGE_MAP(CListControl, CListCtrl)
	ON_WM_ERASEBKGND()
	ON_NOTIFY_REFLECT(LVN_COLUMNCLICK, OnColumnclick)
END_MESSAGE_MAP()

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

//构造函数
CListHeaderCtrl::CListHeaderCtrl()
{
}

//析构函数
CListHeaderCtrl::~CListHeaderCtrl()
{
}

//重画函数
VOID CListHeaderCtrl::OnPaint() 
{
	CPaintDC dc(this);

	//获取位置
	CRect rcRect;
	GetClientRect(&rcRect);

	//设置 DC
	dc.SetBkMode(TRANSPARENT);
	dc.SelectObject(GetFont());

	//绘画子项
	CRect rcItem;
	HDITEM HDItem;
	TCHAR szBuffer[64];
	for (INT i=0;i<GetItemCount();i++)
	{
		//构造变量
		HDItem.mask=HDI_TEXT;
		HDItem.pszText=szBuffer;
		HDItem.cchTextMax=CountArray(szBuffer);

		//获取信息
		GetItem(i,&HDItem);
		GetItemRect(i,&rcItem);

		//绘画背景
		dc.DrawFrameControl(&rcItem,DFC_BUTTON,DFCS_BUTTONPUSH);

		//绘画标题
		rcItem.DeflateRect(3,1,3,1);
		const UINT uFormat=DT_WORDBREAK|DT_EDITCONTROL|DT_END_ELLIPSIS|DT_SINGLELINE|DT_VCENTER|DT_CENTER;
		dc.DrawText(szBuffer,lstrlen(szBuffer),&rcItem,uFormat);
	}


	return;
}

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

//构造函数
CListControl::CListControl()
{
	//设置变量
	m_bAscendSort=true;

	return;
}

//析构函数
CListControl::~CListControl()
{
}

//控件绑定
VOID CListControl::PreSubclassWindow()
{
	__super::PreSubclassWindow();

	//设置属性
	SetBkColor(CR_NORMAL_BK);
	SetExtendedStyle(LIST_STYTE);

	//绑定控件
	m_ListHeaderCtrl.SubclassWindow(GetHeaderCtrl()->GetSafeHwnd());

	//配置列表
	InitListControl();

	return;
}

//绘画函数
VOID CListControl::DrawItem(LPDRAWITEMSTRUCT lpDrawItemStruct)
{
	//变量定义
	CRect rcItem=lpDrawItemStruct->rcItem;
	CDC * pDC=CDC::FromHandle(lpDrawItemStruct->hDC);

	//获取属性
	INT nItemID=lpDrawItemStruct->itemID;
	INT nColumnCount=m_ListHeaderCtrl.GetItemCount();

	//获取颜色
	tagListItemColor ListItemColor;
	GetListItemColor(lpDrawItemStruct->itemData,lpDrawItemStruct->itemState,ListItemColor);

	//绘画区域
	CRect rcClipBox;
	pDC->GetClipBox(&rcClipBox);

	//设置 DC
	pDC->SetBkColor(ListItemColor.rcBackColor);
	pDC->SetTextColor(ListItemColor.rcTextColor);

	//绘画子项
	for (INT i=0;i<nColumnCount;i++)
	{
		//获取位置
		CRect rcSubItem;
		GetSubItemRect(nItemID,i,LVIR_LABEL,rcSubItem);

		//填充空隙
		if ((i==0)&&(rcItem.left<rcSubItem.left))
		{
			INT nYFill=rcItem.bottom-rcItem.top;
			INT nXFill=rcSubItem.left-rcItem.left;
			pDC->FillSolidRect(rcItem.left,rcItem.top,nXFill,nYFill,ListItemColor.rcBackColor);
		}

		//绘画判断
		if (rcSubItem.left>rcClipBox.right) break;
		if (rcSubItem.right<rcClipBox.left) continue;
		if (rcSubItem.left==rcSubItem.right) continue;

		//绘画背景
		pDC->FillSolidRect(&rcSubItem,ListItemColor.rcBackColor);

		//获取文字
		TCHAR szString[256]=TEXT("");
		GetItemText(nItemID,i,szString,CountArray(szString));

		//获取格式
		HDITEM HDItem;
		HDItem.mask=HDI_FORMAT;
		m_ListHeaderCtrl.GetItem(i,&HDItem);

		//设置格式
		UINT uFormat=0;
		if (HDItem.fmt&HDF_LEFT) uFormat|=DT_LEFT;
		if (HDItem.fmt&HDF_RIGHT) uFormat|=DT_RIGHT;
		if (HDItem.fmt&HDF_CENTER) uFormat|=DT_CENTER;

		//绘画文字
		rcSubItem.left+=5;
		pDC->DrawText(szString,lstrlen(szString),&rcSubItem,uFormat|DT_VCENTER|DT_SINGLELINE|DT_END_ELLIPSIS);
	}

	//绘画选择
	if (lpDrawItemStruct->itemState&ODS_FOCUS) pDC->DrawFocusRect(&rcItem);

	return;
}

//排列函数
INT CALLBACK CListControl::SortFunction(LPARAM lParam1, LPARAM lParam2, LPARAM lParamList)
{
	//变量定义
    tagSortParameter * pSortParameter=(tagSortParameter *)lParamList;

	//排序变量
	WORD wColumnIndex=pSortParameter->wColumnIndex;
	CListControl * pListControl=pSortParameter->pListControl;

	//子项排序
	INT nResult=pListControl->SortListItemData(lParam1,lParam2,wColumnIndex);

	return (pSortParameter->bAscendSort)?nResult:-nResult;
}

//绘画背景
BOOL CListControl::OnEraseBkgnd(CDC * pDC)
{
	//绘画背景
	if (m_ListHeaderCtrl.m_hWnd!=NULL)
	{
		//获取位置
		CRect rcClient;
		CRect rcHeaderCtrl;
		GetClientRect(&rcClient);
		m_ListHeaderCtrl.GetWindowRect(&rcHeaderCtrl);

		//绘画背景
		rcClient.top=rcHeaderCtrl.Height();
		pDC->FillSolidRect(&rcClient,GetBkColor());

		return TRUE;
	}

	return __super::OnEraseBkgnd(pDC);
}

//点击消息
VOID CListControl::OnColumnclick(NMHDR * pNMHDR, LRESULT * pResult) 
{
	//变量定义
	NM_LISTVIEW * pNMListView=(NM_LISTVIEW *)pNMHDR;

	//变量定义
	tagSortParameter SortParameter;
	ZeroMemory(&SortParameter,sizeof(SortParameter));

	//设置变量
	SortParameter.pListControl=this;
	SortParameter.bAscendSort=m_bAscendSort;
	SortParameter.wColumnIndex=pNMListView->iSubItem;

	//设置状态
	m_bAscendSort=!m_bAscendSort;

	//排列列表
	SortItems(SortFunction,(LPARAM)&SortParameter);

	return;
}

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