mirror of
https://github.com/fofolee/uTools-quickcommand.git
synced 2025-06-07 21:46:12 +08:00
122 lines
3.6 KiB
JavaScript
122 lines
3.6 KiB
JavaScript
const { nativeImage } = require("electron");
|
||
const { captureScreen } = require("./screenCapture");
|
||
|
||
// 获取显示器缩放比例
|
||
function getDisplayScale() {
|
||
// MacOS 上要考虑缩放比例
|
||
if (process.platform === "darwin") {
|
||
const primaryDisplay = utools.getPrimaryDisplay();
|
||
const { scaleFactor } = primaryDisplay;
|
||
return scaleFactor;
|
||
}
|
||
return 1;
|
||
}
|
||
|
||
// 在屏幕上查找图片
|
||
async function findImage(subDataURL, options = {}) {
|
||
const mainDataURL = await captureScreen();
|
||
if (!mainDataURL) return null;
|
||
// 解析主图和子图
|
||
const mainImg = nativeImage.createFromDataURL(mainDataURL);
|
||
const subImg = nativeImage.createFromDataURL(subDataURL);
|
||
|
||
// 获取图像基本信息
|
||
const mainSize = mainImg.getSize();
|
||
const subSize = subImg.getSize();
|
||
|
||
// 获取像素数据（返回Buffer，RGBA格式）
|
||
const mainPixels = mainImg.getBitmap();
|
||
const subPixels = subImg.getBitmap();
|
||
|
||
// 边界检查
|
||
if (subSize.width > mainSize.width || subSize.height > mainSize.height) {
|
||
throw new Error("要查找图片尺寸大于屏幕");
|
||
}
|
||
|
||
// 预提取子图首像素值（优化点）
|
||
const firstSubPixel = [
|
||
subPixels[0], // R
|
||
subPixels[1], // G
|
||
subPixels[2], // B
|
||
subPixels[3], // A
|
||
];
|
||
|
||
// 主图遍历边界
|
||
const maxX = mainSize.width - subSize.width;
|
||
const maxY = mainSize.height - subSize.height;
|
||
|
||
// 遍历主图每个可能的位置
|
||
for (let y = 0; y <= maxY; y++) {
|
||
for (let x = 0; x <= maxX; x++) {
|
||
// 快速检查首像素（性能优化关键）
|
||
const mainOffset = (y * mainSize.width + x) * 4;
|
||
if (
|
||
mainPixels[mainOffset] !== firstSubPixel[0] ||
|
||
mainPixels[mainOffset + 1] !== firstSubPixel[1] ||
|
||
mainPixels[mainOffset + 2] !== firstSubPixel[2] ||
|
||
mainPixels[mainOffset + 3] !== firstSubPixel[3]
|
||
) {
|
||
continue;
|
||
}
|
||
|
||
// 完整像素比对
|
||
let match = true;
|
||
for (let subY = 0; subY < subSize.height; subY++) {
|
||
for (let subX = 0; subX < subSize.width; subX++) {
|
||
// 计算像素位置
|
||
const mainPixelPos = ((y + subY) * mainSize.width + (x + subX)) * 4;
|
||
const subPixelPos = (subY * subSize.width + subX) * 4;
|
||
|
||
// 精确匹配每个通道
|
||
if (
|
||
mainPixels[mainPixelPos] !== subPixels[subPixelPos] ||
|
||
mainPixels[mainPixelPos + 1] !== subPixels[subPixelPos + 1] ||
|
||
mainPixels[mainPixelPos + 2] !== subPixels[subPixelPos + 2] ||
|
||
mainPixels[mainPixelPos + 3] !== subPixels[subPixelPos + 3]
|
||
) {
|
||
match = false;
|
||
break;
|
||
}
|
||
}
|
||
if (!match) break;
|
||
}
|
||
|
||
if (match) {
|
||
const displayScale = getDisplayScale();
|
||
const position = {
|
||
x: Math.round(x / displayScale),
|
||
y: Math.round(y / displayScale),
|
||
width: Math.round(subSize.width / displayScale),
|
||
height: Math.round(subSize.height / displayScale),
|
||
};
|
||
clickImage(position, options.mouseAction);
|
||
return position;
|
||
}
|
||
}
|
||
}
|
||
|
||
return null;
|
||
}
|
||
|
||
const clickImage = (position, mouseAction) => {
|
||
const centerX = Math.round(position.x + position.width / 2);
|
||
const centerY = Math.round(position.y + position.height / 2);
|
||
|
||
// 根据配置执行鼠标动作
|
||
switch (mouseAction) {
|
||
case "none":
|
||
break;
|
||
case "click":
|
||
window.utools.simulateMouseClick(centerX, centerY);
|
||
break;
|
||
case "dblclick":
|
||
window.utools.simulateMouseDoubleClick(centerX, centerY);
|
||
break;
|
||
case "rightclick":
|
||
window.utools.simulateMouseRightClick(centerX, centerY);
|
||
break;
|
||
}
|
||
};
|
||
|
||
module.exports = { findImage };
|