check/check3.2.py at master · kyrixn/check · 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
104
#coding=utf-8

import random
from time import sleep
import numpy as np
from cv2 import split,merge,cvtColor,inRange,medianBlur,threshold,findContours,boundingRect,waitKey,destroyAllWindows,COLOR_BGR2HSV,RETR_EXTERNAL,CHAIN_APPROX_SIMPLE
from pyautogui import moveTo,doubleClick,size
from PIL import ImageGrab

cnt=int(0)
rt=[[0]*2]*2
kd=[0]*2
pre =[0]*4
now =[0]*4
lower_blue=[0]*2
upper_blue=[0]*2
WIDE,HEIGHT=size()
mark=False

cnt =int(1)
BOX = (0, 0, WIDE, HEIGHT)
kd[0]=True;kd[1]=False
rt[0]=[4.3,4.6];rt[1]=[5.05,5.5]

#HSV: 色相0-180度，饱和度0-255对应0-180，亮度0-255对应0-180 别搞错了
#winddows
lower_blue[1] = np.array([99,123,255])
upper_blue[1] = np.array([105,205,255])
#原：145 #现：205
#android:197,79,96
#629 139 433 97 198 45 143 32 342 78 459 104 646 143 940 207
lower_blue[0] = np.array([98,200,237])
upper_blue[0] = np.array([100,202,245])

orgimg = np.array(ImageGrab.grab(bbox=BOX))

ver=int(input("Mod[0]And, [1]windows: "))
if ver== 3:
    mark=True
    ver=1

def change():           #B,R通道互換
    global orgimg
    b, g, r = split(orgimg)
    orgimg = merge([r, g, b])

def judg(xx,yy,ww,hh):
    global ver
    if hh !=0 and ww/hh >rt[ver][0] and ww/hh <rt[ver][1] and (kd[ver] or xx*yy >=9020):
        return True
    else:
        return False
def delay():
    if mark:
        sleep(0.2)
    else:
        tm =random.random()*2.7+0.3
        print("delay： "+str(tm)+"s")
        sleep(tm)

change()
print("resolution: "+str(WIDE)+"*"+str(HEIGHT))
while True:
    sleep(0.2)
    orgimg = np.array(ImageGrab.grab(bbox=BOX))
    change()
    # orgimg =cv2.imread("img1.jpg")
    #图像是否可拉伸
    # cv2.namedWindow("img", cv2.WINDOW_NORMAL)

    img = cvtColor(orgimg, COLOR_BGR2HSV)
    mask = inRange(img, lower_blue[ver], upper_blue[ver])
    mask = medianBlur(mask,5)
    ret,btton=threshold(mask,254,255,0)
    contours, hiera=findContours(btton, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE)

    x=y=w=h=0
    flag =False
    for tmp in contours:
        x, y, w, h = boundingRect(tmp)
        if judg(x,y,w,h):
            flag=True;break
    # x, y, w, h = cv2.boundingRect(btton)
    if flag:
        now[0]=x;now[1]=y;now[2]=w;now[3]=h
        if now != pre:
            print('\a')
            delay()
            moveTo(int(x+w/2),int(y+h/2))
            doubleClick(button='left')

            print(str(x+w/2), str(y+h/2), str(w), str(h))
            print("CHECK! ---------- 第"+str(++cnt)+"次")

            pre[0]=now[0];pre[1]=now[1];pre[2]=now[2];pre[3]=now[3]
            cnt=cnt+1
        sleep(0.2)

        # cv2.rectangle(orgimg, (x, y), (x + w, y + h), (0, 0, 255), 2)

    # cv2.imshow("img", cv2.cvtColor(btton, cv2.COLOR_BGR2RGB))
    if waitKey(25) & 0xFF == ord("q"):
        destroyAllWindows()
        break