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:
|
"""
|
hwnd = ths_util.get_ths_main_content_hwnd()
|
if not hwnd:
|
raise Exception("看盘页面句柄未获取到")
|
# 句柄截图
|
# 实际位置(左,上,右,下)
|
rect = win32gui.GetWindowRect(hwnd)
|
|
# 格式为:（上边距,宽度,高度,缩放倍数）
|
rect_ = setting.get_ths_auto_code_rect()
|
# 测试
|
# 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)
|
if int(time.time()) % 5 == 0:
|
cv2.imwrite(f"test2.png", opencv_util.gray_img(img))
|
if always_save:
|
cv2.imwrite(f"test22.png", opencv_util.gray_img(img))
|
|
clip_img, details = opencv_util.clip_ths_code_area(img)
|
start_time = time.time()
|
code = ocr_util.recognize_code(clip_img)
|
use_time = round((time.time() - start_time) * 1000)
|
if code is None:
|
code = ocr_util.recognize_code(img)
|
if code != last_ocr_code[0]:
|
print("保存图片", code)
|
debug_logger.info(f"代码识别结果:{code} 识别时间：{use_time}ms")
|
last_ocr_code[0] = code
|
cv2.imwrite(f"datas/test/{code}.png", opencv_util.gray_img(img))
|
return code
|
