#include "L2TradeQueueUtil.h"

#include "Win32Util.h"

list<ImgArea> L2TradeQueueUtil::splitViewElements(cv::Mat img)
{

	list<int*> rowDataList = splitRows(img);
	if (rowDataList.size() != 2) {
		for (list<int*>::iterator ele = rowDataList.begin(); ele != rowDataList.end(); ++ele) {
			free(*ele);
		}
		throw string("·Ö¸ôÐгö´í");
	}

	list<ImgArea> posList;

	//·Ö¸ô³öίÂô¶ÓÁÐʱ¼ä£¬Î¯Âôµ¥¼Û£¬Î¯ÂôÊÖÊý£¬Î¯Âò¶ÓÁÐʱ¼ä£¬Î¯Âòµ¥¼Û£¬Î¯ÂòÊÖÊý
	int* rowData = *(rowDataList.begin());
	//ίÂô¶ÓÁÐʱ¼ä
	list<int*> dataList = splitCols(img, 0, rowData[0], img.cols / 2, rowData[1]);
	if (dataList.size() > 1)
	{
		list<int*>::iterator ele = dataList.begin();
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
	}
	else {
		posList.push_back({ -1,-1,-1,-1 });
	}
	freeData(dataList);
	//ίÂò¶ÓÁÐʱ¼ä
	dataList = splitCols(img, img.cols / 2 + 1, rowData[0], img.cols - 1, rowData[1]);
	if (dataList.size() > 1)
	{
		list<int*>::iterator ele = dataList.begin();
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
	}
	else {
		posList.push_back({ -1,-1,-1,-1 });
	}
	
	free(rowData);
	rowData = *(++rowDataList.begin());
	//ίÂôµ¥¼Û£¬Î¯ÂôÊÖÊý
	freeData(dataList);
	dataList = splitCols(img, 0, rowData[0], img.cols / 2, rowData[1]);
	if (dataList.size() > 2)
	{
		list<int*>::iterator ele = dataList.begin();
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
	}
	else {
		posList.push_back({ -1,-1,-1,-1 });
		posList.push_back({ -1,-1,-1,-1 });
	}
	freeData(dataList);
	//ίÂòµ¥¼Û£¬Î¯ÂòÊÖÊý
	dataList = splitCols(img, img.cols / 2 + 1, rowData[0], img.cols - 1, rowData[1]);
	if (dataList.size() > 2)
	{
		list<int*>::iterator ele = dataList.begin();
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
		++ele;
		posList.push_back({ (*ele)[0],(*ele)[1],(*ele)[2],(*ele)[3] });
	}
	else {
		posList.push_back({-1,-1,-1,-1});
		posList.push_back({-1,-1,-1,-1});
	}
	freeData(dataList);
	free(rowData);

	if (FALSE) {
		//±£´æ
		int index = -1;
		for (list<ImgArea>::iterator ele = posList.begin(); ele != posList.end(); ++ele) {
			index++;
			ImgArea data = *ele;
			if (data.startx>-1) {
				//±£´æÊý¾Ý
				cv::Mat mat = cv::Mat(img, cv::Rect(data.startx, data.starty, data.endx - data.startx + 1, data.endy - data.starty + 1));
				string path = "C:\\Users\\Administrator\\Desktop\\ocr\\l2_trade\\";
				path.append(to_string(index)).append(".png");
				cv::imwrite(path, mat);
			}
		}
	}

	return posList;
}

list<ImgArea> L2TradeQueueUtil::splitBuyQueue(cv::Mat img,int identify)
{
	if (FALSE)
	{
		cv::imwrite(string("C:\\Users\\Administrator\\Desktop\\ocr\\trade_queue\\demo\\").append(to_string(identify)).append("_").append(to_string(rand())).append(".png"), img);
	}
	list<ImgArea> resultList;
	//cv::Mat binary;
	//¶þÖµ»¯£¬·½±ã´¦Àí
	//threshold(img, binary, _IMG_BINARY_THRESHOLD, 255, cv::THRESH_BINARY);
	
	
	//·Ö¸ôµÚÒ»ÐÐ
	int rows = img.rows;
	int startCol = 0;
	int endCol = img.cols - 1;
	list<int*> rowDataList;
	int full_start = -1;
	int full_end = -1;
	for (int r = 1; r < rows; r++) {
		if (ImgDivider::isRowFull(img, r, startCol, endCol, 1,49,51)) {
			if (full_start < 0) {
				full_start = r;
			}
			if (r - full_start > 10) {
				int* pos = (int*)malloc(sizeof(int) * 2);
				pos[0] = full_start + 1;
				pos[1] = r - 1;
				rowDataList.push_back(pos);
				full_start = r;
			}
			else {
				full_start = r;
			}
		}
	}

	if (rowDataList.size() < 1) {
		throw string("ÎÞÊý¾Ý");
	}
	endCol = img.cols - 1;
	int index = 0;
	for (list<int*>::iterator ele = rowDataList.begin(); ele != rowDataList.end(); ++ele) {
		int* pos = *ele;
		int startRow = pos[0];
		int endRow = pos[1];
		free(pos);
		//È¥³ýÉÏÏ¿հ×,ÓÒ²àÁô°×10
		for (int i = startRow; i <= endRow; i++) {
			if (!ImgDivider::isRowEmpty(img, i, startCol, endCol,1, _IMG_BINARY_THRESHOLD)) {
				startRow = i;
				break;
			}
		}
		for (int i = endRow; i >= startRow; i--) {
			if (!ImgDivider::isRowEmpty(img, i, startCol, endCol,1, _IMG_BINARY_THRESHOLD)) {
				endRow = i;
				break;
			}
		}

		//if (identify == 4|| identify ==6) {
		//	printf("½øÈëµ÷ÊÔ");
		//}

		if (endRow - startRow > 5)
		{

			list<int*> elements = splitCols(img, startCol, startRow, endCol - 1, endRow, 3);
			
			for (list<int*>::iterator e = elements.begin(); e != elements.end(); ++e) {
				int* ps = *e;
				ImgArea area;
				area.startx = ps[0];
				area.starty = ps[1];
				area.endx = ps[2];
				area.endy = ps[3];
				resultList.push_back(area);
				free(ps);
				if (FALSE)
				{
					cv::Mat sm = cv::Mat(img, cv::Rect(area.startx, area.starty, area.endx - area.startx + 1, area.endy - area.starty + 1));
					cv::imwrite(string("C:\\Users\\Administrator\\Desktop\\ocr\\trade_queue\\").append(to_string(index)).append(".png"), sm);
				}
				index++;
			}
		}
	}
	return resultList;
}

HWND L2TradeQueueUtil::getTradeQueueBuyHWND(HWND tradeQueueHWND)
{
	//ÔÝʱ·ÅÔÚÕâ¶ù
	HWND temp = NULL;
	while (TRUE)
	{
		temp = FindWindowExA(tradeQueueHWND, temp, "#32770", "ÂòÂô¶ÓÁÐ");
		if (Win32Util::isWindowShow(temp) || temp <= 0) {
			break;
		}
	}
	if (temp <= 0) {
		throw string("δÕÒµ½ÂòÂô¶ÓÁоä±ú");
	}

	HWND childTemp = NULL;


	HWND rightHWND=NULL;
	int rightPOS=-1;

	while (TRUE)
	{
		childTemp = FindWindowExA(temp, childTemp, "#32770", NULL);
		if (childTemp <= 0) {
			break;
		}
		else {
			if (Win32Util::isWindowShow(childTemp)) {
				//»ñÈ¡×îÓÒ²àµÄÄÚÈÝ¿ò¾ä±ú
				RECT rect;
				Win32Util::getWindowRect(childTemp, &rect);
				if (rightPOS == -1) {
					rightHWND = childTemp;
					rightPOS = rect.right;
				}
				else if (rect.right > rightPOS) {
					rightHWND = childTemp;
					rightPOS = rect.right;
				}
			}
		}
	}

	if (rightHWND <= 0) {
		throw string("ÂòÈë³É½»¶ÓÁоä±ú²éÕÒʧ°Ü");
	}
	rightHWND = FindWindowExA(rightHWND, NULL, "SysListView32", NULL);
	if (rightHWND <= 0) {
		throw string("ÂòÈë³É½»¶ÓÁоä±ú²éÕÒʧ°Ü");
	}
	return rightHWND;
}

void L2TradeQueueUtil::filterTradeQueueBuyMat(cv::Mat& grayImg)
{
	list<ImgArea> areas;
	int startR = -1;
	for (int r = 0; r < grayImg.rows; r++) {
		int startc = -1;
		int endc = -1;
		bool empty = TRUE;
		for (int c = 0; c < grayImg.cols; c++) {
			uchar data = grayImg.ptr<uchar>(r)[c];
			if (data >= 178 && data <= 180) {
				empty = FALSE;
				if (startc < 0) {
					startc = c;
					endc = c;
				}
				endc = c;
			}
			else {
				if (startc > 0 && endc - startc > 10 && startR < 0) {
					ImgArea area;
					area.startx = startc;
					area.endx = endc;
					area.starty = r;
					area.endy = -1;
					areas.push_back(area);
					startc = -1;
					endc = -1;
				}
			}
		}
		if (areas.size() > 0 && startR < 0) {
			startR = r;
		}

		if (!empty && areas.size() > 0) {
			continue;
		}

		if (startR > 0 && empty) {
			printf("ÆðʼÐÐ%d, ½áÊøÐУº%d Êý¾ÝÌõÊý£º%d\n", startR, r, areas.size());
			for (list<ImgArea>::iterator ele = areas.begin(); ele != areas.end(); ++ele) {
				ImgArea area = *ele;
				area.endy = r - 1;
				printf("%d,%d,%d,%d\n", area.startx, area.starty, area.endx, area.endy);
				for (int r = area.starty; r <= area.endy; r++) {
					for (int c = area.startx; c <= area.endx; c++) {
						uchar	data = grayImg.ptr<uchar>(r)[c];
						if (data >= _IMG_BINARY_THRESHOLD) {
							grayImg.ptr<uchar>(r)[c] = 0;
						}
						else {
							grayImg.ptr<uchar>(r)[c] = 255;
						}
					}
				}
			}

			printf("------------------------\n");

			startR = -1;
			areas.clear();
		}
	}
}
list<int*> L2TradeQueueUtil::splitRows(cv::Mat img)
{
	list<int*>  rowDataList;

	//ͼÏñ·Ö¸ô
	//ÏÈ·Ö¸ô 
	int start_row = -1;
	int end_row = -1;
	for (int r = 0; r < img.rows; r++) {
		if (ImgDivider::isRowEmpty(img, r, 1, img.cols/4,1, _IMG_BINARY_THRESHOLD)) {
			if (end_row - start_row >=10) {
				//ÓÐÊý¾Ý
				int* rowRata = (int*)malloc(sizeof(int) * 2);
				rowRata[0] = start_row;
				rowRata[1] = end_row;
				rowDataList.push_back(rowRata);
				start_row = -1;
				end_row = -1;
				if (rowDataList.size() >= 2) {
					break;
				}
			}
			start_row = r + 1;
		}
		else {
			if (start_row < 0) {
				start_row = r;
			}
			end_row =r;
		
		}
	}

	return rowDataList;
}

list<int*> L2TradeQueueUtil::splitCols(cv::Mat img, int startx, int starty, int endx, int endy, int minContentWidth)
{
	if (FALSE)
	{
		cv::imwrite("C:\\Users\\Administrator\\Desktop\\ocr\\l2_trade\\test.png", cv::Mat(img, cv::Rect(startx, starty, endx - startx + 1, endy - starty + 1)));
	}
	list<int*> splitList;
	int empty_start = -1;
	int empty_end = -1;
	int start_col = -1;
	int end_col = -1;
	for (int c = startx; c <= endx; c++) {
		if (ImgDivider::isColEmpty(img, c, starty, endy,_IMG_BINARY_THRESHOLD)) {
			if (empty_start < 0) {
				empty_start = c;
			}
			empty_end = c;
			if (empty_end - empty_start > 10) {
				if (end_col - start_col >= minContentWidth) {
					int* itemData = (int*)malloc(sizeof(int) * 4);
					itemData[0] = start_col;
					itemData[1] = starty;
					itemData[2] = empty_start - 1;
					itemData[3] = endy;
					splitList.push_back(itemData);
					start_col = -1;
					end_col = -1;
				}
			}

		}
		else {
			if (start_col < 0) {
				start_col = c;
			}
			end_col = c;
			empty_start = -1;
			empty_end = -1;
		}
	}
	if (end_col - start_col >= minContentWidth) {
		int* itemData = (int*)malloc(sizeof(int) * 4);
		itemData[0] = start_col;
		itemData[1] = starty;
		itemData[2] = empty_start - 1;
		itemData[3] = endy;
		splitList.push_back(itemData);
		start_col = -1;
		end_col = -1;
	}

	return splitList;
}

void L2TradeQueueUtil::freeData(list<int*> dataList)
{
	if (dataList.size()>0) {
	
		for (list<int*>::iterator ele = dataList.begin(); ele != dataList.end(); ele++) {
			free(*ele);
		}

	}

}