#include "stdafx.h"
#include "UIIInplaceEdit.h"

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

#ifndef WM_INPLACEKILLFOCUS 
#define WM_INPLACEKILLFOCUS (WM_USER+0x1234)
#endif

CUIIInplaceEdit::CUIIInplaceEdit()
{
	m_nItem		= -1;
	m_nSubItem	= -1;
	m_sContent	= "";
	m_bVK_ESCAPE = FALSE; 
}

CUIIInplaceEdit::CUIIInplaceEdit(int nItem, int nSubItem, CString &sContent)
{
	m_nItem		= nItem;
	m_nSubItem	= nSubItem;
	m_sContent	= sContent;
	m_bVK_ESCAPE = FALSE; 
}

CUIIInplaceEdit::~CUIIInplaceEdit()
{

}

BEGIN_MESSAGE_MAP(CUIIInplaceEdit, CEdit)
	//{{AFX_MSG_MAP(CUIIInplaceEdit)
	ON_WM_CHAR()
	ON_WM_CREATE()
	ON_WM_KILLFOCUS()
	ON_WM_NCDESTROY()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

void CUIIInplaceEdit::OnChar(UINT nChar, UINT nRepCnt, UINT nFlags) 
{
	if( nChar == VK_ESCAPE ||
		nChar == VK_RETURN)	
	{		
		if( nChar == VK_ESCAPE)
			m_bVK_ESCAPE = 1;		
		GetParent()->SetFocus();
		return;	
	}
	CEdit::OnChar(nChar, nRepCnt, nFlags);
	
	// Resize edit control if needed
	// Get text extent	
	CString str;	
	GetWindowText( str );	

	CWindowDC dc(this);
	CFont *pFont = GetParent()->GetFont();
	CFont *pFontDC = dc.SelectObject(pFont);
	CSize size = dc.GetTextExtent(str);	
	dc.SelectObject(pFontDC);
	size.cx += 5; // add some extra buffer	

	// Get client rect
	CRect rect, rcParent;	
	GetClientRect(&rect);
	GetParent()->GetClientRect(&rcParent);

	// Transform rect to parent coordinates	
	ClientToScreen(&rect);
	GetParent()->ScreenToClient(&rect);
	// Check whether control needs to be resized
	// and whether there is space to grow	
	if(size.cx > rect.Width())	
	{
		if(size.cx + rect.left < rcParent.right)
			rect.right = rect.left + size.cx;		
		else			
			rect.right = rcParent.right;
		MoveWindow(&rect);	
	}
}

int CUIIInplaceEdit::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (CEdit::OnCreate(lpCreateStruct) == -1)
		return -1;

	CFont* font = GetParent()->GetFont();
	SetFont(font);
	SetWindowText(m_sContent);
	SetFocus();	
	SetSel(0, 0);

	return 0;
}

void CUIIInplaceEdit::OnKillFocus(CWnd* pNewWnd) 
{
	CEdit::OnKillFocus(pNewWnd);
	
	CString str;
	GetWindowText(str);

	LV_DISPINFO lvDispInfo;
	lvDispInfo.hdr.hwndFrom = GetParent()->m_hWnd;
	lvDispInfo.hdr.idFrom = GetDlgCtrlID();	
	lvDispInfo.hdr.code = LVN_ENDLABELEDIT;
	lvDispInfo.item.mask = LVIF_TEXT;	
	lvDispInfo.item.iItem = m_nItem;
	lvDispInfo.item.iSubItem = m_nSubItem;
	lvDispInfo.item.pszText = m_bVK_ESCAPE ? NULL : LPTSTR((LPCTSTR)str);
	lvDispInfo.item.cchTextMax = str.GetLength();
	int nCtrlID = GetParent()->GetDlgCtrlID();
	GetParent()->GetParent()->SendMessage( WM_NOTIFY,nCtrlID,(LPARAM)&lvDispInfo);
	GetParent()->GetParent()->SendMessage( WM_INPLACEKILLFOCUS,nCtrlID,(LPARAM)&lvDispInfo);
	DestroyWindow();
}

void CUIIInplaceEdit::OnNcDestroy() 
{
	CEdit::OnNcDestroy();
	delete this;
}

BOOL CUIIInplaceEdit::PreTranslateMessage(MSG* pMsg) 
{
	if( pMsg->message == WM_KEYDOWN )	
	{		
		if( pMsg->wParam == VK_RETURN || 
			pMsg->wParam == VK_DELETE ||
			pMsg->wParam == VK_ESCAPE ||
			GetKeyState( VK_CONTROL))
		{			
			::TranslateMessage(pMsg);
			::DispatchMessage(pMsg);
			return 1;
		}	
	}	
	return CEdit::PreTranslateMessage(pMsg);
}