from __future__ import print_function

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
import math
import time
import os
import cv2
import datetime

scale = 0.25

template = cv2.imread('character.png')
template = cv2.resize(template, (0, 0), fx=scale, fy=scale)
template_size = template.shape[:2]


def search(img):
 result = cv2.matchTemplate(img, template, cv2.TM_SQDIFF)
 min_val, max_val, min_loc, max_loc = cv2.minMaxLoc(result)

 cv2.rectangle(img, (min_loc[0], min_loc[1]), (min_loc[0] + template_size[1], min_loc[1] + template_size[0]), (255, 0, 0), 4)

 return img, min_loc[0] + template_size[1] / 2, min_loc[1] + template_size[0]

def pull_screenshot():
 filename = datetime.datetime.now().strftime("%H%M%S") + '.png'
 os.system('mv autojump.png {}'.format(filename))
 os.system('adb shell screencap -p /sdcard/autojump.png')
 os.system('adb pull /sdcard/autojump.png .')

def jump(distance):
 press_time = distance * 1.35
 press_time = int(press_time)
 cmd = 'adb shell input swipe 320 410 320 410 ' + str(press_time)
 print(cmd)
 os.system(cmd)

def update_data():
 global src_x, src_y

 img = cv2.imread('autojump.png')
 img = cv2.resize(img, (0, 0), fx=scale, fy=scale)

 img, src_x, src_y = search(img)
 return img


fig = plt.figure()
index = 0

# pull_screenshot()
img = update_data()

update = True 
im = plt.imshow(img, animated=True)


def updatefig(*args):
 global update

 if update:
 time.sleep(1)
 pull_screenshot()
 im.set_array(update_data())
 update = False
 return im,

def onClick(event): 
 global update 
 global src_x, src_y
 
 dst_x, dst_y = event.xdata, event.ydata

 distance = (dst_x - src_x)**2 + (dst_y - src_y)**2 
 distance = (distance ** 0.5) / scale
 print('distance = ', distance)
 jump(distance)
 update = True


fig.canvas.mpl_connect('button_press_event', onClick)
ani = animation.FuncAnimation(fig, updatefig, interval=5, blit=True)
plt.show()

# coding: utf-8
import os
import sys
import subprocess
import shutil
import time
import math
from PIL import Image, ImageDraw
import random
import json
import re


# === 思路 ===
# 核心：每次落穩(wěn)之后截圖，根據(jù)截圖算出棋子的坐標(biāo)和下一個(gè)塊頂面的中點(diǎn)坐標(biāo)，
# 根據(jù)兩個(gè)點(diǎn)的距離乘以一個(gè)時(shí)間系數(shù)獲得長按的時(shí)間
# 識(shí)別棋子：靠棋子的顏色來識(shí)別位置，通過截圖發(fā)現(xiàn)最下面一行大概是一條直線，就從上往下一行一行遍歷，
# 比較顏色（顏色用了一個(gè)區(qū)間來比較）找到最下面的那一行的所有點(diǎn)，然后求個(gè)中點(diǎn)，
# 求好之后再讓 Y 軸坐標(biāo)減小棋子底盤的一半高度從而得到中心點(diǎn)的坐標(biāo)
# 識(shí)別棋盤：靠底色和方塊的色差來做，從分?jǐn)?shù)之下的位置開始，一行一行掃描，由于圓形的塊最頂上是一條線，
# 方形的上面大概是一個(gè)點(diǎn)，所以就用類似識(shí)別棋子的做法多識(shí)別了幾個(gè)點(diǎn)求中點(diǎn)，
# 這時(shí)候得到了塊中點(diǎn)的 X 軸坐標(biāo)，這時(shí)候假設(shè)現(xiàn)在棋子在當(dāng)前塊的中心，
# 根據(jù)一個(gè)通過截圖獲取的固定的角度來推出中點(diǎn)的 Y 坐標(biāo)
# 最后：根據(jù)兩點(diǎn)的坐標(biāo)算距離乘以系數(shù)來獲取長按時(shí)間（似乎可以直接用 X 軸距離）


# TODO: 解決定位偏移的問題
# TODO: 看看兩個(gè)塊中心到中軸距離是否相同，如果是的話靠這個(gè)來判斷一下當(dāng)前超前還是落后，便于矯正
# TODO: 一些固定值根據(jù)截圖的具體大小計(jì)算
# TODO: 直接用 X 軸距離簡化邏輯


def open_accordant_config():
 screen_size = _get_screen_size()
 config_file = "{path}/config/{screen_size}/config.json".format(
 path=sys.path[0],
 screen_size=screen_size
 )
 if os.path.exists(config_file):
 with open(config_file, 'r') as f:
 print("Load config file from {}".format(config_file))
 return json.load(f)
 else:
 with open('{}/config/default.json'.format(sys.path[0]), 'r') as f:
 print("Load default config")
 return json.load(f)


def _get_screen_size():
 size_str = os.popen('adb shell wm size').read()
 if not size_str:
 print('請(qǐng)安裝ADB及驅(qū)動(dòng)并配置環(huán)境變量')
 sys.exit()
 m = re.search('(\d+)x(\d+)', size_str)
 if m:
 width = m.group(1)
 height = m.group(2)
 return "{height}x{width}".format(height=height, width=width)


config = open_accordant_config()

# Magic Number，不設(shè)置可能無法正常執(zhí)行，請(qǐng)根據(jù)具體截圖從上到下按需設(shè)置
under_game_score_y = config['under_game_score_y']
press_coefficient = config['press_coefficient'] # 長按的時(shí)間系數(shù)，請(qǐng)自己根據(jù)實(shí)際情況調(diào)節(jié)
piece_base_height_1_2 = config['piece_base_height_1_2'] # 二分之一的棋子底座高度，可能要調(diào)節(jié)
piece_body_width = config['piece_body_width'] # 棋子的寬度，比截圖中量到的稍微大一點(diǎn)比較安全，可能要調(diào)節(jié)

# 模擬按壓的起始點(diǎn)坐標(biāo)，需要自動(dòng)重復(fù)游戲請(qǐng)?jiān)O(shè)置成“再來一局”的坐標(biāo)
if config.get('swipe'):
 swipe = config['swipe']
else:
 swipe = {}
 #設(shè)置模擬按壓各項(xiàng)參數(shù)，經(jīng)過多臺(tái)手機(jī)測試，其中2160x1080建議調(diào)整參數(shù)為320，1210，720，910
 #使用vivox20，夏普全面屏和小米mix2測試過，均可達(dá)到2000+分?jǐn)?shù)（記得在開發(fā)者設(shè)置打開usb安全驗(yàn)證）
 swipe['x1'], swipe['y1'], swipe['x2'], swipe['y2'] = 320, 410, 320, 410


screenshot_way = 2
screenshot_backup_dir = 'screenshot_backups/'
if not os.path.isdir(screenshot_backup_dir):
 os.mkdir(screenshot_backup_dir)


def pull_screenshot():
 global screenshot_way
 # 新的方法請(qǐng)根據(jù)效率及適用性由高到低排序
 if screenshot_way == 2 or screenshot_way == 1:
 process = subprocess.Popen('adb shell screencap -p', shell=True, stdout=subprocess.PIPE)
 screenshot = process.stdout.read()
 if screenshot_way == 2:
 binary_screenshot = screenshot.replace(b'\r\n', b'\n')
 else:
 binary_screenshot = screenshot.replace(b'\r\r\n', b'\n')
 f = open('autojump.png', 'wb')
 f.write(binary_screenshot)
 f.close()
 elif screenshot_way == 0:
 os.system('adb shell screencap -p /sdcard/autojump.png')
 os.system('adb pull /sdcard/autojump.png .')


def backup_screenshot(ts):
 # 為了方便失敗的時(shí)候 debug
 if not os.path.isdir(screenshot_backup_dir):
 os.mkdir(screenshot_backup_dir)
 shutil.copy('autojump.png', '{}{}.png'.format(screenshot_backup_dir, ts))


def save_debug_creenshot(ts, im, piece_x, piece_y, board_x, board_y):
 draw = ImageDraw.Draw(im)
 # 對(duì)debug圖片加上詳細(xì)的注釋
 draw.line((piece_x, piece_y) + (board_x, board_y), fill=2, width=3)
 draw.line((piece_x, 0, piece_x, im.size[1]), fill=(255, 0, 0))
 draw.line((0, piece_y, im.size[0], piece_y), fill=(255, 0, 0))
 draw.line((board_x, 0, board_x, im.size[1]), fill=(0, 0, 255))
 draw.line((0, board_y, im.size[0], board_y), fill=(0, 0, 255))
 draw.ellipse((piece_x - 10, piece_y - 10, piece_x + 10, piece_y + 10), fill=(255, 0, 0))
 draw.ellipse((board_x - 10, board_y - 10, board_x + 10, board_y + 10), fill=(0, 0, 255))
 del draw
 im.save('{}{}_d.png'.format(screenshot_backup_dir, ts))


def set_button_position(im):
 # 將swipe設(shè)置為 `再來一局` 按鈕的位置
 global swipe_x1, swipe_y1, swipe_x2, swipe_y2
 w, h = im.size
 left = w / 2
 top = int(1584 * (h / 1920.0))
 swipe_x1, swipe_y1, swipe_x2, swipe_y2 = left, top, left, top


def jump(distance):
 press_time = distance * press_coefficient
 press_time = max(press_time, 200) # 設(shè)置 200 ms 是最小的按壓時(shí)間
 press_time = int(press_time)
 cmd = 'adb shell input swipe {x1} {y1} {x2} {y2} {duration}'.format(
 x1=swipe_x1,
 y1=swipe_y1,
 x2=swipe_x2,
 y2=swipe_y2,
 duration=press_time
 )
 print(cmd)
 os.system(cmd)
 return press_time

def find_piece_and_board(im):
 w, h = im.size

 piece_x_sum = 0
 piece_x_c = 0
 piece_y_max = 0
 board_x = 0
 board_y = 0
 scan_x_border = int(w / 8) # 掃描棋子時(shí)的左右邊界
 scan_start_y = 0 # 掃描的起始y坐標(biāo)
 im_pixel=im.load()
 # 以50px步長，嘗試探測scan_start_y
 for i in range(int(h / 3), int( h*2 /3 ), 50):
 last_pixel = im_pixel[0,i]
 for j in range(1, w):
 pixel=im_pixel[j,i]
 # 不是純色的線，則記錄scan_start_y的值，準(zhǔn)備跳出循環(huán)
 if pixel[0] != last_pixel[0] or pixel[1] != last_pixel[1] or pixel[2] != last_pixel[2]:
 scan_start_y = i - 50
 break
 if scan_start_y:
 break
 print('scan_start_y: ', scan_start_y)

 # 從scan_start_y開始往下掃描，棋子應(yīng)位于屏幕上半部分，這里暫定不超過2/3
 for i in range(scan_start_y, int(h * 2 / 3)):
 for j in range(scan_x_border, w - scan_x_border): # 橫坐標(biāo)方面也減少了一部分掃描開銷
 pixel = im_pixel[j,i]
 # 根據(jù)棋子的最低行的顏色判斷，找最后一行那些點(diǎn)的平均值，這個(gè)顏色這樣應(yīng)該 OK，暫時(shí)不提出來
 if (50 < pixel[0] < 60) and (53 < pixel[1] < 63) and (95 < pixel[2] < 110):
 piece_x_sum += j
 piece_x_c += 1
 piece_y_max = max(i, piece_y_max)

 if not all((piece_x_sum, piece_x_c)):
 return 0, 0, 0, 0
 piece_x = int(piece_x_sum / piece_x_c);
 piece_y = piece_y_max - piece_base_height_1_2 # 上移棋子底盤高度的一半

 #限制棋盤掃描的橫坐標(biāo)，避免音符bug
 if piece_x < w/2:
 board_x_start = piece_x
 board_x_end = w
 else:
 board_x_start = 0
 board_x_end = piece_x

 for i in range(int(h / 3), int(h * 2 / 3)):
 last_pixel = im_pixel[0, i]
 if board_x or board_y:
 break
 board_x_sum = 0
 board_x_c = 0

 for j in range(int(board_x_start), int(board_x_end)):
 pixel = im_pixel[j,i]
 # 修掉腦袋比下一個(gè)小格子還高的情況的 bug
 if abs(j - piece_x) < piece_body_width:
 continue

 # 修掉圓頂?shù)臅r(shí)候一條線導(dǎo)致的小 bug，這個(gè)顏色判斷應(yīng)該 OK，暫時(shí)不提出來
 if abs(pixel[0] - last_pixel[0]) + abs(pixel[1] - last_pixel[1]) + abs(pixel[2] - last_pixel[2]) > 10:
 board_x_sum += j
 board_x_c += 1
 if board_x_sum:
 board_x = board_x_sum / board_x_c
 last_pixel=im_pixel[board_x,i]

 #從上頂點(diǎn)往下+274的位置開始向上找顏色與上頂點(diǎn)一樣的點(diǎn)，為下頂點(diǎn)
 #該方法對(duì)所有純色平面和部分非純色平面有效，對(duì)高爾夫草坪面、木紋桌面、藥瓶和非菱形的碟機(jī)（好像是）會(huì)判斷錯(cuò)誤
 for k in range(i+274, i, -1): #274取開局時(shí)最大的方塊的上下頂點(diǎn)距離
 pixel = im_pixel[board_x,k]
 if abs(pixel[0] - last_pixel[0]) + abs(pixel[1] - last_pixel[1]) + abs(pixel[2] - last_pixel[2]) < 10:
 break
 board_y = int((i+k) / 2)

 #如果上一跳命中中間，則下個(gè)目標(biāo)中心會(huì)出現(xiàn)r245 g245 b245的點(diǎn)，利用這個(gè)屬性彌補(bǔ)上一段代碼可能存在的判斷錯(cuò)誤
 #若上一跳由于某種原因沒有跳到正中間，而下一跳恰好有無法正確識(shí)別花紋，則有可能游戲失敗，由于花紋面積通常比較大，失敗概率較低
 for l in range(i, i+200):
 pixel = im_pixel[board_x,l]
 if abs(pixel[0] - 245) + abs(pixel[1] - 245) + abs(pixel[2] - 245) == 0:
 board_y = l+10
 break



 if not all((board_x, board_y)):
 return 0, 0, 0, 0

 return piece_x, piece_y, board_x, board_y

def dump_device_info():
 size_str = os.popen('adb shell wm size').read()
 device_str = os.popen('adb shell getprop ro.product.model').read()
 density_str = os.popen('adb shell wm density').read()
 print("如果你的腳本無法工作，上報(bào)issue時(shí)請(qǐng)copy如下信息:\n**********\
 \nScreen: {size}\nDensity: {dpi}\nDeviceType: {type}\nOS: {os}\nPython: {python}\n**********".format(
 size=size_str.strip(),
 type=device_str.strip(),
 dpi=density_str.strip(),
 os=sys.platform,
 python=sys.version
 ))


def check_screenshot():
 global screenshot_way
 if os.path.isfile('autojump.png'):
 os.remove('autojump.png')
 if (screenshot_way < 0):
 print('暫不支持當(dāng)前設(shè)備')
 sys.exit()
 pull_screenshot()
 try:
 Image.open('./autojump.png').load()
 print('采用方式{}獲取截圖'.format(screenshot_way))
 except:
 screenshot_way -= 1
 check_screenshot()

def main():

 dump_device_info()
 check_screenshot()

 while True:
 pull_screenshot()
 im = Image.open('./autojump.png')
 # 獲取棋子和 board 的位置
 piece_x, piece_y, board_x, board_y = find_piece_and_board(im)
 ts = int(time.time())
 print(ts, piece_x, piece_y, board_x, board_y)
 set_button_position(im)
 jump(math.sqrt((board_x - piece_x) ** 2 + (board_y - piece_y) ** 2))
 save_debug_creenshot(ts, im, piece_x, piece_y, board_x, board_y)
 backup_screenshot(ts)
 time.sleep(1) # 為了保證截圖的時(shí)候應(yīng)落穩(wěn)了，多延遲一會(huì)兒


if __name__ == '__main__':
 main()