__author__ = '周雁冰'
import os
import PIL,numpy
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation
import time


need_update = True

# 獲取手機(jī)截圖
def get_screen_image():
 os.system('adb shell screencap -p /sdcard/screen.png') # 獲取手機(jī)當(dāng)前界面截圖
 os.system('adb pull /sdcard/screen.png') # 下載當(dāng)前截圖到電腦當(dāng)前文件夾下
 return numpy.array(PIL.Image.open('screen.png')) #轉(zhuǎn)為array返回


# 計算弦的長度
def jump_to_next(point1, point2):
 x1, y1 = point1; x2, y2 = point2
 distance = ((x2-x1)**2 + (y2-y1)**2)**0.5 # 計算弦長度
 os.system('adb shell input swipe 320 410 320 410 {}'.format(int(distance*1))) # 按下橫縱左邊 放開橫縱坐標(biāo) 按壓時間 2K的屏幕彈跳系數(shù)為1


# 綁定鼠標(biāo)單擊事件
def on_calck(event, coor=[]): # [(x,y),(x2,y2)]
 global need_update
 coor.append((event.xdata, event.ydata)) # 獲取x和y坐標(biāo)位置放入coor數(shù)組中
 if len(coor) == 2:
 jump_to_next(coor.pop(), coor.pop()) # 獲取到兩個坐標(biāo)后計算長度并清空數(shù)組
 need_update = True 


def update_screen(frame): # 更新圖片
 global need_update
 if need_update:
 time.sleep(1) # 因?yàn)樘S需要時間所以這里需要休眠1s，然后重新獲取圖片
 axes_image.set_array(get_screen_image())
 need_update = False
 return axes_image, # 返回元祖


figure = plt.figure() # 創(chuàng)建一個空白的的圖片對象/創(chuàng)建畫布
axes_image = plt.imshow(get_screen_image(), animated=True) # 把獲取的圖片放進(jìn)坐標(biāo)軸
figure.canvas.mpl_connect('button_press_event', on_calck)
ani = FuncAnimation(figure, update_screen, interval=50, blit=True) # 實(shí)例化 FuncAnimation更新畫布圖片 50為50ms
plt.show() # 展示坐標(biāo)圖