111 lines
4.3 KiB
JavaScript
111 lines
4.3 KiB
JavaScript
|
|
// 自定义 basename 函数
|
|
function basename(filePath) {
|
|
const lastSlashIndex = filePath.lastIndexOf('\\'); // 或者使用 '/'，取决于你的路径分隔符
|
|
return filePath.substring(lastSlashIndex + 1);
|
|
}
|
|
|
|
// 检查路径是否存在
|
|
function pathExists(path) {
|
|
try {
|
|
const entries = file.readPathSync(path); // 尝试读取路径内容
|
|
return entries !== undefined && entries.length >= 0;
|
|
} catch (error) {
|
|
return false; // 如果读取失败，返回 false
|
|
}
|
|
}
|
|
|
|
// 自定义 readAllFilePaths 函数
|
|
function readAllFilePaths(dirPath, currentDepth = 0, maxDepth = 3, includeExtensions = ['.png', '.jpg', '.jpeg', '.bmp']) {
|
|
if (!pathExists(dirPath)) {
|
|
log.error(`目录 ${dirPath} 不存在`);
|
|
return [];
|
|
}
|
|
|
|
try {
|
|
const entries = file.readPathSync(dirPath); // 读取目录内容，返回的是完整路径
|
|
const filePaths = [];
|
|
|
|
for (const entry of entries) {
|
|
const isDirectory = pathExists(entry); // 如果路径存在且返回的是数组，则认为是目录
|
|
if (isDirectory && currentDepth < maxDepth) {
|
|
filePaths.push(...readAllFilePaths(entry, currentDepth + 1, maxDepth, includeExtensions)); // 递归读取子目录
|
|
} else if (!isDirectory) {
|
|
const fileExtension = entry.substring(entry.lastIndexOf('.'));
|
|
if (includeExtensions.includes(fileExtension.toLowerCase())) {
|
|
filePaths.push(entry); // 添加文件路径
|
|
}
|
|
}
|
|
}
|
|
|
|
return filePaths;
|
|
} catch (error) {
|
|
log.error(`读取目录 ${dirPath} 时发生错误: ${error}`);
|
|
return [];
|
|
}
|
|
}
|
|
|
|
// 定义一个函数用于识别图像
|
|
async function recognizeImage(recognitionObject, timeout = 5000) {
|
|
let startTime = Date.now();
|
|
while (Date.now() - startTime < timeout) {
|
|
try {
|
|
// 尝试识别图像
|
|
let imageResult = captureGameRegion().find(recognitionObject);
|
|
if (imageResult && imageResult.x !== 0 && imageResult.y !== 0 && imageResult.width !== 0 && imageResult.height !== 0) {
|
|
await drawAndClearRedBox(imageResult, 500);// 调用异步函数绘制红框并延时清除
|
|
log.info(`成功识别图像，坐标: x=${imageResult.x}, y=${imageResult.y}, width=${imageResult.width}, height=${imageResult.height}`);
|
|
return { success: true, x: imageResult.x, y: imageResult.y, width: imageResult.width, height: imageResult.height};
|
|
}
|
|
} catch (error) {
|
|
log.error(`识别图像时发生异常: ${error.message}`);
|
|
}
|
|
await sleep(10); // 短暂延迟，避免过快循环
|
|
}
|
|
log.warn(`经过多次尝试，仍然无法识别图像`);
|
|
return { success: false };
|
|
}
|
|
|
|
// 定义一个异步函数来绘制红框并延时清除
|
|
async function drawAndClearRedBox(result, delay) {
|
|
// 绘制红框
|
|
let drawRegion = captureGameRegion().DeriveCrop(result.x, result.y, result.width, result.height).DrawSelf("icon");
|
|
|
|
// 延时
|
|
await sleep(delay);
|
|
|
|
// 清除红框
|
|
if (drawRegion) {
|
|
drawRegion = captureGameRegion().DeriveCrop(0, 0, 0, 0).DrawSelf("icon");
|
|
drawRegion = null; // 释放对象
|
|
}
|
|
}
|
|
// 主函数
|
|
(async function () {
|
|
setGameMetrics(1920, 1080, 1);
|
|
|
|
// 读取 assets 文件夹下的所有图标路径
|
|
const assetsDir = "assets";
|
|
const imagePaths = readAllFilePaths(assetsDir, 0, 1);
|
|
|
|
// 遍历每个图标
|
|
for (const imagePath of imagePaths) {
|
|
const fileName = basename(imagePath); // 获取文件名
|
|
// log.info(`正在处理图标: ${fileName}`);
|
|
|
|
// 创建识别对象
|
|
let recognitionObject = RecognitionObject.TemplateMatch(file.ReadImageMatSync(imagePath));
|
|
|
|
// 识别图标
|
|
let result = await recognizeImage(recognitionObject, 50);
|
|
if (result.success) {
|
|
// 输出坐标和识图范围推荐
|
|
log.info(`图标 ${fileName} 的坐标: x=${result.x}, y=${result.y}, width=${result.width}, height=${result.height}，识图范围推荐: ${result.x-1}, ${result.y-1}, ${result.width+2}, ${result.height+2}`);
|
|
} else {
|
|
// log.warn(`未能识别到图标 ${fileName}`);
|
|
}
|
|
|
|
await sleep(50); // 确保识别之间有足够的时间间隔
|
|
}
|
|
})();
|