// DataGridEx.cpp: implementation of the CDataGridEx class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "autophs.h"
#include "DataGridEx.h"

#include "Column.h"
#include "Columns.h"

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

/////////////////////////////////////////////////////////////////////////////
// CDataGridEx

CDataGridEx::CDataGridEx()
{
	m_RecordCount=-1;
	m_PreVScrollPos=0;
	m_PreVScrollFlag=FALSE;
}

CDataGridEx::~CDataGridEx()
{
}

CMap<DWORD,DWORD,CDataGridEx::tagHookCountInfo*,CDataGridEx::tagHookCountInfo*> CDataGridEx::m_MouseThreadHookMap;

BEGIN_MESSAGE_MAP(CDataGridEx, CDataGrid)
	//{{AFX_MSG_MAP(CDataGridEx)
	ON_WM_HSCROLL()
	ON_WM_VSCROLL()
	ON_WM_RBUTTONDOWN()
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_MOUSEWHEEL2,OnMouseWheel2)
	ON_COMMAND(IDC_GRID_ASC,OnSortAsc)
	ON_COMMAND(IDC_GRID_DESC,OnSortDesc)
END_MESSAGE_MAP()

////////////////////////////////////////////////
//
// 初始化TOOLTIP控件
//
void CDataGridEx::InitToolTipControl()
{
	if(m_RecordCount<0)
	{
		LPUNKNOWN pUnknown;
		HRESULT hr;
		_RecordsetPtr IRecordset;

		pUnknown=GetDataSource();
		
		if(pUnknown==NULL)
			return;

		try
		{
			hr=pUnknown->QueryInterface(__uuidof(_Recordset),(void**)&IRecordset);

			pUnknown->Release();

			if(FAILED(hr))
			{
				throw _com_error(hr);
			}

			m_RecordCount=IRecordset->GetRecordCount();
		}
		catch(_com_error &e)
		{
			AfxMessageBox(e.ErrorMessage());
		}
	}

	if(IsWindow(m_ToolTip.GetSafeHwnd()))
		return;

	m_ToolTip.Create(this);
	
	TOOLINFO ToolInfo;

	memset(&ToolInfo,0,sizeof(TOOLINFO));

	ToolInfo.cbSize=sizeof(TOOLINFO);
	ToolInfo.uFlags=TTF_TRACK|TTF_IDISHWND| TTF_ABSOLUTE;
	ToolInfo.hwnd=this->m_hWnd;
	ToolInfo.uId=(UINT)this->m_hWnd;
	ToolInfo.lpszText=LPSTR_TEXTCALLBACK;

	m_ToolTip.SendMessage(TTM_ADDTOOL,0,(LPARAM)&ToolInfo);
}


/////////////////////////////////////////////////////////////////////////////
// CDataGridEx message handlers


/////////////////////////////////////////////////////////////////////////////
//
// 响应横向滚动
//
void CDataGridEx::OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
{

	if(nSBCode==SB_THUMBTRACK)
	{
		short Col;

		Col=this->GetLeftCol();

		this->Scroll(nPos-Col,0);
	}

	CDataGrid::OnHScroll(nSBCode, nPos, pScrollBar);
}

///////////////////////////////////////////////////////////////////////////////
//
// 响应垂直滚动
//
void CDataGridEx::OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
{
	if(nSBCode==SB_THUMBTRACK || nSBCode==SB_THUMBPOSITION)
	{
		TOOLINFO ToolInfo;
		_variant_t varTemp;
		POINT pt;

		InitToolTipControl();

		memset(&ToolInfo,0,sizeof(TOOLINFO));

		ToolInfo.cbSize=sizeof(TOOLINFO);
		ToolInfo.uFlags=TTF_TRACK|TTF_IDISHWND| TTF_ABSOLUTE;
		ToolInfo.hwnd=this->m_hWnd;
		ToolInfo.uId=(UINT)this->m_hWnd;
		ToolInfo.lpszText=LPSTR_TEXTCALLBACK;

		if(nSBCode==SB_THUMBTRACK)
		{
//			varTemp=GetFirstRow();
//			Scroll(0,nPos-(long)varTemp+1);

			if(m_PreVScrollFlag==FALSE)
			{
				m_PreVScrollPos=nPos;
				m_PreVScrollFlag=TRUE;
			}
			else
			{
				Scroll(0,nPos-m_PreVScrollPos);

				m_PreVScrollPos=nPos;
			}

			::_stprintf(m_ToolTipInfo,_T("记录：%d共有记录数%d"),nPos+1,m_RecordCount);

			::GetCursorPos(&pt);
			m_ToolTip.SendMessage(TTM_TRACKPOSITION,0,MAKEWPARAM(pt.x+15,pt.y));
			m_ToolTip.SendMessage(TTM_TRACKACTIVATE,TRUE,(LPARAM)&ToolInfo);

			//放置ToolTip控件在合适显示的位置
			PlaceToolTipWndToRightPosition();
		}
		else
		{
			m_ToolTip.SendMessage(TTM_TRACKACTIVATE,FALSE,(LPARAM)&ToolInfo);
		}

		if(nSBCode==SB_THUMBPOSITION)
		{
			m_PreVScrollFlag=FALSE;
		}

	}

	CDataGrid::OnVScroll(nSBCode, nPos, pScrollBar);
}

BOOL CDataGridEx::OnNotify(WPARAM wParam, LPARAM lParam, LRESULT *pResult)
{
	LPNMTTDISPINFO 	lpnmtdi=(LPNMTTDISPINFO)lParam;

	if(lpnmtdi->hdr.code==TTN_NEEDTEXT)
	{
		lpnmtdi->lpszText=m_ToolTipInfo;
	}

	return CDataGrid::OnNotify(wParam,lParam,pResult);
}

///////////////////////////////////////////////////////
//
// 放置ToolTip控件在合适显示的位置
//
void CDataGridEx::PlaceToolTipWndToRightPosition()
{
	CRect ToolTipRect,ScreenRect;

	if(!::IsWindow(this->m_ToolTip.GetSafeHwnd()))
		return;

	m_ToolTip.GetWindowRect(&ToolTipRect);

	if(ToolTipRect.IsRectEmpty())
		return;

	::GetWindowRect(::GetDesktopWindow(),&ScreenRect);

	if(ToolTipRect.right>ScreenRect.right)
	{
		ToolTipRect.OffsetRect(ScreenRect.right-ToolTipRect.right,0);

		m_ToolTip.SendMessage(TTM_TRACKPOSITION,0,MAKEWPARAM(ToolTipRect.left,ToolTipRect.top));
	}


}

//////////////////////////////////////////////
//
// 初始化中间滚动球的滚动响应
//
// 应该在窗口创建已后才调用
//
void CDataGridEx::InitMouseWheel()
{
	HANDLE Handle;
	tagHookCountInfo* pHookCountInfo;

	if(!IsWindow(GetSafeHwnd()))
		return;

	Handle=::GetProp(GetSafeHwnd(),_T("MouseWheel"));

	if(Handle!=NULL)
		return;

	::SetProp(GetSafeHwnd(),_T("MouseWheel"),(HANDLE)this);

	if(!m_MouseThreadHookMap.Lookup(::GetCurrentThreadId(),pHookCountInfo))
	{
		pHookCountInfo=new tagHookCountInfo;

		pHookCountInfo->hHook=::SetWindowsHookEx(WH_GETMESSAGE,CDataGridEx::GetMsgProc,
												 NULL,GetCurrentThreadId());
		
		pHookCountInfo->Count=1;

		m_MouseThreadHookMap.SetAt(GetCurrentThreadId(),pHookCountInfo);
	}
	else
	{
		pHookCountInfo->Count++;
	}
}

/////////////////////////////////////////////////////
//
// 钩子的会调函数
//
LRESULT CALLBACK CDataGridEx::GetMsgProc(int code, WPARAM wParam, LPARAM lParam)
{
	HANDLE Handle;
	tagHookCountInfo* pHookCountInfo;
	HWND hWnd;
	MSG *pMsg;

	pMsg=(MSG*)lParam;

	m_MouseThreadHookMap.Lookup(::GetCurrentThreadId(),pHookCountInfo);

	if(code<0 || pMsg->message!=WM_MOUSEWHEEL)
		return CallNextHookEx(pHookCountInfo->hHook,code,wParam,lParam);


	hWnd=pMsg->hwnd;

	while(TRUE)
	{
		if(!IsWindow(hWnd))
			break;

		Handle=::GetProp(hWnd,_T("MouseWheel"));

		if(Handle==NULL)
			hWnd=::GetParent(hWnd);
		else
		{
			::PostMessage(hWnd,WM_MOUSEWHEEL2,pMsg->wParam,pMsg->lParam);
			break;
		}
	}

	return CallNextHookEx(pHookCountInfo->hHook,code,wParam,lParam); 
}

LRESULT CDataGridEx::OnMouseWheel2(WPARAM wParam, LPARAM lParam)
{
	int Ro;

	Ro=int(wParam);

	if(Ro>=0)
	{
		this->Scroll(0,-2);
	}
	else
	{
		this->Scroll(0,2);
	}

	return 0;	
}

void CDataGridEx::OnRButtonDown(UINT nFlags, CPoint point) 
{
	POINT ScreenPt;
	CMenu menu;

	try
	{
//		m_SelectCol=ColContaining(static_cast<float>(point.x));

		m_SelectCol=this->GetSelStartCol();

		if(m_SelectCol<0)
			return;
	}
	catch(COleDispatchException *e)
	{
		char cStr[255];
		e->GetErrorMessage(cStr,255);
		AfxMessageBox(cStr);
		return;
	}

	ScreenPt.x=point.x;
	ScreenPt.y=point.y;

	ClientToScreen(&ScreenPt);
	

	//LFX 2005.3.6注释掉,暂时不实现此功能
	//menu.CreatePopupMenu();
	//menu.AppendMenu(MF_STRING,IDC_GRID_ASC,"升序(&A)");
	//menu.AppendMenu(MF_STRING,IDC_GRID_DESC,"降序(&C)");
	//menu.TrackPopupMenu(TPM_LEFTALIGN,ScreenPt.x,ScreenPt.y,this);

	CDataGrid::OnRButtonDown(nFlags, point);
}

void CDataGridEx::OnSortAsc()
{
	CString strTemp;
	_RecordsetPtr IRecordset;
	LPUNKNOWN pUnknown;
	HRESULT hr;

	try
	{
		pUnknown=this->GetDataSource();

		hr=pUnknown->QueryInterface(__uuidof(_RecordsetPtr),(void**)&IRecordset);

		if(FAILED(hr))
		{
			pUnknown->Release();
			AfxMessageBox(_com_error(hr).ErrorMessage());
			return;
		}

		pUnknown->Release();

		strTemp=GetColumns().GetItem(_variant_t(m_SelectCol)).GetCaption();

		strTemp+=_T(" ASC");
		IRecordset->Sort=_bstr_t(strTemp);
	}
	catch(_com_error &e)
	{
		AfxMessageBox(e.ErrorMessage());
		return;
	}
	catch(COleDispatchException *e)
	{
		char cStr[255];
		e->GetErrorMessage(cStr,255);
		AfxMessageBox(cStr);
		e->Delete();
		return;
	}

	return;
}

void CDataGridEx::OnSortDesc()
{
	CString strTemp;
	_RecordsetPtr IRecordset;
	LPUNKNOWN pUnknown;
	HRESULT hr;

	try
	{
		pUnknown=this->GetDataSource();

		hr=pUnknown->QueryInterface(__uuidof(_RecordsetPtr),(void**)&IRecordset);

		if(FAILED(hr))
		{
			pUnknown->Release();
			AfxMessageBox(_com_error(hr).ErrorMessage());
			return;
		}

		pUnknown->Release();

		strTemp=GetColumns().GetItem(_variant_t(m_SelectCol)).GetCaption();

		strTemp+=_T(" DESC");
		IRecordset->Sort=_bstr_t(strTemp);
	}
	catch(_com_error &e)
	{
		AfxMessageBox(e.ErrorMessage());
		return;
	}
	catch(COleDispatchException *e)
	{
		char cStr[255];
		e->GetErrorMessage(cStr,255);
		AfxMessageBox(cStr);
		e->Delete();
		return;
	}

	
	return;
}

////////////////////////////////////////////
//
// 释放钩子资源，应在窗口销毁前调用
//
void CDataGridEx::UnInitMouseWheel()
{
	tagHookCountInfo* pHookCountInfo;
	HANDLE Handle;

	if(!IsWindow(GetSafeHwnd()))
	{
		return;
	}

	Handle=::GetProp(GetSafeHwnd(),_T("MouseWheel"));

	if(Handle!=NULL)
	{
		RemoveProp(GetSafeHwnd(),_T("MouseWheel"));
	}
	else
	{
		return;
	}

	//
	// 减少使用此线程钩子的计数,当计数为0时
	// 销毁这个线程的钩子,并从map中移除
	//
	if(m_MouseThreadHookMap.Lookup(::GetCurrentThreadId(),pHookCountInfo))
	{
		pHookCountInfo->Count--;

		if(pHookCountInfo->Count==0)
		{
			::UnhookWindowsHookEx(pHookCountInfo->hHook);

			m_MouseThreadHookMap.RemoveKey(::GetCurrentThreadId());

			delete pHookCountInfo;
		}
	}	
}