1
0

wechat_jump_auto.py 7.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191
  1. # coding: utf-8
  2. import os
  3. import shutil
  4. import time
  5. import math
  6. from PIL import Image, ImageDraw
  7. import random
  8. import json
  9. # === 思路 ===
  10. # 核心:每次落稳之后截图,根据截图算出棋子的坐标和下一个块顶面的中点坐标,
  11. # 根据两个点的距离乘以一个时间系数获得长按的时间
  12. # 识别棋子:靠棋子的颜色来识别位置,通过截图发现最下面一行大概是一条直线,就从上往下一行一行遍历,
  13. # 比较颜色(颜色用了一个区间来比较)找到最下面的那一行的所有点,然后求个中点,
  14. # 求好之后再让 Y 轴坐标减小棋子底盘的一半高度从而得到中心点的坐标
  15. # 识别棋盘:靠底色和方块的色差来做,从分数之下的位置开始,一行一行扫描,由于圆形的块最顶上是一条线,
  16. # 方形的上面大概是一个点,所以就用类似识别棋子的做法多识别了几个点求中点,
  17. # 这时候得到了块中点的 X 轴坐标,这时候假设现在棋子在当前块的中心,
  18. # 根据一个通过截图获取的固定的角度来推出中点的 Y 坐标
  19. # 最后:根据两点的坐标算距离乘以系数来获取长按时间(似乎可以直接用 X 轴距离)
  20. # TODO: 解决定位偏移的问题
  21. # TODO: 看看两个块中心到中轴距离是否相同,如果是的话靠这个来判断一下当前超前还是落后,便于矫正
  22. # TODO: 一些固定值根据截图的具体大小计算
  23. # TODO: 直接用 X 轴距离简化逻辑
  24. with open('config.json', 'r') as f:
  25. config = json.load(f)
  26. # Magic Number,不设置可能无法正常执行,请根据具体截图从上到下按需设置
  27. # 截图中刚好低于分数显示区域的 Y 坐标,300 是 1920x1080 的值,2K 屏、全面屏请根据实际情况修改
  28. under_game_score_y = config['under_game_score_y']
  29. press_coefficient = config['press_coefficient'] # 长按的时间系数,请自己根据实际情况调节
  30. piece_base_height_1_2 = config['piece_base_height_1_2'] # 二分之一的棋子底座高度,可能要调节
  31. # 棋子的宽度,比截图中量到的稍微大一点比较安全,可能要调节
  32. piece_body_width = config['piece_body_width']
  33. # 模拟按压的起始点坐标,需要自动重复游戏请设置成“再来一局”的坐标
  34. swipe_x1, swipe_y1, swipe_x2, swipe_y2 = 320, 410, 320, 410
  35. piece_base_height_1_2 = 25 # 二分之一的棋子底座高度,可能要调节
  36. piece_body_width = 80 # 棋子的宽度,比截图中量到的稍微大一点比较安全,可能要调节
  37. # 下面的 (353, 859) 和 (772, 1100) 是游戏截图里的两个台子的中点坐标,主要用来算角度,可能要调节
  38. sample_board_x1, sample_board_y1, sample_board_x2, sample_board_y2 = 353, 859, 772, 1100
  39. screenshot_backup_dir = 'debug/'
  40. if not os.path.isdir(screenshot_backup_dir):
  41. os.mkdir(screenshot_backup_dir)
  42. def pull_screenshot():
  43. os.system('adb shell screencap -p /sdcard/wx_temp.png')
  44. os.system('adb pull /sdcard/wx_temp.png .')
  45. def backup_screenshot(ts):
  46. # 为了方便失败的时候 debug
  47. if not os.path.isdir(screenshot_backup_dir):
  48. os.mkdir(screenshot_backup_dir)
  49. shutil.copy('wx_temp.png', '{}{}.png'.format(screenshot_backup_dir, ts))
  50. def save_debug_creenshot(ts, im, piece_x, piece_y, board_x, board_y):
  51. draw = ImageDraw.Draw(im)
  52. draw.line((piece_x, piece_y) + (board_x, board_y), fill=2, width=3)
  53. del draw
  54. im.save("{}{}_d.png".format(screenshot_backup_dir, ts))
  55. def set_button_position(im):
  56. # 将swipe设置为 `再来一局` 按钮的位置
  57. global swipe_x1, swipe_y1, swipe_x2, swipe_y2
  58. w, h = im.size
  59. left = w / 2
  60. top = 1003 * (h / 1280.0) + 10
  61. swipe_x1, swipe_y1, swipe_x2, swipe_y2 = left, top, left, top
  62. def jump(distance):
  63. print('distace:', distance)
  64. tmp = 0
  65. if distance < 220:
  66. tmp = 38
  67. elif distance < 380:
  68. tmp = 30
  69. elif distance < 480:
  70. tmp = 25
  71. elif distance < 515:
  72. tmp = 14
  73. elif distance < 580:
  74. tmp = 12
  75. distance = distance + tmp
  76. press_time = distance * press_coefficient
  77. press_time = max(press_time, 200) # 设置 200 ms 是最小的按压时间
  78. press_time = int(press_time)
  79. cmd = 'adb shell input swipe {} {} {} {} {}'.format(
  80. swipe_x1, swipe_y1, swipe_x2, swipe_y2, press_time)
  81. # print(cmd)
  82. os.system(cmd)
  83. def find_piece_and_board(im):
  84. w, h = im.size
  85. piece_x_sum = 0
  86. piece_x_c = 0
  87. piece_y_max = 0
  88. board_x = 0
  89. board_y = 0
  90. scan_x_border = int(w / 8) # 扫描棋子时的左右边界
  91. scan_start_y = 0 # 扫描的起始y坐标
  92. im_pixel = im.load()
  93. # 以50px步长,尝试探测scan_start_y
  94. for i in range(under_game_score_y, h, 50):
  95. last_pixel = im_pixel[0, i]
  96. for j in range(1, w):
  97. pixel = im_pixel[j, i]
  98. # 不是纯色的线,则记录scan_start_y的值,准备跳出循环
  99. if pixel[0] != last_pixel[0] or pixel[1] != last_pixel[1] or pixel[2] != last_pixel[2]:
  100. scan_start_y = i - 50
  101. break
  102. if scan_start_y:
  103. break
  104. # print("scan_start_y: ", scan_start_y)
  105. # 从scan_start_y开始往下扫描,棋子应位于屏幕上半部分,这里暂定不超过2/3
  106. for i in range(scan_start_y, int(h * 2 / 3)):
  107. for j in range(scan_x_border, w - scan_x_border): # 横坐标方面也减少了一部分扫描开销
  108. pixel = im_pixel[j, i]
  109. # 根据棋子的最低行的颜色判断,找最后一行那些点的平均值,这个颜色这样应该 OK,暂时不提出来
  110. if (50 < pixel[0] < 60) and (53 < pixel[1] < 63) and (95 < pixel[2] < 110):
  111. piece_x_sum += j
  112. piece_x_c += 1
  113. piece_y_max = max(i, piece_y_max)
  114. if not all((piece_x_sum, piece_x_c)):
  115. return 0, 0, 0, 0
  116. piece_x = piece_x_sum / piece_x_c
  117. piece_y = piece_y_max - piece_base_height_1_2 # 上移棋子底盘高度的一半
  118. for i in range(scan_start_y, h):
  119. last_pixel = im_pixel[0, i]
  120. if board_x or board_y:
  121. break
  122. board_x_sum = 0
  123. board_x_c = 0
  124. for j in range(w):
  125. pixel = im_pixel[j, i]
  126. # 修掉脑袋比下一个小格子还高的情况的 bug
  127. if abs(j - piece_x) < piece_body_width:
  128. continue
  129. # 修掉圆顶的时候一条线导致的小 bug,这个颜色判断应该 OK,暂时不提出来
  130. if abs(pixel[0] - last_pixel[0]) + abs(pixel[1] - last_pixel[1]) + abs(pixel[2] - last_pixel[2]) > 10:
  131. board_x_sum += j
  132. board_x_c += 1
  133. if board_x_sum:
  134. board_x = board_x_sum / board_x_c
  135. # 按实际的角度来算,找到接近下一个 board 中心的坐标
  136. board_y = piece_y - abs(board_x - piece_x) * abs(sample_board_y1 -
  137. sample_board_y2) / abs(sample_board_x1 - sample_board_x2)
  138. if not all((board_x, board_y)):
  139. return 0, 0, 0, 0
  140. return piece_x, piece_y, board_x, board_y
  141. def main():
  142. while True:
  143. pull_screenshot()
  144. im = Image.open("./wx_temp.png")
  145. # 获取棋子和 board 的位置
  146. piece_x, piece_y, board_x, board_y = find_piece_and_board(im)
  147. ts = int(time.time())
  148. # print(ts, piece_x, piece_y, board_x, board_y)
  149. set_button_position(im)
  150. jump(math.sqrt((board_x - piece_x) ** 2 + (board_y - piece_y) ** 2))
  151. save_debug_creenshot(ts, im, piece_x, piece_y, board_x, board_y)
  152. backup_screenshot(ts)
  153. print('\n')
  154. time.sleep(random.uniform(1, 3.1)) # 为了保证截图的时候应落稳了,多延迟一会儿
  155. if __name__ == '__main__':
  156. main()