# main.py# _*_ coding:utf-8 _*_
__author__ = 'WoLykos'from operations import *
from draw import *
from algorithm import *
import time
import random# 测试截屏
# def test_screen_cap():
#     op = Operation()
#     op.screen_cap()# 测试显示图片
def test_show_pic():draw = Draw()draw.show_pic("img/auto.png")# 测试计算欧式距离
def test_euclidean_distance():algorithm = Algorithm()p1 = (3, 4)p2 = (6, 8)d = algorithm.euclidean_distance(p1, p2)print(d)# 测试寻找关键坐标
def test_find_point():op = Operation()im = op.screen_cap()algorithm = Algorithm()start_x, start_y, end_x, end_y = algorithm.find_point(im)print("start_point:", start_x, start_y)print("end_point:", end_x, end_y)start_point = (start_x, start_y)end_point = (end_x, end_y)distance = algorithm.euclidean_distance(start_point, end_point)# print(distance)press_time = algorithm.distance_to_time(distance)op.jump(start_point, end_point, press_time)if __name__ == "__main__":# test_screen_cap()# test_show_pic()while True:# test_euclidean_distance()
        test_find_point()time.sleep(1 + 2*random.random())

# algorithm.py
# _*_ coding:utf-8 _*_
__author__ = 'WoLykos'class Algorithm:# 构造器def __int__(self):pass# 计算两点之间的欧氏距离# p1和p2表示两个点 用元组来表示def euclidean_distance(self, p1, p2):return ((p1[0]-p2[0])**2+(p1[1]-p2[1])**2)**0.5  # ((p1[0]-p2[0])**2+(p1[1]-p2[1])**2)**0.5# 寻找关键坐标# 返回值1,2 piece_x, piece_y 起跳点的坐标 170,555# 返回值3,4 board_x, board_y 目标点的坐标 395,425def find_point(self, im):# piece_x = piece_y = 0# board_x = board_y = 0# 图像的大小w, h = im.size  # (1080,1920)# 加载图像im_pixel = im.load()# 记录小人所有的点points = []# 记录y的最大值piece_y_max = 0# 1 计算出起跳点 就是小人底座的中心点# 1.1 获取小人的所有像素点中y坐标的最大值# 遍历图像中的每一个点# 遍历每一行for i in range(h // 3, h * 2 // 3):# 遍历每一列for j in range(w):pixel = im_pixel[j, i]# print("i = ", i, ",j = ", j, "pixel = ", pixel)# 判断pixel是否小人所在的位置# 当该点的RGB值约为56,56,82的时候就可以认为是小人所在的像素点了if (51 < pixel[0] < 61 and 51 < pixel[1] < 61 and 72 < pixel[2] < 102):# 把当前的点添加到points数组中points.append((j, i))  # (x,y)# 记录下y的值if i > piece_y_max:piece_y_max = i# print("piece_y_max = %d" % (piece_y_max,))# 1.2 在小人y坐标的最大值那些像素点中，计算出x的平均值，作为小人底座的x的值。bottom_x = []for x, y in points:if y == piece_y_max:bottom_x.append(x)piece_x = sum(bottom_x) // len(bottom_x)# print("piece_x = %d" % (piece_x,))# 1.3 y坐标的最大值减去一个偏移值，就作为小人底座的y值。（注意：该偏移值不同的设备是不同的，同一台设备不同场景下是一样的）piece_y = piece_y_max - 20  # 偏移值1130-110=20# print("piece_y = %d" % (piece_y,))# 2计算 目标格子的中心点# 2.1计算目标格子的x值points = []# 只取中间1/3进行扫描for i in range(h // 3, h * 2 // 3):if len(points) > 0:break# 取坐标的一个点作为背景的参照物last_pixel = im_pixel[0, i]# 逐个扫描右边的点for j in range(w):pixel = im_pixel[j, i]# 把当前点与最左边的点比较 如果RGB差异比较大 则认为是目标点# 排除该点为小人像素点56,56,82的可能性，BUGif not (54 < pixel[0] < 141 and 54 < pixel[1] < 130 and 69 < pixel[2] < 172):if (abs(pixel[0] - last_pixel[0]) + abs(pixel[1] - last_pixel[1])+ abs(pixel[2] - last_pixel[2]) > 10):points.append((j, i))top_x = []for x, y in points:top_x.append(x)board_x = sum(top_x) // len(top_x)# print("board_x = %d" % (board_x,))# 2.2计算目标格式子y值# 屏幕中心的值center_x = w / 2 + 20  # x的偏差是20center_y = h / 2 + 20  # y的偏差是20，园# 格子高和宽的比例height_per_width = 262 / 455# 计算出目标格子的y值(需要转换成整数)# 从piece_x调到board_x 如果piece_x < board_x则表示从左往右跳# 如果piece_x > board_x 则表示从右往左跳if piece_x < board_x:board_y = int(center_y - height_per_width * (board_x - center_x))else:  # 从右往左跳board_y = int(center_y + height_per_width * (board_x - center_x))# print("board_y = %d" % (board_y,))return piece_x, piece_y, board_x, board_y# 距离与时间的转换def distance_to_time(self, distance):# 当0分的时候 距离为 527.5234591939964 时间为713p = 713 / 527.5234591939964  # 该算法后面待优化press_time = distance * preturn press_time

# operations.py
# _*_ coding:utf-8 _*_
__author__ = 'WoLykos'import os
import datetimefrom PIL import Image# 实现控制安卓
class Operation:# 构造方法def __int__(self):pass# 截屏def screen_cap(self):filename = time = datetime.datetime.now().strftime("%H%M%S")+".png"# 截屏并保存到手机cmd = "adb shell screencap -p /sdcard/auto.png"os.system(cmd)# 拷贝到电脑cmd = "adb pull /sdcard/auto.png "+"img/"+filenameos.system(cmd)# 打开图像文件return Image.open("img/"+filename)# 控制屏幕进行跳动def jump(self, src, dst, press_time):# print(press_time)press_time = int(press_time)cmd = "adb shell input swipe %d %d %d %d %d" % (int(src[0]), int(src[1]),int(dst[0]), int(dst[1]),press_time)print(cmd)os.system(cmd)