#ifndef __DTPICKER__H
#define __DTPICKER__H

#pragma once

/////////////////////////////////////////////////////
//	Masked edit controls
//
//	Written by Alexey Shirshov (qqqaqa@mail.ru)
//	Copyright (c) 2003 Alexey Shirshov
//
//
//
//

#ifndef __ATLWIN_H__
#error maskededit.h requires altwin.h to be included first
#endif

#ifndef __ATLMISC_H__
#error maskededit.h requires altmisc.h to be included first
#endif

#include "other\MiniCalendarCtrl.h"

namespace AWTL
{

class CDateTimeCtrl : public CWindowImpl<CDateTimeCtrl,CStatic>,
	public COwnerDraw<CDateTimeCtrl>
{
private:
	typedef CWindowImpl<CDateTimeCtrl,CStatic> _baseClass;

	struct CMaskItem
	{
		CMaskItem()
		{
			data[0] = 0;
			separators[0] = 0;
			rc.SetRectEmpty();
		}
		TCHAR _type[10];
		CRect rc;
		TCHAR data[100];
		TCHAR separators[10];
	};

public:
	DECLARE_WND_CLASS(_T("AWTL_DateTimeCtrl"))
		
	CDateTimeCtrl()
	{
		_xOffset = 0;
		_iActiveItemIdx = 0;
		_fDrawButton = 0;
		fCancel = false;
		fBorder = false;
		Cash[0] = 0;
	}

	BOOL SubclassWindow(HWND hWnd)
	{
#if (defined(_DEBUG) && (WINVER > 0x0400))
		TCHAR buf[100];
		RealGetWindowClass(hWnd,buf,100);
		ATLASSERT(!lstrcmpi(buf,_T("STATIC")));
#endif
		
		BOOL b = _baseClass::SubclassWindow(hWnd);
		Init();
		return b;
	}

	void OnFinalMessage(HWND)
	{
		if (_edit.IsWindow())
			_edit.DestroyWindow();
		if (m_wndCalendar.IsWindow())
			m_wndCalendar.DestroyWindow();
	}

	void SetDateTime(LPSYSTEMTIME pST = NULL)
	{
		Cash[0] = 0;
		for(int i = 0;i < _items.GetSize();i++)
		{
			GetDateFormat(LOCALE_USER_DEFAULT,0,pST,
				_items[i]._type,_items[i].data,100);
			GetTimeFormat(LOCALE_USER_DEFAULT,0,pST,
				_items[i].data,_items[i].data,100);
		}
		Invalidate();
	}

	bool GetDateTime(LPSYSTEMTIME pST)
	{
		ATLASSERT(pST);
		
		TCHAR buf[500];
		GetDateTimeText(buf,500);

		CComVariant v(buf);		
		bool b = SUCCEEDED(v.ChangeType(VT_DATE));
		if (b)
			b = VariantTimeToSystemTime(v.date,pST) == TRUE;
		
		return b;
	}

	bool SetFormat(PTSTR pFormat)
	{
		//TCHAR seps[] = _T(", :/-\\().");
		PCTSTR pCur = pFormat;
		size_t pos = _tcscspn(pCur,GetSeps());
		CMaskItem* mi = 0;
		while(pos){
			ATLASSERT(pos < 5);
			mi = CreateItem(pCur,pCur+pos-1);
			ATLASSERT(mi);
			PCTSTR pNext = _tcscspnn(pCur+pos,GetSeps());
			if (pNext){
				ATLASSERT(pNext-pCur-pos < 10);
				memcpy(mi->separators,pCur+pos,(pNext-pCur-pos)*sizeof(TCHAR));
				mi->separators[pNext-pCur-pos] = 0;
				pCur = pNext;
				pos = _tcscspn(pCur,GetSeps());
			}
			else
				pos = 0;
			_items.Add(*mi);
			delete mi;
		}
/*		mi = CreateItem(pCur,pCur+pos-1);
		if (mi){
			_items.Add(*mi);
			delete mi;
		}*/
		return true;
	}

	bool GetFormat(PTSTR pBuf,size_t& sz,PTSTR pPref = _T(""))
	{
		if (pBuf == NULL) sz = 0;
		size_t cur_sz = 0;

		const size_t pref_sz = lstrlen(pPref);
		for(int i = 0;i < _items.GetSize();i++)
		{
			const size_t fsz = lstrlen(_items[i]._type);
			const size_t ssz = lstrlen(_items[i].separators);
			cur_sz += fsz+ssz+pref_sz;
			if (cur_sz <= sz && pBuf != NULL){
				lstrcat(pBuf,pPref);
				lstrcat(pBuf,_items[i]._type);
				lstrcat(pBuf,_items[i].separators);
			}
		}
		bool b = sz >= cur_sz;
		sz = cur_sz;
		return b;
	}

	bool SetBorder(bool b)
	{
		bool t = fBorder;
		fBorder = b;
		return t;
	}

	virtual PCTSTR GetSeps() const
	{
		return _T(", :;/-\\().");
	}

	static PCTSTR _tcscspnn(PCTSTR pBuf,PCTSTR pSeps)
	{
		for(PCTSTR pF = pBuf;*pF;pF++)
		{
			if (!_tcschr(pSeps,*pF)) return pF;
		}
		return NULL;
	}

protected:
	virtual void Init()
	{		
		_edit.Create(this,2,m_hWnd,&CWindow::rcDefault,
			NULL,WS_CHILD|ES_RIGHT|WS_TABSTOP/*|WS_BORDER*/);
		
		ATLASSERT(_edit.IsWindow());

		_edit.SetMargins(0,0);
		
		//SetFont(AtlGetDefaultGuiFont());
		_edit.SetFont(GetFont());
		
		ModifyStyle(SS_ENHMETAFILE|WS_BORDER, 
			SS_OWNERDRAW|SS_NOTIFY|WS_CLIPSIBLINGS|
			WS_CLIPCHILDREN);

		const DWORD dwStyle = WS_POPUP | 
			MCX_3DBORDER | /*MCX_MULTISELECT | */MCX_HIGHLIGHTTODAY |
			MCX_SHOWNONMONTHDAYS/* | MCX_TODAYBUTTON | MCX_NOSELBUTTON*/;

		CRect r;
		GetClientRect(&r);
		r.OffsetRect(0,r.Height());
		ClientToScreen(&r);
		m_wndCalendar.Create(m_hWnd,r,0,dwStyle);
		m_wndCalendar.SetNotifyWindow(m_hWnd);
		//::SetForegroundWindow(m_wndCalendar);
		
		ATLASSERT(m_wndCalendar.IsWindow());
	}

	int GetDateTimeText(PTSTR pBuf,int iBufSz)
	{
		ATLASSERT(pBuf);
		pBuf[0] = 0;
		const int CashSz = lstrlen(Cash);
		if (CashSz > 0){
			if (CashSz > iBufSz) return CashSz;
			lstrcpy(pBuf,Cash);
			return CashSz;
		}

		int iRealBufSz = 0;
		CRect rc,r;
		GetClientRect(&rc);
		CWindowDC dc(m_hWnd);
		//dc.SelectFont(AtlGetDefaultGuiFont());
		HFONT hOldFont = dc.SelectFont(GetFont());
		int iOffset = 0;
		for(int i = 0;i < _items.GetSize();i++)
		{
			r.SetRectEmpty();
			_items[i].rc.SetRectEmpty();
			dc.DrawText(_items[i].data,-1,&_items[i].rc,
				DT_CALCRECT|DT_SINGLELINE|DT_VCENTER);
			_items[i].rc.top = rc.top;
			_items[i].rc.bottom = rc.bottom;
			_items[i].rc.OffsetRect(iOffset,0);
			dc.DrawText(_items[i].separators,-1,&r,
				DT_CALCRECT|DT_SINGLELINE|DT_VCENTER);
			iOffset = _items[i].rc.right+r.right;
			
			iRealBufSz += lstrlen(_items[i].data);
			const int SepSz = lstrlen(_items[i].separators);
			iRealBufSz += SepSz;
			if (iRealBufSz < iBufSz){
				lstrcat(pBuf,_items[i].data);
				if (SepSz){
					lstrcat(pBuf,_items[i].separators);
				}
			}
		}

		dc.SelectFont(hOldFont);
		
		ATLASSERT(iRealBufSz < 500);
		
		if (iRealBufSz < iBufSz)
			lstrcpy(Cash,pBuf);
		
		return iRealBufSz;
	}

	CMaskItem* CreateItem(PCTSTR pBeg,PCTSTR pEnd)
	{
		TCHAR words[] = _T("yMmdHhst");
		if (!_tcschr(words,pBeg[0])) return NULL;

		for(PCTSTR pf = &pBeg[1];pf <= pEnd;pf++)
		{
			if (*pf != pBeg[0]) return NULL;
		}

		CMaskItem* mi = new CMaskItem;
		const size_t sz = pEnd-pBeg+sizeof(TCHAR);
		memcpy(mi->_type,pBeg,sz*sizeof(TCHAR));
		mi->_type[sz] = 0;
		return mi;
	}

	void ClearAllItems()
	{
		_items.RemoveAll();
	}

	void ActivateItem(int iIdx)
	{
		ATLASSERT(iIdx >= 0 && iIdx < _items.GetSize());
		CRect rc;
		GetClientRect(&rc);
		CRect r = _items[iIdx].rc;
		if (fBorder){
			r.OffsetRect(2,-1);
			r.DeflateRect(0,1);
			r.bottom -= 2;
		}
		_xOffset = rc.right-r.right-rc.Height();
		if (_xOffset > 0) _xOffset = 0;
		Invalidate();
		//CRect r = _items[iIdx].rc;
		
/*		if (lstrlen(_items[iIdx].data) > 2)
			r.right -= 6;

		if (_xOffset < 0)
			r.OffsetRect(-2,0);
*/		
		r.OffsetRect(_xOffset,(rc.Height()<17)?1:2);
		//ClientToScreen(r);
		_edit.SetMargins(0,0);
		_edit.SetWindowPos(NULL,&r,SWP_SHOWWINDOW);
		_edit.SetWindowText(_items[iIdx].data);
//		_edit.ShowWindow(SW_SHOW);
		_edit.SetFocus();
		if (_items[iIdx]._type[0] != _T('t'))
			_edit.ModifyStyle(0,ES_NUMBER);
		else
			_edit.ModifyStyle(ES_NUMBER,0);
		
		_iActiveItemIdx = iIdx;
	}

	int GetItemIdxByPos(POINT p)
	{
		for(int i = 0;i < _items.GetSize();i++)
		{
			const CRect& r = _items[i].rc;
			if (r.PtInRect(p)) return i;
		}
		return -1;
	}

	void NextItem()
	{
		::SetFocus(GetParent());
		if (_iActiveItemIdx+1 == _items.GetSize())
			ActivateItem(0);
		else
			ActivateItem(++_iActiveItemIdx);
	}

	void PrevItem()
	{
		::SetFocus(GetParent());
		if (_iActiveItemIdx == 0)
			ActivateItem(_items.GetSize()-1);
		else
			ActivateItem(--_iActiveItemIdx);
	}

	virtual void ShowCal()
	{
		SetFocus();
		ATLASSERT(m_wndCalendar.IsWindow());
		
		SYSTEMTIME st;
		GetDateTime(&st);
		st.wHour = 0;
		st.wMilliseconds = 0;
		st.wSecond = 0;
		st.wMinute = 0;
		m_wndCalendar.EnsureVisible(st);

		CRect r;
		GetClientRect(&r);
		r.OffsetRect(0,r.Height());
		ClientToScreen(&r);
		m_wndCalendar.SetWindowPos(NULL,&r,SWP_SHOWWINDOW|SWP_NOSIZE);
	}

protected:
	BEGIN_MSG_MAP(CDateTimeCtrl)
		MESSAGE_HANDLER(WM_CREATE,OnCreate)
		//MESSAGE_HANDLER(WM_ERASEBKGND,OnEraseBkgnd)
		MESSAGE_HANDLER(WM_SETFOCUS,OnSetFocus)
		MESSAGE_HANDLER(WM_SETTEXT,OnSetText)
		MESSAGE_HANDLER(WM_GETTEXT,OnGetText)
		MESSAGE_HANDLER(WM_GETTEXTLENGTH, OnGetTextLength)