AutoHS/catch_screen_demo.py at main · redou12/AutoHS · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
"""
主体代码引自 Demon_Hunter 的CSDN博客, 博客URL:https://blog.csdn.net/zhuisui_woxin/article/details/84345036
"""
import sys

import cv2
import time
import math

import get_screen


PRINT_ALL_AREA_LIST = []
# AREA_LIST = [((1495, 465), (1620, 522))]
AREA_LIST = [((690, 290), (710, 310))]

POINT_LIST = [(960, 650), (1090, 1070), (705, 305)]


def get_sum(x):
    return int(x[0]) + int(x[1]) + int(x[2])


def add_line(img, width, height):
    for i in range(1, math.floor(width / 100) + 1):
        cv2.line(img, pt1=(i * 100, 0), pt2=(i * 100, height), color=(200, 200, 200), thickness=1)
        cv2.putText(img, str(i * 100), (i * 100 - 30, 30), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255), 1)

    for i in range(1, math.floor(height / 100) + 1):
        cv2.line(img, pt1=(0, i * 100), pt2=(width, i * 100), color=(200, 200, 200), thickness=1)
        cv2.putText(img, str(i * 100), (0, i * 100), cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 255), 1)

    return img


def add_point(img, point_list):
    for pair in point_list:
        print(str(pair) + " has color: " + str(img[pair[1]][pair[0]]))
        cv2.circle(img, pair, 1, (255, 0, 0), 2, 0)


def show_area(img, top_left, bottom_right, print_out=False):
    x1, y1 = top_left
    x2, y2 = bottom_right
    tmp_img = img[y1:y2, x1:x2]
    tmp_img = tmp_img.copy()

    if print_out:
        count = 0
        for line in tmp_img:
            for pixel in line:
                if pixel[1] > 230:
                    count += 1
        print(count)

    ratio = min(round(800 / (x2 - x1)), round(500 / (y2 - y1)))
    resized_x_length = (x2 - x1) * int(ratio)
    resized_y_length = int(resized_x_length * ((y2 - y1) / (x2 - x1)))
    tmp_img = cv2.resize(tmp_img, (resized_x_length, resized_y_length))

    font_size = int(ratio) / 8
    for x in range(x1, x2, 10):
        temp_x = int(resized_x_length / (x2 - x1) * (x - x1))
        cv2.line(tmp_img, pt1=(temp_x, 0), pt2=(temp_x, resized_y_length), color=(200, 200, 200), thickness=1)
        cv2.putText(tmp_img, str(x), (temp_x, 10), cv2.FONT_HERSHEY_COMPLEX, font_size, (0, 0, 255), 1)

    for y in range(y1, y2, 10):
        temp_y = int(resized_y_length * (y - y1) / (y2 - y1))
        cv2.line(tmp_img, pt1=(0, temp_y), pt2=(resized_x_length, temp_y), color=(200, 200, 200), thickness=1)
        cv2.putText(tmp_img, str(y), (0, temp_y + 5), cv2.FONT_HERSHEY_COMPLEX, font_size, (0, 255, 0), 1)

    cv2.imshow("area", tmp_img)
    cv2.waitKey(0)
    cv2.destroyWindow("area")


if __name__ == "__main__":
    im_opencv = get_screen.catch_screen()
    if im_opencv is None:
        print("未找到应用")
        sys.exit(-1)

    add_line(im_opencv, 1920, 1080)
    add_point(im_opencv, POINT_LIST)

    cv2.imshow("total", im_opencv)  # 显示
    cv2.waitKey(0)
    cv2.destroyWindow("total")
    time.sleep(0.2)

    for area in AREA_LIST:
        if len(area) < 2 or len(area) > 3:
            print("[Usage]: ((left-top), (botton-right), [mouse-position])")
        top_left, bottom_right = area
        show_area(im_opencv, top_left, bottom_right)

    for area in PRINT_ALL_AREA_LIST:
        if len(area) < 2 or len(area) > 3:
            print("[Usage]: ((left-top), (botton-right), [mouse-position])")
        top_left, bottom_right = area
        show_area(im_opencv, top_left, bottom_right, print_out=True)

    cv2.destroyAllWindows()