import time

import cv2
import pytesseract
import win32gui

import setting
import win32_util
from utils import ths_util, opencv_util, ocr_util


def get_image():
    scale = 1
    always_save = False
    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()
    print(time_use_list)
    return clip_img


if __name__ == "__main__":
    pytesseract.pytesseract.tesseract_cmd = r'D:\Program Files\Tesseract-OCR\tesseract.exe'
    img = get_image()
    for i in range(100):
        start_time = time.time()
        # text = pytesseract.image_to_string(img, config=config)
        text = ocr_util.recognize_code(img)
        print((time.time() - start_time) * 1000, text)