#include "L2TradeQueueUtil.h"
|
|
list<ImgArea> L2TradeQueueUtil::splitElements(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<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)
|
{
|
cv::imwrite("C:\\Users\\Administrator\\Desktop\\ocr\\l2_trade\\test.png", cv::Mat(img, cv::Rect(startx, starty, endx - startx, endy - starty)));
|
|
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)) {
|
if (empty_start < 0) {
|
empty_start = c;
|
}
|
empty_end = c;
|
if (empty_end - empty_start > 10) {
|
if (end_col - start_col >= 6) {
|
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;
|
}
|
}
|
|
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);
|
}
|
|
}
|
|
}
|
