bettergi-scripts-list/repo/js/CD-Aware-AutoGather/lib/ocr.js

const defaultReplacementMap = {
    监: "盐",
    卵: "卯",
};

/**
 * 绘制方框以标记OCR区域
 *
 * @param {number[4] | {x: number, y: number, width: number, height: number}} ocrRegion - OCR 区域，数组或对象形式，表示 [x, y, width, height]
 * @param {object|null} [captureRegion=null] - 截图区域对象。为 null 时自动调用 captureGameRegion()
 * @throws {TypeError} 当 ocrRegion 既不是数组也不是对象时抛出。
 */
function drawOcrRegion(ocrRegion, captureRegion = null) {
    let x, y, width, height;
    if (Array.isArray(ocrRegion)) {
        [x, y, width, height] = ocrRegion;
    } else if (typeof ocrRegion === "object" && ocrRegion !== null) {
        ({ x, y, width, height } = ocrRegion);
    } else {
        throw new TypeError("'ocrRegion' must be an array [x, y, width, height] or an object with x, y, width, height properties");
    }

    let auto_created = false;
    if (captureRegion === null) {
        captureRegion = captureGameRegion();
        auto_created = true;
    }
    let region = captureRegion.DeriveCrop(x, y, width, height);
    let name = [x, y, width, height].toString();
    region.DrawSelf(name);
    region.dispose();
    if (auto_created) {
        captureRegion.dispose();
    }
}

async function getTextInRegion(ocrRegion, timeout = 5000, retryInterval = 50, replacementMap = defaultReplacementMap) {
    let x, y, width, height;
    if (Array.isArray(ocrRegion)) {
        [x, y, width, height] = ocrRegion;
    } else if (typeof ocrRegion === "object" && ocrRegion !== null) {
        ({ x, y, width, height } = ocrRegion);
    } else {
        throw new Error("Invalid parameter 'ocrRegion'");
    }

    const debugThreshold = timeout / retryInterval / 3;
    let startTime = Date.now();
    let retryCount = 0; // 重试计数
    while (Date.now() - startTime < timeout) {
        let captureRegion = captureGameRegion();
        try {
            // 尝试 OCR 识别
            let resList = captureRegion.findMulti(RecognitionObject.ocr(x, y, width, height)); // 指定识别区域
            // 遍历识别结果，检查是否找到目标文本
            for (let res of resList) {
                // 后处理：根据替换映射表检查和替换错误识别的字符
                let correctedText = res.text;
                for (let [wrongChar, correctChar] of Object.entries(replacementMap)) {
                    correctedText = correctedText.replace(new RegExp(wrongChar, "g"), correctChar);
                }

                captureRegion.dispose();
                return correctedText.trim();
            }
        } catch (error) {
            log.warn(`页面标志识别失败，正在进行第 ${retryCount} 次重试...`);
        }
        retryCount++; // 增加重试计数
        if (retryCount > debugThreshold) {
            let region = captureRegion.DeriveCrop(x, y, width, height);
            region.DrawSelf("debug");
            region.dispose();
        }
        captureRegion.dispose();
        await sleep(retryInterval);
    }
    return null;
}

async function waitForTextAppear(targetText, ocrRegion, timeout = 5000, retryInterval = 50, replacementMap = defaultReplacementMap) {
    let x, y, width, height;

    if (Array.isArray(ocrRegion)) {
        [x, y, width, height] = ocrRegion;
    } else if (typeof ocrRegion === "object" && ocrRegion !== null) {
        ({ x, y, width, height } = ocrRegion);
    } else {
        throw new Error("Invalid parameter 'ocrRegion'");
    }

    const debugThreshold = timeout / retryInterval / 3;
    let startTime = Date.now();
    let retryCount = 0; // 重试计数
    while (Date.now() - startTime < timeout) {
        let captureRegion = captureGameRegion();
        try {
            // 尝试 OCR 识别
            let resList = captureRegion.findMulti(RecognitionObject.ocr(x, y, width, height)); // 指定识别区域
            // 遍历识别结果，检查是否找到目标文本
            for (let res of resList) {
                // 后处理：根据替换映射表检查和替换错误识别的字符
                let correctedText = res.text;
                for (let [wrongChar, correctChar] of Object.entries(replacementMap)) {
                    correctedText = correctedText.replace(new RegExp(wrongChar, "g"), correctChar);
                }

                if (correctedText.includes(targetText)) {
                    captureRegion.dispose();
                    return { success: true, wait_time: Date.now() - startTime };
                }
            }
        } catch (error) {
            log.warn(`页面标志识别失败，正在进行第 ${retryCount} 次重试...`);
        }
        retryCount++; // 增加重试计数
        if (retryCount > debugThreshold) {
            let region = captureRegion.DeriveCrop(x, y, width, height);
            region.DrawSelf("debug");
            region.dispose();
        }
        captureRegion.dispose();
        await sleep(retryInterval);
    }
    return { success: false };
}

async function recognizeTextAndClick(targetText, ocrRegion, timeout = 5000, retryInterval = 50, replacementMap = defaultReplacementMap) {
    let x, y, width, height;

    if (Array.isArray(ocrRegion)) {
        [x, y, width, height] = ocrRegion;
    } else if (typeof ocrRegion === "object" && ocrRegion !== null) {
        ({ x, y, width, height } = ocrRegion);
    } else {
        throw new Error("Invalid parameter 'ocrRegion'");
    }

    const debugThreshold = timeout / retryInterval / 3;
    let startTime = Date.now();
    let retryCount = 0; // 重试计数
    while (Date.now() - startTime < timeout) {
        let captureRegion = captureGameRegion();
        try {
            // 尝试 OCR 识别
            let resList = captureRegion.findMulti(RecognitionObject.ocr(x, y, width, height)); // 指定识别区域
            // 遍历识别结果，检查是否找到目标文本
            for (let res of resList) {
                // 后处理：根据替换映射表检查和替换错误识别的字符
                let correctedText = res.text;
                for (let [wrongChar, correctChar] of Object.entries(replacementMap)) {
                    correctedText = correctedText.replace(new RegExp(wrongChar, "g"), correctChar);
                }

                if (correctedText.includes(targetText)) {
                    // 如果找到目标文本，计算并点击文字的中心坐标
                    let centerX = Math.round(res.x + res.width / 2);
                    let centerY = Math.round(res.y + res.height / 2);
                    await click(centerX, centerY);
                    await sleep(50);
                    captureRegion.dispose();
                    return { success: true, x: centerX, y: centerY };
                }
            }
        } catch (error) {
            log.warn(`页面标志识别失败，正在进行第 ${retryCount} 次重试...`);
        }
        retryCount++; // 增加重试计数
        if (retryCount > debugThreshold) {
            let region = captureRegion.DeriveCrop(x, y, width, height);
            region.DrawSelf("debug");
            region.dispose();
        }
        captureRegion.dispose();
        await sleep(retryInterval);
    }
    return { success: false };
}

async function isTextExistedInRegion(searchText, ocrRegion) {
    let x, y, width, height;
    if (Array.isArray(ocrRegion)) {
        [x, y, width, height] = ocrRegion;
    } else if (typeof ocrRegion === "object" && ocrRegion !== null) {
        ({ x, y, width, height } = ocrRegion);
    } else {
        throw new Error("Invalid parameter 'ocrRegion'");
    }
    let captureRegion = captureGameRegion();
    const result = captureRegion.find(RecognitionObject.ocr(x, y, width, height));
    captureRegion.dispose();
    if (result.text) {
        if (typeof searchText === "string") {
            return result.text.includes(searchText);
        } else if (searchText instanceof RegExp) {
            return result.text.match(searchText);
        }
    }
    return false;
}