// 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);
	}
}
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);
}
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;
}