MaaAssistantArknights/MeoAssistance/src/WinMacro.cpp

#include "WinMacro.h"

#include <vector>
#include <utility>
#include <ctime>
#include <cassert>
#include <algorithm>

#include <stdint.h>
#include <WinUser.h>

#include <iostream>

#include "Configer.h"

using namespace MeoAssistance;

WinMacro::WinMacro(HandleType type)
	: m_handle_type(type),
	m_rand_engine(time(NULL))
{
}

bool WinMacro::findHandle()
{
	json::array handle_arr;
	switch (m_handle_type) {
	case HandleType::BlueStacksControl:
		handle_arr = Configer::handleObj["BlueStacksControl"].as_array();
		break;
	case HandleType::BlueStacksView:
		handle_arr = Configer::handleObj["BlueStacksView"].as_array();
		break;
	case HandleType::BlueStacksWindow:
		handle_arr = Configer::handleObj["BlueStacksWindow"].as_array();
		break;
	default:
		std::cerr << "handle type error! " << static_cast<int>(m_handle_type) << std::endl;
		return false;
	}

	m_handle = NULL;
	for (auto&& obj : handle_arr)
	{
		m_handle = ::FindWindowExA(m_handle, NULL, obj["class"].as_string().c_str(), obj["window"].as_string().c_str());
	}

#ifdef _DEBUG
	std::cout << "type: " << static_cast<int>(m_handle_type) << ", handle: " << m_handle << std::endl;
#endif

	if (m_handle != NULL) {
		return true;
	}
	else {
		return false;
	}
}

bool WinMacro::resizeWindow(int width, int height)
{
	if (!(static_cast<int>(m_handle_type) & static_cast<int>(HandleType::Window))) {
		return false;
	}

	return ::MoveWindow(m_handle, 0, 0, width / getScreenScale(), height / getScreenScale(), true);
}

double WinMacro::getScreenScale()
{
	static double scale = 0;
	if (scale == 0) {
		// 获取窗口当前显示的监视器
		// 使用桌面的句柄.
		HWND hWnd = GetDesktopWindow();
		HMONITOR hMonitor = MonitorFromWindow(hWnd, MONITOR_DEFAULTTONEAREST);

		// 获取监视器逻辑宽度与高度
		MONITORINFOEX miex;
		miex.cbSize = sizeof(miex);
		GetMonitorInfo(hMonitor, &miex);
		int cxLogical = (miex.rcMonitor.right - miex.rcMonitor.left);
		int cyLogical = (miex.rcMonitor.bottom - miex.rcMonitor.top);

		// 获取监视器物理宽度与高度
		DEVMODE dm;
		dm.dmSize = sizeof(dm);
		dm.dmDriverExtra = 0;
		EnumDisplaySettings(miex.szDevice, ENUM_CURRENT_SETTINGS, &dm);
		int cxPhysical = dm.dmPelsWidth;
		int cyPhysical = dm.dmPelsHeight;

		// 考虑状态栏大小，逻辑尺寸会比实际小
		double horzScale = ((double)cxPhysical / (double)cxLogical);
		double vertScale = ((double)cyPhysical / (double)cyLogical);

		// 考虑状态栏大小，选择里面大的那个
		scale = std::max(horzScale, vertScale);
	}
	return scale;
}

bool WinMacro::click(const Point& p)
{
	if (!(static_cast<int>(m_handle_type) & static_cast<int>(HandleType::Control))) {
		return false;
	}

	int x = p.x / getScreenScale();
	int y = p.y / getScreenScale();
#ifdef _DEBUG
	std::cout << "click: " << x << ", " << y << std::endl;
#endif

	LPARAM lparam = MAKELPARAM(x, y);

	::SendMessage(m_handle, WM_LBUTTONDOWN, MK_LBUTTON, lparam);
	::SendMessage(m_handle, WM_LBUTTONUP, 0, lparam);

	return true;
}

bool WinMacro::clickRange(const Rect& rect)
{
	if (!(static_cast<int>(m_handle_type) & static_cast<int>(HandleType::Control))) {
		return false;
	}

	int x = 0, y = 0;
	if (rect.width == 0) {
		x = rect.x;
	}
	else {
		std::poisson_distribution<int> x_rand(rect.width);
		x = x_rand(m_rand_engine) + rect.x;
	}

	if (rect.height == 0) {
		y = rect.y;
	}
	else {
		std::poisson_distribution<int> y_rand(rect.height);
		y = y_rand(m_rand_engine) + rect.y;
	}

	return click({ x, y });
}

Rect WinMacro::getWindowRect()
{
	RECT rect;
	bool ret = ::GetWindowRect(m_handle, &rect);
	if (!ret) {
		return { 0, 0, 0 ,0 };
	}
	return Rect{ rect.left, rect.top, rect.right - rect.left, rect.bottom - rect.top } *getScreenScale();
}

cv::Mat WinMacro::getImage(const Rect& rect)
{
	if (!(static_cast<int>(m_handle_type) & static_cast<int>(HandleType::View))) {
		return cv::Mat();
	}

	HDC pDC;// 源DC
	pDC = ::GetDC(m_handle);//获取屏幕DC(0为全屏，句柄则为窗口)
	int BitPerPixel = ::GetDeviceCaps(pDC, BITSPIXEL);//获得颜色模式
	HDC memDC;//内存DC
	memDC = ::CreateCompatibleDC(pDC);
	HBITMAP memBitmap, oldmemBitmap;//建立和屏幕兼容的bitmap
	memBitmap = ::CreateCompatibleBitmap(pDC, rect.width, rect.height);
	oldmemBitmap = (HBITMAP)::SelectObject(memDC, memBitmap);//将memBitmap选入内存DC
	::PrintWindow(m_handle, memDC, PW_CLIENTONLY);

	BITMAP bmp;
	GetObject(memBitmap, sizeof(BITMAP), &bmp);
	int nChannels = bmp.bmBitsPixel == 1 ? 1 : bmp.bmBitsPixel / 8;
	cv::Mat dst_mat;
	dst_mat.create(cv::Size(bmp.bmWidth, bmp.bmHeight), CV_MAKETYPE(CV_8U, nChannels));
	GetBitmapBits(memBitmap, bmp.bmHeight * bmp.bmWidth * nChannels, dst_mat.data);

	DeleteObject(memBitmap);
	DeleteDC(memDC);
	ReleaseDC(m_handle, pDC);

#ifdef _DEBUG
	std::string filename = Configer::getCurDir() + "\\test.bmp";
	cv::imwrite(filename, dst_mat);
#endif

	return dst_mat;
}