'功能完善'

admin

2025-04-08 5c9991be21f57781573f04961ec511ac2938ea3d

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
#include "pch.h"
#include "WechatOCR.h"
#include "StringUtil.h"
 
HWND WechatOCR::getWechatWindow()
{
    return HWND();
}
 
cv::Mat WechatOCR::capture(HWND hwnd)
{
    return cv::Mat();
}
 
string WechatOCR::ocrCode(cv::Mat src)
{
    return string();
}
 
WechatOCR::WechatOCR()
{
    tessApi = new tesseract::TessBaseAPI();
    // ÉèÖÃÓïÑÔ
    // eng+chi_sim
    if (tessApi->Init("tessdata", "eng+chi_sim")) {
        std::cerr << "Could not initialize tesseract.\n";
        exit(1);
    }
 
}
 
WechatOCR::~WechatOCR()
{
    if (tessApi != nullptr) {
        // ÇåÀí
        tessApi->End();
        delete tessApi;
    }
}
 
void WechatOCR::ocr(cv::Mat src)
{
 
    Pix* pix = pixCreate(src.cols, src.rows, 8);
    for (int y = 0; y < src.rows; ++y) {
        for (int x = 0; x < src.cols; ++x) {
            pixSetPixel(pix, x, y, src.ptr<uchar>(y)[x]);
        }
    }
 
    // ÉèÖÃͼÏñ
    tessApi->SetImage(pix);
 
    //char *result = tessApi->GetUTF8Text();
 
    //cout << "ʶ±ðµÄÄÚÈÝ£º"<< string(result) << endl;
 
    if (tessApi->Recognize(nullptr) != 0) {
        std::cerr << "Recognition failed." << std::endl;
        tessApi->End();
        pixDestroy(&pix);
        cout << "ʶ±ð³ö´í" << endl;
        return ;
    }
 
    // »ñÈ¡Îı¾
    tesseract::ResultIterator* ri  = tessApi->GetIterator();
    tesseract::PageIteratorLevel level = tesseract::RIL_WORD;
    if (ri != nullptr) {
        do {
            const char* word = ri->GetUTF8Text(level);
            if (word == nullptr) {
                continue;
            }
            float conf = ri->Confidence(level);
            int x1, y1, x2, y2;
            ri->BoundingBox(level, &x1, &y1, &x2, &y2);
 
 
 
            std::cout << "Word: " << word << ", Confidence: " << conf
                << ", Bounding box: (" << x1 << ", " << y1 << ") - (" << x2 << ", " << y2 << ")" << std::endl;
            delete[] word;
        } while (ri->Next(level));
        delete ri;
    }
    tessApi->End();
    pixDestroy(&pix);
}
 
void WechatOCR::test(Pix* src)
{
    // ÉèÖÃͼÏñ
    tessApi->SetImage(src);
    // »ñÈ¡Îı¾
    tesseract::ResultIterator* ri = tessApi->GetIterator();
    tesseract::PageIteratorLevel level = tesseract::RIL_WORD;
    if (ri != nullptr) {
        do {
            const char* word = ri->GetUTF8Text(level);
            float conf = ri->Confidence(level);
            int x1, y1, x2, y2;
            ri->BoundingBox(level, &x1, &y1, &x2, &y2);
 
            std::cout << "Word: " << word << ", Confidence: " << conf
                << ", Bounding box: (" << x1 << ", " << y1 << ") - (" << x2 << ", " << y2 << ")" << std::endl;
            delete[] word;
        } while (ri->Next(level));
        delete ri;
    }
    tessApi->End();
    pixDestroy(&src);
}
 
list<HWND> WechatOCR::getWechatSimulatorHwnds()
{
    list<HWND>  hwnds = Win32Util::searchWindow("΢ÐÅ");
    list<HWND> wechatHWNDs;
    for (list<HWND>::iterator e = hwnds.begin();e != hwnds.end();++e) {
        HWND hwnd = *e;
        hwnd = FindWindowExA(hwnd, 0, "Qt5152QWindowIcon", "Androws");
        if (hwnd > 0) {
            for (int i = 0;i < 10;i++) {
                hwnd = FindWindowExA(hwnd, 0, "Qt5152QWindowIcon", "Androws");
                if (FindWindowExA(hwnd, 0, "subWin", "sub") > 0) {
                    hwnd = FindWindowExA(hwnd, 0, "subWin", "sub");
                    wechatHWNDs.push_back(hwnd);
                    break;
                }
            }
        }
    }
    return wechatHWNDs;
}
 
cv::Mat WechatOCR::grayImage(cv::Mat src)
{
    cv::Mat grayImage;
    if (src.channels() == 3)
    {
        cvtColor(src, grayImage, cv::COLOR_RGB2GRAY);
    }
    else if (src.channels() == 4) {
        cvtColor(src, grayImage, cv::COLOR_RGBA2GRAY);
    }
    else if (src.channels() == 1) {
        return src;
    }
    return grayImage;
}