lyfz_repos/经典版源码/lyfzInternalClient/branches/716/EditListCtrl.cpp

// EditListCtrl.cpp : implementation file
//

#include "stdafx.h"
#include "EditListCtrl.h"
#include <windowsx.h>
#include "ylgl.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#define IDC_EDIT  0X01
#define PROPERTY_ITEM 0x02
#define PROPERTY_SUB  0x03
#define IDC_ComboBox  0X04
 
/////////////////////////////////////////////////////////////////////////////
// CEditListCtrl
extern  CFont g_listctrlfont;
CEditListCtrl::CEditListCtrl()
{
   m_edit.m_hWnd = NULL;
   m_mode=0;
}

CEditListCtrl::~CEditListCtrl()
{

}


BEGIN_MESSAGE_MAP(CEditListCtrl, CListCtrl)
	//{{AFX_MSG_MAP(CEditListCtrl)
	ON_WM_LBUTTONDBLCLK()
	ON_WM_PARENTNOTIFY()
	ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, OnCustomdrawList)
	//}}AFX_MSG_MAP
	ON_MESSAGE(WM_USER_EDIT_END,OnEditEnd) 
	ON_MESSAGE(WM_USER+1003,OnDateEditEnd) 
	ON_MESSAGE(WM_USER_ComboBox_END,OnComboBoxEnd)
END_MESSAGE_MAP()
#undef SubclassWindow  
void CEditListCtrl::PreSubclassWindow()
{
	// the list control must have the report style.
	ASSERT( GetStyle() & LVS_REPORT );

	CListCtrl::PreSubclassWindow();
//	VERIFY( m_ctlHeader.SubclassWindow( GetHeaderCtrl()->GetSafeHwnd() ) );
m_ctlHeader.SubclassWindow( GetHeaderCtrl()->GetSafeHwnd() );

}
void CEditListCtrl::OnCustomdrawList ( NMHDR* pNMHDR, LRESULT* pResult )
{
	NMLVCUSTOMDRAW* pLVCD = reinterpret_cast<NMLVCUSTOMDRAW*>( pNMHDR );
    // Take the default processing unless we set this to something else below.
    *pResult = 0;

    // First thing - check the draw stage. If it's the control's prepaint
    // stage, then tell Windows we want messages for every item.
    if ( CDDS_PREPAINT == pLVCD->nmcd.dwDrawStage )
    {
        *pResult = CDRF_NOTIFYITEMDRAW;
    }
    else if ( CDDS_ITEMPREPAINT == pLVCD->nmcd.dwDrawStage )
        {
        // This is the prepaint stage for an item. Here's where we set the
        // item's text color. Our return value will tell Windows to draw the
        // item itself, but it will use the new color we set here.
        // We'll cycle the colors through red, green, and light blue.
        COLORREF crText;
		int pos=pLVCD->nmcd.dwItemSpec;
		if(pos%2)
		{
			pLVCD->clrTextBk = g_gridcol1;
		}
		else
		{
			pLVCD->clrTextBk = g_gridcol2;
		}
        // Tell Windows to paint the control itself.
        *pResult = CDRF_DODEFAULT;
        }
}

void CEditListCtrl::OnLButtonDblClk(UINT nFlags, CPoint point) 
{
	CRect rcCtrl;
	LVHITTESTINFO lvhti;
    lvhti.pt = point;
	int nItem = CListCtrl::SubItemHitTest(&lvhti);
	if(nItem == -1)
	   return;
	int nSubItem = lvhti.iSubItem;
	CListCtrl::GetSubItemRect(nItem,nSubItem,LVIR_LABEL,rcCtrl); 



	if(nSubItem==2+m_mode)
	{
		ShowEdit(TRUE,nItem,nSubItem,rcCtrl);
	}
	if(nSubItem==3+m_mode)
	{
		ShowEdit(TRUE,nItem,nSubItem,rcCtrl);
	}
	else if(nSubItem==4+m_mode)
	{	 
		rcCtrl.top -=4;
	    rcCtrl.bottom +=200;
		ShowComboBox(TRUE,nItem,nSubItem,rcCtrl);
	}
	else if(nSubItem==5+m_mode)
	{
		ShowDateEdit(TRUE,nItem,nSubItem,rcCtrl);
	}
	CListCtrl::OnLButtonDblClk(nFlags, point);
}

void CEditListCtrl::ShowComboBox(BOOL bShow,int nItem,int nIndex,CRect rcCtrl)
{
    if(m_ComboBox.m_hWnd == NULL)
	{
		m_ComboBox.Create(WS_VSCROLL|WS_CHILD|CBS_DROPDOWNLIST,CRect(0,0,0,0),this,IDC_ComboBox);
		m_ComboBox.ShowWindow(SW_HIDE);
		m_ComboBox.AddString ("¼Ó¼±");
		m_ComboBox.AddString ("");
	}
	if(bShow == TRUE)
	{
 
		CString strItem = CListCtrl::GetItemText(nItem,nIndex);
		m_ComboBox.MoveWindow(rcCtrl);
		m_ComboBox.ShowWindow(SW_SHOW);
		::SetFocus(m_ComboBox.GetSafeHwnd());
 
 
		m_ComboBox.SetCurSel (m_ComboBox.FindString (0, strItem));
		m_ComboBox.SetCtrlData(MAKEWPARAM(nIndex,nItem));	
	}
	else
		m_ComboBox.ShowWindow(SW_HIDE);
}
void CEditListCtrl::ShowDateEdit(BOOL bShow,int nItem,int nIndex,CRect rcCtrl)
{
    if(m_dateedit.m_hWnd == NULL)
	{
		m_dateedit.Create(ES_AUTOHSCROLL|WS_CHILD|ES_LEFT|ES_WANTRETURN|WS_BORDER,CRect(0,0,0,0),this,IDC_EDIT);
		m_dateedit.ShowWindow(SW_HIDE);

		CFont tpFont;
		tpFont.CreateStockObject(DEFAULT_GUI_FONT);
		m_dateedit.SetFont(&tpFont);
		tpFont.DeleteObject();
	}
	if(bShow == TRUE)
	{
		CString strItem = CListCtrl::GetItemText(nItem,nIndex);
		m_dateedit.MoveWindow(rcCtrl);
		m_dateedit.ShowWindow(SW_SHOW);
		m_dateedit.SetWindowText(strItem);
		::SetFocus(m_dateedit.GetSafeHwnd());
		m_dateedit.SetSel(-1);  
		m_dateedit.SetCtrlData(MAKEWPARAM(nIndex,nItem));	
	}
	else
		m_dateedit.ShowWindow(SW_HIDE);
}
void CEditListCtrl::ShowEdit(BOOL bShow,int nItem,int nIndex,CRect rcCtrl)
{
    if(m_edit.m_hWnd == NULL)
	{
		m_edit.Create(ES_AUTOHSCROLL|WS_CHILD|ES_LEFT|ES_WANTRETURN|WS_BORDER,CRect(0,0,0,0),this,IDC_EDIT);
		m_edit.ShowWindow(SW_HIDE);

		CFont tpFont;
		tpFont.CreateStockObject(DEFAULT_GUI_FONT);
		m_edit.SetFont(&tpFont);
		tpFont.DeleteObject();
	}
	if(bShow == TRUE)
	{
		CString strItem = CListCtrl::GetItemText(nItem,nIndex);
		m_edit.MoveWindow(rcCtrl);
		m_edit.ShowWindow(SW_SHOW);
		m_edit.SetWindowText(strItem);
		::SetFocus(m_edit.GetSafeHwnd());
		m_edit.SetSel(-1);  
		m_edit.SetCtrlData(MAKEWPARAM(nIndex,nItem));	
	}
	else
	{
		m_edit.ShowWindow(SW_HIDE);
		GetParent()->SendMessage(WM_USER+100, 0, 0);
	}
}

#ifdef VC_60
void CEditListCtrl::OnDateEditEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		m_dateedit.GetWindowText(strText);
		if(!CheckDateOK(strText))
		{
			strText.Empty ();
			m_dateedit.SetWindowText(strText);
		}
		DWORD dwData = m_dateedit.GetCtrlData();
		int nItem= dwData>>16;
		int nIndex = dwData&0x0000ffff;
		CListCtrl::SetItemText(nItem,nIndex,strText);
	}
	else
	{
	   
	}

    if(lParam == FALSE)
	    m_dateedit.ShowWindow(SW_HIDE);
}
void CEditListCtrl::OnEditEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		m_edit.GetWindowText(strText);
		DWORD dwData = m_edit.GetCtrlData();
		int nItem= dwData>>16;
		int nIndex = dwData&0x0000ffff;
		CListCtrl::SetItemText(nItem,nIndex,strText);
	}
	else
	{
	   
	}
    if(lParam == FALSE)
	{
	    m_edit.ShowWindow(SW_HIDE);
		GetParent()->SendMessage(WM_USER+100, 0, 0);
//	WM_USER+1003
	}
}

void CEditListCtrl::OnComboBoxEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		int pos=m_ComboBox.GetCurSel ();
		if(pos!=-1)
		{
		m_ComboBox.GetLBText (pos, strText);
		DWORD dwData = m_ComboBox.GetCtrlData();
		int nItem= dwData>>16;
		int nIndex = dwData&0x0000ffff;
		CListCtrl::SetItemText(nItem,nIndex,strText);
		}
	}
	else
	{
	   
	}

    if(lParam == FALSE)
	    m_ComboBox.ShowWindow(SW_HIDE);
}
#else
LRESULT CEditListCtrl::OnDateEditEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		m_dateedit.GetWindowText(strText);
		if(!CheckDateOK(strText))
		{
			strText.Empty ();
			m_dateedit.SetWindowText(strText);
		}
		DWORD dwData = m_dateedit.GetCtrlData();
		int nItem= dwData>>16;
		int nIndex = dwData&0x0000ffff;
		CListCtrl::SetItemText(nItem,nIndex,strText);
	}
	else
	{

	}

	if(lParam == FALSE)
		m_dateedit.ShowWindow(SW_HIDE);

	return 0;
}
LRESULT CEditListCtrl::OnEditEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		m_edit.GetWindowText(strText);
		DWORD dwData = m_edit.GetCtrlData();
		int nItem= dwData>>16;
		int nIndex = dwData&0x0000ffff;
		CListCtrl::SetItemText(nItem,nIndex,strText);
	}
	else
	{

	}
	if(lParam == FALSE)
	{
		m_edit.ShowWindow(SW_HIDE);
		GetParent()->SendMessage(WM_USER+100, 0, 0);
		//	WM_USER+1003
	}

	return 0;
}

LRESULT CEditListCtrl::OnComboBoxEnd(WPARAM wParam,LPARAM lParam)
{
	if(wParam == TRUE)
	{
		CString strText(_T(""));
		int pos=m_ComboBox.GetCurSel ();
		if(pos!=-1)
		{
			m_ComboBox.GetLBText (pos, strText);
			DWORD dwData = m_ComboBox.GetCtrlData();
			int nItem= dwData>>16;
			int nIndex = dwData&0x0000ffff;
			CListCtrl::SetItemText(nItem,nIndex,strText);
		}
	}
	else
	{

	}

	if(lParam == FALSE)
		m_ComboBox.ShowWindow(SW_HIDE);

	return 0;
}
#endif
void CEditListCtrl::OnParentNotify(UINT message, LPARAM lParam) 
{
	CListCtrl::OnParentNotify(message, lParam);
	//////////////////////////////////////////////////////////////////////////
	CHeaderCtrl* pHeaderCtrl = CListCtrl::GetHeaderCtrl();
	if(pHeaderCtrl == NULL)
		return;
	CRect rcHeader;
	pHeaderCtrl->GetWindowRect(rcHeader);
	ScreenToClient(rcHeader);
	//The x coordinate is in the low-order word and the y coordinate is in the high-order word.
	CPoint pt;
	pt.x = GET_X_LPARAM(lParam);
	pt.y = GET_Y_LPARAM(lParam);
	if(rcHeader.PtInRect(pt) && message == WM_LBUTTONDOWN)
	{
		if(m_edit.m_hWnd != NULL)
		{
			DWORD dwStyle = m_edit.GetStyle();
			if((dwStyle&WS_VISIBLE) == WS_VISIBLE)
			{
				m_edit.ShowWindow(SW_HIDE);
			}
		}	

		if(m_ComboBox.m_hWnd != NULL)
		{
			DWORD dwStyle = m_ComboBox.GetStyle();
			if((dwStyle&WS_VISIBLE) == WS_VISIBLE)
			{
				m_ComboBox.ShowWindow(SW_HIDE);
			}
		}
	}	
}
BOOL CEditListCtrl::PreTranslateMessage(MSG* pMsg) 
{
	if(0)//pMsg->message == WM_KEYDOWN)
	{
		if(pMsg->wParam == VK_TAB && m_edit.m_hWnd!= NULL)
		{
		 	DWORD dwStyle = m_edit.GetStyle();
			if((dwStyle&WS_VISIBLE) == WS_VISIBLE)
			{
			     OnEditEnd(TRUE,TRUE);
				 CRect rcCtrl;		 
				 int nItem;
				 int nSub;
				 if(FALSE == Key_Ctrl(nItem,nSub))
                    Key_Shift(nItem,nSub);
                 CListCtrl::GetSubItemRect(nItem,nSub,LVIR_LABEL,rcCtrl); 
				 CPoint pt(rcCtrl.left+1,rcCtrl.top+1);
				 OnLButtonDblClk(0,pt);
				 POSITION pos = CListCtrl::GetFirstSelectedItemPosition();
				 if (pos == NULL)
				 {
				 }
				 else
				 {
					 while (pos)
					 {
						 int ntpItem = CListCtrl::GetNextSelectedItem(pos);
						 CListCtrl::SetItemState(ntpItem,0,LVIS_SELECTED);
					 }
				 }
				 CListCtrl::SetItemState(nItem,  LVIS_SELECTED,  LVIS_SELECTED);
				 return TRUE;
			}
		}
	}
	
	return CListCtrl::PreTranslateMessage(pMsg);
}
BOOL CEditListCtrl::Key_Shift(int& nItem,int& nSub)
{
	int nItemCount = CListCtrl::GetItemCount();
	DWORD dwData = m_edit.GetCtrlData();
	nItem= dwData>>16;
	nSub = dwData&0x0000ffff;
	
	CHeaderCtrl* pHeader = CListCtrl::GetHeaderCtrl();
	if(pHeader == NULL)
		return FALSE;
	
	short sRet = GetKeyState(VK_SHIFT);
	int nSubcCount = pHeader->GetItemCount();
	sRet = sRet >>15;
	if(sRet == 0)
	{
		nSub += 1;
		if(nSub >= nSubcCount)
		{
			if(nItem == nItemCount-1)
			{
				nItem = 0;
				nSub  = 0;
			}
			else
			{
				nSub = 0;
				nItem += 1;
			}
		}
		if(nItem >= nItemCount)
			nItem = nItemCount-1;
		return FALSE;
	}
	else
	{
		nSub -= 1;
		if(nSub < 0)
		{
			
			nSub = nSubcCount -1;
			nItem --;
		}
		
		if(nItem < 0)
			nItem = nItemCount-1;
		return TRUE;
		
	}
	return FALSE;
}
BOOL CEditListCtrl::Key_Ctrl(int& nItem,int &nSub)
{
    short sRet = GetKeyState(VK_CONTROL);
	DWORD dwData = m_edit.GetCtrlData();
	nItem= dwData>>16;
	nSub = dwData&0x0000ffff;
	
	sRet = sRet >>15;
	int nItemCount = CListCtrl::GetItemCount();
	if(sRet == 0)
	{
		
	}
	else
	{
		nItem = nItem >=nItemCount-1? 0:nItem+=1;
		return TRUE;
	}
	
	return FALSE;
}
//////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////
//////////////////////////////////////////////////////////////////////////

CListCtrlEdit::CListCtrlEdit()
{
}

CListCtrlEdit::~CListCtrlEdit()
{
}


BEGIN_MESSAGE_MAP(CListCtrlEdit, CEdit)
	//{{AFX_MSG_MAP(CListCtrlEdit)
	ON_WM_KILLFOCUS()
	ON_WM_SETFOCUS()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CListCtrlEdit message handlers
void CListCtrlEdit::SetCtrlData(DWORD dwData)
{
	m_dwData = dwData;
}

DWORD CListCtrlEdit::GetCtrlData()
{
   return m_dwData;
}

void CListCtrlEdit::OnKillFocus(CWnd* pNewWnd) 
{
	CEdit::OnKillFocus(pNewWnd);
	CWnd* pParent = this->GetParent();
	::PostMessage(pParent->GetSafeHwnd(),WM_USER_EDIT_END,m_bExchange,0);
}
BOOL CListCtrlEdit::PreTranslateMessage(MSG* pMsg) 
{
	
	if(pMsg->message == WM_KEYDOWN)
	{
		if(pMsg->wParam == VK_RETURN)
		{
			CWnd* pParent = this->GetParent();
			m_bExchange = TRUE;
			::PostMessage(pParent->GetSafeHwnd(),WM_USER_EDIT_END,m_bExchange,0);
		}
		else if(pMsg->wParam == VK_ESCAPE)
		{
           CWnd* pParent = this->GetParent();
		   m_bExchange = FALSE;
			::PostMessage(pParent->GetSafeHwnd(),WM_USER_EDIT_END,m_bExchange,0);
		}
	}
	
	return CEdit::PreTranslateMessage(pMsg);
}

void CListCtrlEdit::OnSetFocus(CWnd* pOldWnd) 
{
	CEdit::OnSetFocus(pOldWnd);
    m_bExchange = TRUE;	
}

void CEditListCtrl::InitStyle()
{
	SetFont (&g_listctrlfont);
    SetExtendedStyle(LVS_EX_FLATSB|LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES|LVS_EX_HEADERDRAGDROP);
}

    BOOL CEditListCtrl::SetItemText(int nItem, int nSubItem, LPCTSTR lpszText)  
    {  
        ASSERT(::IsWindow(m_hWnd));  
        ASSERT((GetStyle() & LVS_OWNERDATA)==0);  
        LVITEM lvi;  
        lvi.iSubItem = nSubItem;  
        lvi.pszText = (LPTSTR)lpszText;  
        ::SendMessage(m_hWnd, LVM_SETITEMTEXT, nItem, (LPARAM)&lvi);  
        return (BOOL) ::SendMessage(m_hWnd, LVM_SETITEMTEXT, nItem, (LPARAM)&lvi);  
    }  
      
      
    CString CEditListCtrl::GetItemText(int nItem, int nSubItem)  
    {  
        ASSERT(::IsWindow(m_hWnd));  
        LVITEM lvi;  
        memset(&lvi, 0, sizeof(LVITEM));  
        lvi.iSubItem = nSubItem;  
        CString str;  
        int nLen = 1024*4;    //  
        int nRes;  
        do  
        {  
            nLen *= 2;  
            lvi.cchTextMax = nLen;  
            lvi.pszText = str.GetBufferSetLength(nLen);  
            nRes  = (int)::SendMessage(m_hWnd, LVM_GETITEMTEXT, (WPARAM)nItem,(LPARAM)&lvi);  
        } while (nRes == nLen-1);  
        str.ReleaseBuffer();  
        return str;  
    }