js: 识图工具 (#1212)

* js,识图工具

能识别标注图片的位置，并输出推荐识别区域

* Add files via upload

repo/js/识图工具 区域坐标/main.js

@@ -0,0 +1,110 @@
+
+// 自定义 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); // 确保识别之间有足够的时间间隔
+    }
+})();

repo/js/识图工具 区域坐标/manifest.json

@@ -0,0 +1,13 @@
+{
+  "manifest_version": 1,
+  "name": "识图坐标区域",
+  "version": "1.0",
+  "description": "识别图片输出坐标区域",
+  "authors": [
+    {
+      "name": "吉吉喵",
+    }
+  ],
+  "settings_ui": "settings.json",
+  "main": "main.js",
+}

repo/js/识图工具 区域坐标/settings.json

@@ -0,0 +1,7 @@
+[
+    {
+        "name": "A1",
+        "type": "checkbox",
+        "label": "无需勾选= =\n把需要获取识别区域的图片\n放到assets文件夹下"
+    }
+]