|
|
|
QQ灏忔父鎴忚剼鏈紙杩炶繛鐪- import cv2import numpy as npimport win32apiimport win32guiimport win32confrom PIL import ImageGrabimport timeimport randomimport sysWINDOW_TITLE = "QQ娓告垙 - 杩炶繛鐪嬭鑹茬増"TIME_INTERVAL_MAX = 0TIME_INTERVAL_MIN = 0MARGIN_LEFT = 10MARGIN_HEIGHT = 180H_NUM = 19V_NUM = 11POINT_WIDTH = 31POINT_HEIGHT = 35EMPTY_ID = 0SUB_LT_X = 8SUB_LT_Y = 8SUB_RB_X = 27SUB_RB_Y = 27MAX_ROUND = 200def getGameWindow(): window = win32gui.FindWindow(None, WINDOW_TITLE) while not window: print('Failed to locate the game window , reposition the game window after 10 seconds...') time.sleep(10) window = win32gui.FindWindow(None, WINDOW_TITLE) win32gui.SetForegroundWindow(window) pos = win32gui.GetWindowRect(window) print("Game windows at " + str(pos)) return (pos[0], pos[1])def getScreenImage(): print('Shot screen...') scim = ImageGrab.grab() scim.save('screen.png') return cv2.imread("screen.png")def getAllSquare(screen_image, game_pos): print('Processing pictures...') game_x = game_pos[0] + MARGIN_LEFT game_y = game_pos[1] + MARGIN_HEIGHT all_square = [] for x in range(0, H_NUM): for y in range(0, V_NUM): square = screen_image[game_y + y * POINT_HEIGHT:game_y + (y + 1) * POINT_HEIGHT, game_x + x * POINT_WIDTH:game_x + (x + 1) * POINT_WIDTH] all_square.append(square) finalresult = [] for square in all_square: s = square[SUB_LT_Y:SUB_RB_Y, SUB_LT_X:SUB_RB_X] finalresult.append(s) return finalresultdef isImageExist(img, img_list): i = 0 for existed_img in img_list: b = np.subtract(existed_img, img) if not np.any(b): return i i = i + 1 return -1def getAllSquareTypes(all_square): print("Init pictures types...") types = [] number = [] nowid = 0 for square in all_square: nid = isImageExist(square, types) if nid == -1: types.append(square) number.append(1) else: number[nid] = number[nid] + 1 if (number[nid] > number[nowid]): nowid = nid global EMPTY_ID EMPTY_ID = nowid print('EMPTY_ID = ' + str(EMPTY_ID)) return typesdef getAllSquareRecord(all_square_list, types): print("Change map...") record = [] line = [] for square in all_square_list: num = 0 for type in types: res = cv2.subtract(square, type) if not np.any(res): line.append(num) break num += 1 if len(line) == V_NUM: print(line) record.append(line) line = [] return recorddef canConnect(x1, y1, x2, y2, r): result = r[:] if result[x1][y1] == EMPTY_ID or result[x2][y2] == EMPTY_ID: return False if x1 == x2 and y1 == y2: return False if result[x1][y1] != result[x2][y2]: return False if horizontalCheck(x1, y1, x2, y2, result): return True if verticalCheck(x1, y1, x2, y2, result): return True if turnOnceCheck(x1, y1, x2, y2, result): return True if turnTwiceCheck(x1, y1, x2, y2, result): return True return Falsedef horizontalCheck(x1, y1, x2, y2, result): if x1 == x2 and y1 == y2: return False if x1 != x2: return False startY = min(y1, y2) endY = max(y1, y2) if (endY - startY) == 1: return True for i in range(startY + 1, endY): if result[x1][i] != EMPTY_ID: return False return Truedef verticalCheck(x1, y1, x2, y2, result): if x1 == x2 and y1 == y2: return False if y1 != y2: return False startX = min(x1, x2) endX = max(x1, x2) if (endX - startX) == 1: return True for i in range(startX + 1, endX): if result[i][y1] != EMPTY_ID: return False return Truedef turnOnceCheck(x1, y1, x2, y2, result): if x1 == x2 or y1 == y2: return False cx = x1 cy = y2 dx = x2 dy = y1 if result[cx][cy] == EMPTY_ID: if horizontalCheck(x1, y1, cx, cy, result) and verticalCheck(cx, cy, x2, y2, result): return True if result[dx][dy] == EMPTY_ID: if verticalCheck(x1, y1, dx, dy, result) and horizontalCheck(dx, dy, x2, y2, result): return True return Falsedef turnTwiceCheck(x1, y1, x2, y2, result): if x1 == x2 and y1 == y2: return False for i in range(0, len(result)): for j in range(0, len(result[1])): if result[i][j] != EMPTY_ID: continue if i != x1 and i != x2 and j != y1 and j != y2: continue if (i == x1 and j == y2) or (i == x2 and j == y1): continue if turnOnceCheck(x1, y1, i, j, result) and ( horizontalCheck(i, j, x2, y2, result) or verticalCheck(i, j, x2, y2, result)): return True if turnOnceCheck(i, j, x2, y2, result) and ( horizontalCheck(x1, y1, i, j, result) or verticalCheck(x1, y1, i, j, result)): return True return Falsedef autoRelease(result, game_x, game_y): for i in range(0, len(result)): for j in range(0, len(result[0])): if result[i][j] != EMPTY_ID: for m in range(0, len(result)): for n in range(0, len(result[0])): if result[m][n] != EMPTY_ID: if canConnect(i, j, m, n, result): result[i][j] = EMPTY_ID result[m][n] = EMPTY_ID print('Remove 锛' + str(i + 1) + ',' + str(j + 1) + ' and ' + str(m + 1) + ',' + str( n + 1)) x1 = game_x + j * POINT_WIDTH y1 = game_y + i * POINT_HEIGHT x2 = game_x + n * POINT_WIDTH y2 = game_y + m * POINT_HEIGHT win32api.SetCursorPos((x1 + 15, y1 + 18)) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, x1 + 15, y1 + 18, 0, 0) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, x1 + 15, y1 + 18, 0, 0) time.sleep(random.uniform(TIME_INTERVAL_MIN, TIME_INTERVAL_MAX)) win32api.SetCursorPos((x2 + 15, y2 + 18)) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTDOWN, x2 + 15, y2 + 18, 0, 0) win32api.mouse_event(win32con.MOUSEEVENTF_LEFTUP, x2 + 15, y2 + 18, 0, 0) time.sleep(random.uniform(TIME_INTERVAL_MIN, TIME_INTERVAL_MAX)) return True return Falsedef autoRemove(squares, game_pos): game_x = game_pos[0] + MARGIN_LEFT game_y = game_pos[1] + MARGIN_HEIGHT while True: if not autoRelease(squares, game_x, game_y): return sys.exit()if __name__ == '__main__': random.seed() game_pos = getGameWindow() time.sleep(1) screen_image = getScreenImage() all_square_list = getAllSquare(screen_image, game_pos) types = getAllSquareTypes(all_square_list) result = np.transpose(getAllSquareRecord(all_square_list, types)) print('The total elimination amount is ' + str(autoRemove(result, game_pos))) win32api.system("pause")
|
|
发表于 2023-8-26 21:29:47