#include "StdAfx.h"

#ifdef CHART_IMPLEMENT

#define WINDOW_PREFIX "Quotes Chart_"
#define CHART_CTRL_CLASS L"DioksinChart"

struct CTimeConvert
{
	static double Convert(const boost::posix_time::time_duration& v)
	{
		return boost::numeric_cast<double>(v.ticks());
	}

	static tstring ToString(const boost::posix_time::ptime& v)
	{
		tostringstream k;
		k.imbue(std::locale(GetSystemLocale(), new ttime_facet(L"%d/%m/%y %H:%M:%S")));
		k << v;
		return k.str();
	}
};

typedef CChart<boost::posix_time::ptime, double, CTimeConvert> TChart;

inline TChart* get_chart_ptr(HWND hWnd)
{
	TChart* pChart = reinterpret_cast<TChart*>(GetWindowLongPtr(hWnd, GWLP_USERDATA));
	return pChart;
}

bool read_log_file(MCONTACT hContact, TChart& rChart)
{
	tstring sLogFileName = GetContactLogFileName(hContact);
	if (false == sLogFileName.empty()) {
		std::locale loc(GetSystemLocale(), new ttime_input_facet(L"%d.%m.%y %H:%M:%S"));
		boost::posix_time::ptime oDateTime;
		double dRate;

		tifstream file(sLogFileName.c_str());
		file.imbue(loc);
		while ((false == file.fail()) && (false == file.eof())) {
			tstring sLine;
			std::getline(file, sLine);

			tistringstream line(sLine);
			line.imbue(loc);

			tstring sName;
			std::getline(line, sName, '\t');
			line >> oDateTime >> dRate;
			if ((false == line.fail()) && (true == line.eof())) {
				rChart.AddValue(oDateTime, dRate);
			}
		}

		return true;
	}
	return false;
}


enum
{
	ID_CHART = 0x1969,

	srcLogFile = 0,
	srcHistory = 1,

	filterAll = 0,
	filterLastDay = 1,
	filterLastWeek = 2,
	filterLastMonth = 3,
	filterLastYear = 4,
	filterUserDefined = 5,

	CHART_SET_SOURCE = WM_USER + 1,
	CHART_SET_FILTER = WM_USER + 2,
};

LRESULT CALLBACK ChartWndProc(HWND hWnd, UINT msg, WPARAM wp, LPARAM lp)
{
	switch (msg) {
	case WM_CREATE:
		{
			CREATESTRUCT* pCS = reinterpret_cast<CREATESTRUCT*>(lp);
			MCONTACT hContact = reinterpret_cast<HANDLE>(pCS->lpCreateParams);

			TChart* pChart = new TChart;
			read_log_file(hContact, *pChart);

			::SetWindowLongPtr(hWnd, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(pChart));
		}
		return 0;

	case CHART_SET_SOURCE:
		break;

	case CHART_SET_FILTER:
		break;

	case WM_SIZE:
		{
			TChart* pChart = get_chart_ptr(hWnd);
			pChart->SetRect(0, 0, LOWORD(lp), HIWORD(lp));
		}
		return 0;

	case WM_PAINT:
		if (TRUE == ::GetUpdateRect(hWnd, NULL, FALSE)) {
			PAINTSTRUCT ps;
			HDC hdc = ::BeginPaint(hWnd, &ps);
			if (NULL != hdc) {
				TChart* pChart = get_chart_ptr(hWnd);
				pChart->Draw(hdc);
				::EndPaint(hWnd, &ps);
			}
		}

		return 0;
	case WM_DESTROY:
		{
			TChart* pChart = get_chart_ptr(hWnd);
			::SetWindowLongPtr(hWnd, GWLP_USERDATA, 0);
			delete pChart;
		}
		break;
	}

	return ::DefWindowProc(hWnd, msg, wp, lp);
}

void register_chart_control()
{
	static bool g_bRegister = false;
	if (g_bRegister) {
		return;
	}

	WNDCLASS wc;

	wc.style = CS_HREDRAW | CS_VREDRAW;
	wc.lpfnWndProc = ChartWndProc;
	wc.cbClsExtra = 0;
	wc.cbWndExtra = 0;
	wc.hInstance = CModuleInfo::GetInstance().GetModuleHandle();
	wc.hIcon = NULL;
	wc.hCursor = LoadCursor(NULL, IDC_ARROW);
	wc.hbrBackground = static_cast<HBRUSH>(::GetStockObject(WHITE_BRUSH));
	wc.lpszMenuName = NULL;
	wc.lpszClassName = CHART_CTRL_CLASS;

	if (RegisterClass(&wc)) {
		g_bRegister = true;
	}
}


bool screen_2_client(HWND hWnd, LPRECT pRect)
{
	POINT pt;
	pt.x = pRect->left;
	pt.y = pRect->top;
	bool bResult = TRUE == ::ScreenToClient(hWnd, &pt);
	pRect->left = pt.x;
	pRect->top = pt.y;
	pt.x = pRect->right;
	pt.y = pRect->bottom;
	bResult |= TRUE == ::ScreenToClient(hWnd, &pt);
	pRect->right = pt.x;
	pRect->bottom = pt.y;
	return bResult;
}

inline HANDLE get_contact(HWND hWnd)
{
	MCONTACT hContact = reinterpret_cast<HANDLE>(GetWindowLongPtr(hWnd, GWLP_USERDATA));
	return hContact;
}

void update_filter_controls(HWND hDlg)
{
	int nSel = ::SendDlgItemMessage(hDlg, IDC_COMBO_FILTER, CB_GETCURSEL, 0, 0);

	::ShowWindow(::GetDlgItem(hDlg, IDC_EDIT_FROM), (filterUserDefined == nSel) ? SW_SHOW : SW_HIDE);
	::ShowWindow(::GetDlgItem(hDlg, IDC_EDIT_TO), (filterUserDefined == nSel) ? SW_SHOW : SW_HIDE);
}

INT_PTR CALLBACK ChartDlgProc(HWND hDlg, UINT msg, WPARAM wp, LPARAM lp)
{
	switch (msg) {
	case WM_INITDIALOG:
		TranslateDialogDefault(hDlg);
		{
			MCONTACT hContact = reinterpret_cast<HANDLE>(lp);

			tstring sName = get_window_text(hDlg);
			sName += L" - ";
			sName += GetContactName(hContact);
			::SetWindowText(hDlg, sName.c_str());

			HANDLE hWL = CModuleInfo::GetInstance().GetWindowList(WINDOW_PREFIX, false);
			assert(hWL);
			WindowList_Add(hWL, hDlg, hContact);

			::SetWindowLongPtr(hDlg, GWLP_USERDATA, reinterpret_cast<LONG_PTR>(hContact));

			static LPCTSTR szSources[] = { LPGENW("Log File"), LPGENW("Miranda's History") };
			static LPCTSTR szFilters[] = { LPGENW("All"), LPGENW("Last Day"), LPGENW("Last Week"), LPGENW("Last Month"), LPGENW("Last Year"), LPGENW("User-Defined") };

			for (int i = 0; i < sizeof(szSources) / sizeof(szSources[0]); ++i) {
				LPCTSTR p = TranslateTS(szSources[i]);
				::SendDlgItemMessage(hDlg, IDC_COMBO_DATA_SOURCE, CB_INSERTSTRING, -1, reinterpret_cast<LPARAM>(p));
			}

			int nSel = db_get_b(hContact, QUOTES_PROTOCOL_NAME, "Chart_Source", srcLogFile);
			::SendDlgItemMessage(hDlg, IDC_COMBO_DATA_SOURCE, CB_SETCURSEL, nSel, 0);

			for (int i = 0; i < sizeof(szFilters) / sizeof(szFilters[0]); ++i) {
				LPCTSTR p = TranslateTS(szSources[i]);
				::SendDlgItemMessage(hDlg, IDC_COMBO_FILTER, CB_INSERTSTRING, -1, reinterpret_cast<LPARAM>(szFilters[i]));
			}

			nSel = db_get_b(hContact, QUOTES_PROTOCOL_NAME, "Chart_Filter", filterAll);
			::SendDlgItemMessage(hDlg, IDC_COMBO_FILTER, CB_SETCURSEL, nSel, 0);

			update_filter_controls(hDlg);

			register_chart_control();
			HWND hwndImage = ::GetDlgItem(hDlg, IDC_STATIC_IMAGE);
			RECT rcImage;
			::GetWindowRect(hwndImage, &rcImage);
			screen_2_client(hDlg, &rcImage);
			//BOOL bResult = ShowWindow(hwndImage,SW_HIDE);
			//assert(bResult);

			HWND hChart = ::CreateWindowEx(0L, CHART_CTRL_CLASS, NULL, WS_CHILDWINDOW | WS_VISIBLE,
				rcImage.left, rcImage.top, rcImage.right - rcImage.left, rcImage.bottom - rcImage.top,
				hDlg, reinterpret_cast<HMENU>(ID_CHART), CModuleInfo::GetInstance().GetModuleHandle(), hContact);
			assert(NULL != hChart);

			Utils_RestoreWindowPosition(hDlg, hContact, QUOTES_MODULE_NAME, WINDOW_PREFIX);
			BOOL bResult = ::ShowWindow(hDlg, SW_SHOW);
			assert(bResult);
		}
		return (TRUE);
	case WM_CLOSE:
		{
			MCONTACT hContact = get_contact(hDlg);
			SetWindowLongPtr(hDlg, GWLP_USERDATA, 0);

			// 				save_options(hDlg,hContact);

			HANDLE hWL = CModuleInfo::GetInstance().GetWindowList(WINDOW_PREFIX, false);
			assert(hWL);
			WindowList_Remove(hWL, hDlg);
			Utils_SaveWindowPosition(hDlg, hContact, QUOTES_MODULE_NAME, WINDOW_PREFIX);

			HWND hwndChart = ::GetDlgItem(hDlg, ID_CHART);
			BOOL bResult = ::DestroyWindow(hwndChart);
			assert(bResult);

			::EndDialog(hDlg, 0);
		}
		return (TRUE);

	case WM_COMMAND:
		switch (LOWORD(wp)) {
		case IDCANCEL:
			SendMessage(hDlg, WM_CLOSE, 0, 0);
			return (TRUE);

		case IDC_COMBO_FILTER:
			if (CBN_SELCHANGE == HIWORD(wp)) {
				::SendDlgItemMessage(hDlg, ID_CHART, CHART_SET_FILTER, ::SendDlgItemMessage(hDlg, IDC_COMBO_FILTER, CB_GETCURSEL, 0, 0), 0);
				update_filter_controls(hDlg);
			}
			break;
		case IDC_COMBO_DATA_SOURCE:
			if (CBN_SELCHANGE == HIWORD(wp))
				::SendDlgItemMessage(hDlg, ID_CHART, CHART_SET_SOURCE, ::SendDlgItemMessage(hDlg, IDC_COMBO_DATA_SOURCE, CB_GETCURSEL, 0, 0), 0);
			break;
		}
		return (FALSE);

	case WM_NOTIFY:
		{
			LPNMHDR pNMHDR = reinterpret_cast<LPNMHDR>(lp);
			switch (pNMHDR->code) {
			case NM_CLICK:
				if (IDC_SYSLINK_PROVIDER == wp) {
					PNMLINK pNMLink = reinterpret_cast<PNMLINK>(pNMHDR);
					::ShellExecute(hDlg, L"open", pNMLink->item.szUrl, NULL, NULL, SW_SHOWNORMAL);
				}
				break;
			}
		}
		break;

	case WM_SIZE:
		{
			enum { INDENT = 7 };

			int nWidth = LOWORD(lp);
			int nHeight = HIWORD(lp);

			HWND hwndChart = GetDlgItem(hDlg, ID_CHART);
			HWND hwndLink = GetDlgItem(hDlg, IDC_SYSLINK_PROVIDER);
			HWND hwndClose = GetDlgItem(hDlg, IDCANCEL);

			RECT rcDlg;
			GetClientRect(hDlg, &rcDlg);

			RECT rcChart;
			GetWindowRect(hwndChart, &rcChart);
			screen_2_client(hDlg, &rcChart);

			RECT rcLink;
			GetWindowRect(hwndLink, &rcLink);
			screen_2_client(hDlg, &rcLink);
			SetWindowPos(hwndLink, NULL, rcDlg.left + INDENT,
				rcDlg.bottom - INDENT - (rcLink.bottom - rcLink.top),
				0, 0, SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE);

			RECT rcClose;
			GetWindowRect(hwndClose, &rcClose);
			screen_2_client(hDlg, &rcClose);
			SetWindowPos(hwndClose, NULL, rcDlg.right - INDENT - (rcClose.right - rcClose.left),
				rcDlg.bottom - INDENT - (rcClose.bottom - rcClose.top),
				0, 0, SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE);

			SetWindowPos(hwndChart, NULL, rcDlg.left + INDENT,
				rcChart.top,
				(nWidth - INDENT * 2),
				nHeight - (rcClose.bottom - rcClose.top) - INDENT * 2 - rcChart.top,
				SWP_NOZORDER | SWP_NOACTIVATE);

		}
		break;
	}
	return (FALSE);
}

INT_PTR QuotesMenu_Chart(WPARAM wp, LPARAM lp)
{
	MCONTACT hContact = reinterpret_cast<HANDLE>(wp);
	if (NULL == hContact)
		return 0;

	HANDLE hWL = CModuleInfo::GetInstance().GetWindowList(WINDOW_PREFIX, true);
	assert(hWL);
	HWND hWnd = WindowList_Find(hWL, hContact);
	if (NULL != hWnd) {
		SetForegroundWindow(hWnd);
		SetFocus(hWnd);
	}
	else CreateDialogParam(CModuleInfo::GetModuleHandle(), MAKEINTRESOURCE(IDD_DUKASCOPY_CHART), NULL, ChartDlgProc, reinterpret_cast<LPARAM>(hContact));

	return 0;
}

#endif //CHART_IMPLEMENT