Method prepare_batch_ocr

source/MaaFramework/Task/PipelineTask.cpp:333–368

}

std::optional<PipelineTask::BatchOCRPlan> PipelineTask::prepare_batch_ocr(const std::vector<MAA_RES_NS::NodeAttr>& list)
{
    using namespace MAA_RES_NS::Recognition;

    if (!context_) {
        return std::nullopt;
    }

    OCRCollectContext ctx;

    for (const auto& node : list) {
        auto data_opt = context_->get_pipeline_data(node);
        if (!data_opt) {
            continue;
        }
        const auto& data = *data_opt;

        if (!data.enabled) {
            continue;
        }

        if (!context_->check_hit_count(data)) {
            continue;
        }

        collect_ocr_from_reco(ctx, data.name, data.reco_type, data.reco_param);
    }

    if (ctx.plan.entries.size() < 2) {
        LogDebug << "batch OCR not needed, eligible OCR nodes < 2" << VAR(ctx.plan.entries.size());
        return std::nullopt;
    }

    LogInfo << "prepared batch OCR plan" << VAR(ctx.plan.node_names) << VAR(ctx.plan.model);
    return ctx.plan;
}

void PipelineTask::try_add_ocr_node(OCRCollectContext& ctx, const std::string& name, const MAA_VISION_NS::OCRerParam& param)
{