#include <windows.h>
#include <stdint.h>
#include <tlhelp32.h>
#include <stdio.h>
#include <iostream>
#include <vector>
#include <string>
#include <fstream>
#include <stdlib.h>
#include <ctime>
// opencv2
#include <opencv2\core\core.hpp>
#include <opencv2\highgui\highgui.hpp>
#include <opencv2\opencv.hpp>
using namespace std;
using namespace cv;
#define PATH "C:\\Users\\Administrator\\Desktop\\test\\" // 临时文件保存目录
#define MAPSIZE 100
typedef struct EnumHWndsArg
{
std::vector<HWND> *vecHWnds;
DWORD dwProcessId;
}EnumHWndsArg, *LPEnumHWndsArg;
void ReadF(char* str, char* buffer)
{
HANDLE pfile;
pfile = ::CreateFile(str, GENERIC_READ, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
if (pfile == INVALID_HANDLE_VALUE)
{
cout << "打开文件失败" << endl;;
CloseHandle(pfile); // 一定注意在函数退出之前对句柄进行释放
return;
}
DWORD filesize = GetFileSize(pfile, NULL);
DWORD readsize;
ReadFile(pfile, buffer, filesize, &readsize, NULL);
buffer[filesize] = 0;
CloseHandle(pfile); // 关闭句柄
DeleteFile(str);
}
void WriteF(char* str, const char* buffer, int size)
{
HANDLE pfile;
pfile = ::CreateFile(str, GENERIC_WRITE | GENERIC_READ, 0,
NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_HIDDEN | FILE_FLAG_WRITE_THROUGH, NULL);
if (pfile == INVALID_HANDLE_VALUE)
{
cout << "打开文件失败" << endl;;
CloseHandle(pfile); // 一定注意在函数退出之前对句柄进行释放
return;
}
DWORD readsize;
WriteFile(pfile, buffer, size, &readsize, NULL);
CloseHandle(pfile); // 关闭句柄
DeleteFile(str);
}
//通过进程ID获取进程句柄
HANDLE GetProcessHandleByID(int nID)
{
return OpenProcess(PROCESS_ALL_ACCESS, FALSE, nID);
}
// 通过应用程序名称得到进程ID
DWORD GetProcessIDByName(const char* pName)
{
HANDLE hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
if (INVALID_HANDLE_VALUE == hSnapshot) {
return NULL;
}
PROCESSENTRY32 pe = { sizeof(pe) };
for (BOOL ret = Process32First(hSnapshot, &pe); ret; ret = Process32Next(hSnapshot, &pe)) {
if (strcmp((const char*)pe.szExeFile, pName) == 0) {
CloseHandle(hSnapshot);
return pe.th32ProcessID;
}
//printf("%-6d %s\n", pe.th32ProcessID, pe.szExeFile);
}
CloseHandle(hSnapshot);
return 0;
}
BOOL CALLBACK lpEnumFunc(HWND hwnd, LPARAM lParam)
{
EnumHWndsArg *pArg = (LPEnumHWndsArg)lParam;
DWORD processId;
GetWindowThreadProcessId(hwnd, &processId);
if (processId == pArg->dwProcessId)
{
pArg->vecHWnds->push_back(hwnd);
//printf("%p\n", hwnd);
}
return TRUE;
}
// 对指定句柄窗口进行截屏操作
void ShootScreen(const char* filename, HWND hWnd)
{
HDC hdc = CreateDC("DISPLAY", NULL, NULL, NULL);
int32_t ScrWidth = 0, ScrHeight = 0;
RECT rect = { 0 };
if (hWnd == NULL)
{
ScrWidth = GetDeviceCaps(hdc, HORZRES);
ScrHeight = GetDeviceCaps(hdc, VERTRES);
}
else
{
GetWindowRect(hWnd, &rect);
rect.left += 8;
rect.right -= 8;
rect.bottom -= 8;
rect.top += 1;
ScrWidth = rect.right - rect.left;
ScrHeight = rect.bottom - rect.top;
}
HDC hmdc = CreateCompatibleDC(hdc);
HBITMAP hBmpScreen = CreateCompatibleBitmap(hdc, ScrWidth, ScrHeight);
HBITMAP holdbmp = (HBITMAP)SelectObject(hmdc, hBmpScreen);
BITMAP bm;
GetObject(hBmpScreen, sizeof(bm), &bm);
BITMAPINFOHEADER bi = { 0 };
bi.biSize = sizeof(BITMAPINFOHEADER);
bi.biWidth = bm.bmWidth;
bi.biHeight = bm.bmHeight;
bi.biPlanes = bm.bmPlanes;
bi.biBitCount = bm.bmBitsPixel;
bi.biCompression = BI_RGB;
bi.biSizeImage = bm.bmHeight * bm.bmWidthBytes;
// 图片的像素数据
char *buf = new char[bi.biSizeImage];
BitBlt(hmdc, 0, 0, ScrWidth, ScrHeight, hdc, rect.left, rect.top, SRCCOPY);
GetDIBits(hmdc, hBmpScreen, 0L, (DWORD)ScrHeight, buf, (LPBITMAPINFO)&bi, (DWORD)DIB_RGB_COLORS);
BITMAPFILEHEADER bfh = { 0 };
bfh.bfType = ((WORD)('M' << 8) | 'B');
bfh.bfSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + bi.biSizeImage;
bfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);
HANDLE hFile = CreateFile(filename, GENERIC_WRITE, 0, 0, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
DWORD dwWrite;
WriteFile(hFile, &bfh, sizeof(BITMAPFILEHEADER), &dwWrite, NULL);
WriteFile(hFile, &bi, sizeof(BITMAPINFOHEADER), &dwWrite, NULL);
WriteFile(hFile, buf, bi.biSizeImage, &dwWrite, NULL);
CloseHandle(hFile);
hBmpScreen = (HBITMAP)SelectObject(hmdc, holdbmp);
}
// 通过进程号得到当前进程中的所有窗口句柄
void GetHWndsByProcessID(DWORD processID, std::vector<HWND> &vecHWnds)
{
EnumHWndsArg wi;
wi.dwProcessId = processID;
wi.vecHWnds = &vecHWnds;
EnumWindows(lpEnumFunc, (LPARAM)&wi);
}
// 得到程序主窗口
// 主要是根据窗口的大小和窗口的可见性进筛选
HWND getMainHWND(vector<HWND> &vec)
{
HWND res;
int size = 0;
for (auto &hdc : vec)
{
if (!IsWindowVisible(hdc)) continue;
LPRECT lrt = new tagRECT();
GetWindowRect(hdc, lrt);
int h = lrt->bottom - lrt->top;
int w = lrt->right - lrt->left;
if (h * w > size)
{
size = h * w;
res = hdc;
}
}
return res;
}
// 模拟鼠标左键点击
void leftClickScreen(int x, int y)
{
SetCursorPos(x, y);
mouse_event(MOUSEEVENTF_LEFTDOWN, 0, 0, 0, 0);
mouse_event(MOUSEEVENTF_LEFTUP, 0, 0, 0, 0);
return;
}
// 模拟鼠标右键点击
void rightClickScreen(int x, int y)
{
SetCursorPos(x, y);
mouse_event(MOUSEEVENTF_RIGHTDOWN, 0, 0, 0, 0);
mouse_event(MOUSEEVENTF_RIGHTUP, 0, 0, 0, 0);
return;
}
void initMap(vector<vector<int>> &mp, int h, int w)
{
// -1 未点击
// 0 空地
// >0 炸弹数量
// -2 炸弹
mp = vector<vector<int>>(h, vector<int>(w, -1));
}
// 返回true 代表踩到地雷结束游戏
// 根据图片更新二维矩阵
bool updataMap(
vector<vector<int>> &mp,
string path,
int bx,
int by,
int height,
int width,
int cellheight,
int cellwidth
)
{
Mat image = imread(path);
/*for (int i = 1; i <= height; i++)
{
Point beginPoint(0 + bx, i * cellheight + by);
Point endPoint(width * cellwidth + bx, i * cellheight + by);
line(image, beginPoint, endPoint, Scalar(0, 0, 0));
}
for (int i = 1; i <= width; i++)
{
Point beginPoint(i * cellwidth + bx, 0 + by);
Point endPoint(i * cellwidth + bx, height * cellheight + by);
line(image, beginPoint, endPoint, Scalar(0, 0, 0));
}
*/
for (int i = 0; i < height; i++)
{
for (int j = 0; j < width; j++)
{
Point point(j * cellwidth + bx + cellwidth / 2, i * cellheight + by + cellheight / 2);
Point point1(j * cellwidth + bx + 1, i * cellheight + by + 3);
//circle(image, point, 2, Scalar(100, 100, 100));
auto it = image.at<Vec3b>(point);
int r = it[0], g = it[1], b = it[2];
//printf("(%d: r = %d, g = %d, b = %d)", i * width + j, r, g, b);
if (r == 0 && g == 0 && b == 0) return true;
if (r == 192 && g == 192 && b == 192){
it = image.at<Vec3b>(point1);
r = it[0], g = it[1], b = it[2];
//printf("(r = %d, g = %d, b = %d)", r, g, b);
if (r == 192) mp[i][j] = 0;
}
else
{
if (r == 255 && g == 0 && b == 0) mp[i][j] = 1;
else if (r == 0 && g == 128 && b == 0) mp[i][j] = 2;
else if (r == 0 && g == 0 && b == 255) mp[i][j] = 3;
else if (r == 128 && g == 0 && b == 0) mp[i][j] = 4;
else if (r == 0 && g == 0 && b == 128) mp[i][j] = 5;
else mp[i][j] = 6;
}
//printf("\n");
}
}
for (int i = 0; i < height; i++)
{
for (int j = 0; j < width; j++)
{
printf("%d ", mp[i][j]);
}
printf("\n");
}
printf("\n");
imshow("image", image);
waitKey(1);
return false;
}
/*
[gamename] 扫雷的名称
[bx] 第一个小单元格相对于扫雷界面横坐标的偏移
[by] 第一个小单元格相对于扫雷界面纵坐标的偏移
[height] 地图大小（高度）
[width] 地图大小（宽度）
[cellHeight] 小单元格的高度
[cellWidth] 小单元格的宽度
*/
bool playGame(
string gamename,
int bx,
int by,
int height,
int width,
int cellHeight,
int cellWidth
)
{
DWORD pid = GetPro
