find_blood_location.py

# -*- coding: utf-8 -*-
"""
Created on Wed Apr  8 09:45:04 2020

@author: pang
"""

import numpy as np
from PIL import ImageGrab
import cv2
import time
import grabscreen
import os


def boss_blood_count(boss_gray):
    boss_blood = 0
    for boss_bd_num in boss_gray[0]:
    # boss blood gray pixel 65~75
    # 血量灰度值65~75 
        # print(boss_bd_num)
        if boss_bd_num > 65 and boss_bd_num < 75:
            boss_blood += 1
    return boss_blood

wait_time = 5
L_t = 3

window_size = (75, 125, 260, 27)#384,344  192,172 96,86
x, y, width, height = 80, 130, 260, 27

x2,y2,width,height = 370, 130, 260,27
# for i in list(range(wait_time))[::-1]:
#     print(i+1)
#     time.sleep(1)
def self_blood_count(self_gray):
    self_blood = 0
    for self_bd_num in self_gray[10]:
        self_bd = self_bd_num[0]
        # self blood gray pixel 80~98
        # 血量灰度值80~98
        if self_bd > 105 and self_bd < 109:
            self_blood += 2
        if self_bd == 105:
            self_blood += 1
    return self_blood
last_time = time.time()
res = []
res_blood = []
i =1
while(True):

    #printscreen = np.array(ImageGrab.grab(bbox=(window_size)))
    #printscreen_numpy = np.array(printscreen_pil.getdata(),dtype='uint8')\
    #.reshape((printscreen_pil.size[1],printscreen_pil.size[0],3))
    #pil格式耗时太长
    
    screen_gray = grabscreen.grab_screen(x2,y2,width,height)#灰度图像收集
    if screen_gray[10][30][0] ==0:
        cv2.waitKey(50)
        screen_gray = grabscreen.grab_screen(x,y,width,height)
    # print(screen_gray.shape)
    # print(screen_gray[10])
    
    res.append(screen_gray[10][30][0])
    # screen_reshape = cv2.resize(screen_gray,(96,86))
    self_blood = self_blood_count(screen_gray)
    # print("blood ====",self_blood)
    # boss_blood = boss_blood_count(screen_gray)
    res_blood.append((self_blood_count(screen_gray),screen_gray[10][30][0]))
    cv2.imshow('window1',screen_gray)
    #cv2.imshow('window3',printscreen)
    #cv2.imshow('window2',screen_reshape)
    # print("res ===",res)
    # tuples = [tuple(a) for a in res]
    print(np.unique(res))
    print("res_blood ",res_blood)
    #测试时间用
    print('loop took {} seconds'.format(time.time()-last_time))
    last_time = time.time()
    
    
    if cv2.waitKey(5) & 0xFF == ord('q'):
        break
cv2.waitKey()# 视频结束后，按任意键退出
cv2.destroyAllWindows()