#include "THSActionUtil.h"
#include "ImgDivider.h"
#include "Win32Util.h"


//»ñÈ¡¸±ÆÁ
HWND getSecondWindow() {
	list<HWND> wlist = Win32Util::searchWindow("ͬ»¨Ë³(");
	list<HWND>::iterator ele;
	for (ele = wlist.begin();ele != wlist.end();ele++) {
		HWND hwnd = *ele;
		string str = Win32Util::getWindowName(hwnd);
		if (str.find("ͬ»¨Ë³") != string::npos && str.find("¸±ÆÁ") != string::npos)
		{
			return hwnd;
		}
	}
	return 0;
}

void THSActionUtil::openSecondScreen() {

	list<HWND> wlist = Win32Util::searchWindow("ͬ»¨Ë³(");

	bool open = false;
	HWND mainPage = 0;

	list<HWND>::iterator ele;
	for (ele = wlist.begin();ele != wlist.end();ele++) {
		HWND hwnd = *ele;
		string str = Win32Util::getWindowName(hwnd);
		if (str.find("ͬ»¨Ë³") != string::npos && str.find("¸±ÆÁ") != string::npos)
		{
			cout << str << endl;
			SetForegroundWindow(hwnd);
			SetFocus(hwnd);
			open = true;
			break;
		}

		if (str.find("ͬ»¨Ë³") != string::npos && str.find("ÈÈÃŹÉ") != string::npos) {
			cout << hwnd << endl;
			//»ñÈ¡³ß´ç
			RECT rc;
			GetWindowRect(hwnd, &rc);
			if (rc.right - rc.left > 200 && rc.bottom - rc.top > 200) {
				mainPage = hwnd;
			}
		}

	}

	if (!open) {
		Win32Util::focus(mainPage);
		//²éÕÒÖ÷´°¿ÚµÄ¹¤¾ßÀ¸
		HWND tool = FindWindowExA(mainPage, NULL, "AfxControlBar100s", NULL);

		//»ñÈ¡µÚÒ»¸ö×Ó´°¿Ú
		tool = FindWindowExA(tool, NULL, NULL, NULL);
		cout << "²Ù×÷À¸£º" << tool << endl;
		//´ò¿ª¸±ÆÁ£¬¶à´°°´Å¥IDΪ:0x00007AF9
		HWND btn = GetDlgItem(tool, 0x00007AF9);
		RECT rc;
		GetWindowRect(btn, &rc);
		//»ñÈ¡ÐèÒªµã»÷µÄλÖÃ
		int x = rc.left + (rc.right - rc.left) / 3;
		int y = rc.top + (rc.bottom - rc.top) / 3;
		//µã»÷¶à´°
		Win32Util::click(x, y);
		//µã»÷¸±ÆÁ1 
		Win32Util::click(x + 10, y + 21 * 7 + 5, 500);
	}
}
//Ìí¼Ó¹ÉƱ
void THSActionUtil::setGP(std::string quickCode, list<std::string> codeList) {
	//´ò¿ª¸±ÆÁ
	HWND sw = getSecondWindow();
	if (sw <= 0)
	{
		openSecondScreen();
		Sleep(2000);
		sw = getSecondWindow();
	}
	if (sw <= 0) {
		throw("δ´ò¿ª¸±ÆÁ");
	}

	//´ò¿ª°å¿é
	Win32Util::keyboardNum(quickCode);
	Win32Util::keyboard(VK_RETURN, 200);
	Sleep(1000);
	//ÉèÖðå¿éÖеĹÉƱ

	//»ñÈ¡°å¿éÄÚÈݾä±ú

	HWND content = FindWindowExA(sw, NULL, "AfxFrameOrView100s", NULL);
	cv::Mat img = CaptureUtil::capture(content);
	if (img.cols <= 0 || img.rows <= 0) {
		throw("°å¿é½ØÆÁÄÚÈÝΪ¿Õ");
	}

	std::list<GPCodeArea>  areaList = recognitionGPArea(img);

	std::list<string> addList;
	std::list<GPCodeArea> delList;
	std::list<GPCodeArea> resultList = recognitionNum(img, areaList);

	for (std::list<GPCodeArea>::reverse_iterator ele= resultList.rbegin();ele != resultList.rend();++ele) {

		bool contains = false;
		std::list<string>::iterator e;
		for (e = codeList.begin();e != codeList.end();e++) {
			if (*e == (*ele).code) {
				contains = true;
				break;
			}
		}

		if (!contains) {
			delList.push_back(*ele);
		}
		else
		{
			codeList.remove((*ele).code);
		}
	
	}
	//-----ÏÈɾ³ýÐèҪɾ³ýµÄ
	//»ñÈ¡ÄÚÈÝ°å¿é×ø±ê
	RECT rect;
	GetWindowRect(content, &rect);
	

	for (std::list<GPCodeArea>::iterator ele = delList.begin();ele != delList.end();++ele) {     
		int x = rect.left;
		int y = rect.top;
		x += (*ele).startx+5;
		y += (*ele).starty+5;
	    //Ñ¡ÖÐɾ³ý
		Win32Util::click(x,y,50);
		Win32Util::keyboard(VK_DELETE, 50);
	}

	//----Ôö¼Ó
	//½Øͼ£¬Ê¶±ð³öÔö¼Ó°´Å¥Î»Ö㬵ã»÷Ôö¼Ó£¬ÊäÈëÄÚÈÝ

}


void add(string code) {
	//´ò¿ª¸±ÆÁ
	HWND sw = getSecondWindow();
	if (sw <= 0)
	{
		openSecondScreen();
		Sleep(2000);
		sw = getSecondWindow();
	}
	if (sw <= 0) {
		throw("δ´ò¿ª¸±ÆÁ");
	}

	HWND content = FindWindowExA(sw, NULL, "AfxFrameOrView100s", NULL);
	cv::Mat img = CaptureUtil::capture(content);
	if (img.cols <= 0 || img.rows <= 0) {
		throw("°å¿é½ØÆÁÄÚÈÝΪ¿Õ");
	}
	std::list<GPCodeArea>  areaList = recognitionGPArea(img);
	for (std::list<GPCodeArea>::iterator ele = areaList.begin();ele != areaList.end();++ele) {
		GPCodeArea codeArea = *ele;
		if (codeArea.type == IMG_TYPE_ADD) {
			RECT rc;
			GetWindowRect(HWND(0x00161728), &rc);
			int y = rc.top + codeArea.starty + (codeArea.endy - codeArea.starty) / 2;
			int x = rc.left + codeArea.startx + (codeArea.endx - codeArea.startx) / 2;
			Win32Util::click(x, y, 50);
			//ÊäÈë¹ÉƱ´úÂë
			Win32Util::keyboardNum(code,1000);
			Win32Util::keyboard(VK_RETURN, 1500);
			//¹Ø±Õ°´Å¥
			Sleep(100);
			HWND close = FindWindowA(0, "Ìí¼Ó¹ÉƱ");
			SendMessage(close, WM_CLOSE, 0, 0);
			break;
		}
	}
}

//ʶ±ð¹ÉƱ´úÂë
std::list<GPCodeArea>  THSActionUtil::recognitionGPArea(cv::Mat img) {
	//»ñÈ¡title·Ö¸ôÏß
	int rows = img.rows;
	int cols = img.cols;
	int r;
	int contentStartRow = -1;
	for (r = 5;r < img.rows;r++) {
		if (ImgDivider::isRowFull(img, r, cols - 100, cols - 5, 2)) {
			contentStartRow = r;
		}

		if (contentStartRow > -1) {
			break;
		}
	}

	if (contentStartRow < 0) {
		throw("ÆðʼÐзָô³ö´í");
	}

	//·Ö¸ôÁÐ
	int c = 0;
	int startC = -1;
	int endC = -1;
	for (c = 50;c < cols;c++) {
		if (ImgDivider::isColFull(img, c, contentStartRow + 5, contentStartRow + 100, 2)) {
			if (startC < 0) {
				startC = c;
			}
			else {
				if (c - startC < 20) {
					startC = c;
				}
				else {
					endC = c;
					break;
				}
			}
		}
	}

	if (startC < 0 || endC < 0) {
		throw("ÄÚÈÝ¿ò·Ö¸ô³ö´í");
	}
	//cv::imshow("ÄÚÈÝ", cv::Mat(img, cv::Rect(startC, contentStartRow, endC - startC, rows - contentStartRow)));
	std::list<GPCodeArea> resultList;

	startC += 1;
	endC -= 1;

	//·Ö¸ôÐÐÄÚÈÝ
	int emptyStartRow = -1;
	int emptyEndRow = -1;
	int startf = -1;
	int endf = -1;
	std::list<int*> dataItemList;

	for (int i = contentStartRow + 30;i < rows;i++) {
		bool empty = ImgDivider::isRowEmpty(img, i, startC, startC + 50, 1, 64) && ImgDivider::isRowEmpty(img, i, startC + (endC - startC) / 2 - 40, startC + (endC - startC) / 2 + 40, 1, 64);
		if (empty) {
			if (emptyStartRow < 0) {
				emptyStartRow = i;
				emptyEndRow = i;
			}
			else {
				emptyEndRow = i;
			}

			if (emptyEndRow - emptyStartRow > 50 && dataItemList.size() > 0) {
				//ûÓÐÊý¾ÝÁË
				break;
			}

			if (startf > -1 && endf > -1 && emptyEndRow - emptyStartRow > 4) {
				//ÄÚÈÝ×ø±ê
				   //LogUtil::debug("ÄÚÈݵĸ߶ÈΪ£º%d \n", endf - startf);
				int* dd = (int*)malloc(sizeof(int) * 4);
				*dd = startC;
				*(dd + 1) = startf;
				*(dd + 2) = endC;
				*(dd + 3) = endf;
				//ÐÐÊý¾Ý¸ß´óÓÚ6²ÅΪÓÐЧµÄÐиß
				if (endf - startf > 6)
				{
					dataItemList.push_back(dd);
				}
				startf = -1;
				endf = -1;
				emptyStartRow = i;
				emptyEndRow = i;
			}

		}
		else
		{
			//Êý¾Ý¿ªÊ¼
			if (startf < 0) {
				startf = i;
				endf = i;
			}
			else {
				endf = i;
			}
			emptyStartRow = -1;
			//Êý¾Ý½áÊø
		}
	}

	//·Ö¸ôÁÐ
	//·Ö¸ôÿһÁеÄÊý¾Ý
	std::list<int*>::iterator ele;
	int index = 0;
	for (ele = dataItemList.begin(); ele != dataItemList.end();ele++) {
		index++;
		int startRow = *(*ele + 1);
		int startCol = *(*ele + 0);
		int endRow = *(*ele + 3);
		int endCol = *(*ele + 2);
		LogUtil::debug("%d %d %d %d\n", startRow, startCol, endRow, endCol);

		//±£´æÐÐÊý¾Ý
		if (false) {
			std::string path = "C:\\Users\\Administrator\\Desktop\\ocr\\split\\";
			path.append(std::to_string(index)).append(".jpg");
			imwrite(path, cv::Mat(img, cv::Rect(startCol, startRow, endCol - startCol + 1, endRow - startRow + 1)));
		}

		int emptyStart = -1;
		int emptyEnd = -1;
		int dataStart = -1;
		int dataEnd = -1;

		std::list<int*> rowDataList;

		for (int i = startCol;i <= endCol;i++) {
			bool empty = ImgDivider::isColEmpty(img, i, startRow, endRow, 64);
			if (empty) {
				if (emptyStart < 0) {
					emptyStart = i;
					emptyEnd = i;
				}
				else {
					emptyEnd = i;
				}

				//3¸ö¼°ÒÔÉϵĿհ×Êý¾Ý²Å·ÖÁÐ
				if (emptyEnd - emptyStart >= 5 && dataEnd - dataStart > 0) {
					int* dd = (int*)malloc(sizeof(int) * 4);
					*dd = dataStart;
					*(dd + 1) = startRow;
					*(dd + 2) = dataEnd;
					*(dd + 3) = endRow;
					rowDataList.push_back(dd);
					dataEnd = -1;
					dataStart = -1;
					if (rowDataList.size() >= 2) {
						break;
					}
				}
			}
			else {
				if (dataStart < 0) {
					dataStart = i;
					dataEnd = i;
				}
				else {
					dataEnd = i;
				}

				emptyStart = -1;
				emptyEnd = -1;
			}
		}
		//ºóÃæµÄÊý¾ÝûÓÐ×ã¹»µÄ¿Õ°×·Ö¸ô
		if (dataEnd - dataStart > 0) {
			int* dd = (int*)malloc(sizeof(int) * 4);
			*dd = dataStart;
			*(dd + 1) = startRow;
			*(dd + 2) = dataEnd;
			*(dd + 3) = endRow;
			rowDataList.push_back(dd);
		}

		int rowDataSize = rowDataList.size();
		if (rowDataSize == 2) {

			std::list<int*>::iterator ele = rowDataList.begin();
			std::advance(ele, 1);

			GPCodeArea area = GPCodeArea();
			area.type = IMG_TYPE_GP;
			area.startx = **ele;
			area.starty = *(*ele + 1);
			area.endx = *(*ele + 2);
			area.endy = *(*ele + 3);
			resultList.push_back(area);
		}
		else {
			std::list<int*>::iterator ele = rowDataList.begin();
			GPCodeArea area = GPCodeArea();
			area.type = IMG_TYPE_ADD;
			area.startx = **ele;
			area.starty = *(*ele + 1);
			area.endx = *(*ele + 2);
			area.endy = *(*ele + 3);
			resultList.push_back(area);
		}



		//±£´æÊý¾Ý
		if (false)
		{
			std::list<int*>::iterator ele1;
			int cc = 0;
			for (ele1 = rowDataList.begin(); ele1 != rowDataList.end();ele1++) {
				if (*ele1)
				{
					int startCRow = *(*ele1 + 1);
					int startCCol = *(*ele1 + 0);
					int endCRow = *(*ele1 + 3);
					int endCCol = *(*ele1 + 2);
					cv::Mat	temp = cv::Mat(img, cv::Rect(startCCol, startCRow, endCCol - startCCol + 1, endCRow - startCRow + 1));
					std::string path = "C:\\Users\\Administrator\\Desktop\\ocr\\split\\";
					path = path.append(std::to_string(index));
					path = path.append("_");
					path = path.append(std::to_string(cc));
					path = path.append(".jpg");
					imwrite(path.c_str(), temp);
				}
				cc++;
			}
		}

	}

	return resultList;

}

std::list<GPCodeArea>   THSActionUtil::recognitionNum(cv::Mat img, std::list<GPCodeArea> areaList) {

	if (!recognitionManager) {
		recognitionManager = new RecognitionManager();
	}
	//ʶ±ðÊý×Ö
	std::list<GPCodeArea> codeList;

	std::list<GPCodeArea>::iterator ele;

	for (ele = areaList.begin();ele != areaList.end();ele++) {
		GPCodeArea codeArea = *ele;
		if (codeArea.type == IMG_TYPE_GP) {
			cv::Mat nums = cv::Mat(img, cv::Rect(codeArea.startx, codeArea.starty, codeArea.endx - codeArea.startx + 1, codeArea.endy - codeArea.starty + 1));
			std::list<cv::Mat> list2 = ImgUtil::splitNum(nums);
			std::list<uchar> resultList = recognitionManager->recognitionGPCode(list2);
			std::list<uchar>::iterator ele1;
			string code = "";
			for (ele1 = resultList.begin();ele1 != resultList.end();++ele1) {
				code.append(to_string(*ele1));
			}
			codeArea.code = code;
			codeList.push_back(codeArea);
		}
	}

	return codeList;
}