#include "common.h"
#include "pinggraph.h"

HistoryMap history_map;

#define ID_REPAINT_TIMER		10101

struct WindowData {
	DWORD item_id;
	HistoryList list;
	HWND hwnd_chk_grid;
	HWND hwnd_chk_stat;
	bool show_grid;
	bool show_stat;
};

#define WM_REBUILDLIST			(WM_USER + 5)

LRESULT CALLBACK GraphWindowProc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
	switch(msg) {
	case WM_REBUILDLIST:
		{
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);

			bool found = false;
			EnterCriticalSection(&data_list_cs);
			for(PINGLIST::Iterator i = data_list.start(); i.has_val(); i.next()) {
				if(i.val().item_id == wd->item_id) {
					wd->list = history_map[wd->item_id];
					found = true;
					break;
				}
			}
			LeaveCriticalSection(&data_list_cs);

			if(!found) {
				PostMessage(hwnd, WM_CLOSE, 0, 0);
				return TRUE;
			}

			InvalidateRect(hwnd, 0, FALSE);
		}
		return TRUE;
	case WM_SHOWWINDOW:
		if(wParam == TRUE && lParam == 0) {
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);

			if(wd->hwnd_chk_grid == 0) {
				wd->hwnd_chk_grid = CreateWindow(_T("BUTTON"), TranslateT("Show grid lines"), WS_CHILD | WS_VISIBLE | BS_AUTOCHECKBOX, 0, 0, 0, 0, hwnd, 0, hInst, 0);
				SendMessage(wd->hwnd_chk_grid, BM_SETCHECK, wd->show_grid ? BST_CHECKED : BST_UNCHECKED, 0);
			}
			if(wd->hwnd_chk_stat == 0) {
				wd->hwnd_chk_stat = CreateWindow(_T("BUTTON"), TranslateT("Show stats"), WS_CHILD | WS_VISIBLE | BS_AUTOCHECKBOX, 0, 0, 0, 0, hwnd, 0, hInst, 0);
				SendMessage(wd->hwnd_chk_stat, BM_SETCHECK, wd->show_stat ? BST_CHECKED : BST_UNCHECKED, 0);
			}
			KillTimer(hwnd, ID_REPAINT_TIMER);
#ifdef min
			SetTimer(hwnd, ID_REPAINT_TIMER, min(options.ping_period * 1000, 5000), 0);
#else
			SetTimer(hwnd, ID_REPAINT_TIMER, std::min(options.ping_period * 1000, 5000), 0);
#endif

			SendMessage(hwnd, WM_REBUILDLIST, 0, 0);
		}
		break;
	case WM_COMMAND:
		if(HIWORD(wParam) == BN_CLICKED) {
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);
			if((HWND)lParam == wd->hwnd_chk_grid) {
				wd->show_grid = (SendMessage(wd->hwnd_chk_grid, BM_GETCHECK, 0, 0) == BST_CHECKED);
			} else if((HWND)lParam == wd->hwnd_chk_stat) {
				wd->show_stat = (SendMessage(wd->hwnd_chk_stat, BM_GETCHECK, 0, 0) == BST_CHECKED);
			}
			InvalidateRect(hwnd, 0, TRUE);
		}
		return TRUE;
	case WM_TIMER:
		{
			SendMessage(hwnd, WM_REBUILDLIST, 0, 0);
		}
		return TRUE;
	case WM_PAINT:
		{
			PAINTSTRUCT ps;
			HDC hdc;
			RECT r;
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);
			if(wd && (hdc = BeginPaint(hwnd, &ps)) != 0) {
				GetClientRect(hwnd, &r);
				FillRect(hdc, &r, GetSysColorBrush(COLOR_WINDOW));

				short max_value = -1, min_value = (short)0x7FFF,
					graph_height = 0; // this is minimum graph height, in ms
				double avg = 0;
				for(HistoryList::Iterator hli = wd->list.start(); hli.has_val(); hli.next()) {
					if(hli.val().first > max_value) max_value = hli.val().first;
					if(hli.val().first >= 0 && hli.val().first < min_value) min_value = hli.val().first;
					avg += hli.val().first;
				}
				if(wd->list.size())
					avg /= wd->list.size();

				graph_height = (int)(max_value * 1.2f);
				if(graph_height < MIN_GRAPH_HEIGHT) graph_height = MIN_GRAPH_HEIGHT;

#ifdef max
				float unit_width = (r.right - r.left) / (float)max((int)wd->list.size(), MIN_BARS), // space for at least MIN_BARS bars
#else
				float unit_width = (r.right - r.left) / (float)std::max((int)wd->list.size(), MIN_BARS), // space for at least MIN_BARS bars
#endif
					unit_height = (r.bottom - r.top) / (float)graph_height;

				time_t last_time = 0, time, start_time = 0;
				if(wd->list.size())
					start_time = wd->list.start().val().second;

				RECT bar;
				bar.bottom = r.bottom;
				float x = r.right - (unit_width * wd->list.size() + 0.5f);
				bar.left = (int)(x + 0.5f);
				bar.right = (int)(x + unit_width + 0.5f);

				// set up pen for horiz (ping time) and vert (time covered by graph) lines
				HPEN hPenOld, hPen = CreatePen(PS_SOLID, 1, GetSysColor(COLOR_3DDKSHADOW));
				hPenOld = (HPEN)SelectObject(hdc, hPen);

				for(HistoryList::Iterator hi = wd->list.start(); hi.has_val(); hi.next()) {
					if(hi.val().first != -1) {
						bar.top = bar.bottom - (int)(hi.val().first * unit_height + 0.5f);
						FillRect(hdc, &bar, GetSysColorBrush(COLOR_HOTLIGHT));
					}

					time = hi.val().second - start_time;
					
					if(time / MARK_PERIOD != last_time / MARK_PERIOD) { // new minute
						MoveToEx(hdc, bar.left, r.bottom, 0);
						LineTo(hdc, bar.left, r.top);
					}

					last_time = time;

					x += unit_width;
					bar.left = bar.right;
					bar.right = (int)(x + unit_width + 0.5f);
				}

				if(wd->show_grid) {
					// draw horizontal lines to mark every 100ms
					for(int li = 0; li < graph_height; li += MARK_TIME) {
						MoveToEx(hdc, r.left, r.bottom - (int)(li * unit_height + 0.5f), 0);
						LineTo(hdc, r.right, r.bottom - (int)(li * unit_height + 0.5f));
					}
				}
				 
				HPEN hPen2 = CreatePen(PS_SOLID, 1, RGB(255, 0, 0));
				if(wd->show_stat) {
					SelectObject(hdc, hPen2);
					MoveToEx(hdc, r.left, r.bottom - (int)(avg * unit_height + 0.5f), 0);
					LineTo(hdc, r.right, r.bottom - (int)(avg * unit_height + 0.5f));
					if(max_value != avg) {
						MoveToEx(hdc, r.left, r.bottom - (int)(max_value * unit_height + 0.5f), 0);
						LineTo(hdc, r.right, r.bottom - (int)(max_value * unit_height + 0.5f));
						MoveToEx(hdc, r.left, r.bottom - (int)(min_value * unit_height + 0.5f), 0);
						LineTo(hdc, r.right, r.bottom - (int)(min_value * unit_height + 0.5f));
					}
				}

				// restore pen
				SelectObject(hdc, hPenOld);
				DeleteObject(hPen);
				DeleteObject(hPen2);

				SetBkMode(hdc, TRANSPARENT);
				SetTextColor(hdc, GetSysColor(COLOR_3DDKSHADOW));
				char buff[64];
				if(wd->show_grid) {
					sprintf(buff, Translate("%d ms"), MARK_TIME);
					TextOut(hdc, r.right - 100, r.bottom - (int)(unit_height * MARK_TIME + 0.5f), buff, strlen(buff));
				}

				if(wd->show_stat) {
					SetTextColor(hdc, RGB(255, 0, 0));
					sprintf(buff, Translate("AVG %.1lf ms"), avg);
					TextOut(hdc, r.left + 10, r.bottom - (int)(avg * unit_height + 0.5f), buff, strlen(buff));
					if(max_value != avg) {
						sprintf(buff, Translate("MAX %hd ms"), max_value);
						TextOut(hdc, r.left + 10, r.bottom - (int)(max_value * unit_height + 0.5f), buff, strlen(buff));
						sprintf(buff, Translate("MIN %hd ms"), min_value);
						TextOut(hdc, r.left + 10, r.bottom - (int)(min_value * unit_height + 0.5f), buff, strlen(buff));
					}
				}

				EndPaint(hwnd, &ps);
			}
		}
		return TRUE;

	case WM_ERASEBKGND:
		return TRUE;

	case WM_SIZE:
		{
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);
			RECT r;
			GetClientRect(hwnd, &r);
			if(wd->hwnd_chk_grid != 0) SetWindowPos(wd->hwnd_chk_grid, 0, r.right - 150, r.top + 10, 120, 20, SWP_NOZORDER | SWP_NOACTIVATE);
			if(wd->hwnd_chk_stat != 0) SetWindowPos(wd->hwnd_chk_stat, 0, r.right - 150, r.top + 30, 120, 20, SWP_NOZORDER | SWP_NOACTIVATE);
		}
		InvalidateRect(hwnd, 0, FALSE);
		break;
	case WM_CLOSE:
		{
			KillTimer(hwnd, ID_REPAINT_TIMER);
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);
			{
				char buff[30];
				sprintf(buff, "pinggraphwnd%d", wd->item_id);
				Utils_SaveWindowPosition(hwnd, 0, PLUG, buff);
			}
		}
		break;
	case WM_DESTROY:
		{
			WindowData *wd = (WindowData *)GetWindowLongPtr(hwnd, GWLP_USERDATA);
			{
				char buff[30];
				sprintf(buff, "WindowHandle%d", wd->item_id);
				DBWriteContactSettingDword(0, PLUG, buff, 0);
			}
			delete wd;
		}
		// drop through
	};

	return DefWindowProc(hwnd, msg, wParam, lParam);
}

INT_PTR ShowGraph(WPARAM wParam, LPARAM lParam) {
	char buff[30];
	sprintf(buff, "WindowHandle%d", (DWORD)wParam);
	HWND hGraphWnd = (HWND)DBGetContactSettingDword(0, PLUG, buff, 0);
	if(hGraphWnd) {
		ShowWindow(hGraphWnd, SW_SHOW);
		SetWindowPos(hGraphWnd, HWND_TOP, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);
		return 0;
	}

	WNDCLASS wndclass;
    wndclass.style         = 0;
    wndclass.lpfnWndProc   = GraphWindowProc;
    wndclass.cbClsExtra    = 0;
    wndclass.cbWndExtra    = 0;
    wndclass.hInstance     = hInst;
    wndclass.hIcon         = hIconResponding;
    wndclass.hCursor       = LoadCursor (NULL, IDC_ARROW);
    wndclass.hbrBackground = (HBRUSH)(COLOR_3DFACE+1);
    wndclass.lpszMenuName  = NULL;
    wndclass.lpszClassName = _T(PLUG) _T("GraphWindow");
	RegisterClass(&wndclass);

	char title[256];
	strcpy(title, "Ping Graph");
	if(lParam) {
		strcat(title, " - ");
		strncat(title, (char *)lParam, 256 - 13);
	}

	HWND parent = 0;
	hGraphWnd = CreateWindowEx(0, _T(PLUG) _T("GraphWindow"), title,
						(WS_THICKFRAME | WS_CAPTION | WS_SYSMENU | WS_CLIPCHILDREN) & ~CS_VREDRAW & ~CS_HREDRAW,
						0,0,800,600,parent,NULL,hInst,NULL);

	WindowData *wd = new WindowData;
	wd->item_id = (DWORD)wParam; // wParam is destination id
	wd->hwnd_chk_grid = 0;
	wd->hwnd_chk_stat = 0;
	wd->show_grid = DBGetContactSettingByte(0, PLUG, "ShowGridLines", 0) ? true : false;
	wd->show_stat = DBGetContactSettingByte(0, PLUG, "ShowStats", 1) ? true : false;

	DBWriteContactSettingDword(0, PLUG, buff, (DWORD)hGraphWnd);

	SetWindowLongPtr(hGraphWnd, GWLP_USERDATA, (LONG_PTR)wd); 

	sprintf(buff, "pinggraphwnd%d", wd->item_id);
	Utils_RestoreWindowPosition(hGraphWnd, 0, PLUG, buff);

	if(!IsWindowVisible(hGraphWnd))
		ShowWindow(hGraphWnd, SW_SHOW);

	return 0;
}

// save window positions, close windows
void graphs_cleanup() {
	int list_size = GetListSize(0, 0);
	char buff[64];
	HWND hwnd;

	for(int i = 0; i < list_size; i++) {
		sprintf(buff, "WindowHandle%d", i);
		if(hwnd = (HWND)DBGetContactSettingDword(0, PLUG, buff, 0)) {
			DestroyWindow(hwnd);
			DBWriteContactSettingDword(0, PLUG, buff, 0);
			sprintf(buff, "WindowWasOpen%d", i);
			DBWriteContactSettingByte(0, PLUG, buff, 1);
		}	
	}
}

// remove old data, possibly left from a crash
void graphs_init() {
	PINGLIST pl;
	char buff[64];
	CallService(PLUG "/GetPingList", 0, (LPARAM)&pl);
	for(PINGLIST::Iterator i = pl.start(); i.has_val(); i.next()) {
		sprintf(buff, "WindowHandle%d", i.val().item_id); // clean up from possible crash
		DBWriteContactSettingDword(0, PLUG, buff, 0);
		sprintf(buff, "WindowWasOpen%d", i.val().item_id); // restore windows that were open on shutdown
		if(DBGetContactSettingByte(0, PLUG, buff, 0)) {
			DBWriteContactSettingByte(0, PLUG, buff, 0);
			ShowGraph((WPARAM)i.val().item_id, (LPARAM)i.val().pszLabel);
		}
	}
}