From d7e3f882feb9307c22b0cd2698e45d24804b6572 Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期三, 20 七月 2022 18:37:45 +0800 Subject: [PATCH] '完善' --- ConsoleApplication/CaptureUtil.cpp | 13 ++++++++----- 1 files changed, 8 insertions(+), 5 deletions(-) diff --git a/ConsoleApplication/CaptureUtil.cpp b/ConsoleApplication/CaptureUtil.cpp index 15b68d8..3ff0e33 100644 --- a/ConsoleApplication/CaptureUtil.cpp +++ b/ConsoleApplication/CaptureUtil.cpp @@ -97,18 +97,21 @@ cv::Mat img= ImgUtil::grayImage(capture(tradeQueueFrames[i])); string path = "C:\\Users\\Administrator\\Desktop\\ocr\\trade_queue\\"; path.append(to_string(i)).append(".jpg"); - cv::imwrite(path,img); + //cv::imwrite(path,img); } } -HWND CaptureUtil::getHWND(int index) { - return level2Frames[index]; +HWND CaptureUtil::getHWND(int index, CaptureContentType type) { + if (type == CAPTURE_TYPE_L2) + return level2Frames[index]; + else + return tradeQueueFrames[index]; } -cv::Mat CaptureUtil::capture(int index) { - cv::Mat cap = capture(getHWND(index)); +cv::Mat CaptureUtil::capture(int index, CaptureContentType type) { + cv::Mat cap = capture(getHWND(index, type)); return cap; } -- Gitblit v1.8.0