import time
|
|
import cv2
|
import win32gui
|
|
import setting
|
import win32_util
|
from log import debug_logger
|
from utils import ths_util, opencv_util, ocr_util
|
|
scale = setting.get_screen_scale()
|
|
last_ocr_code = ["000000"]
|
|
|
def ocr_ths_code(always_save=False):
|
"""
|
识别同花顺代码
|
:return:
|
"""
|
start_time = time.time()
|
time_use_list = []
|
hwnd = ths_util.get_ths_main_content_hwnd_cached()
|
time_use_list.append(("获取句柄", time.time() - start_time))
|
start_time = time.time()
|
if not hwnd:
|
raise Exception("看盘页面句柄未获取到")
|
# 句柄截图
|
# 实际位置(左,上,右,下)
|
rect = win32gui.GetWindowRect(hwnd)
|
|
# 格式为:（上边距,宽度,高度,缩放倍数）
|
rect_ = setting.get_ths_auto_code_rect()
|
|
time_use_list.append(("获取配置参数", time.time() - start_time))
|
start_time = time.time()
|
# 测试
|
# if len(rect_) > 3:
|
# scale = rect_[3]
|
width = int((rect[2] - rect[0]) * scale)
|
left = width - rect_[1]
|
img = win32_util.window_capture(hwnd, (left, rect_[0], width - 1, rect_[0] + rect_[2]), scale, always_save)
|
time_use_list.append(("截图", time.time() - start_time))
|
start_time = time.time()
|
|
if always_save:
|
cv2.imwrite(f"test_clip_gray.png", opencv_util.gray_img(img))
|
clip_img, details = opencv_util.clip_ths_code_area(img)
|
if always_save:
|
cv2.imwrite(f"test_clip_gray_clip.png", clip_img)
|
|
time_use_list.append(("分割", time.time() - start_time))
|
start_time = time.time()
|
code = ocr_util.recognize_code(clip_img)
|
time_use_list.append(("识别", time.time() - start_time))
|
start_time = time.time()
|
if code is None:
|
code = ocr_util.recognize_code(img)
|
time_use_list.append(("再次识别", time.time() - start_time))
|
start_time = time.time()
|
if code != last_ocr_code[0]:
|
msg = " ".join([f"{x[0]}:{int(x[1]*1000)}ms" for x in time_use_list])
|
debug_logger.info(f"代码识别结果:{code} {msg}")
|
last_ocr_code[0] = code
|
if always_save:
|
cv2.imwrite(f"datas/test/{code}.png", opencv_util.gray_img(img))
|
return code
|
