path: root/plugins/Quotes/src/QuoteChart.cpp

#include "StdAfx.h"

#ifdef CHART_IMPLEMENT

#define WINDOW_PREFIX "Quotes Chart_"
#define CHART_CTRL_CLASS _T("DioksinChart")

namespace
{
	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(_T("%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(_T("%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('\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,NULL);
				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:
			{
				MCONTACT hContact = reinterpret_cast<HANDLE>(lp);

				TranslateDialogDefault(hDlg);

				tstring sName = get_window_text(hDlg);
				sName += _T(" - ");
				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[] = {LPGENT("Log File"), LPGENT("Miranda's History")};
				static LPCTSTR szFilters[] = {LPGENT("All"), LPGENT("Last Day"), LPGENT("Last Week"), LPGENT("Last Month"), LPGENT("Last Year"), LPGENT("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,_T("open"),pNMLink->item.szUrl,NULL,NULL,SW_SHOWNORMAL);					
					}
					break;
				}
			}
			break;
// 		case WM_ERASEBKGND:
// 			{
// 				HDC hdc = reinterpret_cast<HDC>(wp);
// 				TChart* pChart = get_chart_ptr(hDlg);
// 				pChart->DrawBackground(hdc);
// 				return TRUE;
// 			}
// 			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